| 1 | /* ------------------------------------------------------------------------- |
| 2 | * Project: GSocket (Generic Socket) |
| 3 | * Name: src/msw/gsockmsw.cpp |
| 4 | * Copyright: (c) Guilhem Lavaux |
| 5 | * Licence: wxWindows Licence |
| 6 | * Author: Guillermo Rodriguez Garcia <guille@iies.es> |
| 7 | * Purpose: GSocket GUI-specific MSW code |
| 8 | * CVSID: $Id$ |
| 9 | * ------------------------------------------------------------------------- |
| 10 | */ |
| 11 | |
| 12 | // For compilers that support precompilation, includes "wx.h". |
| 13 | #include "wx/wxprec.h" |
| 14 | |
| 15 | #ifdef __BORLANDC__ |
| 16 | #pragma hdrstop |
| 17 | #endif |
| 18 | |
| 19 | /* |
| 20 | * DONE: for WinCE we need to replace WSAAsyncSelect |
| 21 | * (Windows message-based notification of network events for a socket) |
| 22 | * with another mechanism. |
| 23 | * As WSAAsyncSelect is not present on WinCE, it now uses |
| 24 | * WSACreateEvent, WSAEventSelect, WSAWaitForMultipleEvents and WSAEnumNetworkEvents. |
| 25 | * When enabling eventhandling for a socket a new thread it created that keeps track of the events |
| 26 | * and posts a messageto the hidden window to use the standard message loop. |
| 27 | */ |
| 28 | |
| 29 | /* including rasasync.h (included from windows.h itself included from |
| 30 | * wx/setup.h and/or winsock.h results in this warning for |
| 31 | * RPCNOTIFICATION_ROUTINE |
| 32 | */ |
| 33 | #ifdef _MSC_VER |
| 34 | # pragma warning(disable:4115) /* named type definition in parentheses */ |
| 35 | #endif |
| 36 | |
| 37 | /* This needs to be before the wx/defs/h inclusion |
| 38 | * for some reason |
| 39 | */ |
| 40 | |
| 41 | #ifdef __WXWINCE__ |
| 42 | /* windows.h results in tons of warnings at max warning level */ |
| 43 | # ifdef _MSC_VER |
| 44 | # pragma warning(push, 1) |
| 45 | # endif |
| 46 | # include <windows.h> |
| 47 | # ifdef _MSC_VER |
| 48 | # pragma warning(pop) |
| 49 | # pragma warning(disable:4514) |
| 50 | # endif |
| 51 | #endif |
| 52 | |
| 53 | #ifndef __GSOCKET_STANDALONE__ |
| 54 | # include "wx/platform.h" |
| 55 | #endif |
| 56 | |
| 57 | #if wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) |
| 58 | |
| 59 | #ifndef __GSOCKET_STANDALONE__ |
| 60 | |
| 61 | #include "wx/msw/gsockmsw.h" |
| 62 | #include "wx/gsocket.h" |
| 63 | |
| 64 | extern "C" WXDLLIMPEXP_BASE HINSTANCE wxGetInstance(void); |
| 65 | #define INSTANCE wxGetInstance() |
| 66 | |
| 67 | #else /* __GSOCKET_STANDALONE__ */ |
| 68 | |
| 69 | #include "gsockmsw.h" |
| 70 | #include "gsocket.h" |
| 71 | |
| 72 | /* If not using wxWidgets, a global var called hInst must |
| 73 | * be available and it must contain the app's instance |
| 74 | * handle. |
| 75 | */ |
| 76 | extern HINSTANCE hInst; |
| 77 | #define INSTANCE hInst |
| 78 | |
| 79 | #endif /* !__GSOCKET_STANDALONE__/__GSOCKET_STANDALONE__ */ |
| 80 | |
| 81 | #ifndef __WXWINCE__ |
| 82 | #include <assert.h> |
| 83 | #else |
| 84 | #define assert(x) |
| 85 | #include <winsock.h> |
| 86 | #include "wx/msw/wince/net.h" |
| 87 | #include "wx/hashmap.h" |
| 88 | WX_DECLARE_HASH_MAP(int,bool,wxIntegerHash,wxIntegerEqual,SocketHash); |
| 89 | #endif |
| 90 | |
| 91 | #include <string.h> |
| 92 | #include <stdio.h> |
| 93 | #include <stdlib.h> |
| 94 | #include <stddef.h> |
| 95 | #include <ctype.h> |
| 96 | |
| 97 | #include <winsock.h> |
| 98 | |
| 99 | #ifdef _MSC_VER |
| 100 | # pragma warning(default:4115) /* named type definition in parentheses */ |
| 101 | #endif |
| 102 | |
| 103 | #define CLASSNAME TEXT("_GSocket_Internal_Window_Class") |
| 104 | |
| 105 | /* implemented in utils.cpp */ |
| 106 | extern "C" WXDLLIMPEXP_BASE HWND |
| 107 | wxCreateHiddenWindow(LPCTSTR *pclassname, LPCTSTR classname, WNDPROC wndproc); |
| 108 | |
| 109 | /* Maximum number of different GSocket objects at a given time. |
| 110 | * This value can be modified at will, but it CANNOT be greater |
| 111 | * than (0x7FFF - WM_USER + 1) |
| 112 | */ |
| 113 | #define MAXSOCKETS 1024 |
| 114 | |
| 115 | #if (MAXSOCKETS > (0x7FFF - WM_USER + 1)) |
| 116 | #error "MAXSOCKETS is too big!" |
| 117 | #endif |
| 118 | |
| 119 | #ifndef __WXWINCE__ |
| 120 | typedef int (PASCAL *WSAAsyncSelectFunc)(SOCKET,HWND,u_int,long); |
| 121 | #else |
| 122 | /* Typedef the needed function prototypes and the WSANETWORKEVENTS structure |
| 123 | */ |
| 124 | typedef struct _WSANETWORKEVENTS { |
| 125 | long lNetworkEvents; |
| 126 | int iErrorCode[10]; |
| 127 | } WSANETWORKEVENTS, FAR * LPWSANETWORKEVENTS; |
| 128 | typedef HANDLE (PASCAL *WSACreateEventFunc)(void); |
| 129 | typedef int (PASCAL *WSAEventSelectFunc)(SOCKET,HANDLE,long); |
| 130 | typedef int (PASCAL *WSAWaitForMultipleEventsFunc)(long,HANDLE,BOOL,long,BOOL); |
| 131 | typedef int (PASCAL *WSAEnumNetworkEventsFunc)(SOCKET,HANDLE,LPWSANETWORKEVENTS); |
| 132 | #endif //__WXWINCE__ |
| 133 | |
| 134 | LRESULT CALLBACK _GSocket_Internal_WinProc(HWND, UINT, WPARAM, LPARAM); |
| 135 | |
| 136 | /* Global variables */ |
| 137 | |
| 138 | static HWND hWin; |
| 139 | static CRITICAL_SECTION critical; |
| 140 | static GSocket* socketList[MAXSOCKETS]; |
| 141 | static int firstAvailable; |
| 142 | |
| 143 | #ifndef __WXWINCE__ |
| 144 | static WSAAsyncSelectFunc gs_WSAAsyncSelect = NULL; |
| 145 | #else |
| 146 | static SocketHash socketHash; |
| 147 | static unsigned int currSocket; |
| 148 | HANDLE hThread[MAXSOCKETS]; |
| 149 | static WSACreateEventFunc gs_WSACreateEvent = NULL; |
| 150 | static WSAEventSelectFunc gs_WSAEventSelect = NULL; |
| 151 | static WSAWaitForMultipleEventsFunc gs_WSAWaitForMultipleEvents = NULL; |
| 152 | static WSAEnumNetworkEventsFunc gs_WSAEnumNetworkEvents = NULL; |
| 153 | /* This structure will be used to pass data on to the thread that handles socket events. |
| 154 | */ |
| 155 | typedef struct thread_data{ |
| 156 | HWND hEvtWin; |
| 157 | unsigned long msgnumber; |
| 158 | unsigned long fd; |
| 159 | unsigned long lEvent; |
| 160 | }thread_data; |
| 161 | #endif |
| 162 | |
| 163 | static HMODULE gs_wsock32dll = 0; |
| 164 | |
| 165 | |
| 166 | #ifdef __WXWINCE__ |
| 167 | /* This thread handles socket events on WinCE using WSAEventSelect() as WSAAsyncSelect is not supported. |
| 168 | * When an event occures for the socket, it is checked what kind of event happend and the correct message gets posted |
| 169 | * so that the hidden window can handle it as it would in other MSW builds. |
| 170 | */ |
| 171 | DWORD WINAPI SocketThread(LPVOID data) |
| 172 | { |
| 173 | WSANETWORKEVENTS NetworkEvents; |
| 174 | thread_data* d = (thread_data *)data; |
| 175 | |
| 176 | HANDLE NetworkEvent = gs_WSACreateEvent(); |
| 177 | gs_WSAEventSelect(d->fd, NetworkEvent, d->lEvent); |
| 178 | |
| 179 | while(socketHash[d->fd] == true) |
| 180 | { |
| 181 | if ((gs_WSAWaitForMultipleEvents(1, &NetworkEvent, FALSE,INFINITE, FALSE)) == WAIT_FAILED) |
| 182 | { |
| 183 | printf("WSAWaitForMultipleEvents failed with error %d\n", WSAGetLastError()); |
| 184 | return 0; |
| 185 | } |
| 186 | if (gs_WSAEnumNetworkEvents(d->fd ,NetworkEvent, &NetworkEvents) == SOCKET_ERROR) |
| 187 | { |
| 188 | printf("WSAEnumNetworkEvents failed with error %d\n", WSAGetLastError()); |
| 189 | return 0; |
| 190 | } |
| 191 | |
| 192 | long flags = NetworkEvents.lNetworkEvents; |
| 193 | if (flags & FD_READ) |
| 194 | ::PostMessage(d->hEvtWin, d->msgnumber,d->fd, FD_READ); |
| 195 | if (flags & FD_WRITE) |
| 196 | ::PostMessage(d->hEvtWin, d->msgnumber,d->fd, FD_WRITE); |
| 197 | if (flags & FD_OOB) |
| 198 | ::PostMessage(d->hEvtWin, d->msgnumber,d->fd, FD_OOB); |
| 199 | if (flags & FD_ACCEPT) |
| 200 | ::PostMessage(d->hEvtWin, d->msgnumber,d->fd, FD_ACCEPT); |
| 201 | if (flags & FD_CONNECT) |
| 202 | ::PostMessage(d->hEvtWin, d->msgnumber,d->fd, FD_CONNECT); |
| 203 | if (flags & FD_CLOSE) |
| 204 | ::PostMessage(d->hEvtWin, d->msgnumber,d->fd, FD_CLOSE); |
| 205 | |
| 206 | } |
| 207 | gs_WSAEventSelect(d->fd, NetworkEvent, 0); |
| 208 | ExitThread(0); |
| 209 | return 0; |
| 210 | } |
| 211 | #endif |
| 212 | |
| 213 | |
| 214 | bool GSocketGUIFunctionsTableConcrete::CanUseEventLoop() |
| 215 | { |
| 216 | return true; |
| 217 | } |
| 218 | |
| 219 | /* Global initializers */ |
| 220 | |
| 221 | bool GSocketGUIFunctionsTableConcrete::OnInit() |
| 222 | { |
| 223 | static LPCTSTR pclassname = NULL; |
| 224 | int i; |
| 225 | |
| 226 | /* Create internal window for event notifications */ |
| 227 | hWin = wxCreateHiddenWindow(&pclassname, CLASSNAME, _GSocket_Internal_WinProc); |
| 228 | if (!hWin) |
| 229 | return false; |
| 230 | |
| 231 | /* Initialize socket list */ |
| 232 | InitializeCriticalSection(&critical); |
| 233 | |
| 234 | for (i = 0; i < MAXSOCKETS; i++) |
| 235 | { |
| 236 | socketList[i] = NULL; |
| 237 | } |
| 238 | firstAvailable = 0; |
| 239 | |
| 240 | /* Load WSAAsyncSelect from wsock32.dll (we don't link against it |
| 241 | statically to avoid dependency on wsock32.dll for apps that don't use |
| 242 | sockets): */ |
| 243 | #ifndef __WXWINCE__ |
| 244 | gs_wsock32dll = LoadLibrary(wxT("wsock32.dll")); |
| 245 | if (!gs_wsock32dll) |
| 246 | return false; |
| 247 | gs_WSAAsyncSelect =(WSAAsyncSelectFunc)GetProcAddress(gs_wsock32dll, |
| 248 | "WSAAsyncSelect"); |
| 249 | if (!gs_WSAAsyncSelect) |
| 250 | return false; |
| 251 | #else |
| 252 | /* On WinCE we load ws2.dll which will provide the needed functions. |
| 253 | */ |
| 254 | gs_wsock32dll = LoadLibrary(wxT("ws2.dll")); |
| 255 | if (!gs_wsock32dll) |
| 256 | return false; |
| 257 | gs_WSAEventSelect =(WSAEventSelectFunc)GetProcAddress(gs_wsock32dll, |
| 258 | wxT("WSAEventSelect")); |
| 259 | if (!gs_WSAEventSelect) |
| 260 | return false; |
| 261 | |
| 262 | gs_WSACreateEvent =(WSACreateEventFunc)GetProcAddress(gs_wsock32dll, |
| 263 | wxT("WSACreateEvent")); |
| 264 | if (!gs_WSACreateEvent) |
| 265 | return false; |
| 266 | |
| 267 | gs_WSAWaitForMultipleEvents =(WSAWaitForMultipleEventsFunc)GetProcAddress(gs_wsock32dll, |
| 268 | wxT("WSAWaitForMultipleEvents")); |
| 269 | if (!gs_WSAWaitForMultipleEvents) |
| 270 | return false; |
| 271 | |
| 272 | gs_WSAEnumNetworkEvents =(WSAEnumNetworkEventsFunc)GetProcAddress(gs_wsock32dll, |
| 273 | wxT("WSAEnumNetworkEvents")); |
| 274 | if (!gs_WSAEnumNetworkEvents) |
| 275 | return false; |
| 276 | |
| 277 | currSocket = 0; |
| 278 | #endif |
| 279 | |
| 280 | return true; |
| 281 | } |
| 282 | |
| 283 | void GSocketGUIFunctionsTableConcrete::OnExit() |
| 284 | { |
| 285 | #ifdef __WXWINCE__ |
| 286 | /* Delete the threads here */ |
| 287 | for(unsigned int i=0; i < currSocket; i++) |
| 288 | CloseHandle(hThread[i]); |
| 289 | #endif |
| 290 | /* Destroy internal window */ |
| 291 | DestroyWindow(hWin); |
| 292 | UnregisterClass(CLASSNAME, INSTANCE); |
| 293 | |
| 294 | /* Unlock wsock32.dll */ |
| 295 | if (gs_wsock32dll) |
| 296 | { |
| 297 | FreeLibrary(gs_wsock32dll); |
| 298 | gs_wsock32dll = 0; |
| 299 | } |
| 300 | |
| 301 | /* Delete critical section */ |
| 302 | DeleteCriticalSection(&critical); |
| 303 | } |
| 304 | |
| 305 | /* Per-socket GUI initialization / cleanup */ |
| 306 | |
| 307 | bool GSocketGUIFunctionsTableConcrete::Init_Socket(GSocket *socket) |
| 308 | { |
| 309 | int i; |
| 310 | |
| 311 | /* Allocate a new message number for this socket */ |
| 312 | EnterCriticalSection(&critical); |
| 313 | |
| 314 | i = firstAvailable; |
| 315 | while (socketList[i] != NULL) |
| 316 | { |
| 317 | i = (i + 1) % MAXSOCKETS; |
| 318 | |
| 319 | if (i == firstAvailable) /* abort! */ |
| 320 | { |
| 321 | LeaveCriticalSection(&critical); |
| 322 | return false; |
| 323 | } |
| 324 | } |
| 325 | socketList[i] = socket; |
| 326 | firstAvailable = (i + 1) % MAXSOCKETS; |
| 327 | socket->m_msgnumber = (i + WM_USER); |
| 328 | |
| 329 | LeaveCriticalSection(&critical); |
| 330 | |
| 331 | return true; |
| 332 | } |
| 333 | |
| 334 | void GSocketGUIFunctionsTableConcrete::Destroy_Socket(GSocket *socket) |
| 335 | { |
| 336 | /* Remove the socket from the list */ |
| 337 | EnterCriticalSection(&critical); |
| 338 | if ( socket->IsOk() ) |
| 339 | socketList[(socket->m_msgnumber - WM_USER)] = NULL; |
| 340 | LeaveCriticalSection(&critical); |
| 341 | } |
| 342 | |
| 343 | /* Windows proc for asynchronous event handling */ |
| 344 | |
| 345 | LRESULT CALLBACK _GSocket_Internal_WinProc(HWND hWnd, |
| 346 | UINT uMsg, |
| 347 | WPARAM wParam, |
| 348 | LPARAM lParam) |
| 349 | { |
| 350 | GSocket *socket; |
| 351 | GSocketEvent event; |
| 352 | GSocketCallback cback; |
| 353 | char *data; |
| 354 | |
| 355 | if (uMsg >= WM_USER && uMsg <= (WM_USER + MAXSOCKETS - 1)) |
| 356 | { |
| 357 | EnterCriticalSection(&critical); |
| 358 | socket = socketList[(uMsg - WM_USER)]; |
| 359 | event = (GSocketEvent) -1; |
| 360 | cback = NULL; |
| 361 | data = NULL; |
| 362 | |
| 363 | /* Check that the socket still exists (it has not been |
| 364 | * destroyed) and for safety, check that the m_fd field |
| 365 | * is what we expect it to be. |
| 366 | */ |
| 367 | if ((socket != NULL) && (socket->m_fd == wParam)) |
| 368 | { |
| 369 | switch WSAGETSELECTEVENT(lParam) |
| 370 | { |
| 371 | case FD_READ: event = GSOCK_INPUT; break; |
| 372 | case FD_WRITE: event = GSOCK_OUTPUT; break; |
| 373 | case FD_ACCEPT: event = GSOCK_CONNECTION; break; |
| 374 | case FD_CONNECT: |
| 375 | { |
| 376 | if (WSAGETSELECTERROR(lParam) != 0) |
| 377 | event = GSOCK_LOST; |
| 378 | else |
| 379 | event = GSOCK_CONNECTION; |
| 380 | break; |
| 381 | } |
| 382 | case FD_CLOSE: event = GSOCK_LOST; break; |
| 383 | } |
| 384 | |
| 385 | if (event != -1) |
| 386 | { |
| 387 | cback = socket->m_cbacks[event]; |
| 388 | data = socket->m_data[event]; |
| 389 | |
| 390 | if (event == GSOCK_LOST) |
| 391 | socket->m_detected = GSOCK_LOST_FLAG; |
| 392 | else |
| 393 | socket->m_detected |= (1 << event); |
| 394 | } |
| 395 | } |
| 396 | |
| 397 | /* OK, we can now leave the critical section because we have |
| 398 | * already obtained the callback address (we make no further |
| 399 | * accesses to socket->whatever). However, the app should |
| 400 | * be prepared to handle events from a socket that has just |
| 401 | * been closed! |
| 402 | */ |
| 403 | LeaveCriticalSection(&critical); |
| 404 | |
| 405 | if (cback != NULL) |
| 406 | (cback)(socket, event, data); |
| 407 | |
| 408 | return (LRESULT) 0; |
| 409 | } |
| 410 | else |
| 411 | return DefWindowProc(hWnd, uMsg, wParam, lParam); |
| 412 | } |
| 413 | |
| 414 | /* _GSocket_Enable_Events: |
| 415 | * Enable all event notifications; we need to be notified of all |
| 416 | * events for internal processing, but we will only notify users |
| 417 | * when an appropiate callback function has been installed. |
| 418 | */ |
| 419 | void GSocketGUIFunctionsTableConcrete::Enable_Events(GSocket *socket) |
| 420 | { |
| 421 | assert (socket != NULL); |
| 422 | |
| 423 | if (socket->m_fd != INVALID_SOCKET) |
| 424 | { |
| 425 | /* We could probably just subscribe to all events regardless |
| 426 | * of the socket type, but MS recommends to do it this way. |
| 427 | */ |
| 428 | long lEvent = socket->m_server? |
| 429 | FD_ACCEPT : (FD_READ | FD_WRITE | FD_CONNECT | FD_CLOSE); |
| 430 | #ifndef __WXWINCE__ |
| 431 | gs_WSAAsyncSelect(socket->m_fd, hWin, socket->m_msgnumber, lEvent); |
| 432 | #else |
| 433 | /* |
| 434 | * WinCE creates a thread for socket event handling. |
| 435 | * All needed parameters get passed through the thread_data structure. |
| 436 | */ |
| 437 | |
| 438 | thread_data* d = new thread_data; |
| 439 | d->lEvent = lEvent; |
| 440 | d->hEvtWin = hWin; |
| 441 | d->msgnumber = socket->m_msgnumber; |
| 442 | d->fd = socket->m_fd; |
| 443 | socketHash[socket->m_fd] = true; |
| 444 | hThread[currSocket++] = CreateThread(NULL, 0, &SocketThread,(LPVOID)d, 0, NULL); |
| 445 | #endif |
| 446 | } |
| 447 | } |
| 448 | |
| 449 | /* _GSocket_Disable_Events: |
| 450 | * Disable event notifications (when shutdowning the socket) |
| 451 | */ |
| 452 | void GSocketGUIFunctionsTableConcrete::Disable_Events(GSocket *socket) |
| 453 | { |
| 454 | assert (socket != NULL); |
| 455 | |
| 456 | if (socket->m_fd != INVALID_SOCKET) |
| 457 | { |
| 458 | #ifndef __WXWINCE__ |
| 459 | gs_WSAAsyncSelect(socket->m_fd, hWin, socket->m_msgnumber, 0); |
| 460 | #else |
| 461 | //Destroy the thread |
| 462 | socketHash[socket->m_fd] = false; |
| 463 | #endif |
| 464 | } |
| 465 | } |
| 466 | |
| 467 | #else /* !wxUSE_SOCKETS */ |
| 468 | |
| 469 | /* |
| 470 | * Translation unit shouldn't be empty, so include this typedef to make the |
| 471 | * compiler (VC++ 6.0, for example) happy |
| 472 | */ |
| 473 | typedef void (*wxDummy)(); |
| 474 | |
| 475 | #endif /* wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) */ |