Jacek Caban : urlmon: Return MK_S_ASYNCHRONOUS if bindindg is asynchronous.

[Alexandre Julliard]

Module: wine
Branch: refs/heads/master
Commit: cef354a2165d74a422a6baee2e41035cd67762b9
URL:    http://source.winehq.org/git/?p=wine.git;a=commit;h=cef354a2165d74a422a6baee...

Author: Jacek Caban <jacek(a)codeweavers.com>
Date:   Thu May 25 18:34:15 2006 +0200

urlmon: Return MK_S_ASYNCHRONOUS if bindindg is asynchronous.

---

 dlls/urlmon/binding.c |   33 +++++++++++++++++++++++++--------
 1 files changed, 25 insertions(+), 8 deletions(-)

diff --git a/dlls/urlmon/binding.c b/dlls/urlmon/binding.c
index 04d2f7f..2a59b80 100644
--- a/dlls/urlmon/binding.c
+++ b/dlls/urlmon/binding.c
@@ -64,6 +64,7 @@ struct ProtocolStream {
 
     BYTE buf[1024*8];
     DWORD buf_size;
+    BOOL init_buf;
 };
 
 #define BINDING(x)   ((IBinding*)               &(x)->lpBindingVtbl)
@@ -347,6 +348,7 @@ static ProtocolStream *create_stream(IIn
     ret->lpStreamVtbl = &ProtocolStreamVtbl;
     ret->ref = 1;
     ret->buf_size = 0;
+    ret->init_buf = FALSE;
 
     IInternetProtocol_AddRef(protocol);
     ret->protocol = protocol;
@@ -359,10 +361,14 @@ static ProtocolStream *create_stream(IIn
 static void fill_stream_buffer(ProtocolStream *This)
 {
     DWORD read = 0;
+    HRESULT hres;
 
-    IInternetProtocol_Read(This->protocol, This->buf+This->buf_size,
-                           sizeof(This->buf)-This->buf_size, &read);
-    This->buf_size += read;
+    hres = IInternetProtocol_Read(This->protocol, This->buf+This->buf_size,
+                                  sizeof(This->buf)-This->buf_size, &read);
+    if(SUCCEEDED(hres)) {
+        This->buf_size += read;
+        This->init_buf = TRUE;
+    }
 }
 
 static HRESULT WINAPI Binding_QueryInterface(IBinding *iface, REFIID riid, void **ppv)
@@ -908,15 +914,26 @@ HRESULT start_binding(LPCWSTR url, IBind
     hres = IInternetProtocol_Start(binding->protocol, url, PROTSINK(binding),
              BINDINF(binding), 0, 0);
 
-    if(SUCCEEDED(hres)) {
-        IInternetProtocol_UnlockRequest(binding->protocol);
-    }else {
+    if(FAILED(hres)) {
         WARN("Start failed: %08lx\n", hres);
+
+        IInternetProtocol_Terminate(binding->protocol, 0);
         IBindStatusCallback_OnStopBinding(binding->callback, S_OK, NULL);
+        IBinding_Release(BINDING(binding));
+
+        return hres;
     }
 
-    IStream_AddRef(STREAM(binding->stream));
-    *ppv = binding->stream;
+    if(binding->stream->init_buf) {
+        IInternetProtocol_UnlockRequest(binding->protocol);
+
+        IStream_AddRef(STREAM(binding->stream));
+        *ppv = binding->stream;
+
+        hres = S_OK;
+    }else {
+        hres = MK_S_ASYNCHRONOUS;
+    }
 
     IBinding_Release(BINDING(binding));