dlls/mshtml/protocol.c

   1 /*
   2  * Copyright 2005 Jacek Caban
   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 "config.h"
  20
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23
  24 #define COBJMACROS
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "ole2.h"
  30
  31 #include "wine/debug.h"
  32 #include "wine/unicode.h"
  33
  34 #include "mshtml_private.h"
  35
  36 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  37
  38 /********************************************************************
  39  * common ProtocolFactory implementation
  40  */
  41
  42 #define PROTOCOLINFO(x) ((IInternetProtocolInfo*) &(x)->lpInternetProtocolInfoVtbl)
  43 #define CLASSFACTORY(x) ((IClassFactory*)         &(x)->lpClassFactoryVtbl)
  44 #define PROTOCOL(x)     ((IInternetProtocol*)     &(x)->lpInternetProtocolVtbl)
  45
  46 typedef struct {
  47     const IInternetProtocolInfoVtbl *lpInternetProtocolInfoVtbl;
  48     const IClassFactoryVtbl         *lpClassFactoryVtbl;
  49 } ProtocolFactory;
  50
  51 #define PROTOCOLINFO_THIS(iface) DEFINE_THIS(ProtocolFactory, InternetProtocolInfo, iface)
  52
  53 static HRESULT WINAPI InternetProtocolInfo_QueryInterface(IInternetProtocolInfo *iface, REFIID riid, void **ppv)
  54 {
  55     ProtocolFactory *This = PROTOCOLINFO_THIS(iface);
  56
  57     *ppv = NULL;
  58     if(IsEqualGUID(&IID_IUnknown, riid)) {
  59         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
  60         *ppv = PROTOCOLINFO(This);
  61     }else if(IsEqualGUID(&IID_IInternetProtocolInfo, riid)) {
  62         TRACE("(%p)->(IID_IInternetProtocolInfo %p)\n", This, ppv);
  63         *ppv = PROTOCOLINFO(This);
  64     }else if(IsEqualGUID(&IID_IClassFactory, riid)) {
  65         TRACE("(%p)->(IID_IClassFactory %p)\n", This, ppv);
  66         *ppv = CLASSFACTORY(This);
  67     }
  68
  69     if(!*ppv) {
  70         WARN("unknown interface %s\n", debugstr_guid(riid));
  71         return E_NOINTERFACE;
  72     }
  73
  74     IInternetProtocolInfo_AddRef(iface);
  75     return S_OK;
  76 }
  77
  78 static ULONG WINAPI InternetProtocolInfo_AddRef(IInternetProtocolInfo *iface)
  79 {
  80     ProtocolFactory *This = PROTOCOLINFO_THIS(iface);
  81     TRACE("(%p)\n", This);
  82     LOCK_MODULE();
  83     return 2;
  84 }
  85
  86 static ULONG WINAPI InternetProtocolInfo_Release(IInternetProtocolInfo *iface)
  87 {
  88     ProtocolFactory *This = PROTOCOLINFO_THIS(iface);
  89     TRACE("(%p)\n", This);
  90     UNLOCK_MODULE();
  91     return 1;
  92 }
  93
  94 static HRESULT WINAPI InternetProtocolInfo_CombineUrl(IInternetProtocolInfo *iface,
  95         LPCWSTR pwzBaseUrl, LPCWSTR pwzRelativeUrl, DWORD dwCombineFlags, LPWSTR pwzResult,
  96         DWORD cchResult, DWORD* pcchResult, DWORD dwReserved)
  97 {
  98     TRACE("%p)->(%s %s %08x %p %d %p %d)\n", iface, debugstr_w(pwzBaseUrl),
  99             debugstr_w(pwzRelativeUrl), dwCombineFlags, pwzResult, cchResult,
 100             pcchResult, dwReserved);
 101
 102     return INET_E_USE_DEFAULT_PROTOCOLHANDLER;
 103 }
 104
 105 static HRESULT WINAPI InternetProtocolInfo_CompareUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl1,
 106         LPCWSTR pwzUrl2, DWORD dwCompareFlags)
 107 {
 108     TRACE("%p)->(%s %s %08x)\n", iface, debugstr_w(pwzUrl1), debugstr_w(pwzUrl2), dwCompareFlags);
 109     return E_NOTIMPL;
 110 }
 111
 112 #undef PROTOCOLINFO_THIS
 113
 114 #define CLASSFACTORY_THIS(iface) DEFINE_THIS(ProtocolFactory, ClassFactory, iface)
 115
 116 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
 117 {
 118     ProtocolFactory *This = CLASSFACTORY_THIS(iface);
 119     return IInternetProtocolInfo_QueryInterface(PROTOCOLINFO(This), riid, ppv);
 120 }
 121
 122 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
 123 {
 124     ProtocolFactory *This = CLASSFACTORY_THIS(iface);
 125     return IInternetProtocolInfo_AddRef(PROTOCOLINFO(This));
 126 }
 127
 128 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
 129 {
 130     ProtocolFactory *This = CLASSFACTORY_THIS(iface);
 131     return IInternetProtocolInfo_Release(PROTOCOLINFO(This));
 132 }
 133
 134 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
 135 {
 136     ProtocolFactory *This = CLASSFACTORY_THIS(iface);
 137
 138     TRACE("(%p)->(%x)\n", This, dolock);
 139
 140     if(dolock)
 141         LOCK_MODULE();
 142     else
 143         UNLOCK_MODULE();
 144
 145     return S_OK;
 146 }
 147
 148 #undef CLASSFACTORY_THIS
 149
 150 /********************************************************************
 151  * AboutProtocol implementation
 152  */
 153
 154 typedef struct {
 155     const IInternetProtocolVtbl *lpInternetProtocolVtbl;
 156
 157     LONG ref;
 158
 159     BYTE *data;
 160     ULONG data_len;
 161     ULONG cur;
 162
 163     IUnknown *pUnkOuter;
 164 } AboutProtocol;
 165
 166 #define PROTOCOL_THIS(iface) DEFINE_THIS(AboutProtocol, InternetProtocol, iface)
 167
 168 static HRESULT WINAPI AboutProtocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
 169 {
 170     AboutProtocol *This = PROTOCOL_THIS(iface);
 171
 172     *ppv = NULL;
 173
 174     if(IsEqualGUID(&IID_IUnknown, riid)) {
 175         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
 176         if(This->pUnkOuter)
 177             return IUnknown_QueryInterface(This->pUnkOuter, riid, ppv);
 178         *ppv = PROTOCOL(This);
 179     }else if(IsEqualGUID(&IID_IInternetProtocolRoot, riid)) {
 180         TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", iface, ppv);
 181         *ppv = PROTOCOL(This);
 182     }else if(IsEqualGUID(&IID_IInternetProtocol, riid)) {
 183         TRACE("(%p)->(IID_IInternetProtocol %p)\n", iface, ppv);
 184         *ppv = PROTOCOL(This);
 185     }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
 186         FIXME("IServiceProvider is not implemented\n");
 187         return E_NOINTERFACE;
 188     }
 189
 190     if(!*ppv) {
 191         TRACE("unknown interface %s\n", debugstr_guid(riid));
 192         return E_NOINTERFACE;
 193     }
 194
 195     IInternetProtocol_AddRef(iface);
 196     return S_OK;
 197 }
 198
 199 static ULONG WINAPI AboutProtocol_AddRef(IInternetProtocol *iface)
 200 {
 201     AboutProtocol *This = PROTOCOL_THIS(iface);
 202     ULONG ref = InterlockedIncrement(&This->ref);
 203     TRACE("(%p) ref=%d\n", iface, ref);
 204     return This->pUnkOuter ? IUnknown_AddRef(This->pUnkOuter) : ref;
 205 }
 206
 207 static ULONG WINAPI AboutProtocol_Release(IInternetProtocol *iface)
 208 {
 209     AboutProtocol *This = PROTOCOL_THIS(iface);
 210     IUnknown *pUnkOuter = This->pUnkOuter;
 211     ULONG ref = InterlockedDecrement(&This->ref);
 212
 213     TRACE("(%p) ref=%x\n", iface, ref);
 214
 215     if(!ref) {
 216         heap_free(This->data);
 217         heap_free(This);
 218         UNLOCK_MODULE();
 219     }
 220
 221     return pUnkOuter ? IUnknown_Release(pUnkOuter) : ref;
 222 }
 223
 224 static HRESULT WINAPI AboutProtocol_Start(IInternetProtocol *iface, LPCWSTR szUrl,
 225         IInternetProtocolSink* pOIProtSink, IInternetBindInfo* pOIBindInfo,
 226         DWORD grfPI, DWORD dwReserved)
 227 {
 228     AboutProtocol *This = PROTOCOL_THIS(iface);
 229     BINDINFO bindinfo;
 230     DWORD grfBINDF = 0;
 231     LPCWSTR text = NULL;
 232
 233     static const WCHAR html_begin[] = {0xfeff,'<','H','T','M','L','>',0};
 234     static const WCHAR html_end[] = {'<','/','H','T','M','L','>',0};
 235     static const WCHAR wszBlank[] = {'b','l','a','n','k',0};
 236     static const WCHAR wszAbout[] = {'a','b','o','u','t',':'};
 237     static const WCHAR wszTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
 238
 239     /* NOTE:
 240      * the about protocol seems not to work as I would expect. It creates html document
 241      * for a given url, eg. about:some_text -> <HTML>some_text</HTML> except for the case when
 242      * some_text = "blank", when document is blank (<HTML></HMTL>). The same happens
 243      * when the url does not have "about:" in the beginning.
 244      */
 245
 246     TRACE("(%p)->(%s %p %p %08x %d)\n", This, debugstr_w(szUrl), pOIProtSink,
 247             pOIBindInfo, grfPI, dwReserved);
 248
 249     memset(&bindinfo, 0, sizeof(bindinfo));
 250     bindinfo.cbSize = sizeof(BINDINFO);
 251     IInternetBindInfo_GetBindInfo(pOIBindInfo, &grfBINDF, &bindinfo);
 252     ReleaseBindInfo(&bindinfo);
 253
 254     TRACE("bindf %x\n", grfBINDF);
 255
 256     if(strlenW(szUrl)>=sizeof(wszAbout)/sizeof(WCHAR) && !memcmp(wszAbout, szUrl, sizeof(wszAbout))) {
 257         text = szUrl + sizeof(wszAbout)/sizeof(WCHAR);
 258         if(!strcmpW(wszBlank, text))
 259             text = NULL;
 260     }
 261
 262     This->data_len = sizeof(html_begin)+sizeof(html_end)-sizeof(WCHAR)
 263         + (text ? strlenW(text)*sizeof(WCHAR) : 0);
 264     This->data = heap_alloc(This->data_len);
 265
 266     memcpy(This->data, html_begin, sizeof(html_begin));
 267     if(text)
 268         strcatW((LPWSTR)This->data, text);
 269     strcatW((LPWSTR)This->data, html_end);
 270
 271     This->cur = 0;
 272
 273     IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_MIMETYPEAVAILABLE, wszTextHtml);
 274
 275     IInternetProtocolSink_ReportData(pOIProtSink,
 276             BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE,
 277             This->data_len, This->data_len);
 278
 279     IInternetProtocolSink_ReportResult(pOIProtSink, S_OK, 0, NULL);
 280
 281     return S_OK;
 282 }
 283
 284 static HRESULT WINAPI AboutProtocol_Continue(IInternetProtocol *iface, PROTOCOLDATA* pProtocolData)
 285 {
 286     AboutProtocol *This = PROTOCOL_THIS(iface);
 287     FIXME("(%p)->(%p)\n", This, pProtocolData);
 288     return E_NOTIMPL;
 289 }
 290
 291 static HRESULT WINAPI AboutProtocol_Abort(IInternetProtocol *iface, HRESULT hrReason,
 292         DWORD dwOptions)
 293 {
 294     AboutProtocol *This = PROTOCOL_THIS(iface);
 295     FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
 296     return E_NOTIMPL;
 297 }
 298
 299 static HRESULT WINAPI AboutProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
 300 {
 301     AboutProtocol *This = PROTOCOL_THIS(iface);
 302     TRACE("(%p)->(%08x)\n", This, dwOptions);
 303     return S_OK;
 304 }
 305
 306 static HRESULT WINAPI AboutProtocol_Suspend(IInternetProtocol *iface)
 307 {
 308     AboutProtocol *This = PROTOCOL_THIS(iface);
 309     FIXME("(%p)\n", This);
 310     return E_NOTIMPL;
 311 }
 312
 313 static HRESULT WINAPI AboutProtocol_Resume(IInternetProtocol *iface)
 314 {
 315     AboutProtocol *This = PROTOCOL_THIS(iface);
 316     FIXME("(%p)\n", This);
 317     return E_NOTIMPL;
 318 }
 319
 320 static HRESULT WINAPI AboutProtocol_Read(IInternetProtocol *iface, void* pv, ULONG cb, ULONG* pcbRead)
 321 {
 322     AboutProtocol *This = PROTOCOL_THIS(iface);
 323
 324     TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
 325
 326     if(!This->data)
 327         return E_FAIL;
 328
 329     *pcbRead = (cb > This->data_len-This->cur ? This->data_len-This->cur : cb);
 330
 331     if(!*pcbRead)
 332         return S_FALSE;
 333
 334     memcpy(pv, This->data, *pcbRead);
 335     This->cur += *pcbRead;
 336
 337     return S_OK;
 338 }
 339
 340 static HRESULT WINAPI AboutProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
 341         DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition)
 342 {
 343     AboutProtocol *This = PROTOCOL_THIS(iface);
 344     FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
 345     return E_NOTIMPL;
 346 }
 347
 348 static HRESULT WINAPI AboutProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
 349 {
 350     AboutProtocol *This = PROTOCOL_THIS(iface);
 351
 352     TRACE("(%p)->(%d)\n", This, dwOptions);
 353
 354     return S_OK;
 355 }
 356
 357 static HRESULT WINAPI AboutProtocol_UnlockRequest(IInternetProtocol *iface)
 358 {
 359     AboutProtocol *This = PROTOCOL_THIS(iface);
 360
 361     TRACE("(%p)\n", This);
 362
 363     return S_OK;
 364 }
 365
 366 #undef PROTOCOL_THIS
 367
 368 static const IInternetProtocolVtbl AboutProtocolVtbl = {
 369     AboutProtocol_QueryInterface,
 370     AboutProtocol_AddRef,
 371     AboutProtocol_Release,
 372     AboutProtocol_Start,
 373     AboutProtocol_Continue,
 374     AboutProtocol_Abort,
 375     AboutProtocol_Terminate,
 376     AboutProtocol_Suspend,
 377     AboutProtocol_Resume,
 378     AboutProtocol_Read,
 379     AboutProtocol_Seek,
 380     AboutProtocol_LockRequest,
 381     AboutProtocol_UnlockRequest
 382 };
 383
 384 static HRESULT WINAPI AboutProtocolFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter,
 385         REFIID riid, void **ppv)
 386 {
 387     AboutProtocol *ret;
 388     HRESULT hres = S_OK;
 389
 390     TRACE("(%p)->(%p %s %p)\n", iface, pUnkOuter, debugstr_guid(riid), ppv);
 391
 392     ret = heap_alloc(sizeof(AboutProtocol));
 393     ret->lpInternetProtocolVtbl = &AboutProtocolVtbl;
 394     ret->ref = 0;
 395
 396     ret->data = NULL;
 397     ret->data_len = 0;
 398     ret->cur = 0;
 399     ret->pUnkOuter = pUnkOuter;
 400
 401     if(pUnkOuter) {
 402         ret->ref = 1;
 403         if(IsEqualGUID(&IID_IUnknown, riid))
 404             *ppv = PROTOCOL(ret);
 405         else
 406             hres = E_INVALIDARG;
 407     }else {
 408         hres = IInternetProtocol_QueryInterface(PROTOCOL(ret), riid, ppv);
 409     }
 410
 411     if(SUCCEEDED(hres))
 412         LOCK_MODULE();
 413     else
 414         heap_free(ret);
 415
 416     return hres;
 417 }
 418
 419 static HRESULT WINAPI AboutProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
 420         PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult,
 421         DWORD* pcchResult, DWORD dwReserved)
 422 {
 423     TRACE("%p)->(%s %08x %08x %p %d %p %d)\n", iface, debugstr_w(pwzUrl), ParseAction,
 424             dwParseFlags, pwzResult, cchResult, pcchResult, dwReserved);
 425
 426     if(ParseAction == PARSE_SECURITY_URL) {
 427         int len = lstrlenW(pwzUrl);
 428
 429         if(len >= cchResult)
 430             return S_FALSE;
 431
 432         memcpy(pwzResult, pwzUrl, (len+1)*sizeof(WCHAR));
 433         return S_OK;
 434     }
 435
 436     if(ParseAction == PARSE_DOMAIN) {
 437         if(!pcchResult)
 438             return E_POINTER;
 439
 440         if(pwzUrl)
 441             *pcchResult = strlenW(pwzUrl)+1;
 442         else
 443             *pcchResult = 1;
 444         return E_FAIL;
 445     }
 446
 447     return INET_E_DEFAULT_ACTION;
 448 }
 449
 450 static HRESULT WINAPI AboutProtocolInfo_QueryInfo(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
 451         QUERYOPTION QueryOption, DWORD dwQueryFlags, LPVOID pBuffer, DWORD cbBuffer, DWORD* pcbBuf,
 452         DWORD dwReserved)
 453 {
 454     TRACE("%p)->(%s %08x %08x %p %d %p %d)\n", iface, debugstr_w(pwzUrl), QueryOption, dwQueryFlags, pBuffer,
 455           cbBuffer, pcbBuf, dwReserved);
 456
 457     switch(QueryOption) {
 458     case QUERY_CAN_NAVIGATE:
 459         return INET_E_USE_DEFAULT_PROTOCOLHANDLER;
 460
 461     case QUERY_USES_NETWORK:
 462         if(!pBuffer || cbBuffer < sizeof(DWORD))
 463             return E_FAIL;
 464
 465         *(DWORD*)pBuffer = 0;
 466         if(pcbBuf)
 467             *pcbBuf = sizeof(DWORD);
 468
 469         break;
 470
 471     case QUERY_IS_CACHED:
 472         FIXME("Unsupported option QUERY_IS_CACHED\n");
 473         return E_NOTIMPL;
 474     case QUERY_IS_INSTALLEDENTRY:
 475         FIXME("Unsupported option QUERY_IS_INSTALLEDENTRY\n");
 476         return E_NOTIMPL;
 477     case QUERY_IS_CACHED_OR_MAPPED:
 478         FIXME("Unsupported option QUERY_IS_CACHED_OR_MAPPED\n");
 479         return E_NOTIMPL;
 480     case QUERY_IS_SECURE:
 481         FIXME("Unsupported option QUERY_IS_SECURE\n");
 482         return E_NOTIMPL;
 483     case QUERY_IS_SAFE:
 484         FIXME("Unsupported option QUERY_IS_SAFE\n");
 485         return E_NOTIMPL;
 486     default:
 487         return E_FAIL;
 488     }
 489
 490     return S_OK;
 491 }
 492
 493 static const IInternetProtocolInfoVtbl AboutProtocolInfoVtbl = {
 494     InternetProtocolInfo_QueryInterface,
 495     InternetProtocolInfo_AddRef,
 496     InternetProtocolInfo_Release,
 497     AboutProtocolInfo_ParseUrl,
 498     InternetProtocolInfo_CombineUrl,
 499     InternetProtocolInfo_CompareUrl,
 500     AboutProtocolInfo_QueryInfo
 501 };
 502
 503 static const IClassFactoryVtbl AboutProtocolFactoryVtbl = {
 504     ClassFactory_QueryInterface,
 505     ClassFactory_AddRef,
 506     ClassFactory_Release,
 507     AboutProtocolFactory_CreateInstance,
 508     ClassFactory_LockServer
 509 };
 510
 511 static ProtocolFactory AboutProtocolFactory = {
 512     &AboutProtocolInfoVtbl,
 513     &AboutProtocolFactoryVtbl
 514 };
 515
 516 /********************************************************************
 517  * ResProtocol implementation
 518  */
 519
 520 typedef struct {
 521     const IInternetProtocolVtbl *lpInternetProtocolVtbl;
 522     LONG ref;
 523
 524     BYTE *data;
 525     ULONG data_len;
 526     ULONG cur;
 527
 528     IUnknown *pUnkOuter;
 529 } ResProtocol;
 530
 531 #define PROTOCOL_THIS(iface) DEFINE_THIS(ResProtocol, InternetProtocol, iface)
 532
 533 static HRESULT WINAPI ResProtocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
 534 {
 535     ResProtocol *This = PROTOCOL_THIS(iface);
 536
 537     *ppv = NULL;
 538
 539     if(IsEqualGUID(&IID_IUnknown, riid)) {
 540         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
 541         if(This->pUnkOuter)
 542             return IUnknown_QueryInterface(This->pUnkOuter, &IID_IUnknown, ppv);
 543         *ppv = PROTOCOL(This);
 544     }else if(IsEqualGUID(&IID_IInternetProtocolRoot, riid)) {
 545         TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", iface, ppv);
 546         *ppv = PROTOCOL(This);
 547     }else if(IsEqualGUID(&IID_IInternetProtocol, riid)) {
 548         TRACE("(%p)->(IID_IInternetProtocol %p)\n", iface, ppv);
 549         *ppv = PROTOCOL(This);
 550     }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
 551         FIXME("IServiceProvider is not implemented\n");
 552         return E_NOINTERFACE;
 553     }
 554
 555     if(!*ppv) {
 556         TRACE("unknown interface %s\n", debugstr_guid(riid));
 557         return E_NOINTERFACE;
 558     }
 559
 560     IInternetProtocol_AddRef(iface);
 561     return S_OK;
 562 }
 563
 564 static ULONG WINAPI ResProtocol_AddRef(IInternetProtocol *iface)
 565 {
 566     ResProtocol *This = PROTOCOL_THIS(iface);
 567     ULONG ref = InterlockedIncrement(&This->ref);
 568     TRACE("(%p) ref=%d\n", iface, ref);
 569     return This->pUnkOuter ? IUnknown_AddRef(This->pUnkOuter) : ref;
 570 }
 571
 572 static ULONG WINAPI ResProtocol_Release(IInternetProtocol *iface)
 573 {
 574     ResProtocol *This = (ResProtocol*)iface;
 575     IUnknown *pUnkOuter = This->pUnkOuter;
 576     ULONG ref = InterlockedDecrement(&This->ref);
 577
 578     TRACE("(%p) ref=%x\n", iface, ref);
 579
 580     if(!ref) {
 581         heap_free(This->data);
 582         heap_free(This);
 583         UNLOCK_MODULE();
 584     }
 585
 586     return pUnkOuter ? IUnknown_Release(pUnkOuter) : ref;
 587 }
 588
 589 static HRESULT WINAPI ResProtocol_Start(IInternetProtocol *iface, LPCWSTR szUrl,
 590         IInternetProtocolSink* pOIProtSink, IInternetBindInfo* pOIBindInfo,
 591         DWORD grfPI, DWORD dwReserved)
 592 {
 593     ResProtocol *This = PROTOCOL_THIS(iface);
 594     DWORD grfBINDF = 0, len;
 595     BINDINFO bindinfo;
 596     LPWSTR url_dll, url_file, url, mime;
 597     HMODULE hdll;
 598     HRSRC src;
 599     HRESULT hres;
 600
 601     static const WCHAR wszRes[] = {'r','e','s',':','/','/'};
 602
 603     TRACE("(%p)->(%s %p %p %08x %d)\n", This, debugstr_w(szUrl), pOIProtSink,
 604             pOIBindInfo, grfPI, dwReserved);
 605
 606     memset(&bindinfo, 0, sizeof(bindinfo));
 607     bindinfo.cbSize = sizeof(BINDINFO);
 608     IInternetBindInfo_GetBindInfo(pOIBindInfo, &grfBINDF, &bindinfo);
 609     ReleaseBindInfo(&bindinfo);
 610
 611     len = strlenW(szUrl)+16;
 612     url = heap_alloc(len*sizeof(WCHAR));
 613     hres = CoInternetParseUrl(szUrl, PARSE_ENCODE, 0, url, len, &len, 0);
 614     if(FAILED(hres)) {
 615         WARN("CoInternetParseUrl failed: %08x\n", hres);
 616         heap_free(url);
 617         IInternetProtocolSink_ReportResult(pOIProtSink, hres, 0, NULL);
 618         return hres;
 619     }
 620
 621     if(len < sizeof(wszRes)/sizeof(wszRes[0]) || memcmp(url, wszRes, sizeof(wszRes))) {
 622         WARN("Wrong protocol of url: %s\n", debugstr_w(url));
 623         IInternetProtocolSink_ReportResult(pOIProtSink, E_INVALIDARG, 0, NULL);
 624         heap_free(url);
 625         return E_INVALIDARG;
 626     }
 627
 628     url_dll = url + sizeof(wszRes)/sizeof(wszRes[0]);
 629     if(!(url_file = strrchrW(url_dll, '/'))) {
 630         WARN("wrong url: %s\n", debugstr_w(url));
 631         IInternetProtocolSink_ReportResult(pOIProtSink, MK_E_SYNTAX, 0, NULL);
 632         heap_free(url);
 633         return MK_E_SYNTAX;
 634     }
 635
 636     *url_file++ = 0;
 637     hdll = LoadLibraryExW(url_dll, NULL, LOAD_LIBRARY_AS_DATAFILE);
 638     if(!hdll) {
 639         WARN("Could not open dll: %s\n", debugstr_w(url_dll));
 640         IInternetProtocolSink_ReportResult(pOIProtSink, HRESULT_FROM_WIN32(GetLastError()), 0, NULL);
 641         heap_free(url);
 642         return HRESULT_FROM_WIN32(GetLastError());
 643     }
 644
 645     src = FindResourceW(hdll, url_file, (LPCWSTR)RT_HTML);
 646     if(!src) {
 647         LPWSTR endpoint = NULL;
 648         DWORD file_id = strtolW(url_file, &endpoint, 10);
 649         if(endpoint == url_file+strlenW(url_file))
 650             src = FindResourceW(hdll, (LPCWSTR)file_id, (LPCWSTR)RT_HTML);
 651
 652         if(!src) {
 653             WARN("Could not find resource\n");
 654             IInternetProtocolSink_ReportResult(pOIProtSink,
 655                     HRESULT_FROM_WIN32(GetLastError()), 0, NULL);
 656             heap_free(url);
 657             return HRESULT_FROM_WIN32(GetLastError());
 658         }
 659     }
 660
 661     if(This->data) {
 662         WARN("data already loaded\n");
 663         heap_free(This->data);
 664     }
 665
 666     This->data_len = SizeofResource(hdll, src);
 667     This->data = heap_alloc(This->data_len);
 668     memcpy(This->data, LoadResource(hdll, src), This->data_len);
 669     This->cur = 0;
 670
 671     FreeLibrary(hdll);
 672
 673     hres = FindMimeFromData(NULL, url_file, NULL, 0, NULL, 0, &mime, 0);
 674     heap_free(url);
 675     if(SUCCEEDED(hres)) {
 676         IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_MIMETYPEAVAILABLE, mime);
 677         CoTaskMemFree(mime);
 678     }
 679
 680     IInternetProtocolSink_ReportData(pOIProtSink,
 681             BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE,
 682             This->data_len, This->data_len);
 683
 684     IInternetProtocolSink_ReportResult(pOIProtSink, S_OK, 0, NULL);
 685
 686     return S_OK;
 687 }
 688
 689 static HRESULT WINAPI ResProtocol_Continue(IInternetProtocol *iface, PROTOCOLDATA* pProtocolData)
 690 {
 691     ResProtocol *This = PROTOCOL_THIS(iface);
 692     FIXME("(%p)->(%p)\n", This, pProtocolData);
 693     return E_NOTIMPL;
 694 }
 695
 696 static HRESULT WINAPI ResProtocol_Abort(IInternetProtocol *iface, HRESULT hrReason,
 697         DWORD dwOptions)
 698 {
 699     ResProtocol *This = PROTOCOL_THIS(iface);
 700     FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
 701     return E_NOTIMPL;
 702 }
 703
 704 static HRESULT WINAPI ResProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
 705 {
 706     ResProtocol *This = PROTOCOL_THIS(iface);
 707
 708     TRACE("(%p)->(%08x)\n", This, dwOptions);
 709
 710     /* test show that we don't have to do anything here */
 711     return S_OK;
 712 }
 713
 714 static HRESULT WINAPI ResProtocol_Suspend(IInternetProtocol *iface)
 715 {
 716     ResProtocol *This = PROTOCOL_THIS(iface);
 717     FIXME("(%p)\n", This);
 718     return E_NOTIMPL;
 719 }
 720
 721 static HRESULT WINAPI ResProtocol_Resume(IInternetProtocol *iface)
 722 {
 723     ResProtocol *This = PROTOCOL_THIS(iface);
 724     FIXME("(%p)\n", This);
 725     return E_NOTIMPL;
 726 }
 727
 728 static HRESULT WINAPI ResProtocol_Read(IInternetProtocol *iface, void* pv, ULONG cb, ULONG* pcbRead)
 729 {
 730     ResProtocol *This = PROTOCOL_THIS(iface);
 731
 732     TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
 733
 734     if(!This->data)
 735         return E_FAIL;
 736
 737     *pcbRead = (cb > This->data_len-This->cur ? This->data_len-This->cur : cb);
 738
 739     if(!*pcbRead)
 740         return S_FALSE;
 741
 742     memcpy(pv, This->data, *pcbRead);
 743     This->cur += *pcbRead;
 744
 745     return S_OK;
 746 }
 747
 748 static HRESULT WINAPI ResProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
 749         DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition)
 750 {
 751     ResProtocol *This = PROTOCOL_THIS(iface);
 752     FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
 753     return E_NOTIMPL;
 754 }
 755
 756 static HRESULT WINAPI ResProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
 757 {
 758     ResProtocol *This = PROTOCOL_THIS(iface);
 759
 760     TRACE("(%p)->(%d)\n", This, dwOptions);
 761
 762     /* test show that we don't have to do anything here */
 763     return S_OK;
 764 }
 765
 766 static HRESULT WINAPI ResProtocol_UnlockRequest(IInternetProtocol *iface)
 767 {
 768     ResProtocol *This = PROTOCOL_THIS(iface);
 769
 770     TRACE("(%p)\n", This);
 771
 772     /* test show that we don't have to do anything here */
 773     return S_OK;
 774 }
 775
 776 #undef PROTOCOL_THIS
 777
 778 static const IInternetProtocolVtbl ResProtocolVtbl = {
 779     ResProtocol_QueryInterface,
 780     ResProtocol_AddRef,
 781     ResProtocol_Release,
 782     ResProtocol_Start,
 783     ResProtocol_Continue,
 784     ResProtocol_Abort,
 785     ResProtocol_Terminate,
 786     ResProtocol_Suspend,
 787     ResProtocol_Resume,
 788     ResProtocol_Read,
 789     ResProtocol_Seek,
 790     ResProtocol_LockRequest,
 791     ResProtocol_UnlockRequest
 792 };
 793
 794 static HRESULT WINAPI ResProtocolFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter,
 795         REFIID riid, void **ppv)
 796 {
 797     ResProtocol *ret;
 798     HRESULT hres = S_OK;
 799
 800     TRACE("(%p)->(%p %s %p)\n", iface, pUnkOuter, debugstr_guid(riid), ppv);
 801
 802     ret = heap_alloc(sizeof(ResProtocol));
 803     ret->lpInternetProtocolVtbl = &ResProtocolVtbl;
 804     ret->ref = 0;
 805     ret->data = NULL;
 806     ret->data_len = 0;
 807     ret->cur = 0;
 808     ret->pUnkOuter = pUnkOuter;
 809
 810     if(pUnkOuter) {
 811         ret->ref = 1;
 812         if(IsEqualGUID(&IID_IUnknown, riid))
 813             *ppv = PROTOCOL(ret);
 814         else
 815             hres = E_FAIL;
 816     }else {
 817         hres = IInternetProtocol_QueryInterface(PROTOCOL(ret), riid, ppv);
 818     }
 819
 820     if(SUCCEEDED(hres))
 821         LOCK_MODULE();
 822     else
 823         heap_free(ret);
 824
 825     return hres;
 826 }
 827
 828 static HRESULT WINAPI ResProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
 829         PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult,
 830         DWORD* pcchResult, DWORD dwReserved)
 831 {
 832     TRACE("%p)->(%s %d %x %p %d %p %d)\n", iface, debugstr_w(pwzUrl), ParseAction,
 833             dwParseFlags, pwzResult, cchResult, pcchResult, dwReserved);
 834
 835     if(ParseAction == PARSE_SECURITY_URL) {
 836         WCHAR *ptr;
 837         DWORD size;
 838
 839         static const WCHAR wszFile[] = {'f','i','l','e',':','/','/'};
 840         static const WCHAR wszRes[] = {'r','e','s',':','/','/'};
 841
 842         if(strlenW(pwzUrl) <= sizeof(wszRes)/sizeof(WCHAR) || memcmp(pwzUrl, wszRes, sizeof(wszRes)))
 843             return E_INVALIDARG;
 844
 845         ptr = strchrW(pwzUrl + sizeof(wszRes)/sizeof(WCHAR), '/');
 846         if(!ptr)
 847             return E_INVALIDARG;
 848
 849         size = ptr-pwzUrl + sizeof(wszFile)/sizeof(WCHAR) - sizeof(wszRes)/sizeof(WCHAR);
 850         if(size >= cchResult)
 851             return S_FALSE;
 852
 853         /* FIXME: return full path */
 854         memcpy(pwzResult, wszFile, sizeof(wszFile));
 855         memcpy(pwzResult + sizeof(wszFile)/sizeof(WCHAR),
 856                 pwzUrl + sizeof(wszRes)/sizeof(WCHAR),
 857                 size*sizeof(WCHAR) - sizeof(wszFile));
 858         pwzResult[size] = 0;
 859
 860         if(pcchResult)
 861             *pcchResult = size;
 862         return S_OK;
 863     }
 864
 865     if(ParseAction == PARSE_DOMAIN) {
 866         if(!pcchResult)
 867             return E_POINTER;
 868
 869         if(pwzUrl)
 870             *pcchResult = strlenW(pwzUrl)+1;
 871         else
 872             *pcchResult = 1;
 873         return E_FAIL;
 874     }
 875
 876     return INET_E_DEFAULT_ACTION;
 877 }
 878
 879 static HRESULT WINAPI ResProtocolInfo_QueryInfo(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
 880         QUERYOPTION QueryOption, DWORD dwQueryFlags, LPVOID pBuffer, DWORD cbBuffer, DWORD* pcbBuf,
 881         DWORD dwReserved)
 882 {
 883     TRACE("%p)->(%s %08x %08x %p %d %p %d)\n", iface, debugstr_w(pwzUrl), QueryOption, dwQueryFlags, pBuffer,
 884           cbBuffer, pcbBuf, dwReserved);
 885
 886     switch(QueryOption) {
 887     case QUERY_USES_NETWORK:
 888         if(!pBuffer || cbBuffer < sizeof(DWORD))
 889             return E_FAIL;
 890
 891         *(DWORD*)pBuffer = 0;
 892         if(pcbBuf)
 893             *pcbBuf = sizeof(DWORD);
 894         break;
 895
 896     case QUERY_IS_SECURE:
 897         FIXME("not supporte QUERY_IS_SECURE\n");
 898         return E_NOTIMPL;
 899     case QUERY_IS_SAFE:
 900         FIXME("not supporte QUERY_IS_SAFE\n");
 901         return E_NOTIMPL;
 902     default:
 903         return INET_E_USE_DEFAULT_PROTOCOLHANDLER;
 904     }
 905
 906     return S_OK;
 907 }
 908
 909 static const IInternetProtocolInfoVtbl ResProtocolInfoVtbl = {
 910     InternetProtocolInfo_QueryInterface,
 911     InternetProtocolInfo_AddRef,
 912     InternetProtocolInfo_Release,
 913     ResProtocolInfo_ParseUrl,
 914     InternetProtocolInfo_CombineUrl,
 915     InternetProtocolInfo_CompareUrl,
 916     ResProtocolInfo_QueryInfo
 917 };
 918
 919 static const IClassFactoryVtbl ResProtocolFactoryVtbl = {
 920     ClassFactory_QueryInterface,
 921     ClassFactory_AddRef,
 922     ClassFactory_Release,
 923     ResProtocolFactory_CreateInstance,
 924     ClassFactory_LockServer
 925 };
 926
 927 static ProtocolFactory ResProtocolFactory = {
 928     &ResProtocolInfoVtbl,
 929     &ResProtocolFactoryVtbl
 930 };
 931
 932 HRESULT ProtocolFactory_Create(REFCLSID rclsid, REFIID riid, void **ppv)
 933 {
 934     ProtocolFactory *cf = NULL;
 935
 936     if(IsEqualGUID(&CLSID_AboutProtocol, rclsid))
 937         cf = &AboutProtocolFactory;
 938     else if(IsEqualGUID(&CLSID_ResProtocol, rclsid))
 939         cf = &ResProtocolFactory;
 940
 941     if(!cf) {
 942         FIXME("not implemented protocol %s\n", debugstr_guid(rclsid));
 943         return CLASS_E_CLASSNOTAVAILABLE;
 944     }
 945
 946     return IUnknown_QueryInterface((IUnknown*)cf, riid, ppv);
 947 }