diff options
Diffstat (limited to 'src/socket/ntapi_sc_shutdown.c')
| -rw-r--r-- | src/socket/ntapi_sc_shutdown.c | 26 |
1 files changed, 4 insertions, 22 deletions
diff --git a/src/socket/ntapi_sc_shutdown.c b/src/socket/ntapi_sc_shutdown.c index 0aa5c36..acbc09b 100644 --- a/src/socket/ntapi_sc_shutdown.c +++ b/src/socket/ntapi_sc_shutdown.c @@ -13,8 +13,7 @@ int32_t __cdecl __ntapi_sc_shutdown( __in nt_socket * hssocket, - __in uintptr_t psxhow, - __in uintptr_t afdhow, + __in uint32_t afdhow, __out nt_io_status_block * iosb __optional) { nt_afd_disconnect_info afd_disconnect; @@ -22,27 +21,10 @@ int32_t __cdecl __ntapi_sc_shutdown( iosb = iosb ? iosb : &siosb; - if (afdhow == 0) { - switch (psxhow) { - case NT_SHUT_RD: - afdhow = NT_AFD_DISCONNECT_RD; - break; + if (afdhow & ~(NT_AFD_DISCONNECT_RD | NT_AFD_DISCONNECT_WR)) + return NT_STATUS_INVALID_PARAMETER; - case NT_SHUT_WR: - afdhow = NT_AFD_DISCONNECT_WR; - break; - - case NT_SHUT_RDWR: - afdhow = NT_AFD_DISCONNECT_RD | NT_AFD_DISCONNECT_WR; - break; - - default: - return NT_STATUS_INVALID_PARAMETER_2; - break; - } - } - - afd_disconnect.shutdown_flags = (uint32_t)afdhow; + afd_disconnect.shutdown_flags = afdhow; afd_disconnect.unknown[0] = 0xff; afd_disconnect.unknown[1] = 0xff; afd_disconnect.unknown[2] = 0xff; |