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

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

IPPROTO_IP support for setsockopt, WSAEnumNetworkEvents fixes
File size: 24.2 KB
Line 
1/* $Id: asyncapi.cpp,v 1.15 2001-07-08 15:44:27 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 event = WSAMAKESELECTREPLY(event, socket_error);
406
407 if (pThreadParm->u.asyncselect.mode == WSA_SELECT_HWND)
408 {
409 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, event));
410 PostMessageA((HWND)pThreadParm->notifyHandle, (DWORD)pThreadParm->notifyData, (WPARAM)pThreadParm->u.asyncselect.s,
411 (LPARAM)event);
412 }
413 else
414 if (pThreadParm->u.asyncselect.mode == WSA_SELECT_HEVENT)
415 {
416 dprintf(("AsyncSelectNotifyEvent: notifying event semaphore, socket: 0x%x, HEVENT: 0x%x, event: 0x%x", pThreadParm->u.asyncselect.s, pThreadParm->notifyHandle, event));
417 SetEvent(pThreadParm->notifyHandle);
418 }
419 else
420 {
421 dprintf(("AsyncSelectNotifyEvent: error, unknown mode"));
422 }
423}
424//******************************************************************************
425#define nr(i) ((i != -1) ? 1 : 0)
426#define ready(i) ((i != -1) && (sockets[i] != -1))
427//NOTE: Must execute RemoveFromQueue before returning
428//******************************************************************************
429void ASYNCCNV WSAsyncSelectThreadProc(void *pparm)
430{
431 PASYNCTHREADPARM pThreadParm = (PASYNCTHREADPARM)pparm;
432 SOCKET sockets[3];
433 SOCKET s = pThreadParm->u.asyncselect.s;
434 int noread, nowrite, noexcept, state, sockoptlen, sockoptval;
435 int tmp, i, lEventsPending, ret, bytesread;
436
437 while(TRUE)
438 {
439asyncloopstart:
440 i = 0;
441 noread = nowrite = noexcept = -1;
442
443 //break if user cancelled request
444 if(pThreadParm->u.asyncselect.lEvents == 0) {
445 break;
446 }
447
448 lEventsPending = pThreadParm->u.asyncselect.lEventsPending;
449 //block if no events are pending
450 if(lEventsPending == 0)
451 {
452 dprintf2(("WSAsyncSelectThreadProc: waiting for new events"));
453 //wait for events to be enabled
454 pThreadParm->u.asyncselect.asyncSem->wait();
455 //reset event semaphore
456 pThreadParm->u.asyncselect.asyncSem->reset();
457 continue;
458 }
459
460 if(lEventsPending & (FD_READ | FD_CLOSE | FD_ACCEPT)) {
461 noread = i++;
462 sockets[noread] = s;
463 }
464 if((lEventsPending & FD_CONNECT) ||
465 ((lEventsPending & FD_WRITE) &&
466 (!(ioctl(s, FIOBSTATUS, (char *)&tmp, sizeof(tmp)) & SS_CANTSENDMORE))))
467 {
468 nowrite = i++;
469 sockets[nowrite] = s;
470 }
471 if(lEventsPending & FD_OOB) {
472 noexcept = i++;
473 sockets[noexcept] = s;
474 }
475
476 dprintf2(("WSAsyncSelectThreadProc %x rds=%d, wrs=%d, oos =%d, pending = %x", pThreadParm->u.asyncselect.s, noread, nowrite, noexcept, lEventsPending));
477
478 pThreadParm->fWaitSelect = TRUE;
479 ret = select((int *)sockets, nr(noread), nr(nowrite), nr(noexcept), -1);
480 pThreadParm->fWaitSelect = FALSE;
481 if(ret == SOCKET_ERROR) {
482 int selecterr = sock_errno();
483 dprintf2(("WSAsyncSelectThreadProc %x rds=%d, wrs=%d, oos =%d, pending = %x select returned %x", pThreadParm->u.asyncselect.s, noread, nowrite, noexcept, lEventsPending, selecterr));
484 if(selecterr && selecterr < SOCBASEERR) {
485 selecterr += SOCBASEERR;
486 }
487 switch(selecterr)
488 {
489 case SOCEINTR:
490//// state = ioctl(s, FIOBSTATUS, (char *)&tmp, sizeof(tmp));
491//// dprintf(("SOCEINTR; state = %x", state));
492 goto asyncloopstart; //so_cancel was called
493
494 case SOCECONNRESET:
495 case SOCEPIPE:
496 if(lEventsPending & FD_CLOSE) {
497 dprintf(("FD_CLOSE; broken connection"));
498 AsyncSelectNotifyEvent(pThreadParm, FD_CLOSE, WSAECONNRESET);
499 }
500
501 //remote connection broken (so can't receive data anymore)
502 //but can still send
503 pThreadParm->u.asyncselect.lEventsPending &= ~(FD_READ | FD_ACCEPT);
504 goto asyncloopstart;
505
506 case SOCEINVAL:
507 if(lEventsPending & FD_CLOSE) {
508 dprintf(("FD_CLOSE; SOCEINVAL"));
509 AsyncSelectNotifyEvent(pThreadParm, FD_CLOSE, selecterr);
510 }
511 break;
512 default:
513 dprintf(("WSAsyncSelectThreadProc: select SOCKET_ERROR %x", selecterr));
514 break; //something bad happened
515 }
516 break;
517 }
518
519 if(ready(nowrite))
520 {
521 state = ioctl(s, FIOBSTATUS, (char *)&tmp, sizeof(tmp));
522
523 if(lEventsPending & FD_CONNECT) {
524 if(state & SS_ISCONNECTED) {
525 AsyncSelectNotifyEvent(pThreadParm, FD_CONNECT, NO_ERROR);
526 }
527 else {
528 sockoptlen = sizeof(int);
529
530 ret = getsockopt(s, SOL_SOCKET, SO_ERROR,
531 (char *) &sockoptval, &sockoptlen);
532 //SvL: WSeB returns SOCECONNREFUSED, Warp 4 0x3d
533 if(sockoptval == SOCECONNREFUSED || sockoptval == (SOCECONNREFUSED - SOCBASEERR)) {
534 AsyncSelectNotifyEvent(pThreadParm, FD_CONNECT, WSAECONNREFUSED);
535 }
536 }
537 }
538 else
539 if(!(state & SS_CANTSENDMORE) && (lEventsPending & FD_WRITE)) {
540 AsyncSelectNotifyEvent(pThreadParm, FD_WRITE, NO_ERROR);
541 }
542 }
543
544 if(ready(noread))
545 {
546 state = ioctl(s, FIONREAD, (CHAR *) &bytesread, sizeof(bytesread));
547 if(state == SOCKET_ERROR)
548 {
549 if(lEventsPending & FD_CLOSE)
550 {
551 dprintf(("FD_CLOSE; ioctl; socket error"));
552 AsyncSelectNotifyEvent(pThreadParm, FD_CLOSE, NO_ERROR);
553 //remote connection broken (so can't receive data anymore)
554 //but can still send
555 pThreadParm->u.asyncselect.lEventsPending &= ~(FD_READ | FD_ACCEPT);
556 continue;
557 }
558 else {
559 dprintf(("WSAsyncSelectThreadProc: ioctl SOCKET_ERROR!"));
560 break; //todo: correct???
561 }
562 }
563 if(lEventsPending & FD_ACCEPT)
564 {
565 sockoptlen = sizeof(sockoptlen);
566
567 ret = getsockopt(s, SOL_SOCKET, SO_OPTIONS,
568 (char *) &sockoptval, &sockoptlen);
569 if(ret == SOCKET_ERROR) {
570 dprintf(("WSAsyncSelectThreadProc: getsockopt SOCKET_ERROR!"));
571 break;
572 }
573 if((sockoptval & SO_ACCEPTCONN) == SO_ACCEPTCONN) {
574 AsyncSelectNotifyEvent(pThreadParm, FD_ACCEPT, NO_ERROR);
575 }
576 }
577 if((lEventsPending & FD_READ) && bytesread >= 0) {
578 AsyncSelectNotifyEvent(pThreadParm, FD_READ, NO_ERROR);
579 }
580#if 0
581//SvL: This generates FD_CLOSE messages when the connection is just fine
582// (recv executed in another thread when select returns)
583 else
584 if((lEventsPending & FD_CLOSE) && (state == 0 && bytesread == 0)) {
585 dprintf(("FD_CLOSE; state == 0 && bytesread == 0"));
586 AsyncSelectNotifyEvent(pThreadParm, FD_CLOSE, NO_ERROR);
587 }
588#endif
589 }
590 if(ready(noexcept))
591 {
592 if(lEventsPending & FD_OOB) {
593 AsyncSelectNotifyEvent(pThreadParm, FD_OOB, NO_ERROR);
594 }
595 }
596 if((pThreadParm->u.asyncselect.lEventsPending & (FD_ACCEPT|FD_CLOSE|FD_CONNECT)) ==
597 (lEventsPending & (FD_ACCEPT|FD_CLOSE|FD_CONNECT))) {
598 DosSleep(10);
599 }
600 }
601 //remove it first, then delete semaphore object
602 pThreadParm->fActive = FALSE;
603 RemoveFromQueue(pThreadParm);
604 delete pThreadParm->u.asyncselect.asyncSem;
605 pThreadParm->u.asyncselect.asyncSem = 0;
606}
607//******************************************************************************
608//******************************************************************************
609ODINFUNCTION4(int,WSAAsyncSelect,
610 SOCKET,s,
611 HWND,hWnd,
612 u_int,wMsg,
613 long,lEvent)
614{
615 /* just forward call to worker method */
616 return WSAAsyncSelectWorker(s, WSA_SELECT_HWND, (int)hWnd, (int)wMsg, lEvent);
617}
618//******************************************************************************
619//******************************************************************************
620int WSAAsyncSelectWorker(SOCKET s, int mode, int notifyHandle, int notifyData, long lEventMask)
621{
622 PASYNCTHREADPARM pThreadParm;
623 int nonblock = 1;
624 int ret;
625
626#ifdef DEBUG
627 // log all event bits that are set
628 char tmpbuf[300];
629 strcpy(tmpbuf, "");
630 if (lEventMask & FD_READ)
631 strcat(tmpbuf, " FD_READ");
632 if (lEventMask & FD_WRITE)
633 strcat(tmpbuf, " FD_WRITE");
634 if (lEventMask & FD_OOB)
635 strcat(tmpbuf, " FD_OOB");
636 if (lEventMask & FD_ACCEPT)
637 strcat(tmpbuf, " FD_ACCEPT");
638 if (lEventMask & FD_CONNECT)
639 strcat(tmpbuf, " FD_CONNECT");
640 if (lEventMask & FD_CLOSE)
641 strcat(tmpbuf, " FD_CLOSE");
642 if (lEventMask & FD_QOS)
643 strcat(tmpbuf, " FD_QOS");
644 if (lEventMask & FD_GROUP_QOS)
645 strcat(tmpbuf, " FD_GROUP_QOS");
646 if (lEventMask & FD_ROUTING_INTERFACE_CHANGE)
647 strcat(tmpbuf, " FD_ROUTING_INTERFACE_CHANGE");
648 if (lEventMask & FD_ADDRESS_LIST_CHANGE)
649 strcat(tmpbuf, " FD_ADDRESS_LIST_CHANGE");
650 dprintf(("event bits:%s", tmpbuf));
651#endif
652
653 if(!fWSAInitialized)
654 {
655 dprintf(("WSA sockets not initialized"));
656 WSASetLastError(WSANOTINITIALISED);
657 return SOCKET_ERROR;
658 }
659 else
660 if(WSAIsBlocking())
661 {
662 dprintf(("blocking call in progress"));
663 WSASetLastError(WSAEINPROGRESS); // blocking call in progress
664 return SOCKET_ERROR;
665 }
666 else
667 if((mode == WSA_SELECT_HWND) && (HWND)notifyHandle && !IsWindow((HWND)notifyHandle))
668 {
669 dprintf(("invalid window handle"));
670 WSASetLastError(WSAEINVAL); // invalid parameter
671 return SOCKET_ERROR;
672 }
673
674 // Set socket to non-blocking mode
675 ret = ioctl(s, FIONBIO, (char *)&nonblock, sizeof(nonblock));
676 if(ret == SOCKET_ERROR) {
677 dprintf(("setting socket to non blocking mode failed"));
678 WSASetLastError(wsaErrno());
679 return SOCKET_ERROR;
680 }
681 if(FindAndSetAsyncEvent(s, mode, notifyHandle, notifyData, lEventMask) == TRUE)
682 {
683 dprintf(("already got socket, just changing event mask"));
684 //found and changed active async event
685 WSASetLastError(NO_ERROR);
686 return NO_ERROR;
687 }
688 pThreadParm = (PASYNCTHREADPARM)malloc(sizeof(ASYNCTHREADPARM));
689 if(pThreadParm == NULL) {
690 dprintf(("WSAAsyncSelect: malloc failure!"));
691 DebugInt3();
692 WSASetLastError(WSAEFAULT);
693 return SOCKET_ERROR;
694 }
695 memset(pThreadParm, 0, sizeof(*pThreadParm));
696 pThreadParm->request = ASYNC_SELECT;
697 pThreadParm->notifyHandle = notifyHandle;
698 pThreadParm->notifyData = notifyData;
699 pThreadParm->u.asyncselect.mode = mode;
700 pThreadParm->u.asyncselect.lEvents = lEventMask;
701 pThreadParm->u.asyncselect.lEventsPending = lEventMask;
702 pThreadParm->u.asyncselect.s = s;
703 pThreadParm->u.asyncselect.asyncSem = new VSemaphore;
704 if(pThreadParm->u.asyncselect.asyncSem == NULL) {
705 dprintf(("WSAAsyncSelect: VSemaphore alloc failure!"));
706 DebugInt3();
707 WSASetLastError(WSAEFAULT);
708 return SOCKET_ERROR;
709 }
710 if(QueueAsyncJob(WSAsyncSelectThreadProc, pThreadParm) == 0) {
711 delete pThreadParm->u.asyncselect.asyncSem;
712 free(pThreadParm);
713 dprintf(("WSAAsyncSelect: QueueAsyncJob failure!"));
714 DebugInt3();
715 WSASetLastError(WSAEFAULT);
716 return SOCKET_ERROR;
717 }
718 WSASetLastError(NO_ERROR);
719 return NO_ERROR;
720}
721//******************************************************************************
722//******************************************************************************
723int WSAEnumNetworkEventsWorker(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent)
724{
725 PASYNCTHREADPARM pThreadInfo;
726
727 asyncThreadMutex.enter();
728 pThreadInfo = FindAsyncEvent(s);
729 if (pThreadInfo)
730 {
731 // TODO return correct errors!!!
732 for (int i = 0; i < FD_MAX_EVENTS; i++)
733 {
734 lpEvent->iErrorCode[i] = 0;
735 }
736
737 lpEvent->lNetworkEvents = pThreadInfo->u.asyncselect.lEventsPending;
738
739 // TODO is this correct? API says they're cleared
740 pThreadInfo->u.asyncselect.lEventsPending = 0;
741 } else
742 {
743 asyncThreadMutex.leave();
744 dprintf(("no async registration for socket %x", s));
745 // TODO: correct behavior?
746 WSASetLastError(WSAEINVAL);
747 return SOCKET_ERROR;
748 }
749
750 // reset event semaphore if present
751 if (hEvent)
752 {
753// dprintf(("posting event semaphore 0x%x", hEvent));
754 ResetEvent(hEvent);
755 }
756
757 asyncThreadMutex.leave();
758
759 WSASetLastError(NO_ERROR);
760 return NO_ERROR;
761}
762//******************************************************************************
763//******************************************************************************
Note: See TracBrowser for help on using the repository browser.