summaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
Diffstat
-rw-r--r--include/ntapi/nt_socket.h5
-rw-r--r--include/ntapi/ntapi.h1
-rw-r--r--project/common.mk2
-rw-r--r--src/internal/ntapi.c2
-rw-r--r--src/internal/ntapi_fnapi.h2
-rw-r--r--src/socket/ntapi_sc_getpeername_v1.c80
-rw-r--r--src/socket/ntapi_sc_getpeername_v2.c42
7 files changed, 134 insertions, 0 deletions
diff --git a/include/ntapi/nt_socket.h b/include/ntapi/nt_socket.h
index fcf6d96..c984b92 100644
--- a/include/ntapi/nt_socket.h
+++ b/include/ntapi/nt_socket.h
@@ -413,6 +413,11 @@ typedef int32_t __cdecl ntapi_sc_getsockname(
__in uint16_t * addrlen,
__out nt_io_status_block * iosb __optional);
+typedef int32_t __cdecl ntapi_sc_getpeername(
+ __in nt_socket * hssocket,
+ __in nt_sockaddr * addr,
+ __in uint16_t * addrlen,
+ __out nt_io_status_block * iosb __optional);
typedef int32_t __cdecl ntapi_sc_server_accept_connection(
__in nt_socket * hssocket,
diff --git a/include/ntapi/ntapi.h b/include/ntapi/ntapi.h
index d6e5f65..17d9871 100644
--- a/include/ntapi/ntapi.h
+++ b/include/ntapi/ntapi.h
@@ -544,6 +544,7 @@ typedef struct _ntapi_vtbl {
ntapi_sc_recv * sc_recv;
ntapi_sc_shutdown * sc_shutdown;
ntapi_sc_getsockname * sc_getsockname;
+ ntapi_sc_getpeername * sc_getpeername;
ntapi_sc_server_accept_connection * sc_server_accept_connection;
ntapi_sc_server_duplicate_socket * sc_server_duplicate_socket;
ntapi_sc_wait * sc_wait;
diff --git a/project/common.mk b/project/common.mk
index abbbb20..0e3bb9a 100644
--- a/project/common.mk
+++ b/project/common.mk
@@ -51,6 +51,8 @@ COMMON_SRCS = \
src/socket/ntapi_sc_bind_v2.c \
src/socket/ntapi_sc_connect_v1.c \
src/socket/ntapi_sc_connect_v2.c \
+ src/socket/ntapi_sc_getpeername_v1.c \
+ src/socket/ntapi_sc_getpeername_v2.c \
src/socket/ntapi_sc_getsockname_v1.c \
src/socket/ntapi_sc_getsockname_v2.c \
src/socket/ntapi_sc_listen.c \
diff --git a/src/internal/ntapi.c b/src/internal/ntapi.c
index 048a8db..e8173ce 100644
--- a/src/internal/ntapi.c
+++ b/src/internal/ntapi.c
@@ -355,6 +355,7 @@ static int32_t __fastcall __ntapi_init_once(ntapi_vtbl ** pvtbl)
__ntapi->sc_connect = __ntapi_sc_connect_v2;
__ntapi->sc_server_accept_connection = __ntapi_sc_server_accept_connection_v2;
__ntapi->sc_getsockname = __ntapi_sc_getsockname_v2;
+ __ntapi->sc_getpeername = __ntapi_sc_getpeername_v2;
} else {
__ntapi->tt_fork = __ntapi_tt_fork_v1;
__ntapi->tt_create_native_process = __ntapi_tt_create_native_process_v1;
@@ -364,6 +365,7 @@ static int32_t __fastcall __ntapi_init_once(ntapi_vtbl ** pvtbl)
__ntapi->sc_connect = __ntapi_sc_connect_v1;
__ntapi->sc_server_accept_connection = __ntapi_sc_server_accept_connection_v1;
__ntapi->sc_getsockname = __ntapi_sc_getsockname_v1;
+ __ntapi->sc_getpeername = __ntapi_sc_getpeername_v1;
}
/* internals */
diff --git a/src/internal/ntapi_fnapi.h b/src/internal/ntapi_fnapi.h
index 178f2ba..a22c659 100644
--- a/src/internal/ntapi_fnapi.h
+++ b/src/internal/ntapi_fnapi.h
@@ -198,6 +198,8 @@ ntapi_sc_connect __ntapi_sc_connect_v1;
ntapi_sc_connect __ntapi_sc_connect_v2;
ntapi_sc_getsockname __ntapi_sc_getsockname_v1;
ntapi_sc_getsockname __ntapi_sc_getsockname_v2;
+ntapi_sc_getpeername __ntapi_sc_getpeername_v1;
+ntapi_sc_getpeername __ntapi_sc_getpeername_v2;
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;
diff --git a/src/socket/ntapi_sc_getpeername_v1.c b/src/socket/ntapi_sc_getpeername_v1.c
new file mode 100644
index 0000000..f38e11a
--- /dev/null
+++ b/src/socket/ntapi_sc_getpeername_v1.c
@@ -0,0 +1,80 @@
+/********************************************************/
+/* ntapi: Native API core library */
+/* Copyright (C) 2013--2016 Z. Gilboa */
+/* Released under GPLv2 and GPLv3; see COPYING.NTAPI. */
+/********************************************************/
+
+#include <psxtypes/psxtypes.h>
+#include <ntapi/nt_object.h>
+#include <ntapi/nt_file.h>
+#include <ntapi/nt_socket.h>
+#include <ntapi/ntapi.h>
+#include "ntapi_impl.h"
+
+typedef struct _nt_afd_server_socket_name_info {
+ uint32_t unknown;
+ uint32_t type;
+ uint32_t service_flags;
+ char sa_data[14];
+} nt_afd_server_socket_name_info;
+
+
+struct __addr_memcpy {
+ uint16_t d0;
+ uint16_t d1;
+ uint16_t d2;
+ uint16_t d3;
+ uint16_t d4;
+ uint16_t d5;
+ uint16_t d6;
+ uint16_t d7;
+};
+
+
+int32_t __cdecl __ntapi_sc_getpeername_v1(
+ __in nt_socket * hssocket,
+ __in nt_sockaddr * addr,
+ __in uint16_t * addrlen,
+ __out nt_io_status_block * iosb __optional)
+{
+ nt_io_status_block siosb;
+ nt_afd_server_socket_name_info sock_name_info;
+
+ struct __addr_memcpy * asrc;
+ struct __addr_memcpy * adst;
+
+ iosb = iosb ? iosb : &siosb;
+
+ hssocket->iostatus = __ntapi->zw_device_io_control_file(
+ hssocket->hsocket,
+ hssocket->hevent,
+ 0,
+ 0,
+ iosb,
+ NT_AFD_IOCTL_GET_PEER_NAME,
+ 0,
+ 0,
+ &sock_name_info,
+ sizeof(sock_name_info));
+
+ __ntapi->sc_wait(hssocket,iosb,0);
+
+ if (!hssocket->iostatus) {
+ addr->sa_addr_in4.sa_family = hssocket->domain;
+
+ asrc = (struct __addr_memcpy *)&(sock_name_info.sa_data);
+ adst = (struct __addr_memcpy *)addr;
+
+ adst->d1 = asrc->d0;
+ adst->d2 = asrc->d1;
+ adst->d3 = asrc->d2;
+ adst->d4 = asrc->d3;
+ adst->d5 = asrc->d4;
+ adst->d6 = asrc->d5;
+ adst->d7 = asrc->d6;
+
+ *addrlen = (uint16_t)iosb->info;
+ };
+
+ return hssocket->iostatus;
+}
diff --git a/src/socket/ntapi_sc_getpeername_v2.c b/src/socket/ntapi_sc_getpeername_v2.c
new file mode 100644
index 0000000..58c3662
--- /dev/null
+++ b/src/socket/ntapi_sc_getpeername_v2.c
@@ -0,0 +1,42 @@
+/********************************************************/
+/* ntapi: Native API core library */
+/* Copyright (C) 2013--2016 Z. Gilboa */
+/* Released under GPLv2 and GPLv3; see COPYING.NTAPI. */
+/********************************************************/
+
+#include <psxtypes/psxtypes.h>
+#include <ntapi/nt_object.h>
+#include <ntapi/nt_file.h>
+#include <ntapi/nt_socket.h>
+#include <ntapi/ntapi.h>
+#include "ntapi_impl.h"
+
+int32_t __cdecl __ntapi_sc_getpeername_v2(
+ __in nt_socket * hssocket,
+ __in nt_sockaddr * addr,
+ __in uint16_t * addrlen,
+ __out nt_io_status_block * iosb __optional)
+{
+ nt_iosb siosb;
+
+ iosb = iosb ? iosb : &siosb;
+
+ hssocket->iostatus = __ntapi->zw_device_io_control_file(
+ hssocket->hsocket,
+ hssocket->hevent,
+ 0,
+ 0,
+ iosb,
+ NT_AFD_IOCTL_GET_PEER_NAME,
+ 0,
+ 0,
+ addr,
+ sizeof(*addr));
+
+ __ntapi->sc_wait(hssocket,iosb,0);
+
+ if (!hssocket->iostatus)
+ *addrlen = (uint16_t)iosb->info;
+
+ return hssocket->iostatus;
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-25 01:11:29 +0000