urlmon: Get rid of URLMonikerImpl_BindToStorage_hack.
This commit is contained in:
Jacek Caban
Alexandre Julliard
parent
0da20926c3
commit
96a34b586f
|
|
@ -33,251 +33,6 @@
|
||||||
|
|
||||||
WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
|
WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
|
||||||
|
|
||||||
/* native urlmon.dll uses this key, too */
|
|
||||||
static WCHAR BSCBHolder[] = { '_','B','S','C','B','_','H','o','l','d','e','r','_',0 };
|
|
||||||
|
|
||||||
/*static BOOL registered_wndclass = FALSE;*/
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
const IBindingVtbl *lpVtbl;
|
|
||||||
|
|
||||||
LONG ref;
|
|
||||||
|
|
||||||
LPWSTR URLName;
|
|
||||||
|
|
||||||
HWND hwndCallback;
|
|
||||||
IBindCtx *pBC;
|
|
||||||
HINTERNET hinternet, hconnect, hrequest;
|
|
||||||
HANDLE hCacheFile;
|
|
||||||
IUMCacheStream *pstrCache;
|
|
||||||
IBindStatusCallback *pbscb;
|
|
||||||
DWORD total_read, expected_size;
|
|
||||||
} Binding;
|
|
||||||
|
|
||||||
static HRESULT WINAPI Binding_QueryInterface(IBinding* iface, REFIID riid, void **ppvObject)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)->(%s,%p)\n", This, debugstr_guid(riid), ppvObject);
|
|
||||||
|
|
||||||
if((This == NULL) || (ppvObject == NULL))
|
|
||||||
return E_INVALIDARG;
|
|
||||||
|
|
||||||
if (IsEqualIID(&IID_IUnknown, riid) || IsEqualIID(&IID_IBinding, riid)) {
|
|
||||||
*ppvObject = iface;
|
|
||||||
IBinding_AddRef(iface);
|
|
||||||
return S_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
*ppvObject = NULL;
|
|
||||||
return E_NOINTERFACE;
|
|
||||||
}
|
|
||||||
|
|
||||||
static ULONG WINAPI Binding_AddRef(IBinding* iface)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
ULONG ref = InterlockedIncrement(&This->ref);
|
|
||||||
|
|
||||||
TRACE("(%p) ref=%d\n", This, ref);
|
|
||||||
|
|
||||||
return ref;
|
|
||||||
}
|
|
||||||
|
|
||||||
static ULONG WINAPI Binding_Release(IBinding* iface)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
ULONG ref = InterlockedDecrement(&This->ref);
|
|
||||||
|
|
||||||
TRACE("(%p) ref=%d\n",This, ref);
|
|
||||||
|
|
||||||
if(!ref) {
|
|
||||||
heap_free(This->URLName);
|
|
||||||
if (This->hCacheFile)
|
|
||||||
CloseHandle(This->hCacheFile);
|
|
||||||
if (This->pstrCache)
|
|
||||||
{
|
|
||||||
UMCloseCacheFileStream(This->pstrCache);
|
|
||||||
IStream_Release((IStream *)This->pstrCache);
|
|
||||||
}
|
|
||||||
if (This->pbscb)
|
|
||||||
IBindStatusCallback_Release(This->pbscb);
|
|
||||||
|
|
||||||
heap_free(This);
|
|
||||||
|
|
||||||
URLMON_UnlockModule();
|
|
||||||
}
|
|
||||||
|
|
||||||
return ref;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI Binding_Abort(IBinding* iface)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
|
|
||||||
FIXME("(%p): stub\n", This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI Binding_GetBindResult(IBinding* iface, CLSID* pclsidProtocol, DWORD* pdwResult, LPOLESTR* pszResult, DWORD* pdwReserved)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
|
|
||||||
FIXME("(%p)->(%p, %p, %p, %p): stub\n", This, pclsidProtocol, pdwResult, pszResult, pdwReserved);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI Binding_GetPriority(IBinding* iface, LONG* pnPriority)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
|
|
||||||
FIXME("(%p)->(%p): stub\n", This, pnPriority);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI Binding_Resume(IBinding* iface)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
|
|
||||||
FIXME("(%p): stub\n", This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI Binding_SetPriority(IBinding* iface, LONG nPriority)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
|
|
||||||
FIXME("(%p)->(%d): stub\n", This, nPriority);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI Binding_Suspend(IBinding* iface)
|
|
||||||
{
|
|
||||||
Binding *This = (Binding*)iface;
|
|
||||||
|
|
||||||
FIXME("(%p): stub\n", This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void Binding_CloseCacheDownload(Binding *This)
|
|
||||||
{
|
|
||||||
CloseHandle(This->hCacheFile);
|
|
||||||
This->hCacheFile = 0;
|
|
||||||
UMCloseCacheFileStream(This->pstrCache);
|
|
||||||
IStream_Release((IStream *)This->pstrCache);
|
|
||||||
This->pstrCache = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT Binding_MoreCacheData(Binding *This, const char *buf, DWORD dwBytes)
|
|
||||||
{
|
|
||||||
DWORD written;
|
|
||||||
|
|
||||||
if (WriteFile(This->hCacheFile, buf, dwBytes, &written, NULL) && written == dwBytes)
|
|
||||||
{
|
|
||||||
HRESULT hr;
|
|
||||||
|
|
||||||
This->total_read += written;
|
|
||||||
hr = IBindStatusCallback_OnProgress(This->pbscb,
|
|
||||||
This->total_read + written,
|
|
||||||
This->expected_size,
|
|
||||||
(This->total_read == written) ?
|
|
||||||
BINDSTATUS_BEGINDOWNLOADDATA :
|
|
||||||
BINDSTATUS_DOWNLOADINGDATA,
|
|
||||||
This->URLName);
|
|
||||||
if (hr == S_OK)
|
|
||||||
{
|
|
||||||
STGMEDIUM stg;
|
|
||||||
FORMATETC fmt;
|
|
||||||
|
|
||||||
fmt.cfFormat = 0;
|
|
||||||
fmt.ptd = NULL;
|
|
||||||
fmt.dwAspect = 0;
|
|
||||||
fmt.lindex = -1;
|
|
||||||
fmt.tymed = TYMED_ISTREAM;
|
|
||||||
|
|
||||||
stg.tymed = TYMED_ISTREAM;
|
|
||||||
stg.u.pstm = (IStream *)This->pstrCache;
|
|
||||||
stg.pUnkForRelease = NULL;
|
|
||||||
|
|
||||||
hr = IBindStatusCallback_OnDataAvailable(This->pbscb,
|
|
||||||
(This->total_read == written) ?
|
|
||||||
BSCF_FIRSTDATANOTIFICATION :
|
|
||||||
BSCF_INTERMEDIATEDATANOTIFICATION,
|
|
||||||
This->total_read + written,
|
|
||||||
&fmt,
|
|
||||||
&stg);
|
|
||||||
}
|
|
||||||
if (written < dwBytes)
|
|
||||||
return STG_E_MEDIUMFULL;
|
|
||||||
else
|
|
||||||
return hr;
|
|
||||||
}
|
|
||||||
return HRESULT_FROM_WIN32(GetLastError());
|
|
||||||
}
|
|
||||||
|
|
||||||
static void Binding_FinishedDownload(Binding *This, HRESULT hr)
|
|
||||||
{
|
|
||||||
STGMEDIUM stg;
|
|
||||||
FORMATETC fmt;
|
|
||||||
|
|
||||||
fmt.ptd = NULL;
|
|
||||||
fmt.dwAspect = 0;
|
|
||||||
fmt.lindex = -1;
|
|
||||||
fmt.tymed = TYMED_ISTREAM;
|
|
||||||
|
|
||||||
stg.tymed = TYMED_ISTREAM;
|
|
||||||
stg.u.pstm = (IStream *)This->pstrCache;
|
|
||||||
stg.pUnkForRelease = NULL;
|
|
||||||
|
|
||||||
IBindStatusCallback_OnProgress(This->pbscb, This->total_read, This->expected_size,
|
|
||||||
BINDSTATUS_ENDDOWNLOADDATA, This->URLName);
|
|
||||||
IBindStatusCallback_OnDataAvailable(This->pbscb, BSCF_LASTDATANOTIFICATION, This->total_read, &fmt, &stg);
|
|
||||||
if (hr != S_OK)
|
|
||||||
{
|
|
||||||
WCHAR *pwchError = 0;
|
|
||||||
|
|
||||||
FormatMessageW (FORMAT_MESSAGE_FROM_SYSTEM |
|
|
||||||
FORMAT_MESSAGE_ALLOCATE_BUFFER,
|
|
||||||
NULL, (DWORD) hr,
|
|
||||||
0, (LPWSTR) &pwchError,
|
|
||||||
0, NULL);
|
|
||||||
if (!pwchError)
|
|
||||||
{
|
|
||||||
static const WCHAR achFormat[] = { '%', '0', '8', 'x', 0 };
|
|
||||||
|
|
||||||
pwchError =(WCHAR *) LocalAlloc(LMEM_FIXED, sizeof(WCHAR) * 9);
|
|
||||||
wsprintfW(pwchError, achFormat, hr);
|
|
||||||
}
|
|
||||||
IBindStatusCallback_OnStopBinding(This->pbscb, hr, pwchError);
|
|
||||||
LocalFree(pwchError);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
IBindStatusCallback_OnStopBinding(This->pbscb, hr, NULL);
|
|
||||||
}
|
|
||||||
IBindStatusCallback_Release(This->pbscb);
|
|
||||||
This->pbscb = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static const IBindingVtbl BindingVtbl =
|
|
||||||
{
|
|
||||||
Binding_QueryInterface,
|
|
||||||
Binding_AddRef,
|
|
||||||
Binding_Release,
|
|
||||||
Binding_Abort,
|
|
||||||
Binding_Suspend,
|
|
||||||
Binding_Resume,
|
|
||||||
Binding_SetPriority,
|
|
||||||
Binding_GetPriority,
|
|
||||||
Binding_GetBindResult
|
|
||||||
};
|
|
||||||
|
|
||||||
/* filemoniker data structure */
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
|
||||||
const IMonikerVtbl* lpvtbl; /* VTable relative to the IMoniker interface.*/
|
const IMonikerVtbl* lpvtbl; /* VTable relative to the IMoniker interface.*/
|
||||||
|
|
@ -493,158 +248,6 @@ static HRESULT WINAPI URLMonikerImpl_BindToObject(IMoniker* iface,
|
||||||
/******************************************************************************
|
/******************************************************************************
|
||||||
* URLMoniker_BindToStorage
|
* URLMoniker_BindToStorage
|
||||||
******************************************************************************/
|
******************************************************************************/
|
||||||
static HRESULT URLMonikerImpl_BindToStorage_hack(LPCWSTR URLName, IBindCtx* pbc, VOID** ppvObject)
|
|
||||||
{
|
|
||||||
HRESULT hres;
|
|
||||||
BINDINFO bi;
|
|
||||||
DWORD bindf;
|
|
||||||
WCHAR szFileName[MAX_PATH + 1];
|
|
||||||
Binding *bind;
|
|
||||||
int len;
|
|
||||||
|
|
||||||
WARN("(%s %p %p)\n", debugstr_w(URLName), pbc, ppvObject);
|
|
||||||
|
|
||||||
bind = heap_alloc_zero(sizeof(Binding));
|
|
||||||
bind->lpVtbl = &BindingVtbl;
|
|
||||||
bind->ref = 1;
|
|
||||||
URLMON_LockModule();
|
|
||||||
|
|
||||||
len = lstrlenW(URLName)+1;
|
|
||||||
bind->URLName = heap_alloc(len*sizeof(WCHAR));
|
|
||||||
memcpy(bind->URLName, URLName, len*sizeof(WCHAR));
|
|
||||||
|
|
||||||
hres = UMCreateStreamOnCacheFile(bind->URLName, 0, szFileName, &bind->hCacheFile, &bind->pstrCache);
|
|
||||||
|
|
||||||
if(SUCCEEDED(hres)) {
|
|
||||||
TRACE("Created stream...\n");
|
|
||||||
|
|
||||||
*ppvObject = (void *) bind->pstrCache;
|
|
||||||
IStream_AddRef((IStream *) bind->pstrCache);
|
|
||||||
|
|
||||||
hres = IBindCtx_GetObjectParam(pbc, BSCBHolder, (IUnknown**)&bind->pbscb);
|
|
||||||
if(SUCCEEDED(hres)) {
|
|
||||||
TRACE("Got IBindStatusCallback...\n");
|
|
||||||
|
|
||||||
memset(&bi, 0, sizeof(bi));
|
|
||||||
bi.cbSize = sizeof(bi);
|
|
||||||
bindf = 0;
|
|
||||||
hres = IBindStatusCallback_GetBindInfo(bind->pbscb, &bindf, &bi);
|
|
||||||
if(SUCCEEDED(hres)) {
|
|
||||||
URL_COMPONENTSW url;
|
|
||||||
WCHAR *host, *path, *user, *pass;
|
|
||||||
|
|
||||||
TRACE("got bindinfo. bindf = %08x extrainfo = %s bindinfof = %08x bindverb = %08x iid %s\n",
|
|
||||||
bindf, debugstr_w(bi.szExtraInfo), bi.grfBindInfoF, bi.dwBindVerb, debugstr_guid(&bi.iid));
|
|
||||||
hres = IBindStatusCallback_OnStartBinding(bind->pbscb, 0, (IBinding*)bind);
|
|
||||||
TRACE("OnStartBinding rets %08x\n", hres);
|
|
||||||
|
|
||||||
bind->expected_size = 0;
|
|
||||||
bind->total_read = 0;
|
|
||||||
|
|
||||||
memset(&url, 0, sizeof(url));
|
|
||||||
url.dwStructSize = sizeof(url);
|
|
||||||
url.dwSchemeLength = url.dwHostNameLength = url.dwUrlPathLength = url.dwUserNameLength = url.dwPasswordLength = 1;
|
|
||||||
InternetCrackUrlW(URLName, 0, ICU_ESCAPE, &url);
|
|
||||||
host = heap_alloc((url.dwHostNameLength + 1) * sizeof(WCHAR));
|
|
||||||
memcpy(host, url.lpszHostName, url.dwHostNameLength * sizeof(WCHAR));
|
|
||||||
host[url.dwHostNameLength] = '\0';
|
|
||||||
path = heap_alloc((url.dwUrlPathLength + 1) * sizeof(WCHAR));
|
|
||||||
memcpy(path, url.lpszUrlPath, url.dwUrlPathLength * sizeof(WCHAR));
|
|
||||||
path[url.dwUrlPathLength] = '\0';
|
|
||||||
if (url.dwUserNameLength)
|
|
||||||
{
|
|
||||||
user = heap_alloc(((url.dwUserNameLength + 1) * sizeof(WCHAR)));
|
|
||||||
memcpy(user, url.lpszUserName, url.dwUserNameLength * sizeof(WCHAR));
|
|
||||||
user[url.dwUserNameLength] = 0;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
user = 0;
|
|
||||||
}
|
|
||||||
if (url.dwPasswordLength)
|
|
||||||
{
|
|
||||||
pass = heap_alloc(((url.dwPasswordLength + 1) * sizeof(WCHAR)));
|
|
||||||
memcpy(pass, url.lpszPassword, url.dwPasswordLength * sizeof(WCHAR));
|
|
||||||
pass[url.dwPasswordLength] = 0;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
pass = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
do {
|
|
||||||
bind->hinternet = InternetOpenA("User Agent", 0, NULL, NULL, 0);
|
|
||||||
if (!bind->hinternet)
|
|
||||||
{
|
|
||||||
hres = HRESULT_FROM_WIN32(GetLastError());
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!url.nPort)
|
|
||||||
url.nPort = INTERNET_DEFAULT_GOPHER_PORT;
|
|
||||||
bind->hconnect = InternetConnectW(bind->hinternet, host, url.nPort, user, pass,
|
|
||||||
INTERNET_SERVICE_GOPHER, 0, (DWORD_PTR)bind);
|
|
||||||
if (!bind->hconnect)
|
|
||||||
{
|
|
||||||
hres = HRESULT_FROM_WIN32(GetLastError());
|
|
||||||
CloseHandle(bind->hinternet);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
hres = IBindStatusCallback_OnProgress(bind->pbscb, 0, 0, 0x22, NULL);
|
|
||||||
hres = IBindStatusCallback_OnProgress(bind->pbscb, 0, 0, BINDSTATUS_FINDINGRESOURCE, NULL);
|
|
||||||
hres = IBindStatusCallback_OnProgress(bind->pbscb, 0, 0, BINDSTATUS_CONNECTING, NULL);
|
|
||||||
hres = IBindStatusCallback_OnProgress(bind->pbscb, 0, 0, BINDSTATUS_SENDINGREQUEST, NULL);
|
|
||||||
|
|
||||||
bind->hrequest = GopherOpenFileW(bind->hconnect,
|
|
||||||
path,
|
|
||||||
0,
|
|
||||||
INTERNET_FLAG_RELOAD,
|
|
||||||
0);
|
|
||||||
if (bind->hrequest)
|
|
||||||
{
|
|
||||||
TRACE("res = %d gle = %u url len = %d\n", hres, GetLastError(), bind->expected_size);
|
|
||||||
|
|
||||||
IBindStatusCallback_OnProgress(bind->pbscb, 0, 0, BINDSTATUS_CACHEFILENAMEAVAILABLE, szFileName);
|
|
||||||
|
|
||||||
while(1) {
|
|
||||||
char buf[4096];
|
|
||||||
DWORD bufread;
|
|
||||||
if(InternetReadFile(bind->hrequest, buf, sizeof(buf), &bufread)) {
|
|
||||||
TRACE("read %d bytes %s...\n", bufread, debugstr_an(buf, 10));
|
|
||||||
if(bufread == 0) break;
|
|
||||||
hres = Binding_MoreCacheData(bind, buf, bufread);
|
|
||||||
} else
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
InternetCloseHandle(bind->hrequest);
|
|
||||||
hres = S_OK;
|
|
||||||
}else {
|
|
||||||
hres = HRESULT_FROM_WIN32(GetLastError());
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
InternetCloseHandle(bind->hconnect);
|
|
||||||
InternetCloseHandle(bind->hinternet);
|
|
||||||
} while(0);
|
|
||||||
|
|
||||||
Binding_FinishedDownload(bind, hres);
|
|
||||||
Binding_CloseCacheDownload(bind);
|
|
||||||
|
|
||||||
heap_free(user);
|
|
||||||
heap_free(pass);
|
|
||||||
heap_free(path);
|
|
||||||
heap_free(host);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
IBinding_Release((IBinding*)bind);
|
|
||||||
|
|
||||||
return hres;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI URLMonikerImpl_BindToStorage(IMoniker* iface,
|
static HRESULT WINAPI URLMonikerImpl_BindToStorage(IMoniker* iface,
|
||||||
IBindCtx* pbc,
|
IBindCtx* pbc,
|
||||||
IMoniker* pmkToLeft,
|
IMoniker* pmkToLeft,
|
||||||
|
|
@ -652,26 +255,12 @@ static HRESULT WINAPI URLMonikerImpl_BindToStorage(IMoniker* iface,
|
||||||
VOID** ppvObject)
|
VOID** ppvObject)
|
||||||
{
|
{
|
||||||
URLMonikerImpl *This = (URLMonikerImpl*)iface;
|
URLMonikerImpl *This = (URLMonikerImpl*)iface;
|
||||||
WCHAR schema[64];
|
|
||||||
BOOL bret;
|
|
||||||
|
|
||||||
URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW), schema,
|
TRACE("(%p)->(%p %p %s %p)\n", This, pbc, pmkToLeft, debugstr_guid(riid), ppvObject);
|
||||||
sizeof(schema)/sizeof(WCHAR), 0, NULL, 0, 0, NULL, 0, NULL, 0, NULL, 0, NULL, 0};
|
|
||||||
|
|
||||||
if(pmkToLeft)
|
if(pmkToLeft)
|
||||||
FIXME("Unsupported pmkToLeft\n");
|
FIXME("Unsupported pmkToLeft\n");
|
||||||
|
|
||||||
bret = InternetCrackUrlW(This->URLName, 0, ICU_ESCAPE, &url);
|
|
||||||
if(!bret) {
|
|
||||||
ERR("InternetCrackUrl failed: %u\n", GetLastError());
|
|
||||||
return E_FAIL;
|
|
||||||
}
|
|
||||||
|
|
||||||
if(IsEqualGUID(&IID_IStream, riid) && url.nScheme == INTERNET_SCHEME_GOPHER)
|
|
||||||
return URLMonikerImpl_BindToStorage_hack(This->URLName, pbc, ppvObject);
|
|
||||||
|
|
||||||
TRACE("(%p)->(%p %p %s %p)\n", This, pbc, pmkToLeft, debugstr_guid(riid), ppvObject);
|
|
||||||
|
|
||||||
return bind_to_storage(This->URLName, pbc, riid, ppvObject);
|
return bind_to_storage(This->URLName, pbc, riid, ppvObject);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -30,342 +30,6 @@
|
||||||
|
|
||||||
WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
|
WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
|
||||||
|
|
||||||
static const IStreamVtbl stvt;
|
|
||||||
|
|
||||||
HRESULT UMCreateStreamOnCacheFile(LPCWSTR pszURL,
|
|
||||||
DWORD dwSize,
|
|
||||||
LPWSTR pszFileName,
|
|
||||||
HANDLE *phfile,
|
|
||||||
IUMCacheStream **ppstr)
|
|
||||||
{
|
|
||||||
IUMCacheStream* ucstr;
|
|
||||||
HANDLE handle;
|
|
||||||
DWORD size;
|
|
||||||
LPWSTR url, c, ext = NULL;
|
|
||||||
HRESULT hr;
|
|
||||||
|
|
||||||
size = (strlenW(pszURL)+1)*sizeof(WCHAR);
|
|
||||||
url = heap_alloc(size);
|
|
||||||
memcpy(url, pszURL, size);
|
|
||||||
|
|
||||||
for (c = url; *c && *c != '#' && *c != '?'; ++c)
|
|
||||||
{
|
|
||||||
if (*c == '.')
|
|
||||||
ext = c+1;
|
|
||||||
else if(*c == '/')
|
|
||||||
ext = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
*c = 0;
|
|
||||||
|
|
||||||
if(!CreateUrlCacheEntryW(url, dwSize, ext, pszFileName, 0))
|
|
||||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
|
||||||
else
|
|
||||||
hr = S_OK;
|
|
||||||
|
|
||||||
heap_free(url);
|
|
||||||
|
|
||||||
if (hr != S_OK)
|
|
||||||
return hr;
|
|
||||||
|
|
||||||
TRACE("Opening %s\n", debugstr_w(pszFileName) );
|
|
||||||
|
|
||||||
handle = CreateFileW( pszFileName, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, NULL );
|
|
||||||
if( handle == INVALID_HANDLE_VALUE )
|
|
||||||
return HRESULT_FROM_WIN32(GetLastError());
|
|
||||||
|
|
||||||
if (phfile)
|
|
||||||
{
|
|
||||||
/* Call CreateFileW again because we need a handle with its own file pointer, and DuplicateHandle will return
|
|
||||||
* a handle that shares its file pointer with the original.
|
|
||||||
*/
|
|
||||||
*phfile = CreateFileW( pszFileName, GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL );
|
|
||||||
|
|
||||||
if (*phfile == (HANDLE) HFILE_ERROR)
|
|
||||||
{
|
|
||||||
DWORD dwError = GetLastError();
|
|
||||||
|
|
||||||
CloseHandle(handle);
|
|
||||||
return HRESULT_FROM_WIN32(dwError);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
ucstr = heap_alloc_zero(sizeof(IUMCacheStream));
|
|
||||||
if(ucstr)
|
|
||||||
{
|
|
||||||
ucstr->pszURL = heap_alloc_zero(sizeof(WCHAR) * (lstrlenW(pszURL) + 1));
|
|
||||||
if (ucstr->pszURL)
|
|
||||||
{
|
|
||||||
ucstr->pszFileName = heap_alloc_zero(sizeof(WCHAR) * (lstrlenW(pszFileName) + 1));
|
|
||||||
if (ucstr->pszFileName)
|
|
||||||
{
|
|
||||||
ucstr->lpVtbl=&stvt;
|
|
||||||
ucstr->ref = 1;
|
|
||||||
ucstr->handle = handle;
|
|
||||||
ucstr->closed = 0;
|
|
||||||
lstrcpyW(ucstr->pszURL, pszURL);
|
|
||||||
lstrcpyW(ucstr->pszFileName, pszFileName);
|
|
||||||
|
|
||||||
*ppstr = ucstr;
|
|
||||||
|
|
||||||
return S_OK;
|
|
||||||
}
|
|
||||||
heap_free(ucstr->pszURL);
|
|
||||||
}
|
|
||||||
heap_free(ucstr);
|
|
||||||
}
|
|
||||||
CloseHandle(handle);
|
|
||||||
if (phfile)
|
|
||||||
CloseHandle(*phfile);
|
|
||||||
return E_OUTOFMEMORY;
|
|
||||||
}
|
|
||||||
|
|
||||||
void UMCloseCacheFileStream(IUMCacheStream *This)
|
|
||||||
{
|
|
||||||
if (!This->closed)
|
|
||||||
{
|
|
||||||
FILETIME ftZero;
|
|
||||||
|
|
||||||
ftZero.dwLowDateTime = ftZero.dwHighDateTime = 0;
|
|
||||||
|
|
||||||
This->closed = 1;
|
|
||||||
CommitUrlCacheEntryW(This->pszURL,
|
|
||||||
This->pszFileName,
|
|
||||||
ftZero,
|
|
||||||
ftZero,
|
|
||||||
NORMAL_CACHE_ENTRY,
|
|
||||||
0,
|
|
||||||
0,
|
|
||||||
0,
|
|
||||||
0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**************************************************************************
|
|
||||||
* IStream_fnQueryInterface
|
|
||||||
*/
|
|
||||||
static HRESULT WINAPI IStream_fnQueryInterface(IStream *iface,
|
|
||||||
REFIID riid,
|
|
||||||
LPVOID *ppvObj)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)->(\n\tIID:\t%s,%p)\n",This,debugstr_guid(riid),ppvObj);
|
|
||||||
|
|
||||||
*ppvObj = NULL;
|
|
||||||
|
|
||||||
if(IsEqualIID(riid, &IID_IUnknown) ||
|
|
||||||
IsEqualIID(riid, &IID_IStream))
|
|
||||||
{
|
|
||||||
*ppvObj = This;
|
|
||||||
}
|
|
||||||
|
|
||||||
if(*ppvObj)
|
|
||||||
{
|
|
||||||
IStream_AddRef((IStream*)*ppvObj);
|
|
||||||
TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
|
|
||||||
return S_OK;
|
|
||||||
}
|
|
||||||
TRACE("-- Interface: E_NOINTERFACE\n");
|
|
||||||
return E_NOINTERFACE;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**************************************************************************
|
|
||||||
* IStream_fnAddRef
|
|
||||||
*/
|
|
||||||
static ULONG WINAPI IStream_fnAddRef(IStream *iface)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
ULONG refCount = InterlockedIncrement(&This->ref);
|
|
||||||
|
|
||||||
TRACE("(%p)->(count=%u)\n", This, refCount - 1);
|
|
||||||
|
|
||||||
return refCount;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**************************************************************************
|
|
||||||
* IStream_fnRelease
|
|
||||||
*/
|
|
||||||
static ULONG WINAPI IStream_fnRelease(IStream *iface)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
ULONG refCount = InterlockedDecrement(&This->ref);
|
|
||||||
|
|
||||||
TRACE("(%p)->(count=%u)\n", This, refCount + 1);
|
|
||||||
|
|
||||||
if (!refCount)
|
|
||||||
{
|
|
||||||
TRACE(" destroying UMCacheStream (%p)\n",This);
|
|
||||||
UMCloseCacheFileStream(This);
|
|
||||||
CloseHandle(This->handle);
|
|
||||||
heap_free(This->pszFileName);
|
|
||||||
heap_free(This->pszURL);
|
|
||||||
heap_free(This);
|
|
||||||
}
|
|
||||||
return refCount;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI IStream_fnRead (IStream * iface,
|
|
||||||
void* pv,
|
|
||||||
ULONG cb,
|
|
||||||
ULONG* pcbRead)
|
|
||||||
{
|
|
||||||
ULONG dwBytesRead;
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)->(%p,0x%08x,%p)\n",This, pv, cb, pcbRead);
|
|
||||||
|
|
||||||
if ( !pv )
|
|
||||||
return STG_E_INVALIDPOINTER;
|
|
||||||
|
|
||||||
if ( !pcbRead)
|
|
||||||
pcbRead = &dwBytesRead;
|
|
||||||
|
|
||||||
if ( ! ReadFile( This->handle, pv, cb, pcbRead, NULL ) )
|
|
||||||
return S_FALSE;
|
|
||||||
|
|
||||||
if (!*pcbRead)
|
|
||||||
return This->closed ? S_FALSE : E_PENDING;
|
|
||||||
return S_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI IStream_fnWrite (IStream * iface,
|
|
||||||
const void* pv,
|
|
||||||
ULONG cb,
|
|
||||||
ULONG* pcbWritten)
|
|
||||||
{
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI IStream_fnSeek (IStream * iface,
|
|
||||||
LARGE_INTEGER dlibMove,
|
|
||||||
DWORD dwOrigin,
|
|
||||||
ULARGE_INTEGER* plibNewPosition)
|
|
||||||
{
|
|
||||||
LARGE_INTEGER newpos;
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
if (!SetFilePointerEx( This->handle, dlibMove, &newpos, dwOrigin ))
|
|
||||||
return E_FAIL;
|
|
||||||
|
|
||||||
if (plibNewPosition)
|
|
||||||
plibNewPosition->QuadPart = newpos.QuadPart;
|
|
||||||
|
|
||||||
return S_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI IStream_fnSetSize (IStream * iface,
|
|
||||||
ULARGE_INTEGER libNewSize)
|
|
||||||
{
|
|
||||||
LARGE_INTEGER newpos;
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
newpos.QuadPart = libNewSize.QuadPart;
|
|
||||||
if( ! SetFilePointerEx( This->handle, newpos, NULL, FILE_BEGIN ) )
|
|
||||||
return E_FAIL;
|
|
||||||
|
|
||||||
if( ! SetEndOfFile( This->handle ) )
|
|
||||||
return E_FAIL;
|
|
||||||
|
|
||||||
return S_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI IStream_fnCopyTo (IStream * iface,
|
|
||||||
IStream* pstm,
|
|
||||||
ULARGE_INTEGER cb,
|
|
||||||
ULARGE_INTEGER* pcbRead,
|
|
||||||
ULARGE_INTEGER* pcbWritten)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI IStream_fnCommit (IStream * iface,
|
|
||||||
DWORD grfCommitFlags)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static HRESULT WINAPI IStream_fnRevert (IStream * iface)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
static HRESULT WINAPI IStream_fnLockRegion (IStream * iface,
|
|
||||||
ULARGE_INTEGER libOffset,
|
|
||||||
ULARGE_INTEGER cb,
|
|
||||||
DWORD dwLockType)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
static HRESULT WINAPI IStream_fnUnlockRegion (IStream * iface,
|
|
||||||
ULARGE_INTEGER libOffset,
|
|
||||||
ULARGE_INTEGER cb,
|
|
||||||
DWORD dwLockType)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
static HRESULT WINAPI IStream_fnStat (IStream * iface,
|
|
||||||
STATSTG* pstatstg,
|
|
||||||
DWORD grfStatFlag)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
static HRESULT WINAPI IStream_fnClone (IStream * iface,
|
|
||||||
IStream** ppstm)
|
|
||||||
{
|
|
||||||
IUMCacheStream *This = (IUMCacheStream *)iface;
|
|
||||||
|
|
||||||
TRACE("(%p)\n",This);
|
|
||||||
|
|
||||||
return E_NOTIMPL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static const IStreamVtbl stvt =
|
|
||||||
{
|
|
||||||
IStream_fnQueryInterface,
|
|
||||||
IStream_fnAddRef,
|
|
||||||
IStream_fnRelease,
|
|
||||||
IStream_fnRead,
|
|
||||||
IStream_fnWrite,
|
|
||||||
IStream_fnSeek,
|
|
||||||
IStream_fnSetSize,
|
|
||||||
IStream_fnCopyTo,
|
|
||||||
IStream_fnCommit,
|
|
||||||
IStream_fnRevert,
|
|
||||||
IStream_fnLockRegion,
|
|
||||||
IStream_fnUnlockRegion,
|
|
||||||
IStream_fnStat,
|
|
||||||
IStream_fnClone
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
typedef struct ProxyBindStatusCallback
|
typedef struct ProxyBindStatusCallback
|
||||||
{
|
{
|
||||||
const IBindStatusCallbackVtbl *lpVtbl;
|
const IBindStatusCallbackVtbl *lpVtbl;
|
||||||
|
|
|
||||||
|
|
@ -56,19 +56,6 @@ static inline void URLMON_UnlockModule(void) { InterlockedDecrement( &URLMON_ref
|
||||||
#define DEFINE_THIS2(cls,ifc,iface) ((cls*)((BYTE*)(iface)-offsetof(cls,ifc)))
|
#define DEFINE_THIS2(cls,ifc,iface) ((cls*)((BYTE*)(iface)-offsetof(cls,ifc)))
|
||||||
#define DEFINE_THIS(cls,ifc,iface) DEFINE_THIS2(cls,lp ## ifc ## Vtbl,iface)
|
#define DEFINE_THIS(cls,ifc,iface) DEFINE_THIS2(cls,lp ## ifc ## Vtbl,iface)
|
||||||
|
|
||||||
typedef struct
|
|
||||||
{
|
|
||||||
const IStreamVtbl *lpVtbl;
|
|
||||||
LONG ref;
|
|
||||||
HANDLE handle;
|
|
||||||
BOOL closed;
|
|
||||||
WCHAR *pszFileName;
|
|
||||||
WCHAR *pszURL;
|
|
||||||
} IUMCacheStream;
|
|
||||||
|
|
||||||
HRESULT UMCreateStreamOnCacheFile(LPCWSTR pszURL, DWORD dwSize, LPWSTR pszFileName, HANDLE *phfile, IUMCacheStream **ppstr);
|
|
||||||
void UMCloseCacheFileStream(IUMCacheStream *pstr);
|
|
||||||
|
|
||||||
IInternetProtocolInfo *get_protocol_info(LPCWSTR url);
|
IInternetProtocolInfo *get_protocol_info(LPCWSTR url);
|
||||||
HRESULT get_protocol_handler(LPCWSTR url, CLSID *clsid, IClassFactory **ret);
|
HRESULT get_protocol_handler(LPCWSTR url, CLSID *clsid, IClassFactory **ret);
|
||||||
BOOL is_registered_protocol(LPCWSTR);
|
BOOL is_registered_protocol(LPCWSTR);
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue