trunk/Src/Win32/Sockets/SocketInput.cpp

   1 // SocketInput.cpp
   2
   3 // (C) Copyright Seb Wills 2005
   4
   5 //
   6
   7 // Win32-specific socket input class
   8
   9
  10
  11 #include "..\Common\WinCommon.h"
  12
  13
  14
  15 #include "./SocketInput.h"
  16
  17 #include "../../Win32/resource.h" // Probably shouldn't be including a file outside DasherCore. This is needed purely for the resource strings (IDS_...) used in the error message titles
  18
  19
  20
  21 #include <process.h>
  22
  23 #include <atlstr.h>
  24
  25 #include <stdlib.h>
  26
  27 #include <stddef.h>
  28
  29
  30
  31 using namespace Dasher;
  32
  33
  34
  35 // Track memory leaks on Windows to the line that new'd the memory
  36
  37 #ifdef _WIN32
  38
  39 #ifdef _DEBUG
  40
  41 #define DEBUG_NEW new( _NORMAL_BLOCK, THIS_FILE, __LINE__ )
  42
  43 #define new DEBUG_NEW
  44
  45 #undef THIS_FILE
  46
  47 static char THIS_FILE[] = __FILE__;
  48
  49 #endif
  50
  51 #endif
  52
  53
  54
  55 CSocketInput::CSocketInput(CEventHandler * pEventHandler, CSettingsStore * pSettingsStore)
  56
  57 :CSocketInputBase(pEventHandler, pSettingsStore) {
  58
  59
  60
  61   allocatedConsole = false;
  62
  63
  64
  65   // Although SetDebug is called in the constructore for CSocketInputBase, that invokes CSocketInputBase::SetDebug,
  66
  67   // and we need to invoke CSocketInput::SetDebug to bring up the debug messages window if debugging is enabled.
  68
  69   SetDebug(GetBoolParameter(BP_SOCKET_DEBUG));
  70
  71
  72
  73   // Windows-specific initialisation of Winsock:
  74
  75   WORD wVersionRequested;
  76
  77   WSADATA wsaData;
  78
  79   int err;
  80
  81   wVersionRequested = MAKEWORD( 2, 2 );
  82
  83   err = WSAStartup( wVersionRequested, &wsaData );
  84
  85   if ( err != 0 ) {
  86
  87     Tstring ErrTitle;
  88
  89     WinLocalisation::GetResourceString(IDS_ERR_SOCKET_TITLE, &ErrTitle);
  90
  91     MessageBox(NULL, _TEXT("Failed to initialize Winsock DLL"),  ErrTitle.c_str(), MB_OK | MB_ICONEXCLAMATION);
  92
  93   }
  94
  95 }
  96
  97
  98
  99 CSocketInput::~CSocketInput() {
 100
 101   StopListening();
 102
 103 }
 104
 105
 106
 107
 108
 109 bool CSocketInput::LaunchReaderThread() {
 110
 111
 112
 113   unsigned threadId;
 114
 115   m_readerThreadHandle = (HANDLE) _beginthreadex(NULL, // security attributes
 116
 117                  0,    // stack size (0 = use default)
 118
 119                  ThreadLauncherStub, // function to invoke
 120
 121                  this, // argument
 122
 123                  0,    // initflag
 124
 125                  &threadId); // receives thread ID
 126
 127   if(m_readerThreadHandle == 0) {
 128
 129     ReportErrnoError("Error creating reader thread");
 130
 131     return false;
 132
 133   }
 134
 135   return true;
 136
 137 }
 138
 139
 140
 141 void CSocketInput::CancelReaderThread() {
 142
 143   // FIXME: this is ugly; should have a way of cleanly terminating the thread
 144
 145   // The thread will need to use select to check the socket input buffer
 146
 147   // (rather than just calling recv which blocks indefinitely) and monitor a pleaseStop
 148
 149   // variable or something. But this seems to work OK...
 150
 151   TerminateThread(m_readerThreadHandle, 1);
 152
 153 }
 154
 155
 156
 157 void CSocketInput::ReportError(std::string s) {
 158
 159   CString msg(s.c_str()); //FIXME: check what goes on here with encodings. Is s in UTF-8? Do we need to call mbstowcs or something on it?
 160
 161   Tstring ErrTitle;
 162
 163   WinLocalisation::GetResourceString(IDS_ERR_SOCKET_TITLE, &ErrTitle);
 164
 165   MessageBox(NULL, msg, ErrTitle.c_str(), MB_OK | MB_ICONEXCLAMATION);
 166
 167 }
 168
 169
 170
 171 void CSocketInput::ReportErrnoError(std::string prefix) {
 172
 173   CString msg(prefix.c_str());
 174
 175   //wchar_t *tmp = new wchar_t[mbstowcs(NULL, (const char *) prefix.c_str(), 1024)];
 176
 177
 178
 179   msg += _TEXT(": Errno reports '");
 180
 181   msg += _tcserror(errno); // retrieve localised error message for the most recent socket error
 182
 183   CString msg2;
 184
 185   msg2.Format(_TEXT("'. WSAGetLastError returns %d."), WSAGetLastError());
 186
 187   msg += msg2;
 188
 189
 190
 191   Tstring ErrTitle;
 192
 193   WinLocalisation::GetResourceString(IDS_ERR_SOCKET_TITLE, &ErrTitle);
 194
 195   MessageBox(NULL, msg, ErrTitle.c_str(), MB_OK | MB_ICONEXCLAMATION);
 196
 197 }
 198
 199
 200
 201 void CSocketInput::SetDebug(bool _debug) {
 202   if(_debug && !allocatedConsole) {
 203     AllocConsole();
 204     freopen("conout$", "w", stderr); // makes stderr go to the new console window (but not std::cerr, it seems!)
 205     allocatedConsole = true;
 206   }
 207   CSocketInputBase::SetDebug(_debug);
 208 }