diff --git a/include/ntapi/nt_socket.h b/include/ntapi/nt_socket.h index 1f8dd14..d56f370 100644 --- a/include/ntapi/nt_socket.h +++ b/include/ntapi/nt_socket.h @@ -518,6 +518,15 @@ typedef int32_t __cdecl ntapi_sc_setsockopt( __out nt_io_status_block * iosb __optional); +typedef int32_t __cdecl ntapi_sc_getsockopt( + __in nt_socket * hssocket, + __in int32_t level, + __in int32_t optname, + __out void * optval, + __in uint32_t optlen, + __out nt_io_status_block * iosb); + + typedef int32_t __cdecl ntapi_sc_wait(nt_socket * hssocket, nt_iosb * iosb, nt_timeout * timeout); #endif diff --git a/include/ntapi/ntapi.h b/include/ntapi/ntapi.h index 904a563..054d66a 100644 --- a/include/ntapi/ntapi.h +++ b/include/ntapi/ntapi.h @@ -548,6 +548,7 @@ typedef struct _ntapi_vtbl { ntapi_sc_server_accept_connection * sc_server_accept_connection; ntapi_sc_server_duplicate_socket * sc_server_duplicate_socket; ntapi_sc_setsockopt * sc_setsockopt; + ntapi_sc_getsockopt * sc_getsockopt; ntapi_sc_wait * sc_wait; /* nt_mount.h */ diff --git a/project/common.mk b/project/common.mk index bc41360..1f5d599 100644 --- a/project/common.mk +++ b/project/common.mk @@ -62,6 +62,7 @@ COMMON_SRCS = \ src/socket/ntapi_sc_server_accept_connection_v2.c \ src/socket/ntapi_sc_server_duplicate_socket.c \ src/socket/ntapi_sc_setsockopt.c \ + src/socket/ntapi_sc_getsockopt.c \ src/socket/ntapi_sc_shutdown.c \ src/socket/ntapi_sc_socket_v1.c \ src/socket/ntapi_sc_socket_v2.c \ diff --git a/src/internal/ntapi.c b/src/internal/ntapi.c index 451ccd7..07f0da7 100644 --- a/src/internal/ntapi.c +++ b/src/internal/ntapi.c @@ -309,6 +309,7 @@ static int32_t __fastcall __ntapi_init_once(ntapi_vtbl ** pvtbl) __ntapi->sc_shutdown = __ntapi_sc_shutdown; __ntapi->sc_server_duplicate_socket = __ntapi_sc_server_duplicate_socket; __ntapi->sc_setsockopt = __ntapi_sc_setsockopt; + __ntapi->sc_getsockopt = __ntapi_sc_getsockopt; __ntapi->sc_wait = __ntapi_sc_wait; /* nt_mount.h */ diff --git a/src/internal/ntapi_fnapi.h b/src/internal/ntapi_fnapi.h index 05f4707..396fb59 100644 --- a/src/internal/ntapi_fnapi.h +++ b/src/internal/ntapi_fnapi.h @@ -204,6 +204,7 @@ ntapi_sc_server_accept_connection __ntapi_sc_server_accept_connection_v1; ntapi_sc_server_accept_connection __ntapi_sc_server_accept_connection_v2; ntapi_sc_server_duplicate_socket __ntapi_sc_server_duplicate_socket; ntapi_sc_setsockopt __ntapi_sc_setsockopt; +ntapi_sc_getsockopt __ntapi_sc_getsockopt; ntapi_sc_listen __ntapi_sc_listen; ntapi_sc_accept __ntapi_sc_accept; ntapi_sc_send __ntapi_sc_send; diff --git a/src/socket/ntapi_sc_getsockopt.c b/src/socket/ntapi_sc_getsockopt.c new file mode 100644 index 0000000..3fa59d4 --- /dev/null +++ b/src/socket/ntapi_sc_getsockopt.c @@ -0,0 +1,50 @@ +/********************************************************/ +/* ntapi: Native API core library */ +/* Copyright (C) 2013--2016 Z. Gilboa */ +/* Released under GPLv2 and GPLv3; see COPYING.NTAPI. */ +/********************************************************/ + +#include +#include +#include +#include +#include +#include "ntapi_impl.h" + +int32_t __cdecl __ntapi_sc_getsockopt( + __in nt_socket * hssocket, + __in int32_t level, + __in int32_t optname, + __out void * optval, + __in uint32_t optlen, + __out nt_io_status_block * iosb) +{ + nt_afd_sockopt_info afd_sockopt; + + if (!iosb) + return NT_STATUS_INVALID_PARAMETER_6; + + /* afd_sockopt */ + afd_sockopt.mode = NT_AFD_SOCKOPT_GET; + afd_sockopt.level = level; + afd_sockopt.optname = optname; + afd_sockopt.optval = 0; + afd_sockopt.optlen = 0; + afd_sockopt.ding = 1; + + hssocket->iostatus = __ntapi->zw_device_io_control_file( + hssocket->hsocket, + hssocket->hevent, + 0, + 0, + iosb, + NT_AFD_IOCTL_SOCKOPT, + &afd_sockopt, + sizeof(afd_sockopt), + optval, + optlen); + + return hssocket->iostatus + ? __ntapi->sc_wait(hssocket,iosb,0) + : NT_STATUS_SUCCESS; +}