Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749/* * Copyright 2017 Jacek Caban for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */
#define COBJMACROS#define NONAMELESSUNION
#include <assert.h>
#include "mimeole.h"#include "inetcomm_private.h"
#include "wine/debug.h"#include "wine/heap.h"
WINE_DEFAULT_DEBUG_CHANNEL(inetcomm);
typedef struct { IUnknown IUnknown_inner; IInternetProtocol IInternetProtocol_iface; IInternetProtocolInfo IInternetProtocolInfo_iface;
LONG ref; IUnknown *outer_unk;
WCHAR *location; IStream *stream; IInternetProtocolSink *sink;} MimeHtmlProtocol;
typedef struct { const WCHAR *mhtml; size_t mhtml_len; const WCHAR *location;} mhtml_url_t;
typedef struct { IBindStatusCallback IBindStatusCallback_iface;
LONG ref;
MimeHtmlProtocol *protocol; HRESULT status; IStream *stream; WCHAR url[1];} MimeHtmlBinding;
static const WCHAR mhtml_prefixW[] = {'m','h','t','m','l',':'};static const WCHAR mhtml_separatorW[] = {'!','x','-','u','s','c',':'};
static inline LPWSTR heap_strdupW(LPCWSTR str){ LPWSTR ret = NULL;
if(str) { DWORD size;
size = (lstrlenW(str)+1)*sizeof(WCHAR); ret = heap_alloc(size); if(ret) memcpy(ret, str, size); }
return ret;}
static HRESULT parse_mhtml_url(const WCHAR *url, mhtml_url_t *r){ const WCHAR *p;
if(_wcsnicmp(url, mhtml_prefixW, ARRAY_SIZE(mhtml_prefixW))) return E_FAIL;
r->mhtml = url + ARRAY_SIZE(mhtml_prefixW); p = wcschr(r->mhtml, '!'); if(p) { r->mhtml_len = p - r->mhtml; /* FIXME: We handle '!' and '!x-usc:' in URLs as the same thing. Those should not be the same. */ if(!wcsncmp(p, mhtml_separatorW, ARRAY_SIZE(mhtml_separatorW))) p += ARRAY_SIZE(mhtml_separatorW); else p++; }else { r->mhtml_len = lstrlenW(r->mhtml); }
r->location = p; return S_OK;}
static HRESULT report_result(MimeHtmlProtocol *protocol, HRESULT result){ if(protocol->sink) { IInternetProtocolSink_ReportResult(protocol->sink, result, ERROR_SUCCESS, NULL); IInternetProtocolSink_Release(protocol->sink); protocol->sink = NULL; }
return result;}
static HRESULT on_mime_message_available(MimeHtmlProtocol *protocol, IMimeMessage *mime_message){ FINDBODY find = {NULL}; IMimeBody *mime_body; PROPVARIANT value; HBODY body; HRESULT hres;
hres = IMimeMessage_FindFirst(mime_message, &find, &body); if(FAILED(hres)) return report_result(protocol, hres);
if(protocol->location) { BOOL found = FALSE; do { hres = IMimeMessage_FindNext(mime_message, &find, &body); if(FAILED(hres)) { WARN("location %s not found\n", debugstr_w(protocol->location)); return report_result(protocol, hres); }
value.vt = VT_LPWSTR; hres = IMimeMessage_GetBodyProp(mime_message, body, "content-location", 0, &value); if(hres == MIME_E_NOT_FOUND) continue; if(FAILED(hres)) return report_result(protocol, hres);
found = !lstrcmpW(protocol->location, value.u.pwszVal); PropVariantClear(&value); }while(!found); }else { hres = IMimeMessage_FindNext(mime_message, &find, &body); if(FAILED(hres)) { WARN("location %s not found\n", debugstr_w(protocol->location)); return report_result(protocol, hres); } }
hres = IMimeMessage_BindToObject(mime_message, body, &IID_IMimeBody, (void**)&mime_body); if(FAILED(hres)) return report_result(protocol, hres);
value.vt = VT_LPWSTR; hres = IMimeBody_GetProp(mime_body, "content-type", 0, &value); if(SUCCEEDED(hres)) { hres = IInternetProtocolSink_ReportProgress(protocol->sink, BINDSTATUS_MIMETYPEAVAILABLE, value.u.pwszVal); PropVariantClear(&value); }
/* FIXME: Create and report cache file. */
hres = IMimeBody_GetData(mime_body, IET_DECODED, &protocol->stream); if(FAILED(hres)) return report_result(protocol, hres);
IInternetProtocolSink_ReportData(protocol->sink, BSCF_FIRSTDATANOTIFICATION | BSCF_INTERMEDIATEDATANOTIFICATION | BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE | BSCF_AVAILABLEDATASIZEUNKNOWN, 0, 0);
return report_result(protocol, S_OK);}
static HRESULT load_mime_message(IStream *stream, IMimeMessage **ret){ IMimeMessage *mime_message; HRESULT hres;
hres = MimeMessage_create(NULL, (void**)&mime_message); if(FAILED(hres)) return hres;
IMimeMessage_InitNew(mime_message);
hres = IMimeMessage_Load(mime_message, stream); if(FAILED(hres)) { IMimeMessage_Release(mime_message); return hres; }
*ret = mime_message; return S_OK;}
static inline MimeHtmlBinding *impl_from_IBindStatusCallback(IBindStatusCallback *iface){ return CONTAINING_RECORD(iface, MimeHtmlBinding, IBindStatusCallback_iface);}
static HRESULT WINAPI BindStatusCallback_QueryInterface(IBindStatusCallback *iface, REFIID riid, void **ppv){ MimeHtmlBinding *This = impl_from_IBindStatusCallback(iface);
if(IsEqualGUID(&IID_IUnknown, riid)) { TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv); *ppv = &This->IBindStatusCallback_iface; }else if(IsEqualGUID(&IID_IBindStatusCallback, riid)) { TRACE("(%p)->(IID_IBindStatusCallback %p)\n", This, ppv); *ppv = &This->IBindStatusCallback_iface; }else { TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv); *ppv = NULL; return E_NOINTERFACE; }
IUnknown_AddRef((IUnknown*)*ppv); return S_OK;}
static ULONG WINAPI BindStatusCallback_AddRef(IBindStatusCallback *iface){ MimeHtmlBinding *This = impl_from_IBindStatusCallback(iface); LONG ref = InterlockedIncrement(&This->ref);
TRACE("(%p) ref=%d\n", This, ref);
return ref;}
static ULONG WINAPI BindStatusCallback_Release(IBindStatusCallback *iface){ MimeHtmlBinding *This = impl_from_IBindStatusCallback(iface); LONG ref = InterlockedDecrement(&This->ref);
TRACE("(%p) ref=%d\n", This, ref);
if(!ref) { if(This->protocol) IInternetProtocol_Release(&This->protocol->IInternetProtocol_iface); if(This->stream) IStream_Release(This->stream); heap_free(This); }
return ref;}
static HRESULT WINAPI BindStatusCallback_OnStartBinding(IBindStatusCallback *iface, DWORD dwReserved, IBinding *pib){ MimeHtmlBinding *This = impl_from_IBindStatusCallback(iface);
TRACE("(%p)->(%x %p)\n", This, dwReserved, pib);
assert(!This->stream); return CreateStreamOnHGlobal(NULL, TRUE, &This->stream);}
static HRESULT WINAPI BindStatusCallback_GetPriority(IBindStatusCallback *iface, LONG *pnPriority){ return E_NOTIMPL;}
static HRESULT WINAPI BindStatusCallback_OnLowResource(IBindStatusCallback *iface, DWORD dwReserved){ return E_NOTIMPL;}
static HRESULT WINAPI BindStatusCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress, ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText){ MimeHtmlBinding *This = impl_from_IBindStatusCallback(iface); TRACE("(%p)->(%u/%u %u %s)\n", This, ulProgress, ulProgressMax, ulStatusCode, debugstr_w(szStatusText)); return S_OK;}
static HRESULT WINAPI BindStatusCallback_OnStopBinding(IBindStatusCallback *iface, HRESULT hresult, LPCWSTR szError){ MimeHtmlBinding *This = impl_from_IBindStatusCallback(iface); IMimeMessage *mime_message = NULL;
TRACE("(%p)->(%x %s)\n", This, hresult, debugstr_w(szError));
if(SUCCEEDED(hresult)) { hresult = load_mime_message(This->stream, &mime_message); IStream_Release(This->stream); This->stream = NULL; }
This->status = hresult;
if(mime_message) on_mime_message_available(This->protocol, mime_message); else report_result(This->protocol, hresult);
if(mime_message) IMimeMessage_Release(mime_message); IInternetProtocol_Release(&This->protocol->IInternetProtocol_iface); This->protocol = NULL; return S_OK;}
static HRESULT WINAPI BindStatusCallback_GetBindInfo(IBindStatusCallback *iface, DWORD* grfBINDF, BINDINFO* pbindinfo){ MimeHtmlBinding *This = impl_from_IBindStatusCallback(iface);
TRACE("(%p)\n", This);
*grfBINDF = BINDF_ASYNCHRONOUS; return S_OK;}
static HRESULT WINAPI BindStatusCallback_OnDataAvailable(IBindStatusCallback *iface, DWORD grfBSCF, DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed){ MimeHtmlBinding *This = impl_from_IBindStatusCallback(iface); BYTE buf[4*1024]; DWORD read; HRESULT hres;
TRACE("(%p)\n", This);
assert(pstgmed->tymed == TYMED_ISTREAM);
while(1) { hres = IStream_Read(pstgmed->u.pstm, buf, sizeof(buf), &read); if(FAILED(hres)) return hres; if(!read) break; hres = IStream_Write(This->stream, buf, read, NULL); if(FAILED(hres)) return hres; } return S_OK;}
static HRESULT WINAPI BindStatusCallback_OnObjectAvailable(IBindStatusCallback *iface, REFIID riid, IUnknown* punk){ ERR("\n"); return E_NOTIMPL;}
static const IBindStatusCallbackVtbl BindStatusCallbackVtbl = { BindStatusCallback_QueryInterface, BindStatusCallback_AddRef, BindStatusCallback_Release, BindStatusCallback_OnStartBinding, BindStatusCallback_GetPriority, BindStatusCallback_OnLowResource, BindStatusCallback_OnProgress, BindStatusCallback_OnStopBinding, BindStatusCallback_GetBindInfo, BindStatusCallback_OnDataAvailable, BindStatusCallback_OnObjectAvailable};
static inline MimeHtmlProtocol *impl_from_IUnknown(IUnknown *iface){ return CONTAINING_RECORD(iface, MimeHtmlProtocol, IUnknown_inner);}
static HRESULT WINAPI MimeHtmlProtocol_QueryInterface(IUnknown *iface, REFIID riid, void **ppv){ MimeHtmlProtocol *This = impl_from_IUnknown(iface);
if(IsEqualGUID(&IID_IUnknown, riid)) { TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv); *ppv = &This->IInternetProtocol_iface; }else if(IsEqualGUID(&IID_IInternetProtocolRoot, riid)) { TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This, ppv); *ppv = &This->IInternetProtocol_iface; }else if(IsEqualGUID(&IID_IInternetProtocol, riid)) { TRACE("(%p)->(IID_IInternetProtocol %p)\n", This, ppv); *ppv = &This->IInternetProtocol_iface; }else if(IsEqualGUID(&IID_IInternetProtocolInfo, riid)) { TRACE("(%p)->(IID_IInternetProtocolInfo %p)\n", This, ppv); *ppv = &This->IInternetProtocolInfo_iface; }else { FIXME("unknown interface %s\n", debugstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; }
IUnknown_AddRef((IUnknown*)*ppv); return S_OK;}
static ULONG WINAPI MimeHtmlProtocol_AddRef(IUnknown *iface){ MimeHtmlProtocol *This = impl_from_IUnknown(iface); ULONG ref = InterlockedIncrement(&This->ref);
TRACE("(%p) ref=%d\n", This, ref);
return ref;}
static ULONG WINAPI MimeHtmlProtocol_Release(IUnknown *iface){ MimeHtmlProtocol *This = impl_from_IUnknown(iface); ULONG ref = InterlockedDecrement(&This->ref);
TRACE("(%p) ref=%x\n", This, ref);
if(!ref) { if(This->sink) IInternetProtocolSink_Release(This->sink); if(This->stream) IStream_Release(This->stream); heap_free(This->location); heap_free(This); }
return ref;}
static const IUnknownVtbl MimeHtmlProtocolInnerVtbl = { MimeHtmlProtocol_QueryInterface, MimeHtmlProtocol_AddRef, MimeHtmlProtocol_Release};
static inline MimeHtmlProtocol *impl_from_IInternetProtocol(IInternetProtocol *iface){ return CONTAINING_RECORD(iface, MimeHtmlProtocol, IInternetProtocol_iface);}
static HRESULT WINAPI InternetProtocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); return IUnknown_QueryInterface(This->outer_unk, riid, ppv);}
static ULONG WINAPI InternetProtocol_AddRef(IInternetProtocol *iface){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); return IUnknown_AddRef(This->outer_unk);}
static ULONG WINAPI InternetProtocol_Release(IInternetProtocol *iface){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); return IUnknown_Release(This->outer_unk);}
static HRESULT WINAPI MimeHtmlProtocol_Start(IInternetProtocol *iface, const WCHAR *szUrl, IInternetProtocolSink* pOIProtSink, IInternetBindInfo* pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); BINDINFO bindinfo = { sizeof(bindinfo) }; MimeHtmlBinding *binding; IBindCtx *bind_ctx; IStream *stream; mhtml_url_t url; DWORD bindf = 0; IMoniker *mon; HRESULT hres;
TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink, pOIBindInfo, grfPI, dwReserved);
hres = parse_mhtml_url(szUrl, &url); if(FAILED(hres)) return hres;
if(url.location && !(This->location = heap_strdupW(url.location))) return E_OUTOFMEMORY;
hres = IInternetBindInfo_GetBindInfo(pOIBindInfo, &bindf, &bindinfo); if(FAILED(hres)) { WARN("GetBindInfo failed: %08x\n", hres); return hres; } if((bindf & (BINDF_ASYNCHRONOUS|BINDF_FROMURLMON|BINDF_NEEDFILE)) != (BINDF_ASYNCHRONOUS|BINDF_FROMURLMON|BINDF_NEEDFILE)) FIXME("unsupported bindf %x\n", bindf);
IInternetProtocolSink_AddRef(This->sink = pOIProtSink);
binding = heap_alloc(FIELD_OFFSET(MimeHtmlBinding, url[url.mhtml_len+1])); if(!binding) return E_OUTOFMEMORY; memcpy(binding->url, url.mhtml, url.mhtml_len*sizeof(WCHAR)); binding->url[url.mhtml_len] = 0;
hres = CreateURLMoniker(NULL, binding->url, &mon); if(FAILED(hres)) { heap_free(binding); return hres; }
binding->IBindStatusCallback_iface.lpVtbl = &BindStatusCallbackVtbl; binding->ref = 1; binding->status = E_PENDING; binding->stream = NULL; binding->protocol = NULL;
hres = CreateAsyncBindCtx(0, &binding->IBindStatusCallback_iface, NULL, &bind_ctx); if(FAILED(hres)) { IMoniker_Release(mon); IBindStatusCallback_Release(&binding->IBindStatusCallback_iface); return hres; }
IInternetProtocol_AddRef(&This->IInternetProtocol_iface); binding->protocol = This;
hres = IMoniker_BindToStorage(mon, bind_ctx, NULL, &IID_IStream, (void**)&stream); IBindCtx_Release(bind_ctx); IMoniker_Release(mon); if(stream) IStream_Release(stream); hres = binding->status; IBindStatusCallback_Release(&binding->IBindStatusCallback_iface); if(FAILED(hres) && hres != E_PENDING) report_result(This, hres); return hres;}
static HRESULT WINAPI MimeHtmlProtocol_Continue(IInternetProtocol *iface, PROTOCOLDATA *pProtocolData){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)->(%p)\n", This, pProtocolData); return E_NOTIMPL;}
static HRESULT WINAPI MimeHtmlProtocol_Abort(IInternetProtocol *iface, HRESULT hrReason, DWORD dwOptions){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions); return E_NOTIMPL;}
static HRESULT WINAPI MimeHtmlProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); TRACE("(%p)->(%08x)\n", This, dwOptions); return S_OK;}
static HRESULT WINAPI MimeHtmlProtocol_Suspend(IInternetProtocol *iface){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)\n", This); return E_NOTIMPL;}
static HRESULT WINAPI MimeHtmlProtocol_Resume(IInternetProtocol *iface){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)\n", This); return E_NOTIMPL;}
static HRESULT WINAPI MimeHtmlProtocol_Read(IInternetProtocol *iface, void* pv, ULONG cb, ULONG* pcbRead){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); ULONG read = 0; HRESULT hres;
TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
hres = IStream_Read(This->stream, pv, cb, &read); if(pcbRead) *pcbRead = read; if(hres != S_OK) return hres;
return read ? S_OK : S_FALSE;}
static HRESULT WINAPI MimeHtmlProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition); return E_NOTIMPL;}
static HRESULT WINAPI MimeHtmlProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)->(%d)\n", This, dwOptions); return S_OK;}
static HRESULT WINAPI MimeHtmlProtocol_UnlockRequest(IInternetProtocol *iface){ MimeHtmlProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)\n", This); return S_OK;}
static const IInternetProtocolVtbl MimeHtmlProtocolVtbl = { InternetProtocol_QueryInterface, InternetProtocol_AddRef, InternetProtocol_Release, MimeHtmlProtocol_Start, MimeHtmlProtocol_Continue, MimeHtmlProtocol_Abort, MimeHtmlProtocol_Terminate, MimeHtmlProtocol_Suspend, MimeHtmlProtocol_Resume, MimeHtmlProtocol_Read, MimeHtmlProtocol_Seek, MimeHtmlProtocol_LockRequest, MimeHtmlProtocol_UnlockRequest};
static inline MimeHtmlProtocol *impl_from_IInternetProtocolInfo(IInternetProtocolInfo *iface){ return CONTAINING_RECORD(iface, MimeHtmlProtocol, IInternetProtocolInfo_iface);}
static HRESULT WINAPI MimeHtmlProtocolInfo_QueryInterface(IInternetProtocolInfo *iface, REFIID riid, void **ppv){ MimeHtmlProtocol *This = impl_from_IInternetProtocolInfo(iface); return IUnknown_QueryInterface(This->outer_unk, riid, ppv);}
static ULONG WINAPI MimeHtmlProtocolInfo_AddRef(IInternetProtocolInfo *iface){ MimeHtmlProtocol *This = impl_from_IInternetProtocolInfo(iface); return IUnknown_AddRef(This->outer_unk);}
static ULONG WINAPI MimeHtmlProtocolInfo_Release(IInternetProtocolInfo *iface){ MimeHtmlProtocol *This = impl_from_IInternetProtocolInfo(iface); return IUnknown_Release(This->outer_unk);}
static HRESULT WINAPI MimeHtmlProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl, PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult, DWORD* pcchResult, DWORD dwReserved){ MimeHtmlProtocol *This = impl_from_IInternetProtocolInfo(iface); FIXME("(%p)->(%s %d %x %p %d %p %d)\n", This, debugstr_w(pwzUrl), ParseAction, dwParseFlags, pwzResult, cchResult, pcchResult, dwReserved); return INET_E_DEFAULT_ACTION;}
static HRESULT WINAPI MimeHtmlProtocolInfo_CombineUrl(IInternetProtocolInfo *iface, LPCWSTR pwzBaseUrl, LPCWSTR pwzRelativeUrl, DWORD dwCombineFlags, LPWSTR pwzResult, DWORD cchResult, DWORD* pcchResult, DWORD dwReserved){ MimeHtmlProtocol *This = impl_from_IInternetProtocolInfo(iface); size_t len = ARRAY_SIZE(mhtml_prefixW); mhtml_url_t url; WCHAR *p; HRESULT hres;
TRACE("(%p)->(%s %s %08x %p %d %p %d)\n", This, debugstr_w(pwzBaseUrl), debugstr_w(pwzRelativeUrl), dwCombineFlags, pwzResult, cchResult, pcchResult, dwReserved);
hres = parse_mhtml_url(pwzBaseUrl, &url); if(FAILED(hres)) return hres;
if(!_wcsnicmp(pwzRelativeUrl, mhtml_prefixW, ARRAY_SIZE(mhtml_prefixW))) { FIXME("Relative URL is mhtml protocol\n"); return INET_E_USE_DEFAULT_PROTOCOLHANDLER; }
len += url.mhtml_len; if(*pwzRelativeUrl) len += lstrlenW(pwzRelativeUrl) + ARRAY_SIZE(mhtml_separatorW); if(len >= cchResult) { *pcchResult = 0; return E_FAIL; }
memcpy(pwzResult, mhtml_prefixW, sizeof(mhtml_prefixW)); p = pwzResult + ARRAY_SIZE(mhtml_prefixW); memcpy(p, url.mhtml, url.mhtml_len*sizeof(WCHAR)); p += url.mhtml_len; if(*pwzRelativeUrl) { memcpy(p, mhtml_separatorW, sizeof(mhtml_separatorW)); p += ARRAY_SIZE(mhtml_separatorW); lstrcpyW(p, pwzRelativeUrl); }else { *p = 0; }
*pcchResult = len; return S_OK;}
static HRESULT WINAPI MimeHtmlProtocolInfo_CompareUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags){ MimeHtmlProtocol *This = impl_from_IInternetProtocolInfo(iface); FIXME("(%p)->(%s %s %08x)\n", This, debugstr_w(pwzUrl1), debugstr_w(pwzUrl2), dwCompareFlags); return E_NOTIMPL;}
static HRESULT WINAPI MimeHtmlProtocolInfo_QueryInfo(IInternetProtocolInfo *iface, LPCWSTR pwzUrl, QUERYOPTION QueryOption, DWORD dwQueryFlags, LPVOID pBuffer, DWORD cbBuffer, DWORD* pcbBuf, DWORD dwReserved){ MimeHtmlProtocol *This = impl_from_IInternetProtocolInfo(iface); FIXME("(%p)->(%s %08x %08x %p %d %p %d)\n", This, debugstr_w(pwzUrl), QueryOption, dwQueryFlags, pBuffer, cbBuffer, pcbBuf, dwReserved); return INET_E_USE_DEFAULT_PROTOCOLHANDLER;}
static const IInternetProtocolInfoVtbl MimeHtmlProtocolInfoVtbl = { MimeHtmlProtocolInfo_QueryInterface, MimeHtmlProtocolInfo_AddRef, MimeHtmlProtocolInfo_Release, MimeHtmlProtocolInfo_ParseUrl, MimeHtmlProtocolInfo_CombineUrl, MimeHtmlProtocolInfo_CompareUrl, MimeHtmlProtocolInfo_QueryInfo};
HRESULT MimeHtmlProtocol_create(IUnknown *outer, void **obj){ MimeHtmlProtocol *protocol;
protocol = heap_alloc(sizeof(*protocol)); if(!protocol) return E_OUTOFMEMORY;
protocol->IUnknown_inner.lpVtbl = &MimeHtmlProtocolInnerVtbl; protocol->IInternetProtocol_iface.lpVtbl = &MimeHtmlProtocolVtbl; protocol->IInternetProtocolInfo_iface.lpVtbl = &MimeHtmlProtocolInfoVtbl; protocol->ref = 1; protocol->outer_unk = outer ? outer : &protocol->IUnknown_inner; protocol->location = NULL; protocol->stream = NULL; protocol->sink = NULL;
*obj = &protocol->IUnknown_inner; return S_OK;}