commit 4f405b228d94c79cebc506053c7d892b7e18a7fa
author Misha Koshelev <mk144210@bcm.edu> Mon Aug 13 00:24:57 2007 -0500
committer Alexandre Julliard <julliard@winehq.org> Tue Aug 14 12:25:26 2007 +0200
tree dc5cda950339d554daab25cfdeba8dbf840478cb
parent 70edf2d2234159f021465f52de52cb4cfcf6e1ab

urlmon: Fix timing issue with FLAG_REQUEST_COMPLETE and InternetQueryDataAvailable in HttpProtocol_Continue.

dlls/urlmon/http.c

diff --git a/dlls/urlmon/http.c b/dlls/urlmon/http.c
index 08d6419..2d1929d 100644
--- a/dlls/urlmon/http.c
+++ b/dlls/urlmon/http.c
@@ -661,20 +661,22 @@
 
     if (pProtocolData->pData >= (LPVOID)BINDSTATUS_DOWNLOADINGDATA)
     {
+        /* InternetQueryDataAvailable may immediately fork and perform its asynchronous
+         * read, so clear the flag _before_ calling so it does not incorrectly get cleared
+         * after the status callback is called */
+        This->flags &= ~FLAG_REQUEST_COMPLETE;
         if (!InternetQueryDataAvailable(This->request, &This->available_bytes, 0, 0))
         {
-            if (GetLastError() == ERROR_IO_PENDING)
+            if (GetLastError() != ERROR_IO_PENDING)
             {
-                This->flags &= ~FLAG_REQUEST_COMPLETE;
-            }
-            else
-            {
+                This->flags |= FLAG_REQUEST_COMPLETE;
                 WARN("InternetQueryDataAvailable failed: %d\n", GetLastError());
                 HTTPPROTOCOL_ReportResult(This, INET_E_DATA_NOT_AVAILABLE);
             }
         }
         else
         {
+            This->flags |= FLAG_REQUEST_COMPLETE;
             HTTPPROTOCOL_ReportData(This);
         }
     }