Mike Kaplinskiy : ws2_32/tests: Test closing listener during pending AcceptEx.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 4acc8729c456e238c90afa6d677d539082111aca
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=4acc8729c456e238c90afa6d67...

Author: Mike Kaplinskiy <mike.kaplinskiy(a)gmail.com>
Date:   Sun Jul 19 20:52:50 2009 -0400

ws2_32/tests: Test closing listener during pending AcceptEx.

---

 dlls/ws2_32/tests/sock.c |   19 +++++++++++++++++++
 1 files changed, 19 insertions(+), 0 deletions(-)

diff --git a/dlls/ws2_32/tests/sock.c b/dlls/ws2_32/tests/sock.c
index cecda0d..bd672c5 100644
--- a/dlls/ws2_32/tests/sock.c
+++ b/dlls/ws2_32/tests/sock.c
@@ -2972,6 +2972,25 @@ static void test_AcceptEx(void)
     ok(bret, "GetOverlappedResult failed, error %d\n", GetLastError());
     ok(bytesReturned == 0, "bytesReturned isn't supposed to be %d\n", bytesReturned);
 
+    acceptor = socket(AF_INET, SOCK_STREAM, 0);
+    if (acceptor == INVALID_SOCKET) {
+        skip("could not create acceptor socket, error %d\n", WSAGetLastError());
+        goto end;
+    }
+    bret = pAcceptEx(listener, acceptor, buffer, sizeof(buffer) - 2*(sizeof(struct sockaddr_in) + 16),
+        sizeof(struct sockaddr_in) + 16, sizeof(struct sockaddr_in) + 16,
+        &bytesReturned, &overlapped);
+    ok(bret == FALSE && WSAGetLastError() == ERROR_IO_PENDING, "AcceptEx returned %d + errno %d\n", bret, WSAGetLastError());
+
+    closesocket(listener);
+    listener = INVALID_SOCKET;
+
+    dwret = WaitForSingleObject(overlapped.hEvent, 1000);
+    ok(dwret == WAIT_OBJECT_0, "Waiting for accept event failed with %d + errno %d\n", dwret, GetLastError());
+
+    bret = GetOverlappedResult((HANDLE)listener, &overlapped, &bytesReturned, FALSE);
+    ok(!bret && GetLastError() == ERROR_OPERATION_ABORTED, "GetOverlappedResult failed, error %d\n", GetLastError());
+
 end:
     if (overlapped.hEvent)
         WSACloseEvent(overlapped.hEvent);