Implemented res protocol.

2005-06-30 10:21:58 +00:00 · 2005-06-30 10:21:58 +00:00 · 8a46c8edc9
parent f7cf32eb4d
commit 8a46c8edc9
1 changed files with 132 additions and 26 deletions
--- a/dlls/mshtml/protocol.c
+++ b/dlls/mshtml/protocol.c
@ -32,6 +32,7 @@
 #include "mshtml.h"

 #include "wine/debug.h"
+#include "wine/unicode.h"

 #include "mshtml_private.h"

@ -49,13 +50,12 @@ typedef struct {
    const IClassFactoryVtbl         *lpClassFactoryVtbl;
 } ProtocolFactory;

-#define PROTOCOLINFO_THIS \
-        ProtocolFactory* const This= \
-            (ProtocolFactory*)((BYTE*)(iface)-offsetof(ProtocolFactory,lpInternetProtocolInfoVtbl));
+#define PROTOCOLINFO_THIS(iface) \
+    (ProtocolFactory*)((char*)(iface)-offsetof(ProtocolFactory,lpInternetProtocolInfoVtbl))

 static HRESULT WINAPI InternetProtocolInfo_QueryInterface(IInternetProtocolInfo *iface, REFIID riid, void **ppv)
 {
-    PROTOCOLINFO_THIS
+    ProtocolFactory *This = PROTOCOLINFO_THIS(iface);

    *ppv = NULL;
    if(IsEqualGUID(&IID_IUnknown, riid)) {
@ -80,44 +80,50 @@ static HRESULT WINAPI InternetProtocolInfo_QueryInterface(IInternetProtocolInfo

 static ULONG WINAPI InternetProtocolInfo_AddRef(IInternetProtocolInfo *iface)
 {
-    TRACE("(%p)\n", iface);
+    ProtocolFactory *This = PROTOCOLINFO_THIS(iface);
+    TRACE("(%p)\n", This);
    return 2;
 }

 static ULONG WINAPI InternetProtocolInfo_Release(IInternetProtocolInfo *iface)
 {
-    TRACE("(%p)\n", iface);
+    ProtocolFactory *This = PROTOCOLINFO_THIS(iface);
+    TRACE("(%p)\n", This);
    return 1;
 }

-#define CLASSFACTORY_THIS \
-    ProtocolFactory* const This= \
-        (ProtocolFactory*)((BYTE*)(iface)-offsetof(ProtocolFactory,lpClassFactoryVtbl));
+#undef PROTOCOLINFO_THIS
+
+#define CLASSFACTORY_THIS(iface) \
+    (ProtocolFactory*)((char*)(iface)-offsetof(ProtocolFactory,lpClassFactoryVtbl))

 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
 {
-    CLASSFACTORY_THIS
+    ProtocolFactory *This = CLASSFACTORY_THIS(iface);
    return IInternetProtocolInfo_QueryInterface(PROTOCOLINFO(This), riid, ppv);
 }

 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
 {
-    CLASSFACTORY_THIS
+    ProtocolFactory *This = CLASSFACTORY_THIS(iface);
    return IInternetProtocolInfo_AddRef(PROTOCOLINFO(This));
 }

 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
 {
-    CLASSFACTORY_THIS
+    ProtocolFactory *This = CLASSFACTORY_THIS(iface);
    return IInternetProtocolInfo_Release(PROTOCOLINFO(This));
 }

 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
 {
-    FIXME("(%p)->(%x)\n", iface, dolock);
+    ProtocolFactory *This = CLASSFACTORY_THIS(iface);
+    FIXME("(%p)->(%x)\n", This, dolock);
    return S_OK;
 }

+#undef CLASSFACTORY_THIS
+
 /********************************************************************
 * AboutProtocol implementation
 */
@ -272,7 +278,7 @@ static HRESULT WINAPI AboutProtocolFactory_CreateInstance(IClassFactory *iface,
    AboutProtocol *ret;
    HRESULT hres;

-    FIXME("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
+    TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);

    ret = HeapAlloc(GetProcessHeap(), 0, sizeof(AboutProtocol));
    ret->lpInternetProtocolVtbl = &AboutProtocolVtbl;
@ -350,6 +356,10 @@ static ProtocolFactory AboutProtocolFactory = {
 typedef struct {
    const IInternetProtocolVtbl *lpInternetProtocolVtbl;
    ULONG ref;
+
+    BYTE *data;
+    ULONG data_len;
+    ULONG cur;
 } ResProtocol;

 static HRESULT WINAPI ResProtocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
@ -394,8 +404,10 @@ static ULONG WINAPI ResProtocol_Release(IInternetProtocol *iface)

    TRACE("(%p) ref=%lx\n", iface, ref);

-    if(!ref)
+    if(!ref) {
+        HeapFree(GetProcessHeap(), 0, This->data);
        HeapFree(GetProcessHeap(), 0, This);
+    }

    return ref;
 }
@ -405,9 +417,78 @@ static HRESULT WINAPI ResProtocol_Start(IInternetProtocol *iface, LPCWSTR szUrl,
        DWORD grfPI, DWORD dwReserved)
 {
    ResProtocol *This = (ResProtocol*)iface;
-    FIXME("(%p)->(%s %p %p %08lx %ld)\n", This, debugstr_w(szUrl), pOIProtSink,
+    DWORD grfBINDF = 0;
+    BINDINFO bindinfo;
+    int len;
+    WCHAR dll[MAX_PATH];
+    LPCWSTR url_dll, url_file;
+    HMODULE hdll;
+    HRSRC src;
+
+    static const WCHAR wszRes[] = {'r','e','s',':','/','/'};
+
+    TRACE("(%p)->(%s %p %p %08lx %ld)\n", This, debugstr_w(szUrl), pOIProtSink,
            pOIBindInfo, grfPI, dwReserved);
-    return E_NOTIMPL;
+
+    memset(&bindinfo, 0, sizeof(bindinfo));
+    bindinfo.cbSize = sizeof(BINDINFO);
+    IInternetBindInfo_GetBindInfo(pOIBindInfo, &grfBINDF, &bindinfo);
+
+    /* FIXME:
+     * Implement MIME type checking
+     * Use CoInternetParseUrl (not implemented yet)
+     */
+
+    len = strlenW(szUrl);
+    if(len < sizeof(wszRes)/sizeof(wszRes[0]) || memcmp(szUrl, wszRes, sizeof(wszRes))) {
+        WARN("Wrong protocol of url: %s\n", debugstr_w(szUrl));
+        IInternetProtocolSink_ReportResult(pOIProtSink, MK_E_SYNTAX, 0, NULL);
+        return MK_E_SYNTAX;
+    }
+
+    url_dll = szUrl + sizeof(wszRes)/sizeof(wszRes[0]);
+    if(!(url_file = strchrW(url_dll, '/'))) {
+        WARN("wrong url: %s\n", debugstr_w(szUrl));
+        IInternetProtocolSink_ReportResult(pOIProtSink, MK_E_SYNTAX, 0, NULL);
+        return MK_E_SYNTAX;
+    }
+
+    memcpy(dll, url_dll, (url_file-url_dll)*sizeof(WCHAR));
+    dll[url_file-url_dll] = 0;
+
+    hdll = LoadLibraryExW(dll, NULL, LOAD_LIBRARY_AS_DATAFILE);
+    if(!hdll) {
+        WARN("Could not open dll: %s\n", debugstr_w(dll));
+        IInternetProtocolSink_ReportResult(pOIProtSink, HRESULT_FROM_WIN32(GetLastError()), 0, NULL);
+        return HRESULT_FROM_WIN32(GetLastError());
+    }
+
+    src = FindResourceW(hdll, ++url_file, (LPCWSTR)23);
+    if(!src) {
+        WARN("Could not find resource: %s\n", debugstr_w(url_file));
+        IInternetProtocolSink_ReportResult(pOIProtSink, HRESULT_FROM_WIN32(GetLastError()), 0, NULL);
+        return HRESULT_FROM_WIN32(GetLastError());
+    }
+
+    if(This->data) {
+        WARN("data already loaded\n");
+        HeapFree(GetProcessHeap(), 0, This->data);
+    }
+
+    This->data_len = SizeofResource(hdll, src);
+    This->data = HeapAlloc(GetProcessHeap(), 0, This->data_len);
+    memcpy(This->data, LoadResource(hdll, src), This->data_len);
+    This->cur = 0;
+
+    FreeLibrary(hdll);
+
+    IInternetProtocolSink_ReportData(pOIProtSink,
+            BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE,
+            This->data_len, This->data_len);
+
+    IInternetProtocolSink_ReportResult(pOIProtSink, S_OK, 0, NULL);
+    
+    return S_OK;
 }

 static HRESULT WINAPI ResProtocol_Continue(IInternetProtocol *iface, PROTOCOLDATA* pProtocolData)
@ -428,8 +509,11 @@ static HRESULT WINAPI ResProtocol_Abort(IInternetProtocol *iface, HRESULT hrReas
 static HRESULT WINAPI ResProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
 {
    ResProtocol *This = (ResProtocol*)iface;
-    FIXME("(%p)->(%08lx)\n", This, dwOptions);
-    return E_NOTIMPL;
+
+    TRACE("(%p)->(%08lx)\n", This, dwOptions);
+
+    /* test show that we don't have to do anything here */
+    return S_OK;
 }

 static HRESULT WINAPI ResProtocol_Suspend(IInternetProtocol *iface)
@ -449,8 +533,21 @@ static HRESULT WINAPI ResProtocol_Resume(IInternetProtocol *iface)
 static HRESULT WINAPI ResProtocol_Read(IInternetProtocol *iface, void* pv, ULONG cb, ULONG* pcbRead)
 {
    ResProtocol *This = (ResProtocol*)iface;
-    FIXME("(%p)->(%lu %p)\n", This, cb, pcbRead);
-    return E_NOTIMPL;
+
+    TRACE("(%p)->(%p %lu %p)\n", This, pv, cb, pcbRead);
+
+    if(!This->data)
+        return E_FAIL;
+
+    *pcbRead = (cb > This->data_len-This->cur ? This->data_len-This->cur : cb);
+
+    if(!*pcbRead)
+        return S_FALSE;
+
+    memcpy(pv, This->data, *pcbRead);
+    This->cur += *pcbRead;
+
+    return S_OK;
 }

 static HRESULT WINAPI ResProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
@ -464,15 +561,21 @@ static HRESULT WINAPI ResProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER d
 static HRESULT WINAPI ResProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
 {
    ResProtocol *This = (ResProtocol*)iface;
-    FIXME("(%p)->(%ld)\n", This, dwOptions);
-    return E_NOTIMPL;
+
+    TRACE("(%p)->(%ld)\n", This, dwOptions);
+
+    /* test show that we don't have to do anything here */
+    return S_OK;
 }

 static HRESULT WINAPI ResProtocol_UnlockRequest(IInternetProtocol *iface)
 {
    ResProtocol *This = (ResProtocol*)iface;
-    FIXME("(%p)\n", This);
-    return E_NOTIMPL;
+
+    TRACE("(%p)\n", This);
+
+    /* test show that we don't have to do anything here */
+    return S_OK;
 }

 static const IInternetProtocolVtbl ResProtocolVtbl = {
@ -497,11 +600,14 @@ static HRESULT WINAPI ResProtocolFactory_CreateInstance(IClassFactory *iface, IU
    ResProtocol *ret;
    HRESULT hres;

-    FIXME("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
+    TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);

    ret = HeapAlloc(GetProcessHeap(), 0, sizeof(ResProtocol));
    ret->lpInternetProtocolVtbl = &ResProtocolVtbl;
    ret->ref = 0;
+    ret->data = NULL;
+    ret->data_len = 0;
+    ret->cur = 0;

    hres = IUnknown_QueryInterface((IUnknown*)ret, riid, ppv);