[PATCH 6/8] netio.sys: Implement wsk_socket() function for listen sockets.
Paul Gofman
pgofman at codeweavers.com
Sun Jun 14 13:58:53 CDT 2020
Signed-off-by: Paul Gofman <pgofman at codeweavers.com>
---
dlls/netio.sys/netio.c | 133 +++++++++++++++++++++++++++++++++++++++--
include/ddk/wsk.h | 14 +++++
2 files changed, 142 insertions(+), 5 deletions(-)
diff --git a/dlls/netio.sys/netio.c b/dlls/netio.sys/netio.c
index 00dbec76642..3ac2008c7e3 100644
--- a/dlls/netio.sys/netio.c
+++ b/dlls/netio.sys/netio.c
@@ -44,6 +44,14 @@ struct _WSK_CLIENT
WSK_CLIENT_NPI *client_npi;
};
+struct wsk_socket_internal
+{
+ WSK_SOCKET wsk_socket;
+ SOCKET s;
+ const void *client_dispatch;
+ void *client_context;
+};
+
static NTSTATUS sock_error_to_ntstatus(DWORD err)
{
switch (err)
@@ -92,16 +100,131 @@ static void dispatch_irp(IRP *irp, NTSTATUS status)
IoCompleteRequest(irp, IO_NO_INCREMENT);
}
+static NTSTATUS WINAPI wsk_control_socket(WSK_SOCKET *socket, WSK_CONTROL_SOCKET_TYPE request_type,
+ ULONG control_code, ULONG level, SIZE_T input_size, void *input_buffer, SIZE_T output_size,
+ void *output_buffer, SIZE_T *output_size_returned, IRP *irp)
+{
+ FIXME("socket %p, request_type %u, control_code %#x, level %u, input_size %lu, input_buffer %p, "
+ "output_size %lu, output_buffer %p, output_size_returned %p, irp %p stub.\n",
+ socket, request_type, control_code, level, input_size, input_buffer, output_size,
+ output_buffer, output_size_returned, irp);
+
+ return STATUS_NOT_IMPLEMENTED;
+}
+
+static NTSTATUS WINAPI wsk_close_socket(WSK_SOCKET *socket, IRP *irp)
+{
+ FIXME("socket %p, irp %p stub.\n", socket, irp);
+
+ return STATUS_NOT_IMPLEMENTED;
+}
+
+static NTSTATUS WINAPI wsk_bind(WSK_SOCKET *socket, SOCKADDR *local_address, ULONG flags, IRP *irp)
+{
+ FIXME("socket %p, local_address %p, flags %#x, irp %p stub.\n",
+ socket, local_address, flags, irp);
+
+ return STATUS_NOT_IMPLEMENTED;
+}
+
+static NTSTATUS WINAPI wsk_accept(WSK_SOCKET *listen_socket, ULONG flags, void *accept_socket_context,
+ const WSK_CLIENT_CONNECTION_DISPATCH *accept_socket_dispatch, SOCKADDR *local_address,
+ SOCKADDR *remote_address, IRP *irp)
+{
+ FIXME("listen_socket %p, flags %#x, accept_socket_context %p, accept_socket_dispatch %p, "
+ "local_address %p, remote_address %p, irp %p stub.\n",
+ listen_socket, flags, accept_socket_context, accept_socket_dispatch, local_address,
+ remote_address, irp);
+
+ return STATUS_NOT_IMPLEMENTED;
+}
+
+static NTSTATUS WINAPI wsk_inspect_complete(WSK_SOCKET *listen_socket, WSK_INSPECT_ID *inspect_id,
+ WSK_INSPECT_ACTION action, IRP *irp)
+{
+ FIXME("listen_socket %p, inspect_id %p, action %u, irp %p stub.\n",
+ listen_socket, inspect_id, action, irp);
+
+ return STATUS_NOT_IMPLEMENTED;
+}
+
+static NTSTATUS WINAPI wsk_get_local_address(WSK_SOCKET *socket, SOCKADDR *local_address, IRP *irp)
+{
+ FIXME("socket %p, local_address %p, irp %p stub.\n", socket, local_address, irp);
+
+ return STATUS_NOT_IMPLEMENTED;
+}
+
+static const WSK_PROVIDER_LISTEN_DISPATCH wsk_provider_listen_dispatch =
+{
+ {
+ wsk_control_socket,
+ wsk_close_socket,
+ },
+ wsk_bind,
+ wsk_accept,
+ wsk_inspect_complete,
+ wsk_get_local_address,
+};
+
static NTSTATUS WINAPI wsk_socket(WSK_CLIENT *client, ADDRESS_FAMILY address_family, USHORT socket_type,
- ULONG protocol, ULONG Flags, void *socket_context, const void *dispatch, PEPROCESS owning_process,
+ ULONG protocol, ULONG flags, void *socket_context, const void *dispatch, PEPROCESS owning_process,
PETHREAD owning_thread, SECURITY_DESCRIPTOR *security_descriptor, IRP *irp)
{
- FIXME("client %p, address_family %#x, socket_type %#x, protocol %#x, Flags %#x, socket_context %p, dispatch %p, "
- "owning_process %p, owning_thread %p, security_descriptor %p, irp %p stub.\n",
- client, address_family, socket_type, protocol, Flags, socket_context, dispatch, owning_process,
+ struct wsk_socket_internal *socket;
+ NTSTATUS status;
+ SOCKET s;
+
+ TRACE("client %p, address_family %#x, socket_type %#x, protocol %#x, flags %#x, socket_context %p, dispatch %p, "
+ "owning_process %p, owning_thread %p, security_descriptor %p, irp %p.\n",
+ client, address_family, socket_type, protocol, flags, socket_context, dispatch, owning_process,
owning_thread, security_descriptor, irp);
- return STATUS_NOT_IMPLEMENTED;
+ if (!irp)
+ return STATUS_INVALID_PARAMETER;
+
+ if (!client)
+ return STATUS_INVALID_HANDLE;
+
+ irp->IoStatus.Information = 0;
+
+ if ((s = WSASocketW(address_family, socket_type, protocol, NULL, 0, 0)) == INVALID_SOCKET)
+ {
+ status = sock_error_to_ntstatus(WSAGetLastError());
+ goto done;
+ }
+
+ if (!(socket = heap_alloc(sizeof(*socket))))
+ {
+ status = STATUS_NO_MEMORY;
+ closesocket(s);
+ goto done;
+ }
+
+ socket->s = s;
+ socket->client_dispatch = dispatch;
+ socket->client_context = socket_context;
+
+ switch (flags)
+ {
+ case WSK_FLAG_LISTEN_SOCKET:
+ socket->wsk_socket.Dispatch = &wsk_provider_listen_dispatch;
+ break;
+
+ default:
+ FIXME("Flags %#x not implemented.\n", flags);
+ closesocket(s);
+ heap_free(socket);
+ status = STATUS_NOT_IMPLEMENTED;
+ goto done;
+ }
+
+ irp->IoStatus.Information = (ULONG_PTR)&socket->wsk_socket;
+ status = STATUS_SUCCESS;
+
+done:
+ dispatch_irp(irp, status);
+ return status ? status : STATUS_PENDING;
}
static NTSTATUS WINAPI wsk_socket_connect(WSK_CLIENT *client, USHORT socket_type, ULONG protocol,
diff --git a/include/ddk/wsk.h b/include/ddk/wsk.h
index 97ab8ab3a51..3e0debb1c0f 100644
--- a/include/ddk/wsk.h
+++ b/include/ddk/wsk.h
@@ -136,6 +136,20 @@ typedef struct _WSK_CLIENT_CONNECTION_DISPATCH
PFN_WSK_SEND_BACKLOG_EVENT WskSendBacklogEvent;
} WSK_CLIENT_CONNECTION_DISPATCH, *PWSK_CLIENT_CONNECTION_DISPATCH;
+typedef NTSTATUS (WINAPI *PFN_WSK_ACCEPT_EVENT)(void *socket_context, ULONG flags, SOCKADDR *local_address,
+ SOCKADDR *remote_address, WSK_SOCKET *accept_socket, void *accept_socket_context,
+ const WSK_CLIENT_CONNECTION_DISPATCH **accept_socket_dispatch);
+typedef WSK_INSPECT_ACTION (WINAPI *PFN_WSK_INSPECT_EVENT)(void *socket_context, SOCKADDR *local_address,
+ SOCKADDR *remote_address, WSK_INSPECT_ID *inspect_id);
+typedef NTSTATUS (WINAPI *PFN_WSK_ABORT_EVENT)(void *socket_context, WSK_INSPECT_ID *inspect_id);
+
+typedef struct _WSK_CLIENT_LISTEN_DISPATCH
+{
+ PFN_WSK_ACCEPT_EVENT WskAcceptEvent;
+ PFN_WSK_INSPECT_EVENT WskInspectEvent;
+ PFN_WSK_ABORT_EVENT WskAbortEvent;
+} WSK_CLIENT_LISTEN_DISPATCH, *PWSK_CLIENT_LISTEN_DISPATCH;
+
typedef struct _WSK_CLIENT_DISPATCH
{
USHORT Version;
--
2.26.2
More information about the wine-devel
mailing list