2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS Ancillary Function Driver DLL
4 * FILE: dll/win32/msafd/misc/dllmain.c
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 #include <wine/debug.h>
19 WINE_DEFAULT_DEBUG_CHANNEL(msafd
);
22 WSPUPCALLTABLE Upcalls
;
23 DWORD CatalogEntryId
; /* CatalogEntryId for 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
= NULL
;
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
;
73 PSOCK_SHARED_INFO SharedData
= NULL
;
75 TRACE("Creating Socket, getting TDI Name - AddressFamily (%d) SocketType (%d) Protocol (%d).\n",
76 AddressFamily
, SocketType
, Protocol
);
78 if (lpProtocolInfo
&& lpProtocolInfo
->dwServiceFlags3
!= 0 && lpProtocolInfo
->dwServiceFlags4
!= 0)
80 /* Duplpicating socket from different process */
81 if ((HANDLE
)lpProtocolInfo
->dwServiceFlags3
== INVALID_HANDLE_VALUE
)
86 if ((HANDLE
)lpProtocolInfo
->dwServiceFlags4
== INVALID_HANDLE_VALUE
)
91 SharedData
= MapViewOfFile((HANDLE
)lpProtocolInfo
->dwServiceFlags3
,
95 sizeof(SOCK_SHARED_INFO
));
101 InterlockedIncrement(&SharedData
->RefCount
);
102 AddressFamily
= SharedData
->AddressFamily
;
103 SocketType
= SharedData
->SocketType
;
104 Protocol
= SharedData
->Protocol
;
107 if (AddressFamily
== AF_UNSPEC
&& SocketType
== 0 && Protocol
== 0)
113 /* Set the defaults */
114 if (AddressFamily
== AF_UNSPEC
)
115 AddressFamily
= AF_INET
;
122 SocketType
= SOCK_STREAM
;
125 SocketType
= SOCK_DGRAM
;
128 SocketType
= SOCK_RAW
;
131 TRACE("Unknown Protocol (%d). We will try SOCK_STREAM.\n", Protocol
);
132 SocketType
= SOCK_STREAM
;
142 Protocol
= IPPROTO_TCP
;
145 Protocol
= IPPROTO_UDP
;
148 Protocol
= IPPROTO_RAW
;
151 TRACE("Unknown SocketType (%d). We will try IPPROTO_TCP.\n", SocketType
);
152 Protocol
= IPPROTO_TCP
;
157 /* Get Helper Data and Transport */
158 Status
= SockGetTdiName (&AddressFamily
,
168 /* Check for error */
169 if (Status
!= NO_ERROR
)
171 ERR("SockGetTdiName: Status %x\n", Status
);
175 /* AFD Device Name */
176 RtlInitUnicodeString(&DevName
, L
"\\Device\\Afd\\Endpoint");
178 /* Set Socket Data */
179 Socket
= HeapAlloc(GlobalHeap
, 0, sizeof(*Socket
));
185 RtlZeroMemory(Socket
, sizeof(*Socket
));
188 Socket
->SharedData
= SharedData
;
189 Socket
->SharedDataHandle
= (HANDLE
)lpProtocolInfo
->dwServiceFlags3
;
190 Sock
= (HANDLE
)lpProtocolInfo
->dwServiceFlags4
;
191 Socket
->Handle
= (SOCKET
)lpProtocolInfo
->dwServiceFlags4
;
195 Socket
->SharedDataHandle
= INVALID_HANDLE_VALUE
;
196 Socket
->SharedData
= HeapAlloc(GlobalHeap
, 0, sizeof(*Socket
->SharedData
));
197 if (!Socket
->SharedData
)
202 RtlZeroMemory(Socket
->SharedData
, sizeof(*Socket
->SharedData
));
203 Socket
->SharedData
->State
= SocketOpen
;
204 Socket
->SharedData
->RefCount
= 1L;
205 Socket
->SharedData
->Listening
= FALSE
;
206 Socket
->SharedData
->AddressFamily
= AddressFamily
;
207 Socket
->SharedData
->SocketType
= SocketType
;
208 Socket
->SharedData
->Protocol
= Protocol
;
209 Socket
->SharedData
->SizeOfLocalAddress
= HelperData
->MaxWSAddressLength
;
210 Socket
->SharedData
->SizeOfRemoteAddress
= HelperData
->MaxWSAddressLength
;
211 Socket
->SharedData
->UseDelayedAcceptance
= HelperData
->UseDelayedAcceptance
;
212 Socket
->SharedData
->CreateFlags
= dwFlags
;
213 Socket
->SharedData
->ServiceFlags1
= lpProtocolInfo
->dwServiceFlags1
;
214 Socket
->SharedData
->ProviderFlags
= lpProtocolInfo
->dwProviderFlags
;
215 Socket
->SharedData
->UseSAN
= FALSE
;
216 Socket
->SharedData
->NonBlocking
= FALSE
; /* Sockets start blocking */
217 Socket
->SharedData
->RecvTimeout
= INFINITE
;
218 Socket
->SharedData
->SendTimeout
= INFINITE
;
219 Socket
->SharedData
->OobInline
= FALSE
;
221 /* Ask alex about this */
222 if( Socket
->SharedData
->SocketType
== SOCK_DGRAM
||
223 Socket
->SharedData
->SocketType
== SOCK_RAW
)
225 TRACE("Connectionless socket\n");
226 Socket
->SharedData
->ServiceFlags1
|= XP1_CONNECTIONLESS
;
228 Socket
->Handle
= INVALID_SOCKET
;
231 Socket
->HelperContext
= HelperDLLContext
;
232 Socket
->HelperData
= HelperData
;
233 Socket
->HelperEvents
= HelperEvents
;
234 Socket
->LocalAddress
= &Socket
->SharedData
->WSLocalAddress
;
235 Socket
->RemoteAddress
= &Socket
->SharedData
->WSRemoteAddress
;
236 Socket
->SanData
= NULL
;
237 RtlCopyMemory(&Socket
->ProtocolInfo
, lpProtocolInfo
, sizeof(Socket
->ProtocolInfo
));
242 SizeOfPacket
= TransportName
.Length
+ sizeof(AFD_CREATE_PACKET
) + sizeof(WCHAR
);
245 SizeOfEA
= SizeOfPacket
+ sizeof(FILE_FULL_EA_INFORMATION
) + AFD_PACKET_COMMAND_LENGTH
;
247 /* Set up EA Buffer */
248 EABuffer
= HeapAlloc(GlobalHeap
, 0, SizeOfEA
);
255 RtlZeroMemory(EABuffer
, SizeOfEA
);
256 EABuffer
->NextEntryOffset
= 0;
258 EABuffer
->EaNameLength
= AFD_PACKET_COMMAND_LENGTH
;
259 RtlCopyMemory (EABuffer
->EaName
,
261 AFD_PACKET_COMMAND_LENGTH
+ 1);
262 EABuffer
->EaValueLength
= SizeOfPacket
;
264 /* Set up AFD Packet */
265 AfdPacket
= (PAFD_CREATE_PACKET
)(EABuffer
->EaName
+ EABuffer
->EaNameLength
+ 1);
266 AfdPacket
->SizeOfTransportName
= TransportName
.Length
;
267 RtlCopyMemory (AfdPacket
->TransportName
,
268 TransportName
.Buffer
,
269 TransportName
.Length
+ sizeof(WCHAR
));
270 AfdPacket
->GroupID
= g
;
272 /* Set up Endpoint Flags */
273 if ((Socket
->SharedData
->ServiceFlags1
& XP1_CONNECTIONLESS
) != 0)
275 if ((SocketType
!= SOCK_DGRAM
) && (SocketType
!= SOCK_RAW
))
277 /* Only RAW or UDP can be Connectionless */
281 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_CONNECTIONLESS
;
284 if ((Socket
->SharedData
->ServiceFlags1
& XP1_MESSAGE_ORIENTED
) != 0)
286 if (SocketType
== SOCK_STREAM
)
288 if ((Socket
->SharedData
->ServiceFlags1
& XP1_PSEUDO_STREAM
) == 0)
290 /* The Provider doesn't actually support Message Oriented Streams */
295 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_MESSAGE_ORIENTED
;
298 if (SocketType
== SOCK_RAW
) AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_RAW
;
300 if (dwFlags
& (WSA_FLAG_MULTIPOINT_C_ROOT
|
301 WSA_FLAG_MULTIPOINT_C_LEAF
|
302 WSA_FLAG_MULTIPOINT_D_ROOT
|
303 WSA_FLAG_MULTIPOINT_D_LEAF
))
305 if ((Socket
->SharedData
->ServiceFlags1
& XP1_SUPPORT_MULTIPOINT
) == 0)
307 /* The Provider doesn't actually support Multipoint */
311 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_MULTIPOINT
;
313 if (dwFlags
& WSA_FLAG_MULTIPOINT_C_ROOT
)
315 if (((Socket
->SharedData
->ServiceFlags1
& XP1_MULTIPOINT_CONTROL_PLANE
) == 0)
316 || ((dwFlags
& WSA_FLAG_MULTIPOINT_C_LEAF
) != 0))
318 /* The Provider doesn't support Control Planes, or you already gave a leaf */
322 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_C_ROOT
;
325 if (dwFlags
& WSA_FLAG_MULTIPOINT_D_ROOT
)
327 if (((Socket
->SharedData
->ServiceFlags1
& XP1_MULTIPOINT_DATA_PLANE
) == 0)
328 || ((dwFlags
& WSA_FLAG_MULTIPOINT_D_LEAF
) != 0))
330 /* The Provider doesn't support Data Planes, or you already gave a leaf */
334 AfdPacket
->EndpointFlags
|= AFD_ENDPOINT_D_ROOT
;
338 /* Set up Object Attributes */
339 InitializeObjectAttributes (&Object
,
341 OBJ_CASE_INSENSITIVE
| OBJ_INHERIT
,
345 /* Create the Socket as asynchronous. That means we have to block
346 ourselves after every call to NtDeviceIoControlFile. This is
347 because the kernel doesn't support overlapping synchronous I/O
348 requests (made from multiple threads) at this time (Sep 2005) */
349 Status
= NtCreateFile(&Sock
,
350 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
,
355 FILE_SHARE_READ
| FILE_SHARE_WRITE
,
361 HeapFree(GlobalHeap
, 0, EABuffer
);
363 if (!NT_SUCCESS(Status
))
365 ERR("Failed to open socket. Status 0x%08x\n", Status
);
366 Status
= TranslateNtStatusError(Status
);
371 Socket
->Handle
= (SOCKET
)Sock
;
373 /* Save Group Info */
376 GetSocketInformation(Socket
,
377 AFD_INFO_GROUP_ID_TYPE
,
383 Socket
->SharedData
->GroupID
= GroupData
.u
.LowPart
;
384 Socket
->SharedData
->GroupType
= GroupData
.u
.HighPart
;
387 /* Get Window Sizes and Save them */
388 GetSocketInformation (Socket
,
389 AFD_INFO_SEND_WINDOW_SIZE
,
391 &Socket
->SharedData
->SizeOfSendBuffer
,
396 GetSocketInformation (Socket
,
397 AFD_INFO_RECEIVE_WINDOW_SIZE
,
399 &Socket
->SharedData
->SizeOfRecvBuffer
,
405 /* Save in Process Sockets List */
406 EnterCriticalSection(&SocketListLock
);
407 Socket
->NextSocket
= SocketListHead
;
408 SocketListHead
= Socket
;
409 LeaveCriticalSection(&SocketListLock
);
411 /* Create the Socket Context */
412 CreateContext(Socket
);
415 Upcalls
.lpWPUModifyIFSHandle(Socket
->ProtocolInfo
.dwCatalogEntryId
, (SOCKET
)Sock
, lpErrno
);
417 /* Return Socket Handle */
418 TRACE("Success %x\n", Sock
);
423 ERR("Ending %x\n", Status
);
427 UnmapViewOfFile(SharedData
);
428 NtClose((HANDLE
)lpProtocolInfo
->dwServiceFlags3
);
432 if( Socket
&& Socket
->SharedData
)
433 HeapFree(GlobalHeap
, 0, Socket
->SharedData
);
437 HeapFree(GlobalHeap
, 0, Socket
);
440 HeapFree(GlobalHeap
, 0, EABuffer
);
445 return INVALID_SOCKET
;
453 IN DWORD dwProcessId
,
454 OUT LPWSAPROTOCOL_INFOW lpProtocolInfo
,
457 HANDLE hProcess
, hDuplicatedSharedData
, hDuplicatedHandle
;
458 PSOCKET_INFORMATION Socket
;
459 PSOCK_SHARED_INFO pSharedData
, pOldSharedData
;
462 if (Handle
== INVALID_SOCKET
)
463 return MsafdReturnWithErrno(STATUS_INVALID_PARAMETER
, lpErrno
, 0, NULL
);
464 Socket
= GetSocketStructure(Handle
);
468 *lpErrno
= WSAENOTSOCK
;
471 if ( !(hProcess
= OpenProcess(PROCESS_DUP_HANDLE
, FALSE
, dwProcessId
)) )
472 return MsafdReturnWithErrno(STATUS_INVALID_PARAMETER
, lpErrno
, 0, NULL
);
474 /* It is a not yet duplicated socket, so map the memory, copy the SharedData and free heap */
475 if( Socket
->SharedDataHandle
== INVALID_HANDLE_VALUE
)
477 Socket
->SharedDataHandle
= CreateFileMapping(INVALID_HANDLE_VALUE
,
479 PAGE_READWRITE
| SEC_COMMIT
,
481 (sizeof(SOCK_SHARED_INFO
) + PAGE_SIZE
- 1) & ~(PAGE_SIZE
- 1),
483 if( Socket
->SharedDataHandle
== INVALID_HANDLE_VALUE
)
484 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
485 pSharedData
= MapViewOfFile(Socket
->SharedDataHandle
,
489 sizeof(SOCK_SHARED_INFO
));
491 RtlCopyMemory(pSharedData
, Socket
->SharedData
, sizeof(SOCK_SHARED_INFO
));
492 pOldSharedData
= Socket
->SharedData
;
493 Socket
->SharedData
= pSharedData
;
494 HeapFree(GlobalHeap
, 0, pOldSharedData
);
496 /* Duplicate the handles for the new process */
497 bDuplicated
= DuplicateHandle(GetCurrentProcess(),
498 Socket
->SharedDataHandle
,
500 (LPHANDLE
)&hDuplicatedSharedData
,
503 DUPLICATE_SAME_ACCESS
);
507 return MsafdReturnWithErrno(STATUS_ACCESS_DENIED
, lpErrno
, 0, NULL
);
509 bDuplicated
= DuplicateHandle(GetCurrentProcess(),
510 (HANDLE
)Socket
->Handle
,
512 (LPHANDLE
)&hDuplicatedHandle
,
515 DUPLICATE_SAME_ACCESS
);
518 return MsafdReturnWithErrno(STATUS_ACCESS_DENIED
, lpErrno
, 0, NULL
);
522 return MsafdReturnWithErrno(STATUS_ACCESS_VIOLATION
, lpErrno
, 0, NULL
);
524 RtlCopyMemory(lpProtocolInfo
, &Socket
->ProtocolInfo
, sizeof(*lpProtocolInfo
));
526 lpProtocolInfo
->iAddressFamily
= Socket
->SharedData
->AddressFamily
;
527 lpProtocolInfo
->iProtocol
= Socket
->SharedData
->Protocol
;
528 lpProtocolInfo
->iSocketType
= Socket
->SharedData
->SocketType
;
529 lpProtocolInfo
->dwServiceFlags3
= (DWORD
)hDuplicatedSharedData
;
530 lpProtocolInfo
->dwServiceFlags4
= (DWORD
)hDuplicatedHandle
;
539 TranslateNtStatusError(NTSTATUS Status
)
543 case STATUS_CANT_WAIT
:
544 return WSAEWOULDBLOCK
;
552 case STATUS_FILE_CLOSED
:
553 return WSAECONNRESET
;
555 case STATUS_END_OF_FILE
:
559 return WSA_IO_PENDING
;
561 case STATUS_BUFFER_TOO_SMALL
:
562 case STATUS_BUFFER_OVERFLOW
:
565 case STATUS_NO_MEMORY
:
566 case STATUS_INSUFFICIENT_RESOURCES
:
569 case STATUS_INVALID_CONNECTION
:
572 case STATUS_PROTOCOL_NOT_SUPPORTED
:
573 return WSAEAFNOSUPPORT
;
575 case STATUS_INVALID_ADDRESS
:
576 return WSAEADDRNOTAVAIL
;
578 case STATUS_REMOTE_NOT_LISTENING
:
579 case STATUS_REMOTE_DISCONNECT
:
580 return WSAECONNREFUSED
;
582 case STATUS_NETWORK_UNREACHABLE
:
583 return WSAENETUNREACH
;
585 case STATUS_INVALID_PARAMETER
:
588 case STATUS_CANCELLED
:
589 return WSA_OPERATION_ABORTED
;
591 case STATUS_ADDRESS_ALREADY_EXISTS
:
592 return WSAEADDRINUSE
;
594 case STATUS_LOCAL_DISCONNECT
:
595 return WSAECONNABORTED
;
597 case STATUS_ACCESS_VIOLATION
:
600 case STATUS_ACCESS_DENIED
:
603 case STATUS_NOT_IMPLEMENTED
:
604 return WSAEOPNOTSUPP
;
607 ERR("MSAFD: Unhandled NTSTATUS value: 0x%x\n", Status
);
613 * FUNCTION: Closes an open socket
615 * s = Socket descriptor
616 * lpErrno = Address of buffer for error information
618 * NO_ERROR, or SOCKET_ERROR if the socket could not be closed
622 WSPCloseSocket(IN SOCKET Handle
,
625 IO_STATUS_BLOCK IoStatusBlock
;
626 PSOCKET_INFORMATION Socket
= NULL
, CurrentSocket
;
629 AFD_DISCONNECT_INFO DisconnectInfo
;
630 SOCKET_STATE OldState
;
631 LONG LingerWait
= -1;
634 /* Get the Socket Structure associate to this Socket*/
635 Socket
= GetSocketStructure(Handle
);
638 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
642 /* Create the Wait Event */
643 Status
= NtCreateEvent(&SockEvent
,
649 if(!NT_SUCCESS(Status
))
651 ERR("NtCreateEvent failed: 0x%08x", Status
);
655 if (Socket
->HelperEvents
& WSH_NOTIFY_CLOSE
)
657 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
659 Socket
->TdiAddressHandle
,
660 Socket
->TdiConnectionHandle
,
665 if (lpErrno
) *lpErrno
= Status
;
666 ERR("WSHNotify failed. Error 0x%#x", Status
);
672 /* If a Close is already in Process, give up */
673 if (Socket
->SharedData
->State
== SocketClosed
)
675 WARN("Socket is closing.\n");
677 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
681 /* Decrement reference count on SharedData */
682 References
= InterlockedDecrement(&Socket
->SharedData
->RefCount
);
686 /* Set the state to close */
687 OldState
= Socket
->SharedData
->State
;
688 Socket
->SharedData
->State
= SocketClosed
;
690 /* If SO_LINGER is ON and the Socket is connected, we need to disconnect */
691 /* FIXME: Should we do this on Datagram Sockets too? */
692 if ((OldState
== SocketConnected
) && (Socket
->SharedData
->LingerData
.l_onoff
))
694 ULONG SendsInProgress
;
697 /* We need to respect the timeout */
699 LingerWait
= Socket
->SharedData
->LingerData
.l_linger
* 1000;
701 /* Loop until no more sends are pending, within the timeout */
704 /* Find out how many Sends are in Progress */
705 if (GetSocketInformation(Socket
,
706 AFD_INFO_SENDS_IN_PROGRESS
,
713 /* Bail out if anything but NO_ERROR */
718 /* Bail out if no more sends are pending */
719 if (!SendsInProgress
)
726 * We have to execute a sleep, so it's kind of like
727 * a block. If the socket is Nonblock, we cannot
728 * go on since asynchronous operation is expected
729 * and we cannot offer it
731 if (Socket
->SharedData
->NonBlocking
)
733 WARN("Would block!\n");
735 Socket
->SharedData
->State
= OldState
;
736 if (lpErrno
) *lpErrno
= WSAEWOULDBLOCK
;
740 /* Now we can sleep, and decrement the linger wait */
742 * FIXME: It seems Windows does some funky acceleration
743 * since the waiting seems to be longer and longer. I
744 * don't think this improves performance so much, so we
745 * wait a fixed time instead.
748 LingerWait
-= SleepWait
;
752 if (OldState
== SocketConnected
)
756 DisconnectInfo
.Timeout
= RtlConvertLongToLargeInteger(0);
757 DisconnectInfo
.DisconnectType
= LingerWait
< 0 ? AFD_DISCONNECT_SEND
: AFD_DISCONNECT_ABORT
;
759 if (((DisconnectInfo
.DisconnectType
& AFD_DISCONNECT_SEND
) && (!Socket
->SharedData
->SendShutdown
)) ||
760 ((DisconnectInfo
.DisconnectType
& AFD_DISCONNECT_ABORT
) && (!Socket
->SharedData
->ReceiveShutdown
)))
763 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
768 IOCTL_AFD_DISCONNECT
,
770 sizeof(DisconnectInfo
),
774 /* Wait for return */
775 if (Status
== STATUS_PENDING
)
777 WaitForSingleObject(SockEvent
, INFINITE
);
778 Status
= IoStatusBlock
.Status
;
785 Socket
->HelperContext
= NULL
;
786 Socket
->SharedData
->AsyncDisabledEvents
= -1;
787 NtClose(Socket
->TdiAddressHandle
);
788 Socket
->TdiAddressHandle
= NULL
;
789 NtClose(Socket
->TdiConnectionHandle
);
790 Socket
->TdiConnectionHandle
= NULL
;
792 EnterCriticalSection(&SocketListLock
);
793 if (SocketListHead
== Socket
)
795 SocketListHead
= SocketListHead
->NextSocket
;
799 CurrentSocket
= SocketListHead
;
800 while (CurrentSocket
->NextSocket
)
802 if (CurrentSocket
->NextSocket
== Socket
)
804 CurrentSocket
->NextSocket
= CurrentSocket
->NextSocket
->NextSocket
;
808 CurrentSocket
= CurrentSocket
->NextSocket
;
811 LeaveCriticalSection(&SocketListLock
);
813 /* Close the handle */
814 NtClose((HANDLE
)Handle
);
817 if( Socket
->SharedDataHandle
!= INVALID_HANDLE_VALUE
)
819 /* It is a duplicated socket, so unmap the memory */
820 UnmapViewOfFile(Socket
->SharedData
);
821 NtClose(Socket
->SharedDataHandle
);
822 Socket
->SharedData
= NULL
;
824 if( !References
&& Socket
->SharedData
)
826 HeapFree(GlobalHeap
, 0, Socket
->SharedData
);
828 HeapFree(GlobalHeap
, 0, Socket
);
829 return MsafdReturnWithErrno(Status
, lpErrno
, 0, NULL
);
834 * FUNCTION: Associates a local address with a socket
836 * s = Socket descriptor
837 * name = Pointer to local address
838 * namelen = Length of name
839 * lpErrno = Address of buffer for error information
841 * 0, or SOCKET_ERROR if the socket could not be bound
845 WSPBind(SOCKET Handle
,
846 const struct sockaddr
*SocketAddress
,
847 int SocketAddressLength
,
850 IO_STATUS_BLOCK IOSB
;
851 PAFD_BIND_DATA BindData
;
852 PSOCKET_INFORMATION Socket
= NULL
;
854 SOCKADDR_INFO SocketInfo
;
857 /* Get the Socket Structure associate to this Socket*/
858 Socket
= GetSocketStructure(Handle
);
861 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
864 if (Socket
->SharedData
->State
!= SocketOpen
)
866 if (lpErrno
) *lpErrno
= WSAEINVAL
;
869 if (!SocketAddress
|| SocketAddressLength
< Socket
->SharedData
->SizeOfLocalAddress
)
871 if (lpErrno
) *lpErrno
= WSAEINVAL
;
875 /* Get Address Information */
876 Socket
->HelperData
->WSHGetSockaddrType ((PSOCKADDR
)SocketAddress
,
880 if (SocketInfo
.AddressInfo
== SockaddrAddressInfoBroadcast
&& !Socket
->SharedData
->Broadcast
)
882 if (lpErrno
) *lpErrno
= WSAEADDRNOTAVAIL
;
886 Status
= NtCreateEvent(&SockEvent
,
892 if (!NT_SUCCESS(Status
))
898 BindData
= HeapAlloc(GlobalHeap
, 0, 0xA + SocketAddressLength
);
901 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
904 /* Set up Address in TDI Format */
905 BindData
->Address
.TAAddressCount
= 1;
906 BindData
->Address
.Address
[0].AddressLength
= SocketAddressLength
- sizeof(SocketAddress
->sa_family
);
907 BindData
->Address
.Address
[0].AddressType
= SocketAddress
->sa_family
;
908 RtlCopyMemory (BindData
->Address
.Address
[0].Address
,
909 SocketAddress
->sa_data
,
910 SocketAddressLength
- sizeof(SocketAddress
->sa_family
));
912 /* Set the Share Type */
913 if (Socket
->SharedData
->ExclusiveAddressUse
)
915 BindData
->ShareType
= AFD_SHARE_EXCLUSIVE
;
917 else if (SocketInfo
.EndpointInfo
== SockaddrEndpointInfoWildcard
)
919 BindData
->ShareType
= AFD_SHARE_WILDCARD
;
921 else if (Socket
->SharedData
->ReuseAddresses
)
923 BindData
->ShareType
= AFD_SHARE_REUSE
;
927 BindData
->ShareType
= AFD_SHARE_UNIQUE
;
931 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
938 0xA + Socket
->SharedData
->SizeOfLocalAddress
, /* Can't figure out a way to calculate this in C*/
940 0xA + Socket
->SharedData
->SizeOfLocalAddress
); /* Can't figure out a way to calculate this C */
942 /* Wait for return */
943 if (Status
== STATUS_PENDING
)
945 WaitForSingleObject(SockEvent
, INFINITE
);
946 Status
= IOSB
.Status
;
949 NtClose( SockEvent
);
950 HeapFree(GlobalHeap
, 0, BindData
);
952 Socket
->SharedData
->SocketLastError
= TranslateNtStatusError(Status
);
953 if (Status
!= STATUS_SUCCESS
)
954 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
956 /* Set up Socket Data */
957 Socket
->SharedData
->State
= SocketBound
;
958 Socket
->TdiAddressHandle
= (HANDLE
)IOSB
.Information
;
960 if (Socket
->HelperEvents
& WSH_NOTIFY_BIND
)
962 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
964 Socket
->TdiAddressHandle
,
965 Socket
->TdiConnectionHandle
,
970 if (lpErrno
) *lpErrno
= Status
;
975 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
980 WSPListen(SOCKET Handle
,
984 IO_STATUS_BLOCK IOSB
;
985 AFD_LISTEN_DATA ListenData
;
986 PSOCKET_INFORMATION Socket
= NULL
;
990 /* Get the Socket Structure associate to this Socket*/
991 Socket
= GetSocketStructure(Handle
);
994 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
998 if (Socket
->SharedData
->Listening
)
1001 Status
= NtCreateEvent(&SockEvent
,
1007 if( !NT_SUCCESS(Status
) )
1008 return SOCKET_ERROR
;
1010 /* Set Up Listen Structure */
1011 ListenData
.UseSAN
= FALSE
;
1012 ListenData
.UseDelayedAcceptance
= Socket
->SharedData
->UseDelayedAcceptance
;
1013 ListenData
.Backlog
= Backlog
;
1016 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1021 IOCTL_AFD_START_LISTEN
,
1027 /* Wait for return */
1028 if (Status
== STATUS_PENDING
)
1030 WaitForSingleObject(SockEvent
, INFINITE
);
1031 Status
= IOSB
.Status
;
1034 NtClose( SockEvent
);
1036 Socket
->SharedData
->SocketLastError
= TranslateNtStatusError(Status
);
1037 if (Status
!= STATUS_SUCCESS
)
1038 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
1040 /* Set to Listening */
1041 Socket
->SharedData
->Listening
= TRUE
;
1043 if (Socket
->HelperEvents
& WSH_NOTIFY_LISTEN
)
1045 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
1047 Socket
->TdiAddressHandle
,
1048 Socket
->TdiConnectionHandle
,
1053 if (lpErrno
) *lpErrno
= Status
;
1054 return SOCKET_ERROR
;
1058 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
1064 WSPSelect(IN
int nfds
,
1065 IN OUT fd_set
*readfds OPTIONAL
,
1066 IN OUT fd_set
*writefds OPTIONAL
,
1067 IN OUT fd_set
*exceptfds OPTIONAL
,
1068 IN
const struct timeval
*timeout OPTIONAL
,
1071 IO_STATUS_BLOCK IOSB
;
1072 PAFD_POLL_INFO PollInfo
;
1075 ULONG PollBufferSize
;
1079 LARGE_INTEGER Timeout
;
1080 PSOCKET_INFORMATION Socket
;
1085 /* Find out how many sockets we have, and how large the buffer needs
1087 FD_ZERO(&selectfds
);
1088 if (readfds
!= NULL
)
1090 for (i
= 0; i
< readfds
->fd_count
; i
++)
1092 FD_SET(readfds
->fd_array
[i
], &selectfds
);
1095 if (writefds
!= NULL
)
1097 for (i
= 0; i
< writefds
->fd_count
; i
++)
1099 FD_SET(writefds
->fd_array
[i
], &selectfds
);
1102 if (exceptfds
!= NULL
)
1104 for (i
= 0; i
< exceptfds
->fd_count
; i
++)
1106 FD_SET(exceptfds
->fd_array
[i
], &selectfds
);
1110 HandleCount
= selectfds
.fd_count
;
1112 if ( HandleCount
== 0 )
1114 WARN("No handles! Returning SOCKET_ERROR\n", HandleCount
);
1115 if (lpErrno
) *lpErrno
= WSAEINVAL
;
1116 return SOCKET_ERROR
;
1119 PollBufferSize
= sizeof(*PollInfo
) + ((HandleCount
- 1) * sizeof(AFD_HANDLE
));
1121 TRACE("HandleCount: %u BufferSize: %u\n", HandleCount
, PollBufferSize
);
1123 /* Convert Timeout to NT Format */
1124 if (timeout
== NULL
)
1126 Timeout
.u
.LowPart
= -1;
1127 Timeout
.u
.HighPart
= 0x7FFFFFFF;
1128 TRACE("Infinite timeout\n");
1132 Timeout
= RtlEnlargedIntegerMultiply
1133 ((timeout
->tv_sec
* 1000) + (timeout
->tv_usec
/ 1000), -10000);
1134 /* Negative timeouts are illegal. Since the kernel represents an
1135 * incremental timeout as a negative number, we check for a positive
1138 if (Timeout
.QuadPart
> 0)
1140 if (lpErrno
) *lpErrno
= WSAEINVAL
;
1141 return SOCKET_ERROR
;
1143 TRACE("Timeout: Orig %d.%06d kernel %d\n",
1144 timeout
->tv_sec
, timeout
->tv_usec
,
1148 Status
= NtCreateEvent(&SockEvent
,
1154 if(!NT_SUCCESS(Status
))
1157 *lpErrno
= WSAEFAULT
;
1159 ERR("NtCreateEvent failed, 0x%08x\n", Status
);
1160 return SOCKET_ERROR
;
1164 PollBuffer
= HeapAlloc(GlobalHeap
, 0, PollBufferSize
);
1169 *lpErrno
= WSAEFAULT
;
1171 return SOCKET_ERROR
;
1174 PollInfo
= (PAFD_POLL_INFO
)PollBuffer
;
1176 RtlZeroMemory( PollInfo
, PollBufferSize
);
1178 /* Number of handles for AFD to Check */
1179 PollInfo
->Exclusive
= FALSE
;
1180 PollInfo
->Timeout
= Timeout
;
1182 for (i
= 0; i
< selectfds
.fd_count
; i
++)
1184 PollInfo
->Handles
[i
].Handle
= selectfds
.fd_array
[i
];
1186 if (readfds
!= NULL
) {
1187 for (i
= 0; i
< readfds
->fd_count
; i
++)
1189 for (j
= 0; j
< HandleCount
; j
++)
1191 if (PollInfo
->Handles
[j
].Handle
== readfds
->fd_array
[i
])
1194 if (j
>= HandleCount
)
1196 ERR("Error while counting readfds %ld > %ld\n", j
, HandleCount
);
1197 if (lpErrno
) *lpErrno
= WSAEFAULT
;
1198 HeapFree(GlobalHeap
, 0, PollBuffer
);
1200 return SOCKET_ERROR
;
1202 Socket
= GetSocketStructure(readfds
->fd_array
[i
]);
1205 ERR("Invalid socket handle provided in readfds %d\n", readfds
->fd_array
[i
]);
1206 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
1207 HeapFree(GlobalHeap
, 0, PollBuffer
);
1209 return SOCKET_ERROR
;
1211 PollInfo
->Handles
[j
].Events
|= AFD_EVENT_RECEIVE
|
1212 AFD_EVENT_DISCONNECT
|
1216 //if (Socket->SharedData->OobInline != 0)
1217 // PollInfo->Handles[j].Events |= AFD_EVENT_OOB_RECEIVE;
1220 if (writefds
!= NULL
)
1222 for (i
= 0; i
< writefds
->fd_count
; i
++)
1224 for (j
= 0; j
< HandleCount
; j
++)
1226 if (PollInfo
->Handles
[j
].Handle
== writefds
->fd_array
[i
])
1229 if (j
>= HandleCount
)
1231 ERR("Error while counting writefds %ld > %ld\n", j
, HandleCount
);
1232 if (lpErrno
) *lpErrno
= WSAEFAULT
;
1233 HeapFree(GlobalHeap
, 0, PollBuffer
);
1235 return SOCKET_ERROR
;
1237 Socket
= GetSocketStructure(writefds
->fd_array
[i
]);
1240 ERR("Invalid socket handle provided in writefds %d\n", writefds
->fd_array
[i
]);
1241 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
1242 HeapFree(GlobalHeap
, 0, PollBuffer
);
1244 return SOCKET_ERROR
;
1246 PollInfo
->Handles
[j
].Handle
= writefds
->fd_array
[i
];
1247 PollInfo
->Handles
[j
].Events
|= AFD_EVENT_SEND
;
1248 if (Socket
->SharedData
->NonBlocking
!= 0)
1249 PollInfo
->Handles
[j
].Events
|= AFD_EVENT_CONNECT
;
1252 if (exceptfds
!= NULL
)
1254 for (i
= 0; i
< exceptfds
->fd_count
; i
++)
1256 for (j
= 0; j
< HandleCount
; j
++)
1258 if (PollInfo
->Handles
[j
].Handle
== exceptfds
->fd_array
[i
])
1261 if (j
> HandleCount
)
1263 ERR("Error while counting exceptfds %ld > %ld\n", j
, HandleCount
);
1264 if (lpErrno
) *lpErrno
= WSAEFAULT
;
1265 HeapFree(GlobalHeap
, 0, PollBuffer
);
1267 return SOCKET_ERROR
;
1269 Socket
= GetSocketStructure(exceptfds
->fd_array
[i
]);
1272 TRACE("Invalid socket handle provided in exceptfds %d\n", exceptfds
->fd_array
[i
]);
1273 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
1274 HeapFree(GlobalHeap
, 0, PollBuffer
);
1276 return SOCKET_ERROR
;
1278 PollInfo
->Handles
[j
].Handle
= exceptfds
->fd_array
[i
];
1279 if (Socket
->SharedData
->OobInline
== 0)
1280 PollInfo
->Handles
[j
].Events
|= AFD_EVENT_OOB_RECEIVE
;
1281 if (Socket
->SharedData
->NonBlocking
!= 0)
1282 PollInfo
->Handles
[j
].Events
|= AFD_EVENT_CONNECT_FAIL
;
1286 PollInfo
->HandleCount
= HandleCount
;
1287 PollBufferSize
= FIELD_OFFSET(AFD_POLL_INFO
, Handles
) + PollInfo
->HandleCount
* sizeof(AFD_HANDLE
);
1290 Status
= NtDeviceIoControlFile((HANDLE
)PollInfo
->Handles
[0].Handle
,
1301 TRACE("DeviceIoControlFile => %x\n", Status
);
1303 /* Wait for Completion */
1304 if (Status
== STATUS_PENDING
)
1306 WaitForSingleObject(SockEvent
, INFINITE
);
1307 Status
= IOSB
.Status
;
1310 /* Clear the Structures */
1318 /* Loop through return structure */
1319 HandleCount
= PollInfo
->HandleCount
;
1321 /* Return in FDSET Format */
1322 for (i
= 0; i
< HandleCount
; i
++)
1324 Events
= PollInfo
->Handles
[i
].Events
;
1325 Handle
= PollInfo
->Handles
[i
].Handle
;
1326 for(x
= 1; x
; x
<<=1)
1328 Socket
= GetSocketStructure(Handle
);
1331 TRACE("Invalid socket handle found %d\n", Handle
);
1332 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
1333 HeapFree(GlobalHeap
, 0, PollBuffer
);
1335 return SOCKET_ERROR
;
1339 case AFD_EVENT_RECEIVE
:
1340 case AFD_EVENT_DISCONNECT
:
1341 case AFD_EVENT_ABORT
:
1342 case AFD_EVENT_ACCEPT
:
1343 case AFD_EVENT_CLOSE
:
1344 TRACE("Event %x on handle %x\n",
1347 if ((Events
& x
) == AFD_EVENT_DISCONNECT
|| (Events
& x
) == AFD_EVENT_CLOSE
)
1348 Socket
->SharedData
->SocketLastError
= WSAECONNRESET
;
1349 if ((Events
& x
) == AFD_EVENT_ABORT
)
1350 Socket
->SharedData
->SocketLastError
= WSAECONNABORTED
;
1352 FD_SET(Handle
, readfds
);
1354 case AFD_EVENT_SEND
:
1355 TRACE("Event %x on handle %x\n",
1359 FD_SET(Handle
, writefds
);
1361 case AFD_EVENT_CONNECT
:
1362 TRACE("Event %x on handle %x\n",
1365 if( writefds
&& Socket
->SharedData
->NonBlocking
!= 0 )
1366 FD_SET(Handle
, writefds
);
1368 case AFD_EVENT_OOB_RECEIVE
:
1369 TRACE("Event %x on handle %x\n",
1372 if( readfds
&& Socket
->SharedData
->OobInline
!= 0 )
1373 FD_SET(Handle
, readfds
);
1374 if( exceptfds
&& Socket
->SharedData
->OobInline
== 0 )
1375 FD_SET(Handle
, exceptfds
);
1377 case AFD_EVENT_CONNECT_FAIL
:
1378 TRACE("Event %x on handle %x\n",
1381 if( exceptfds
&& Socket
->SharedData
->NonBlocking
!= 0 )
1382 FD_SET(Handle
, exceptfds
);
1388 HeapFree( GlobalHeap
, 0, PollBuffer
);
1389 NtClose( SockEvent
);
1393 switch( IOSB
.Status
)
1395 case STATUS_SUCCESS
:
1396 case STATUS_TIMEOUT
:
1400 *lpErrno
= WSAEINVAL
;
1403 TRACE("*lpErrno = %x\n", *lpErrno
);
1406 HandleCount
= (readfds
? readfds
->fd_count
: 0) +
1407 (writefds
&& writefds
!= readfds
? writefds
->fd_count
: 0) +
1408 (exceptfds
&& exceptfds
!= readfds
&& exceptfds
!= writefds
? exceptfds
->fd_count
: 0);
1410 TRACE("%d events\n", HandleCount
);
1416 GetCurrentTimeInSeconds(VOID
)
1418 SYSTEMTIME st1970
= { 1970, 1, 0, 1, 0, 0, 0, 0 };
1425 GetSystemTimeAsFileTime(&Time
.ft
);
1426 SystemTimeToFileTime(&st1970
, &u1970
.ft
);
1427 return (DWORD
)((Time
.ll
- u1970
.ll
) / 10000000ULL);
1432 WSPAccept(SOCKET Handle
,
1433 struct sockaddr
*SocketAddress
,
1434 int *SocketAddressLength
,
1435 LPCONDITIONPROC lpfnCondition
,
1436 DWORD dwCallbackData
,
1439 IO_STATUS_BLOCK IOSB
;
1440 PAFD_RECEIVED_ACCEPT_DATA ListenReceiveData
;
1441 AFD_ACCEPT_DATA AcceptData
;
1442 AFD_DEFER_ACCEPT_DATA DeferData
;
1443 AFD_PENDING_ACCEPT_DATA PendingAcceptData
;
1444 PSOCKET_INFORMATION Socket
= NULL
;
1446 struct fd_set ReadSet
;
1447 struct timeval Timeout
;
1448 PVOID PendingData
= NULL
;
1449 ULONG PendingDataLength
= 0;
1450 PVOID CalleeDataBuffer
;
1451 WSABUF CallerData
, CalleeID
, CallerID
, CalleeData
;
1452 PSOCKADDR RemoteAddress
= NULL
;
1455 SOCKET AcceptSocket
;
1456 PSOCKET_INFORMATION AcceptSocketInfo
;
1457 UCHAR ReceiveBuffer
[0x1A];
1460 /* Get the Socket Structure associate to this Socket*/
1461 Socket
= GetSocketStructure(Handle
);
1464 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
1465 return SOCKET_ERROR
;
1467 if (!Socket
->SharedData
->Listening
)
1469 if (lpErrno
) *lpErrno
= WSAEINVAL
;
1470 return SOCKET_ERROR
;
1472 if ((SocketAddress
&& !SocketAddressLength
) ||
1473 (SocketAddressLength
&& !SocketAddress
) ||
1474 (SocketAddressLength
&& *SocketAddressLength
< sizeof(SOCKADDR
)))
1476 if (lpErrno
) *lpErrno
= WSAEFAULT
;
1477 return INVALID_SOCKET
;
1480 Status
= NtCreateEvent(&SockEvent
,
1486 if( !NT_SUCCESS(Status
) )
1488 return SOCKET_ERROR
;
1491 /* Dynamic Structure...ugh */
1492 ListenReceiveData
= (PAFD_RECEIVED_ACCEPT_DATA
)ReceiveBuffer
;
1494 /* If this is non-blocking, make sure there's something for us to accept */
1496 FD_SET(Socket
->Handle
, &ReadSet
);
1500 if (WSPSelect(0, &ReadSet
, NULL
, NULL
, &Timeout
, lpErrno
) == SOCKET_ERROR
)
1503 return SOCKET_ERROR
;
1506 if (ReadSet
.fd_array
[0] != Socket
->Handle
)
1509 if (lpErrno
) *lpErrno
= WSAEWOULDBLOCK
;
1510 return SOCKET_ERROR
;
1514 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1519 IOCTL_AFD_WAIT_FOR_LISTEN
,
1523 0xA + sizeof(*ListenReceiveData
));
1525 /* Wait for return */
1526 if (Status
== STATUS_PENDING
)
1528 WaitForSingleObject(SockEvent
, INFINITE
);
1529 Status
= IOSB
.Status
;
1532 if (!NT_SUCCESS(Status
))
1534 NtClose( SockEvent
);
1535 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1538 if (lpfnCondition
!= NULL
)
1540 if ((Socket
->SharedData
->ServiceFlags1
& XP1_CONNECT_DATA
) != 0)
1542 /* Find out how much data is pending */
1543 PendingAcceptData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1544 PendingAcceptData
.ReturnSize
= TRUE
;
1547 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1552 IOCTL_AFD_GET_PENDING_CONNECT_DATA
,
1554 sizeof(PendingAcceptData
),
1556 sizeof(PendingAcceptData
));
1558 /* Wait for return */
1559 if (Status
== STATUS_PENDING
)
1561 WaitForSingleObject(SockEvent
, INFINITE
);
1562 Status
= IOSB
.Status
;
1565 if (!NT_SUCCESS(Status
))
1567 NtClose( SockEvent
);
1568 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1571 /* How much data to allocate */
1572 PendingDataLength
= IOSB
.Information
;
1574 if (PendingDataLength
)
1576 /* Allocate needed space */
1577 PendingData
= HeapAlloc(GlobalHeap
, 0, PendingDataLength
);
1580 return MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1583 /* We want the data now */
1584 PendingAcceptData
.ReturnSize
= FALSE
;
1587 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1592 IOCTL_AFD_GET_PENDING_CONNECT_DATA
,
1594 sizeof(PendingAcceptData
),
1598 /* Wait for return */
1599 if (Status
== STATUS_PENDING
)
1601 WaitForSingleObject(SockEvent
, INFINITE
);
1602 Status
= IOSB
.Status
;
1605 if (!NT_SUCCESS(Status
))
1607 NtClose( SockEvent
);
1608 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1613 if ((Socket
->SharedData
->ServiceFlags1
& XP1_QOS_SUPPORTED
) != 0)
1615 /* I don't support this yet */
1618 /* Build Callee ID */
1619 CalleeID
.buf
= (PVOID
)Socket
->LocalAddress
;
1620 CalleeID
.len
= Socket
->SharedData
->SizeOfLocalAddress
;
1622 RemoteAddress
= HeapAlloc(GlobalHeap
, 0, sizeof(*RemoteAddress
));
1625 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1628 /* Set up Address in SOCKADDR Format */
1629 RtlCopyMemory (RemoteAddress
,
1630 &ListenReceiveData
->Address
.Address
[0].AddressType
,
1631 sizeof(*RemoteAddress
));
1633 /* Build Caller ID */
1634 CallerID
.buf
= (PVOID
)RemoteAddress
;
1635 CallerID
.len
= sizeof(*RemoteAddress
);
1637 /* Build Caller Data */
1638 CallerData
.buf
= PendingData
;
1639 CallerData
.len
= PendingDataLength
;
1641 /* Check if socket supports Conditional Accept */
1642 if (Socket
->SharedData
->UseDelayedAcceptance
!= 0)
1644 /* Allocate Buffer for Callee Data */
1645 CalleeDataBuffer
= HeapAlloc(GlobalHeap
, 0, 4096);
1646 if (!CalleeDataBuffer
) {
1647 return MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1649 CalleeData
.buf
= CalleeDataBuffer
;
1650 CalleeData
.len
= 4096;
1659 /* Call the Condition Function */
1660 CallBack
= (lpfnCondition
)(&CallerID
,
1661 CallerData
.buf
== NULL
? NULL
: &CallerData
,
1665 CalleeData
.buf
== NULL
? NULL
: &CalleeData
,
1669 if (((CallBack
== CF_ACCEPT
) && GroupID
) != 0)
1671 /* TBD: Check for Validity */
1674 if (CallBack
== CF_ACCEPT
)
1676 if ((Socket
->SharedData
->ServiceFlags1
& XP1_QOS_SUPPORTED
) != 0)
1678 /* I don't support this yet */
1682 // SockSetConnectData Sockets(SocketID), IOCTL_AFD_SET_CONNECT_DATA, CalleeData.Buffer, CalleeData.BuffSize, 0
1687 /* Callback rejected. Build Defer Structure */
1688 DeferData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1689 DeferData
.RejectConnection
= (CallBack
== CF_REJECT
);
1692 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1697 IOCTL_AFD_DEFER_ACCEPT
,
1703 /* Wait for return */
1704 if (Status
== STATUS_PENDING
)
1706 WaitForSingleObject(SockEvent
, INFINITE
);
1707 Status
= IOSB
.Status
;
1710 NtClose( SockEvent
);
1712 if (!NT_SUCCESS(Status
))
1714 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1717 if (CallBack
== CF_REJECT
)
1719 if (lpErrno
) *lpErrno
= WSAECONNREFUSED
;
1720 return SOCKET_ERROR
;
1724 if (lpErrno
) *lpErrno
= WSAECONNREFUSED
;
1725 return SOCKET_ERROR
;
1730 /* Create a new Socket */
1731 AcceptSocket
= WSPSocket (Socket
->SharedData
->AddressFamily
,
1732 Socket
->SharedData
->SocketType
,
1733 Socket
->SharedData
->Protocol
,
1734 &Socket
->ProtocolInfo
,
1736 Socket
->SharedData
->CreateFlags
,
1738 if (AcceptSocket
== INVALID_SOCKET
)
1739 return SOCKET_ERROR
;
1741 /* Set up the Accept Structure */
1742 AcceptData
.ListenHandle
= (HANDLE
)AcceptSocket
;
1743 AcceptData
.SequenceNumber
= ListenReceiveData
->SequenceNumber
;
1745 /* Send IOCTL to Accept */
1746 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
1757 /* Wait for return */
1758 if (Status
== STATUS_PENDING
)
1760 WaitForSingleObject(SockEvent
, INFINITE
);
1761 Status
= IOSB
.Status
;
1764 Socket
->SharedData
->SocketLastError
= TranslateNtStatusError(Status
);
1765 if (!NT_SUCCESS(Status
))
1768 WSPCloseSocket( AcceptSocket
, lpErrno
);
1769 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
1772 AcceptSocketInfo
= GetSocketStructure(AcceptSocket
);
1773 if (!AcceptSocketInfo
)
1776 WSPCloseSocket( AcceptSocket
, lpErrno
);
1777 return MsafdReturnWithErrno( STATUS_PROTOCOL_NOT_SUPPORTED
, lpErrno
, 0, NULL
);
1780 AcceptSocketInfo
->SharedData
->State
= SocketConnected
;
1781 AcceptSocketInfo
->SharedData
->ConnectTime
= GetCurrentTimeInSeconds();
1783 /* Return Address in SOCKADDR FORMAT */
1786 RtlCopyMemory (SocketAddress
,
1787 &ListenReceiveData
->Address
.Address
[0].AddressType
,
1788 sizeof(*RemoteAddress
));
1789 if( SocketAddressLength
)
1790 *SocketAddressLength
= sizeof(*RemoteAddress
);
1793 NtClose( SockEvent
);
1795 /* Re-enable Async Event */
1796 SockReenableAsyncSelectEvent(Socket
, FD_ACCEPT
);
1798 TRACE("Socket %x\n", AcceptSocket
);
1800 if (Status
== STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_ACCEPT
))
1802 Status
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
1804 Socket
->TdiAddressHandle
,
1805 Socket
->TdiConnectionHandle
,
1810 if (lpErrno
) *lpErrno
= Status
;
1811 return SOCKET_ERROR
;
1815 if (lpErrno
) *lpErrno
= NO_ERROR
;
1818 return AcceptSocket
;
1823 WSPConnect(SOCKET Handle
,
1824 const struct sockaddr
* SocketAddress
,
1825 int SocketAddressLength
,
1826 LPWSABUF lpCallerData
,
1827 LPWSABUF lpCalleeData
,
1832 IO_STATUS_BLOCK IOSB
;
1833 PAFD_CONNECT_INFO ConnectInfo
= NULL
;
1834 PSOCKET_INFORMATION Socket
;
1837 ULONG ConnectDataLength
;
1838 ULONG InConnectDataLength
;
1839 INT BindAddressLength
;
1840 PSOCKADDR BindAddress
;
1842 int SocketDataLength
;
1844 TRACE("Called (%lx) %lx:%d\n", Handle
, ((const struct sockaddr_in
*)SocketAddress
)->sin_addr
, ((const struct sockaddr_in
*)SocketAddress
)->sin_port
);
1846 /* Get the Socket Structure associate to this Socket*/
1847 Socket
= GetSocketStructure(Handle
);
1850 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
1851 return SOCKET_ERROR
;
1854 Status
= NtCreateEvent(&SockEvent
,
1860 if (!NT_SUCCESS(Status
))
1861 return SOCKET_ERROR
;
1864 if (Socket
->SharedData
->State
== SocketOpen
)
1866 /* Get the Wildcard Address */
1867 BindAddressLength
= Socket
->HelperData
->MaxWSAddressLength
;
1868 BindAddress
= HeapAlloc(GetProcessHeap(), 0, BindAddressLength
);
1872 return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES
, lpErrno
, 0, NULL
);
1874 Socket
->HelperData
->WSHGetWildcardSockaddr (Socket
->HelperContext
,
1876 &BindAddressLength
);
1878 if (WSPBind(Handle
, BindAddress
, BindAddressLength
, lpErrno
) == SOCKET_ERROR
)
1879 return SOCKET_ERROR
;
1882 /* Set the Connect Data */
1883 if (lpCallerData
!= NULL
)
1885 ConnectDataLength
= lpCallerData
->len
;
1886 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1891 IOCTL_AFD_SET_CONNECT_DATA
,
1896 /* Wait for return */
1897 if (Status
== STATUS_PENDING
)
1899 WaitForSingleObject(SockEvent
, INFINITE
);
1900 Status
= IOSB
.Status
;
1903 if (Status
!= STATUS_SUCCESS
)
1907 /* Calculate the size of SocketAddress->sa_data */
1908 SocketDataLength
= SocketAddressLength
- FIELD_OFFSET(struct sockaddr
, sa_data
);
1910 /* Allocate a connection info buffer with SocketDataLength bytes of payload */
1911 ConnectInfo
= HeapAlloc(GetProcessHeap(), 0,
1912 FIELD_OFFSET(AFD_CONNECT_INFO
,
1913 RemoteAddress
.Address
[0].Address
[SocketDataLength
]));
1916 Status
= STATUS_INSUFFICIENT_RESOURCES
;
1920 /* Set up Address in TDI Format */
1921 ConnectInfo
->RemoteAddress
.TAAddressCount
= 1;
1922 ConnectInfo
->RemoteAddress
.Address
[0].AddressLength
= SocketDataLength
;
1923 ConnectInfo
->RemoteAddress
.Address
[0].AddressType
= SocketAddress
->sa_family
;
1924 RtlCopyMemory(ConnectInfo
->RemoteAddress
.Address
[0].Address
,
1925 SocketAddress
->sa_data
,
1929 * Disable FD_WRITE and FD_CONNECT
1930 * The latter fixes a race condition where the FD_CONNECT is re-enabled
1931 * at the end of this function right after the Async Thread disables it.
1932 * This should only happen at the *next* WSPConnect
1934 if (Socket
->SharedData
->AsyncEvents
& FD_CONNECT
)
1936 Socket
->SharedData
->AsyncDisabledEvents
|= FD_CONNECT
| FD_WRITE
;
1939 /* Tell AFD that we want Connection Data back, have it allocate a buffer */
1940 if (lpCalleeData
!= NULL
)
1942 InConnectDataLength
= lpCalleeData
->len
;
1943 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1948 IOCTL_AFD_SET_CONNECT_DATA_SIZE
,
1949 &InConnectDataLength
,
1950 sizeof(InConnectDataLength
),
1954 /* Wait for return */
1955 if (Status
== STATUS_PENDING
)
1957 WaitForSingleObject(SockEvent
, INFINITE
);
1958 Status
= IOSB
.Status
;
1961 if (Status
!= STATUS_SUCCESS
)
1965 /* AFD doesn't seem to care if these are invalid, but let's 0 them anyways */
1966 ConnectInfo
->Root
= 0;
1967 ConnectInfo
->UseSAN
= FALSE
;
1968 ConnectInfo
->Unknown
= 0;
1970 /* FIXME: Handle Async Connect */
1971 if (Socket
->SharedData
->NonBlocking
)
1973 ERR("Async Connect UNIMPLEMENTED!\n");
1977 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
1987 /* Wait for return */
1988 if (Status
== STATUS_PENDING
)
1990 WaitForSingleObject(SockEvent
, INFINITE
);
1991 Status
= IOSB
.Status
;
1994 Socket
->SharedData
->SocketLastError
= TranslateNtStatusError(Status
);
1995 if (Status
!= STATUS_SUCCESS
)
1998 Socket
->SharedData
->State
= SocketConnected
;
1999 Socket
->TdiConnectionHandle
= (HANDLE
)IOSB
.Information
;
2000 Socket
->SharedData
->ConnectTime
= GetCurrentTimeInSeconds();
2002 /* Get any pending connect data */
2003 if (lpCalleeData
!= NULL
)
2005 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
2010 IOCTL_AFD_GET_CONNECT_DATA
,
2015 /* Wait for return */
2016 if (Status
== STATUS_PENDING
)
2018 WaitForSingleObject(SockEvent
, INFINITE
);
2019 Status
= IOSB
.Status
;
2023 TRACE("Ending %lx\n", IOSB
.Status
);
2026 if (ConnectInfo
) HeapFree(GetProcessHeap(), 0, ConnectInfo
);
2028 /* Re-enable Async Event */
2029 SockReenableAsyncSelectEvent(Socket
, FD_WRITE
);
2031 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
2032 SockReenableAsyncSelectEvent(Socket
, FD_CONNECT
);
2036 if (Status
== STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_CONNECT
))
2038 Errno
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
2040 Socket
->TdiAddressHandle
,
2041 Socket
->TdiConnectionHandle
,
2042 WSH_NOTIFY_CONNECT
);
2046 if (lpErrno
) *lpErrno
= Errno
;
2047 return SOCKET_ERROR
;
2050 else if (Status
!= STATUS_SUCCESS
&& (Socket
->HelperEvents
& WSH_NOTIFY_CONNECT_ERROR
))
2052 Errno
= Socket
->HelperData
->WSHNotify(Socket
->HelperContext
,
2054 Socket
->TdiAddressHandle
,
2055 Socket
->TdiConnectionHandle
,
2056 WSH_NOTIFY_CONNECT_ERROR
);
2060 if (lpErrno
) *lpErrno
= Errno
;
2061 return SOCKET_ERROR
;
2065 return MsafdReturnWithErrno(Status
, lpErrno
, 0, NULL
);
2069 WSPShutdown(SOCKET Handle
,
2074 IO_STATUS_BLOCK IOSB
;
2075 AFD_DISCONNECT_INFO DisconnectInfo
;
2076 PSOCKET_INFORMATION Socket
= NULL
;
2082 /* Get the Socket Structure associate to this Socket*/
2083 Socket
= GetSocketStructure(Handle
);
2086 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
2087 return SOCKET_ERROR
;
2090 Status
= NtCreateEvent(&SockEvent
,
2096 if( !NT_SUCCESS(Status
) )
2097 return SOCKET_ERROR
;
2099 /* Set AFD Disconnect Type */
2103 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_RECV
;
2104 Socket
->SharedData
->ReceiveShutdown
= TRUE
;
2107 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_SEND
;
2108 Socket
->SharedData
->SendShutdown
= TRUE
;
2111 DisconnectInfo
.DisconnectType
= AFD_DISCONNECT_RECV
| AFD_DISCONNECT_SEND
;
2112 Socket
->SharedData
->ReceiveShutdown
= TRUE
;
2113 Socket
->SharedData
->SendShutdown
= TRUE
;
2117 DisconnectInfo
.Timeout
= RtlConvertLongToLargeInteger(-1000000);
2120 Status
= NtDeviceIoControlFile((HANDLE
)Handle
,
2125 IOCTL_AFD_DISCONNECT
,
2127 sizeof(DisconnectInfo
),
2131 /* Wait for return */
2132 if (Status
== STATUS_PENDING
)
2134 WaitForSingleObject(SockEvent
, INFINITE
);
2135 Status
= IOSB
.Status
;
2140 NtClose( SockEvent
);
2142 Socket
->SharedData
->SocketLastError
= TranslateNtStatusError(Status
);
2143 return MsafdReturnWithErrno( Status
, lpErrno
, 0, NULL
);
2149 WSPGetSockName(IN SOCKET Handle
,
2150 OUT LPSOCKADDR Name
,
2151 IN OUT LPINT NameLength
,
2154 IO_STATUS_BLOCK IOSB
;
2155 ULONG TdiAddressSize
;
2156 PTDI_ADDRESS_INFO TdiAddress
;
2157 PTRANSPORT_ADDRESS SocketAddress
;
2158 PSOCKET_INFORMATION Socket
= NULL
;
2162 /* Get the Socket Structure associate to this Socket*/
2163 Socket
= GetSocketStructure(Handle
);
2166 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
2167 return SOCKET_ERROR
;
2170 if (!Name
|| !NameLength
)
2172 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2173 return SOCKET_ERROR
;
2176 Status
= NtCreateEvent(&SockEvent
,
2182 if( !NT_SUCCESS(Status
) )
2183 return SOCKET_ERROR
;
2185 /* Allocate a buffer for the address */
2187 sizeof(TRANSPORT_ADDRESS
) + Socket
->SharedData
->SizeOfLocalAddress
;
2188 TdiAddress
= HeapAlloc(GlobalHeap
, 0, TdiAddressSize
);
2190 if ( TdiAddress
== NULL
)
2192 NtClose( SockEvent
);
2193 if (lpErrno
) *lpErrno
= WSAENOBUFS
;
2194 return SOCKET_ERROR
;
2197 SocketAddress
= &TdiAddress
->Address
;
2200 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
2205 IOCTL_AFD_GET_SOCK_NAME
,
2211 /* Wait for return */
2212 if (Status
== STATUS_PENDING
)
2214 WaitForSingleObject(SockEvent
, INFINITE
);
2215 Status
= IOSB
.Status
;
2218 NtClose( SockEvent
);
2220 if (NT_SUCCESS(Status
))
2222 if (*NameLength
>= Socket
->SharedData
->SizeOfLocalAddress
)
2224 Name
->sa_family
= SocketAddress
->Address
[0].AddressType
;
2225 RtlCopyMemory (Name
->sa_data
,
2226 SocketAddress
->Address
[0].Address
,
2227 SocketAddress
->Address
[0].AddressLength
);
2228 *NameLength
= Socket
->SharedData
->SizeOfLocalAddress
;
2229 TRACE("NameLength %d Address: %x Port %x\n",
2230 *NameLength
, ((struct sockaddr_in
*)Name
)->sin_addr
.s_addr
,
2231 ((struct sockaddr_in
*)Name
)->sin_port
);
2232 HeapFree(GlobalHeap
, 0, TdiAddress
);
2237 HeapFree(GlobalHeap
, 0, TdiAddress
);
2238 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2239 return SOCKET_ERROR
;
2243 HeapFree(GlobalHeap
, 0, TdiAddress
);
2245 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
2251 WSPGetPeerName(IN SOCKET s
,
2252 OUT LPSOCKADDR Name
,
2253 IN OUT LPINT NameLength
,
2256 IO_STATUS_BLOCK IOSB
;
2257 ULONG TdiAddressSize
;
2258 PTRANSPORT_ADDRESS SocketAddress
;
2259 PSOCKET_INFORMATION Socket
= NULL
;
2263 /* Get the Socket Structure associate to this Socket*/
2264 Socket
= GetSocketStructure(s
);
2267 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
2268 return SOCKET_ERROR
;
2271 if (Socket
->SharedData
->State
!= SocketConnected
)
2273 if (lpErrno
) *lpErrno
= WSAENOTCONN
;
2274 return SOCKET_ERROR
;
2277 if (!Name
|| !NameLength
)
2279 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2280 return SOCKET_ERROR
;
2283 Status
= NtCreateEvent(&SockEvent
,
2289 if( !NT_SUCCESS(Status
) )
2290 return SOCKET_ERROR
;
2292 /* Allocate a buffer for the address */
2293 TdiAddressSize
= sizeof(TRANSPORT_ADDRESS
) + Socket
->SharedData
->SizeOfRemoteAddress
;
2294 SocketAddress
= HeapAlloc(GlobalHeap
, 0, TdiAddressSize
);
2296 if ( SocketAddress
== NULL
)
2298 NtClose( SockEvent
);
2299 if (lpErrno
) *lpErrno
= WSAENOBUFS
;
2300 return SOCKET_ERROR
;
2304 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
2309 IOCTL_AFD_GET_PEER_NAME
,
2315 /* Wait for return */
2316 if (Status
== STATUS_PENDING
)
2318 WaitForSingleObject(SockEvent
, INFINITE
);
2319 Status
= IOSB
.Status
;
2322 NtClose( SockEvent
);
2324 if (NT_SUCCESS(Status
))
2326 if (*NameLength
>= Socket
->SharedData
->SizeOfRemoteAddress
)
2328 Name
->sa_family
= SocketAddress
->Address
[0].AddressType
;
2329 RtlCopyMemory (Name
->sa_data
,
2330 SocketAddress
->Address
[0].Address
,
2331 SocketAddress
->Address
[0].AddressLength
);
2332 *NameLength
= Socket
->SharedData
->SizeOfRemoteAddress
;
2333 TRACE("NameLength %d Address: %x Port %x\n",
2334 *NameLength
, ((struct sockaddr_in
*)Name
)->sin_addr
.s_addr
,
2335 ((struct sockaddr_in
*)Name
)->sin_port
);
2336 HeapFree(GlobalHeap
, 0, SocketAddress
);
2341 HeapFree(GlobalHeap
, 0, SocketAddress
);
2342 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2343 return SOCKET_ERROR
;
2347 HeapFree(GlobalHeap
, 0, SocketAddress
);
2349 return MsafdReturnWithErrno ( Status
, lpErrno
, 0, NULL
);
2354 WSPIoctl(IN SOCKET Handle
,
2355 IN DWORD dwIoControlCode
,
2356 IN LPVOID lpvInBuffer
,
2357 IN DWORD cbInBuffer
,
2358 OUT LPVOID lpvOutBuffer
,
2359 IN DWORD cbOutBuffer
,
2360 OUT LPDWORD lpcbBytesReturned
,
2361 IN LPWSAOVERLAPPED lpOverlapped
,
2362 IN LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine
,
2363 IN LPWSATHREADID lpThreadId
,
2366 PSOCKET_INFORMATION Socket
= NULL
;
2367 BOOL NeedsCompletion
= lpOverlapped
!= NULL
;
2368 BOOLEAN NonBlocking
;
2369 INT Errno
= NO_ERROR
, Ret
= SOCKET_ERROR
;
2372 /* Get the Socket Structure associate to this Socket*/
2373 Socket
= GetSocketStructure(Handle
);
2377 *lpErrno
= WSAENOTSOCK
;
2378 return SOCKET_ERROR
;
2381 if (!lpcbBytesReturned
&& !lpOverlapped
)
2384 *lpErrno
= WSAEFAULT
;
2385 return SOCKET_ERROR
;
2388 switch( dwIoControlCode
)
2391 if( cbInBuffer
< sizeof(INT
) || IS_INTRESOURCE(lpvInBuffer
) )
2396 NonBlocking
= *((PULONG
)lpvInBuffer
) ? TRUE
: FALSE
;
2397 /* Don't allow to go in blocking mode if WSPAsyncSelect or WSPEventSelect is pending */
2400 /* If there is an WSPAsyncSelect pending, fail with WSAEINVAL */
2401 if (Socket
->SharedData
->AsyncEvents
& (~Socket
->SharedData
->AsyncDisabledEvents
))
2406 /* If there is an WSPEventSelect pending, fail with WSAEINVAL */
2407 if (Socket
->NetworkEvents
)
2413 Socket
->SharedData
->NonBlocking
= NonBlocking
? 1 : 0;
2414 NeedsCompletion
= FALSE
;
2415 Errno
= SetSocketInformation(Socket
, AFD_INFO_BLOCKING_MODE
, &NonBlocking
, NULL
, NULL
, lpOverlapped
, lpCompletionRoutine
);
2416 if (Errno
== NO_ERROR
)
2420 if (IS_INTRESOURCE(lpvOutBuffer
) || cbOutBuffer
== 0)
2422 cbRet
= sizeof(ULONG
);
2426 if (cbOutBuffer
< sizeof(ULONG
))
2431 NeedsCompletion
= FALSE
;
2432 Errno
= GetSocketInformation(Socket
, AFD_INFO_RECEIVE_CONTENT_SIZE
, NULL
, (PULONG
)lpvOutBuffer
, NULL
, lpOverlapped
, lpCompletionRoutine
);
2433 if (Errno
== NO_ERROR
)
2435 cbRet
= sizeof(ULONG
);
2440 if (IS_INTRESOURCE(lpvOutBuffer
) || cbOutBuffer
== 0)
2442 cbRet
= sizeof(BOOL
);
2446 if (cbOutBuffer
< sizeof(BOOL
))
2451 if (Socket
->SharedData
->SocketType
!= SOCK_STREAM
)
2457 /* FIXME: Return false if OOBINLINE is true for now
2458 We should MSG_PEEK|MSG_OOB check with driver
2460 *(BOOL
*)lpvOutBuffer
= !Socket
->SharedData
->OobInline
;
2462 cbRet
= sizeof(BOOL
);
2466 case SIO_GET_EXTENSION_FUNCTION_POINTER
:
2469 case SIO_ADDRESS_LIST_QUERY
:
2470 if (IS_INTRESOURCE(lpvOutBuffer
) || cbOutBuffer
== 0)
2472 cbRet
= sizeof(SOCKET_ADDRESS_LIST
) + sizeof(Socket
->SharedData
->WSLocalAddress
);
2476 if (cbOutBuffer
< sizeof(INT
))
2482 cbRet
= sizeof(SOCKET_ADDRESS_LIST
) + sizeof(Socket
->SharedData
->WSLocalAddress
);
2484 ((SOCKET_ADDRESS_LIST
*)lpvOutBuffer
)->iAddressCount
= 1;
2486 if (cbOutBuffer
< (sizeof(SOCKET_ADDRESS_LIST
) + sizeof(Socket
->SharedData
->WSLocalAddress
)))
2492 ((SOCKET_ADDRESS_LIST
*)lpvOutBuffer
)->Address
[0].iSockaddrLength
= sizeof(Socket
->SharedData
->WSLocalAddress
);
2493 ((SOCKET_ADDRESS_LIST
*)lpvOutBuffer
)->Address
[0].lpSockaddr
= &Socket
->SharedData
->WSLocalAddress
;
2499 Errno
= Socket
->HelperData
->WSHIoctl(Socket
->HelperContext
,
2501 Socket
->TdiAddressHandle
,
2502 Socket
->TdiConnectionHandle
,
2510 lpCompletionRoutine
,
2513 if (Errno
== NO_ERROR
)
2517 if (lpOverlapped
&& NeedsCompletion
)
2519 lpOverlapped
->Internal
= Errno
;
2520 lpOverlapped
->InternalHigh
= cbRet
;
2521 if (lpCompletionRoutine
!= NULL
)
2523 lpCompletionRoutine(Errno
, cbRet
, lpOverlapped
, 0);
2525 if (lpOverlapped
->hEvent
)
2526 SetEvent(lpOverlapped
->hEvent
);
2527 if (!PostQueuedCompletionStatus((HANDLE
)Handle
, cbRet
, 0, lpOverlapped
))
2529 ERR("PostQueuedCompletionStatus failed %d\n", GetLastError());
2535 if (lpcbBytesReturned
)
2536 *lpcbBytesReturned
= cbRet
;
2543 WSPGetSockOpt(IN SOCKET Handle
,
2546 OUT CHAR FAR
* OptionValue
,
2547 IN OUT LPINT OptionLength
,
2550 PSOCKET_INFORMATION Socket
= NULL
;
2559 /* Get the Socket Structure associate to this Socket*/
2560 Socket
= GetSocketStructure(Handle
);
2563 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
2564 return SOCKET_ERROR
;
2566 if (!OptionLength
|| !OptionValue
)
2568 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2569 return SOCKET_ERROR
;
2578 Buffer
= &Socket
->SharedData
->SocketType
;
2579 BufferSize
= sizeof(INT
);
2583 Buffer
= &Socket
->SharedData
->SizeOfRecvBuffer
;
2584 BufferSize
= sizeof(INT
);
2588 Buffer
= &Socket
->SharedData
->SizeOfSendBuffer
;
2589 BufferSize
= sizeof(INT
);
2593 BoolBuffer
= Socket
->SharedData
->Listening
;
2594 Buffer
= &BoolBuffer
;
2595 BufferSize
= sizeof(BOOL
);
2599 BoolBuffer
= Socket
->SharedData
->Broadcast
;
2600 Buffer
= &BoolBuffer
;
2601 BufferSize
= sizeof(BOOL
);
2605 BoolBuffer
= Socket
->SharedData
->Debug
;
2606 Buffer
= &BoolBuffer
;
2607 BufferSize
= sizeof(BOOL
);
2611 BoolBuffer
= (Socket
->SharedData
->LingerData
.l_onoff
== 0);
2612 Buffer
= &BoolBuffer
;
2613 BufferSize
= sizeof(BOOL
);
2617 if (Socket
->SharedData
->SocketType
== SOCK_DGRAM
)
2619 if (lpErrno
) *lpErrno
= WSAENOPROTOOPT
;
2620 return SOCKET_ERROR
;
2622 Buffer
= &Socket
->SharedData
->LingerData
;
2623 BufferSize
= sizeof(struct linger
);
2627 BoolBuffer
= (Socket
->SharedData
->OobInline
!= 0);
2628 Buffer
= &BoolBuffer
;
2629 BufferSize
= sizeof(BOOL
);
2634 /* These guys go directly to the helper */
2637 case SO_CONDITIONAL_ACCEPT
:
2638 BoolBuffer
= (Socket
->SharedData
->UseDelayedAcceptance
!= 0);
2639 Buffer
= &BoolBuffer
;
2640 BufferSize
= sizeof(BOOL
);
2644 BoolBuffer
= (Socket
->SharedData
->ReuseAddresses
!= 0);
2645 Buffer
= &BoolBuffer
;
2646 BufferSize
= sizeof(BOOL
);
2649 case SO_EXCLUSIVEADDRUSE
:
2650 BoolBuffer
= (Socket
->SharedData
->ExclusiveAddressUse
!= 0);
2651 Buffer
= &BoolBuffer
;
2652 BufferSize
= sizeof(BOOL
);
2656 Buffer
= &Socket
->SharedData
->SocketLastError
;
2657 BufferSize
= sizeof(INT
);
2660 case SO_CONNECT_TIME
:
2661 DwordBuffer
= GetCurrentTimeInSeconds() - Socket
->SharedData
->ConnectTime
;
2662 Buffer
= &DwordBuffer
;
2663 BufferSize
= sizeof(DWORD
);
2667 Buffer
= &Socket
->SharedData
->SendTimeout
;
2668 BufferSize
= sizeof(DWORD
);
2671 Buffer
= &Socket
->SharedData
->RecvTimeout
;
2672 BufferSize
= sizeof(DWORD
);
2674 case SO_PROTOCOL_INFOW
:
2675 Buffer
= &Socket
->ProtocolInfo
;
2676 BufferSize
= sizeof(Socket
->ProtocolInfo
);
2680 case SO_GROUP_PRIORITY
:
2681 case SO_MAX_MSG_SIZE
:
2684 DbgPrint("MSAFD: Get unknown optname %x\n", OptionName
);
2685 if (lpErrno
) *lpErrno
= WSAENOPROTOOPT
;
2686 return SOCKET_ERROR
;
2689 if (*OptionLength
< BufferSize
)
2691 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2692 *OptionLength
= BufferSize
;
2693 return SOCKET_ERROR
;
2695 RtlCopyMemory(OptionValue
, Buffer
, BufferSize
);
2700 if (lpErrno
) *lpErrno
= WSAEINVAL
;
2701 return SOCKET_ERROR
;
2705 Errno
= Socket
->HelperData
->WSHGetSocketInformation(Socket
->HelperContext
,
2707 Socket
->TdiAddressHandle
,
2708 Socket
->TdiConnectionHandle
,
2712 (LPINT
)OptionLength
);
2713 if (lpErrno
) *lpErrno
= Errno
;
2714 return (Errno
== NO_ERROR
) ? NO_ERROR
: SOCKET_ERROR
;
2723 IN CONST CHAR FAR
* optval
,
2727 PSOCKET_INFORMATION Socket
;
2730 /* Get the Socket Structure associate to this Socket*/
2731 Socket
= GetSocketStructure(s
);
2734 if (lpErrno
) *lpErrno
= WSAENOTSOCK
;
2735 return SOCKET_ERROR
;
2739 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2740 return SOCKET_ERROR
;
2744 /* FIXME: We should handle some more cases here */
2745 if (level
== SOL_SOCKET
)
2750 if (optlen
< sizeof(BOOL
))
2752 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2753 return SOCKET_ERROR
;
2755 Socket
->SharedData
->Broadcast
= (*optval
!= 0) ? 1 : 0;
2759 if (optlen
< sizeof(BOOL
))
2761 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2762 return SOCKET_ERROR
;
2764 Socket
->SharedData
->OobInline
= (*optval
!= 0) ? 1 : 0;
2768 if (optlen
< sizeof(BOOL
))
2770 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2771 return SOCKET_ERROR
;
2773 Socket
->SharedData
->LingerData
.l_onoff
= (*optval
!= 0) ? 0 : 1;
2777 if (optlen
< sizeof(BOOL
))
2779 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2780 return SOCKET_ERROR
;
2782 Socket
->SharedData
->ReuseAddresses
= (*optval
!= 0) ? 1 : 0;
2785 case SO_EXCLUSIVEADDRUSE
:
2786 if (optlen
< sizeof(BOOL
))
2788 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2789 return SOCKET_ERROR
;
2791 Socket
->SharedData
->ExclusiveAddressUse
= (*optval
!= 0) ? 1 : 0;
2795 if (optlen
< sizeof(struct linger
))
2797 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2798 return SOCKET_ERROR
;
2800 RtlCopyMemory(&Socket
->SharedData
->LingerData
,
2802 sizeof(struct linger
));
2806 if (optlen
< sizeof(DWORD
))
2808 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2809 return SOCKET_ERROR
;
2812 /* TODO: The total per-socket buffer space reserved for sends */
2813 ERR("Setting send buf to %x is not implemented yet\n", optval
);
2817 if (optlen
< sizeof(INT
))
2819 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2820 return SOCKET_ERROR
;
2823 RtlCopyMemory(&Socket
->SharedData
->SocketLastError
,
2829 if (optlen
< sizeof(DWORD
))
2831 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2832 return SOCKET_ERROR
;
2835 RtlCopyMemory(&Socket
->SharedData
->SendTimeout
,
2841 if (optlen
< sizeof(DWORD
))
2843 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2844 return SOCKET_ERROR
;
2847 RtlCopyMemory(&Socket
->SharedData
->RecvTimeout
,
2854 /* These go directly to the helper dll */
2858 /* Obviously this is a hack */
2859 ERR("MSAFD: Set unknown optname %x\n", optname
);
2865 Errno
= Socket
->HelperData
->WSHSetSocketInformation(Socket
->HelperContext
,
2867 Socket
->TdiAddressHandle
,
2868 Socket
->TdiConnectionHandle
,
2873 if (lpErrno
) *lpErrno
= Errno
;
2874 return (Errno
== NO_ERROR
) ? NO_ERROR
: SOCKET_ERROR
;
2878 * FUNCTION: Initialize service provider for a client
2880 * wVersionRequested = Highest WinSock SPI version that the caller can use
2881 * lpWSPData = Address of WSPDATA structure to initialize
2882 * lpProtocolInfo = Pointer to structure that defines the desired protocol
2883 * UpcallTable = Pointer to upcall table of the WinSock DLL
2884 * lpProcTable = Address of procedure table to initialize
2886 * Status of operation
2890 WSPStartup(IN WORD wVersionRequested
,
2891 OUT LPWSPDATA lpWSPData
,
2892 IN LPWSAPROTOCOL_INFOW lpProtocolInfo
,
2893 IN WSPUPCALLTABLE UpcallTable
,
2894 OUT LPWSPPROC_TABLE lpProcTable
)
2899 if (((LOBYTE(wVersionRequested
) == 2) && (HIBYTE(wVersionRequested
) < 2)) ||
2900 (LOBYTE(wVersionRequested
) < 2))
2902 ERR("WSPStartup NOT SUPPORTED for version 0x%X\n", wVersionRequested
);
2903 return WSAVERNOTSUPPORTED
;
2907 /* FIXME: Enable all cases of WSPStartup status */
2908 Upcalls
= UpcallTable
;
2910 if (Status
== NO_ERROR
)
2912 lpProcTable
->lpWSPAccept
= WSPAccept
;
2913 lpProcTable
->lpWSPAddressToString
= WSPAddressToString
;
2914 lpProcTable
->lpWSPAsyncSelect
= WSPAsyncSelect
;
2915 lpProcTable
->lpWSPBind
= WSPBind
;
2916 lpProcTable
->lpWSPCancelBlockingCall
= WSPCancelBlockingCall
;
2917 lpProcTable
->lpWSPCleanup
= WSPCleanup
;
2918 lpProcTable
->lpWSPCloseSocket
= WSPCloseSocket
;
2919 lpProcTable
->lpWSPConnect
= WSPConnect
;
2920 lpProcTable
->lpWSPDuplicateSocket
= WSPDuplicateSocket
;
2921 lpProcTable
->lpWSPEnumNetworkEvents
= WSPEnumNetworkEvents
;
2922 lpProcTable
->lpWSPEventSelect
= WSPEventSelect
;
2923 lpProcTable
->lpWSPGetOverlappedResult
= WSPGetOverlappedResult
;
2924 lpProcTable
->lpWSPGetPeerName
= WSPGetPeerName
;
2925 lpProcTable
->lpWSPGetSockName
= WSPGetSockName
;
2926 lpProcTable
->lpWSPGetSockOpt
= WSPGetSockOpt
;
2927 lpProcTable
->lpWSPGetQOSByName
= WSPGetQOSByName
;
2928 lpProcTable
->lpWSPIoctl
= WSPIoctl
;
2929 lpProcTable
->lpWSPJoinLeaf
= WSPJoinLeaf
;
2930 lpProcTable
->lpWSPListen
= WSPListen
;
2931 lpProcTable
->lpWSPRecv
= WSPRecv
;
2932 lpProcTable
->lpWSPRecvDisconnect
= WSPRecvDisconnect
;
2933 lpProcTable
->lpWSPRecvFrom
= WSPRecvFrom
;
2934 lpProcTable
->lpWSPSelect
= WSPSelect
;
2935 lpProcTable
->lpWSPSend
= WSPSend
;
2936 lpProcTable
->lpWSPSendDisconnect
= WSPSendDisconnect
;
2937 lpProcTable
->lpWSPSendTo
= WSPSendTo
;
2938 lpProcTable
->lpWSPSetSockOpt
= WSPSetSockOpt
;
2939 lpProcTable
->lpWSPShutdown
= WSPShutdown
;
2940 lpProcTable
->lpWSPSocket
= WSPSocket
;
2941 lpProcTable
->lpWSPStringToAddress
= WSPStringToAddress
;
2942 lpWSPData
->wVersion
= MAKEWORD(2, 2);
2943 lpWSPData
->wHighVersion
= MAKEWORD(2, 2);
2944 /* Save CatalogEntryId for all upcalls */
2945 CatalogEntryId
= lpProtocolInfo
->dwCatalogEntryId
;
2948 TRACE("Status (%d).\n", Status
);
2955 WSPAddressToString(IN LPSOCKADDR lpsaAddress
,
2956 IN DWORD dwAddressLength
,
2957 IN LPWSAPROTOCOL_INFOW lpProtocolInfo
,
2958 OUT LPWSTR lpszAddressString
,
2959 IN OUT LPDWORD lpdwAddressStringLength
,
2963 WCHAR buffer
[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */
2966 if (!lpsaAddress
|| !lpszAddressString
|| !lpdwAddressStringLength
)
2968 if (lpErrno
) *lpErrno
= WSAEFAULT
;
2969 return SOCKET_ERROR
;
2972 switch (lpsaAddress
->sa_family
)
2975 if (dwAddressLength
< sizeof(SOCKADDR_IN
))
2977 if (lpErrno
) *lpErrno
= WSAEINVAL
;
2978 return SOCKET_ERROR
;
2982 (unsigned int)(ntohl(((SOCKADDR_IN
*)lpsaAddress
)->sin_addr
.s_addr
) >> 24 & 0xff),
2983 (unsigned int)(ntohl(((SOCKADDR_IN
*)lpsaAddress
)->sin_addr
.s_addr
) >> 16 & 0xff),
2984 (unsigned int)(ntohl(((SOCKADDR_IN
*)lpsaAddress
)->sin_addr
.s_addr
) >> 8 & 0xff),
2985 (unsigned int)(ntohl(((SOCKADDR_IN
*)lpsaAddress
)->sin_addr
.s_addr
) & 0xff),
2986 ntohs(((SOCKADDR_IN
*)lpsaAddress
)->sin_port
));
2988 p
= wcschr(buffer
, L
':');
2989 if (!((SOCKADDR_IN
*)lpsaAddress
)->sin_port
)
2995 if (lpErrno
) *lpErrno
= WSAEINVAL
;
2996 return SOCKET_ERROR
;
2999 size
= wcslen(buffer
) + 1;
3001 if (*lpdwAddressStringLength
< size
)
3003 *lpdwAddressStringLength
= size
;
3004 if (lpErrno
) *lpErrno
= WSAEFAULT
;
3005 return SOCKET_ERROR
;
3008 *lpdwAddressStringLength
= size
;
3009 wcscpy(lpszAddressString
, buffer
);
3015 WSPStringToAddress(IN LPWSTR AddressString
,
3016 IN INT AddressFamily
,
3017 IN LPWSAPROTOCOL_INFOW lpProtocolInfo
,
3018 OUT LPSOCKADDR lpAddress
,
3019 IN OUT LPINT lpAddressLength
,
3024 LONG inetaddr
= 0, ip_part
;
3026 SOCKADDR_IN
*sockaddr
;
3028 if (!lpAddressLength
|| !lpAddress
|| !AddressString
)
3030 if (lpErrno
) *lpErrno
= WSAEINVAL
;
3031 return SOCKET_ERROR
;
3034 sockaddr
= (SOCKADDR_IN
*)lpAddress
;
3036 /* Set right address family */
3037 if (lpProtocolInfo
!= NULL
)
3039 sockaddr
->sin_family
= lpProtocolInfo
->iAddressFamily
;
3043 sockaddr
->sin_family
= AddressFamily
;
3047 if (AddressFamily
== AF_INET
)
3049 if (*lpAddressLength
< (INT
)sizeof(SOCKADDR_IN
))
3051 if (lpErrno
) *lpErrno
= WSAEFAULT
;
3055 // translate ip string to ip
3058 bp
= &AddressString
;
3061 while (*bp
< &AddressString
[wcslen(AddressString
)])
3063 ip_part
= wcstol(*bp
, bp
, 10);
3064 /* ip part number should be in range 0-255 */
3065 if (ip_part
< 0 || ip_part
> 255)
3067 if (lpErrno
) *lpErrno
= WSAEINVAL
;
3068 return SOCKET_ERROR
;
3070 inetaddr
= (inetaddr
<< 8) + ip_part
;
3071 /* we end on string end or port separator */
3072 if ((*bp
)[0] == 0 || (*bp
)[0] == L
':')
3074 /* ip parts are dot separated. verify it */
3075 if ((*bp
)[0] != L
'.')
3077 if (lpErrno
) *lpErrno
= WSAEINVAL
;
3078 return SOCKET_ERROR
;
3080 /* count the dots */
3082 /* move over the dot to next ip part */
3086 /* check dots count */
3089 if (lpErrno
) *lpErrno
= WSAEINVAL
;
3090 return SOCKET_ERROR
;
3093 /* Get port number */
3094 if ((*bp
)[0] == L
':')
3096 /* move over the column to port part */
3098 /* next char should be numeric */
3099 if ((*bp
)[0] < L
'0' || (*bp
)[0] > L
'9')
3101 if (lpErrno
) *lpErrno
= WSAEINVAL
;
3102 return SOCKET_ERROR
;
3104 port
= wcstol(*bp
, bp
, 10);
3111 if (lpErrno
) *lpErrno
= NO_ERROR
;
3112 /* rest sockaddr.sin_addr.s_addr
3113 for we need to be sure it is zero when we come to while */
3114 *lpAddressLength
= sizeof(*sockaddr
);
3115 memset(lpAddress
, 0, sizeof(*sockaddr
));
3116 sockaddr
->sin_family
= AF_INET
;
3117 sockaddr
->sin_addr
.s_addr
= inetaddr
;
3118 sockaddr
->sin_port
= port
;
3122 if (lpErrno
&& !*lpErrno
)
3127 return SOCKET_ERROR
;
3131 * FUNCTION: Cleans up service provider for a client
3133 * lpErrno = Address of buffer for error information
3135 * 0 if successful, or SOCKET_ERROR if not
3139 WSPCleanup(OUT LPINT lpErrno
)
3142 TRACE("Leaving.\n");
3144 if (lpErrno
) *lpErrno
= NO_ERROR
;
3151 AfdInfoAPC(PVOID ApcContext
,
3152 PIO_STATUS_BLOCK IoStatusBlock
,
3155 PAFDAPCCONTEXT Context
= ApcContext
;
3157 Context
->lpCompletionRoutine(IoStatusBlock
->Status
, IoStatusBlock
->Information
, Context
->lpOverlapped
, 0);
3158 HeapFree(GlobalHeap
, 0, ApcContext
);
3162 GetSocketInformation(PSOCKET_INFORMATION Socket
,
3163 ULONG AfdInformationClass
,
3164 PBOOLEAN Boolean OPTIONAL
,
3165 PULONG Ulong OPTIONAL
,
3166 PLARGE_INTEGER LargeInteger OPTIONAL
,
3167 LPWSAOVERLAPPED Overlapped OPTIONAL
,
3168 LPWSAOVERLAPPED_COMPLETION_ROUTINE CompletionRoutine OPTIONAL
)
3170 PIO_STATUS_BLOCK IOSB
;
3171 IO_STATUS_BLOCK DummyIOSB
;
3174 PAFDAPCCONTEXT APCContext
;
3175 PIO_APC_ROUTINE APCFunction
;
3176 HANDLE Event
= NULL
;
3179 Status
= NtCreateEvent(&SockEvent
,
3185 if( !NT_SUCCESS(Status
) )
3186 return SOCKET_ERROR
;
3188 /* Set Info Class */
3189 InfoData
.InformationClass
= AfdInformationClass
;
3191 /* Verify if we should use APC */
3192 if (Overlapped
== NULL
)
3194 /* Not using Overlapped structure, so use normal blocking on event */
3202 /* Overlapped request for non overlapped opened socket */
3203 if ((Socket
->SharedData
->CreateFlags
& SO_SYNCHRONOUS_NONALERT
) != 0)
3205 TRACE("Opened without flag WSA_FLAG_OVERLAPPED. Do nothing.\n");
3208 if (CompletionRoutine
== NULL
)
3210 /* Using Overlapped Structure, but no Completition Routine, so no need for APC */
3211 APCContext
= (PAFDAPCCONTEXT
)Overlapped
;
3213 Event
= Overlapped
->hEvent
;
3217 /* Using Overlapped Structure and a Completition Routine, so use an APC */
3218 APCFunction
= &AfdInfoAPC
; // should be a private io completition function inside us
3219 APCContext
= HeapAlloc(GlobalHeap
, 0, sizeof(AFDAPCCONTEXT
));
3222 ERR("Not enough memory for APC Context\n");
3225 APCContext
->lpCompletionRoutine
= CompletionRoutine
;
3226 APCContext
->lpOverlapped
= Overlapped
;
3227 APCContext
->lpSocket
= Socket
;
3230 IOSB
= (PIO_STATUS_BLOCK
)&Overlapped
->Internal
;
3233 IOSB
->Status
= STATUS_PENDING
;
3236 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
3247 /* Wait for return */
3248 if (Status
== STATUS_PENDING
&& Overlapped
== NULL
)
3250 WaitForSingleObject(SockEvent
, INFINITE
);
3251 Status
= IOSB
->Status
;
3254 TRACE("Status %x Information %d\n", Status
, IOSB
->Information
);
3256 if (Status
== STATUS_PENDING
)
3258 TRACE("Leaving (Pending)\n");
3259 return WSA_IO_PENDING
;
3262 if (Status
!= STATUS_SUCCESS
)
3263 return SOCKET_ERROR
;
3265 /* Return Information */
3268 *Ulong
= InfoData
.Information
.Ulong
;
3270 if (LargeInteger
!= NULL
)
3272 *LargeInteger
= InfoData
.Information
.LargeInteger
;
3274 if (Boolean
!= NULL
)
3276 *Boolean
= InfoData
.Information
.Boolean
;
3279 NtClose( SockEvent
);
3287 SetSocketInformation(PSOCKET_INFORMATION Socket
,
3288 ULONG AfdInformationClass
,
3289 PBOOLEAN Boolean OPTIONAL
,
3290 PULONG Ulong OPTIONAL
,
3291 PLARGE_INTEGER LargeInteger OPTIONAL
,
3292 LPWSAOVERLAPPED Overlapped OPTIONAL
,
3293 LPWSAOVERLAPPED_COMPLETION_ROUTINE CompletionRoutine OPTIONAL
)
3295 PIO_STATUS_BLOCK IOSB
;
3296 IO_STATUS_BLOCK DummyIOSB
;
3299 PAFDAPCCONTEXT APCContext
;
3300 PIO_APC_ROUTINE APCFunction
;
3301 HANDLE Event
= NULL
;
3304 Status
= NtCreateEvent(&SockEvent
,
3310 if( !NT_SUCCESS(Status
) )
3311 return SOCKET_ERROR
;
3313 /* Set Info Class */
3314 InfoData
.InformationClass
= AfdInformationClass
;
3316 /* Set Information */
3319 InfoData
.Information
.Ulong
= *Ulong
;
3321 if (LargeInteger
!= NULL
)
3323 InfoData
.Information
.LargeInteger
= *LargeInteger
;
3325 if (Boolean
!= NULL
)
3327 InfoData
.Information
.Boolean
= *Boolean
;
3330 /* Verify if we should use APC */
3331 if (Overlapped
== NULL
)
3333 /* Not using Overlapped structure, so use normal blocking on event */
3341 /* Overlapped request for non overlapped opened socket */
3342 if ((Socket
->SharedData
->CreateFlags
& SO_SYNCHRONOUS_NONALERT
) != 0)
3344 TRACE("Opened without flag WSA_FLAG_OVERLAPPED. Do nothing.\n");
3347 if (CompletionRoutine
== NULL
)
3349 /* Using Overlapped Structure, but no Completition Routine, so no need for APC */
3350 APCContext
= (PAFDAPCCONTEXT
)Overlapped
;
3352 Event
= Overlapped
->hEvent
;
3356 /* Using Overlapped Structure and a Completition Routine, so use an APC */
3357 APCFunction
= &AfdInfoAPC
; // should be a private io completition function inside us
3358 APCContext
= HeapAlloc(GlobalHeap
, 0, sizeof(AFDAPCCONTEXT
));
3361 ERR("Not enough memory for APC Context\n");
3364 APCContext
->lpCompletionRoutine
= CompletionRoutine
;
3365 APCContext
->lpOverlapped
= Overlapped
;
3366 APCContext
->lpSocket
= Socket
;
3369 IOSB
= (PIO_STATUS_BLOCK
)&Overlapped
->Internal
;
3372 IOSB
->Status
= STATUS_PENDING
;
3375 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
3386 /* Wait for return */
3387 if (Status
== STATUS_PENDING
&& Overlapped
== NULL
)
3389 WaitForSingleObject(SockEvent
, INFINITE
);
3390 Status
= IOSB
->Status
;
3393 NtClose( SockEvent
);
3395 TRACE("Status %x Information %d\n", Status
, IOSB
->Information
);
3397 if (Status
== STATUS_PENDING
)
3399 TRACE("Leaving (Pending)\n");
3400 return WSA_IO_PENDING
;
3403 return Status
== STATUS_SUCCESS
? NO_ERROR
: SOCKET_ERROR
;
3408 GetSocketStructure(SOCKET Handle
)
3410 PSOCKET_INFORMATION CurrentSocket
;
3412 EnterCriticalSection(&SocketListLock
);
3414 CurrentSocket
= SocketListHead
;
3415 while (CurrentSocket
)
3417 if (CurrentSocket
->Handle
== Handle
)
3419 LeaveCriticalSection(&SocketListLock
);
3420 return CurrentSocket
;
3423 CurrentSocket
= CurrentSocket
->NextSocket
;
3426 LeaveCriticalSection(&SocketListLock
);
3431 int CreateContext(PSOCKET_INFORMATION Socket
)
3433 IO_STATUS_BLOCK IOSB
;
3434 SOCKET_CONTEXT ContextData
;
3438 Status
= NtCreateEvent(&SockEvent
,
3444 if( !NT_SUCCESS(Status
) )
3445 return SOCKET_ERROR
;
3447 /* Create Context */
3448 ContextData
.SharedData
= *Socket
->SharedData
;
3449 ContextData
.SizeOfHelperData
= 0;
3450 RtlCopyMemory (&ContextData
.LocalAddress
,
3451 Socket
->LocalAddress
,
3452 Socket
->SharedData
->SizeOfLocalAddress
);
3453 RtlCopyMemory (&ContextData
.RemoteAddress
,
3454 Socket
->RemoteAddress
,
3455 Socket
->SharedData
->SizeOfRemoteAddress
);
3458 Status
= NtDeviceIoControlFile((HANDLE
)Socket
->Handle
,
3463 IOCTL_AFD_SET_CONTEXT
,
3465 sizeof(ContextData
),
3469 /* Wait for Completion */
3470 if (Status
== STATUS_PENDING
)
3472 WaitForSingleObject(SockEvent
, INFINITE
);
3473 Status
= IOSB
.Status
;
3476 NtClose( SockEvent
);
3478 return Status
== STATUS_SUCCESS
? NO_ERROR
: SOCKET_ERROR
;
3481 BOOLEAN
SockCreateOrReferenceAsyncThread(VOID
)
3483 HANDLE hAsyncThread
;
3484 DWORD AsyncThreadId
;
3486 OBJECT_HANDLE_ATTRIBUTE_INFORMATION HandleFlags
;
3489 /* Check if the Thread Already Exists */
3490 if (SockAsyncThreadRefCount
)
3492 ASSERT(SockAsyncCompletionPort
);
3496 /* Create the Completion Port */
3497 if (!SockAsyncCompletionPort
)
3499 Status
= NtCreateIoCompletion(&SockAsyncCompletionPort
,
3500 IO_COMPLETION_ALL_ACCESS
,
3502 2); // Allow 2 threads only
3503 if (!NT_SUCCESS(Status
))
3505 ERR("Failed to create completion port: 0x%08x\n", Status
);
3508 /* Protect Handle */
3509 HandleFlags
.ProtectFromClose
= TRUE
;
3510 HandleFlags
.Inherit
= FALSE
;
3511 Status
= NtSetInformationObject(SockAsyncCompletionPort
,
3512 ObjectHandleFlagInformation
,
3514 sizeof(HandleFlags
));
3517 /* Create the Async Event */
3518 Status
= NtCreateEvent(&AsyncEvent
,
3524 /* Create the Async Thread */
3525 hAsyncThread
= CreateThread(NULL
,
3532 /* Close the Handle */
3533 NtClose(hAsyncThread
);
3535 /* Increase the Reference Count */
3536 SockAsyncThreadRefCount
++;
3542 SockAsyncThread(PVOID ThreadParam
)
3545 PASYNC_COMPLETION_ROUTINE AsyncCompletionRoutine
;
3546 IO_STATUS_BLOCK IOSB
;
3549 /* Make the Thread Higher Priority */
3550 SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_ABOVE_NORMAL
);
3552 /* Do a KQUEUE/WorkItem Style Loop, thanks to IoCompletion Ports */
3555 Status
= NtRemoveIoCompletion (SockAsyncCompletionPort
,
3556 (PVOID
*)&AsyncCompletionRoutine
,
3560 /* Call the Async Function */
3561 if (NT_SUCCESS(Status
))
3563 (*AsyncCompletionRoutine
)(AsyncContext
, &IOSB
);
3567 /* It Failed, sleep for a second */
3570 } while ((Status
!= STATUS_TIMEOUT
));
3572 /* The Thread has Ended */
3576 BOOLEAN
SockGetAsyncSelectHelperAfdHandle(VOID
)
3578 UNICODE_STRING AfdHelper
;
3579 OBJECT_ATTRIBUTES ObjectAttributes
;
3580 IO_STATUS_BLOCK IoSb
;
3581 FILE_COMPLETION_INFORMATION CompletionInfo
;
3582 OBJECT_HANDLE_ATTRIBUTE_INFORMATION HandleFlags
;
3584 /* First, make sure we're not already initialized */
3585 if (SockAsyncHelperAfdHandle
)
3590 /* Set up Handle Name and Object */
3591 RtlInitUnicodeString(&AfdHelper
, L
"\\Device\\Afd\\AsyncSelectHlp" );
3592 InitializeObjectAttributes(&ObjectAttributes
,
3594 OBJ_INHERIT
| OBJ_CASE_INSENSITIVE
,
3598 /* Open the Handle to AFD */
3599 NtCreateFile(&SockAsyncHelperAfdHandle
,
3600 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
,
3605 FILE_SHARE_READ
| FILE_SHARE_WRITE
,
3612 * Now Set up the Completion Port Information
3613 * This means that whenever a Poll is finished, the routine will be executed
3615 CompletionInfo
.Port
= SockAsyncCompletionPort
;
3616 CompletionInfo
.Key
= SockAsyncSelectCompletionRoutine
;
3617 NtSetInformationFile(SockAsyncHelperAfdHandle
,
3620 sizeof(CompletionInfo
),
3621 FileCompletionInformation
);
3624 /* Protect the Handle */
3625 HandleFlags
.ProtectFromClose
= TRUE
;
3626 HandleFlags
.Inherit
= FALSE
;
3627 NtSetInformationObject(SockAsyncCompletionPort
,
3628 ObjectHandleFlagInformation
,
3630 sizeof(HandleFlags
));
3633 /* Set this variable to true so that Send/Recv/Accept will know wether to renable disabled events */
3634 SockAsyncSelectCalled
= TRUE
;
3638 VOID
SockAsyncSelectCompletionRoutine(PVOID Context
, PIO_STATUS_BLOCK IoStatusBlock
)
3641 PASYNC_DATA AsyncData
= Context
;
3642 PSOCKET_INFORMATION Socket
;
3645 /* Get the Socket */
3646 Socket
= AsyncData
->ParentSocket
;
3648 /* Check if the Sequence Number Changed behind our back */
3649 if (AsyncData
->SequenceNumber
!= Socket
->SharedData
->SequenceNumber
)
3654 /* Check we were manually called b/c of a failure */
3655 if (!NT_SUCCESS(IoStatusBlock
->Status
))
3657 /* FIXME: Perform Upcall */
3661 for (x
= 1; x
; x
<<=1)
3663 switch (AsyncData
->AsyncSelectInfo
.Handles
[0].Events
& x
)
3665 case AFD_EVENT_RECEIVE
:
3666 if (0 != (Socket
->SharedData
->AsyncEvents
& FD_READ
) &&
3667 0 == (Socket
->SharedData
->AsyncDisabledEvents
& FD_READ
))
3669 /* Make the Notification */
3670 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
->hWnd
,
3671 Socket
->SharedData
->wMsg
,
3673 WSAMAKESELECTREPLY(FD_READ
, 0));
3674 /* Disable this event until the next read(); */
3675 Socket
->SharedData
->AsyncDisabledEvents
|= FD_READ
;
3679 case AFD_EVENT_OOB_RECEIVE
:
3680 if (0 != (Socket
->SharedData
->AsyncEvents
& FD_OOB
) &&
3681 0 == (Socket
->SharedData
->AsyncDisabledEvents
& FD_OOB
))
3683 /* Make the Notification */
3684 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
->hWnd
,
3685 Socket
->SharedData
->wMsg
,
3687 WSAMAKESELECTREPLY(FD_OOB
, 0));
3688 /* Disable this event until the next read(); */
3689 Socket
->SharedData
->AsyncDisabledEvents
|= FD_OOB
;
3693 case AFD_EVENT_SEND
:
3694 if (0 != (Socket
->SharedData
->AsyncEvents
& FD_WRITE
) &&
3695 0 == (Socket
->SharedData
->AsyncDisabledEvents
& FD_WRITE
))
3697 /* Make the Notification */
3698 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
->hWnd
,
3699 Socket
->SharedData
->wMsg
,
3701 WSAMAKESELECTREPLY(FD_WRITE
, 0));
3702 /* Disable this event until the next write(); */
3703 Socket
->SharedData
->AsyncDisabledEvents
|= FD_WRITE
;
3707 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
3708 case AFD_EVENT_CONNECT
:
3709 case AFD_EVENT_CONNECT_FAIL
:
3710 if (0 != (Socket
->SharedData
->AsyncEvents
& FD_CONNECT
) &&
3711 0 == (Socket
->SharedData
->AsyncDisabledEvents
& FD_CONNECT
))
3713 /* Make the Notification */
3714 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
->hWnd
,
3715 Socket
->SharedData
->wMsg
,
3717 WSAMAKESELECTREPLY(FD_CONNECT
, 0));
3718 /* Disable this event forever; */
3719 Socket
->SharedData
->AsyncDisabledEvents
|= FD_CONNECT
;
3723 case AFD_EVENT_ACCEPT
:
3724 if (0 != (Socket
->SharedData
->AsyncEvents
& FD_ACCEPT
) &&
3725 0 == (Socket
->SharedData
->AsyncDisabledEvents
& FD_ACCEPT
))
3727 /* Make the Notification */
3728 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
->hWnd
,
3729 Socket
->SharedData
->wMsg
,
3731 WSAMAKESELECTREPLY(FD_ACCEPT
, 0));
3732 /* Disable this event until the next accept(); */
3733 Socket
->SharedData
->AsyncDisabledEvents
|= FD_ACCEPT
;
3737 case AFD_EVENT_DISCONNECT
:
3738 case AFD_EVENT_ABORT
:
3739 case AFD_EVENT_CLOSE
:
3740 if (0 != (Socket
->SharedData
->AsyncEvents
& FD_CLOSE
) &&
3741 0 == (Socket
->SharedData
->AsyncDisabledEvents
& FD_CLOSE
))
3743 /* Make the Notification */
3744 (Upcalls
.lpWPUPostMessage
)(Socket
->SharedData
->hWnd
,
3745 Socket
->SharedData
->wMsg
,
3747 WSAMAKESELECTREPLY(FD_CLOSE
, 0));
3748 /* Disable this event forever; */
3749 Socket
->SharedData
->AsyncDisabledEvents
|= FD_CLOSE
;
3752 /* FIXME: Support QOS */
3756 /* Check if there are any events left for us to check */
3757 if ((Socket
->SharedData
->AsyncEvents
& (~Socket
->SharedData
->AsyncDisabledEvents
)) == 0 )
3763 SockProcessAsyncSelect(Socket
, AsyncData
);
3767 VOID
SockProcessAsyncSelect(PSOCKET_INFORMATION Socket
, PASYNC_DATA AsyncData
)
3770 ULONG lNetworkEvents
;
3773 /* Set up the Async Data Event Info */
3774 AsyncData
->AsyncSelectInfo
.Timeout
.HighPart
= 0x7FFFFFFF;
3775 AsyncData
->AsyncSelectInfo
.Timeout
.LowPart
= 0xFFFFFFFF;
3776 AsyncData
->AsyncSelectInfo
.HandleCount
= 1;
3777 AsyncData
->AsyncSelectInfo
.Exclusive
= TRUE
;
3778 AsyncData
->AsyncSelectInfo
.Handles
[0].Handle
= Socket
->Handle
;
3779 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
= 0;
3781 /* Remove unwanted events */
3782 lNetworkEvents
= Socket
->SharedData
->AsyncEvents
& (~Socket
->SharedData
->AsyncDisabledEvents
);
3784 /* Set Events to wait for */
3785 if (lNetworkEvents
& FD_READ
)
3787 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_RECEIVE
;
3790 if (lNetworkEvents
& FD_WRITE
)
3792 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_SEND
;
3795 if (lNetworkEvents
& FD_OOB
)
3797 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_OOB_RECEIVE
;
3800 if (lNetworkEvents
& FD_ACCEPT
)
3802 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_ACCEPT
;
3805 /* FIXME: THIS IS NOT RIGHT!!! HACK HACK HACK! */
3806 if (lNetworkEvents
& FD_CONNECT
)
3808 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_CONNECT
| AFD_EVENT_CONNECT_FAIL
;
3811 if (lNetworkEvents
& FD_CLOSE
)
3813 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_DISCONNECT
| AFD_EVENT_ABORT
| AFD_EVENT_CLOSE
;
3816 if (lNetworkEvents
& FD_QOS
)
3818 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_QOS
;
3821 if (lNetworkEvents
& FD_GROUP_QOS
)
3823 AsyncData
->AsyncSelectInfo
.Handles
[0].Events
|= AFD_EVENT_GROUP_QOS
;
3827 Status
= NtDeviceIoControlFile (SockAsyncHelperAfdHandle
,
3831 &AsyncData
->IoStatusBlock
,
3833 &AsyncData
->AsyncSelectInfo
,
3834 sizeof(AsyncData
->AsyncSelectInfo
),
3835 &AsyncData
->AsyncSelectInfo
,
3836 sizeof(AsyncData
->AsyncSelectInfo
));
3838 /* I/O Manager Won't call the completion routine, let's do it manually */
3839 if (NT_SUCCESS(Status
))
3845 AsyncData
->IoStatusBlock
.Status
= Status
;
3846 SockAsyncSelectCompletionRoutine(AsyncData
, &AsyncData
->IoStatusBlock
);
3850 VOID
SockProcessQueuedAsyncSelect(PVOID Context
, PIO_STATUS_BLOCK IoStatusBlock
)
3852 PASYNC_DATA AsyncData
= Context
;
3853 BOOL FreeContext
= TRUE
;
3854 PSOCKET_INFORMATION Socket
;
3856 /* Get the Socket */
3857 Socket
= AsyncData
->ParentSocket
;
3859 /* If someone closed it, stop the function */
3860 if (Socket
->SharedData
->State
!= SocketClosed
)
3862 /* Check if the Sequence Number changed by now, in which case quit */
3863 if (AsyncData
->SequenceNumber
== Socket
->SharedData
->SequenceNumber
)
3865 /* Do the actual select, if needed */
3866 if ((Socket
->SharedData
->AsyncEvents
& (~Socket
->SharedData
->AsyncDisabledEvents
)))
3868 SockProcessAsyncSelect(Socket
, AsyncData
);
3869 FreeContext
= FALSE
;
3874 /* Free the Context */
3877 HeapFree(GetProcessHeap(), 0, AsyncData
);
3884 SockReenableAsyncSelectEvent (IN PSOCKET_INFORMATION Socket
,
3887 PASYNC_DATA AsyncData
;
3889 /* Make sure the event is actually disabled */
3890 if (!(Socket
->SharedData
->AsyncDisabledEvents
& Event
))
3896 Socket
->SharedData
->AsyncDisabledEvents
&= ~Event
;
3898 /* Return if no more events are being polled */
3899 if ((Socket
->SharedData
->AsyncEvents
& (~Socket
->SharedData
->AsyncDisabledEvents
)) == 0 )
3904 /* Wait on new events */
3905 AsyncData
= HeapAlloc(GetProcessHeap(), 0, sizeof(ASYNC_DATA
));
3906 if (!AsyncData
) return;
3908 /* Create the Asynch Thread if Needed */
3909 SockCreateOrReferenceAsyncThread();
3911 /* Increase the sequence number to stop anything else */
3912 Socket
->SharedData
->SequenceNumber
++;
3914 /* Set up the Async Data */
3915 AsyncData
->ParentSocket
= Socket
;
3916 AsyncData
->SequenceNumber
= Socket
->SharedData
->SequenceNumber
;
3918 /* Begin Async Select by using I/O Completion */
3919 NtSetIoCompletion(SockAsyncCompletionPort
,
3920 (PVOID
)&SockProcessQueuedAsyncSelect
,
3931 DllMain(HANDLE hInstDll
,
3938 case DLL_PROCESS_ATTACH
:
3940 TRACE("Loading MSAFD.DLL \n");
3942 /* Don't need thread attach notifications
3943 so disable them to improve performance */
3944 DisableThreadLibraryCalls(hInstDll
);
3946 /* List of DLL Helpers */
3947 InitializeListHead(&SockHelpersListHead
);
3949 /* Heap to use when allocating */
3950 GlobalHeap
= GetProcessHeap();
3952 /* Initialize the lock that protects our socket list */
3953 InitializeCriticalSection(&SocketListLock
);
3955 TRACE("MSAFD.DLL has been loaded\n");
3959 case DLL_THREAD_ATTACH
:
3962 case DLL_THREAD_DETACH
:
3965 case DLL_PROCESS_DETACH
:
3967 /* Delete the socket list lock */
3968 DeleteCriticalSection(&SocketListLock
);
3973 TRACE("DllMain of msafd.dll (leaving)\n");