git.oblomov.eu Git - wine/blob - dlls/mshtml/persist.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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 #include "config.h"
  20
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23
  24 #define COBJMACROS
  25 #define NONAMELESSUNION
  26 #define NONAMELESSSTRUCT
  27
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "winuser.h"
  31 #include "ole2.h"
  32 #include "shlguid.h"
  33
  34 #include "wine/debug.h"
  35
  36 #include "mshtml_private.h"
  37
  38 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  39
  40 struct BindStatusCallback {
  41     const IBindStatusCallbackVtbl *lpBindStatusCallbackVtbl;
  42
  43     LONG ref;
  44
  45     HTMLDocument *doc;
  46     IBinding *binding;
  47     IStream *stream;
  48     LPOLESTR url;
  49 };
  50
  51 #define STATUSCLB_THIS(iface) DEFINE_THIS(BindStatusCallback, BindStatusCallback, iface)
  52
  53 static HRESULT WINAPI BindStatusCallback_QueryInterface(IBindStatusCallback *iface,
  54         REFIID riid, void **ppv)
  55 {
  56     BindStatusCallback *This = STATUSCLB_THIS(iface);
  57
  58     *ppv = NULL;
  59     if(IsEqualGUID(&IID_IUnknown, riid)) {
  60         TRACE("(%p)->(IID_IUnknown, %p)\n", This, ppv);
  61         *ppv = STATUSCLB(This);
  62     }else if(IsEqualGUID(&IID_IBindStatusCallback, riid)) {
  63         TRACE("(%p)->(IID_IBindStatusCallback, %p)\n", This, ppv);
  64         *ppv = STATUSCLB(This);
  65     }
  66
  67     if(*ppv) {
  68         IBindStatusCallback_AddRef(STATUSCLB(This));
  69         return S_OK;
  70     }
  71
  72     TRACE("Unsupported riid = %s\n", debugstr_guid(riid));
  73     return E_NOINTERFACE;
  74 }
  75
  76 static ULONG WINAPI BindStatusCallback_AddRef(IBindStatusCallback *iface)
  77 {
  78     BindStatusCallback *This = STATUSCLB_THIS(iface);
  79     LONG ref = InterlockedIncrement(&This->ref);
  80
  81     TRACE("(%p) ref = %ld\n", This, ref);
  82
  83     return ref;
  84 }
  85
  86 static ULONG WINAPI BindStatusCallback_Release(IBindStatusCallback *iface)
  87 {
  88     BindStatusCallback *This = STATUSCLB_THIS(iface);
  89     LONG ref = InterlockedDecrement(&This->ref);
  90
  91     TRACE("(%p) ref = %ld\n", This, ref);
  92
  93     if(!ref) {
  94         if(This->doc->status_callback == This)
  95             This->doc->status_callback = NULL;
  96         IHTMLDocument2_Release(HTMLDOC(This->doc));
  97         if(This->stream)
  98             IStream_Release(This->stream);
  99         CoTaskMemFree(This->url);
 100         HeapFree(GetProcessHeap(), 0, This);
 101     }
 102
 103     return ref;
 104 }
 105
 106 static HRESULT WINAPI BindStatusCallback_OnStartBinding(IBindStatusCallback *iface,
 107         DWORD dwReserved, IBinding *pbind)
 108 {
 109     BindStatusCallback *This = STATUSCLB_THIS(iface);
 110
 111     TRACE("(%p)->(%ld %p)\n", This, dwReserved, pbind);
 112
 113     This->binding = pbind;
 114     IBinding_AddRef(pbind);
 115
 116     if(This->doc->nscontainer && This->doc->nscontainer->stream) {
 117         nsACString *strTextHtml;
 118         nsresult nsres;
 119         nsIURI *uri = get_nsIURI(This->url);
 120
 121         strTextHtml = nsACString_Create();
 122         /* FIXME: Set it correctly */
 123         nsACString_SetData(strTextHtml, "text/html");
 124
 125         nsres = nsIWebBrowserStream_OpenStream(This->doc->nscontainer->stream, uri, strTextHtml);
 126         if(NS_FAILED(nsres))
 127             ERR("OpenStream failed: %08lx\n", nsres);
 128
 129         nsIURI_Release(uri);
 130     }
 131
 132     return S_OK;
 133 }
 134
 135 static HRESULT WINAPI BindStatusCallback_GetPriority(IBindStatusCallback *iface, LONG *pnPriority)
 136 {
 137     BindStatusCallback *This = STATUSCLB_THIS(iface);
 138     FIXME("(%p)->(%p)\n", This, pnPriority);
 139     return E_NOTIMPL;
 140 }
 141
 142 static HRESULT WINAPI BindStatusCallback_OnLowResource(IBindStatusCallback *iface, DWORD reserved)
 143 {
 144     BindStatusCallback *This = STATUSCLB_THIS(iface);
 145     FIXME("(%p)->(%ld)\n", This, reserved);
 146     return E_NOTIMPL;
 147 }
 148
 149 static HRESULT WINAPI BindStatusCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
 150         ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
 151 {
 152     BindStatusCallback *This = STATUSCLB_THIS(iface);
 153     TRACE("%p)->(%lu %lu %lu %s)\n", This, ulProgress, ulProgressMax, ulStatusCode,
 154             debugstr_w(szStatusText));
 155     return S_OK;
 156 }
 157
 158 static HRESULT WINAPI BindStatusCallback_OnStopBinding(IBindStatusCallback *iface,
 159         HRESULT hresult, LPCWSTR szError)
 160 {
 161     BindStatusCallback *This = STATUSCLB_THIS(iface);
 162
 163     TRACE("(%p)->(%08lx %s)\n", This, hresult, debugstr_w(szError));
 164
 165     if(This->doc->nscontainer && This->doc->nscontainer->stream)
 166         nsIWebBrowserStream_CloseStream(This->doc->nscontainer->stream);
 167
 168     IBinding_Release(This->binding);
 169     This->binding = NULL;
 170     return S_OK;
 171 }
 172
 173 static HRESULT WINAPI BindStatusCallback_GetBindInfo(IBindStatusCallback *iface,
 174         DWORD *grfBINDF, BINDINFO *pbindinfo)
 175 {
 176     BindStatusCallback *This = STATUSCLB_THIS(iface);
 177     DWORD size;
 178
 179     TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
 180
 181     *grfBINDF = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA;
 182     size = pbindinfo->cbSize;
 183     memset(pbindinfo, 0, size);
 184     pbindinfo->cbSize = size;
 185
 186     return S_OK;
 187 }
 188
 189 static HRESULT WINAPI BindStatusCallback_OnDataAvailable(IBindStatusCallback *iface,
 190         DWORD grfBSCF, DWORD dwSize, FORMATETC *pformatetc, STGMEDIUM *pstgmed)
 191 {
 192     BindStatusCallback *This = STATUSCLB_THIS(iface);
 193
 194     TRACE("(%p)->(%08lx %ld %p %p)\n", This, grfBSCF, dwSize, pformatetc, pstgmed);
 195
 196     if(!This->stream) {
 197         This->stream = pstgmed->u.pstm;
 198         IStream_AddRef(This->stream);
 199     }
 200
 201     if(This->doc->nscontainer && This->doc->nscontainer->stream) {
 202         BYTE buf[1024];
 203         DWORD size;
 204         HRESULT hres;
 205
 206         do {
 207             size = sizeof(buf);
 208             hres = IStream_Read(This->stream, buf, size, &size);
 209             nsIWebBrowserStream_AppendToStream(This->doc->nscontainer->stream, buf, size);
 210         }while(hres == S_OK);
 211     }
 212
 213     return S_OK;
 214 }
 215
 216 static HRESULT WINAPI BindStatusCallback_OnObjectAvailable(IBindStatusCallback *iface,
 217         REFIID riid, IUnknown *punk)
 218 {
 219     BindStatusCallback *This = STATUSCLB_THIS(iface);
 220     FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), punk);
 221     return E_NOTIMPL;
 222 }
 223
 224 static const IBindStatusCallbackVtbl BindStatusCallbackVtbl = {
 225     BindStatusCallback_QueryInterface,
 226     BindStatusCallback_AddRef,
 227     BindStatusCallback_Release,
 228     BindStatusCallback_OnStartBinding,
 229     BindStatusCallback_GetPriority,
 230     BindStatusCallback_OnLowResource,
 231     BindStatusCallback_OnProgress,
 232     BindStatusCallback_OnStopBinding,
 233     BindStatusCallback_GetBindInfo,
 234     BindStatusCallback_OnDataAvailable,
 235     BindStatusCallback_OnObjectAvailable
 236 };
 237
 238 static BindStatusCallback *BindStatusCallback_Create(HTMLDocument *doc, LPOLESTR url)
 239 {
 240     BindStatusCallback *ret = HeapAlloc(GetProcessHeap(), 0, sizeof(BindStatusCallback));
 241     ret->lpBindStatusCallbackVtbl = &BindStatusCallbackVtbl;
 242     ret->ref = 0;
 243     ret->url = url;
 244     ret->doc = doc;
 245     ret->stream = NULL;
 246     IHTMLDocument2_AddRef(HTMLDOC(doc));
 247     return ret;
 248 }
 249
 250 /**********************************************************
 251  * IPersistMoniker implementation
 252  */
 253
 254 #define PERSISTMON_THIS(iface) DEFINE_THIS(HTMLDocument, PersistMoniker, iface)
 255
 256 static HRESULT WINAPI PersistMoniker_QueryInterface(IPersistMoniker *iface, REFIID riid,
 257                                                             void **ppvObject)
 258 {
 259     HTMLDocument *This = PERSISTMON_THIS(iface);
 260     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
 261 }
 262
 263 static ULONG WINAPI PersistMoniker_AddRef(IPersistMoniker *iface)
 264 {
 265     HTMLDocument *This = PERSISTMON_THIS(iface);
 266     return IHTMLDocument2_AddRef(HTMLDOC(This));
 267 }
 268
 269 static ULONG WINAPI PersistMoniker_Release(IPersistMoniker *iface)
 270 {
 271     HTMLDocument *This = PERSISTMON_THIS(iface);
 272     return IHTMLDocument2_Release(HTMLDOC(This));
 273 }
 274
 275 static HRESULT WINAPI PersistMoniker_GetClassID(IPersistMoniker *iface, CLSID *pClassID)
 276 {
 277     HTMLDocument *This = PERSISTMON_THIS(iface);
 278     return IPersist_GetClassID(PERSIST(This), pClassID);
 279 }
 280
 281 static HRESULT WINAPI PersistMoniker_IsDirty(IPersistMoniker *iface)
 282 {
 283     HTMLDocument *This = PERSISTMON_THIS(iface);
 284     FIXME("(%p)\n", This);
 285     return E_NOTIMPL;
 286 }
 287
 288 static HRESULT WINAPI PersistMoniker_Load(IPersistMoniker *iface, BOOL fFullyAvailable,
 289         IMoniker *pimkName, LPBC pibc, DWORD grfMode)
 290 {
 291     HTMLDocument *This = PERSISTMON_THIS(iface);
 292     IBindCtx *pbind;
 293     BindStatusCallback *callback;
 294     IStream *str = NULL;
 295     LPOLESTR url;
 296     HRESULT hres;
 297     nsresult nsres;
 298
 299     TRACE("(%p)->(%x %p %p %08lx)\n", This, fFullyAvailable, pimkName, pibc, grfMode);
 300
 301     if(pibc) {
 302         IUnknown *unk = NULL;
 303
 304         static WCHAR wszClientSiteParam[] = {'{','d','4','d','b','6','8','5','0','-',
 305             '5','3','8','5','-','1','1','d','0','-','8','9','e','9','-','0','0','a',
 306             '0','c','9','0','a','9','0','a','c','}',0};
 307
 308         /* FIXME:
 309          * Use params:
 310          * "__PrecreatedObject"
 311          * "BIND_CONTEXT_PARAM"
 312          * "__HTMLLOADOPTIONS"
 313          * "__DWNBINDINFO"
 314          * "URL Context"
 315          * "CBinding Context"
 316          * "_ITransData_Object_"
 317          * "_EnumFORMATETC_"
 318          */
 319
 320         IBindCtx_GetObjectParam(pibc, wszClientSiteParam, &unk);
 321         if(unk) {
 322             IOleClientSite *client = NULL;
 323
 324             hres = IUnknown_QueryInterface(unk, &IID_IOleClientSite, (void**)&client);
 325             if(SUCCEEDED(hres)) {
 326                 TRACE("Got client site %p\n", client);
 327                 IOleObject_SetClientSite(OLEOBJ(This), client);
 328                 IOleClientSite_Release(client);
 329             }
 330
 331             IUnknown_Release(unk);
 332         }
 333     }
 334
 335     HTMLDocument_LockContainer(This, TRUE);
 336
 337     hres = IMoniker_GetDisplayName(pimkName, pibc, NULL, &url);
 338     if(FAILED(hres)) {
 339         WARN("GetDiaplayName failed: %08lx\n", hres);
 340         return hres;
 341     }
 342
 343     TRACE("got url: %s\n", debugstr_w(url));
 344
 345     if(This->client) {
 346         IOleCommandTarget *cmdtrg = NULL;
 347
 348         hres = IOleClientSite_QueryInterface(This->client, &IID_IOleCommandTarget,
 349                 (void**)&cmdtrg);
 350         if(SUCCEEDED(hres)) {
 351             VARIANT var;
 352
 353             V_VT(&var) = VT_I4;
 354             V_I4(&var) = 0;
 355             IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 37, 0, &var, NULL);
 356         }
 357     }
 358
 359     if(This->nscontainer && !This->nscontainer->stream) {
 360         /*
 361          * This is a workaround for older Gecko that doesn't support nsIWebBrowserStream.
 362          * It uses Gecko's LoadURI instead of IMoniker's BindToStorage. Should we improve
 363          * it (to do so we'd have to use not frozen interfaces)?
 364          */
 365         nsres = nsIWebNavigation_LoadURI(This->nscontainer->navigation, url,
 366                 LOAD_FLAGS_NONE, NULL, NULL, NULL);
 367         if(NS_SUCCEEDED(nsres)) {
 368             CoTaskMemFree(url);
 369             return S_OK;
 370         }else {
 371             WARN("LoadURI failed: %08lx\n", nsres);
 372         }
 373     }
 374
 375     /* FIXME: Use grfMode */
 376
 377     if(fFullyAvailable)
 378         FIXME("not supported fFullyAvailable\n");
 379
 380     if(This->status_callback && This->status_callback->binding)
 381         IBinding_Abort(This->status_callback->binding);
 382
 383     callback = This->status_callback = BindStatusCallback_Create(This, url);
 384
 385     if(pibc) {
 386         pbind = pibc;
 387         RegisterBindStatusCallback(pbind, STATUSCLB(callback), NULL, 0);
 388     }else {
 389         CreateAsyncBindCtx(0, STATUSCLB(callback), NULL, &pbind);
 390     }
 391
 392     hres = IMoniker_BindToStorage(pimkName, pbind, NULL, &IID_IStream, (void**)&str);
 393
 394     if(!pibc)
 395         IBindCtx_Release(pbind);
 396     if(str)
 397         IStream_Release(str);
 398     if(FAILED(hres)) {
 399         WARN("BindToStorage failed: %08lx\n", hres);
 400         return hres;
 401     }
 402
 403     return S_OK;
 404 }
 405
 406 static HRESULT WINAPI PersistMoniker_Save(IPersistMoniker *iface, IMoniker *pimkName,
 407         LPBC pbc, BOOL fRemember)
 408 {
 409     HTMLDocument *This = PERSISTMON_THIS(iface);
 410     FIXME("(%p)->(%p %p %x)\n", This, pimkName, pbc, fRemember);
 411     return E_NOTIMPL;
 412 }
 413
 414 static HRESULT WINAPI PersistMoniker_SaveCompleted(IPersistMoniker *iface, IMoniker *pimkName, LPBC pibc)
 415 {
 416     HTMLDocument *This = PERSISTMON_THIS(iface);
 417     FIXME("(%p)->(%p %p)\n", This, pimkName, pibc);
 418     return E_NOTIMPL;
 419 }
 420
 421 static HRESULT WINAPI PersistMoniker_GetCurMoniker(IPersistMoniker *iface, IMoniker **ppimkName)
 422 {
 423     HTMLDocument *This = PERSISTMON_THIS(iface);
 424     FIXME("(%p)->(%p)\n", This, ppimkName);
 425     return E_NOTIMPL;
 426 }
 427
 428 static const IPersistMonikerVtbl PersistMonikerVtbl = {
 429     PersistMoniker_QueryInterface,
 430     PersistMoniker_AddRef,
 431     PersistMoniker_Release,
 432     PersistMoniker_GetClassID,
 433     PersistMoniker_IsDirty,
 434     PersistMoniker_Load,
 435     PersistMoniker_Save,
 436     PersistMoniker_SaveCompleted,
 437     PersistMoniker_GetCurMoniker
 438 };
 439
 440 /**********************************************************
 441  * IMonikerProp implementation
 442  */
 443
 444 #define MONPROP_THIS(iface) DEFINE_THIS(HTMLDocument, MonikerProp, iface)
 445
 446 static HRESULT WINAPI MonikerProp_QueryInterface(IMonikerProp *iface, REFIID riid, void **ppvObject)
 447 {
 448     HTMLDocument *This = MONPROP_THIS(iface);
 449     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
 450 }
 451
 452 static ULONG WINAPI MonikerProp_AddRef(IMonikerProp *iface)
 453 {
 454     HTMLDocument *This = MONPROP_THIS(iface);
 455     return IHTMLDocument2_AddRef(HTMLDOC(This));
 456 }
 457
 458 static ULONG WINAPI MonikerProp_Release(IMonikerProp *iface)
 459 {
 460     HTMLDocument *This = MONPROP_THIS(iface);
 461     return IHTMLDocument_Release(HTMLDOC(This));
 462 }
 463
 464 static HRESULT WINAPI MonikerProp_PutProperty(IMonikerProp *iface, MONIKERPROPERTY mkp, LPCWSTR val)
 465 {
 466     HTMLDocument *This = MONPROP_THIS(iface);
 467     FIXME("(%p)->(%d %s)\n", This, mkp, debugstr_w(val));
 468     return E_NOTIMPL;
 469 }
 470
 471 static const IMonikerPropVtbl MonikerPropVtbl = {
 472     MonikerProp_QueryInterface,
 473     MonikerProp_AddRef,
 474     MonikerProp_Release,
 475     MonikerProp_PutProperty
 476 };
 477
 478 /**********************************************************
 479  * IPersistFile implementation
 480  */
 481
 482 #define PERSISTFILE_THIS(iface) DEFINE_THIS(HTMLDocument, PersistFile, iface)
 483
 484 static HRESULT WINAPI PersistFile_QueryInterface(IPersistFile *iface, REFIID riid, void **ppvObject)
 485 {
 486     HTMLDocument *This = PERSISTFILE_THIS(iface);
 487     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
 488 }
 489
 490 static ULONG WINAPI PersistFile_AddRef(IPersistFile *iface)
 491 {
 492     HTMLDocument *This = PERSISTFILE_THIS(iface);
 493     return IHTMLDocument2_AddRef(HTMLDOC(This));
 494 }
 495
 496 static ULONG WINAPI PersistFile_Release(IPersistFile *iface)
 497 {
 498     HTMLDocument *This = PERSISTFILE_THIS(iface);
 499     return IHTMLDocument2_Release(HTMLDOC(This));
 500 }
 501
 502 static HRESULT WINAPI PersistFile_GetClassID(IPersistFile *iface, CLSID *pClassID)
 503 {
 504     HTMLDocument *This = PERSISTFILE_THIS(iface);
 505
 506     TRACE("(%p)->(%p)\n", This, pClassID);
 507
 508     if(!pClassID)
 509         return E_INVALIDARG;
 510
 511     memcpy(pClassID, &CLSID_HTMLDocument, sizeof(CLSID));
 512     return S_OK;
 513 }
 514
 515 static HRESULT WINAPI PersistFile_IsDirty(IPersistFile *iface)
 516 {
 517     HTMLDocument *This = PERSISTFILE_THIS(iface);
 518     FIXME("(%p)\n", This);
 519     return E_NOTIMPL;
 520 }
 521
 522 static HRESULT WINAPI PersistFile_Load(IPersistFile *iface, LPCOLESTR pszFileName, DWORD dwMode)
 523 {
 524     HTMLDocument *This = PERSISTFILE_THIS(iface);
 525     FIXME("(%p)->(%s %08lx)\n", This, debugstr_w(pszFileName), dwMode);
 526     return E_NOTIMPL;
 527 }
 528
 529 static HRESULT WINAPI PersistFile_Save(IPersistFile *iface, LPCOLESTR pszFileName, BOOL fRemember)
 530 {
 531     HTMLDocument *This = PERSISTFILE_THIS(iface);
 532     FIXME("(%p)->(%s %x)\n", This, debugstr_w(pszFileName), fRemember);
 533     return E_NOTIMPL;
 534 }
 535
 536 static HRESULT WINAPI PersistFile_SaveCompleted(IPersistFile *iface, LPCOLESTR pszFileName)
 537 {
 538     HTMLDocument *This = PERSISTFILE_THIS(iface);
 539     FIXME("(%p)->(%s)\n", This, debugstr_w(pszFileName));
 540     return E_NOTIMPL;
 541 }
 542
 543 static HRESULT WINAPI PersistFile_GetCurFile(IPersistFile *iface, LPOLESTR *pszFileName)
 544 {
 545     HTMLDocument *This = PERSISTFILE_THIS(iface);
 546     FIXME("(%p)->(%p)\n", This, pszFileName);
 547     return E_NOTIMPL;
 548 }
 549
 550 static const IPersistFileVtbl PersistFileVtbl = {
 551     PersistFile_QueryInterface,
 552     PersistFile_AddRef,
 553     PersistFile_Release,
 554     PersistFile_GetClassID,
 555     PersistFile_IsDirty,
 556     PersistFile_Load,
 557     PersistFile_Save,
 558     PersistFile_SaveCompleted,
 559     PersistFile_GetCurFile
 560 };
 561
 562 void HTMLDocument_Persist_Init(HTMLDocument *This)
 563 {
 564     This->lpPersistMonikerVtbl = &PersistMonikerVtbl;
 565     This->lpPersistFileVtbl = &PersistFileVtbl;
 566     This->lpMonikerPropVtbl = &MonikerPropVtbl;
 567
 568     This->status_callback = NULL;
 569 }