Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572/* * Copyright 2006-2007 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 */
#include <stdarg.h>
#define COBJMACROS
#include "windef.h"#include "winbase.h"#include "winuser.h"#include "winreg.h"#include "ole2.h"#include "urlmon.h"#include "shlwapi.h"#include "itsstor.h"#include "chm_lib.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(itss);
typedef struct { IUnknown IUnknown_inner; IInternetProtocol IInternetProtocol_iface; IInternetProtocolInfo IInternetProtocolInfo_iface;
LONG ref; IUnknown *outer;
ULONG offset; struct chmFile *chm_file; struct chmUnitInfo chm_object;} ITSProtocol;
static inline ITSProtocol *impl_from_IUnknown(IUnknown *iface){ return CONTAINING_RECORD(iface, ITSProtocol, IUnknown_inner);}
static inline ITSProtocol *impl_from_IInternetProtocol(IInternetProtocol *iface){ return CONTAINING_RECORD(iface, ITSProtocol, IInternetProtocol_iface);}
static inline ITSProtocol *impl_from_IInternetProtocolInfo(IInternetProtocolInfo *iface){ return CONTAINING_RECORD(iface, ITSProtocol, IInternetProtocolInfo_iface);}
static void release_chm(ITSProtocol *This){ if(This->chm_file) { chm_close(This->chm_file); This->chm_file = NULL; } This->offset = 0;}
static HRESULT WINAPI ITSProtocol_QueryInterface(IUnknown *iface, REFIID riid, void **ppv){ ITSProtocol *This = impl_from_IUnknown(iface);
if(IsEqualGUID(&IID_IUnknown, riid)) { TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv); *ppv = &This->IUnknown_inner; }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 { *ppv = NULL; WARN("not supported interface %s\n", debugstr_guid(riid)); return E_NOINTERFACE; }
IUnknown_AddRef((IUnknown*)*ppv); return S_OK;}
static ULONG WINAPI ITSProtocol_AddRef(IUnknown *iface){ ITSProtocol *This = impl_from_IUnknown(iface); LONG ref = InterlockedIncrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); return ref;}
static ULONG WINAPI ITSProtocol_Release(IUnknown *iface){ ITSProtocol *This = impl_from_IUnknown(iface); LONG ref = InterlockedDecrement(&This->ref);
TRACE("(%p) ref=%d\n", This, ref);
if(!ref) { release_chm(This); HeapFree(GetProcessHeap(), 0, This);
ITSS_UnlockModule(); }
return ref;}
static const IUnknownVtbl ITSProtocolUnkVtbl = { ITSProtocol_QueryInterface, ITSProtocol_AddRef, ITSProtocol_Release};
static HRESULT WINAPI ITSInternetProtocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv){ ITSProtocol *This = impl_from_IInternetProtocol(iface); return IUnknown_QueryInterface(This->outer, riid, ppv);}
static ULONG WINAPI ITSInternetProtocol_AddRef(IInternetProtocol *iface){ ITSProtocol *This = impl_from_IInternetProtocol(iface); return IUnknown_AddRef(This->outer);}
static ULONG WINAPI ITSInternetProtocol_Release(IInternetProtocol *iface){ ITSProtocol *This = impl_from_IInternetProtocol(iface); return IUnknown_Release(This->outer);}
static LPCWSTR skip_schema(LPCWSTR url){ static const WCHAR its_schema[] = {'i','t','s',':'}; static const WCHAR msits_schema[] = {'m','s','-','i','t','s',':'}; static const WCHAR mk_schema[] = {'m','k',':','@','M','S','I','T','S','t','o','r','e',':'};
if(!_wcsnicmp(its_schema, url, ARRAY_SIZE(its_schema))) return url + ARRAY_SIZE(its_schema); if(!_wcsnicmp(msits_schema, url, ARRAY_SIZE(msits_schema))) return url + ARRAY_SIZE(msits_schema); if(!_wcsnicmp(mk_schema, url, ARRAY_SIZE(mk_schema))) return url + ARRAY_SIZE(mk_schema);
return NULL;}
/* Adopted from urlmon */static void remove_dot_segments(WCHAR *path) { const WCHAR *in = path; WCHAR *out = path;
while(1) { /* Move the first path segment in the input buffer to the end of * the output buffer, and any subsequent characters up to, including * the next "/" character (if any) or the end of the input buffer. */ while(*in != '/') { if(!(*out++ = *in++)) return; }
*out++ = *in++;
while(*in) { if(*in != '.') break;
/* Handle ending "/." */ if(!in[1]) { ++in; break; }
/* Handle "/./" */ if(in[1] == '/') { in += 2; continue; }
/* If we don't have "/../" or ending "/.." */ if(in[1] != '.' || (in[2] && in[2] != '/')) break;
in += *in ? 3 : 2;
/* Find the slash preceding out pointer and move out pointer to it */ if(out > path+1 && *--out == '/') --out; while(out > path && *(--out) != '/'); if(*out == '/') ++out; } }}
static HRESULT report_result(IInternetProtocolSink *sink, HRESULT hres){ IInternetProtocolSink_ReportResult(sink, hres, 0, NULL); return hres;}
static HRESULT WINAPI ITSProtocol_Start(IInternetProtocol *iface, LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved){ ITSProtocol *This = impl_from_IInternetProtocol(iface); BINDINFO bindinfo; DWORD bindf = 0, len; LPWSTR file_name, mime, object_name, p; LPCWSTR ptr; struct chmFile *chm_file; struct chmUnitInfo chm_object; int res; HRESULT hres;
static const WCHAR separator[] = {':',':',0};
TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink, pOIBindInfo, grfPI, dwReserved);
ptr = skip_schema(szUrl); if(!ptr) return INET_E_USE_DEFAULT_PROTOCOLHANDLER;
memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(BINDINFO); hres = IInternetBindInfo_GetBindInfo(pOIBindInfo, &bindf, &bindinfo); if(FAILED(hres)) { WARN("GetBindInfo failed: %08x\n", hres); return hres; }
ReleaseBindInfo(&bindinfo);
len = lstrlenW(ptr)+3; file_name = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR)); memcpy(file_name, ptr, len*sizeof(WCHAR)); hres = UrlUnescapeW(file_name, NULL, &len, URL_UNESCAPE_INPLACE); if(FAILED(hres)) { WARN("UrlUnescape failed: %08x\n", hres); HeapFree(GetProcessHeap(), 0, file_name); return hres; }
p = wcsstr(file_name, separator); if(!p) { WARN("invalid url\n"); HeapFree(GetProcessHeap(), 0, file_name); return report_result(pOIProtSink, STG_E_FILENOTFOUND); }
*p = 0; chm_file = chm_openW(file_name); if(!chm_file) { WARN("Could not open chm file\n"); HeapFree(GetProcessHeap(), 0, file_name); return report_result(pOIProtSink, STG_E_FILENOTFOUND); }
object_name = p+2; len = lstrlenW(object_name);
if(*object_name != '/' && *object_name != '\\') { memmove(object_name+1, object_name, (len+1)*sizeof(WCHAR)); *object_name = '/'; len++; }
if(object_name[len-1] == '/') object_name[--len] = 0;
for(p=object_name; *p; p++) { if(*p == '\\') *p = '/'; }
remove_dot_segments(object_name);
TRACE("Resolving %s\n", debugstr_w(object_name));
memset(&chm_object, 0, sizeof(chm_object)); res = chm_resolve_object(chm_file, object_name, &chm_object); if(res != CHM_RESOLVE_SUCCESS) { WARN("Could not resolve chm object\n"); HeapFree(GetProcessHeap(), 0, file_name); chm_close(chm_file); return report_result(pOIProtSink, STG_E_FILENOTFOUND); }
IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_SENDINGREQUEST, wcsrchr(object_name, '/')+1);
/* FIXME: Native doesn't use FindMimeFromData */ hres = FindMimeFromData(NULL, object_name, NULL, 0, NULL, 0, &mime, 0); HeapFree(GetProcessHeap(), 0, file_name); if(SUCCEEDED(hres)) { IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_MIMETYPEAVAILABLE, mime); CoTaskMemFree(mime); }
release_chm(This); /* Native leaks handle here */ This->chm_file = chm_file; This->chm_object = chm_object;
hres = IInternetProtocolSink_ReportData(pOIProtSink, BSCF_FIRSTDATANOTIFICATION|BSCF_DATAFULLYAVAILABLE, chm_object.length, chm_object.length); if(FAILED(hres)) { WARN("ReportData failed: %08x\n", hres); release_chm(This); return report_result(pOIProtSink, hres); }
hres = IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_BEGINDOWNLOADDATA, NULL);
return report_result(pOIProtSink, hres);}
static HRESULT WINAPI ITSProtocol_Continue(IInternetProtocol *iface, PROTOCOLDATA *pProtocolData){ ITSProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)->(%p)\n", This, pProtocolData); return E_NOTIMPL;}
static HRESULT WINAPI ITSProtocol_Abort(IInternetProtocol *iface, HRESULT hrReason, DWORD dwOptions){ ITSProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions); return E_NOTIMPL;}
static HRESULT WINAPI ITSProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions){ ITSProtocol *This = impl_from_IInternetProtocol(iface);
TRACE("(%p)->(%08x)\n", This, dwOptions);
return S_OK;}
static HRESULT WINAPI ITSProtocol_Suspend(IInternetProtocol *iface){ ITSProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)\n", This); return E_NOTIMPL;}
static HRESULT WINAPI ITSProtocol_Resume(IInternetProtocol *iface){ ITSProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)\n", This); return E_NOTIMPL;}
static HRESULT WINAPI ITSProtocol_Read(IInternetProtocol *iface, void *pv, ULONG cb, ULONG *pcbRead){ ITSProtocol *This = impl_from_IInternetProtocol(iface);
TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
if(!This->chm_file) return INET_E_DATA_NOT_AVAILABLE;
*pcbRead = chm_retrieve_object(This->chm_file, &This->chm_object, pv, This->offset, cb); This->offset += *pcbRead;
return *pcbRead ? S_OK : S_FALSE;}
static HRESULT WINAPI ITSProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition){ ITSProtocol *This = impl_from_IInternetProtocol(iface); FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition); return E_NOTIMPL;}
static HRESULT WINAPI ITSProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions){ ITSProtocol *This = impl_from_IInternetProtocol(iface);
TRACE("(%p)->(%08x)\n", This, dwOptions);
return S_OK;}
static HRESULT WINAPI ITSProtocol_UnlockRequest(IInternetProtocol *iface){ ITSProtocol *This = impl_from_IInternetProtocol(iface);
TRACE("(%p)\n", This);
return S_OK;}
static const IInternetProtocolVtbl ITSProtocolVtbl = { ITSInternetProtocol_QueryInterface, ITSInternetProtocol_AddRef, ITSInternetProtocol_Release, ITSProtocol_Start, ITSProtocol_Continue, ITSProtocol_Abort, ITSProtocol_Terminate, ITSProtocol_Suspend, ITSProtocol_Resume, ITSProtocol_Read, ITSProtocol_Seek, ITSProtocol_LockRequest, ITSProtocol_UnlockRequest};
static HRESULT WINAPI ITSProtocolInfo_QueryInterface(IInternetProtocolInfo *iface, REFIID riid, void **ppv){ ITSProtocol *This = impl_from_IInternetProtocolInfo(iface); return IInternetProtocol_QueryInterface(&This->IInternetProtocol_iface, riid, ppv);}
static ULONG WINAPI ITSProtocolInfo_AddRef(IInternetProtocolInfo *iface){ ITSProtocol *This = impl_from_IInternetProtocolInfo(iface); return IInternetProtocol_AddRef(&This->IInternetProtocol_iface);}
static ULONG WINAPI ITSProtocolInfo_Release(IInternetProtocolInfo *iface){ ITSProtocol *This = impl_from_IInternetProtocolInfo(iface); return IInternetProtocol_Release(&This->IInternetProtocol_iface);}
static HRESULT WINAPI ITSProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl, PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult, DWORD *pcchResult, DWORD dwReserved){ ITSProtocol *This = impl_from_IInternetProtocolInfo(iface);
TRACE("(%p)->(%s %x %08x %p %d %p %d)\n", This, debugstr_w(pwzUrl), ParseAction, dwParseFlags, pwzResult, cchResult, pcchResult, dwReserved);
switch(ParseAction) { case PARSE_CANONICALIZE: FIXME("PARSE_CANONICALIZE\n"); return E_NOTIMPL; case PARSE_SECURITY_URL: FIXME("PARSE_SECURITY_URL\n"); return E_NOTIMPL; default: return INET_E_DEFAULT_ACTION; }
return S_OK;}
static HRESULT WINAPI ITSProtocolInfo_CombineUrl(IInternetProtocolInfo *iface, LPCWSTR pwzBaseUrl, LPCWSTR pwzRelativeUrl, DWORD dwCombineFlags, LPWSTR pwzResult, DWORD cchResult, DWORD* pcchResult, DWORD dwReserved){ ITSProtocol *This = impl_from_IInternetProtocolInfo(iface); LPCWSTR base_end, ptr; DWORD rel_len;
static const WCHAR separator[] = {':',':',0};
TRACE("(%p)->(%s %s %08x %p %d %p %d)\n", This, debugstr_w(pwzBaseUrl), debugstr_w(pwzRelativeUrl), dwCombineFlags, pwzResult, cchResult, pcchResult, dwReserved);
base_end = wcsstr(pwzBaseUrl, separator); if(!base_end) return 0x80041001; base_end += 2;
if(!skip_schema(pwzBaseUrl)) return INET_E_USE_DEFAULT_PROTOCOLHANDLER;
if(wcschr(pwzRelativeUrl, ':')) return STG_E_INVALIDNAME;
if(pwzRelativeUrl[0] == '#') { base_end += lstrlenW(base_end); }else if(pwzRelativeUrl[0] != '/') { ptr = wcsrchr(base_end, '/'); if(ptr) base_end = ptr+1; else base_end += lstrlenW(base_end); }
rel_len = lstrlenW(pwzRelativeUrl)+1;
*pcchResult = rel_len + (base_end-pwzBaseUrl);
if(*pcchResult > cchResult) return E_OUTOFMEMORY;
memcpy(pwzResult, pwzBaseUrl, (base_end-pwzBaseUrl)*sizeof(WCHAR)); lstrcpyW(pwzResult + (base_end-pwzBaseUrl), pwzRelativeUrl);
return S_OK;}
static HRESULT WINAPI ITSProtocolInfo_CompareUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags){ ITSProtocol *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 ITSProtocolInfo_QueryInfo(IInternetProtocolInfo *iface, LPCWSTR pwzUrl, QUERYOPTION QueryOption, DWORD dwQueryFlags, LPVOID pBuffer, DWORD cbBuffer, DWORD* pcbBuf, DWORD dwReserved){ ITSProtocol *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 E_NOTIMPL;}
static const IInternetProtocolInfoVtbl ITSProtocolInfoVtbl = { ITSProtocolInfo_QueryInterface, ITSProtocolInfo_AddRef, ITSProtocolInfo_Release, ITSProtocolInfo_ParseUrl, ITSProtocolInfo_CombineUrl, ITSProtocolInfo_CompareUrl, ITSProtocolInfo_QueryInfo};
HRESULT ITSProtocol_create(IUnknown *outer, void **ppv){ ITSProtocol *ret;
TRACE("(%p %p)\n", outer, ppv);
ITSS_LockModule();
ret = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ITSProtocol)); if(!ret) return E_OUTOFMEMORY;
ret->IUnknown_inner.lpVtbl = &ITSProtocolUnkVtbl; ret->IInternetProtocol_iface.lpVtbl = &ITSProtocolVtbl; ret->IInternetProtocolInfo_iface.lpVtbl = &ITSProtocolInfoVtbl; ret->ref = 1; ret->outer = outer ? outer : &ret->IUnknown_inner;
*ppv = &ret->IUnknown_inner; return S_OK;}