Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-02-24 09:25:10 +00:00
Code Issues Projects Releases Packages Wiki Activity

[MSAFD] Add more checks for lpErrno. Rearrange start of some functions to avoid creation and disposal of event of socket handle or parameter(s) is wrong. Other small changes

Browse source 
CORE-12104

svn path=/trunk/; revision=73096
This commit is contained in:
Peter Hater 2016-11-02 15:36:26 +00:00
parent 494db7a55d
commit 0c396fc83e
2 changed files with 213 additions and 207 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

403
reactos/dll/win32/msafd/misc/dllmain.c
View file

@ -202,6 +202,9 @@ WSPSocket(int AddressFamily,
Socket->SharedData->ProviderFlags = lpProtocolInfo->dwProviderFlags; Socket->SharedData->ProviderFlags = lpProtocolInfo->dwProviderFlags;
Socket->SharedData->UseSAN = FALSE; Socket->SharedData->UseSAN = FALSE;
Socket->SharedData->NonBlocking = FALSE; /* Sockets start blocking */ Socket->SharedData->NonBlocking = FALSE; /* Sockets start blocking */
Socket->SharedData->RecvTimeout = INFINITE;
Socket->SharedData->SendTimeout = INFINITE;
Socket->SharedData->OobInline = FALSE;
/* Ask alex about this */ /* Ask alex about this */
if( Socket->SharedData->SocketType == SOCK_DGRAM || if( Socket->SharedData->SocketType == SOCK_DGRAM ||
@ -210,7 +213,7 @@ WSPSocket(int AddressFamily,
TRACE("Connectionless socket\n"); TRACE("Connectionless socket\n");
Socket->SharedData->ServiceFlags1 |= XP1_CONNECTIONLESS; Socket->SharedData->ServiceFlags1 |= XP1_CONNECTIONLESS;
} }
Socket->Handle = -1; Socket->Handle = INVALID_SOCKET;
} }
Socket->HelperContext = HelperDLLContext; Socket->HelperContext = HelperDLLContext;
@ -579,6 +582,9 @@ TranslateNtStatusError(NTSTATUS Status)
case STATUS_ACCESS_DENIED: case STATUS_ACCESS_DENIED:
return WSAEACCES; return WSAEACCES;
case STATUS_NOT_IMPLEMENTED:
return WSAEOPNOTSUPP;
default: default:
ERR("MSAFD: Unhandled NTSTATUS value: 0x%x\n", Status); ERR("MSAFD: Unhandled NTSTATUS value: 0x%x\n", Status);
return WSAENETDOWN; return WSAENETDOWN;
@ -607,6 +613,14 @@ WSPCloseSocket(IN SOCKET Handle,
LONG LingerWait = -1; LONG LingerWait = -1;
DWORD References; DWORD References;
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
/* Create the Wait Event */ /* Create the Wait Event */
Status = NtCreateEvent(&SockEvent, Status = NtCreateEvent(&SockEvent,
EVENT_ALL_ACCESS, EVENT_ALL_ACCESS,
@ -619,14 +633,6 @@ WSPCloseSocket(IN SOCKET Handle,
ERR("NtCreateEvent failed: 0x%08x", Status); ERR("NtCreateEvent failed: 0x%08x", Status);
return SOCKET_ERROR; return SOCKET_ERROR;
} }
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
NtClose(SockEvent);
if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
if (Socket->HelperEvents & WSH_NOTIFY_CLOSE) if (Socket->HelperEvents & WSH_NOTIFY_CLOSE)
{ {
@ -829,11 +835,12 @@ WSPBind(SOCKET Handle,
SOCKADDR_INFO SocketInfo; SOCKADDR_INFO SocketInfo;
HANDLE SockEvent; HANDLE SockEvent;
/* See below */ /* Get the Socket Structure associate to this Socket*/
BindData = HeapAlloc(GlobalHeap, 0, 0xA + SocketAddressLength); Socket = GetSocketStructure(Handle);
if (!BindData) if (!Socket)
{ {
return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES, lpErrno, 0, NULL); if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
} }
Status = NtCreateEvent(&SockEvent, Status = NtCreateEvent(&SockEvent,
@ -844,17 +851,14 @@ WSPBind(SOCKET Handle,
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
HeapFree(GlobalHeap, 0, BindData);
return SOCKET_ERROR; return SOCKET_ERROR;
} }
/* Get the Socket Structure associate to this Socket*/ /* See below */
Socket = GetSocketStructure(Handle); BindData = HeapAlloc(GlobalHeap, 0, 0xA + SocketAddressLength);
if (!Socket) if (!BindData)
{ {
HeapFree(GlobalHeap, 0, BindData); return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES, lpErrno, 0, NULL);
*lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
} }
/* Set up Address in TDI Format */ /* Set up Address in TDI Format */
@ -951,12 +955,12 @@ WSPListen(SOCKET Handle,
Socket = GetSocketStructure(Handle); Socket = GetSocketStructure(Handle);
if (!Socket) if (!Socket)
{ {
*lpErrno = WSAENOTSOCK; if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
if (Socket->SharedData->Listening) if (Socket->SharedData->Listening)
return 0; return NO_ERROR;
Status = NtCreateEvent(&SockEvent, Status = NtCreateEvent(&SockEvent,
EVENT_ALL_ACCESS, EVENT_ALL_ACCESS,
@ -965,7 +969,7 @@ WSPListen(SOCKET Handle,
FALSE); FALSE);
if( !NT_SUCCESS(Status) ) if( !NT_SUCCESS(Status) )
return -1; return SOCKET_ERROR;
/* Set Up Listen Structure */ /* Set Up Listen Structure */
ListenData.UseSAN = FALSE; ListenData.UseSAN = FALSE;
@ -1340,6 +1344,14 @@ WSPAccept(SOCKET Handle,
UCHAR ReceiveBuffer[0x1A]; UCHAR ReceiveBuffer[0x1A];
HANDLE SockEvent; HANDLE SockEvent;
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
Status = NtCreateEvent(&SockEvent, Status = NtCreateEvent(&SockEvent,
EVENT_ALL_ACCESS, EVENT_ALL_ACCESS,
NULL, NULL,
@ -1348,22 +1360,12 @@ WSPAccept(SOCKET Handle,
if( !NT_SUCCESS(Status) ) if( !NT_SUCCESS(Status) )
{ {
MsafdReturnWithErrno( Status, lpErrno, 0, NULL ); return SOCKET_ERROR;
return INVALID_SOCKET;
} }
/* Dynamic Structure...ugh */ /* Dynamic Structure...ugh */
ListenReceiveData = (PAFD_RECEIVED_ACCEPT_DATA)ReceiveBuffer; ListenReceiveData = (PAFD_RECEIVED_ACCEPT_DATA)ReceiveBuffer;
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
NtClose(SockEvent);
*lpErrno = WSAENOTSOCK;
return INVALID_SOCKET;
}
/* If this is non-blocking, make sure there's something for us to accept */ /* If this is non-blocking, make sure there's something for us to accept */
FD_ZERO(&ReadSet); FD_ZERO(&ReadSet);
FD_SET(Socket->Handle, &ReadSet); FD_SET(Socket->Handle, &ReadSet);
@ -1373,14 +1375,14 @@ WSPAccept(SOCKET Handle,
if (WSPSelect(0, &ReadSet, NULL, NULL, &Timeout, lpErrno) == SOCKET_ERROR) if (WSPSelect(0, &ReadSet, NULL, NULL, &Timeout, lpErrno) == SOCKET_ERROR)
{ {
NtClose(SockEvent); NtClose(SockEvent);
return INVALID_SOCKET; return SOCKET_ERROR;
} }
if (ReadSet.fd_array[0] != Socket->Handle) if (ReadSet.fd_array[0] != Socket->Handle)
{ {
NtClose(SockEvent); NtClose(SockEvent);
*lpErrno = WSAEWOULDBLOCK; if (lpErrno) *lpErrno = WSAEWOULDBLOCK;
return INVALID_SOCKET; return SOCKET_ERROR;
} }
/* Send IOCTL */ /* Send IOCTL */
@ -1405,8 +1407,7 @@ WSPAccept(SOCKET Handle,
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
NtClose( SockEvent ); NtClose( SockEvent );
MsafdReturnWithErrno( Status, lpErrno, 0, NULL ); return MsafdReturnWithErrno( Status, lpErrno, 0, NULL );
return INVALID_SOCKET;
} }
if (lpfnCondition != NULL) if (lpfnCondition != NULL)
@ -1439,8 +1440,7 @@ WSPAccept(SOCKET Handle,
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
NtClose( SockEvent ); NtClose( SockEvent );
MsafdReturnWithErrno( Status, lpErrno, 0, NULL ); return MsafdReturnWithErrno( Status, lpErrno, 0, NULL );
return INVALID_SOCKET;
} }
/* How much data to allocate */ /* How much data to allocate */
@ -1452,8 +1452,7 @@ WSPAccept(SOCKET Handle,
PendingData = HeapAlloc(GlobalHeap, 0, PendingDataLength); PendingData = HeapAlloc(GlobalHeap, 0, PendingDataLength);
if (!PendingData) if (!PendingData)
{ {
MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES, lpErrno, 0, NULL ); return MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES, lpErrno, 0, NULL );
return INVALID_SOCKET;
} }
/* We want the data now */ /* We want the data now */
@ -1481,8 +1480,7 @@ WSPAccept(SOCKET Handle,
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
NtClose( SockEvent ); NtClose( SockEvent );
MsafdReturnWithErrno( Status, lpErrno, 0, NULL ); return MsafdReturnWithErrno( Status, lpErrno, 0, NULL );
return INVALID_SOCKET;
} }
} }
} }
@ -1499,8 +1497,7 @@ WSPAccept(SOCKET Handle,
RemoteAddress = HeapAlloc(GlobalHeap, 0, sizeof(*RemoteAddress)); RemoteAddress = HeapAlloc(GlobalHeap, 0, sizeof(*RemoteAddress));
if (!RemoteAddress) if (!RemoteAddress)
{ {
MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES, lpErrno, 0, NULL); return MsafdReturnWithErrno(STATUS_INSUFFICIENT_RESOURCES, lpErrno, 0, NULL);
return INVALID_SOCKET;
} }
/* Set up Address in SOCKADDR Format */ /* Set up Address in SOCKADDR Format */
@ -1522,8 +1519,7 @@ WSPAccept(SOCKET Handle,
/* Allocate Buffer for Callee Data */ /* Allocate Buffer for Callee Data */
CalleeDataBuffer = HeapAlloc(GlobalHeap, 0, 4096); CalleeDataBuffer = HeapAlloc(GlobalHeap, 0, 4096);
if (!CalleeDataBuffer) { if (!CalleeDataBuffer) {
MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES, lpErrno, 0, NULL ); return MsafdReturnWithErrno( STATUS_INSUFFICIENT_RESOURCES, lpErrno, 0, NULL );
return INVALID_SOCKET;
} }
CalleeData.buf = CalleeDataBuffer; CalleeData.buf = CalleeDataBuffer;
CalleeData.len = 4096; CalleeData.len = 4096;
@ -1590,19 +1586,18 @@ WSPAccept(SOCKET Handle,
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
MsafdReturnWithErrno( Status, lpErrno, 0, NULL ); return MsafdReturnWithErrno( Status, lpErrno, 0, NULL );
return INVALID_SOCKET;
} }
if (CallBack == CF_REJECT ) if (CallBack == CF_REJECT )
{ {
*lpErrno = WSAECONNREFUSED; if (lpErrno) *lpErrno = WSAECONNREFUSED;
return INVALID_SOCKET; return SOCKET_ERROR;
} }
else else
{ {
*lpErrno = WSAECONNREFUSED; if (lpErrno) *lpErrno = WSAECONNREFUSED;
return INVALID_SOCKET; return SOCKET_ERROR;
} }
} }
} }
@ -1616,7 +1611,7 @@ WSPAccept(SOCKET Handle,
Socket->SharedData->CreateFlags, Socket->SharedData->CreateFlags,
lpErrno); lpErrno);
if (AcceptSocket == INVALID_SOCKET) if (AcceptSocket == INVALID_SOCKET)
return INVALID_SOCKET; return SOCKET_ERROR;
/* Set up the Accept Structure */ /* Set up the Accept Structure */
AcceptData.ListenHandle = (HANDLE)AcceptSocket; AcceptData.ListenHandle = (HANDLE)AcceptSocket;
@ -1645,8 +1640,7 @@ WSPAccept(SOCKET Handle,
{ {
NtClose(SockEvent); NtClose(SockEvent);
WSPCloseSocket( AcceptSocket, lpErrno ); WSPCloseSocket( AcceptSocket, lpErrno );
MsafdReturnWithErrno( Status, lpErrno, 0, NULL ); return MsafdReturnWithErrno( Status, lpErrno, 0, NULL );
return INVALID_SOCKET;
} }
AcceptSocketInfo = GetSocketStructure(AcceptSocket); AcceptSocketInfo = GetSocketStructure(AcceptSocket);
@ -1654,8 +1648,7 @@ WSPAccept(SOCKET Handle,
{ {
NtClose(SockEvent); NtClose(SockEvent);
WSPCloseSocket( AcceptSocket, lpErrno ); WSPCloseSocket( AcceptSocket, lpErrno );
MsafdReturnWithErrno( STATUS_PROTOCOL_NOT_SUPPORTED, lpErrno, 0, NULL ); return MsafdReturnWithErrno( STATUS_PROTOCOL_NOT_SUPPORTED, lpErrno, 0, NULL );
return INVALID_SOCKET;
} }
AcceptSocketInfo->SharedData->State = SocketConnected; AcceptSocketInfo->SharedData->State = SocketConnected;
@ -1688,11 +1681,11 @@ WSPAccept(SOCKET Handle,
if (Status) if (Status)
{ {
if (lpErrno) *lpErrno = Status; if (lpErrno) *lpErrno = Status;
return INVALID_SOCKET; return SOCKET_ERROR;
} }
} }
*lpErrno = 0; if (lpErrno) *lpErrno = NO_ERROR;
/* Return Socket */ /* Return Socket */
return AcceptSocket; return AcceptSocket;
@ -1721,6 +1714,16 @@ WSPConnect(SOCKET Handle,
HANDLE SockEvent; HANDLE SockEvent;
int SocketDataLength; int SocketDataLength;
TRACE("Called (%lx) %lx:%d\n", Handle, ((const struct sockaddr_in *)SocketAddress)->sin_addr, ((const struct sockaddr_in *)SocketAddress)->sin_port);
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
Status = NtCreateEvent(&SockEvent, Status = NtCreateEvent(&SockEvent,
EVENT_ALL_ACCESS, EVENT_ALL_ACCESS,
NULL, NULL,
@ -1728,18 +1731,7 @@ WSPConnect(SOCKET Handle,
FALSE); FALSE);
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
return MsafdReturnWithErrno(Status, lpErrno, 0, NULL);
TRACE("Called\n");
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
NtClose(SockEvent);
if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR; return SOCKET_ERROR;
}
/* Bind us First */ /* Bind us First */
if (Socket->SharedData->State == SocketOpen) if (Socket->SharedData->State == SocketOpen)
@ -1899,7 +1891,7 @@ WSPConnect(SOCKET Handle,
} }
} }
TRACE("Ending\n"); TRACE("Ending %lx\n", IOSB.Status);
notify: notify:
if (ConnectInfo) HeapFree(GetProcessHeap(), 0, ConnectInfo); if (ConnectInfo) HeapFree(GetProcessHeap(), 0, ConnectInfo);
@ -1956,6 +1948,16 @@ WSPShutdown(SOCKET Handle,
NTSTATUS Status; NTSTATUS Status;
HANDLE SockEvent; HANDLE SockEvent;
TRACE("Called\n");
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
Status = NtCreateEvent(&SockEvent, Status = NtCreateEvent(&SockEvent,
EVENT_ALL_ACCESS, EVENT_ALL_ACCESS,
NULL, NULL,
@ -1963,18 +1965,7 @@ WSPShutdown(SOCKET Handle,
FALSE); FALSE);
if( !NT_SUCCESS(Status) ) if( !NT_SUCCESS(Status) )
return -1; return SOCKET_ERROR;
TRACE("Called\n");
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
NtClose(SockEvent);
*lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
/* Set AFD Disconnect Type */ /* Set AFD Disconnect Type */
switch (HowTo) switch (HowTo)
@ -2038,6 +2029,20 @@ WSPGetSockName(IN SOCKET Handle,
NTSTATUS Status; NTSTATUS Status;
HANDLE SockEvent; HANDLE SockEvent;
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
if (!Name || !NameLength)
{
if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR;
}
Status = NtCreateEvent(&SockEvent, Status = NtCreateEvent(&SockEvent,
EVENT_ALL_ACCESS, EVENT_ALL_ACCESS,
NULL, NULL,
@ -2047,22 +2052,6 @@ WSPGetSockName(IN SOCKET Handle,
if( !NT_SUCCESS(Status) ) if( !NT_SUCCESS(Status) )
return SOCKET_ERROR; return SOCKET_ERROR;
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle);
if (!Socket)
{
NtClose(SockEvent);
*lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
if (!Name || !NameLength)
{
NtClose(SockEvent);
*lpErrno = WSAEFAULT;
return SOCKET_ERROR;
}
/* Allocate a buffer for the address */ /* Allocate a buffer for the address */
TdiAddressSize = TdiAddressSize =
sizeof(TRANSPORT_ADDRESS) + Socket->SharedData->SizeOfLocalAddress; sizeof(TRANSPORT_ADDRESS) + Socket->SharedData->SizeOfLocalAddress;
@ -2071,7 +2060,7 @@ WSPGetSockName(IN SOCKET Handle,
if ( TdiAddress == NULL ) if ( TdiAddress == NULL )
{ {
NtClose( SockEvent ); NtClose( SockEvent );
*lpErrno = WSAENOBUFS; if (lpErrno) *lpErrno = WSAENOBUFS;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2116,7 +2105,7 @@ WSPGetSockName(IN SOCKET Handle,
else else
{ {
HeapFree(GlobalHeap, 0, TdiAddress); HeapFree(GlobalHeap, 0, TdiAddress);
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
} }
@ -2141,6 +2130,26 @@ WSPGetPeerName(IN SOCKET s,
NTSTATUS Status; NTSTATUS Status;
HANDLE SockEvent; HANDLE SockEvent;
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(s);
if (!Socket)
{
if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
if (Socket->SharedData->State != SocketConnected)
{
if (lpErrno) *lpErrno = WSAENOTCONN;
return SOCKET_ERROR;
}
if (!Name || !NameLength)
{
if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR;
}
Status = NtCreateEvent(&SockEvent, Status = NtCreateEvent(&SockEvent,
EVENT_ALL_ACCESS, EVENT_ALL_ACCESS,
NULL, NULL,
@ -2150,29 +2159,6 @@ WSPGetPeerName(IN SOCKET s,
if( !NT_SUCCESS(Status) ) if( !NT_SUCCESS(Status) )
return SOCKET_ERROR; return SOCKET_ERROR;
/* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(s);
if (!Socket)
{
NtClose(SockEvent);
*lpErrno = WSAENOTSOCK;
return SOCKET_ERROR;
}
if (Socket->SharedData->State != SocketConnected)
{
NtClose(SockEvent);
*lpErrno = WSAENOTCONN;
return SOCKET_ERROR;
}
if (!Name || !NameLength)
{
NtClose(SockEvent);
*lpErrno = WSAEFAULT;
return SOCKET_ERROR;
}
/* Allocate a buffer for the address */ /* Allocate a buffer for the address */
TdiAddressSize = sizeof(TRANSPORT_ADDRESS) + Socket->SharedData->SizeOfRemoteAddress; TdiAddressSize = sizeof(TRANSPORT_ADDRESS) + Socket->SharedData->SizeOfRemoteAddress;
SocketAddress = HeapAlloc(GlobalHeap, 0, TdiAddressSize); SocketAddress = HeapAlloc(GlobalHeap, 0, TdiAddressSize);
@ -2180,7 +2166,7 @@ WSPGetPeerName(IN SOCKET s,
if ( SocketAddress == NULL ) if ( SocketAddress == NULL )
{ {
NtClose( SockEvent ); NtClose( SockEvent );
*lpErrno = WSAENOBUFS; if (lpErrno) *lpErrno = WSAENOBUFS;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2223,7 +2209,7 @@ WSPGetPeerName(IN SOCKET s,
else else
{ {
HeapFree(GlobalHeap, 0, SocketAddress); HeapFree(GlobalHeap, 0, SocketAddress);
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
} }
@ -2316,9 +2302,16 @@ WSPIoctl(IN SOCKET Handle,
Errno = WSAEINVAL; Errno = WSAEINVAL;
break; break;
} }
if (Socket->SharedData->SocketType != SOCK_STREAM)
{
Errno = WSAEINVAL;
break;
}
/* FIXME: Return false for now */ /* FIXME: Return false if OOBINLINE is true for now
*(BOOL*)lpvOutBuffer = FALSE; We should MSG_PEEK|MSG_OOB check with driver
*/
*(BOOL*)lpvOutBuffer = !Socket->SharedData->OobInline;
cbRet = sizeof(BOOL); cbRet = sizeof(BOOL);
Errno = NO_ERROR; Errno = NO_ERROR;
@ -2413,12 +2406,13 @@ WSPGetSockOpt(IN SOCKET Handle,
INT BufferSize; INT BufferSize;
BOOL BoolBuffer; BOOL BoolBuffer;
INT IntBuffer; INT IntBuffer;
INT Errno;
/* Get the Socket Structure associate to this Socket*/ /* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle); Socket = GetSocketStructure(Handle);
if (Socket == NULL) if (Socket == NULL)
{ {
*lpErrno = WSAENOTSOCK; if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2471,7 +2465,7 @@ WSPGetSockOpt(IN SOCKET Handle,
case SO_LINGER: case SO_LINGER:
if (Socket->SharedData->SocketType == SOCK_DGRAM) if (Socket->SharedData->SocketType == SOCK_DGRAM)
{ {
*lpErrno = WSAENOPROTOOPT; if (lpErrno) *lpErrno = WSAENOPROTOOPT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
Buffer = &Socket->SharedData->LingerData; Buffer = &Socket->SharedData->LingerData;
@ -2534,13 +2528,13 @@ WSPGetSockOpt(IN SOCKET Handle,
default: default:
DbgPrint("MSAFD: Get unknown optname %x\n", OptionName); DbgPrint("MSAFD: Get unknown optname %x\n", OptionName);
*lpErrno = WSAENOPROTOOPT; if (lpErrno) *lpErrno = WSAENOPROTOOPT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
if (*OptionLength < BufferSize) if (*OptionLength < BufferSize)
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
*OptionLength = BufferSize; *OptionLength = BufferSize;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2549,20 +2543,21 @@ WSPGetSockOpt(IN SOCKET Handle,
return 0; return 0;
default: default:
*lpErrno = WSAEINVAL; if (lpErrno) *lpErrno = WSAEINVAL;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
SendToHelper: SendToHelper:
*lpErrno = Socket->HelperData->WSHGetSocketInformation(Socket->HelperContext, Errno = Socket->HelperData->WSHGetSocketInformation(Socket->HelperContext,
Handle, Handle,
Socket->TdiAddressHandle, Socket->TdiAddressHandle,
Socket->TdiConnectionHandle, Socket->TdiConnectionHandle,
Level, Level,
OptionName, OptionName,
OptionValue, OptionValue,
(LPINT)OptionLength); (LPINT)OptionLength);
return (*lpErrno == 0) ? 0 : SOCKET_ERROR; if (lpErrno) *lpErrno = Errno;
return (Errno == NO_ERROR) ? NO_ERROR : SOCKET_ERROR;
} }
INT INT
@ -2576,12 +2571,13 @@ WSPSetSockOpt(
OUT LPINT lpErrno) OUT LPINT lpErrno)
{ {
PSOCKET_INFORMATION Socket; PSOCKET_INFORMATION Socket;
INT Errno;
/* Get the Socket Structure associate to this Socket*/ /* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(s); Socket = GetSocketStructure(s);
if (Socket == NULL) if (Socket == NULL)
{ {
*lpErrno = WSAENOTSOCK; if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2594,84 +2590,93 @@ WSPSetSockOpt(
case SO_BROADCAST: case SO_BROADCAST:
if (optlen < sizeof(BOOL)) if (optlen < sizeof(BOOL))
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
Socket->SharedData->Broadcast = (*optval != 0) ? 1 : 0; Socket->SharedData->Broadcast = (*optval != 0) ? 1 : 0;
return 0; return NO_ERROR;
case SO_OOBINLINE:
if (optlen < sizeof(BOOL))
{
if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR;
}
Socket->SharedData->OobInline = (*optval != 0) ? 1 : 0;
return NO_ERROR;
case SO_DONTLINGER: case SO_DONTLINGER:
if (optlen < sizeof(BOOL)) if (optlen < sizeof(BOOL))
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
Socket->SharedData->LingerData.l_onoff = (*optval != 0) ? 0 : 1; Socket->SharedData->LingerData.l_onoff = (*optval != 0) ? 0 : 1;
return 0; return NO_ERROR;
case SO_REUSEADDR: case SO_REUSEADDR:
if (optlen < sizeof(BOOL)) if (optlen < sizeof(BOOL))
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
Socket->SharedData->ReuseAddresses = (*optval != 0) ? 1 : 0; Socket->SharedData->ReuseAddresses = (*optval != 0) ? 1 : 0;
return 0; return NO_ERROR;
case SO_EXCLUSIVEADDRUSE: case SO_EXCLUSIVEADDRUSE:
if (optlen < sizeof(BOOL)) if (optlen < sizeof(BOOL))
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
Socket->SharedData->ExclusiveAddressUse = (*optval != 0) ? 1 : 0; Socket->SharedData->ExclusiveAddressUse = (*optval != 0) ? 1 : 0;
return 0; return NO_ERROR;
case SO_LINGER: case SO_LINGER:
if (optlen < sizeof(struct linger)) if (optlen < sizeof(struct linger))
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
RtlCopyMemory(&Socket->SharedData->LingerData, RtlCopyMemory(&Socket->SharedData->LingerData,
optval, optval,
sizeof(struct linger)); sizeof(struct linger));
return 0; return NO_ERROR;
case SO_SNDBUF: case SO_SNDBUF:
if (optlen < sizeof(DWORD)) if (optlen < sizeof(DWORD))
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
/* TODO: The total per-socket buffer space reserved for sends */ /* TODO: The total per-socket buffer space reserved for sends */
ERR("Setting send buf to %x is not implemented yet\n", optval); ERR("Setting send buf to %x is not implemented yet\n", optval);
return 0; return NO_ERROR;
case SO_SNDTIMEO: case SO_SNDTIMEO:
if (optlen < sizeof(DWORD)) if (optlen < sizeof(DWORD))
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
RtlCopyMemory(&Socket->SharedData->SendTimeout, RtlCopyMemory(&Socket->SharedData->SendTimeout,
optval, optval,
sizeof(DWORD)); sizeof(DWORD));
return 0; return NO_ERROR;
case SO_RCVTIMEO: case SO_RCVTIMEO:
if (optlen < sizeof(DWORD)) if (optlen < sizeof(DWORD))
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
RtlCopyMemory(&Socket->SharedData->RecvTimeout, RtlCopyMemory(&Socket->SharedData->RecvTimeout,
optval, optval,
sizeof(DWORD)); sizeof(DWORD));
return 0; return NO_ERROR;
case SO_KEEPALIVE: case SO_KEEPALIVE:
case SO_DONTROUTE: case SO_DONTROUTE:
@ -2681,20 +2686,21 @@ WSPSetSockOpt(
default: default:
/* Obviously this is a hack */ /* Obviously this is a hack */
ERR("MSAFD: Set unknown optname %x\n", optname); ERR("MSAFD: Set unknown optname %x\n", optname);
return 0; return NO_ERROR;
} }
} }
SendToHelper: SendToHelper:
*lpErrno = Socket->HelperData->WSHSetSocketInformation(Socket->HelperContext, Errno = Socket->HelperData->WSHSetSocketInformation(Socket->HelperContext,
s, s,
Socket->TdiAddressHandle, Socket->TdiAddressHandle,
Socket->TdiConnectionHandle, Socket->TdiConnectionHandle,
level, level,
optname, optname,
(PCHAR)optval, (PCHAR)optval,
optlen); optlen);
return (*lpErrno == 0) ? 0 : SOCKET_ERROR; if (lpErrno) *lpErrno = Errno;
return (Errno == NO_ERROR) ? NO_ERROR : SOCKET_ERROR;
} }
/* /*
@ -2788,7 +2794,7 @@ WSPAddressToString(IN LPSOCKADDR lpsaAddress,
if (!lpsaAddress || !lpszAddressString || !lpdwAddressStringLength) if (!lpsaAddress || !lpszAddressString || !lpdwAddressStringLength)
{ {
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2797,7 +2803,7 @@ WSPAddressToString(IN LPSOCKADDR lpsaAddress,
case AF_INET: case AF_INET:
if (dwAddressLength < sizeof(SOCKADDR_IN)) if (dwAddressLength < sizeof(SOCKADDR_IN))
{ {
*lpErrno = WSAEINVAL; if (lpErrno) *lpErrno = WSAEINVAL;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
swprintf(buffer, swprintf(buffer,
@ -2815,7 +2821,7 @@ WSPAddressToString(IN LPSOCKADDR lpsaAddress,
} }
break; break;
default: default:
*lpErrno = WSAEINVAL; if (lpErrno) *lpErrno = WSAEINVAL;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2824,7 +2830,7 @@ WSPAddressToString(IN LPSOCKADDR lpsaAddress,
if (*lpdwAddressStringLength < size) if (*lpdwAddressStringLength < size)
{ {
*lpdwAddressStringLength = size; *lpdwAddressStringLength = size;
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2849,7 +2855,7 @@ WSPStringToAddress(IN LPWSTR AddressString,
if (!lpAddressLength || !lpAddress || !AddressString) if (!lpAddressLength || !lpAddress || !AddressString)
{ {
*lpErrno = WSAEINVAL; if (lpErrno) *lpErrno = WSAEINVAL;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
@ -2871,7 +2877,7 @@ WSPStringToAddress(IN LPWSTR AddressString,
if (*lpAddressLength < (INT)sizeof(SOCKADDR_IN)) if (*lpAddressLength < (INT)sizeof(SOCKADDR_IN))
{ {
*lpAddressLength = sizeof(SOCKADDR_IN); *lpAddressLength = sizeof(SOCKADDR_IN);
*lpErrno = WSAEFAULT; if (lpErrno) *lpErrno = WSAEFAULT;
} }
else else
{ {
@ -2908,13 +2914,13 @@ WSPStringToAddress(IN LPWSTR AddressString,
pos += wcscspn(&AddressString[pos], L".") + 1; pos += wcscspn(&AddressString[pos], L".") + 1;
} }
*lpErrno = 0; if (lpErrno) *lpErrno = NO_ERROR;
sockaddr->sin_addr.s_addr = inetaddr; sockaddr->sin_addr.s_addr = inetaddr;
} }
} }
if (!*lpErrno) if (lpErrno && !*lpErrno)
{ {
return 0; return 0;
} }
@ -2935,7 +2941,8 @@ WSPCleanup(OUT LPINT lpErrno)
{ {
TRACE("Leaving.\n"); TRACE("Leaving.\n");
*lpErrno = NO_ERROR;
if (lpErrno) *lpErrno = NO_ERROR;
return 0; return 0;
} }
@ -2965,7 +2972,7 @@ GetSocketInformation(PSOCKET_INFORMATION Socket,
IO_STATUS_BLOCK DummyIOSB; IO_STATUS_BLOCK DummyIOSB;
AFD_INFO InfoData; AFD_INFO InfoData;
NTSTATUS Status; NTSTATUS Status;
PVOID APCContext; PAFDAPCCONTEXT APCContext;
PIO_APC_ROUTINE APCFunction; PIO_APC_ROUTINE APCFunction;
HANDLE Event = NULL; HANDLE Event = NULL;
HANDLE SockEvent; HANDLE SockEvent;
@ -2977,7 +2984,7 @@ GetSocketInformation(PSOCKET_INFORMATION Socket,
FALSE); FALSE);
if( !NT_SUCCESS(Status) ) if( !NT_SUCCESS(Status) )
return -1; return SOCKET_ERROR;
/* Set Info Class */ /* Set Info Class */
InfoData.InformationClass = AfdInformationClass; InfoData.InformationClass = AfdInformationClass;
@ -3002,7 +3009,7 @@ GetSocketInformation(PSOCKET_INFORMATION Socket,
if (CompletionRoutine == NULL) if (CompletionRoutine == NULL)
{ {
/* Using Overlapped Structure, but no Completition Routine, so no need for APC */ /* Using Overlapped Structure, but no Completition Routine, so no need for APC */
APCContext = Overlapped; APCContext = (PAFDAPCCONTEXT)Overlapped;
APCFunction = NULL; APCFunction = NULL;
Event = Overlapped->hEvent; Event = Overlapped->hEvent;
} }
@ -3016,9 +3023,9 @@ GetSocketInformation(PSOCKET_INFORMATION Socket,
ERR("Not enough memory for APC Context\n"); ERR("Not enough memory for APC Context\n");
return WSAEFAULT; return WSAEFAULT;
} }
((PAFDAPCCONTEXT)APCContext)->lpCompletionRoutine = CompletionRoutine; APCContext->lpCompletionRoutine = CompletionRoutine;
((PAFDAPCCONTEXT)APCContext)->lpOverlapped = Overlapped; APCContext->lpOverlapped = Overlapped;
((PAFDAPCCONTEXT)APCContext)->lpSocket = Socket; APCContext->lpSocket = Socket;
} }
IOSB = (PIO_STATUS_BLOCK)&Overlapped->Internal; IOSB = (PIO_STATUS_BLOCK)&Overlapped->Internal;
@ -3054,7 +3061,7 @@ GetSocketInformation(PSOCKET_INFORMATION Socket,
} }
if (Status != STATUS_SUCCESS) if (Status != STATUS_SUCCESS)
return -1; return SOCKET_ERROR;
/* Return Information */ /* Return Information */
if (Ulong != NULL) if (Ulong != NULL)
@ -3072,7 +3079,7 @@ GetSocketInformation(PSOCKET_INFORMATION Socket,
NtClose( SockEvent ); NtClose( SockEvent );
return 0; return NO_ERROR;
} }
@ -3090,7 +3097,7 @@ SetSocketInformation(PSOCKET_INFORMATION Socket,
IO_STATUS_BLOCK DummyIOSB; IO_STATUS_BLOCK DummyIOSB;
AFD_INFO InfoData; AFD_INFO InfoData;
NTSTATUS Status; NTSTATUS Status;
PVOID APCContext; PAFDAPCCONTEXT APCContext;
PIO_APC_ROUTINE APCFunction; PIO_APC_ROUTINE APCFunction;
HANDLE Event = NULL; HANDLE Event = NULL;
HANDLE SockEvent; HANDLE SockEvent;
@ -3102,7 +3109,7 @@ SetSocketInformation(PSOCKET_INFORMATION Socket,
FALSE); FALSE);
if( !NT_SUCCESS(Status) ) if( !NT_SUCCESS(Status) )
return -1; return SOCKET_ERROR;
/* Set Info Class */ /* Set Info Class */
InfoData.InformationClass = AfdInformationClass; InfoData.InformationClass = AfdInformationClass;
@ -3141,7 +3148,7 @@ SetSocketInformation(PSOCKET_INFORMATION Socket,
if (CompletionRoutine == NULL) if (CompletionRoutine == NULL)
{ {
/* Using Overlapped Structure, but no Completition Routine, so no need for APC */ /* Using Overlapped Structure, but no Completition Routine, so no need for APC */
APCContext = Overlapped; APCContext = (PAFDAPCCONTEXT)Overlapped;
APCFunction = NULL; APCFunction = NULL;
Event = Overlapped->hEvent; Event = Overlapped->hEvent;
} }
@ -3155,9 +3162,9 @@ SetSocketInformation(PSOCKET_INFORMATION Socket,
ERR("Not enough memory for APC Context\n"); ERR("Not enough memory for APC Context\n");
return WSAEFAULT; return WSAEFAULT;
} }
((PAFDAPCCONTEXT)APCContext)->lpCompletionRoutine = CompletionRoutine; APCContext->lpCompletionRoutine = CompletionRoutine;
((PAFDAPCCONTEXT)APCContext)->lpOverlapped = Overlapped; APCContext->lpOverlapped = Overlapped;
((PAFDAPCCONTEXT)APCContext)->lpSocket = Socket; APCContext->lpSocket = Socket;
} }
IOSB = (PIO_STATUS_BLOCK)&Overlapped->Internal; IOSB = (PIO_STATUS_BLOCK)&Overlapped->Internal;
@ -3194,7 +3201,7 @@ SetSocketInformation(PSOCKET_INFORMATION Socket,
return WSA_IO_PENDING; return WSA_IO_PENDING;
} }
return Status == STATUS_SUCCESS ? 0 : -1; return Status == STATUS_SUCCESS ? NO_ERROR : SOCKET_ERROR;
} }
@ -3236,7 +3243,7 @@ int CreateContext(PSOCKET_INFORMATION Socket)
FALSE); FALSE);
if( !NT_SUCCESS(Status) ) if( !NT_SUCCESS(Status) )
return -1; return SOCKET_ERROR;
/* Create Context */ /* Create Context */
ContextData.SharedData = *Socket->SharedData; ContextData.SharedData = *Socket->SharedData;
@ -3269,7 +3276,7 @@ int CreateContext(PSOCKET_INFORMATION Socket)
NtClose( SockEvent ); NtClose( SockEvent );
return Status == STATUS_SUCCESS ? 0 : -1; return Status == STATUS_SUCCESS ? NO_ERROR : SOCKET_ERROR;
} }
BOOLEAN SockCreateOrReferenceAsyncThread(VOID) BOOLEAN SockCreateOrReferenceAsyncThread(VOID)

17
reactos/dll/win32/msafd/misc/event.c
View file

@ -30,20 +30,19 @@ WSPEventSelect(
BOOLEAN BlockMode; BOOLEAN BlockMode;
HANDLE SockEvent; HANDLE SockEvent;
Status = NtCreateEvent( &SockEvent, EVENT_ALL_ACCESS,
NULL, 1, FALSE );
if( !NT_SUCCESS(Status) ) return -1;
/* Get the Socket Structure associate to this Socket*/ /* Get the Socket Structure associate to this Socket*/
Socket = GetSocketStructure(Handle); Socket = GetSocketStructure(Handle);
if (!Socket) if (!Socket)
{ {
NtClose(SockEvent); if (lpErrno) *lpErrno = WSAENOTSOCK;
*lpErrno = WSAENOTSOCK;
return SOCKET_ERROR; return SOCKET_ERROR;
} }
Status = NtCreateEvent( &SockEvent, EVENT_ALL_ACCESS,
NULL, 1, FALSE );
if( !NT_SUCCESS(Status) ) return SOCKET_ERROR;
/* Set Socket to Non-Blocking */ /* Set Socket to Non-Blocking */
BlockMode = TRUE; BlockMode = TRUE;
SetSocketInformation(Socket, AFD_INFO_BLOCKING_MODE, &BlockMode, NULL, NULL, NULL, NULL); SetSocketInformation(Socket, AFD_INFO_BLOCKING_MODE, &BlockMode, NULL, NULL, NULL, NULL);
@ -157,7 +156,7 @@ WSPEnumNetworkEvents(
if( !NT_SUCCESS(Status) ) { if( !NT_SUCCESS(Status) ) {
ERR("Could not make an event %x\n", Status); ERR("Could not make an event %x\n", Status);
return -1; return SOCKET_ERROR;
} }
/* Get the Socket Structure associate to this Socket*/ /* Get the Socket Structure associate to this Socket*/
@ -165,7 +164,7 @@ WSPEnumNetworkEvents(
if (!Socket) if (!Socket)
{ {
NtClose(SockEvent); NtClose(SockEvent);
*lpErrno = WSAENOTSOCK; if (lpErrno) *lpErrno = WSAENOTSOCK;
return SOCKET_ERROR; return SOCKET_ERROR;
} }