dlls/urlmon/umstream.c

   1 /*
   2  * Based on ../shell32/memorystream.c
   3  *
   4  * Copyright 1999 Juergen Schmied
   5  * Copyright 2003 Mike McCormack for CodeWeavers
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include "urlmon_main.h"
  23
  24 #include "winreg.h"
  25 #include "winternl.h"
  26 #include "wininet.h"
  27 #include "shlwapi.h"
  28
  29 #include "wine/debug.h"
  30
  31 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
  32
  33 typedef struct ProxyBindStatusCallback
  34 {
  35     const IBindStatusCallbackVtbl *lpVtbl;
  36
  37     IBindStatusCallback *pBSC;
  38 } ProxyBindStatusCallback;
  39
  40 static HRESULT WINAPI ProxyBindStatusCallback_QueryInterface(IBindStatusCallback *iface, REFIID riid, void **ppv)
  41 {
  42     if (IsEqualGUID(&IID_IBindStatusCallback, riid) ||
  43         IsEqualGUID(&IID_IUnknown, riid))
  44     {
  45         *ppv = iface;
  46         IUnknown_AddRef(iface);
  47         return S_OK;
  48     }
  49
  50     *ppv = NULL;
  51     return E_NOINTERFACE;
  52 }
  53
  54 static ULONG WINAPI ProxyBindStatusCallback_AddRef(IBindStatusCallback *iface)
  55 {
  56     return 2;
  57 }
  58
  59 static ULONG WINAPI ProxyBindStatusCallback_Release(IBindStatusCallback *iface)
  60 {
  61     return 1;
  62 }
  63
  64 static HRESULT WINAPI ProxyBindStatusCallback_OnStartBinding(IBindStatusCallback *iface, DWORD dwReserved,
  65                                                IBinding *pib)
  66 {
  67     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
  68
  69     if(This->pBSC)
  70         return IBindStatusCallback_OnStartBinding(This->pBSC, dwReserved, pib);
  71
  72     return S_OK;
  73 }
  74
  75 static HRESULT WINAPI ProxyBindStatusCallback_GetPriority(IBindStatusCallback *iface, LONG *pnPriority)
  76 {
  77     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
  78
  79     if(This->pBSC)
  80         return IBindStatusCallback_GetPriority(This->pBSC, pnPriority);
  81
  82     return S_OK;
  83 }
  84
  85 static HRESULT WINAPI ProxyBindStatusCallback_OnLowResource(IBindStatusCallback *iface, DWORD reserved)
  86 {
  87     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
  88
  89     if(This->pBSC)
  90         return IBindStatusCallback_OnLowResource(This->pBSC, reserved);
  91
  92     return S_OK;
  93 }
  94
  95 static HRESULT WINAPI ProxyBindStatusCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
  96                                            ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
  97 {
  98     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
  99
 100     if(This->pBSC)
 101         return IBindStatusCallback_OnProgress(This->pBSC, ulProgress,
 102                                           ulProgressMax, ulStatusCode,
 103                                           szStatusText);
 104
 105     return S_OK;
 106 }
 107
 108 static HRESULT WINAPI ProxyBindStatusCallback_OnStopBinding(IBindStatusCallback *iface, HRESULT hresult, LPCWSTR szError)
 109 {
 110     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
 111
 112     if(This->pBSC)
 113         return IBindStatusCallback_OnStopBinding(This->pBSC, hresult, szError);
 114
 115     return S_OK;
 116 }
 117
 118 static HRESULT WINAPI ProxyBindStatusCallback_GetBindInfo(IBindStatusCallback *iface, DWORD *grfBINDF, BINDINFO *pbindinfo)
 119 {
 120     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
 121
 122     if(This->pBSC)
 123         return IBindStatusCallback_GetBindInfo(This->pBSC, grfBINDF, pbindinfo);
 124
 125     return E_INVALIDARG;
 126 }
 127
 128 static HRESULT WINAPI ProxyBindStatusCallback_OnDataAvailable(IBindStatusCallback *iface, DWORD grfBSCF,
 129                                                               DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
 130 {
 131     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
 132
 133     if(This->pBSC)
 134         return IBindStatusCallback_OnDataAvailable(This->pBSC, grfBSCF, dwSize,
 135                                                pformatetc, pstgmed);
 136
 137     return S_OK;
 138 }
 139
 140 static HRESULT WINAPI ProxyBindStatusCallback_OnObjectAvailable(IBindStatusCallback *iface, REFIID riid, IUnknown *punk)
 141 {
 142     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
 143
 144     if(This->pBSC)
 145         return IBindStatusCallback_OnObjectAvailable(This->pBSC, riid, punk);
 146
 147     return S_OK;
 148 }
 149
 150 static HRESULT WINAPI BlockingBindStatusCallback_OnDataAvailable(IBindStatusCallback *iface, DWORD grfBSCF,
 151                                                                  DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
 152 {
 153     return S_OK;
 154 }
 155
 156 static const IBindStatusCallbackVtbl BlockingBindStatusCallbackVtbl =
 157 {
 158     ProxyBindStatusCallback_QueryInterface,
 159     ProxyBindStatusCallback_AddRef,
 160     ProxyBindStatusCallback_Release,
 161     ProxyBindStatusCallback_OnStartBinding,
 162     ProxyBindStatusCallback_GetPriority,
 163     ProxyBindStatusCallback_OnLowResource,
 164     ProxyBindStatusCallback_OnProgress,
 165     ProxyBindStatusCallback_OnStopBinding,
 166     ProxyBindStatusCallback_GetBindInfo,
 167     BlockingBindStatusCallback_OnDataAvailable,
 168     ProxyBindStatusCallback_OnObjectAvailable
 169 };
 170
 171 static HRESULT WINAPI AsyncBindStatusCallback_GetBindInfo(IBindStatusCallback *iface, DWORD *grfBINDF, BINDINFO *pbindinfo)
 172 {
 173     ProxyBindStatusCallback *This = (ProxyBindStatusCallback *)iface;
 174     HRESULT hr = IBindStatusCallback_GetBindInfo(This->pBSC, grfBINDF, pbindinfo);
 175     *grfBINDF |= BINDF_PULLDATA | BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE;
 176     return hr;
 177 }
 178
 179 static const IBindStatusCallbackVtbl AsyncBindStatusCallbackVtbl =
 180 {
 181     ProxyBindStatusCallback_QueryInterface,
 182     ProxyBindStatusCallback_AddRef,
 183     ProxyBindStatusCallback_Release,
 184     ProxyBindStatusCallback_OnStartBinding,
 185     ProxyBindStatusCallback_GetPriority,
 186     ProxyBindStatusCallback_OnLowResource,
 187     ProxyBindStatusCallback_OnProgress,
 188     ProxyBindStatusCallback_OnStopBinding,
 189     AsyncBindStatusCallback_GetBindInfo,
 190     ProxyBindStatusCallback_OnDataAvailable,
 191     ProxyBindStatusCallback_OnObjectAvailable
 192 };
 193
 194 static HRESULT URLStartDownload(LPCWSTR szURL, LPSTREAM *ppStream, IBindStatusCallback *pBSC)
 195 {
 196     HRESULT hr;
 197     IMoniker *pMoniker;
 198     IBindCtx *pbc;
 199
 200     *ppStream = NULL;
 201
 202     hr = CreateURLMoniker(NULL, szURL, &pMoniker);
 203     if (FAILED(hr))
 204         return hr;
 205
 206     hr = CreateBindCtx(0, &pbc);
 207     if (FAILED(hr))
 208     {
 209         IMoniker_Release(pMoniker);
 210         return hr;
 211     }
 212
 213     hr = RegisterBindStatusCallback(pbc, pBSC, NULL, 0);
 214     if (FAILED(hr))
 215     {
 216         IBindCtx_Release(pbc);
 217         IMoniker_Release(pMoniker);
 218         return hr;
 219     }
 220
 221     hr = IMoniker_BindToStorage(pMoniker, pbc, NULL, &IID_IStream, (void **)ppStream);
 222
 223     /* BindToStorage returning E_PENDING because it's asynchronous is not an error */
 224     if (hr == E_PENDING) hr = S_OK;
 225
 226     IBindCtx_Release(pbc);
 227     IMoniker_Release(pMoniker);
 228
 229     return hr;
 230 }
 231
 232 /***********************************************************************
 233  *              URLOpenBlockingStreamA (URLMON.@)
 234  */
 235 HRESULT WINAPI URLOpenBlockingStreamA(LPUNKNOWN pCaller, LPCSTR szURL,
 236                                       LPSTREAM *ppStream, DWORD dwReserved,
 237                                       LPBINDSTATUSCALLBACK lpfnCB)
 238 {
 239     LPWSTR szURLW;
 240     int len;
 241     HRESULT hr;
 242
 243     TRACE("(%p, %s, %p, 0x%x, %p)\n", pCaller, szURL, ppStream, dwReserved, lpfnCB);
 244
 245     if (!szURL || !ppStream)
 246         return E_INVALIDARG;
 247
 248     len = MultiByteToWideChar(CP_ACP, 0, szURL, -1, NULL, 0);
 249     szURLW = heap_alloc(len * sizeof(WCHAR));
 250     if (!szURLW)
 251     {
 252         *ppStream = NULL;
 253         return E_OUTOFMEMORY;
 254     }
 255     MultiByteToWideChar(CP_ACP, 0, szURL, -1, szURLW, len);
 256
 257     hr = URLOpenBlockingStreamW(pCaller, szURLW, ppStream, dwReserved, lpfnCB);
 258
 259     heap_free(szURLW);
 260
 261     return hr;
 262 }
 263
 264 /***********************************************************************
 265  *              URLOpenBlockingStreamW (URLMON.@)
 266  */
 267 HRESULT WINAPI URLOpenBlockingStreamW(LPUNKNOWN pCaller, LPCWSTR szURL,
 268                                       LPSTREAM *ppStream, DWORD dwReserved,
 269                                       LPBINDSTATUSCALLBACK lpfnCB)
 270 {
 271     ProxyBindStatusCallback blocking_bsc;
 272
 273     TRACE("(%p, %s, %p, 0x%x, %p)\n", pCaller, debugstr_w(szURL), ppStream,
 274           dwReserved, lpfnCB);
 275
 276     if (!szURL || !ppStream)
 277         return E_INVALIDARG;
 278
 279     blocking_bsc.lpVtbl = &BlockingBindStatusCallbackVtbl;
 280     blocking_bsc.pBSC = lpfnCB;
 281
 282     return URLStartDownload(szURL, ppStream, (IBindStatusCallback *)&blocking_bsc);
 283 }
 284
 285 /***********************************************************************
 286  *              URLOpenStreamA (URLMON.@)
 287  */
 288 HRESULT WINAPI URLOpenStreamA(LPUNKNOWN pCaller, LPCSTR szURL, DWORD dwReserved,
 289                               LPBINDSTATUSCALLBACK lpfnCB)
 290 {
 291     LPWSTR szURLW;
 292     int len;
 293     HRESULT hr;
 294
 295     TRACE("(%p, %s, 0x%x, %p)\n", pCaller, szURL, dwReserved, lpfnCB);
 296
 297     if (!szURL)
 298         return E_INVALIDARG;
 299
 300     len = MultiByteToWideChar(CP_ACP, 0, szURL, -1, NULL, 0);
 301     szURLW = heap_alloc(len * sizeof(WCHAR));
 302     if (!szURLW)
 303         return E_OUTOFMEMORY;
 304     MultiByteToWideChar(CP_ACP, 0, szURL, -1, szURLW, len);
 305
 306     hr = URLOpenStreamW(pCaller, szURLW, dwReserved, lpfnCB);
 307
 308     heap_free(szURLW);
 309
 310     return hr;
 311 }
 312
 313 /***********************************************************************
 314  *              URLOpenStreamW (URLMON.@)
 315  */
 316 HRESULT WINAPI URLOpenStreamW(LPUNKNOWN pCaller, LPCWSTR szURL, DWORD dwReserved,
 317                               LPBINDSTATUSCALLBACK lpfnCB)
 318 {
 319     HRESULT hr;
 320     ProxyBindStatusCallback async_bsc;
 321     IStream *pStream;
 322
 323     TRACE("(%p, %s, 0x%x, %p)\n", pCaller, debugstr_w(szURL), dwReserved,
 324           lpfnCB);
 325
 326     if (!szURL)
 327         return E_INVALIDARG;
 328
 329     async_bsc.lpVtbl = &AsyncBindStatusCallbackVtbl;
 330     async_bsc.pBSC = lpfnCB;
 331
 332     hr = URLStartDownload(szURL, &pStream, (IBindStatusCallback *)&async_bsc);
 333     if (SUCCEEDED(hr) && pStream)
 334         IStream_Release(pStream);
 335
 336     return hr;
 337 }