Paul Gofman : netio.sys: Add WSK provider stub interface.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 10331eb2b76c1b0220ac93a1be1d78771495c2fa
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=10331eb2b76c1b0220ac93a1b...

Author: Paul Gofman <pgofman(a)codeweavers.com>
Date:   Thu Jun 11 17:47:46 2020 +0300

netio.sys: Add WSK provider stub interface.

Signed-off-by: Paul Gofman <pgofman(a)codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard(a)winehq.org>

---

 dlls/netio.sys/netio.c            | 118 ++++++++++++++++++++++++++++++++++++--
 dlls/ntoskrnl.exe/tests/driver4.c |   8 ++-
 2 files changed, 118 insertions(+), 8 deletions(-)

diff --git a/dlls/netio.sys/netio.c b/dlls/netio.sys/netio.c
index 839a06d30d..fac91736ab 100644
--- a/dlls/netio.sys/netio.c
+++ b/dlls/netio.sys/netio.c
@@ -32,32 +32,138 @@
 #include "ddk/wsk.h"
 #include "wine/debug.h"
 
+#include "wine/heap.h"
+
 WINE_DEFAULT_DEBUG_CHANNEL(netio);
 
+struct _WSK_CLIENT
+{
+    WSK_REGISTRATION *registration;
+    WSK_CLIENT_NPI *client_npi;
+};
+
+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,
+        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,
+            owning_thread, security_descriptor, irp);
+
+    return STATUS_NOT_IMPLEMENTED;
+}
+
+static NTSTATUS WINAPI wsk_socket_connect(WSK_CLIENT *client, USHORT socket_type, ULONG protocol,
+        SOCKADDR *local_address, SOCKADDR *remote_address, ULONG flags, void *socket_context,
+        const WSK_CLIENT_CONNECTION_DISPATCH *dispatch, PEPROCESS owning_process, PETHREAD owning_thread,
+        SECURITY_DESCRIPTOR *security_descriptor, IRP *irp)
+{
+    FIXME("client %p, socket_type %#x, protocol %#x, local_address %p, remote_address %p, "
+            "flags %#x, socket_context %p, dispatch %p, owning_process %p, owning_thread %p, "
+            "security_descriptor %p, irp %p stub.\n",
+            client, socket_type, protocol, local_address, remote_address, flags, socket_context,
+            dispatch, owning_process, owning_thread, security_descriptor, irp);
+
+    return STATUS_NOT_IMPLEMENTED;
+}
+
+static NTSTATUS WINAPI wsk_control_client(WSK_CLIENT *client, ULONG control_code, SIZE_T input_size,
+        void *input_buffer, SIZE_T output_size, void *output_buffer, SIZE_T *output_size_returned,
+        IRP *irp
+)
+{
+    FIXME("client %p, control_code %#x, input_size %lu, input_buffer %p, output_size %lu, "
+            "output_buffer %p, output_size_returned %p, irp %p, stub.\n",
+            client, control_code, input_size, input_buffer, output_size, output_buffer,
+            output_size_returned, irp);
+
+    return STATUS_NOT_IMPLEMENTED;
+}
+
+static NTSTATUS WINAPI wsk_get_address_info(WSK_CLIENT *client, UNICODE_STRING *node_name,
+        UNICODE_STRING *service_name, ULONG name_space, GUID *provider, ADDRINFOEXW *hints,
+        ADDRINFOEXW **result, PEPROCESS owning_process, PETHREAD owning_thread, IRP *irp)
+{
+    FIXME("client %p, node_name %p, service_name %p, name_space %#x, provider %p, hints %p, "
+            "result %p, owning_process %p, owning_thread %p, irp %p stub.\n",
+            client, node_name, service_name, name_space, provider, hints, result,
+            owning_process, owning_thread, irp);
+
+    return STATUS_NOT_IMPLEMENTED;
+}
+
+static void WINAPI wsk_free_address_info(WSK_CLIENT *client, ADDRINFOEXW *addr_info)
+{
+    FIXME("client %p, addr_info %p stub.\n", client, addr_info);
+}
+
+static NTSTATUS WINAPI wsk_get_name_info(WSK_CLIENT *client, SOCKADDR *sock_addr, ULONG sock_addr_length,
+        UNICODE_STRING *node_name, UNICODE_STRING *service_name, ULONG flags, PEPROCESS owning_process,
+        PETHREAD owning_thread, IRP *irp)
+{
+    FIXME("client %p, sock_addr %p, sock_addr_length %u, node_name %p, service_name %p, "
+            "flags %#x, owning_process %p, owning_thread %p, irp %p stub.\n",
+            client, sock_addr, sock_addr_length, node_name, service_name, flags,
+            owning_process, owning_thread, irp);
+
+    return STATUS_NOT_IMPLEMENTED;
+}
+
+static const WSK_PROVIDER_DISPATCH wsk_dispatch =
+{
+    MAKE_WSK_VERSION(1, 0), 0,
+    wsk_socket,
+    wsk_socket_connect,
+    wsk_control_client,
+    wsk_get_address_info,
+    wsk_free_address_info,
+    wsk_get_name_info,
+};
+
 NTSTATUS WINAPI WskCaptureProviderNPI(WSK_REGISTRATION *wsk_registration, ULONG wait_timeout,
         WSK_PROVIDER_NPI *wsk_provider_npi)
 {
-    FIXME("wsk_registration %p, wait_timeout %u, wsk_provider_npi %p stub.\n",
+    WSK_CLIENT *client = wsk_registration->ReservedRegistrationContext;
+
+    TRACE("wsk_registration %p, wait_timeout %u, wsk_provider_npi %p.\n",
             wsk_registration, wait_timeout, wsk_provider_npi);
 
-    return STATUS_NOT_IMPLEMENTED;
+    wsk_provider_npi->Client = client;
+    wsk_provider_npi->Dispatch = &wsk_dispatch;
+    return STATUS_SUCCESS;
 }
 
 void WINAPI WskReleaseProviderNPI(WSK_REGISTRATION *wsk_registration)
 {
-    FIXME("wsk_registration %p stub.\n", wsk_registration);
+    TRACE("wsk_registration %p.\n", wsk_registration);
+
 }
 
 NTSTATUS WINAPI WskRegister(WSK_CLIENT_NPI *wsk_client_npi, WSK_REGISTRATION *wsk_registration)
 {
-    FIXME("wsk_client_npi %p, wsk_registration %p stub.\n", wsk_client_npi, wsk_registration);
+    WSK_CLIENT *client;
 
-    return STATUS_NOT_IMPLEMENTED;
+    TRACE("wsk_client_npi %p, wsk_registration %p.\n", wsk_client_npi, wsk_registration);
+
+    if (!(client = heap_alloc(sizeof(*client))))
+    {
+        ERR("No memory.\n");
+        return STATUS_NO_MEMORY;
+    }
+
+    client->registration = wsk_registration;
+    client->client_npi = wsk_client_npi;
+    wsk_registration->ReservedRegistrationContext = client;
+
+    return STATUS_SUCCESS;
 }
 
 void WINAPI WskDeregister(WSK_REGISTRATION *wsk_registration)
 {
-    FIXME("wsk_registration %p stub.\n", wsk_registration);
+    TRACE("wsk_registration %p.\n", wsk_registration);
+
+    heap_free(wsk_registration->ReservedRegistrationContext);
 }
 
 static void WINAPI driver_unload(DRIVER_OBJECT *driver)
diff --git a/dlls/ntoskrnl.exe/tests/driver4.c b/dlls/ntoskrnl.exe/tests/driver4.c
index 2fe6072ec5..d863d74836 100644
--- a/dlls/ntoskrnl.exe/tests/driver4.c
+++ b/dlls/ntoskrnl.exe/tests/driver4.c
@@ -66,10 +66,14 @@ static void netio_init(void)
 
     client_npi.Dispatch = &client_dispatch;
     status = WskRegister(&client_npi, ®istration);
-    todo_wine ok(status == STATUS_SUCCESS, "Got unexpected status %#x.\n", status);
+    ok(status == STATUS_SUCCESS, "Got unexpected status %#x.\n", status);
 
     status = WskCaptureProviderNPI(®istration, WSK_INFINITE_WAIT, &provider_npi);
-    todo_wine ok(status == STATUS_SUCCESS, "Got unexpected status %#x.\n", status);
+    ok(status == STATUS_SUCCESS, "Got unexpected status %#x.\n", status);
+
+    ok(provider_npi.Dispatch->Version >= MAKE_WSK_VERSION(1, 0), "Got unexpected version %#x.\n",
+            provider_npi.Dispatch->Version);
+    ok(!!provider_npi.Client, "Got null WSK_CLIENT.\n");
 }
 
 static void netio_uninit(void)