ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ws2_32: Use IOCTL_AFD_GET_EVENTS in WSAEnumNetworkEvents(). 

Signed-off-by: Zebediah Figura <z.figura12@gmail.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Zebediah Figura 2021-06-15 10:39:45 -05:00 committed by Alexandre Julliard
parent 2daf76432e
commit a916887b94
1 changed files with 60 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
dlls/ws2_32/socket.c
View File

@ -3826,6 +3826,33 @@ int WINAPI WS_select(int nfds, WS_fd_set *ws_readfds,
return ret; return ret;
} }
static unsigned int afd_poll_flag_to_win32( unsigned int flags )
{
static const unsigned int map[] =
{
FD_READ, /* READ */
FD_OOB, /* OOB */
FD_WRITE, /* WRITE */
FD_CLOSE, /* HUP */
FD_CLOSE, /* RESET */
0, /* CLOSE */
FD_CONNECT, /* CONNECT */
FD_ACCEPT, /* ACCEPT */
FD_CONNECT, /* CONNECT_ERR */
};
unsigned int i, ret = 0;
for (i = 0; i < ARRAY_SIZE(map); ++i)
{
if (flags & (1 << i)) ret |= map[i];
}
return ret;
}
/*********************************************************************** /***********************************************************************
* WSAPoll * WSAPoll
*/ */
@ -4347,36 +4374,47 @@ SOCKET WINAPI WS_socket(int af, int type, int protocol)
/*********************************************************************** /***********************************************************************
* WSAEnumNetworkEvents (WS2_32.36) * WSAEnumNetworkEvents (ws2_32.@)
*/ */
int WINAPI WSAEnumNetworkEvents(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent) int WINAPI WSAEnumNetworkEvents( SOCKET s, WSAEVENT event, WSANETWORKEVENTS *ret_events )
{ {
int ret; struct afd_get_events_params params;
int i; IO_STATUS_BLOCK io;
int errors[FD_MAX_EVENTS]; NTSTATUS status;
TRACE("%04lx, hEvent %p, lpEvent %p\n", s, hEvent, lpEvent ); TRACE( "socket %#lx, event %p, events %p\n", s, event, ret_events );
SERVER_START_REQ( get_socket_event ) ResetEvent( event );
status = NtDeviceIoControlFile( (HANDLE)s, NULL, NULL, NULL, &io, IOCTL_AFD_GET_EVENTS,
NULL, 0, &params, sizeof(params) );
if (!status)
{ {
req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); ret_events->lNetworkEvents = afd_poll_flag_to_win32( params.flags );
req->service = TRUE;
req->c_event = wine_server_obj_handle( hEvent ); if (ret_events->lNetworkEvents & FD_READ)
wine_server_set_reply( req, errors, sizeof(errors) ); ret_events->iErrorCode[FD_READ_BIT] = NtStatusToWSAError( params.status[AFD_POLL_BIT_READ] );
if (!(ret = wine_server_call(req))) lpEvent->lNetworkEvents = reply->pmask & reply->mask;
} if (ret_events->lNetworkEvents & FD_WRITE)
SERVER_END_REQ; ret_events->iErrorCode[FD_WRITE_BIT] = NtStatusToWSAError( params.status[AFD_POLL_BIT_WRITE] );
if (!ret)
{ if (ret_events->lNetworkEvents & FD_OOB)
for (i = 0; i < FD_MAX_EVENTS; i++) ret_events->iErrorCode[FD_OOB_BIT] = NtStatusToWSAError( params.status[AFD_POLL_BIT_OOB] );
if (ret_events->lNetworkEvents & FD_ACCEPT)
ret_events->iErrorCode[FD_ACCEPT_BIT] = NtStatusToWSAError( params.status[AFD_POLL_BIT_ACCEPT] );
if (ret_events->lNetworkEvents & FD_CONNECT)
ret_events->iErrorCode[FD_CONNECT_BIT] = NtStatusToWSAError( params.status[AFD_POLL_BIT_CONNECT_ERR] );
if (ret_events->lNetworkEvents & FD_CLOSE)
{ {
if (lpEvent->lNetworkEvents & (1 << i)) if (!(ret_events->iErrorCode[FD_CLOSE_BIT] = NtStatusToWSAError( params.status[AFD_POLL_BIT_HUP] )))
lpEvent->iErrorCode[i] = errors[i]; ret_events->iErrorCode[FD_CLOSE_BIT] = NtStatusToWSAError( params.status[AFD_POLL_BIT_RESET] );
} }
return 0;
} }
SetLastError(WSAEINVAL); SetLastError( NtStatusToWSAError( status ) );
return SOCKET_ERROR; return status ? -1 : 0;
} }