Zebediah Figura : ws2_32/tests: Test selecting for FD_ACCEPT while there is a pending AcceptEx() call.

[Alexandre Julliard]

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

Author: Zebediah Figura 
Date:   Thu Apr 28 22:02:29 2022 -0500

ws2_32/tests: Test selecting for FD_ACCEPT while there is a pending AcceptEx() call.

Signed-off-by: Zebediah Figura 
Signed-off-by: Alexandre Julliard

---

dlls/ws2_32/tests/sock.c | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/dlls/ws2_32/tests/sock.c b/dlls/ws2_32/tests/sock.c
index 3e8b1f21c1d..f17b4bb0ffa 100644
--- a/dlls/ws2_32/tests/sock.c
+++ b/dlls/ws2_32/tests/sock.c
@@ -5308,12 +5308,23 @@ static void test_accept_events(struct event_test_ctx *ctx)
 {
     const struct sockaddr_in addr = {.sin_family = AF_INET, .sin_addr.s_addr = htonl(INADDR_LOOPBACK)};
     SOCKET listener, server, client, client2;
+    GUID acceptex_guid = WSAID_ACCEPTEX;
     struct sockaddr_in destaddr;
+    OVERLAPPED overlapped = {0};
+    LPFN_ACCEPTEX pAcceptEx;
+    char buffer[32];
     int len, ret;
+    DWORD size;
+
+    overlapped.hEvent = CreateEventA(NULL, TRUE, FALSE, NULL);

listener = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
     ok(listener != -1, "failed to create socket, error %u\n", WSAGetLastError());

+    ret = WSAIoctl(listener, SIO_GET_EXTENSION_FUNCTION_POINTER, &acceptex_guid, sizeof(acceptex_guid),
+            &pAcceptEx, sizeof(pAcceptEx), &size, NULL, NULL);
+    ok(!ret, "failed to get AcceptEx, error %u\n", WSAGetLastError());
+
     select_events(ctx, listener, FD_CONNECT | FD_READ | FD_OOB | FD_ACCEPT);

ret = bind(listener, (const struct sockaddr *)&addr, sizeof(addr));
@@ -5511,7 +5522,44 @@ static void test_accept_events(struct event_test_ctx *ctx)
     closesocket(server);
     closesocket(client);

+    /* Connect while there is a pending AcceptEx(). */
+
+    select_events(ctx, listener, FD_CONNECT | FD_READ | FD_OOB | FD_ACCEPT);
+
+    server = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+    ret = pAcceptEx(listener, server, buffer, 0, 0, sizeof(buffer), NULL, &overlapped);
+    ok(!ret, "got %d\n", ret);
+    ok(WSAGetLastError() == ERROR_IO_PENDING, "got error %u\n", WSAGetLastError());
+
+    client = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+    ret = connect(client, (struct sockaddr *)&destaddr, sizeof(destaddr));
+    ok(!ret, "got error %u\n", WSAGetLastError());
+
+    ret = WaitForSingleObject(overlapped.hEvent, 200);
+    ok(!ret, "got %d\n", ret);
+    ret = GetOverlappedResult((HANDLE)listener, &overlapped, &size, FALSE);
+    ok(ret, "got error %lu\n", GetLastError());
+    ok(!size, "got size %lu\n", size);
+
+    check_events_todo(ctx, 0, 0, 0);
+
+    closesocket(server);
+    closesocket(client);
+
+    client = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+    ret = connect(client, (struct sockaddr *)&destaddr, sizeof(destaddr));
+    ok(!ret, "got error %u\n", WSAGetLastError());
+
+    check_events_todo(ctx, FD_ACCEPT, 0, 200);
+    check_events(ctx, 0, 0, 0);
+
+    server = accept(listener, NULL, NULL);
+    ok(server != -1, "failed to accept, error %u\n", WSAGetLastError());
+    closesocket(server);
+    closesocket(client);
+
     closesocket(listener);
+    CloseHandle(overlapped.hEvent);
 }

static void test_connect_events(struct event_test_ctx *ctx)