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

Last change on this file since 7977 was 7977, checked in by sandervl, 24 years ago

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