2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS Ancillary Function Driver DLL
5 * PURPOSE: DLL entry point
6 * PROGRAMMERS: Casper S. Hornstrup (chorns@users.sourceforge.net)
7 * Alex Ionescu (alex@relsoft.net)
9 * CSH 01/09-2000 Created
10 * Alex 16/07/2004 - Complete Rewrite
18 //DWORD DebugTraceLevel = DEBUG_ULTRA;
19 DWORD DebugTraceLevel
= 0;
23 WSPUPCALLTABLE Upcalls
;
24 LPWPUCOMPLETEOVERLAPPEDREQUEST lpWPUCompleteOverlappedRequest
;
25 PSOCKET_INFORMATION SocketListHead
= NULL
;
26 CRITICAL_SECTION SocketListLock
;
27 LIST_ENTRY SockHelpersListHead
= { NULL
, NULL
};
28 ULONG SockAsyncThreadRefCount
;
29 HANDLE SockAsyncHelperAfdHandle
;
30 HANDLE SockAsyncCompletionPort
;
31 BOOLEAN SockAsyncSelectCalled
;
36 * FUNCTION: Creates a new socket
40 * protocol = Protocol type
41 * lpProtocolInfo = Pointer to protocol information
43 * dwFlags = Socket flags
44 * lpErrno = Address of buffer for error information
46 * Created socket, or INVALID_SOCKET if it could not be created
50 WSPSocket(int AddressFamily
,
53 LPWSAPROTOCOL_INFOW lpProtocolInfo
,
58 OBJECT_ATTRIBUTES Object
;
62 PAFD_CREATE_PACKET AfdPacket
;
64 PSOCKET_INFORMATION Socket
= NULL
;
65 PFILE_FULL_EA_INFORMATION EABuffer
= NULL
;
66 PHELPER_DATA HelperData
;
67 PVOID HelperDLLContext
;
69 UNICODE_STRING TransportName
;
70 UNICODE_STRING DevName
;
71 LARGE_INTEGER GroupData
;
74 AFD_DbgPrint(MAX_TRACE
, ("Creating Socket, getting TDI Name\n"));
75 AFD_DbgPrint(MAX_TRACE
, ("AddressFamily (%d) SocketType (%d) Protocol (%d).\n",
76 AddressFamily
, SocketType
, Protocol
));
78 /* Get Helper Data and Transport */
79 Status
= SockGetTdiName (&AddressFamily
,
90 if (Status
!= NO_ERROR
)
92 AFD_DbgPrint(MID_TRACE
,("SockGetTdiName: Status %x\n", Status
));
97 RtlInitUnicodeString(&DevName
, L
"\\Device\\Afd\\Endpoint");
100 Socket
= HeapAlloc(GlobalHeap
, 0, sizeof(*Socket
));
102 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
104 RtlZeroMemory(Socket
, sizeof(*Socket
));
105 Socket
->RefCount
= 2;
107 Socket
->SharedData
.Listening
= FALSE
;
108 Socket
->SharedData
.State
= SocketOpen
;
109 Socket
->SharedData
.AddressFamily
= AddressFamily
;
110 Socket
->SharedData
.SocketType
= SocketType
;
111 Socket
->SharedData
.Protocol
= Protocol
;
112 Socket
->HelperContext
= HelperDLLContext
;
113 Socket
->HelperData
= HelperData
;
114 Socket
->HelperEvents
= HelperEvents
;
115 Socket
->LocalAddress
= &Socket
->WSLocalAddress
;
116 Socket
->SharedData
.SizeOfLocalAddress
= HelperData
->MaxWSAddressLength
;
117 Socket
->RemoteAddress
= &Socket
->WSRemoteAddress
;
118 Socket
->SharedData
.SizeOfRemoteAddress
= HelperData
->MaxWSAddressLength
;
119 Socket
->SharedData
.UseDelayedAcceptance
= HelperData
->UseDelayedAcceptance
;
120 Socket
->SharedData
.CreateFlags
= dwFlags
;
121 Socket
->SharedData
.CatalogEntryId
= lpProtocolInfo
->dwCatalogEntryId
;
122 Socket
->SharedData
.ServiceFlags1
= lpProtocolInfo
->dwServiceFlags1
;
123 Socket
->SharedData
.ProviderFlags
= lpProtocolInfo
->dwProviderFlags
;
124 Socket
->SharedData
.GroupID
= g
;
125 Socket
->SharedData
.GroupType
= 0;
126 Socket
->SharedData
.UseSAN
= FALSE
;
127 Socket
->SharedData
.NonBlocking
= FALSE
; /* Sockets start blocking */
128 Socket
->SanData
= NULL
;
130 /* Ask alex about this */
131 if( Socket
->SharedData
.SocketType
== SOCK_DGRAM
||
132 Socket
->SharedData
.SocketType
== SOCK_RAW
)
134 AFD_DbgPrint(MID_TRACE
,("Connectionless socket\n"));
135 Socket
->SharedData
.ServiceFlags1
|= XP1_CONNECTIONLESS
;
139 SizeOfPacket
= TransportName
.Length
+ sizeof(AFD_CREATE_PACKET
) + sizeof(WCHAR
);
142 SizeOfEA
= SizeOfPacket
+ sizeof(FILE_FULL_EA_INFORMATION
) + AFD_PACKET_COMMAND_LENGTH
;
144 /* Set up EA Buffer */
145 EABuffer
= HeapAlloc(GlobalHeap
, 0, SizeOfEA
);
147 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
149 RtlZeroMemory(EABuffer
, SizeOfEA
);
150 EABuffer
->NextEntryOffset
= 0;
152 EABuffer
->EaNameLength
= AFD_PACKET_COMMAND_LENGTH
;
153 RtlCopyMemory (EABuffer
->EaName
,
155 AFD_PACKET_COMMAND_LENGTH
+ 1);
156 EABuffer
->EaValueLength
= SizeOfPacket
;
158 /* Set up AFD Packet */
159 AfdPacket
= (PAFD_CREATE_PACKET
)(EABuffer
->EaName
+ EABuffer
->EaNameLength
+ 1);
160 AfdPacket
->SizeOfTransportName
= TransportName
.Length
;
161 RtlCopyMemory (AfdPacket
->TransportName
,
162 TransportName
.Buffer
,
163 TransportName
.Length
+ sizeof(WCHAR
));
164 AfdPacket
->GroupID
= g
;
166 /* Set up Endpoint Flags */
167 if ((Socket
->SharedData
.ServiceFlags1
& XP1_CONNECTIONLESS
) != 0)
169 if ((SocketType
!= SOCK_DGRAM
) && (SocketType
!= SOCK_RAW
))
171 /* Only RAW or UDP can be Connectionless */
174 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_CONNECTIONLESS
;
177 if ((Socket
->SharedData
.ServiceFlags1
& XP1_MESSAGE_ORIENTED
) != 0)
179 if (SocketType
== SOCK_STREAM
)
181 if ((Socket
->SharedData
.ServiceFlags1
& XP1_PSEUDO_STREAM
) == 0)
183 /* The Provider doesn't actually support Message Oriented Streams */
187 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_MESSAGE_ORIENTED
;
190 if (SocketType
== SOCK_RAW
) AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_RAW
;
192 if (dwFlags
& (WSA_FLAG_MULTIPOINT_C_ROOT
|
193 WSA_FLAG_MULTIPOINT_C_LEAF
|
194 WSA_FLAG_MULTIPOINT_D_ROOT
|
195 WSA_FLAG_MULTIPOINT_D_LEAF
))
197 if ((Socket
->SharedData
.ServiceFlags1
& XP1_SUPPORT_MULTIPOINT
) == 0)
199 /* The Provider doesn't actually support Multipoint */
202 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_MULTIPOINT
;
204 if (dwFlags
& WSA_FLAG_MULTIPOINT_C_ROOT
)
206 if (((Socket
->SharedData
.ServiceFlags1
& XP1_MULTIPOINT_CONTROL_PLANE
) == 0)
207 || ((dwFlags
& WSA_FLAG_MULTIPOINT_C_LEAF
) != 0))
209 /* The Provider doesn't support Control Planes, or you already gave a leaf */
212 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_C_ROOT
;
215 if (dwFlags
& WSA_FLAG_MULTIPOINT_D_ROOT
)
217 if (((Socket
->SharedData
.ServiceFlags1
& XP1_MULTIPOINT_DATA_PLANE
) == 0)
218 || ((dwFlags
& WSA_FLAG_MULTIPOINT_D_LEAF
) != 0))
220 /* The Provider doesn't support Data Planes, or you already gave a leaf */
223 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_D_ROOT
;
227 /* Set up Object Attributes */
228 InitializeObjectAttributes (&Object
,
230 OBJ_CASE_INSENSITIVE
| OBJ_INHERIT
,
234 /* Create the Socket as asynchronous. That means we have to block
235 ourselves after every call to NtDeviceIoControlFile. This is
236 because the kernel doesn't support overlapping synchronous I/O
237 requests (made from multiple threads) at this time (Sep 2005) */
238 Status
= NtCreateFile(&Sock
,
239 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
,
244 FILE_SHARE_READ
| FILE_SHARE_WRITE
,
250 HeapFree(GlobalHeap
, 0, EABuffer
);
252 if (Status
!= STATUS_SUCCESS
)
254 AFD_DbgPrint(MIN_TRACE
, ("Failed to open socket\n"));
256 HeapFree(GlobalHeap
, 0, Socket
);
258 return MsafdReturnWithErrno(Status
, lpErrno
, 0, NULL
);
262 Socket
->Handle
= (SOCKET
)Sock
;
264 /* Save Group Info */
267 GetSocketInformation(Socket
, AFD_INFO_GROUP_ID_TYPE
, 0, &GroupData
);
268 Socket
->SharedData
.GroupID
= GroupData
.u
.LowPart
;
269 Socket
->SharedData
.GroupType
= GroupData
.u
.HighPart
;
272 /* Get Window Sizes and Save them */
273 GetSocketInformation (Socket
,
274 AFD_INFO_SEND_WINDOW_SIZE
,
275 &Socket
->SharedData
.SizeOfSendBuffer
,
278 GetSocketInformation (Socket
,
279 AFD_INFO_RECEIVE_WINDOW_SIZE
,
280 &Socket
->SharedData
.SizeOfRecvBuffer
,
283 /* Save in Process Sockets List */
284 EnterCriticalSection(&SocketListLock
);
285 Socket
->NextSocket
= SocketListHead
;
286 SocketListHead
= Socket
;
287 LeaveCriticalSection(&SocketListLock
);
289 /* Create the Socket Context */
290 CreateContext(Socket
);
293 Upcalls
.lpWPUModifyIFSHandle(1, (SOCKET
)Sock
, lpErrno
);
295 /* Return Socket Handle */
296 AFD_DbgPrint(MID_TRACE
,("Success %x\n", Sock
));
301 AFD_DbgPrint(MID_TRACE
,("Ending %x\n", Status
));
304 HeapFree(GlobalHeap
, 0, Socket
);
309 return INVALID_SOCKET
;
313 TranslateNtStatusError(NTSTATUS Status
)
317 case STATUS_CANT_WAIT
:
318 return WSAEWOULDBLOCK
;
326 case STATUS_FILE_CLOSED
:
327 case STATUS_END_OF_FILE
:
331 return WSA_IO_PENDING
;
333 case STATUS_BUFFER_TOO_SMALL
:
334 case STATUS_BUFFER_OVERFLOW
:
335 DbgPrint("MSAFD: STATUS_BUFFER_TOO_SMALL/STATUS_BUFFER_OVERFLOW\n");
338 case STATUS_NO_MEMORY
:
339 case STATUS_INSUFFICIENT_RESOURCES
:
340 DbgPrint("MSAFD: STATUS_NO_MEMORY/STATUS_INSUFFICIENT_RESOURCES\n");
343 case STATUS_INVALID_CONNECTION
:
344 DbgPrint("MSAFD: STATUS_INVALID_CONNECTION\n");
345 return WSAEAFNOSUPPORT
;
347 case STATUS_INVALID_ADDRESS
:
348 DbgPrint("MSAFD: STATUS_INVALID_ADDRESS\n");
349 return WSAEADDRNOTAVAIL
;
351 case STATUS_REMOTE_NOT_LISTENING
:
352 DbgPrint("MSAFD: STATUS_REMOTE_NOT_LISTENING\n");
353 return WSAECONNREFUSED
;
355 case STATUS_NETWORK_UNREACHABLE
:
356 DbgPrint("MSAFD: STATUS_NETWORK_UNREACHABLE\n");
357 return WSAENETUNREACH
;
359 case STATUS_INVALID_PARAMETER
:
360 DbgPrint("MSAFD: STATUS_INVALID_PARAMETER\n");
363 case STATUS_CANCELLED
:
364 DbgPrint("MSAFD: STATUS_CANCELLED\n");
365 return WSA_OPERATION_ABORTED
;
367 case STATUS_ADDRESS_ALREADY_EXISTS
:
368 DbgPrint("MSAFD: STATUS_ADDRESS_ALREADY_EXISTS\n");
369 return WSAEADDRINUSE
;
371 case STATUS_LOCAL_DISCONNECT
:
372 DbgPrint("MSAFD: STATUS_LOCAL_DISCONNECT\n");
373 return WSAECONNABORTED
;
375 case STATUS_REMOTE_DISCONNECT
:
376 DbgPrint("MSAFD: STATUS_REMOTE_DISCONNECT\n");
377 return WSAECONNRESET
;
380 DbgPrint("MSAFD: Unhandled NTSTATUS value: 0x%x\n", Status
);
385 DWORD
MsafdReturnWithErrno(NTSTATUS Status
,
388 LPDWORD ReturnedBytes
)
390 *Errno
= TranslateNtStatusError(Status
);
395 *ReturnedBytes
= Received
;
400 return *Errno
? SOCKET_ERROR
: 0;
404 * FUNCTION: Closes an open socket
406 * s = Socket descriptor
407 * lpErrno = Address of buffer for error information
409 * NO_ERROR, or SOCKET_ERROR if the socket could not be closed
413 WSPCloseSocket(IN SOCKET Handle
,
416 IO_STATUS_BLOCK IoStatusBlock
;
417 PSOCKET_INFORMATION Socket
= NULL
, CurrentSocket
;
420 AFD_DISCONNECT_INFO DisconnectInfo
;
421 SOCKET_STATE OldState
;
423 /* Create the Wait Event */
424 Status
= NtCreateEvent(&SockEvent
,
425 GENERIC_READ
| GENERIC_WRITE
,
430 if(!NT_SUCCESS(Status
))
433 /* Get the Socket Structure associate to this Socket*/
434 Socket
= GetSocketStructure(Handle
);
438 *lpErrno
= WSAENOTSOCK
;
442 if (Socket
->HelperEvents
& WSH_NOTIFY_CLOSE
)
444 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
446 Socket
->TdiAddressHandle
,
447 Socket
->TdiConnectionHandle
,
452 if (lpErrno
) *lpErrno
= Status
;
458 /* If a Close is already in Process, give up */
459 if (Socket
->SharedData
.State
== SocketClosed
)
462 *lpErrno
= WSAENOTSOCK
;
466 /* Set the state to close */
467 OldState
= Socket
->SharedData
.State
;
468 Socket
->SharedData
.State
= SocketClosed
;
470 /* If SO_LINGER is ON and the Socket is connected, we need to disconnect */
471 /* FIXME: Should we do this on Datagram Sockets too? */
472 if ((OldState
== SocketConnected
) && (Socket
->SharedData
.LingerData
.l_onoff
))
475 ULONG SendsInProgress
;
478 /* We need to respect the timeout */
480 LingerWait
= Socket
->SharedData
.LingerData
.l_linger
* 1000;
482 /* Loop until no more sends are pending, within the timeout */
485 /* Find out how many Sends are in Progress */
486 if (GetSocketInformation(Socket
,
487 AFD_INFO_SENDS_IN_PROGRESS
,
491 /* Bail out if anything but NO_ERROR */
496 /* Bail out if no more sends are pending */
497 if (!SendsInProgress
)
500 * We have to execute a sleep, so it's kind of like
501 * a block. If the socket is Nonblock, we cannot
502 * go on since asyncronous operation is expected
503 * and we cannot offer it
505 if (Socket
->SharedData
.NonBlocking
)
508 Socket
->SharedData
.State
= OldState
;
509 *lpErrno
= WSAEWOULDBLOCK
;
513 /* Now we can sleep, and decrement the linger wait */
515 * FIXME: It seems Windows does some funky acceleration
516 * since the waiting seems to be longer and longer. I
517 * don't think this improves performance so much, so we
518 * wait a fixed time instead.
521 LingerWait
-= SleepWait
;
525 * We have reached the timeout or sends are over.
526 * Disconnect if the timeout has been reached.
530 DisconnectInfo
.Timeout
= RtlConvertLongToLargeInteger(0);
531 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_ABORT
;
534 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
539 IOCTL_AFD_DISCONNECT
,
541 sizeof(DisconnectInfo
),
545 /* Wait for return */
546 if (Status
== STATUS_PENDING
)
548 WaitForSingleObject(SockEvent
, INFINITE
);
549 Status
= IoStatusBlock
.Status
;
555 Socket
->HelperContext
= NULL
;
556 Socket
->SharedData
.AsyncDisabledEvents
= -1;
557 NtClose(Socket
->TdiAddressHandle
);
558 Socket
->TdiAddressHandle
= NULL
;
559 NtClose(Socket
->TdiConnectionHandle
);
560 Socket
->TdiConnectionHandle
= NULL
;
562 EnterCriticalSection(&SocketListLock
);
563 if (SocketListHead
== Socket
)
565 SocketListHead
= SocketListHead
->NextSocket
;
569 CurrentSocket
= SocketListHead
;
570 while (CurrentSocket
->NextSocket
)
572 if (CurrentSocket
->NextSocket
== Socket
)
574 CurrentSocket
->NextSocket
= CurrentSocket
->NextSocket
->NextSocket
;
578 CurrentSocket
= CurrentSocket
->NextSocket
;
581 LeaveCriticalSection(&SocketListLock
);
583 HeapFree(GlobalHeap
, 0, Socket
);
585 /* Close the handle */
586 NtClose((HANDLE
)Handle
);
589 return MsafdReturnWithErrno(Status
, lpErrno
, 0, NULL
);
594 * FUNCTION: Associates a local address with a socket
596 * s = Socket descriptor
597 * name = Pointer to local address
598 * namelen = Length of name
599 * lpErrno = Address of buffer for error information
601 * 0, or SOCKET_ERROR if the socket could not be bound
605 WSPBind(SOCKET Handle
,
606 const struct sockaddr
*SocketAddress
,
607 int SocketAddressLength
,
610 IO_STATUS_BLOCK IOSB
;
611 PAFD_BIND_DATA BindData
;
612 PSOCKET_INFORMATION Socket
= NULL
;
614 SOCKADDR_INFO SocketInfo
;
618 BindData
= HeapAlloc(GlobalHeap
, 0, 0xA + SocketAddressLength
);
621 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
624 Status
= NtCreateEvent(&SockEvent
,
625 GENERIC_READ
| GENERIC_WRITE
,
630 if (!NT_SUCCESS(Status
))
632 HeapFree(GlobalHeap
, 0, BindData
);
636 /* Get the Socket Structure associate to this Socket*/
637 Socket
= GetSocketStructure(Handle
);
640 HeapFree(GlobalHeap
, 0, BindData
);
641 *lpErrno
= WSAENOTSOCK
;
645 /* Set up Address in TDI Format */
646 BindData
->Address
.TAAddressCount
= 1;
647 BindData
->Address
.Address
[0].AddressLength
= SocketAddressLength
- sizeof(SocketAddress
->sa_family
);
648 BindData
->Address
.Address
[0].AddressType
= SocketAddress
->sa_family
;
649 RtlCopyMemory (BindData
->Address
.Address
[0].Address
,
650 SocketAddress
->sa_data
,
651 SocketAddressLength
- sizeof(SocketAddress
->sa_family
));
653 /* Get Address Information */
654 Socket
->HelperData
->WSHGetSockaddrType ((PSOCKADDR
)SocketAddress
,
658 /* Set the Share Type */
659 if (Socket
->SharedData
.ExclusiveAddressUse
)
661 BindData
->ShareType
= AFD_SHARE_EXCLUSIVE
;
663 else if (SocketInfo
.EndpointInfo
== SockaddrEndpointInfoWildcard
)
665 BindData
->ShareType
= AFD_SHARE_WILDCARD
;
667 else if (Socket
->SharedData
.ReuseAddresses
)
669 BindData
->ShareType
= AFD_SHARE_REUSE
;
673 BindData
->ShareType
= AFD_SHARE_UNIQUE
;
677 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
684 0xA + Socket
->SharedData
.SizeOfLocalAddress
, /* Can't figure out a way to calculate this in C*/
686 0xA + Socket
->SharedData
.SizeOfLocalAddress
); /* Can't figure out a way to calculate this C */
688 /* Wait for return */
689 if (Status
== STATUS_PENDING
)
691 WaitForSingleObject(SockEvent
, INFINITE
);
692 Status
= IOSB
.Status
;
695 NtClose( SockEvent
);
696 HeapFree(GlobalHeap
, 0, BindData
);
698 if (Status
!= STATUS_SUCCESS
)
699 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
701 /* Set up Socket Data */
702 Socket
->SharedData
.State
= SocketBound
;
703 Socket
->TdiAddressHandle
= (HANDLE
)IOSB
.Information
;
705 if (Socket
->HelperEvents
& WSH_NOTIFY_BIND
)
707 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
709 Socket
->TdiAddressHandle
,
710 Socket
->TdiConnectionHandle
,
715 if (lpErrno
) *lpErrno
= Status
;
720 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
725 WSPListen(SOCKET Handle
,
729 IO_STATUS_BLOCK IOSB
;
730 AFD_LISTEN_DATA ListenData
;
731 PSOCKET_INFORMATION Socket
= NULL
;
735 /* Get the Socket Structure associate to this Socket*/
736 Socket
= GetSocketStructure(Handle
);
739 *lpErrno
= WSAENOTSOCK
;
743 if (Socket
->SharedData
.Listening
)
746 Status
= NtCreateEvent(&SockEvent
,
747 GENERIC_READ
| GENERIC_WRITE
,
752 if( !NT_SUCCESS(Status
) )
755 /* Set Up Listen Structure */
756 ListenData
.UseSAN
= FALSE
;
757 ListenData
.UseDelayedAcceptance
= Socket
->SharedData
.UseDelayedAcceptance
;
758 ListenData
.Backlog
= Backlog
;
761 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
766 IOCTL_AFD_START_LISTEN
,
772 /* Wait for return */
773 if (Status
== STATUS_PENDING
)
775 WaitForSingleObject(SockEvent
, INFINITE
);
776 Status
= IOSB
.Status
;
779 NtClose( SockEvent
);
781 if (Status
!= STATUS_SUCCESS
)
782 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
784 /* Set to Listening */
785 Socket
->SharedData
.Listening
= TRUE
;
787 if (Socket
->HelperEvents
& WSH_NOTIFY_LISTEN
)
789 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
791 Socket
->TdiAddressHandle
,
792 Socket
->TdiConnectionHandle
,
797 if (lpErrno
) *lpErrno
= Status
;
802 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
808 WSPSelect(IN
int nfds
,
809 IN OUT fd_set
*readfds OPTIONAL
,
810 IN OUT fd_set
*writefds OPTIONAL
,
811 IN OUT fd_set
*exceptfds OPTIONAL
,
812 IN
const struct timeval
*timeout OPTIONAL
,
815 IO_STATUS_BLOCK IOSB
;
816 PAFD_POLL_INFO PollInfo
;
818 LONG HandleCount
, OutCount
= 0;
819 ULONG PollBufferSize
;
824 LARGE_INTEGER Timeout
;
826 /* Find out how many sockets we have, and how large the buffer needs
829 HandleCount
= ( readfds
? readfds
->fd_count
: 0 ) +
830 ( writefds
? writefds
->fd_count
: 0 ) +
831 ( exceptfds
? exceptfds
->fd_count
: 0 );
833 if ( HandleCount
== 0 )
835 AFD_DbgPrint(MAX_TRACE
,("HandleCount: %d. Return SOCKET_ERROR\n",
837 if (lpErrno
) *lpErrno
= WSAEINVAL
;
841 PollBufferSize
= sizeof(*PollInfo
) + ((HandleCount
- 1) * sizeof(AFD_HANDLE
));
843 AFD_DbgPrint(MID_TRACE
,("HandleCount: %d BufferSize: %d\n",
844 HandleCount
, PollBufferSize
));
846 /* Convert Timeout to NT Format */
849 Timeout
.u
.LowPart
= -1;
850 Timeout
.u
.HighPart
= 0x7FFFFFFF;
851 AFD_DbgPrint(MAX_TRACE
,("Infinite timeout\n"));
855 Timeout
= RtlEnlargedIntegerMultiply
856 ((timeout
->tv_sec
* 1000) + (timeout
->tv_usec
/ 1000), -10000);
857 /* Negative timeouts are illegal. Since the kernel represents an
858 * incremental timeout as a negative number, we check for a positive
861 if (Timeout
.QuadPart
> 0)
863 if (lpErrno
) *lpErrno
= WSAEINVAL
;
866 AFD_DbgPrint(MAX_TRACE
,("Timeout: Orig %d.%06d kernel %d\n",
867 timeout
->tv_sec
, timeout
->tv_usec
,
871 Status
= NtCreateEvent(&SockEvent
,
872 GENERIC_READ
| GENERIC_WRITE
,
877 if( !NT_SUCCESS(Status
) )
881 PollBuffer
= HeapAlloc(GlobalHeap
, 0, PollBufferSize
);
886 *lpErrno
= WSAEFAULT
;
891 PollInfo
= (PAFD_POLL_INFO
)PollBuffer
;
893 RtlZeroMemory( PollInfo
, PollBufferSize
);
895 /* Number of handles for AFD to Check */
896 PollInfo
->Exclusive
= FALSE
;
897 PollInfo
->Timeout
= Timeout
;
899 if (readfds
!= NULL
) {
900 for (i
= 0; i
< readfds
->fd_count
; i
++, j
++)
902 PollInfo
->Handles
[j
].Handle
= readfds
->fd_array
[i
];
903 PollInfo
->Handles
[j
].Events
= AFD_EVENT_RECEIVE
|
904 AFD_EVENT_DISCONNECT
|
910 if (writefds
!= NULL
)
912 for (i
= 0; i
< writefds
->fd_count
; i
++, j
++)
914 PollInfo
->Handles
[j
].Handle
= writefds
->fd_array
[i
];
915 PollInfo
->Handles
[j
].Events
= AFD_EVENT_SEND
| AFD_EVENT_CONNECT
;
918 if (exceptfds
!= NULL
)
920 for (i
= 0; i
< exceptfds
->fd_count
; i
++, j
++)
922 PollInfo
->Handles
[j
].Handle
= exceptfds
->fd_array
[i
];
923 PollInfo
->Handles
[j
].Events
= AFD_EVENT_OOB_RECEIVE
| AFD_EVENT_CONNECT_FAIL
;
927 PollInfo
->HandleCount
= j
;
928 PollBufferSize
= ((PCHAR
)&PollInfo
->Handles
[j
+1]) - ((PCHAR
)PollInfo
);
931 Status
= NtDeviceIoControlFile((HANDLE
)PollInfo
->Handles
[0].Handle
,
942 AFD_DbgPrint(MID_TRACE
,("DeviceIoControlFile => %x\n", Status
));
944 /* Wait for Completition */
945 if (Status
== STATUS_PENDING
)
947 WaitForSingleObject(SockEvent
, INFINITE
);
948 Status
= IOSB
.Status
;
951 /* Clear the Structures */
959 /* Loop through return structure */
960 HandleCount
= PollInfo
->HandleCount
;
962 /* Return in FDSET Format */
963 for (i
= 0; i
< HandleCount
; i
++)
965 HandleCounted
= FALSE
;
968 switch (PollInfo
->Handles
[i
].Events
& x
)
970 case AFD_EVENT_RECEIVE
:
971 case AFD_EVENT_DISCONNECT
:
972 case AFD_EVENT_ABORT
:
973 case AFD_EVENT_ACCEPT
:
974 case AFD_EVENT_CLOSE
:
975 AFD_DbgPrint(MID_TRACE
,("Event %x on handle %x\n",
976 PollInfo
->Handles
[i
].Events
,
977 PollInfo
->Handles
[i
].Handle
));
981 HandleCounted
= TRUE
;
984 FD_SET(PollInfo
->Handles
[i
].Handle
, readfds
);
987 case AFD_EVENT_CONNECT
:
988 AFD_DbgPrint(MID_TRACE
,("Event %x on handle %x\n",
989 PollInfo
->Handles
[i
].Events
,
990 PollInfo
->Handles
[i
].Handle
));
994 HandleCounted
= TRUE
;
997 FD_SET(PollInfo
->Handles
[i
].Handle
, writefds
);
999 case AFD_EVENT_OOB_RECEIVE
:
1000 case AFD_EVENT_CONNECT_FAIL
:
1001 AFD_DbgPrint(MID_TRACE
,("Event %x on handle %x\n",
1002 PollInfo
->Handles
[i
].Events
,
1003 PollInfo
->Handles
[i
].Handle
));
1004 if (! HandleCounted
)
1007 HandleCounted
= TRUE
;
1010 FD_SET(PollInfo
->Handles
[i
].Handle
, exceptfds
);
1016 HeapFree( GlobalHeap
, 0, PollBuffer
);
1017 NtClose( SockEvent
);
1021 switch( IOSB
.Status
)
1023 case STATUS_SUCCESS
:
1024 case STATUS_TIMEOUT
:
1028 *lpErrno
= WSAEINVAL
;
1031 AFD_DbgPrint(MID_TRACE
,("*lpErrno = %x\n", *lpErrno
));
1034 AFD_DbgPrint(MID_TRACE
,("%d events\n", OutCount
));
1041 WSPAccept(SOCKET Handle
,
1042 struct sockaddr
*SocketAddress
,
1043 int *SocketAddressLength
,
1044 LPCONDITIONPROC lpfnCondition
,
1045 DWORD dwCallbackData
,
1048 IO_STATUS_BLOCK IOSB
;
1049 PAFD_RECEIVED_ACCEPT_DATA ListenReceiveData
;
1050 AFD_ACCEPT_DATA AcceptData
;
1051 AFD_DEFER_ACCEPT_DATA DeferData
;
1052 AFD_PENDING_ACCEPT_DATA PendingAcceptData
;
1053 PSOCKET_INFORMATION Socket
= NULL
;
1055 struct fd_set ReadSet
;
1056 struct timeval Timeout
;
1057 PVOID PendingData
= NULL
;
1058 ULONG PendingDataLength
= 0;
1059 PVOID CalleeDataBuffer
;
1060 WSABUF CallerData
, CalleeID
, CallerID
, CalleeData
;
1061 PSOCKADDR RemoteAddress
= NULL
;
1064 WSAPROTOCOL_INFOW ProtocolInfo
;
1065 SOCKET AcceptSocket
;
1066 UCHAR ReceiveBuffer
[0x1A];
1069 Status
= NtCreateEvent(&SockEvent
,
1070 GENERIC_READ
| GENERIC_WRITE
,
1075 if( !NT_SUCCESS(Status
) )
1077 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1078 return INVALID_SOCKET
;
1081 /* Dynamic Structure...ugh */
1082 ListenReceiveData
= (PAFD_RECEIVED_ACCEPT_DATA
)ReceiveBuffer
;
1084 /* Get the Socket Structure associate to this Socket*/
1085 Socket
= GetSocketStructure(Handle
);
1089 *lpErrno
= WSAENOTSOCK
;
1090 return INVALID_SOCKET
;
1093 /* If this is non-blocking, make sure there's something for us to accept */
1095 FD_SET(Socket
->Handle
, &ReadSet
);
1099 if (WSPSelect(0, &ReadSet
, NULL
, NULL
, &Timeout
, lpErrno
) == SOCKET_ERROR
)
1102 return INVALID_SOCKET
;
1105 if (ReadSet
.fd_array
[0] != Socket
->Handle
)
1108 *lpErrno
= WSAEWOULDBLOCK
;
1109 return INVALID_SOCKET
;
1113 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1118 IOCTL_AFD_WAIT_FOR_LISTEN
,
1122 0xA + sizeof(*ListenReceiveData
));
1124 /* Wait for return */
1125 if (Status
== STATUS_PENDING
)
1127 WaitForSingleObject(SockEvent
, INFINITE
);
1128 Status
= IOSB
.Status
;
1131 if (!NT_SUCCESS(Status
))
1133 NtClose( SockEvent
);
1134 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1135 return INVALID_SOCKET
;
1138 if (lpfnCondition
!= NULL
)
1140 if ((Socket
->SharedData
.ServiceFlags1
& XP1_CONNECT_DATA
) != 0)
1142 /* Find out how much data is pending */
1143 PendingAcceptData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1144 PendingAcceptData
.ReturnSize
= TRUE
;
1147 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1152 IOCTL_AFD_GET_PENDING_CONNECT_DATA
,
1154 sizeof(PendingAcceptData
),
1156 sizeof(PendingAcceptData
));
1158 /* Wait for return */
1159 if (Status
== STATUS_PENDING
)
1161 WaitForSingleObject(SockEvent
, INFINITE
);
1162 Status
= IOSB
.Status
;
1165 if (!NT_SUCCESS(Status
))
1167 NtClose( SockEvent
);
1168 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1169 return INVALID_SOCKET
;
1172 /* How much data to allocate */
1173 PendingDataLength
= IOSB
.Information
;
1175 if (PendingDataLength
)
1177 /* Allocate needed space */
1178 PendingData
= HeapAlloc(GlobalHeap
, 0, PendingDataLength
);
1181 MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1182 return INVALID_SOCKET
;
1185 /* We want the data now */
1186 PendingAcceptData
.ReturnSize
= FALSE
;
1189 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1194 IOCTL_AFD_GET_PENDING_CONNECT_DATA
,
1196 sizeof(PendingAcceptData
),
1200 /* Wait for return */
1201 if (Status
== STATUS_PENDING
)
1203 WaitForSingleObject(SockEvent
, INFINITE
);
1204 Status
= IOSB
.Status
;
1207 if (!NT_SUCCESS(Status
))
1209 NtClose( SockEvent
);
1210 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1211 return INVALID_SOCKET
;
1216 if ((Socket
->SharedData
.ServiceFlags1
& XP1_QOS_SUPPORTED
) != 0)
1218 /* I don't support this yet */
1221 /* Build Callee ID */
1222 CalleeID
.buf
= (PVOID
)Socket
->LocalAddress
;
1223 CalleeID
.len
= Socket
->SharedData
.SizeOfLocalAddress
;
1225 RemoteAddress
= HeapAlloc(GlobalHeap
, 0, sizeof(*RemoteAddress
));
1228 MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1229 return INVALID_SOCKET
;
1232 /* Set up Address in SOCKADDR Format */
1233 RtlCopyMemory (RemoteAddress
,
1234 &ListenReceiveData
->Address
.Address
[0].AddressType
,
1235 sizeof(*RemoteAddress
));
1237 /* Build Caller ID */
1238 CallerID
.buf
= (PVOID
)RemoteAddress
;
1239 CallerID
.len
= sizeof(*RemoteAddress
);
1241 /* Build Caller Data */
1242 CallerData
.buf
= PendingData
;
1243 CallerData
.len
= PendingDataLength
;
1245 /* Check if socket supports Conditional Accept */
1246 if (Socket
->SharedData
.UseDelayedAcceptance
!= 0)
1248 /* Allocate Buffer for Callee Data */
1249 CalleeDataBuffer
= HeapAlloc(GlobalHeap
, 0, 4096);
1250 if (!CalleeDataBuffer
) {
1251 MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1252 return INVALID_SOCKET
;
1254 CalleeData
.buf
= CalleeDataBuffer
;
1255 CalleeData
.len
= 4096;
1264 /* Call the Condition Function */
1265 CallBack
= (lpfnCondition
)(&CallerID
,
1266 CallerData
.buf
== NULL
? NULL
: &CallerData
,
1270 CalleeData
.buf
== NULL
? NULL
: &CalleeData
,
1274 if (((CallBack
== CF_ACCEPT
) && GroupID
) != 0)
1276 /* TBD: Check for Validity */
1279 if (CallBack
== CF_ACCEPT
)
1281 if ((Socket
->SharedData
.ServiceFlags1
& XP1_QOS_SUPPORTED
) != 0)
1283 /* I don't support this yet */
1287 // SockSetConnectData Sockets(SocketID), IOCTL_AFD_SET_CONNECT_DATA, CalleeData.Buffer, CalleeData.BuffSize, 0
1292 /* Callback rejected. Build Defer Structure */
1293 DeferData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1294 DeferData
.RejectConnection
= (CallBack
== CF_REJECT
);
1297 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1302 IOCTL_AFD_DEFER_ACCEPT
,
1308 /* Wait for return */
1309 if (Status
== STATUS_PENDING
)
1311 WaitForSingleObject(SockEvent
, INFINITE
);
1312 Status
= IOSB
.Status
;
1315 NtClose( SockEvent
);
1317 if (!NT_SUCCESS(Status
))
1319 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1320 return INVALID_SOCKET
;
1323 if (CallBack
== CF_REJECT
)
1325 *lpErrno
= WSAECONNREFUSED
;
1326 return INVALID_SOCKET
;
1330 *lpErrno
= WSAECONNREFUSED
;
1331 return INVALID_SOCKET
;
1336 /* Create a new Socket */
1337 ProtocolInfo
.dwCatalogEntryId
= Socket
->SharedData
.CatalogEntryId
;
1338 ProtocolInfo
.dwServiceFlags1
= Socket
->SharedData
.ServiceFlags1
;
1339 ProtocolInfo
.dwProviderFlags
= Socket
->SharedData
.ProviderFlags
;
1341 AcceptSocket
= WSPSocket (Socket
->SharedData
.AddressFamily
,
1342 Socket
->SharedData
.SocketType
,
1343 Socket
->SharedData
.Protocol
,
1346 Socket
->SharedData
.CreateFlags
,
1348 if (AcceptSocket
== INVALID_SOCKET
)
1349 return INVALID_SOCKET
;
1351 /* Set up the Accept Structure */
1352 AcceptData
.ListenHandle
= (HANDLE
)AcceptSocket
;
1353 AcceptData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1355 /* Send IOCTL to Accept */
1356 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1367 /* Wait for return */
1368 if (Status
== STATUS_PENDING
)
1370 WaitForSingleObject(SockEvent
, INFINITE
);
1371 Status
= IOSB
.Status
;
1374 if (!NT_SUCCESS(Status
))
1377 WSPCloseSocket( AcceptSocket
, lpErrno
);
1378 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1379 return INVALID_SOCKET
;
1382 /* Return Address in SOCKADDR FORMAT */
1385 RtlCopyMemory (SocketAddress
,
1386 &ListenReceiveData
->Address
.Address
[0].AddressType
,
1387 sizeof(*RemoteAddress
));
1388 if( SocketAddressLength
)
1389 *SocketAddressLength
= ListenReceiveData
->Address
.Address
[0].AddressLength
;
1392 NtClose( SockEvent
);
1394 /* Re-enable Async Event */
1395 SockReenableAsyncSelectEvent(Socket
, FD_ACCEPT
);
1397 AFD_DbgPrint(MID_TRACE
,("Socket %x\n", AcceptSocket
));
1399 if (Status
== STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_ACCEPT
))
1401 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
1403 Socket
->TdiAddressHandle
,
1404 Socket
->TdiConnectionHandle
,
1409 if (lpErrno
) *lpErrno
= Status
;
1410 return INVALID_SOCKET
;
1417 return AcceptSocket
;
1422 WSPConnect(SOCKET Handle
,
1423 const struct sockaddr
* SocketAddress
,
1424 int SocketAddressLength
,
1425 LPWSABUF lpCallerData
,
1426 LPWSABUF lpCalleeData
,
1431 IO_STATUS_BLOCK IOSB
;
1432 PAFD_CONNECT_INFO ConnectInfo
;
1433 PSOCKET_INFORMATION Socket
= NULL
;
1435 UCHAR ConnectBuffer
[0x22];
1436 ULONG ConnectDataLength
;
1437 ULONG InConnectDataLength
;
1438 INT BindAddressLength
;
1439 PSOCKADDR BindAddress
;
1442 Status
= NtCreateEvent(&SockEvent
,
1443 GENERIC_READ
| GENERIC_WRITE
,
1448 if( !NT_SUCCESS(Status
) )
1451 AFD_DbgPrint(MID_TRACE
,("Called\n"));
1453 /* Get the Socket Structure associate to this Socket*/
1454 Socket
= GetSocketStructure(Handle
);
1458 *lpErrno
= WSAENOTSOCK
;
1459 return SOCKET_ERROR
;
1463 if (Socket
->SharedData
.State
== SocketOpen
)
1465 /* Get the Wildcard Address */
1466 BindAddressLength
= Socket
->HelperData
->MaxWSAddressLength
;
1467 BindAddress
= HeapAlloc(GetProcessHeap(), 0, BindAddressLength
);
1470 MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1471 return INVALID_SOCKET
;
1473 Socket
->HelperData
->WSHGetWildcardSockaddr (Socket
->HelperContext
,
1475 &BindAddressLength
);
1477 if (WSPBind(Handle
, BindAddress
, BindAddressLength
, lpErrno
) == SOCKET_ERROR
)
1478 return INVALID_SOCKET
;
1481 /* Set the Connect Data */
1482 if (lpCallerData
!= NULL
)
1484 ConnectDataLength
= lpCallerData
->len
;
1485 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1490 IOCTL_AFD_SET_CONNECT_DATA
,
1495 /* Wait for return */
1496 if (Status
== STATUS_PENDING
)
1498 WaitForSingleObject(SockEvent
, INFINITE
);
1499 Status
= IOSB
.Status
;
1502 if (Status
!= STATUS_SUCCESS
)
1506 /* Dynamic Structure...ugh */
1507 ConnectInfo
= (PAFD_CONNECT_INFO
)ConnectBuffer
;
1509 /* Set up Address in TDI Format */
1510 ConnectInfo
->RemoteAddress
.TAAddressCount
= 1;
1511 ConnectInfo
->RemoteAddress
.Address
[0].AddressLength
= SocketAddressLength
- sizeof(SocketAddress
->sa_family
);
1512 ConnectInfo
->RemoteAddress
.Address
[0].AddressType
= SocketAddress
->sa_family
;
1513 RtlCopyMemory (ConnectInfo
->RemoteAddress
.Address
[0].Address
,
1514 SocketAddress
->sa_data
,
1515 SocketAddressLength
- sizeof(SocketAddress
->sa_family
));
1518 * Disable FD_WRITE and FD_CONNECT
1519 * The latter fixes a race condition where the FD_CONNECT is re-enabled
1520 * at the end of this function right after the Async Thread disables it.
1521 * This should only happen at the *next* WSPConnect
1523 if (Socket
->SharedData
.AsyncEvents
& FD_CONNECT
)
1525 Socket
->SharedData
.AsyncDisabledEvents
|= FD_CONNECT
| FD_WRITE
;
1528 /* Tell AFD that we want Connection Data back, have it allocate a buffer */
1529 if (lpCalleeData
!= NULL
)
1531 InConnectDataLength
= lpCalleeData
->len
;
1532 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1537 IOCTL_AFD_SET_CONNECT_DATA_SIZE
,
1538 &InConnectDataLength
,
1539 sizeof(InConnectDataLength
),
1543 /* Wait for return */
1544 if (Status
== STATUS_PENDING
)
1546 WaitForSingleObject(SockEvent
, INFINITE
);
1547 Status
= IOSB
.Status
;
1550 if (Status
!= STATUS_SUCCESS
)
1554 /* AFD doesn't seem to care if these are invalid, but let's 0 them anyways */
1555 ConnectInfo
->Root
= 0;
1556 ConnectInfo
->UseSAN
= FALSE
;
1557 ConnectInfo
->Unknown
= 0;
1559 /* FIXME: Handle Async Connect */
1560 if (Socket
->SharedData
.NonBlocking
)
1562 AFD_DbgPrint(MIN_TRACE
, ("Async Connect UNIMPLEMENTED!\n"));
1566 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1576 /* Wait for return */
1577 if (Status
== STATUS_PENDING
)
1579 WaitForSingleObject(SockEvent
, INFINITE
);
1580 Status
= IOSB
.Status
;
1583 if (Status
!= STATUS_SUCCESS
)
1586 Socket
->TdiConnectionHandle
= (HANDLE
)IOSB
.Information
;
1588 /* Get any pending connect data */
1589 if (lpCalleeData
!= NULL
)
1591 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1596 IOCTL_AFD_GET_CONNECT_DATA
,
1601 /* Wait for return */
1602 if (Status
== STATUS_PENDING
)
1604 WaitForSingleObject(SockEvent
, INFINITE
);
1605 Status
= IOSB
.Status
;
1609 AFD_DbgPrint(MID_TRACE
,("Ending\n"));
1612 /* Re-enable Async Event */
1613 SockReenableAsyncSelectEvent(Socket
, FD_WRITE
);
1615 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
1616 SockReenableAsyncSelectEvent(Socket
, FD_CONNECT
);
1618 NtClose( SockEvent
);
1620 if (Status
== STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_CONNECT
))
1622 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
1624 Socket
->TdiAddressHandle
,
1625 Socket
->TdiConnectionHandle
,
1626 WSH_NOTIFY_CONNECT
);
1630 if (lpErrno
) *lpErrno
= Status
;
1631 return SOCKET_ERROR
;
1634 else if (Status
!= STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_CONNECT_ERROR
))
1636 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
1638 Socket
->TdiAddressHandle
,
1639 Socket
->TdiConnectionHandle
,
1640 WSH_NOTIFY_CONNECT_ERROR
);
1644 if (lpErrno
) *lpErrno
= Status
;
1645 return SOCKET_ERROR
;
1649 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1653 WSPShutdown(SOCKET Handle
,
1658 IO_STATUS_BLOCK IOSB
;
1659 AFD_DISCONNECT_INFO DisconnectInfo
;
1660 PSOCKET_INFORMATION Socket
= NULL
;
1664 Status
= NtCreateEvent(&SockEvent
,
1665 GENERIC_READ
| GENERIC_WRITE
,
1670 if( !NT_SUCCESS(Status
) )
1673 AFD_DbgPrint(MID_TRACE
,("Called\n"));
1675 /* Get the Socket Structure associate to this Socket*/
1676 Socket
= GetSocketStructure(Handle
);
1680 *lpErrno
= WSAENOTSOCK
;
1681 return SOCKET_ERROR
;
1684 /* Set AFD Disconnect Type */
1688 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_RECV
;
1689 Socket
->SharedData
.ReceiveShutdown
= TRUE
;
1692 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_SEND
;
1693 Socket
->SharedData
.SendShutdown
= TRUE
;
1696 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_RECV
| AFD_DISCONNECT_SEND
;
1697 Socket
->SharedData
.ReceiveShutdown
= TRUE
;
1698 Socket
->SharedData
.SendShutdown
= TRUE
;
1702 DisconnectInfo
.Timeout
= RtlConvertLongToLargeInteger(-1);
1705 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1710 IOCTL_AFD_DISCONNECT
,
1712 sizeof(DisconnectInfo
),
1716 /* Wait for return */
1717 if (Status
== STATUS_PENDING
)
1719 WaitForSingleObject(SockEvent
, INFINITE
);
1720 Status
= IOSB
.Status
;
1723 AFD_DbgPrint(MID_TRACE
,("Ending\n"));
1725 NtClose( SockEvent
);
1727 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1733 WSPGetSockName(IN SOCKET Handle
,
1734 OUT LPSOCKADDR Name
,
1735 IN OUT LPINT NameLength
,
1738 IO_STATUS_BLOCK IOSB
;
1739 ULONG TdiAddressSize
;
1740 PTDI_ADDRESS_INFO TdiAddress
;
1741 PTRANSPORT_ADDRESS SocketAddress
;
1742 PSOCKET_INFORMATION Socket
= NULL
;
1746 Status
= NtCreateEvent(&SockEvent
,
1747 GENERIC_READ
| GENERIC_WRITE
,
1752 if( !NT_SUCCESS(Status
) )
1753 return SOCKET_ERROR
;
1755 /* Get the Socket Structure associate to this Socket*/
1756 Socket
= GetSocketStructure(Handle
);
1760 *lpErrno
= WSAENOTSOCK
;
1761 return SOCKET_ERROR
;
1764 /* Allocate a buffer for the address */
1766 sizeof(TRANSPORT_ADDRESS
) + Socket
->SharedData
.SizeOfLocalAddress
;
1767 TdiAddress
= HeapAlloc(GlobalHeap
, 0, TdiAddressSize
);
1769 if ( TdiAddress
== NULL
)
1771 NtClose( SockEvent
);
1772 *lpErrno
= WSAENOBUFS
;
1773 return SOCKET_ERROR
;
1776 SocketAddress
= &TdiAddress
->Address
;
1779 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1784 IOCTL_AFD_GET_SOCK_NAME
,
1790 /* Wait for return */
1791 if (Status
== STATUS_PENDING
)
1793 WaitForSingleObject(SockEvent
, INFINITE
);
1794 Status
= IOSB
.Status
;
1797 NtClose( SockEvent
);
1799 if (NT_SUCCESS(Status
))
1801 if (*NameLength
>= SocketAddress
->Address
[0].AddressLength
)
1803 Name
->sa_family
= SocketAddress
->Address
[0].AddressType
;
1804 RtlCopyMemory (Name
->sa_data
,
1805 SocketAddress
->Address
[0].Address
,
1806 SocketAddress
->Address
[0].AddressLength
);
1807 *NameLength
= 2 + SocketAddress
->Address
[0].AddressLength
;
1808 AFD_DbgPrint (MID_TRACE
, ("NameLength %d Address: %x Port %x\n",
1809 *NameLength
, ((struct sockaddr_in
*)Name
)->sin_addr
.s_addr
,
1810 ((struct sockaddr_in
*)Name
)->sin_port
));
1811 HeapFree(GlobalHeap
, 0, TdiAddress
);
1816 HeapFree(GlobalHeap
, 0, TdiAddress
);
1817 *lpErrno
= WSAEFAULT
;
1818 return SOCKET_ERROR
;
1822 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
1828 WSPGetPeerName(IN SOCKET s
,
1829 OUT LPSOCKADDR Name
,
1830 IN OUT LPINT NameLength
,
1833 IO_STATUS_BLOCK IOSB
;
1834 ULONG TdiAddressSize
;
1835 PTRANSPORT_ADDRESS SocketAddress
;
1836 PSOCKET_INFORMATION Socket
= NULL
;
1840 Status
= NtCreateEvent(&SockEvent
,
1841 GENERIC_READ
| GENERIC_WRITE
,
1846 if( !NT_SUCCESS(Status
) )
1847 return SOCKET_ERROR
;
1849 /* Get the Socket Structure associate to this Socket*/
1850 Socket
= GetSocketStructure(s
);
1854 *lpErrno
= WSAENOTSOCK
;
1855 return SOCKET_ERROR
;
1858 /* Allocate a buffer for the address */
1859 TdiAddressSize
= sizeof(TRANSPORT_ADDRESS
) + *NameLength
;
1860 SocketAddress
= HeapAlloc(GlobalHeap
, 0, TdiAddressSize
);
1862 if ( SocketAddress
== NULL
)
1864 NtClose( SockEvent
);
1865 *lpErrno
= WSAENOBUFS
;
1866 return SOCKET_ERROR
;
1870 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1875 IOCTL_AFD_GET_PEER_NAME
,
1881 /* Wait for return */
1882 if (Status
== STATUS_PENDING
)
1884 WaitForSingleObject(SockEvent
, INFINITE
);
1885 Status
= IOSB
.Status
;
1888 NtClose( SockEvent
);
1890 if (NT_SUCCESS(Status
))
1892 if (*NameLength
>= SocketAddress
->Address
[0].AddressLength
)
1894 Name
->sa_family
= SocketAddress
->Address
[0].AddressType
;
1895 RtlCopyMemory (Name
->sa_data
,
1896 SocketAddress
->Address
[0].Address
,
1897 SocketAddress
->Address
[0].AddressLength
);
1898 *NameLength
= 2 + SocketAddress
->Address
[0].AddressLength
;
1899 AFD_DbgPrint (MID_TRACE
, ("NameLength %d Address: %s Port %x\n",
1900 *NameLength
, ((struct sockaddr_in
*)Name
)->sin_addr
.s_addr
,
1901 ((struct sockaddr_in
*)Name
)->sin_port
));
1902 HeapFree(GlobalHeap
, 0, SocketAddress
);
1907 HeapFree(GlobalHeap
, 0, SocketAddress
);
1908 *lpErrno
= WSAEFAULT
;
1909 return SOCKET_ERROR
;
1913 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
1918 WSPIoctl(IN SOCKET Handle
,
1919 IN DWORD dwIoControlCode
,
1920 IN LPVOID lpvInBuffer
,
1921 IN DWORD cbInBuffer
,
1922 OUT LPVOID lpvOutBuffer
,
1923 IN DWORD cbOutBuffer
,
1924 OUT LPDWORD lpcbBytesReturned
,
1925 IN LPWSAOVERLAPPED lpOverlapped
,
1926 IN LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine
,
1927 IN LPWSATHREADID lpThreadId
,
1930 PSOCKET_INFORMATION Socket
= NULL
;
1932 /* Get the Socket Structure associate to this Socket*/
1933 Socket
= GetSocketStructure(Handle
);
1936 *lpErrno
= WSAENOTSOCK
;
1937 return SOCKET_ERROR
;
1940 switch( dwIoControlCode
)
1943 if( cbInBuffer
< sizeof(INT
) || IS_INTRESOURCE(lpvInBuffer
) )
1945 *lpErrno
= WSAEFAULT
;
1946 return SOCKET_ERROR
;
1948 Socket
->SharedData
.NonBlocking
= *((PULONG
)lpvInBuffer
) ? 1 : 0;
1949 return SetSocketInformation(Socket
, AFD_INFO_BLOCKING_MODE
, (PULONG
)lpvInBuffer
, NULL
);
1951 if( cbOutBuffer
< sizeof(INT
) || IS_INTRESOURCE(lpvOutBuffer
) )
1953 *lpErrno
= WSAEFAULT
;
1954 return SOCKET_ERROR
;
1956 return GetSocketInformation(Socket
, AFD_INFO_RECEIVE_CONTENT_SIZE
, (PULONG
)lpvOutBuffer
, NULL
);
1958 *lpErrno
= WSAEINVAL
;
1959 return SOCKET_ERROR
;
1966 WSPGetSockOpt(IN SOCKET Handle
,
1969 OUT CHAR FAR
* OptionValue
,
1970 IN OUT LPINT OptionLength
,
1973 PSOCKET_INFORMATION Socket
= NULL
;
1978 /* Get the Socket Structure associate to this Socket*/
1979 Socket
= GetSocketStructure(Handle
);
1982 *lpErrno
= WSAENOTSOCK
;
1983 return SOCKET_ERROR
;
1986 AFD_DbgPrint(MID_TRACE
, ("Called\n"));
1994 Buffer
= &Socket
->SharedData
.SocketType
;
1995 BufferSize
= sizeof(INT
);
1999 Buffer
= &Socket
->SharedData
.SizeOfRecvBuffer
;
2000 BufferSize
= sizeof(INT
);
2004 Buffer
= &Socket
->SharedData
.SizeOfSendBuffer
;
2005 BufferSize
= sizeof(INT
);
2009 BoolBuffer
= Socket
->SharedData
.Listening
;
2010 Buffer
= &BoolBuffer
;
2011 BufferSize
= sizeof(BOOLEAN
);
2015 BoolBuffer
= Socket
->SharedData
.Broadcast
;
2016 Buffer
= &BoolBuffer
;
2017 BufferSize
= sizeof(BOOLEAN
);
2021 BoolBuffer
= Socket
->SharedData
.Debug
;
2022 Buffer
= &BoolBuffer
;
2023 BufferSize
= sizeof(BOOLEAN
);
2026 /* case SO_CONDITIONAL_ACCEPT: */
2031 case SO_GROUP_PRIORITY
:
2034 case SO_MAX_MSG_SIZE
:
2036 case SO_PROTOCOL_INFO
:
2038 AFD_DbgPrint(MID_TRACE
, ("Unimplemented option (%x)\n",
2042 *lpErrno
= WSAEINVAL
;
2043 return SOCKET_ERROR
;
2046 if (*OptionLength
< BufferSize
)
2048 *lpErrno
= WSAEFAULT
;
2049 *OptionLength
= BufferSize
;
2050 return SOCKET_ERROR
;
2052 RtlCopyMemory(OptionValue
, Buffer
, BufferSize
);
2056 case IPPROTO_TCP
: /* FIXME */
2058 *lpErrno
= Socket
->HelperData
->WSHGetSocketInformation(Socket
->HelperContext
,
2060 Socket
->TdiAddressHandle
,
2061 Socket
->TdiConnectionHandle
,
2065 (LPINT
)OptionLength
);
2066 return (*lpErrno
== 0) ? 0 : SOCKET_ERROR
;
2076 IN CONST CHAR FAR
* optval
,
2080 PSOCKET_INFORMATION Socket
;
2082 /* Get the Socket Structure associate to this Socket*/
2083 Socket
= GetSocketStructure(s
);
2086 *lpErrno
= WSAENOTSOCK
;
2087 return SOCKET_ERROR
;
2091 /* FIXME: We should handle some cases here */
2094 *lpErrno
= Socket
->HelperData
->WSHSetSocketInformation(Socket
->HelperContext
,
2096 Socket
->TdiAddressHandle
,
2097 Socket
->TdiConnectionHandle
,
2102 return (*lpErrno
== 0) ? 0 : SOCKET_ERROR
;
2106 * FUNCTION: Initialize service provider for a client
2108 * wVersionRequested = Highest WinSock SPI version that the caller can use
2109 * lpWSPData = Address of WSPDATA structure to initialize
2110 * lpProtocolInfo = Pointer to structure that defines the desired protocol
2111 * UpcallTable = Pointer to upcall table of the WinSock DLL
2112 * lpProcTable = Address of procedure table to initialize
2114 * Status of operation
2118 WSPStartup(IN WORD wVersionRequested
,
2119 OUT LPWSPDATA lpWSPData
,
2120 IN LPWSAPROTOCOL_INFOW lpProtocolInfo
,
2121 IN WSPUPCALLTABLE UpcallTable
,
2122 OUT LPWSPPROC_TABLE lpProcTable
)
2127 AFD_DbgPrint(MAX_TRACE
, ("wVersionRequested (0x%X) \n", wVersionRequested
));
2129 Upcalls
= UpcallTable
;
2131 if (Status
== NO_ERROR
)
2133 lpProcTable
->lpWSPAccept
= WSPAccept
;
2134 lpProcTable
->lpWSPAddressToString
= WSPAddressToString
;
2135 lpProcTable
->lpWSPAsyncSelect
= WSPAsyncSelect
;
2136 lpProcTable
->lpWSPBind
= WSPBind
;
2137 lpProcTable
->lpWSPCancelBlockingCall
= WSPCancelBlockingCall
;
2138 lpProcTable
->lpWSPCleanup
= WSPCleanup
;
2139 lpProcTable
->lpWSPCloseSocket
= WSPCloseSocket
;
2140 lpProcTable
->lpWSPConnect
= WSPConnect
;
2141 lpProcTable
->lpWSPDuplicateSocket
= WSPDuplicateSocket
;
2142 lpProcTable
->lpWSPEnumNetworkEvents
= WSPEnumNetworkEvents
;
2143 lpProcTable
->lpWSPEventSelect
= WSPEventSelect
;
2144 lpProcTable
->lpWSPGetOverlappedResult
= WSPGetOverlappedResult
;
2145 lpProcTable
->lpWSPGetPeerName
= WSPGetPeerName
;
2146 lpProcTable
->lpWSPGetSockName
= WSPGetSockName
;
2147 lpProcTable
->lpWSPGetSockOpt
= WSPGetSockOpt
;
2148 lpProcTable
->lpWSPGetQOSByName
= WSPGetQOSByName
;
2149 lpProcTable
->lpWSPIoctl
= WSPIoctl
;
2150 lpProcTable
->lpWSPJoinLeaf
= WSPJoinLeaf
;
2151 lpProcTable
->lpWSPListen
= WSPListen
;
2152 lpProcTable
->lpWSPRecv
= WSPRecv
;
2153 lpProcTable
->lpWSPRecvDisconnect
= WSPRecvDisconnect
;
2154 lpProcTable
->lpWSPRecvFrom
= WSPRecvFrom
;
2155 lpProcTable
->lpWSPSelect
= WSPSelect
;
2156 lpProcTable
->lpWSPSend
= WSPSend
;
2157 lpProcTable
->lpWSPSendDisconnect
= WSPSendDisconnect
;
2158 lpProcTable
->lpWSPSendTo
= WSPSendTo
;
2159 lpProcTable
->lpWSPSetSockOpt
= WSPSetSockOpt
;
2160 lpProcTable
->lpWSPShutdown
= WSPShutdown
;
2161 lpProcTable
->lpWSPSocket
= WSPSocket
;
2162 lpProcTable
->lpWSPStringToAddress
= WSPStringToAddress
;
2163 lpWSPData
->wVersion
= MAKEWORD(2, 2);
2164 lpWSPData
->wHighVersion
= MAKEWORD(2, 2);
2167 AFD_DbgPrint(MAX_TRACE
, ("Status (%d).\n", Status
));
2174 * FUNCTION: Cleans up service provider for a client
2176 * lpErrno = Address of buffer for error information
2178 * 0 if successful, or SOCKET_ERROR if not
2182 WSPCleanup(OUT LPINT lpErrno
)
2185 AFD_DbgPrint(MAX_TRACE
, ("\n"));
2186 AFD_DbgPrint(MAX_TRACE
, ("Leaving.\n"));
2187 *lpErrno
= NO_ERROR
;
2195 GetSocketInformation(PSOCKET_INFORMATION Socket
,
2196 ULONG AfdInformationClass
,
2197 PULONG Ulong OPTIONAL
,
2198 PLARGE_INTEGER LargeInteger OPTIONAL
)
2200 IO_STATUS_BLOCK IOSB
;
2205 Status
= NtCreateEvent(&SockEvent
,
2206 GENERIC_READ
| GENERIC_WRITE
,
2211 if( !NT_SUCCESS(Status
) )
2214 /* Set Info Class */
2215 InfoData
.InformationClass
= AfdInformationClass
;
2218 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
2229 /* Wait for return */
2230 if (Status
== STATUS_PENDING
)
2232 WaitForSingleObject(SockEvent
, INFINITE
);
2233 Status
= IOSB
.Status
;
2236 if (Status
!= STATUS_SUCCESS
)
2239 /* Return Information */
2242 *Ulong
= InfoData
.Information
.Ulong
;
2244 if (LargeInteger
!= NULL
)
2246 *LargeInteger
= InfoData
.Information
.LargeInteger
;
2249 NtClose( SockEvent
);
2257 SetSocketInformation(PSOCKET_INFORMATION Socket
,
2258 ULONG AfdInformationClass
,
2259 PULONG Ulong OPTIONAL
,
2260 PLARGE_INTEGER LargeInteger OPTIONAL
)
2262 IO_STATUS_BLOCK IOSB
;
2267 Status
= NtCreateEvent(&SockEvent
,
2268 GENERIC_READ
| GENERIC_WRITE
,
2273 if( !NT_SUCCESS(Status
) )
2276 /* Set Info Class */
2277 InfoData
.InformationClass
= AfdInformationClass
;
2279 /* Set Information */
2282 InfoData
.Information
.Ulong
= *Ulong
;
2284 if (LargeInteger
!= NULL
)
2286 InfoData
.Information
.LargeInteger
= *LargeInteger
;
2289 AFD_DbgPrint(MID_TRACE
,("XXX Info %x (Data %x)\n",
2290 AfdInformationClass
, *Ulong
));
2293 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
2304 /* Wait for return */
2305 if (Status
== STATUS_PENDING
)
2307 WaitForSingleObject(SockEvent
, INFINITE
);
2308 Status
= IOSB
.Status
;
2311 NtClose( SockEvent
);
2313 return Status
== STATUS_SUCCESS
? 0 : -1;
2318 GetSocketStructure(SOCKET Handle
)
2320 PSOCKET_INFORMATION CurrentSocket
;
2322 EnterCriticalSection(&SocketListLock
);
2324 CurrentSocket
= SocketListHead
;
2325 while (CurrentSocket
)
2327 if (CurrentSocket
->Handle
== Handle
)
2329 LeaveCriticalSection(&SocketListLock
);
2330 return CurrentSocket
;
2333 CurrentSocket
= CurrentSocket
->NextSocket
;
2336 LeaveCriticalSection(&SocketListLock
);
2341 int CreateContext(PSOCKET_INFORMATION Socket
)
2343 IO_STATUS_BLOCK IOSB
;
2344 SOCKET_CONTEXT ContextData
;
2348 Status
= NtCreateEvent(&SockEvent
,
2349 GENERIC_READ
| GENERIC_WRITE
,
2354 if( !NT_SUCCESS(Status
) )
2357 /* Create Context */
2358 ContextData
.SharedData
= Socket
->SharedData
;
2359 ContextData
.SizeOfHelperData
= 0;
2360 RtlCopyMemory (&ContextData
.LocalAddress
,
2361 Socket
->LocalAddress
,
2362 Socket
->SharedData
.SizeOfLocalAddress
);
2363 RtlCopyMemory (&ContextData
.RemoteAddress
,
2364 Socket
->RemoteAddress
,
2365 Socket
->SharedData
.SizeOfRemoteAddress
);
2368 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
2373 IOCTL_AFD_SET_CONTEXT
,
2375 sizeof(ContextData
),
2379 /* Wait for Completition */
2380 if (Status
== STATUS_PENDING
)
2382 WaitForSingleObject(SockEvent
, INFINITE
);
2383 Status
= IOSB
.Status
;
2386 NtClose( SockEvent
);
2388 return Status
== STATUS_SUCCESS
? 0 : -1;
2391 BOOLEAN
SockCreateOrReferenceAsyncThread(VOID
)
2393 HANDLE hAsyncThread
;
2394 DWORD AsyncThreadId
;
2396 OBJECT_HANDLE_ATTRIBUTE_INFORMATION HandleFlags
;
2399 /* Check if the Thread Already Exists */
2400 if (SockAsyncThreadRefCount
)
2405 /* Create the Completion Port */
2406 if (!SockAsyncCompletionPort
)
2408 Status
= NtCreateIoCompletion(&SockAsyncCompletionPort
,
2409 IO_COMPLETION_ALL_ACCESS
,
2411 2); // Allow 2 threads only
2413 /* Protect Handle */
2414 HandleFlags
.ProtectFromClose
= TRUE
;
2415 HandleFlags
.Inherit
= FALSE
;
2416 Status
= NtSetInformationObject(SockAsyncCompletionPort
,
2417 ObjectHandleFlagInformation
,
2419 sizeof(HandleFlags
));
2422 /* Create the Async Event */
2423 Status
= NtCreateEvent(&AsyncEvent
,
2429 /* Create the Async Thread */
2430 hAsyncThread
= CreateThread(NULL
,
2432 (LPTHREAD_START_ROUTINE
)SockAsyncThread
,
2437 /* Close the Handle */
2438 NtClose(hAsyncThread
);
2440 /* Increase the Reference Count */
2441 SockAsyncThreadRefCount
++;
2445 int SockAsyncThread(PVOID ThreadParam
)
2448 PASYNC_COMPLETION_ROUTINE AsyncCompletionRoutine
;
2449 IO_STATUS_BLOCK IOSB
;
2452 /* Make the Thread Higher Priority */
2453 SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_ABOVE_NORMAL
);
2455 /* Do a KQUEUE/WorkItem Style Loop, thanks to IoCompletion Ports */
2458 Status
= NtRemoveIoCompletion (SockAsyncCompletionPort
,
2459 (PVOID
*)&AsyncCompletionRoutine
,
2463 /* Call the Async Function */
2464 if (NT_SUCCESS(Status
))
2466 (*AsyncCompletionRoutine
)(AsyncContext
, &IOSB
);
2470 /* It Failed, sleep for a second */
2473 } while ((Status
!= STATUS_TIMEOUT
));
2475 /* The Thread has Ended */
2479 BOOLEAN
SockGetAsyncSelectHelperAfdHandle(VOID
)
2481 UNICODE_STRING AfdHelper
;
2482 OBJECT_ATTRIBUTES ObjectAttributes
;
2483 IO_STATUS_BLOCK IoSb
;
2485 FILE_COMPLETION_INFORMATION CompletionInfo
;
2486 OBJECT_HANDLE_ATTRIBUTE_INFORMATION HandleFlags
;
2488 /* First, make sure we're not already intialized */
2489 if (SockAsyncHelperAfdHandle
)
2494 /* Set up Handle Name and Object */
2495 RtlInitUnicodeString(&AfdHelper
, L
"\\Device\\Afd\\AsyncSelectHlp" );
2496 InitializeObjectAttributes(&ObjectAttributes
,
2498 OBJ_INHERIT
| OBJ_CASE_INSENSITIVE
,
2502 /* Open the Handle to AFD */
2503 Status
= NtCreateFile(&SockAsyncHelperAfdHandle
,
2504 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
,
2509 FILE_SHARE_READ
| FILE_SHARE_WRITE
,
2516 * Now Set up the Completion Port Information
2517 * This means that whenever a Poll is finished, the routine will be executed
2519 CompletionInfo
.Port
= SockAsyncCompletionPort
;
2520 CompletionInfo
.Key
= SockAsyncSelectCompletionRoutine
;
2521 Status
= NtSetInformationFile(SockAsyncHelperAfdHandle
,
2524 sizeof(CompletionInfo
),
2525 FileCompletionInformation
);
2528 /* Protect the Handle */
2529 HandleFlags
.ProtectFromClose
= TRUE
;
2530 HandleFlags
.Inherit
= FALSE
;
2531 Status
= NtSetInformationObject(SockAsyncCompletionPort
,
2532 ObjectHandleFlagInformation
,
2534 sizeof(HandleFlags
));
2537 /* Set this variable to true so that Send/Recv/Accept will know wether to renable disabled events */
2538 SockAsyncSelectCalled
= TRUE
;
2542 VOID
SockAsyncSelectCompletionRoutine(PVOID Context
, PIO_STATUS_BLOCK IoStatusBlock
)
2545 PASYNC_DATA AsyncData
= Context
;
2546 PSOCKET_INFORMATION Socket
;
2549 /* Get the Socket */
2550 Socket
= AsyncData
->ParentSocket
;
2552 /* Check if the Sequence Number Changed behind our back */
2553 if (AsyncData
->SequenceNumber
!= Socket
->SharedData
.SequenceNumber
)
2558 /* Check we were manually called b/c of a failure */
2559 if (!NT_SUCCESS(IoStatusBlock
->Status
))
2561 /* FIXME: Perform Upcall */
2565 for (x
= 1; x
; x
<<=1)
2567 switch (AsyncData
->AsyncSelectInfo
.Handles
[0].Events
& x
)
2569 case AFD_EVENT_RECEIVE
:
2570 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_READ
) &&
2571 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_READ
))
2573 /* Make the Notifcation */
2574 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2575 Socket
->SharedData
.wMsg
,
2577 WSAMAKESELECTREPLY(FD_READ
, 0));
2578 /* Disable this event until the next read(); */
2579 Socket
->SharedData
.AsyncDisabledEvents
|= FD_READ
;
2583 case AFD_EVENT_OOB_RECEIVE
:
2584 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_OOB
) &&
2585 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_OOB
))
2587 /* Make the Notifcation */
2588 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2589 Socket
->SharedData
.wMsg
,
2591 WSAMAKESELECTREPLY(FD_OOB
, 0));
2592 /* Disable this event until the next read(); */
2593 Socket
->SharedData
.AsyncDisabledEvents
|= FD_OOB
;
2597 case AFD_EVENT_SEND
:
2598 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_WRITE
) &&
2599 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_WRITE
))
2601 /* Make the Notifcation */
2602 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2603 Socket
->SharedData
.wMsg
,
2605 WSAMAKESELECTREPLY(FD_WRITE
, 0));
2606 /* Disable this event until the next write(); */
2607 Socket
->SharedData
.AsyncDisabledEvents
|= FD_WRITE
;
2611 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
2612 case AFD_EVENT_CONNECT
:
2613 case AFD_EVENT_CONNECT_FAIL
:
2614 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_CONNECT
) &&
2615 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_CONNECT
))
2617 /* Make the Notifcation */
2618 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2619 Socket
->SharedData
.wMsg
,
2621 WSAMAKESELECTREPLY(FD_CONNECT
, 0));
2622 /* Disable this event forever; */
2623 Socket
->SharedData
.AsyncDisabledEvents
|= FD_CONNECT
;
2627 case AFD_EVENT_ACCEPT
:
2628 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_ACCEPT
) &&
2629 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_ACCEPT
))
2631 /* Make the Notifcation */
2632 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2633 Socket
->SharedData
.wMsg
,
2635 WSAMAKESELECTREPLY(FD_ACCEPT
, 0));
2636 /* Disable this event until the next accept(); */
2637 Socket
->SharedData
.AsyncDisabledEvents
|= FD_ACCEPT
;
2641 case AFD_EVENT_DISCONNECT
:
2642 case AFD_EVENT_ABORT
:
2643 case AFD_EVENT_CLOSE
:
2644 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_CLOSE
) &&
2645 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_CLOSE
))
2647 /* Make the Notifcation */
2648 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2649 Socket
->SharedData
.wMsg
,
2651 WSAMAKESELECTREPLY(FD_CLOSE
, 0));
2652 /* Disable this event forever; */
2653 Socket
->SharedData
.AsyncDisabledEvents
|= FD_CLOSE
;
2656 /* FIXME: Support QOS */
2660 /* Check if there are any events left for us to check */
2661 if ((Socket
->SharedData
.AsyncEvents
& (~Socket
->SharedData
.AsyncDisabledEvents
)) == 0 )
2667 SockProcessAsyncSelect(Socket
, AsyncData
);
2671 VOID
SockProcessAsyncSelect(PSOCKET_INFORMATION Socket
, PASYNC_DATA AsyncData
)
2674 ULONG lNetworkEvents
;
2677 /* Set up the Async Data Event Info */
2678 AsyncData
->AsyncSelectInfo
.Timeout
.HighPart
= 0x7FFFFFFF;
2679 AsyncData
->AsyncSelectInfo
.Timeout
.LowPart
= 0xFFFFFFFF;
2680 AsyncData
->AsyncSelectInfo
.HandleCount
= 1;
2681 AsyncData
->AsyncSelectInfo
.Exclusive
= TRUE
;
2682 AsyncData
->AsyncSelectInfo
.Handles
[0].Handle
= Socket
->Handle
;
2683 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
= 0;
2685 /* Remove unwanted events */
2686 lNetworkEvents
= Socket
->SharedData
.AsyncEvents
& (~Socket
->SharedData
.AsyncDisabledEvents
);
2688 /* Set Events to wait for */
2689 if (lNetworkEvents
& FD_READ
)
2691 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_RECEIVE
;
2694 if (lNetworkEvents
& FD_WRITE
)
2696 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_SEND
;
2699 if (lNetworkEvents
& FD_OOB
)
2701 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_OOB_RECEIVE
;
2704 if (lNetworkEvents
& FD_ACCEPT
)
2706 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_ACCEPT
;
2709 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
2710 if (lNetworkEvents
& FD_CONNECT
)
2712 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_CONNECT
| AFD_EVENT_CONNECT_FAIL
;
2715 if (lNetworkEvents
& FD_CLOSE
)
2717 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_DISCONNECT
| AFD_EVENT_ABORT
| AFD_EVENT_CLOSE
;
2720 if (lNetworkEvents
& FD_QOS
)
2722 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_QOS
;
2725 if (lNetworkEvents
& FD_GROUP_QOS
)
2727 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_GROUP_QOS
;
2731 Status
= NtDeviceIoControlFile (SockAsyncHelperAfdHandle
,
2735 &AsyncData
->IoStatusBlock
,
2737 &AsyncData
->AsyncSelectInfo
,
2738 sizeof(AsyncData
->AsyncSelectInfo
),
2739 &AsyncData
->AsyncSelectInfo
,
2740 sizeof(AsyncData
->AsyncSelectInfo
));
2742 /* I/O Manager Won't call the completion routine, let's do it manually */
2743 if (NT_SUCCESS(Status
))
2749 AsyncData
->IoStatusBlock
.Status
= Status
;
2750 SockAsyncSelectCompletionRoutine(AsyncData
, &AsyncData
->IoStatusBlock
);
2754 VOID
SockProcessQueuedAsyncSelect(PVOID Context
, PIO_STATUS_BLOCK IoStatusBlock
)
2756 PASYNC_DATA AsyncData
= Context
;
2757 BOOL FreeContext
= TRUE
;
2758 PSOCKET_INFORMATION Socket
;
2760 /* Get the Socket */
2761 Socket
= AsyncData
->ParentSocket
;
2763 /* If someone closed it, stop the function */
2764 if (Socket
->SharedData
.State
!= SocketClosed
)
2766 /* Check if the Sequence Number changed by now, in which case quit */
2767 if (AsyncData
->SequenceNumber
== Socket
->SharedData
.SequenceNumber
)
2769 /* Do the actuall select, if needed */
2770 if ((Socket
->SharedData
.AsyncEvents
& (~Socket
->SharedData
.AsyncDisabledEvents
)))
2772 SockProcessAsyncSelect(Socket
, AsyncData
);
2773 FreeContext
= FALSE
;
2778 /* Free the Context */
2781 HeapFree(GetProcessHeap(), 0, AsyncData
);
2788 SockReenableAsyncSelectEvent (IN PSOCKET_INFORMATION Socket
,
2791 PASYNC_DATA AsyncData
;
2793 /* Make sure the event is actually disabled */
2794 if (!(Socket
->SharedData
.AsyncDisabledEvents
& Event
))
2800 Socket
->SharedData
.AsyncDisabledEvents
&= ~Event
;
2802 /* Return if no more events are being polled */
2803 if ((Socket
->SharedData
.AsyncEvents
& (~Socket
->SharedData
.AsyncDisabledEvents
)) == 0 )
2808 /* Wait on new events */
2809 AsyncData
= HeapAlloc(GetProcessHeap(), 0, sizeof(ASYNC_DATA
));
2810 if (!AsyncData
) return;
2812 /* Create the Asynch Thread if Needed */
2813 SockCreateOrReferenceAsyncThread();
2815 /* Increase the sequence number to stop anything else */
2816 Socket
->SharedData
.SequenceNumber
++;
2818 /* Set up the Async Data */
2819 AsyncData
->ParentSocket
= Socket
;
2820 AsyncData
->SequenceNumber
= Socket
->SharedData
.SequenceNumber
;
2822 /* Begin Async Select by using I/O Completion */
2823 NtSetIoCompletion(SockAsyncCompletionPort
,
2824 (PVOID
)&SockProcessQueuedAsyncSelect
,
2835 DllMain(HANDLE hInstDll
,
2842 case DLL_PROCESS_ATTACH
:
2844 AFD_DbgPrint(MAX_TRACE
, ("Loading MSAFD.DLL \n"));
2846 /* Don't need thread attach notifications
2847 so disable them to improve performance */
2848 DisableThreadLibraryCalls(hInstDll
);
2850 /* List of DLL Helpers */
2851 InitializeListHead(&SockHelpersListHead
);
2853 /* Heap to use when allocating */
2854 GlobalHeap
= GetProcessHeap();
2856 /* Initialize the lock that protects our socket list */
2857 InitializeCriticalSection(&SocketListLock
);
2859 AFD_DbgPrint(MAX_TRACE
, ("MSAFD.DLL has been loaded\n"));
2863 case DLL_THREAD_ATTACH
:
2866 case DLL_THREAD_DETACH
:
2869 case DLL_PROCESS_DETACH
:
2871 /* Delete the socket list lock */
2872 DeleteCriticalSection(&SocketListLock
);
2877 AFD_DbgPrint(MAX_TRACE
, ("DllMain of msafd.dll (leaving)\n"));