source: trunk/src/wsock32/asyncapi.cpp@ 7140

Last change on this file since 7140 was 7140, checked in by achimha, 24 years ago

implemented async select event error code notification for semaphore notifications
File size: 25.0 KB
Line 
1/* $Id: asyncapi.cpp,v 1.17 2001-10-19 18:34:47 achimha Exp $ */
2
3/*
4 *
5 * Win32 SOCK32 for OS/2 (Async apis)
6 *
7 * Copyright (C) 2000 Sander van Leeuwen (sandervl@xs4all.nl)
8 *
9 * Parts based on Wine code: (dlls\winsock\async.c)
10 * (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka.
11 * (C) 1999 Marcus Meissner
12 *
13 * Project Odin Software License can be found in LICENSE.TXT
14 *
15 */
16#define INCL_BASE
17#include <os2wrap.h>
18#include <stdio.h>
19#include <stdlib.h>
20#include <string.h>
21#include <odin.h>
22#include <odinwrap.h>
23#include <os2sel.h>
24#include <misc.h>
25#include <win32api.h>
26#include "wsock32.h"
27#include "asyncthread.h"
28#include "wsastruct.h"
29
30#define DBG_LOCALLOG DBG_async
31#include "dbglocal.h"
32
33//kso: dirty fix to make this compile! not permanent!
34BOOL WINAPI QueryPerformanceCounter(LARGE_INTEGER *p);
35#define LowPart u.LowPart
36
37
38ODINDEBUGCHANNEL(WSOCK32-ASYNC)
39
40//******************************************************************************
41//NOTE: Must execute RemoveFromQueue before returning
42//******************************************************************************
43void ASYNCCNV WSAsyncThreadProc(void *pparm)
44{
45 PASYNCTHREADPARM pThreadParm = (PASYNCTHREADPARM)pparm;
46 LPARAM lParam;
47 int size = 0;
48 int fail = NO_ERROR;
49
50 if(!pThreadParm->fCancelled)
51 {
52 switch(pThreadParm->request)
53 {
54 case ASYNC_GETHOSTBYNAME:
55 case ASYNC_GETHOSTBYADDR:
56 {
57 struct hostent *ret;
58
59 if(pThreadParm->request == ASYNC_GETHOSTBYNAME) {
60 ret = gethostbyname(pThreadParm->u.gethostbyname.name);
61
62 free(pThreadParm->u.gethostbyname.name);
63 pThreadParm->u.gethostbyname.name = 0;
64 }
65 else {
66 ret = gethostbyaddr(pThreadParm->u.gethostbyaddr.addr,
67 pThreadParm->u.gethostbyaddr.len,
68 pThreadParm->u.gethostbyaddr.type);
69
70 free(pThreadParm->u.gethostbyaddr.addr);
71 pThreadParm->u.gethostbyaddr.addr = 0;
72 }
73 if (ret != NULL) {
74 size = WS_copy_he((struct ws_hostent *)pThreadParm->buf,
75 pThreadParm->buf,
76 pThreadParm->buflen,
77 ret);
78 if(size < 0) {
79 fail = WSAENOBUFS;
80 size = -size;
81 }
82 }
83 else fail = wsaHerrno();
84 break;
85 }
86
87 case ASYNC_GETPROTOBYNAME:
88 case ASYNC_GETPROTOBYNUMBER:
89 {
90 struct protoent *ret;
91
92 if(pThreadParm->request == ASYNC_GETPROTOBYNAME) {
93 ret = getprotobyname(pThreadParm->u.getprotobyname.name);
94
95 free(pThreadParm->u.getprotobyname.name);
96 pThreadParm->u.getprotobyname.name = 0;
97 }
98 else ret = getprotobynumber(pThreadParm->u.getprotobynumber.number);
99
100 if (ret != NULL) {
101 size = WS_copy_pe((struct ws_protoent *)pThreadParm->buf,
102 pThreadParm->buf,
103 pThreadParm->buflen,
104 ret);
105 if(size < 0) {
106 fail = WSAENOBUFS;
107 size = -size;
108 }
109 }
110 else fail = WSANO_DATA;
111
112 break;
113 }
114
115 case ASYNC_GETSERVBYNAME:
116 case ASYNC_GETSERVBYPORT:
117 {
118 struct servent *ret;
119
120 if(pThreadParm->request == ASYNC_GETSERVBYNAME) {
121 ret = getservbyname(pThreadParm->u.getservbyname.name, pThreadParm->u.getservbyname.proto);
122
123 free(pThreadParm->u.getservbyname.name);
124 pThreadParm->u.getservbyname.name = 0;
125
126 if(pThreadParm->u.getservbyname.proto) {
127 free(pThreadParm->u.getservbyname.proto);
128 pThreadParm->u.getservbyname.proto = 0;
129 }
130
131 }
132 else {
133 ret = getservbyport(pThreadParm->u.getservbyport.port, pThreadParm->u.getservbyport.proto);
134 if(pThreadParm->u.getservbyport.proto) {
135 free(pThreadParm->u.getservbyport.proto);
136 pThreadParm->u.getservbyport.proto = 0;
137 }
138 }
139
140 if (ret != NULL) {
141 size = WS_copy_se((struct ws_servent *)pThreadParm->buf,
142 pThreadParm->buf,
143 pThreadParm->buflen,
144 ret);
145 if(size < 0) {
146 fail = WSAENOBUFS;
147 size = -size;
148 }
149 }
150 else fail = WSANO_DATA;
151 break;
152 }
153 }
154 }
155 lParam = (fail << 16) | size;
156
157 if(!pThreadParm->fCancelled) {
158 dprintf(("WSAsyncThreadProc %x %x %x %x", pThreadParm->notifyHandle, pThreadParm->notifyData, pThreadParm->hAsyncTaskHandle, lParam));
159 PostMessageA((HWND)pThreadParm->notifyHandle, (DWORD)pThreadParm->notifyData,
160 (WPARAM)pThreadParm->hAsyncTaskHandle, lParam);
161 }
162 pThreadParm->fActive = FALSE;
163 RemoveFromQueue(pThreadParm);
164}
165//******************************************************************************
166//******************************************************************************
167LHANDLE WSAAsyncRequest(AsyncRequestType requesttype, HWND hwnd, int msg, char *buf,
168 int buflen, PVOID param1, PVOID param2, PVOID param3,
169 PVOID param4)
170{
171 PASYNCTHREADPARM pThreadParm;
172 LHANDLE hAsyncRequest = 0;
173 LPSTR tempname = 0;
174 LPSTR tempproto = 0;
175 LPSTR tempaddr = 0;
176
177 if(!fWSAInitialized)
178 {
179 WSASetLastError(WSANOTINITIALISED);
180 }
181 else
182 if(WSAIsBlocking())
183 {
184 WSASetLastError(WSAEINPROGRESS); // blocking call in progress
185 }
186 else
187 if(!IsWindow(hwnd))
188 {
189 WSASetLastError(WSAEINVAL); // invalid parameter
190 }
191 else
192 if(buf == NULL || buflen == 0) {
193 WSASetLastError(WSAENOBUFS); // invalid parameter
194 }
195 else
196 {
197 pThreadParm = (PASYNCTHREADPARM)malloc(sizeof(ASYNCTHREADPARM));
198 if(pThreadParm == NULL) {
199 dprintf(("WSAAsyncRequest: malloc failure!"));
200 DebugInt3();
201 WSASetLastError(WSAEFAULT);
202 return 0;
203 }
204 memset(pThreadParm, 0, sizeof(*pThreadParm));
205 pThreadParm->request = requesttype;
206 pThreadParm->notifyHandle = (int)hwnd;
207 pThreadParm->notifyData = (int)msg;
208 pThreadParm->buf = buf;
209 pThreadParm->buflen = buflen;
210
211 switch(requesttype) {
212 case ASYNC_GETHOSTBYNAME:
213 tempname = (LPSTR)malloc(strlen((char *)param1)+1);
214 if(tempname == NULL)
215 {
216 dprintf(("WSAAsyncGetHostByName: malloc failure!"));
217 DebugInt3();
218 WSASetLastError(WSAEFAULT);
219 return 0;
220 }
221 pThreadParm->u.gethostbyname.name = tempname;
222 strcpy(pThreadParm->u.gethostbyname.name, (char *)param1);
223 break;
224
225 case ASYNC_GETHOSTBYADDR:
226 tempaddr = (LPSTR)malloc(strlen((char *)param1)+1);
227 if(tempaddr == NULL)
228 {
229 dprintf(("WSAAsyncGetHostByAddr: malloc failure!"));
230 DebugInt3();
231 WSASetLastError(WSAEFAULT);
232 return 0;
233 }
234
235 pThreadParm->u.gethostbyaddr.len = (int)param2;
236 pThreadParm->u.gethostbyaddr.type = (int)param3;
237 pThreadParm->u.gethostbyaddr.addr = tempaddr;
238 strcpy(pThreadParm->u.gethostbyaddr.addr, (char *)param1);
239 break;
240
241 case ASYNC_GETPROTOBYNAME:
242 tempname = (LPSTR)malloc(strlen((char *)param1)+1);
243 if(tempname == NULL)
244 {
245 dprintf(("WSAAsyncGetProtoByName: malloc failure!"));
246 DebugInt3();
247 WSASetLastError(WSAEFAULT);
248 return 0;
249 }
250 pThreadParm->u.getprotobyname.name = tempname;
251 strcpy(pThreadParm->u.getprotobyname.name, (char *)param1);
252 break;
253
254 case ASYNC_GETPROTOBYNUMBER:
255 pThreadParm->u.getprotobynumber.number = (int)param1;
256 break;
257
258 case ASYNC_GETSERVBYNAME:
259 tempname = (LPSTR)malloc(strlen((char *)param1)+1);
260 tempproto = NULL;
261 if(param2) {
262 tempproto = (LPSTR)malloc(strlen((char *)param2)+1);
263 }
264 if(tempname == NULL || (param2 && !tempproto))
265 {
266 dprintf(("WSAAsyncGetServByName: malloc failure!"));
267 DebugInt3();
268 WSASetLastError(WSAEFAULT);
269 return 0;
270 }
271 pThreadParm->u.getservbyname.name = tempname;
272 strcpy(pThreadParm->u.getservbyname.name, (char *)param1);
273 pThreadParm->u.getservbyname.proto = tempproto;
274 if(param2)
275 strcpy(pThreadParm->u.getservbyname.proto, (char *)param2);
276 break;
277
278 case ASYNC_GETSERVBYPORT:
279 if(param2) {
280 tempproto = (LPSTR)malloc(strlen((char *)param2)+1);
281 }
282 if(param2 && !tempproto)
283 {
284 dprintf(("WSAAsyncGetServByPort: malloc failure!"));
285 DebugInt3();
286 WSASetLastError(WSAEFAULT);
287 return 0;
288 }
289 pThreadParm->u.getservbyport.port = (int)param1;
290 pThreadParm->u.getservbyport.proto = tempproto;
291 if(param2)
292 strcpy(pThreadParm->u.getservbyport.proto, (char *)param2);
293 break;
294 }
295 hAsyncRequest = (LHANDLE)QueueAsyncJob(WSAsyncThreadProc, pThreadParm);
296 if(hAsyncRequest == 0) {
297 free(pThreadParm);
298 if(tempname) free(tempname);
299 if(tempaddr) free(tempaddr);
300 if(tempproto) free(tempproto);
301
302 dprintf(("WSAAsyncRequest: QueueAsyncJob failure!"));
303 DebugInt3();
304 WSASetLastError(WSAEFAULT);
305 return 0;
306 }
307 return hAsyncRequest;
308 }
309 return 0;
310}
311//******************************************************************************
312//******************************************************************************
313ODINFUNCTION5(LHANDLE,WSAAsyncGetHostByName,
314 HWND,hWnd,
315 u_int,wMsg,
316 const char *,name,
317 char *,buf,
318 int,buflen)
319{
320 dprintf(("WSAAsyncGetHostByName %s", name));
321
322 return WSAAsyncRequest(ASYNC_GETHOSTBYNAME, hWnd, wMsg, buf, buflen,
323 (PVOID)name, 0, 0, 0);
324}
325//******************************************************************************
326//******************************************************************************
327ODINFUNCTION7(LHANDLE,WSAAsyncGetHostByAddr,
328 HWND,hWnd,
329 u_int,wMsg,
330 const char *,addr,
331 int,len,
332 int,type,
333 char *,buf,
334 int,buflen)
335{
336 dprintf(("WSAAsyncGetHostByAddr %s", addr));
337
338 return WSAAsyncRequest(ASYNC_GETHOSTBYADDR, hWnd, wMsg, buf, buflen,
339 (PVOID)addr, (PVOID)len, (PVOID)type, 0);
340}
341//******************************************************************************
342//******************************************************************************
343ODINFUNCTION5(LHANDLE,WSAAsyncGetProtoByName,
344 HWND,hWnd,
345 u_int,wMsg,
346 const char *,name,
347 char *,buf,
348 int,buflen)
349{
350 dprintf(("WSAAsyncGetProtoByName %s", name));
351
352 return WSAAsyncRequest(ASYNC_GETPROTOBYNAME, hWnd, wMsg, buf, buflen,
353 (PVOID)name, 0, 0, 0);
354}
355//******************************************************************************
356//******************************************************************************
357ODINFUNCTION5(LHANDLE,WSAAsyncGetProtoByNumber,
358 HWND,hWnd,
359 u_int,wMsg,
360 int,number,
361 char *,buf,
362 int,buflen)
363{
364 dprintf(("WSAAsyncGetProtoByNumber %d", number));
365
366 return WSAAsyncRequest(ASYNC_GETPROTOBYNUMBER, hWnd, wMsg, buf, buflen,
367 (PVOID)number, 0, 0, 0);
368}
369//******************************************************************************
370//******************************************************************************
371ODINFUNCTION6(LHANDLE,WSAAsyncGetServByName,
372 HWND,hWnd,
373 u_int,wMsg,
374 const char *,name,
375 const char *,proto,
376 char *,buf,
377 int,buflen)
378{
379 dprintf(("WSAAsyncGetServByName %s", name));
380
381 return WSAAsyncRequest(ASYNC_GETSERVBYNAME, hWnd, wMsg, buf, buflen,
382 (PVOID)name, (PVOID)proto, 0, 0);
383}
384//******************************************************************************
385//******************************************************************************
386ODINFUNCTION6(LHANDLE,WSAAsyncGetServByPort,
387 HWND,hWnd,
388 u_int,wMsg,
389 int,port,
390 const char *,proto,
391 char *,buf,
392 int,buflen)
393{
394 dprintf(("WSAAsyncGetServByPort %d %s", port, proto));
395
396 return WSAAsyncRequest(ASYNC_GETSERVBYPORT, hWnd, wMsg, buf, buflen,
397 (PVOID)port, (PVOID)proto, 0, 0);
398}
399//******************************************************************************
400//******************************************************************************
401void AsyncSelectNotifyEvent(PASYNCTHREADPARM pThreadParm, ULONG event, ULONG socket_error)
402{
403 pThreadParm->u.asyncselect.lEventsPending &= ~event;
404
405 // @@@AH 20011019 this changes the event code. Don' think we can do this
406 // for the semaphore notification...
407 ULONG eventReply = WSAMAKESELECTREPLY(event, socket_error);
408
409 if (pThreadParm->u.asyncselect.mode == WSA_SELECT_HWND)
410 {
411 dprintf(("AsyncSelectNotifyEvent: notifying window, socket: 0x%x, window handle: 0x%x, window message: 0x%x, event: 0x%x", pThreadParm->u.asyncselect.s, pThreadParm->notifyHandle, pThreadParm->notifyData, eventReply));
412 PostMessageA((HWND)pThreadParm->notifyHandle, (DWORD)pThreadParm->notifyData, (WPARAM)pThreadParm->u.asyncselect.s,
413 (LPARAM)eventReply);
414 }
415 else
416 if (pThreadParm->u.asyncselect.mode == WSA_SELECT_HEVENT)
417 {
418 dprintf(("AsyncSelectNotifyEvent: notifying event semaphore, socket: 0x%x, HEVENT: 0x%x, event: 0x%x", pThreadParm->u.asyncselect.s, pThreadParm->notifyHandle, event));
419 // set the event bit in the mask
420 pThreadParm->u.asyncselect.lLastEvent |= event;
421 // set the error code for the right value
422 // for this, we first have to find out which bit field to write to
423 char slot = 0;
424 ULONG eventValue = event;
425 while (eventValue > 0)
426 {
427 slot++;
428 eventValue = eventValue >> 1;
429 }
430 pThreadParm->u.asyncselect.iErrorCode[slot] = socket_error;
431 // now post the semaphore so that the client can check the event
432 SetEvent(pThreadParm->notifyHandle);
433 }
434 else
435 {
436 dprintf(("AsyncSelectNotifyEvent: error, unknown mode"));
437 }
438}
439//******************************************************************************
440#define nr(i) ((i != -1) ? 1 : 0)
441#define ready(i) ((i != -1) && (sockets[i] != -1))
442//NOTE: Must execute RemoveFromQueue before returning
443//******************************************************************************
444void ASYNCCNV WSAsyncSelectThreadProc(void *pparm)
445{
446 PASYNCTHREADPARM pThreadParm = (PASYNCTHREADPARM)pparm;
447 SOCKET sockets[3];
448 SOCKET s = pThreadParm->u.asyncselect.s;
449 int noread, nowrite, noexcept, state, sockoptlen, sockoptval;
450 int tmp, i, lEventsPending, ret, bytesread;
451
452 while(TRUE)
453 {
454asyncloopstart:
455 i = 0;
456 noread = nowrite = noexcept = -1;
457
458 //break if user cancelled request
459 if(pThreadParm->u.asyncselect.lEvents == 0 || pThreadParm->fCancelled) {
460 break;
461 }
462
463 lEventsPending = pThreadParm->u.asyncselect.lEventsPending;
464 //block if no events are pending
465 if(lEventsPending == 0)
466 {
467 dprintf2(("WSAsyncSelectThreadProc: waiting for new events"));
468 //wait for events to be enabled
469 pThreadParm->u.asyncselect.asyncSem->wait();
470 //reset event semaphore
471 pThreadParm->u.asyncselect.asyncSem->reset();
472 continue;
473 }
474
475 if(lEventsPending & (FD_READ | FD_CLOSE | FD_ACCEPT)) {
476 noread = i++;
477 sockets[noread] = s;
478 }
479 if((lEventsPending & FD_CONNECT) ||
480 ((lEventsPending & FD_WRITE) &&
481 (!(ioctl(s, FIOBSTATUS, (char *)&tmp, sizeof(tmp)) & SS_CANTSENDMORE))))
482 {
483 nowrite = i++;
484 sockets[nowrite] = s;
485 }
486 if(lEventsPending & FD_OOB) {
487 noexcept = i++;
488 sockets[noexcept] = s;
489 }
490
491 dprintf2(("WSAsyncSelectThreadProc %x rds=%d, wrs=%d, oos =%d, pending = %x", pThreadParm->u.asyncselect.s, noread, nowrite, noexcept, lEventsPending));
492
493 pThreadParm->fWaitSelect = TRUE;
494 ret = select((int *)sockets, nr(noread), nr(nowrite), nr(noexcept), -1);
495 pThreadParm->fWaitSelect = FALSE;
496 if(ret == SOCKET_ERROR) {
497 int selecterr = sock_errno();
498 dprintf2(("WSAsyncSelectThreadProc %x rds=%d, wrs=%d, oos =%d, pending = %x select returned %x", pThreadParm->u.asyncselect.s, noread, nowrite, noexcept, lEventsPending, selecterr));
499 if(selecterr && selecterr < SOCBASEERR) {
500 selecterr += SOCBASEERR;
501 }
502 switch(selecterr)
503 {
504 case SOCEINTR:
505//// state = ioctl(s, FIOBSTATUS, (char *)&tmp, sizeof(tmp));
506//// dprintf(("SOCEINTR; state = %x", state));
507 goto asyncloopstart; //so_cancel was called
508
509 case SOCECONNRESET:
510 case SOCEPIPE:
511 if(lEventsPending & FD_CLOSE)
512 {
513 dprintf(("FD_CLOSE; broken connection"));
514 AsyncSelectNotifyEvent(pThreadParm, FD_CLOSE, WSAECONNRESET);
515 }
516
517 //remote connection broken (so can't receive data anymore)
518 //but can still send
519 pThreadParm->u.asyncselect.lEventsPending &= ~(FD_READ | FD_ACCEPT);
520 goto asyncloopstart;
521
522 case SOCEINVAL:
523 if(lEventsPending & FD_CLOSE) {
524 dprintf(("FD_CLOSE; SOCEINVAL"));
525 AsyncSelectNotifyEvent(pThreadParm, FD_CLOSE, selecterr);
526 }
527 break;
528 default:
529 dprintf(("WSAsyncSelectThreadProc: select SOCKET_ERROR %x", selecterr));
530 break; //something bad happened
531 }
532 break;
533 }
534
535 if(ready(nowrite))
536 {
537 state = ioctl(s, FIOBSTATUS, (char *)&tmp, sizeof(tmp));
538
539 if(lEventsPending & FD_CONNECT) {
540 if(state & SS_ISCONNECTED) {
541 AsyncSelectNotifyEvent(pThreadParm, FD_CONNECT, NO_ERROR);
542 }
543 else {
544 sockoptlen = sizeof(int);
545
546 ret = getsockopt(s, SOL_SOCKET, SO_ERROR,
547 (char *) &sockoptval, &sockoptlen);
548 //SvL: WSeB returns SOCECONNREFUSED, Warp 4 0x3d
549 if(sockoptval == SOCECONNREFUSED || sockoptval == (SOCECONNREFUSED - SOCBASEERR)) {
550 AsyncSelectNotifyEvent(pThreadParm, FD_CONNECT, WSAECONNREFUSED);
551 }
552 }
553 }
554 else
555 if(!(state & SS_CANTSENDMORE) && (lEventsPending & FD_WRITE)) {
556 AsyncSelectNotifyEvent(pThreadParm, FD_WRITE, NO_ERROR);
557 }
558 }
559
560 if(ready(noread))
561 {
562 state = ioctl(s, FIONREAD, (CHAR *) &bytesread, sizeof(bytesread));
563 if(state == SOCKET_ERROR)
564 {
565 if(lEventsPending & FD_CLOSE)
566 {
567 dprintf(("FD_CLOSE; ioctl; socket error"));
568 AsyncSelectNotifyEvent(pThreadParm, FD_CLOSE, NO_ERROR);
569 //remote connection broken (so can't receive data anymore)
570 //but can still send
571 pThreadParm->u.asyncselect.lEventsPending &= ~(FD_READ | FD_ACCEPT);
572 continue;
573 }
574 else {
575 dprintf(("WSAsyncSelectThreadProc: ioctl SOCKET_ERROR!"));
576 break; //todo: correct???
577 }
578 }
579 if(lEventsPending & FD_ACCEPT)
580 {
581 sockoptlen = sizeof(sockoptlen);
582
583 ret = getsockopt(s, SOL_SOCKET, SO_OPTIONS,
584 (char *) &sockoptval, &sockoptlen);
585 if(ret == SOCKET_ERROR) {
586 dprintf(("WSAsyncSelectThreadProc: getsockopt SOCKET_ERROR!"));
587 break;
588 }
589 if((sockoptval & SO_ACCEPTCONN) == SO_ACCEPTCONN) {
590 AsyncSelectNotifyEvent(pThreadParm, FD_ACCEPT, NO_ERROR);
591 }
592 }
593 if((lEventsPending & FD_READ) && bytesread >= 0) {
594 AsyncSelectNotifyEvent(pThreadParm, FD_READ, NO_ERROR);
595 }
596#if 0
597//SvL: This generates FD_CLOSE messages when the connection is just fine
598// (recv executed in another thread when select returns)
599 else
600 if((lEventsPending & FD_CLOSE) && (state == 0 && bytesread == 0)) {
601 dprintf(("FD_CLOSE; state == 0 && bytesread == 0"));
602 AsyncSelectNotifyEvent(pThreadParm, FD_CLOSE, NO_ERROR);
603 }
604#endif
605 }
606 if(ready(noexcept))
607 {
608 if(lEventsPending & FD_OOB) {
609 AsyncSelectNotifyEvent(pThreadParm, FD_OOB, NO_ERROR);
610 }
611 }
612 if((pThreadParm->u.asyncselect.lEventsPending & (FD_ACCEPT|FD_CLOSE|FD_CONNECT)) ==
613 (lEventsPending & (FD_ACCEPT|FD_CLOSE|FD_CONNECT))) {
614 DosSleep(10);
615 }
616 }
617 //remove it first, then delete semaphore object
618 pThreadParm->fActive = FALSE;
619 RemoveFromQueue(pThreadParm);
620 delete pThreadParm->u.asyncselect.asyncSem;
621 pThreadParm->u.asyncselect.asyncSem = 0;
622}
623//******************************************************************************
624//******************************************************************************
625ODINFUNCTION4(int,WSAAsyncSelect,
626 SOCKET,s,
627 HWND,hWnd,
628 u_int,wMsg,
629 long,lEvent)
630{
631 /* just forward call to worker method */
632 return WSAAsyncSelectWorker(s, WSA_SELECT_HWND, (int)hWnd, (int)wMsg, lEvent);
633}
634//******************************************************************************
635//******************************************************************************
636int WSAAsyncSelectWorker(SOCKET s, int mode, int notifyHandle, int notifyData, long lEventMask)
637{
638 PASYNCTHREADPARM pThreadParm;
639 int nonblock = 1;
640 int ret;
641
642#ifdef DEBUG
643 // log all event bits that are set
644 char tmpbuf[300];
645 strcpy(tmpbuf, "");
646 if (lEventMask & FD_READ)
647 strcat(tmpbuf, " FD_READ");
648 if (lEventMask & FD_WRITE)
649 strcat(tmpbuf, " FD_WRITE");
650 if (lEventMask & FD_OOB)
651 strcat(tmpbuf, " FD_OOB");
652 if (lEventMask & FD_ACCEPT)
653 strcat(tmpbuf, " FD_ACCEPT");
654 if (lEventMask & FD_CONNECT)
655 strcat(tmpbuf, " FD_CONNECT");
656 if (lEventMask & FD_CLOSE)
657 strcat(tmpbuf, " FD_CLOSE");
658 if (lEventMask & FD_QOS)
659 strcat(tmpbuf, " FD_QOS");
660 if (lEventMask & FD_GROUP_QOS)
661 strcat(tmpbuf, " FD_GROUP_QOS");
662 if (lEventMask & FD_ROUTING_INTERFACE_CHANGE)
663 strcat(tmpbuf, " FD_ROUTING_INTERFACE_CHANGE");
664 if (lEventMask & FD_ADDRESS_LIST_CHANGE)
665 strcat(tmpbuf, " FD_ADDRESS_LIST_CHANGE");
666 dprintf(("event bits:%s", tmpbuf));
667#endif
668
669 if(!fWSAInitialized)
670 {
671 dprintf(("WSA sockets not initialized"));
672 WSASetLastError(WSANOTINITIALISED);
673 return SOCKET_ERROR;
674 }
675 else
676 if(WSAIsBlocking())
677 {
678 dprintf(("blocking call in progress"));
679 WSASetLastError(WSAEINPROGRESS); // blocking call in progress
680 return SOCKET_ERROR;
681 }
682 else
683 if((mode == WSA_SELECT_HWND) && (HWND)notifyHandle && !IsWindow((HWND)notifyHandle))
684 {
685 dprintf(("invalid window handle"));
686 WSASetLastError(WSAEINVAL); // invalid parameter
687 return SOCKET_ERROR;
688 }
689
690 // Set socket to non-blocking mode
691 ret = ioctl(s, FIONBIO, (char *)&nonblock, sizeof(nonblock));
692 if(ret == SOCKET_ERROR) {
693 dprintf(("setting socket to non blocking mode failed"));
694 WSASetLastError(wsaErrno());
695 return SOCKET_ERROR;
696 }
697 if(FindAndSetAsyncEvent(s, mode, notifyHandle, notifyData, lEventMask) == TRUE)
698 {
699 dprintf(("already got socket, just changing event mask"));
700 //found and changed active async event
701 WSASetLastError(NO_ERROR);
702 return NO_ERROR;
703 }
704 pThreadParm = (PASYNCTHREADPARM)malloc(sizeof(ASYNCTHREADPARM));
705 if(pThreadParm == NULL) {
706 dprintf(("WSAAsyncSelect: malloc failure!"));
707 DebugInt3();
708 WSASetLastError(WSAEFAULT);
709 return SOCKET_ERROR;
710 }
711 memset(pThreadParm, 0, sizeof(*pThreadParm));
712 pThreadParm->request = ASYNC_SELECT;
713 pThreadParm->notifyHandle = notifyHandle;
714 pThreadParm->notifyData = notifyData;
715 pThreadParm->u.asyncselect.mode = mode;
716 pThreadParm->u.asyncselect.lEvents = lEventMask;
717 pThreadParm->u.asyncselect.lEventsPending = lEventMask;
718 pThreadParm->u.asyncselect.s = s;
719 // reset all event bits
720 pThreadParm->u.asyncselect.lLastEvent = 0;
721 // reset all error bits
722 memset(&pThreadParm->u.asyncselect.iErrorCode, 0, sizeof(int) * FD_MAX_EVENTS);
723 pThreadParm->u.asyncselect.asyncSem = new VSemaphore;
724 if(pThreadParm->u.asyncselect.asyncSem == NULL) {
725 dprintf(("WSAAsyncSelect: VSemaphore alloc failure!"));
726 DebugInt3();
727 WSASetLastError(WSAEFAULT);
728 return SOCKET_ERROR;
729 }
730 if(QueueAsyncJob(WSAsyncSelectThreadProc, pThreadParm) == 0) {
731 delete pThreadParm->u.asyncselect.asyncSem;
732 free(pThreadParm);
733 dprintf(("WSAAsyncSelect: QueueAsyncJob failure!"));
734 DebugInt3();
735 WSASetLastError(WSAEFAULT);
736 return SOCKET_ERROR;
737 }
738 WSASetLastError(NO_ERROR);
739 return NO_ERROR;
740}
741//******************************************************************************
742//******************************************************************************
743int WSAEnumNetworkEventsWorker(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent)
744{
745 PASYNCTHREADPARM pThreadInfo;
746
747 asyncThreadMutex.enter();
748 pThreadInfo = FindAsyncEvent(s);
749 if (pThreadInfo)
750 {
751 // return our internal error bit representation
752 memcpy(&lpEvent->iErrorCode, &pThreadInfo->u.asyncselect.iErrorCode, sizeof(int) * FD_MAX_EVENTS);
753 // return our internal event bit representation
754 lpEvent->lNetworkEvents = InterlockedExchange((LPLONG)&pThreadInfo->u.asyncselect.lLastEvent, 0);
755 }
756 else
757 {
758 asyncThreadMutex.leave();
759 dprintf(("no async registration for socket %x", s));
760 // TODO: correct behavior?
761 WSASetLastError(WSAEINVAL);
762 return SOCKET_ERROR;
763 }
764 asyncThreadMutex.leave();
765
766 // reset event semaphore if present
767 if (hEvent)
768 {
769// dprintf(("posting event semaphore 0x%x", hEvent));
770 ResetEvent(hEvent);
771 }
772
773 WSASetLastError(NO_ERROR);
774 return NO_ERROR;
775}
776//******************************************************************************
777//******************************************************************************
Note: See TracBrowser for help on using the repository browser.