[PATCH v2 4/6] winhttp: Return error from WinHttpWebSocketQueryCloseStatus() if close frame was not received.

[Paul Gofman]

Signed-off-by: Paul Gofman 
---
     v2:
          - no changes.

dlls/winhttp/request.c            | 6 ++++++
 dlls/winhttp/tests/notification.c | 6 +++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/dlls/winhttp/request.c b/dlls/winhttp/request.c
index 39223d41bbb..279e0522166 100644
--- a/dlls/winhttp/request.c
+++ b/dlls/winhttp/request.c
@@ -3909,6 +3909,12 @@ DWORD WINAPI WinHttpWebSocketQueryCloseStatus( HINTERNET hsocket, USHORT *status
         return ERROR_INVALID_OPERATION;
     }

+    if (!socket->close_frame_received || socket->close_frame_receive_err)
+    {
+        ret = socket->close_frame_received ? socket->close_frame_receive_err : ERROR_INVALID_OPERATION;
+        release_object( &socket->hdr );
+        return ret;
+    }
     *status = socket->status;
     *ret_len = socket->reason_len;
     if (socket->reason_len > len) ret = ERROR_INSUFFICIENT_BUFFER;
diff --git a/dlls/winhttp/tests/notification.c b/dlls/winhttp/tests/notification.c
index 2db5a094f87..1abc8cf377f 100644
--- a/dlls/winhttp/tests/notification.c
+++ b/dlls/winhttp/tests/notification.c
@@ -971,9 +971,9 @@ static void test_websocket(BOOL secure)
     close_status = 0xdead;
     size = sizeof(buffer) + 1;
     err = pWinHttpWebSocketQueryCloseStatus( socket, &close_status, buffer, sizeof(buffer), &size );
-    todo_wine ok( err == ERROR_INVALID_OPERATION, "got %u\n", err );
-    todo_wine ok( close_status == 0xdead, "got %u\n", close_status );
-    todo_wine ok( size == sizeof(buffer) + 1, "got %u\n", size );
+    ok( err == ERROR_INVALID_OPERATION, "got %u\n", err );
+    ok( close_status == 0xdead, "got %u\n", close_status );
+    ok( size == sizeof(buffer) + 1, "got %u\n", size );

WaitForSingleObject( info.wait, INFINITE );