Zebediah Figura : ws2_32/tests: Add some tests for IOCTL_AFD_EVENT_SELECT.

[Alexandre Julliard]

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

Author: Zebediah Figura <z.figura12(a)gmail.com>
Date:   Mon Jun 14 00:41:51 2021 -0500

ws2_32/tests: Add some tests for IOCTL_AFD_EVENT_SELECT.

Signed-off-by: Zebediah Figura <z.figura12(a)gmail.com>
Signed-off-by: Alexandre Julliard <julliard(a)winehq.org>

---

 dlls/ws2_32/tests/afd.c | 94 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)

diff --git a/dlls/ws2_32/tests/afd.c b/dlls/ws2_32/tests/afd.c
index 87e4779b980..e08fc977ad4 100644
--- a/dlls/ws2_32/tests/afd.c
+++ b/dlls/ws2_32/tests/afd.c
@@ -1136,6 +1136,99 @@ static void test_recv(void)
     CloseHandle(event);
 }
 
+static void test_event_select(void)
+{
+    struct afd_event_select_params params;
+    WSANETWORKEVENTS events;
+    SOCKET client, server;
+    IO_STATUS_BLOCK io;
+    HANDLE event;
+    int ret;
+
+    event = CreateEventW(NULL, TRUE, FALSE, NULL);
+
+    tcp_socketpair(&client, &server);
+
+    memset(&io, 0xcc, sizeof(io));
+    ret = NtDeviceIoControlFile((HANDLE)client, NULL, NULL, NULL, &io,
+            IOCTL_AFD_EVENT_SELECT, NULL, 0, NULL, 0);
+    ok(ret == STATUS_INVALID_PARAMETER, "got %#x\n", ret);
+    ok(io.Status == STATUS_INVALID_PARAMETER, "got status %#x\n", io.Status);
+    ok(!io.Information, "got information %#Ix\n", io.Information);
+
+    params.event = 0;
+    params.mask = ~0;
+    memset(&io, 0xcc, sizeof(io));
+    ret = NtDeviceIoControlFile((HANDLE)client, NULL, NULL, NULL, &io,
+            IOCTL_AFD_EVENT_SELECT, ¶ms, sizeof(params), NULL, 0);
+    ok(ret == STATUS_INVALID_PARAMETER, "got %#x\n", ret);
+    ok(io.Status == STATUS_INVALID_PARAMETER, "got status %#x\n", io.Status);
+    ok(!io.Information, "got information %#Ix\n", io.Information);
+
+    params.event = event;
+    params.mask = ~0;
+    memset(&io, 0xcc, sizeof(io));
+    ret = NtDeviceIoControlFile((HANDLE)client, NULL, NULL, NULL, &io,
+            IOCTL_AFD_EVENT_SELECT, ¶ms, sizeof(params), NULL, 0);
+    ok(!ret, "got %#x\n", ret);
+    ok(!io.Status, "got status %#x\n", io.Status);
+    ok(!io.Information, "got information %#Ix\n", io.Information);
+
+    ret = WSAEnumNetworkEvents(client, event, &events);
+    ok(!ret, "got error %u\n", WSAGetLastError());
+    ok(events.lNetworkEvents == (FD_CONNECT | FD_WRITE), "got events %#x\n", events.lNetworkEvents);
+
+    closesocket(client);
+    closesocket(server);
+
+    tcp_socketpair(&client, &server);
+
+    params.event = event;
+    params.mask = AFD_POLL_CONNECT;
+    memset(&io, 0xcc, sizeof(io));
+    ret = NtDeviceIoControlFile((HANDLE)client, NULL, NULL, NULL, &io,
+            IOCTL_AFD_EVENT_SELECT, ¶ms, sizeof(params), NULL, 0);
+    ok(!ret, "got %#x\n", ret);
+    ok(!io.Status, "got status %#x\n", io.Status);
+    ok(!io.Information, "got information %#Ix\n", io.Information);
+
+    ret = WSAEnumNetworkEvents(client, event, &events);
+    ok(!ret, "got error %u\n", WSAGetLastError());
+    ok(events.lNetworkEvents == FD_CONNECT, "got events %#x\n", events.lNetworkEvents);
+
+    closesocket(client);
+    closesocket(server);
+
+    tcp_socketpair(&client, &server);
+
+    params.event = event;
+    params.mask = ~0;
+    memset(&io, 0xcc, sizeof(io));
+    ret = NtDeviceIoControlFile((HANDLE)client, NULL, NULL, NULL, &io,
+            IOCTL_AFD_EVENT_SELECT, ¶ms, sizeof(params), NULL, 0);
+    ok(!ret, "got %#x\n", ret);
+    ok(!io.Status, "got status %#x\n", io.Status);
+    ok(!io.Information, "got information %#Ix\n", io.Information);
+
+    params.event = 0;
+    params.mask = 0;
+    memset(&io, 0xcc, sizeof(io));
+    ret = NtDeviceIoControlFile((HANDLE)client, NULL, NULL, NULL, &io,
+            IOCTL_AFD_EVENT_SELECT, ¶ms, sizeof(params), NULL, 0);
+    ok(!ret, "got %#x\n", ret);
+    ok(!io.Status, "got status %#x\n", io.Status);
+    ok(!io.Information, "got information %#Ix\n", io.Information);
+
+    ret = WSAEnumNetworkEvents(client, event, &events);
+    ok(!ret, "got error %u\n", WSAGetLastError());
+    ok(!events.lNetworkEvents, "got events %#x\n", events.lNetworkEvents);
+
+    closesocket(client);
+    closesocket(server);
+
+    CloseHandle(event);
+}
+
 START_TEST(afd)
 {
     WSADATA data;
@@ -1146,6 +1239,7 @@ START_TEST(afd)
     test_poll();
     test_poll_completion_port();
     test_recv();
+    test_event_select();
 
     WSACleanup();
 }