dlls/urlmon/mk.c

   1 /*
   2  * Copyright 2007 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "urlmon_main.h"
  20 #include "wine/debug.h"
  21
  22 #define NO_SHLWAPI_REG
  23 #include "shlwapi.h"
  24
  25 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
  26
  27 typedef struct {
  28     IInternetProtocolEx IInternetProtocolEx_iface;
  29
  30     LONG ref;
  31
  32     IStream *stream;
  33 } MkProtocol;
  34
  35 static inline MkProtocol *impl_from_IInternetProtocolEx(IInternetProtocolEx *iface)
  36 {
  37     return CONTAINING_RECORD(iface, MkProtocol, IInternetProtocolEx_iface);
  38 }
  39
  40 static HRESULT WINAPI MkProtocol_QueryInterface(IInternetProtocolEx *iface, REFIID riid, void **ppv)
  41 {
  42     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
  43
  44     *ppv = NULL;
  45     if(IsEqualGUID(&IID_IUnknown, riid)) {
  46         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
  47         *ppv = &This->IInternetProtocolEx_iface;
  48     }else if(IsEqualGUID(&IID_IInternetProtocolRoot, riid)) {
  49         TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This, ppv);
  50         *ppv = &This->IInternetProtocolEx_iface;
  51     }else if(IsEqualGUID(&IID_IInternetProtocol, riid)) {
  52         TRACE("(%p)->(IID_IInternetProtocol %p)\n", This, ppv);
  53         *ppv = &This->IInternetProtocolEx_iface;
  54     }else if(IsEqualGUID(&IID_IInternetProtocolEx, riid)) {
  55         TRACE("(%p)->(IID_IInternetProtocolEx %p)\n", This, ppv);
  56         *ppv = &This->IInternetProtocolEx_iface;
  57     }
  58
  59     if(*ppv) {
  60         IInternetProtocolEx_AddRef(iface);
  61         return S_OK;
  62     }
  63
  64     WARN("not supported interface %s\n", debugstr_guid(riid));
  65     return E_NOINTERFACE;
  66 }
  67
  68 static ULONG WINAPI MkProtocol_AddRef(IInternetProtocolEx *iface)
  69 {
  70     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
  71     LONG ref = InterlockedIncrement(&This->ref);
  72     TRACE("(%p) ref=%d\n", This, ref);
  73     return ref;
  74 }
  75
  76 static ULONG WINAPI MkProtocol_Release(IInternetProtocolEx *iface)
  77 {
  78     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
  79     LONG ref = InterlockedDecrement(&This->ref);
  80
  81     TRACE("(%p) ref=%d\n", This, ref);
  82
  83     if(!ref) {
  84         if(This->stream)
  85             IStream_Release(This->stream);
  86
  87         heap_free(This);
  88
  89         URLMON_UnlockModule();
  90     }
  91
  92     return ref;
  93 }
  94
  95 static HRESULT report_result(IInternetProtocolSink *sink, HRESULT hres, DWORD dwError)
  96 {
  97     IInternetProtocolSink_ReportResult(sink, hres, dwError, NULL);
  98     return hres;
  99 }
 100
 101 static HRESULT WINAPI MkProtocol_Start(IInternetProtocolEx *iface, LPCWSTR szUrl,
 102         IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
 103         DWORD grfPI, HANDLE_PTR dwReserved)
 104 {
 105     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 106     HRESULT hres;
 107     IUri *uri;
 108
 109     TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink,
 110             pOIBindInfo, grfPI, dwReserved);
 111
 112     hres = CreateUri(szUrl, 0, 0, &uri);
 113     if(FAILED(hres))
 114         return hres;
 115
 116     hres = IInternetProtocolEx_StartEx(&This->IInternetProtocolEx_iface, uri, pOIProtSink,
 117             pOIBindInfo, grfPI, (HANDLE*)dwReserved);
 118
 119     IUri_Release(uri);
 120     return hres;
 121 }
 122
 123 static HRESULT WINAPI MkProtocol_Continue(IInternetProtocolEx *iface, PROTOCOLDATA *pProtocolData)
 124 {
 125     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 126     FIXME("(%p)->(%p)\n", This, pProtocolData);
 127     return E_NOTIMPL;
 128 }
 129
 130 static HRESULT WINAPI MkProtocol_Abort(IInternetProtocolEx *iface, HRESULT hrReason,
 131         DWORD dwOptions)
 132 {
 133     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 134     FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
 135     return E_NOTIMPL;
 136 }
 137
 138 static HRESULT WINAPI MkProtocol_Terminate(IInternetProtocolEx *iface, DWORD dwOptions)
 139 {
 140     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 141
 142     TRACE("(%p)->(%08x)\n", This, dwOptions);
 143
 144     return S_OK;
 145 }
 146
 147 static HRESULT WINAPI MkProtocol_Suspend(IInternetProtocolEx *iface)
 148 {
 149     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 150     FIXME("(%p)\n", This);
 151     return E_NOTIMPL;
 152 }
 153
 154 static HRESULT WINAPI MkProtocol_Resume(IInternetProtocolEx *iface)
 155 {
 156     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 157     FIXME("(%p)\n", This);
 158     return E_NOTIMPL;
 159 }
 160
 161 static HRESULT WINAPI MkProtocol_Read(IInternetProtocolEx *iface, void *pv,
 162         ULONG cb, ULONG *pcbRead)
 163 {
 164     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 165
 166     TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
 167
 168     if(!This->stream)
 169         return E_FAIL;
 170
 171     return IStream_Read(This->stream, pv, cb, pcbRead);
 172 }
 173
 174 static HRESULT WINAPI MkProtocol_Seek(IInternetProtocolEx *iface, LARGE_INTEGER dlibMove,
 175         DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
 176 {
 177     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 178     FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
 179     return E_NOTIMPL;
 180 }
 181
 182 static HRESULT WINAPI MkProtocol_LockRequest(IInternetProtocolEx *iface, DWORD dwOptions)
 183 {
 184     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 185
 186     TRACE("(%p)->(%08x)\n", This, dwOptions);
 187
 188     return S_OK;
 189 }
 190
 191 static HRESULT WINAPI MkProtocol_UnlockRequest(IInternetProtocolEx *iface)
 192 {
 193     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 194
 195     TRACE("(%p)\n", This);
 196
 197     return S_OK;
 198 }
 199
 200 static HRESULT WINAPI MkProtocol_StartEx(IInternetProtocolEx *iface, IUri *pUri,
 201         IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
 202         DWORD grfPI, HANDLE *dwReserved)
 203 {
 204     MkProtocol *This = impl_from_IInternetProtocolEx(iface);
 205     LPWSTR mime, progid, display_name, colon_ptr;
 206     DWORD bindf=0, eaten=0, scheme=0, len;
 207     BSTR url, path = NULL;
 208     IParseDisplayName *pdn;
 209     BINDINFO bindinfo;
 210     STATSTG statstg;
 211     IMoniker *mon;
 212     HRESULT hres;
 213     CLSID clsid;
 214
 215     TRACE("(%p)->(%p %p %p %08x %p)\n", This, pUri, pOIProtSink,
 216             pOIBindInfo, grfPI, dwReserved);
 217
 218     hres = IUri_GetScheme(pUri, &scheme);
 219     if(FAILED(hres))
 220         return hres;
 221     if(scheme != URL_SCHEME_MK)
 222         return INET_E_INVALID_URL;
 223
 224     memset(&bindinfo, 0, sizeof(bindinfo));
 225     bindinfo.cbSize = sizeof(BINDINFO);
 226     hres = IInternetBindInfo_GetBindInfo(pOIBindInfo, &bindf, &bindinfo);
 227     if(FAILED(hres)) {
 228         WARN("GetBindInfo failed: %08x\n", hres);
 229         return hres;
 230     }
 231
 232     ReleaseBindInfo(&bindinfo);
 233
 234     IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_SENDINGREQUEST, NULL);
 235
 236     hres = IUri_GetDisplayUri(pUri, &url);
 237     if(FAILED(hres))
 238         return hres;
 239     hres = FindMimeFromData(NULL, url, NULL, 0, NULL, 0, &mime, 0);
 240     SysFreeString(url);
 241     if(SUCCEEDED(hres)) {
 242         IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_MIMETYPEAVAILABLE, mime);
 243         CoTaskMemFree(mime);
 244     }
 245
 246     hres = IUri_GetPath(pUri, &path);
 247     if(FAILED(hres))
 248         return hres;
 249     len = SysStringLen(path)+1;
 250     hres = UrlUnescapeW(path, NULL, &len, URL_UNESCAPE_INPLACE);
 251     if (FAILED(hres)) {
 252         SysFreeString(path);
 253         return report_result(pOIProtSink, INET_E_RESOURCE_NOT_FOUND, ERROR_INVALID_PARAMETER);
 254     }
 255
 256     progid = path+1; /* skip '@' symbol */
 257     colon_ptr = strchrW(path, ':');
 258     if(!colon_ptr) {
 259         SysFreeString(path);
 260         return report_result(pOIProtSink, INET_E_RESOURCE_NOT_FOUND, ERROR_INVALID_PARAMETER);
 261     }
 262
 263     len = strlenW(path);
 264     display_name = heap_alloc((len+1)*sizeof(WCHAR));
 265     memcpy(display_name, path, (len+1)*sizeof(WCHAR));
 266
 267     progid[colon_ptr-progid] = 0; /* overwrite ':' with NULL terminator */
 268     hres = CLSIDFromProgID(progid, &clsid);
 269     SysFreeString(path);
 270     if(FAILED(hres))
 271     {
 272         heap_free(display_name);
 273         return report_result(pOIProtSink, INET_E_RESOURCE_NOT_FOUND, ERROR_INVALID_PARAMETER);
 274     }
 275
 276     hres = CoCreateInstance(&clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 277             &IID_IParseDisplayName, (void**)&pdn);
 278     if(FAILED(hres)) {
 279         WARN("Could not create object %s\n", debugstr_guid(&clsid));
 280         heap_free(display_name);
 281         return report_result(pOIProtSink, hres, ERROR_INVALID_PARAMETER);
 282     }
 283
 284     hres = IParseDisplayName_ParseDisplayName(pdn, NULL /* FIXME */, display_name, &eaten, &mon);
 285     heap_free(display_name);
 286     IParseDisplayName_Release(pdn);
 287     if(FAILED(hres)) {
 288         WARN("ParseDisplayName failed: %08x\n", hres);
 289         return report_result(pOIProtSink, hres, ERROR_INVALID_PARAMETER);
 290     }
 291
 292     if(This->stream) {
 293         IStream_Release(This->stream);
 294         This->stream = NULL;
 295     }
 296
 297     hres = IMoniker_BindToStorage(mon, NULL /* FIXME */, NULL, &IID_IStream, (void**)&This->stream);
 298     IMoniker_Release(mon);
 299     if(FAILED(hres)) {
 300         WARN("BindToStorage failed: %08x\n", hres);
 301         return report_result(pOIProtSink, hres, ERROR_INVALID_PARAMETER);
 302     }
 303
 304     hres = IStream_Stat(This->stream, &statstg, STATFLAG_NONAME);
 305     if(FAILED(hres)) {
 306         WARN("Stat failed: %08x\n", hres);
 307         return report_result(pOIProtSink, hres, ERROR_INVALID_PARAMETER);
 308     }
 309
 310     IInternetProtocolSink_ReportData(pOIProtSink,
 311             BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION,
 312             statstg.cbSize.u.LowPart, statstg.cbSize.u.LowPart);
 313     return report_result(pOIProtSink, S_OK, ERROR_SUCCESS);
 314 }
 315
 316 static const IInternetProtocolExVtbl MkProtocolVtbl = {
 317     MkProtocol_QueryInterface,
 318     MkProtocol_AddRef,
 319     MkProtocol_Release,
 320     MkProtocol_Start,
 321     MkProtocol_Continue,
 322     MkProtocol_Abort,
 323     MkProtocol_Terminate,
 324     MkProtocol_Suspend,
 325     MkProtocol_Resume,
 326     MkProtocol_Read,
 327     MkProtocol_Seek,
 328     MkProtocol_LockRequest,
 329     MkProtocol_UnlockRequest,
 330     MkProtocol_StartEx
 331 };
 332
 333 HRESULT MkProtocol_Construct(IUnknown *pUnkOuter, LPVOID *ppobj)
 334 {
 335     MkProtocol *ret;
 336
 337     TRACE("(%p %p)\n", pUnkOuter, ppobj);
 338
 339     URLMON_LockModule();
 340
 341     ret = heap_alloc(sizeof(MkProtocol));
 342
 343     ret->IInternetProtocolEx_iface.lpVtbl = &MkProtocolVtbl;
 344     ret->ref = 1;
 345     ret->stream = NULL;
 346
 347     /* NOTE:
 348      * Native returns NULL ppobj and S_OK in CreateInstance if called with IID_IUnknown riid.
 349      */
 350     *ppobj = &ret->IInternetProtocolEx_iface;
 351
 352     return S_OK;
 353 }