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