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 LIST_ENTRY SockHelpersListHead
= { NULL
, NULL
};
27 ULONG SockAsyncThreadRefCount
;
28 HANDLE SockAsyncHelperAfdHandle
;
29 HANDLE SockAsyncCompletionPort
;
30 BOOLEAN SockAsyncSelectCalled
;
35 * FUNCTION: Creates a new socket
39 * protocol = Protocol type
40 * lpProtocolInfo = Pointer to protocol information
42 * dwFlags = Socket flags
43 * lpErrno = Address of buffer for error information
45 * Created socket, or INVALID_SOCKET if it could not be created
49 WSPSocket(int AddressFamily
,
52 LPWSAPROTOCOL_INFOW lpProtocolInfo
,
57 OBJECT_ATTRIBUTES Object
;
61 PAFD_CREATE_PACKET AfdPacket
;
63 PSOCKET_INFORMATION Socket
= NULL
, PrevSocket
= NULL
;
64 PFILE_FULL_EA_INFORMATION EABuffer
= NULL
;
65 PHELPER_DATA HelperData
;
66 PVOID HelperDLLContext
;
68 UNICODE_STRING TransportName
;
69 UNICODE_STRING DevName
;
70 LARGE_INTEGER GroupData
;
73 AFD_DbgPrint(MAX_TRACE
, ("Creating Socket, getting TDI Name\n"));
74 AFD_DbgPrint(MAX_TRACE
, ("AddressFamily (%d) SocketType (%d) Protocol (%d).\n",
75 AddressFamily
, SocketType
, Protocol
));
77 /* Get Helper Data and Transport */
78 Status
= SockGetTdiName (&AddressFamily
,
89 if (Status
!= NO_ERROR
)
91 AFD_DbgPrint(MID_TRACE
,("SockGetTdiName: Status %x\n", Status
));
96 RtlInitUnicodeString(&DevName
, L
"\\Device\\Afd\\Endpoint");
99 Socket
= HeapAlloc(GlobalHeap
, 0, sizeof(*Socket
));
101 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
103 RtlZeroMemory(Socket
, sizeof(*Socket
));
104 Socket
->RefCount
= 2;
106 Socket
->SharedData
.Listening
= FALSE
;
107 Socket
->SharedData
.State
= SocketOpen
;
108 Socket
->SharedData
.AddressFamily
= AddressFamily
;
109 Socket
->SharedData
.SocketType
= SocketType
;
110 Socket
->SharedData
.Protocol
= Protocol
;
111 Socket
->HelperContext
= HelperDLLContext
;
112 Socket
->HelperData
= HelperData
;
113 Socket
->HelperEvents
= HelperEvents
;
114 Socket
->LocalAddress
= &Socket
->WSLocalAddress
;
115 Socket
->SharedData
.SizeOfLocalAddress
= HelperData
->MaxWSAddressLength
;
116 Socket
->RemoteAddress
= &Socket
->WSRemoteAddress
;
117 Socket
->SharedData
.SizeOfRemoteAddress
= HelperData
->MaxWSAddressLength
;
118 Socket
->SharedData
.UseDelayedAcceptance
= HelperData
->UseDelayedAcceptance
;
119 Socket
->SharedData
.CreateFlags
= dwFlags
;
120 Socket
->SharedData
.CatalogEntryId
= lpProtocolInfo
->dwCatalogEntryId
;
121 Socket
->SharedData
.ServiceFlags1
= lpProtocolInfo
->dwServiceFlags1
;
122 Socket
->SharedData
.ProviderFlags
= lpProtocolInfo
->dwProviderFlags
;
123 Socket
->SharedData
.GroupID
= g
;
124 Socket
->SharedData
.GroupType
= 0;
125 Socket
->SharedData
.UseSAN
= FALSE
;
126 Socket
->SharedData
.NonBlocking
= FALSE
; /* Sockets start blocking */
127 Socket
->SanData
= NULL
;
129 /* Ask alex about this */
130 if( Socket
->SharedData
.SocketType
== SOCK_DGRAM
||
131 Socket
->SharedData
.SocketType
== SOCK_RAW
)
133 AFD_DbgPrint(MID_TRACE
,("Connectionless socket\n"));
134 Socket
->SharedData
.ServiceFlags1
|= XP1_CONNECTIONLESS
;
138 SizeOfPacket
= TransportName
.Length
+ sizeof(AFD_CREATE_PACKET
) + sizeof(WCHAR
);
141 SizeOfEA
= SizeOfPacket
+ sizeof(FILE_FULL_EA_INFORMATION
) + AFD_PACKET_COMMAND_LENGTH
;
143 /* Set up EA Buffer */
144 EABuffer
= HeapAlloc(GlobalHeap
, 0, SizeOfEA
);
146 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
148 RtlZeroMemory(EABuffer
, SizeOfEA
);
149 EABuffer
->NextEntryOffset
= 0;
151 EABuffer
->EaNameLength
= AFD_PACKET_COMMAND_LENGTH
;
152 RtlCopyMemory (EABuffer
->EaName
,
154 AFD_PACKET_COMMAND_LENGTH
+ 1);
155 EABuffer
->EaValueLength
= SizeOfPacket
;
157 /* Set up AFD Packet */
158 AfdPacket
= (PAFD_CREATE_PACKET
)(EABuffer
->EaName
+ EABuffer
->EaNameLength
+ 1);
159 AfdPacket
->SizeOfTransportName
= TransportName
.Length
;
160 RtlCopyMemory (AfdPacket
->TransportName
,
161 TransportName
.Buffer
,
162 TransportName
.Length
+ sizeof(WCHAR
));
163 AfdPacket
->GroupID
= g
;
165 /* Set up Endpoint Flags */
166 if ((Socket
->SharedData
.ServiceFlags1
& XP1_CONNECTIONLESS
) != 0)
168 if ((SocketType
!= SOCK_DGRAM
) && (SocketType
!= SOCK_RAW
))
170 /* Only RAW or UDP can be Connectionless */
173 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_CONNECTIONLESS
;
176 if ((Socket
->SharedData
.ServiceFlags1
& XP1_MESSAGE_ORIENTED
) != 0)
178 if (SocketType
== SOCK_STREAM
)
180 if ((Socket
->SharedData
.ServiceFlags1
& XP1_PSEUDO_STREAM
) == 0)
182 /* The Provider doesn't actually support Message Oriented Streams */
186 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_MESSAGE_ORIENTED
;
189 if (SocketType
== SOCK_RAW
) AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_RAW
;
191 if (dwFlags
& (WSA_FLAG_MULTIPOINT_C_ROOT
|
192 WSA_FLAG_MULTIPOINT_C_LEAF
|
193 WSA_FLAG_MULTIPOINT_D_ROOT
|
194 WSA_FLAG_MULTIPOINT_D_LEAF
))
196 if ((Socket
->SharedData
.ServiceFlags1
& XP1_SUPPORT_MULTIPOINT
) == 0)
198 /* The Provider doesn't actually support Multipoint */
201 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_MULTIPOINT
;
203 if (dwFlags
& WSA_FLAG_MULTIPOINT_C_ROOT
)
205 if (((Socket
->SharedData
.ServiceFlags1
& XP1_MULTIPOINT_CONTROL_PLANE
) == 0)
206 || ((dwFlags
& WSA_FLAG_MULTIPOINT_C_LEAF
) != 0))
208 /* The Provider doesn't support Control Planes, or you already gave a leaf */
211 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_C_ROOT
;
214 if (dwFlags
& WSA_FLAG_MULTIPOINT_D_ROOT
)
216 if (((Socket
->SharedData
.ServiceFlags1
& XP1_MULTIPOINT_DATA_PLANE
) == 0)
217 || ((dwFlags
& WSA_FLAG_MULTIPOINT_D_LEAF
) != 0))
219 /* The Provider doesn't support Data Planes, or you already gave a leaf */
222 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_D_ROOT
;
226 /* Set up Object Attributes */
227 InitializeObjectAttributes (&Object
,
229 OBJ_CASE_INSENSITIVE
| OBJ_INHERIT
,
233 /* Create the Socket as asynchronous. That means we have to block
234 ourselves after every call to NtDeviceIoControlFile. This is
235 because the kernel doesn't support overlapping synchronous I/O
236 requests (made from multiple threads) at this time (Sep 2005) */
237 Status
= NtCreateFile(&Sock
,
238 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
,
243 FILE_SHARE_READ
| FILE_SHARE_WRITE
,
249 HeapFree(GlobalHeap
, 0, EABuffer
);
251 if (Status
!= STATUS_SUCCESS
)
253 AFD_DbgPrint(MIN_TRACE
, ("Failed to open socket\n"));
255 HeapFree(GlobalHeap
, 0, Socket
);
257 return MsafdReturnWithErrno(Status
, lpErrno
, 0, NULL
);
261 Socket
->Handle
= (SOCKET
)Sock
;
263 /* XXX See if there's a structure we can reuse -- We need to do this
265 PrevSocket
= GetSocketStructure( (SOCKET
)Sock
);
269 RtlCopyMemory( PrevSocket
, Socket
, sizeof(*Socket
) );
270 RtlFreeHeap( GlobalHeap
, 0, Socket
);
274 /* Save Group Info */
277 GetSocketInformation(Socket
, AFD_INFO_GROUP_ID_TYPE
, 0, &GroupData
);
278 Socket
->SharedData
.GroupID
= GroupData
.u
.LowPart
;
279 Socket
->SharedData
.GroupType
= GroupData
.u
.HighPart
;
282 /* Get Window Sizes and Save them */
283 GetSocketInformation (Socket
,
284 AFD_INFO_SEND_WINDOW_SIZE
,
285 &Socket
->SharedData
.SizeOfSendBuffer
,
288 GetSocketInformation (Socket
,
289 AFD_INFO_RECEIVE_WINDOW_SIZE
,
290 &Socket
->SharedData
.SizeOfRecvBuffer
,
293 /* Save in Process Sockets List */
294 Socket
->NextSocket
= SocketListHead
;
295 SocketListHead
= Socket
;
297 /* Create the Socket Context */
298 CreateContext(Socket
);
301 Upcalls
.lpWPUModifyIFSHandle(1, (SOCKET
)Sock
, lpErrno
);
303 /* Return Socket Handle */
304 AFD_DbgPrint(MID_TRACE
,("Success %x\n", Sock
));
309 AFD_DbgPrint(MID_TRACE
,("Ending %x\n", Status
));
312 HeapFree(GlobalHeap
, 0, Socket
);
317 return INVALID_SOCKET
;
321 TranslateNtStatusError(NTSTATUS Status
)
325 case STATUS_CANT_WAIT
:
326 return WSAEWOULDBLOCK
;
334 case STATUS_FILE_CLOSED
:
335 case STATUS_END_OF_FILE
:
339 return WSA_IO_PENDING
;
341 case STATUS_BUFFER_TOO_SMALL
:
342 case STATUS_BUFFER_OVERFLOW
:
343 DbgPrint("MSAFD: STATUS_BUFFER_TOO_SMALL/STATUS_BUFFER_OVERFLOW\n");
346 case STATUS_NO_MEMORY
:
347 case STATUS_INSUFFICIENT_RESOURCES
:
348 DbgPrint("MSAFD: STATUS_NO_MEMORY/STATUS_INSUFFICIENT_RESOURCES\n");
351 case STATUS_INVALID_CONNECTION
:
352 DbgPrint("MSAFD: STATUS_INVALID_CONNECTION\n");
353 return WSAEAFNOSUPPORT
;
355 case STATUS_INVALID_ADDRESS
:
356 DbgPrint("MSAFD: STATUS_INVALID_ADDRESS\n");
357 return WSAEADDRNOTAVAIL
;
359 case STATUS_REMOTE_NOT_LISTENING
:
360 DbgPrint("MSAFD: STATUS_REMOTE_NOT_LISTENING\n");
361 return WSAECONNREFUSED
;
363 case STATUS_NETWORK_UNREACHABLE
:
364 DbgPrint("MSAFD: STATUS_NETWORK_UNREACHABLE\n");
365 return WSAENETUNREACH
;
367 case STATUS_INVALID_PARAMETER
:
368 DbgPrint("MSAFD: STATUS_INVALID_PARAMETER\n");
371 case STATUS_CANCELLED
:
372 DbgPrint("MSAFD: STATUS_CANCELLED\n");
373 return WSA_OPERATION_ABORTED
;
375 case STATUS_ADDRESS_ALREADY_EXISTS
:
376 DbgPrint("MSAFD: STATUS_ADDRESS_ALREADY_EXISTS\n");
377 return WSAEADDRINUSE
;
379 case STATUS_LOCAL_DISCONNECT
:
380 DbgPrint("MSAFD: STATUS_LOCAL_DISCONNECT\n");
381 return WSAECONNABORTED
;
383 case STATUS_REMOTE_DISCONNECT
:
384 DbgPrint("MSAFD: STATUS_REMOTE_DISCONNECT\n");
385 return WSAECONNRESET
;
388 DbgPrint("MSAFD: Unhandled NTSTATUS value: 0x%x\n", Status
);
393 DWORD
MsafdReturnWithErrno(NTSTATUS Status
,
396 LPDWORD ReturnedBytes
)
398 *Errno
= TranslateNtStatusError(Status
);
403 *ReturnedBytes
= Received
;
408 return *Errno
? SOCKET_ERROR
: 0;
412 * FUNCTION: Closes an open socket
414 * s = Socket descriptor
415 * lpErrno = Address of buffer for error information
417 * NO_ERROR, or SOCKET_ERROR if the socket could not be closed
421 WSPCloseSocket(IN SOCKET Handle
,
424 IO_STATUS_BLOCK IoStatusBlock
;
425 PSOCKET_INFORMATION Socket
= NULL
, CurrentSocket
;
428 AFD_DISCONNECT_INFO DisconnectInfo
;
429 SOCKET_STATE OldState
;
431 /* Create the Wait Event */
432 Status
= NtCreateEvent(&SockEvent
,
433 GENERIC_READ
| GENERIC_WRITE
,
438 if(!NT_SUCCESS(Status
))
441 /* Get the Socket Structure associate to this Socket*/
442 Socket
= GetSocketStructure(Handle
);
446 *lpErrno
= WSAENOTSOCK
;
450 if (Socket
->HelperEvents
& WSH_NOTIFY_CLOSE
)
452 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
454 Socket
->TdiAddressHandle
,
455 Socket
->TdiConnectionHandle
,
460 if (lpErrno
) *lpErrno
= Status
;
466 /* If a Close is already in Process, give up */
467 if (Socket
->SharedData
.State
== SocketClosed
)
470 *lpErrno
= WSAENOTSOCK
;
474 /* Set the state to close */
475 OldState
= Socket
->SharedData
.State
;
476 Socket
->SharedData
.State
= SocketClosed
;
478 /* If SO_LINGER is ON and the Socket is connected, we need to disconnect */
479 /* FIXME: Should we do this on Datagram Sockets too? */
480 if ((OldState
== SocketConnected
) && (Socket
->SharedData
.LingerData
.l_onoff
))
483 ULONG SendsInProgress
;
486 /* We need to respect the timeout */
488 LingerWait
= Socket
->SharedData
.LingerData
.l_linger
* 1000;
490 /* Loop until no more sends are pending, within the timeout */
493 /* Find out how many Sends are in Progress */
494 if (GetSocketInformation(Socket
,
495 AFD_INFO_SENDS_IN_PROGRESS
,
499 /* Bail out if anything but NO_ERROR */
504 /* Bail out if no more sends are pending */
505 if (!SendsInProgress
)
508 * We have to execute a sleep, so it's kind of like
509 * a block. If the socket is Nonblock, we cannot
510 * go on since asyncronous operation is expected
511 * and we cannot offer it
513 if (Socket
->SharedData
.NonBlocking
)
516 Socket
->SharedData
.State
= OldState
;
517 *lpErrno
= WSAEWOULDBLOCK
;
521 /* Now we can sleep, and decrement the linger wait */
523 * FIXME: It seems Windows does some funky acceleration
524 * since the waiting seems to be longer and longer. I
525 * don't think this improves performance so much, so we
526 * wait a fixed time instead.
529 LingerWait
-= SleepWait
;
533 * We have reached the timeout or sends are over.
534 * Disconnect if the timeout has been reached.
538 DisconnectInfo
.Timeout
= RtlConvertLongToLargeInteger(0);
539 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_ABORT
;
542 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
547 IOCTL_AFD_DISCONNECT
,
549 sizeof(DisconnectInfo
),
553 /* Wait for return */
554 if (Status
== STATUS_PENDING
)
556 WaitForSingleObject(SockEvent
, INFINITE
);
557 Status
= IoStatusBlock
.Status
;
563 Socket
->HelperContext
= NULL
;
564 Socket
->SharedData
.AsyncDisabledEvents
= -1;
565 NtClose(Socket
->TdiAddressHandle
);
566 Socket
->TdiAddressHandle
= NULL
;
567 NtClose(Socket
->TdiConnectionHandle
);
568 Socket
->TdiConnectionHandle
= NULL
;
570 if (SocketListHead
== Socket
)
572 SocketListHead
= SocketListHead
->NextSocket
;
576 CurrentSocket
= SocketListHead
;
577 while (CurrentSocket
->NextSocket
)
579 if (CurrentSocket
->NextSocket
== Socket
)
581 CurrentSocket
->NextSocket
= CurrentSocket
->NextSocket
->NextSocket
;
585 CurrentSocket
= CurrentSocket
->NextSocket
;
589 HeapFree(GlobalHeap
, 0, Socket
);
591 /* Close the handle */
592 NtClose((HANDLE
)Handle
);
595 return MsafdReturnWithErrno(Status
, lpErrno
, 0, NULL
);
600 * FUNCTION: Associates a local address with a socket
602 * s = Socket descriptor
603 * name = Pointer to local address
604 * namelen = Length of name
605 * lpErrno = Address of buffer for error information
607 * 0, or SOCKET_ERROR if the socket could not be bound
611 WSPBind(SOCKET Handle
,
612 const struct sockaddr
*SocketAddress
,
613 int SocketAddressLength
,
616 IO_STATUS_BLOCK IOSB
;
617 PAFD_BIND_DATA BindData
;
618 PSOCKET_INFORMATION Socket
= NULL
;
620 SOCKADDR_INFO SocketInfo
;
624 BindData
= HeapAlloc(GlobalHeap
, 0, 0xA + SocketAddressLength
);
627 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
630 Status
= NtCreateEvent(&SockEvent
,
631 GENERIC_READ
| GENERIC_WRITE
,
636 if (!NT_SUCCESS(Status
))
638 HeapFree(GlobalHeap
, 0, BindData
);
642 /* Get the Socket Structure associate to this Socket*/
643 Socket
= GetSocketStructure(Handle
);
646 HeapFree(GlobalHeap
, 0, BindData
);
647 *lpErrno
= WSAENOTSOCK
;
651 /* Set up Address in TDI Format */
652 BindData
->Address
.TAAddressCount
= 1;
653 BindData
->Address
.Address
[0].AddressLength
= SocketAddressLength
- sizeof(SocketAddress
->sa_family
);
654 BindData
->Address
.Address
[0].AddressType
= SocketAddress
->sa_family
;
655 RtlCopyMemory (BindData
->Address
.Address
[0].Address
,
656 SocketAddress
->sa_data
,
657 SocketAddressLength
- sizeof(SocketAddress
->sa_family
));
659 /* Get Address Information */
660 Socket
->HelperData
->WSHGetSockaddrType ((PSOCKADDR
)SocketAddress
,
664 /* Set the Share Type */
665 if (Socket
->SharedData
.ExclusiveAddressUse
)
667 BindData
->ShareType
= AFD_SHARE_EXCLUSIVE
;
669 else if (SocketInfo
.EndpointInfo
== SockaddrEndpointInfoWildcard
)
671 BindData
->ShareType
= AFD_SHARE_WILDCARD
;
673 else if (Socket
->SharedData
.ReuseAddresses
)
675 BindData
->ShareType
= AFD_SHARE_REUSE
;
679 BindData
->ShareType
= AFD_SHARE_UNIQUE
;
683 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
690 0xA + Socket
->SharedData
.SizeOfLocalAddress
, /* Can't figure out a way to calculate this in C*/
692 0xA + Socket
->SharedData
.SizeOfLocalAddress
); /* Can't figure out a way to calculate this C */
694 /* Wait for return */
695 if (Status
== STATUS_PENDING
)
697 WaitForSingleObject(SockEvent
, INFINITE
);
698 Status
= IOSB
.Status
;
701 NtClose( SockEvent
);
702 HeapFree(GlobalHeap
, 0, BindData
);
704 if (Status
!= STATUS_SUCCESS
)
705 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
707 /* Set up Socket Data */
708 Socket
->SharedData
.State
= SocketBound
;
709 Socket
->TdiAddressHandle
= (HANDLE
)IOSB
.Information
;
711 if (Socket
->HelperEvents
& WSH_NOTIFY_BIND
)
713 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
715 Socket
->TdiAddressHandle
,
716 Socket
->TdiConnectionHandle
,
721 if (lpErrno
) *lpErrno
= Status
;
726 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
731 WSPListen(SOCKET Handle
,
735 IO_STATUS_BLOCK IOSB
;
736 AFD_LISTEN_DATA ListenData
;
737 PSOCKET_INFORMATION Socket
= NULL
;
741 /* Get the Socket Structure associate to this Socket*/
742 Socket
= GetSocketStructure(Handle
);
745 *lpErrno
= WSAENOTSOCK
;
749 if (Socket
->SharedData
.Listening
)
752 Status
= NtCreateEvent(&SockEvent
,
753 GENERIC_READ
| GENERIC_WRITE
,
758 if( !NT_SUCCESS(Status
) )
761 /* Set Up Listen Structure */
762 ListenData
.UseSAN
= FALSE
;
763 ListenData
.UseDelayedAcceptance
= Socket
->SharedData
.UseDelayedAcceptance
;
764 ListenData
.Backlog
= Backlog
;
767 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
772 IOCTL_AFD_START_LISTEN
,
778 /* Wait for return */
779 if (Status
== STATUS_PENDING
)
781 WaitForSingleObject(SockEvent
, INFINITE
);
782 Status
= IOSB
.Status
;
785 NtClose( SockEvent
);
787 if (Status
!= STATUS_SUCCESS
)
788 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
790 /* Set to Listening */
791 Socket
->SharedData
.Listening
= TRUE
;
793 if (Socket
->HelperEvents
& WSH_NOTIFY_LISTEN
)
795 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
797 Socket
->TdiAddressHandle
,
798 Socket
->TdiConnectionHandle
,
803 if (lpErrno
) *lpErrno
= Status
;
808 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
818 const LPTIMEVAL timeout
,
821 IO_STATUS_BLOCK IOSB
;
822 PAFD_POLL_INFO PollInfo
;
824 LONG HandleCount
, OutCount
= 0;
825 ULONG PollBufferSize
;
830 LARGE_INTEGER Timeout
;
832 /* Find out how many sockets we have, and how large the buffer needs
835 HandleCount
= ( readfds
? readfds
->fd_count
: 0 ) +
836 ( writefds
? writefds
->fd_count
: 0 ) +
837 ( exceptfds
? exceptfds
->fd_count
: 0 );
839 if ( HandleCount
== 0 )
841 AFD_DbgPrint(MAX_TRACE
,("HandleCount: %d. Return SOCKET_ERROR\n",
843 if (lpErrno
) *lpErrno
= WSAEINVAL
;
847 PollBufferSize
= sizeof(*PollInfo
) + ((HandleCount
- 1) * sizeof(AFD_HANDLE
));
849 AFD_DbgPrint(MID_TRACE
,("HandleCount: %d BufferSize: %d\n",
850 HandleCount
, PollBufferSize
));
852 /* Convert Timeout to NT Format */
855 Timeout
.u
.LowPart
= -1;
856 Timeout
.u
.HighPart
= 0x7FFFFFFF;
857 AFD_DbgPrint(MAX_TRACE
,("Infinite timeout\n"));
861 Timeout
= RtlEnlargedIntegerMultiply
862 ((timeout
->tv_sec
* 1000) + (timeout
->tv_usec
/ 1000), -10000);
863 /* Negative timeouts are illegal. Since the kernel represents an
864 * incremental timeout as a negative number, we check for a positive
867 if (Timeout
.QuadPart
> 0)
869 if (lpErrno
) *lpErrno
= WSAEINVAL
;
872 AFD_DbgPrint(MAX_TRACE
,("Timeout: Orig %d.%06d kernel %d\n",
873 timeout
->tv_sec
, timeout
->tv_usec
,
877 Status
= NtCreateEvent(&SockEvent
,
878 GENERIC_READ
| GENERIC_WRITE
,
883 if( !NT_SUCCESS(Status
) )
887 PollBuffer
= HeapAlloc(GlobalHeap
, 0, PollBufferSize
);
892 *lpErrno
= WSAEFAULT
;
897 PollInfo
= (PAFD_POLL_INFO
)PollBuffer
;
899 RtlZeroMemory( PollInfo
, PollBufferSize
);
901 /* Number of handles for AFD to Check */
902 PollInfo
->Exclusive
= FALSE
;
903 PollInfo
->Timeout
= Timeout
;
905 if (readfds
!= NULL
) {
906 for (i
= 0; i
< readfds
->fd_count
; i
++, j
++)
908 PollInfo
->Handles
[j
].Handle
= readfds
->fd_array
[i
];
909 PollInfo
->Handles
[j
].Events
= AFD_EVENT_RECEIVE
|
910 AFD_EVENT_DISCONNECT
|
916 if (writefds
!= NULL
)
918 for (i
= 0; i
< writefds
->fd_count
; i
++, j
++)
920 PollInfo
->Handles
[j
].Handle
= writefds
->fd_array
[i
];
921 PollInfo
->Handles
[j
].Events
= AFD_EVENT_SEND
| AFD_EVENT_CONNECT
;
924 if (exceptfds
!= NULL
)
926 for (i
= 0; i
< exceptfds
->fd_count
; i
++, j
++)
928 PollInfo
->Handles
[j
].Handle
= exceptfds
->fd_array
[i
];
929 PollInfo
->Handles
[j
].Events
= AFD_EVENT_OOB_RECEIVE
| AFD_EVENT_CONNECT_FAIL
;
933 PollInfo
->HandleCount
= j
;
934 PollBufferSize
= ((PCHAR
)&PollInfo
->Handles
[j
+1]) - ((PCHAR
)PollInfo
);
937 Status
= NtDeviceIoControlFile((HANDLE
)PollInfo
->Handles
[0].Handle
,
948 AFD_DbgPrint(MID_TRACE
,("DeviceIoControlFile => %x\n", Status
));
950 /* Wait for Completition */
951 if (Status
== STATUS_PENDING
)
953 WaitForSingleObject(SockEvent
, INFINITE
);
954 Status
= IOSB
.Status
;
957 /* Clear the Structures */
965 /* Loop through return structure */
966 HandleCount
= PollInfo
->HandleCount
;
968 /* Return in FDSET Format */
969 for (i
= 0; i
< HandleCount
; i
++)
971 HandleCounted
= FALSE
;
974 switch (PollInfo
->Handles
[i
].Events
& x
)
976 case AFD_EVENT_RECEIVE
:
977 case AFD_EVENT_DISCONNECT
:
978 case AFD_EVENT_ABORT
:
979 case AFD_EVENT_ACCEPT
:
980 case AFD_EVENT_CLOSE
:
981 AFD_DbgPrint(MID_TRACE
,("Event %x on handle %x\n",
982 PollInfo
->Handles
[i
].Events
,
983 PollInfo
->Handles
[i
].Handle
));
987 HandleCounted
= TRUE
;
990 FD_SET(PollInfo
->Handles
[i
].Handle
, readfds
);
993 case AFD_EVENT_CONNECT
:
994 AFD_DbgPrint(MID_TRACE
,("Event %x on handle %x\n",
995 PollInfo
->Handles
[i
].Events
,
996 PollInfo
->Handles
[i
].Handle
));
1000 HandleCounted
= TRUE
;
1003 FD_SET(PollInfo
->Handles
[i
].Handle
, writefds
);
1005 case AFD_EVENT_OOB_RECEIVE
:
1006 case AFD_EVENT_CONNECT_FAIL
:
1007 AFD_DbgPrint(MID_TRACE
,("Event %x on handle %x\n",
1008 PollInfo
->Handles
[i
].Events
,
1009 PollInfo
->Handles
[i
].Handle
));
1010 if (! HandleCounted
)
1013 HandleCounted
= TRUE
;
1016 FD_SET(PollInfo
->Handles
[i
].Handle
, exceptfds
);
1022 HeapFree( GlobalHeap
, 0, PollBuffer
);
1023 NtClose( SockEvent
);
1027 switch( IOSB
.Status
)
1029 case STATUS_SUCCESS
:
1030 case STATUS_TIMEOUT
:
1034 *lpErrno
= WSAEINVAL
;
1037 AFD_DbgPrint(MID_TRACE
,("*lpErrno = %x\n", *lpErrno
));
1040 AFD_DbgPrint(MID_TRACE
,("%d events\n", OutCount
));
1047 WSPAccept(SOCKET Handle
,
1048 struct sockaddr
*SocketAddress
,
1049 int *SocketAddressLength
,
1050 LPCONDITIONPROC lpfnCondition
,
1051 DWORD dwCallbackData
,
1054 IO_STATUS_BLOCK IOSB
;
1055 PAFD_RECEIVED_ACCEPT_DATA ListenReceiveData
;
1056 AFD_ACCEPT_DATA AcceptData
;
1057 AFD_DEFER_ACCEPT_DATA DeferData
;
1058 AFD_PENDING_ACCEPT_DATA PendingAcceptData
;
1059 PSOCKET_INFORMATION Socket
= NULL
;
1061 struct fd_set ReadSet
;
1062 struct timeval Timeout
;
1063 PVOID PendingData
= NULL
;
1064 ULONG PendingDataLength
= 0;
1065 PVOID CalleeDataBuffer
;
1066 WSABUF CallerData
, CalleeID
, CallerID
, CalleeData
;
1067 PSOCKADDR RemoteAddress
= NULL
;
1070 WSAPROTOCOL_INFOW ProtocolInfo
;
1071 SOCKET AcceptSocket
;
1072 UCHAR ReceiveBuffer
[0x1A];
1075 Status
= NtCreateEvent(&SockEvent
,
1076 GENERIC_READ
| GENERIC_WRITE
,
1081 if( !NT_SUCCESS(Status
) )
1083 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1084 return INVALID_SOCKET
;
1087 /* Dynamic Structure...ugh */
1088 ListenReceiveData
= (PAFD_RECEIVED_ACCEPT_DATA
)ReceiveBuffer
;
1090 /* Get the Socket Structure associate to this Socket*/
1091 Socket
= GetSocketStructure(Handle
);
1095 *lpErrno
= WSAENOTSOCK
;
1096 return INVALID_SOCKET
;
1099 /* If this is non-blocking, make sure there's something for us to accept */
1101 FD_SET(Socket
->Handle
, &ReadSet
);
1105 WSPSelect(0, &ReadSet
, NULL
, NULL
, &Timeout
, NULL
);
1107 if (ReadSet
.fd_array
[0] != Socket
->Handle
)
1114 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1119 IOCTL_AFD_WAIT_FOR_LISTEN
,
1123 0xA + sizeof(*ListenReceiveData
));
1125 /* Wait for return */
1126 if (Status
== STATUS_PENDING
)
1128 WaitForSingleObject(SockEvent
, INFINITE
);
1129 Status
= IOSB
.Status
;
1132 if (!NT_SUCCESS(Status
))
1134 NtClose( SockEvent
);
1135 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1136 return INVALID_SOCKET
;
1139 if (lpfnCondition
!= NULL
)
1141 if ((Socket
->SharedData
.ServiceFlags1
& XP1_CONNECT_DATA
) != 0)
1143 /* Find out how much data is pending */
1144 PendingAcceptData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1145 PendingAcceptData
.ReturnSize
= TRUE
;
1148 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1153 IOCTL_AFD_GET_PENDING_CONNECT_DATA
,
1155 sizeof(PendingAcceptData
),
1157 sizeof(PendingAcceptData
));
1159 /* Wait for return */
1160 if (Status
== STATUS_PENDING
)
1162 WaitForSingleObject(SockEvent
, INFINITE
);
1163 Status
= IOSB
.Status
;
1166 if (!NT_SUCCESS(Status
))
1168 NtClose( SockEvent
);
1169 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1170 return INVALID_SOCKET
;
1173 /* How much data to allocate */
1174 PendingDataLength
= IOSB
.Information
;
1176 if (PendingDataLength
)
1178 /* Allocate needed space */
1179 PendingData
= HeapAlloc(GlobalHeap
, 0, PendingDataLength
);
1182 MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1183 return INVALID_SOCKET
;
1186 /* We want the data now */
1187 PendingAcceptData
.ReturnSize
= FALSE
;
1190 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1195 IOCTL_AFD_GET_PENDING_CONNECT_DATA
,
1197 sizeof(PendingAcceptData
),
1201 /* Wait for return */
1202 if (Status
== STATUS_PENDING
)
1204 WaitForSingleObject(SockEvent
, INFINITE
);
1205 Status
= IOSB
.Status
;
1208 if (!NT_SUCCESS(Status
))
1210 NtClose( SockEvent
);
1211 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1212 return INVALID_SOCKET
;
1217 if ((Socket
->SharedData
.ServiceFlags1
& XP1_QOS_SUPPORTED
) != 0)
1219 /* I don't support this yet */
1222 /* Build Callee ID */
1223 CalleeID
.buf
= (PVOID
)Socket
->LocalAddress
;
1224 CalleeID
.len
= Socket
->SharedData
.SizeOfLocalAddress
;
1226 RemoteAddress
= HeapAlloc(GlobalHeap
, 0, sizeof(*RemoteAddress
));
1229 MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1230 return INVALID_SOCKET
;
1233 /* Set up Address in SOCKADDR Format */
1234 RtlCopyMemory (RemoteAddress
,
1235 &ListenReceiveData
->Address
.Address
[0].AddressType
,
1236 sizeof(*RemoteAddress
));
1238 /* Build Caller ID */
1239 CallerID
.buf
= (PVOID
)RemoteAddress
;
1240 CallerID
.len
= sizeof(*RemoteAddress
);
1242 /* Build Caller Data */
1243 CallerData
.buf
= PendingData
;
1244 CallerData
.len
= PendingDataLength
;
1246 /* Check if socket supports Conditional Accept */
1247 if (Socket
->SharedData
.UseDelayedAcceptance
!= 0)
1249 /* Allocate Buffer for Callee Data */
1250 CalleeDataBuffer
= HeapAlloc(GlobalHeap
, 0, 4096);
1251 if (!CalleeDataBuffer
) {
1252 MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1253 return INVALID_SOCKET
;
1255 CalleeData
.buf
= CalleeDataBuffer
;
1256 CalleeData
.len
= 4096;
1265 /* Call the Condition Function */
1266 CallBack
= (lpfnCondition
)(&CallerID
,
1267 CallerData
.buf
== NULL
? NULL
: &CallerData
,
1271 CalleeData
.buf
== NULL
? NULL
: &CalleeData
,
1275 if (((CallBack
== CF_ACCEPT
) && GroupID
) != 0)
1277 /* TBD: Check for Validity */
1280 if (CallBack
== CF_ACCEPT
)
1282 if ((Socket
->SharedData
.ServiceFlags1
& XP1_QOS_SUPPORTED
) != 0)
1284 /* I don't support this yet */
1288 // SockSetConnectData Sockets(SocketID), IOCTL_AFD_SET_CONNECT_DATA, CalleeData.Buffer, CalleeData.BuffSize, 0
1293 /* Callback rejected. Build Defer Structure */
1294 DeferData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1295 DeferData
.RejectConnection
= (CallBack
== CF_REJECT
);
1298 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1303 IOCTL_AFD_DEFER_ACCEPT
,
1309 /* Wait for return */
1310 if (Status
== STATUS_PENDING
)
1312 WaitForSingleObject(SockEvent
, INFINITE
);
1313 Status
= IOSB
.Status
;
1316 NtClose( SockEvent
);
1318 if (!NT_SUCCESS(Status
))
1320 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1321 return INVALID_SOCKET
;
1324 if (CallBack
== CF_REJECT
)
1326 *lpErrno
= WSAECONNREFUSED
;
1327 return INVALID_SOCKET
;
1331 *lpErrno
= WSAECONNREFUSED
;
1332 return INVALID_SOCKET
;
1337 /* Create a new Socket */
1338 ProtocolInfo
.dwCatalogEntryId
= Socket
->SharedData
.CatalogEntryId
;
1339 ProtocolInfo
.dwServiceFlags1
= Socket
->SharedData
.ServiceFlags1
;
1340 ProtocolInfo
.dwProviderFlags
= Socket
->SharedData
.ProviderFlags
;
1342 AcceptSocket
= WSPSocket (Socket
->SharedData
.AddressFamily
,
1343 Socket
->SharedData
.SocketType
,
1344 Socket
->SharedData
.Protocol
,
1347 Socket
->SharedData
.CreateFlags
,
1350 /* Set up the Accept Structure */
1351 AcceptData
.ListenHandle
= (HANDLE
)AcceptSocket
;
1352 AcceptData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1354 /* Send IOCTL to Accept */
1355 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1366 /* Wait for return */
1367 if (Status
== STATUS_PENDING
)
1369 WaitForSingleObject(SockEvent
, INFINITE
);
1370 Status
= IOSB
.Status
;
1373 if (!NT_SUCCESS(Status
))
1376 WSPCloseSocket( AcceptSocket
, lpErrno
);
1377 MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1378 return INVALID_SOCKET
;
1381 /* Return Address in SOCKADDR FORMAT */
1384 RtlCopyMemory (SocketAddress
,
1385 &ListenReceiveData
->Address
.Address
[0].AddressType
,
1386 sizeof(*RemoteAddress
));
1387 if( SocketAddressLength
)
1388 *SocketAddressLength
= ListenReceiveData
->Address
.Address
[0].AddressLength
;
1391 NtClose( SockEvent
);
1393 /* Re-enable Async Event */
1394 SockReenableAsyncSelectEvent(Socket
, FD_ACCEPT
);
1396 AFD_DbgPrint(MID_TRACE
,("Socket %x\n", AcceptSocket
));
1398 if (Status
== STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_ACCEPT
))
1400 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
1402 Socket
->TdiAddressHandle
,
1403 Socket
->TdiConnectionHandle
,
1408 if (lpErrno
) *lpErrno
= Status
;
1409 return INVALID_SOCKET
;
1416 return AcceptSocket
;
1421 WSPConnect(SOCKET Handle
,
1422 const struct sockaddr
* SocketAddress
,
1423 int SocketAddressLength
,
1424 LPWSABUF lpCallerData
,
1425 LPWSABUF lpCalleeData
,
1430 IO_STATUS_BLOCK IOSB
;
1431 PAFD_CONNECT_INFO ConnectInfo
;
1432 PSOCKET_INFORMATION Socket
= NULL
;
1434 UCHAR ConnectBuffer
[0x22];
1435 ULONG ConnectDataLength
;
1436 ULONG InConnectDataLength
;
1437 INT BindAddressLength
;
1438 PSOCKADDR BindAddress
;
1441 Status
= NtCreateEvent(&SockEvent
,
1442 GENERIC_READ
| GENERIC_WRITE
,
1447 if( !NT_SUCCESS(Status
) )
1450 AFD_DbgPrint(MID_TRACE
,("Called\n"));
1452 /* Get the Socket Structure associate to this Socket*/
1453 Socket
= GetSocketStructure(Handle
);
1457 *lpErrno
= WSAENOTSOCK
;
1458 return SOCKET_ERROR
;
1462 if (Socket
->SharedData
.State
== SocketOpen
)
1464 /* Get the Wildcard Address */
1465 BindAddressLength
= Socket
->HelperData
->MaxWSAddressLength
;
1466 BindAddress
= HeapAlloc(GetProcessHeap(), 0, BindAddressLength
);
1469 MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1470 return INVALID_SOCKET
;
1472 Socket
->HelperData
->WSHGetWildcardSockaddr (Socket
->HelperContext
,
1474 &BindAddressLength
);
1476 if (WSPBind(Handle
, BindAddress
, BindAddressLength
, lpErrno
) == SOCKET_ERROR
)
1477 return INVALID_SOCKET
;
1480 /* Set the Connect Data */
1481 if (lpCallerData
!= NULL
)
1483 ConnectDataLength
= lpCallerData
->len
;
1484 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1489 IOCTL_AFD_SET_CONNECT_DATA
,
1494 /* Wait for return */
1495 if (Status
== STATUS_PENDING
)
1497 WaitForSingleObject(SockEvent
, INFINITE
);
1498 Status
= IOSB
.Status
;
1501 if (Status
!= STATUS_SUCCESS
)
1505 /* Dynamic Structure...ugh */
1506 ConnectInfo
= (PAFD_CONNECT_INFO
)ConnectBuffer
;
1508 /* Set up Address in TDI Format */
1509 ConnectInfo
->RemoteAddress
.TAAddressCount
= 1;
1510 ConnectInfo
->RemoteAddress
.Address
[0].AddressLength
= SocketAddressLength
- sizeof(SocketAddress
->sa_family
);
1511 ConnectInfo
->RemoteAddress
.Address
[0].AddressType
= SocketAddress
->sa_family
;
1512 RtlCopyMemory (ConnectInfo
->RemoteAddress
.Address
[0].Address
,
1513 SocketAddress
->sa_data
,
1514 SocketAddressLength
- sizeof(SocketAddress
->sa_family
));
1517 * Disable FD_WRITE and FD_CONNECT
1518 * The latter fixes a race condition where the FD_CONNECT is re-enabled
1519 * at the end of this function right after the Async Thread disables it.
1520 * This should only happen at the *next* WSPConnect
1522 if (Socket
->SharedData
.AsyncEvents
& FD_CONNECT
)
1524 Socket
->SharedData
.AsyncDisabledEvents
|= FD_CONNECT
| FD_WRITE
;
1527 /* Tell AFD that we want Connection Data back, have it allocate a buffer */
1528 if (lpCalleeData
!= NULL
)
1530 InConnectDataLength
= lpCalleeData
->len
;
1531 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1536 IOCTL_AFD_SET_CONNECT_DATA_SIZE
,
1537 &InConnectDataLength
,
1538 sizeof(InConnectDataLength
),
1542 /* Wait for return */
1543 if (Status
== STATUS_PENDING
)
1545 WaitForSingleObject(SockEvent
, INFINITE
);
1546 Status
= IOSB
.Status
;
1549 if (Status
!= STATUS_SUCCESS
)
1553 /* AFD doesn't seem to care if these are invalid, but let's 0 them anyways */
1554 ConnectInfo
->Root
= 0;
1555 ConnectInfo
->UseSAN
= FALSE
;
1556 ConnectInfo
->Unknown
= 0;
1558 /* FIXME: Handle Async Connect */
1559 if (Socket
->SharedData
.NonBlocking
)
1561 AFD_DbgPrint(MIN_TRACE
, ("Async Connect UNIMPLEMENTED!\n"));
1565 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1575 /* Wait for return */
1576 if (Status
== STATUS_PENDING
)
1578 WaitForSingleObject(SockEvent
, INFINITE
);
1579 Status
= IOSB
.Status
;
1582 if (Status
!= STATUS_SUCCESS
)
1585 Socket
->TdiConnectionHandle
= (HANDLE
)IOSB
.Information
;
1587 /* Get any pending connect data */
1588 if (lpCalleeData
!= NULL
)
1590 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1595 IOCTL_AFD_GET_CONNECT_DATA
,
1600 /* Wait for return */
1601 if (Status
== STATUS_PENDING
)
1603 WaitForSingleObject(SockEvent
, INFINITE
);
1604 Status
= IOSB
.Status
;
1608 AFD_DbgPrint(MID_TRACE
,("Ending\n"));
1611 /* Re-enable Async Event */
1612 SockReenableAsyncSelectEvent(Socket
, FD_WRITE
);
1614 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
1615 SockReenableAsyncSelectEvent(Socket
, FD_CONNECT
);
1617 NtClose( SockEvent
);
1619 if (Status
== STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_CONNECT
))
1621 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
1623 Socket
->TdiAddressHandle
,
1624 Socket
->TdiConnectionHandle
,
1625 WSH_NOTIFY_CONNECT
);
1629 if (lpErrno
) *lpErrno
= Status
;
1630 return SOCKET_ERROR
;
1633 else if (Status
!= STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_CONNECT_ERROR
))
1635 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
1637 Socket
->TdiAddressHandle
,
1638 Socket
->TdiConnectionHandle
,
1639 WSH_NOTIFY_CONNECT_ERROR
);
1643 if (lpErrno
) *lpErrno
= Status
;
1644 return SOCKET_ERROR
;
1648 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1652 WSPShutdown(SOCKET Handle
,
1657 IO_STATUS_BLOCK IOSB
;
1658 AFD_DISCONNECT_INFO DisconnectInfo
;
1659 PSOCKET_INFORMATION Socket
= NULL
;
1663 Status
= NtCreateEvent(&SockEvent
,
1664 GENERIC_READ
| GENERIC_WRITE
,
1669 if( !NT_SUCCESS(Status
) )
1672 AFD_DbgPrint(MID_TRACE
,("Called\n"));
1674 /* Get the Socket Structure associate to this Socket*/
1675 Socket
= GetSocketStructure(Handle
);
1679 *lpErrno
= WSAENOTSOCK
;
1680 return SOCKET_ERROR
;
1683 /* Set AFD Disconnect Type */
1687 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_RECV
;
1688 Socket
->SharedData
.ReceiveShutdown
= TRUE
;
1691 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_SEND
;
1692 Socket
->SharedData
.SendShutdown
= TRUE
;
1695 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_RECV
| AFD_DISCONNECT_SEND
;
1696 Socket
->SharedData
.ReceiveShutdown
= TRUE
;
1697 Socket
->SharedData
.SendShutdown
= TRUE
;
1701 DisconnectInfo
.Timeout
= RtlConvertLongToLargeInteger(-1);
1704 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1709 IOCTL_AFD_DISCONNECT
,
1711 sizeof(DisconnectInfo
),
1715 /* Wait for return */
1716 if (Status
== STATUS_PENDING
)
1718 WaitForSingleObject(SockEvent
, INFINITE
);
1719 Status
= IOSB
.Status
;
1722 AFD_DbgPrint(MID_TRACE
,("Ending\n"));
1724 NtClose( SockEvent
);
1726 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1732 WSPGetSockName(IN SOCKET Handle
,
1733 OUT LPSOCKADDR Name
,
1734 IN OUT LPINT NameLength
,
1737 IO_STATUS_BLOCK IOSB
;
1738 ULONG TdiAddressSize
;
1739 PTDI_ADDRESS_INFO TdiAddress
;
1740 PTRANSPORT_ADDRESS SocketAddress
;
1741 PSOCKET_INFORMATION Socket
= NULL
;
1745 Status
= NtCreateEvent(&SockEvent
,
1746 GENERIC_READ
| GENERIC_WRITE
,
1751 if( !NT_SUCCESS(Status
) )
1752 return SOCKET_ERROR
;
1754 /* Get the Socket Structure associate to this Socket*/
1755 Socket
= GetSocketStructure(Handle
);
1759 *lpErrno
= WSAENOTSOCK
;
1760 return SOCKET_ERROR
;
1763 /* Allocate a buffer for the address */
1765 sizeof(TRANSPORT_ADDRESS
) + Socket
->SharedData
.SizeOfLocalAddress
;
1766 TdiAddress
= HeapAlloc(GlobalHeap
, 0, TdiAddressSize
);
1768 if ( TdiAddress
== NULL
)
1770 NtClose( SockEvent
);
1771 *lpErrno
= WSAENOBUFS
;
1772 return SOCKET_ERROR
;
1775 SocketAddress
= &TdiAddress
->Address
;
1778 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1783 IOCTL_AFD_GET_SOCK_NAME
,
1789 /* Wait for return */
1790 if (Status
== STATUS_PENDING
)
1792 WaitForSingleObject(SockEvent
, INFINITE
);
1793 Status
= IOSB
.Status
;
1796 NtClose( SockEvent
);
1798 if (NT_SUCCESS(Status
))
1800 if (*NameLength
>= SocketAddress
->Address
[0].AddressLength
)
1802 Name
->sa_family
= SocketAddress
->Address
[0].AddressType
;
1803 RtlCopyMemory (Name
->sa_data
,
1804 SocketAddress
->Address
[0].Address
,
1805 SocketAddress
->Address
[0].AddressLength
);
1806 *NameLength
= 2 + SocketAddress
->Address
[0].AddressLength
;
1807 AFD_DbgPrint (MID_TRACE
, ("NameLength %d Address: %x Port %x\n",
1808 *NameLength
, ((struct sockaddr_in
*)Name
)->sin_addr
.s_addr
,
1809 ((struct sockaddr_in
*)Name
)->sin_port
));
1810 HeapFree(GlobalHeap
, 0, TdiAddress
);
1815 HeapFree(GlobalHeap
, 0, TdiAddress
);
1816 *lpErrno
= WSAEFAULT
;
1817 return SOCKET_ERROR
;
1821 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
1827 WSPGetPeerName(IN SOCKET s
,
1828 OUT LPSOCKADDR Name
,
1829 IN OUT LPINT NameLength
,
1832 IO_STATUS_BLOCK IOSB
;
1833 ULONG TdiAddressSize
;
1834 PTRANSPORT_ADDRESS SocketAddress
;
1835 PSOCKET_INFORMATION Socket
= NULL
;
1839 Status
= NtCreateEvent(&SockEvent
,
1840 GENERIC_READ
| GENERIC_WRITE
,
1845 if( !NT_SUCCESS(Status
) )
1846 return SOCKET_ERROR
;
1848 /* Get the Socket Structure associate to this Socket*/
1849 Socket
= GetSocketStructure(s
);
1853 *lpErrno
= WSAENOTSOCK
;
1854 return SOCKET_ERROR
;
1857 /* Allocate a buffer for the address */
1858 TdiAddressSize
= sizeof(TRANSPORT_ADDRESS
) + *NameLength
;
1859 SocketAddress
= HeapAlloc(GlobalHeap
, 0, TdiAddressSize
);
1861 if ( SocketAddress
== NULL
)
1863 NtClose( SockEvent
);
1864 *lpErrno
= WSAENOBUFS
;
1865 return SOCKET_ERROR
;
1869 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1874 IOCTL_AFD_GET_PEER_NAME
,
1880 /* Wait for return */
1881 if (Status
== STATUS_PENDING
)
1883 WaitForSingleObject(SockEvent
, INFINITE
);
1884 Status
= IOSB
.Status
;
1887 NtClose( SockEvent
);
1889 if (NT_SUCCESS(Status
))
1891 if (*NameLength
>= SocketAddress
->Address
[0].AddressLength
)
1893 Name
->sa_family
= SocketAddress
->Address
[0].AddressType
;
1894 RtlCopyMemory (Name
->sa_data
,
1895 SocketAddress
->Address
[0].Address
,
1896 SocketAddress
->Address
[0].AddressLength
);
1897 *NameLength
= 2 + SocketAddress
->Address
[0].AddressLength
;
1898 AFD_DbgPrint (MID_TRACE
, ("NameLength %d Address: %s Port %x\n",
1899 *NameLength
, ((struct sockaddr_in
*)Name
)->sin_addr
.s_addr
,
1900 ((struct sockaddr_in
*)Name
)->sin_port
));
1901 HeapFree(GlobalHeap
, 0, SocketAddress
);
1906 HeapFree(GlobalHeap
, 0, SocketAddress
);
1907 *lpErrno
= WSAEFAULT
;
1908 return SOCKET_ERROR
;
1912 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
1917 WSPIoctl(IN SOCKET Handle
,
1918 IN DWORD dwIoControlCode
,
1919 IN LPVOID lpvInBuffer
,
1920 IN DWORD cbInBuffer
,
1921 OUT LPVOID lpvOutBuffer
,
1922 IN DWORD cbOutBuffer
,
1923 OUT LPDWORD lpcbBytesReturned
,
1924 IN LPWSAOVERLAPPED lpOverlapped
,
1925 IN LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine
,
1926 IN LPWSATHREADID lpThreadId
,
1929 PSOCKET_INFORMATION Socket
= NULL
;
1931 /* Get the Socket Structure associate to this Socket*/
1932 Socket
= GetSocketStructure(Handle
);
1935 *lpErrno
= WSAENOTSOCK
;
1936 return SOCKET_ERROR
;
1939 switch( dwIoControlCode
)
1942 if( cbInBuffer
< sizeof(INT
) || IS_INTRESOURCE(lpvInBuffer
) )
1944 *lpErrno
= WSAEFAULT
;
1945 return SOCKET_ERROR
;
1947 Socket
->SharedData
.NonBlocking
= *((PULONG
)lpvInBuffer
) ? 1 : 0;
1948 return SetSocketInformation(Socket
, AFD_INFO_BLOCKING_MODE
, (PULONG
)lpvInBuffer
, NULL
);
1950 if( cbOutBuffer
< sizeof(INT
) || IS_INTRESOURCE(lpvOutBuffer
) )
1952 *lpErrno
= WSAEFAULT
;
1953 return SOCKET_ERROR
;
1955 return GetSocketInformation(Socket
, AFD_INFO_RECEIVE_CONTENT_SIZE
, (PULONG
)lpvOutBuffer
, NULL
);
1957 *lpErrno
= WSAEINVAL
;
1958 return SOCKET_ERROR
;
1965 WSPGetSockOpt(IN SOCKET Handle
,
1968 OUT CHAR FAR
* OptionValue
,
1969 IN OUT LPINT OptionLength
,
1972 PSOCKET_INFORMATION Socket
= NULL
;
1977 /* Get the Socket Structure associate to this Socket*/
1978 Socket
= GetSocketStructure(Handle
);
1981 *lpErrno
= WSAENOTSOCK
;
1982 return SOCKET_ERROR
;
1985 AFD_DbgPrint(MID_TRACE
, ("Called\n"));
1993 Buffer
= &Socket
->SharedData
.SocketType
;
1994 BufferSize
= sizeof(INT
);
1998 Buffer
= &Socket
->SharedData
.SizeOfRecvBuffer
;
1999 BufferSize
= sizeof(INT
);
2003 Buffer
= &Socket
->SharedData
.SizeOfSendBuffer
;
2004 BufferSize
= sizeof(INT
);
2008 BoolBuffer
= Socket
->SharedData
.Listening
;
2009 Buffer
= &BoolBuffer
;
2010 BufferSize
= sizeof(BOOLEAN
);
2014 BoolBuffer
= Socket
->SharedData
.Broadcast
;
2015 Buffer
= &BoolBuffer
;
2016 BufferSize
= sizeof(BOOLEAN
);
2020 BoolBuffer
= Socket
->SharedData
.Debug
;
2021 Buffer
= &BoolBuffer
;
2022 BufferSize
= sizeof(BOOLEAN
);
2025 /* case SO_CONDITIONAL_ACCEPT: */
2030 case SO_GROUP_PRIORITY
:
2033 case SO_MAX_MSG_SIZE
:
2035 case SO_PROTOCOL_INFO
:
2037 AFD_DbgPrint(MID_TRACE
, ("Unimplemented option (%x)\n",
2041 *lpErrno
= WSAEINVAL
;
2042 return SOCKET_ERROR
;
2045 if (*OptionLength
< BufferSize
)
2047 *lpErrno
= WSAEFAULT
;
2048 *OptionLength
= BufferSize
;
2049 return SOCKET_ERROR
;
2051 RtlCopyMemory(OptionValue
, Buffer
, BufferSize
);
2055 case IPPROTO_TCP
: /* FIXME */
2057 *lpErrno
= Socket
->HelperData
->WSHGetSocketInformation(Socket
->HelperContext
,
2059 Socket
->TdiAddressHandle
,
2060 Socket
->TdiConnectionHandle
,
2064 (LPINT
)OptionLength
);
2065 return (*lpErrno
== 0) ? 0 : SOCKET_ERROR
;
2075 IN CONST CHAR FAR
* optval
,
2079 PSOCKET_INFORMATION Socket
;
2081 /* Get the Socket Structure associate to this Socket*/
2082 Socket
= GetSocketStructure(s
);
2085 *lpErrno
= WSAENOTSOCK
;
2086 return SOCKET_ERROR
;
2090 /* FIXME: We should handle some cases here */
2093 *lpErrno
= Socket
->HelperData
->WSHSetSocketInformation(Socket
->HelperContext
,
2095 Socket
->TdiAddressHandle
,
2096 Socket
->TdiConnectionHandle
,
2101 return (*lpErrno
== 0) ? 0 : SOCKET_ERROR
;
2105 * FUNCTION: Initialize service provider for a client
2107 * wVersionRequested = Highest WinSock SPI version that the caller can use
2108 * lpWSPData = Address of WSPDATA structure to initialize
2109 * lpProtocolInfo = Pointer to structure that defines the desired protocol
2110 * UpcallTable = Pointer to upcall table of the WinSock DLL
2111 * lpProcTable = Address of procedure table to initialize
2113 * Status of operation
2117 WSPStartup(IN WORD wVersionRequested
,
2118 OUT LPWSPDATA lpWSPData
,
2119 IN LPWSAPROTOCOL_INFOW lpProtocolInfo
,
2120 IN WSPUPCALLTABLE UpcallTable
,
2121 OUT LPWSPPROC_TABLE lpProcTable
)
2126 AFD_DbgPrint(MAX_TRACE
, ("wVersionRequested (0x%X) \n", wVersionRequested
));
2128 Upcalls
= UpcallTable
;
2130 if (Status
== NO_ERROR
)
2132 lpProcTable
->lpWSPAccept
= WSPAccept
;
2133 lpProcTable
->lpWSPAddressToString
= WSPAddressToString
;
2134 lpProcTable
->lpWSPAsyncSelect
= WSPAsyncSelect
;
2135 lpProcTable
->lpWSPBind
= WSPBind
;
2136 lpProcTable
->lpWSPCancelBlockingCall
= WSPCancelBlockingCall
;
2137 lpProcTable
->lpWSPCleanup
= WSPCleanup
;
2138 lpProcTable
->lpWSPCloseSocket
= WSPCloseSocket
;
2139 lpProcTable
->lpWSPConnect
= WSPConnect
;
2140 lpProcTable
->lpWSPDuplicateSocket
= WSPDuplicateSocket
;
2141 lpProcTable
->lpWSPEnumNetworkEvents
= WSPEnumNetworkEvents
;
2142 lpProcTable
->lpWSPEventSelect
= WSPEventSelect
;
2143 lpProcTable
->lpWSPGetOverlappedResult
= WSPGetOverlappedResult
;
2144 lpProcTable
->lpWSPGetPeerName
= WSPGetPeerName
;
2145 lpProcTable
->lpWSPGetSockName
= WSPGetSockName
;
2146 lpProcTable
->lpWSPGetSockOpt
= WSPGetSockOpt
;
2147 lpProcTable
->lpWSPGetQOSByName
= WSPGetQOSByName
;
2148 lpProcTable
->lpWSPIoctl
= WSPIoctl
;
2149 lpProcTable
->lpWSPJoinLeaf
= WSPJoinLeaf
;
2150 lpProcTable
->lpWSPListen
= WSPListen
;
2151 lpProcTable
->lpWSPRecv
= WSPRecv
;
2152 lpProcTable
->lpWSPRecvDisconnect
= WSPRecvDisconnect
;
2153 lpProcTable
->lpWSPRecvFrom
= WSPRecvFrom
;
2154 lpProcTable
->lpWSPSelect
= WSPSelect
;
2155 lpProcTable
->lpWSPSend
= WSPSend
;
2156 lpProcTable
->lpWSPSendDisconnect
= WSPSendDisconnect
;
2157 lpProcTable
->lpWSPSendTo
= WSPSendTo
;
2158 lpProcTable
->lpWSPSetSockOpt
= WSPSetSockOpt
;
2159 lpProcTable
->lpWSPShutdown
= WSPShutdown
;
2160 lpProcTable
->lpWSPSocket
= WSPSocket
;
2161 lpProcTable
->lpWSPStringToAddress
= WSPStringToAddress
;
2162 lpWSPData
->wVersion
= MAKEWORD(2, 2);
2163 lpWSPData
->wHighVersion
= MAKEWORD(2, 2);
2166 AFD_DbgPrint(MAX_TRACE
, ("Status (%d).\n", Status
));
2173 * FUNCTION: Cleans up service provider for a client
2175 * lpErrno = Address of buffer for error information
2177 * 0 if successful, or SOCKET_ERROR if not
2181 WSPCleanup(OUT LPINT lpErrno
)
2184 AFD_DbgPrint(MAX_TRACE
, ("\n"));
2185 AFD_DbgPrint(MAX_TRACE
, ("Leaving.\n"));
2186 *lpErrno
= NO_ERROR
;
2194 GetSocketInformation(PSOCKET_INFORMATION Socket
,
2195 ULONG AfdInformationClass
,
2196 PULONG Ulong OPTIONAL
,
2197 PLARGE_INTEGER LargeInteger OPTIONAL
)
2199 IO_STATUS_BLOCK IOSB
;
2204 Status
= NtCreateEvent(&SockEvent
,
2205 GENERIC_READ
| GENERIC_WRITE
,
2210 if( !NT_SUCCESS(Status
) )
2213 /* Set Info Class */
2214 InfoData
.InformationClass
= AfdInformationClass
;
2217 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
2228 /* Wait for return */
2229 if (Status
== STATUS_PENDING
)
2231 WaitForSingleObject(SockEvent
, INFINITE
);
2232 Status
= IOSB
.Status
;
2235 if (Status
!= STATUS_SUCCESS
)
2238 /* Return Information */
2241 *Ulong
= InfoData
.Information
.Ulong
;
2243 if (LargeInteger
!= NULL
)
2245 *LargeInteger
= InfoData
.Information
.LargeInteger
;
2248 NtClose( SockEvent
);
2256 SetSocketInformation(PSOCKET_INFORMATION Socket
,
2257 ULONG AfdInformationClass
,
2258 PULONG Ulong OPTIONAL
,
2259 PLARGE_INTEGER LargeInteger OPTIONAL
)
2261 IO_STATUS_BLOCK IOSB
;
2266 Status
= NtCreateEvent(&SockEvent
,
2267 GENERIC_READ
| GENERIC_WRITE
,
2272 if( !NT_SUCCESS(Status
) )
2275 /* Set Info Class */
2276 InfoData
.InformationClass
= AfdInformationClass
;
2278 /* Set Information */
2281 InfoData
.Information
.Ulong
= *Ulong
;
2283 if (LargeInteger
!= NULL
)
2285 InfoData
.Information
.LargeInteger
= *LargeInteger
;
2288 AFD_DbgPrint(MID_TRACE
,("XXX Info %x (Data %x)\n",
2289 AfdInformationClass
, *Ulong
));
2292 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
2303 /* Wait for return */
2304 if (Status
== STATUS_PENDING
)
2306 WaitForSingleObject(SockEvent
, INFINITE
);
2307 Status
= IOSB
.Status
;
2310 NtClose( SockEvent
);
2312 return Status
== STATUS_SUCCESS
? 0 : -1;
2317 GetSocketStructure(SOCKET Handle
)
2319 PSOCKET_INFORMATION CurrentSocket
;
2321 if (!SocketListHead
)
2324 /* This is a special case */
2325 if (SocketListHead
->Handle
== Handle
)
2326 return SocketListHead
;
2328 CurrentSocket
= SocketListHead
;
2329 while (CurrentSocket
->NextSocket
)
2331 if (CurrentSocket
->Handle
== Handle
)
2332 return CurrentSocket
;
2334 CurrentSocket
= CurrentSocket
->NextSocket
;
2340 int CreateContext(PSOCKET_INFORMATION Socket
)
2342 IO_STATUS_BLOCK IOSB
;
2343 SOCKET_CONTEXT ContextData
;
2347 Status
= NtCreateEvent(&SockEvent
,
2348 GENERIC_READ
| GENERIC_WRITE
,
2353 if( !NT_SUCCESS(Status
) )
2356 /* Create Context */
2357 ContextData
.SharedData
= Socket
->SharedData
;
2358 ContextData
.SizeOfHelperData
= 0;
2359 RtlCopyMemory (&ContextData
.LocalAddress
,
2360 Socket
->LocalAddress
,
2361 Socket
->SharedData
.SizeOfLocalAddress
);
2362 RtlCopyMemory (&ContextData
.RemoteAddress
,
2363 Socket
->RemoteAddress
,
2364 Socket
->SharedData
.SizeOfRemoteAddress
);
2367 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
2372 IOCTL_AFD_SET_CONTEXT
,
2374 sizeof(ContextData
),
2378 /* Wait for Completition */
2379 if (Status
== STATUS_PENDING
)
2381 WaitForSingleObject(SockEvent
, INFINITE
);
2382 Status
= IOSB
.Status
;
2385 NtClose( SockEvent
);
2387 return Status
== STATUS_SUCCESS
? 0 : -1;
2390 BOOLEAN
SockCreateOrReferenceAsyncThread(VOID
)
2392 HANDLE hAsyncThread
;
2393 DWORD AsyncThreadId
;
2395 OBJECT_HANDLE_ATTRIBUTE_INFORMATION HandleFlags
;
2398 /* Check if the Thread Already Exists */
2399 if (SockAsyncThreadRefCount
)
2404 /* Create the Completion Port */
2405 if (!SockAsyncCompletionPort
)
2407 Status
= NtCreateIoCompletion(&SockAsyncCompletionPort
,
2408 IO_COMPLETION_ALL_ACCESS
,
2410 2); // Allow 2 threads only
2412 /* Protect Handle */
2413 HandleFlags
.ProtectFromClose
= TRUE
;
2414 HandleFlags
.Inherit
= FALSE
;
2415 Status
= NtSetInformationObject(SockAsyncCompletionPort
,
2416 ObjectHandleFlagInformation
,
2418 sizeof(HandleFlags
));
2421 /* Create the Async Event */
2422 Status
= NtCreateEvent(&AsyncEvent
,
2428 /* Create the Async Thread */
2429 hAsyncThread
= CreateThread(NULL
,
2431 (LPTHREAD_START_ROUTINE
)SockAsyncThread
,
2436 /* Close the Handle */
2437 NtClose(hAsyncThread
);
2439 /* Increase the Reference Count */
2440 SockAsyncThreadRefCount
++;
2444 int SockAsyncThread(PVOID ThreadParam
)
2447 PASYNC_COMPLETION_ROUTINE AsyncCompletionRoutine
;
2448 IO_STATUS_BLOCK IOSB
;
2451 /* Make the Thread Higher Priority */
2452 SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_ABOVE_NORMAL
);
2454 /* Do a KQUEUE/WorkItem Style Loop, thanks to IoCompletion Ports */
2457 Status
= NtRemoveIoCompletion (SockAsyncCompletionPort
,
2458 (PVOID
*)&AsyncCompletionRoutine
,
2462 /* Call the Async Function */
2463 if (NT_SUCCESS(Status
))
2465 (*AsyncCompletionRoutine
)(AsyncContext
, &IOSB
);
2469 /* It Failed, sleep for a second */
2472 } while ((Status
!= STATUS_TIMEOUT
));
2474 /* The Thread has Ended */
2478 BOOLEAN
SockGetAsyncSelectHelperAfdHandle(VOID
)
2480 UNICODE_STRING AfdHelper
;
2481 OBJECT_ATTRIBUTES ObjectAttributes
;
2482 IO_STATUS_BLOCK IoSb
;
2484 FILE_COMPLETION_INFORMATION CompletionInfo
;
2485 OBJECT_HANDLE_ATTRIBUTE_INFORMATION HandleFlags
;
2487 /* First, make sure we're not already intialized */
2488 if (SockAsyncHelperAfdHandle
)
2493 /* Set up Handle Name and Object */
2494 RtlInitUnicodeString(&AfdHelper
, L
"\\Device\\Afd\\AsyncSelectHlp" );
2495 InitializeObjectAttributes(&ObjectAttributes
,
2497 OBJ_INHERIT
| OBJ_CASE_INSENSITIVE
,
2501 /* Open the Handle to AFD */
2502 Status
= NtCreateFile(&SockAsyncHelperAfdHandle
,
2503 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
,
2508 FILE_SHARE_READ
| FILE_SHARE_WRITE
,
2515 * Now Set up the Completion Port Information
2516 * This means that whenever a Poll is finished, the routine will be executed
2518 CompletionInfo
.Port
= SockAsyncCompletionPort
;
2519 CompletionInfo
.Key
= SockAsyncSelectCompletionRoutine
;
2520 Status
= NtSetInformationFile(SockAsyncHelperAfdHandle
,
2523 sizeof(CompletionInfo
),
2524 FileCompletionInformation
);
2527 /* Protect the Handle */
2528 HandleFlags
.ProtectFromClose
= TRUE
;
2529 HandleFlags
.Inherit
= FALSE
;
2530 Status
= NtSetInformationObject(SockAsyncCompletionPort
,
2531 ObjectHandleFlagInformation
,
2533 sizeof(HandleFlags
));
2536 /* Set this variable to true so that Send/Recv/Accept will know wether to renable disabled events */
2537 SockAsyncSelectCalled
= TRUE
;
2541 VOID
SockAsyncSelectCompletionRoutine(PVOID Context
, PIO_STATUS_BLOCK IoStatusBlock
)
2544 PASYNC_DATA AsyncData
= Context
;
2545 PSOCKET_INFORMATION Socket
;
2548 /* Get the Socket */
2549 Socket
= AsyncData
->ParentSocket
;
2551 /* Check if the Sequence Number Changed behind our back */
2552 if (AsyncData
->SequenceNumber
!= Socket
->SharedData
.SequenceNumber
)
2557 /* Check we were manually called b/c of a failure */
2558 if (!NT_SUCCESS(IoStatusBlock
->Status
))
2560 /* FIXME: Perform Upcall */
2564 for (x
= 1; x
; x
<<=1)
2566 switch (AsyncData
->AsyncSelectInfo
.Handles
[0].Events
& x
)
2568 case AFD_EVENT_RECEIVE
:
2569 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_READ
) &&
2570 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_READ
))
2572 /* Make the Notifcation */
2573 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2574 Socket
->SharedData
.wMsg
,
2576 WSAMAKESELECTREPLY(FD_READ
, 0));
2577 /* Disable this event until the next read(); */
2578 Socket
->SharedData
.AsyncDisabledEvents
|= FD_READ
;
2582 case AFD_EVENT_OOB_RECEIVE
:
2583 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_OOB
) &&
2584 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_OOB
))
2586 /* Make the Notifcation */
2587 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2588 Socket
->SharedData
.wMsg
,
2590 WSAMAKESELECTREPLY(FD_OOB
, 0));
2591 /* Disable this event until the next read(); */
2592 Socket
->SharedData
.AsyncDisabledEvents
|= FD_OOB
;
2596 case AFD_EVENT_SEND
:
2597 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_WRITE
) &&
2598 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_WRITE
))
2600 /* Make the Notifcation */
2601 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2602 Socket
->SharedData
.wMsg
,
2604 WSAMAKESELECTREPLY(FD_WRITE
, 0));
2605 /* Disable this event until the next write(); */
2606 Socket
->SharedData
.AsyncDisabledEvents
|= FD_WRITE
;
2610 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
2611 case AFD_EVENT_CONNECT
:
2612 case AFD_EVENT_CONNECT_FAIL
:
2613 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_CONNECT
) &&
2614 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_CONNECT
))
2616 /* Make the Notifcation */
2617 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2618 Socket
->SharedData
.wMsg
,
2620 WSAMAKESELECTREPLY(FD_CONNECT
, 0));
2621 /* Disable this event forever; */
2622 Socket
->SharedData
.AsyncDisabledEvents
|= FD_CONNECT
;
2626 case AFD_EVENT_ACCEPT
:
2627 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_ACCEPT
) &&
2628 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_ACCEPT
))
2630 /* Make the Notifcation */
2631 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2632 Socket
->SharedData
.wMsg
,
2634 WSAMAKESELECTREPLY(FD_ACCEPT
, 0));
2635 /* Disable this event until the next accept(); */
2636 Socket
->SharedData
.AsyncDisabledEvents
|= FD_ACCEPT
;
2640 case AFD_EVENT_DISCONNECT
:
2641 case AFD_EVENT_ABORT
:
2642 case AFD_EVENT_CLOSE
:
2643 if (0 != (Socket
->SharedData
.AsyncEvents
& FD_CLOSE
) &&
2644 0 == (Socket
->SharedData
.AsyncDisabledEvents
& FD_CLOSE
))
2646 /* Make the Notifcation */
2647 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
.hWnd
,
2648 Socket
->SharedData
.wMsg
,
2650 WSAMAKESELECTREPLY(FD_CLOSE
, 0));
2651 /* Disable this event forever; */
2652 Socket
->SharedData
.AsyncDisabledEvents
|= FD_CLOSE
;
2655 /* FIXME: Support QOS */
2659 /* Check if there are any events left for us to check */
2660 if ((Socket
->SharedData
.AsyncEvents
& (~Socket
->SharedData
.AsyncDisabledEvents
)) == 0 )
2666 SockProcessAsyncSelect(Socket
, AsyncData
);
2670 VOID
SockProcessAsyncSelect(PSOCKET_INFORMATION Socket
, PASYNC_DATA AsyncData
)
2673 ULONG lNetworkEvents
;
2676 /* Set up the Async Data Event Info */
2677 AsyncData
->AsyncSelectInfo
.Timeout
.HighPart
= 0x7FFFFFFF;
2678 AsyncData
->AsyncSelectInfo
.Timeout
.LowPart
= 0xFFFFFFFF;
2679 AsyncData
->AsyncSelectInfo
.HandleCount
= 1;
2680 AsyncData
->AsyncSelectInfo
.Exclusive
= TRUE
;
2681 AsyncData
->AsyncSelectInfo
.Handles
[0].Handle
= Socket
->Handle
;
2682 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
= 0;
2684 /* Remove unwanted events */
2685 lNetworkEvents
= Socket
->SharedData
.AsyncEvents
& (~Socket
->SharedData
.AsyncDisabledEvents
);
2687 /* Set Events to wait for */
2688 if (lNetworkEvents
& FD_READ
)
2690 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_RECEIVE
;
2693 if (lNetworkEvents
& FD_WRITE
)
2695 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_SEND
;
2698 if (lNetworkEvents
& FD_OOB
)
2700 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_OOB_RECEIVE
;
2703 if (lNetworkEvents
& FD_ACCEPT
)
2705 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_ACCEPT
;
2708 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
2709 if (lNetworkEvents
& FD_CONNECT
)
2711 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_CONNECT
| AFD_EVENT_CONNECT_FAIL
;
2714 if (lNetworkEvents
& FD_CLOSE
)
2716 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_DISCONNECT
| AFD_EVENT_ABORT
| AFD_EVENT_CLOSE
;
2719 if (lNetworkEvents
& FD_QOS
)
2721 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_QOS
;
2724 if (lNetworkEvents
& FD_GROUP_QOS
)
2726 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_GROUP_QOS
;
2730 Status
= NtDeviceIoControlFile (SockAsyncHelperAfdHandle
,
2734 &AsyncData
->IoStatusBlock
,
2736 &AsyncData
->AsyncSelectInfo
,
2737 sizeof(AsyncData
->AsyncSelectInfo
),
2738 &AsyncData
->AsyncSelectInfo
,
2739 sizeof(AsyncData
->AsyncSelectInfo
));
2741 /* I/O Manager Won't call the completion routine, let's do it manually */
2742 if (NT_SUCCESS(Status
))
2748 AsyncData
->IoStatusBlock
.Status
= Status
;
2749 SockAsyncSelectCompletionRoutine(AsyncData
, &AsyncData
->IoStatusBlock
);
2753 VOID
SockProcessQueuedAsyncSelect(PVOID Context
, PIO_STATUS_BLOCK IoStatusBlock
)
2755 PASYNC_DATA AsyncData
= Context
;
2756 BOOL FreeContext
= TRUE
;
2757 PSOCKET_INFORMATION Socket
;
2759 /* Get the Socket */
2760 Socket
= AsyncData
->ParentSocket
;
2762 /* If someone closed it, stop the function */
2763 if (Socket
->SharedData
.State
!= SocketClosed
)
2765 /* Check if the Sequence Number changed by now, in which case quit */
2766 if (AsyncData
->SequenceNumber
== Socket
->SharedData
.SequenceNumber
)
2768 /* Do the actuall select, if needed */
2769 if ((Socket
->SharedData
.AsyncEvents
& (~Socket
->SharedData
.AsyncDisabledEvents
)))
2771 SockProcessAsyncSelect(Socket
, AsyncData
);
2772 FreeContext
= FALSE
;
2777 /* Free the Context */
2780 HeapFree(GetProcessHeap(), 0, AsyncData
);
2787 SockReenableAsyncSelectEvent (IN PSOCKET_INFORMATION Socket
,
2790 PASYNC_DATA AsyncData
;
2792 /* Make sure the event is actually disabled */
2793 if (!(Socket
->SharedData
.AsyncDisabledEvents
& Event
))
2799 Socket
->SharedData
.AsyncDisabledEvents
&= ~Event
;
2801 /* Return if no more events are being polled */
2802 if ((Socket
->SharedData
.AsyncEvents
& (~Socket
->SharedData
.AsyncDisabledEvents
)) == 0 )
2807 /* Wait on new events */
2808 AsyncData
= HeapAlloc(GetProcessHeap(), 0, sizeof(ASYNC_DATA
));
2809 if (!AsyncData
) return;
2811 /* Create the Asynch Thread if Needed */
2812 SockCreateOrReferenceAsyncThread();
2814 /* Increase the sequence number to stop anything else */
2815 Socket
->SharedData
.SequenceNumber
++;
2817 /* Set up the Async Data */
2818 AsyncData
->ParentSocket
= Socket
;
2819 AsyncData
->SequenceNumber
= Socket
->SharedData
.SequenceNumber
;
2821 /* Begin Async Select by using I/O Completion */
2822 NtSetIoCompletion(SockAsyncCompletionPort
,
2823 (PVOID
)&SockProcessQueuedAsyncSelect
,
2834 DllMain(HANDLE hInstDll
,
2841 case DLL_PROCESS_ATTACH
:
2843 AFD_DbgPrint(MAX_TRACE
, ("Loading MSAFD.DLL \n"));
2845 /* Don't need thread attach notifications
2846 so disable them to improve performance */
2847 DisableThreadLibraryCalls(hInstDll
);
2849 /* List of DLL Helpers */
2850 InitializeListHead(&SockHelpersListHead
);
2852 /* Heap to use when allocating */
2853 GlobalHeap
= GetProcessHeap();
2855 AFD_DbgPrint(MAX_TRACE
, ("MSAFD.DLL has been loaded\n"));
2859 case DLL_THREAD_ATTACH
:
2862 case DLL_THREAD_DETACH
:
2865 case DLL_PROCESS_DETACH
:
2869 AFD_DbgPrint(MAX_TRACE
, ("DllMain of msafd.dll (leaving)\n"));