+typedef struct {
+ const char *url;
+ const char *content;
+ const char *mime;
+ const char *data;
+} mhtml_binding_test_t;
+
+static const mhtml_binding_test_t binding_tests[] = {
+ {
+ "mhtml:file://%s",
+ mhtml_page1,
+ "text/html",
+ "<HTML></HTML>"
+ },
+ {
+ "mhtml:file://%s!http://winehq.org/mhtmltest.html",
+ mhtml_page1,
+ "Image/Jpeg",
+ "Test"
+ }
+};
+
+static const mhtml_binding_test_t *current_binding_test;
+static IInternetProtocol *current_binding_protocol;
+
+static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface, REFIID riid, void **ppv)
+{
+ if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetBindInfo, riid)) {
+ *ppv = iface;
+ return S_OK;
+ }
+
+ *ppv = NULL;
+ ok(0, "unexpected riid %s\n", wine_dbgstr_guid(riid));
+ return E_NOINTERFACE;
+}
+
+static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface)
+{
+ return 2;
+}
+
+static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface)
+{
+ return 1;
+}
+
+static HRESULT WINAPI BindInfo_GetBindInfo(IInternetBindInfo *iface, DWORD *grfBINDF, BINDINFO *pbindinfo)
+{
+ CHECK_EXPECT(GetBindInfo);
+
+ ok(grfBINDF != NULL, "grfBINDF == NULL\n");
+ ok(pbindinfo != NULL, "pbindinfo == NULL\n");
+ ok(pbindinfo->cbSize == sizeof(BINDINFO), "wrong size of pbindinfo: %d\n", pbindinfo->cbSize);
+
+ *grfBINDF = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NEEDFILE;
+ return S_OK;
+}
+
+static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface, ULONG ulStringType, LPOLESTR *ppwzStr,
+ ULONG cEl, ULONG *pcElFetched)
+{
+ ok(0, "unexpected call\n");
+ return E_NOTIMPL;
+}
+
+static IInternetBindInfoVtbl InternetBindInfoVtbl = {
+ BindInfo_QueryInterface,
+ BindInfo_AddRef,
+ BindInfo_Release,
+ BindInfo_GetBindInfo,
+ BindInfo_GetBindString
+};
+
+static IInternetBindInfo bind_info = {
+ &InternetBindInfoVtbl
+};
+
+static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv)
+{
+ ok(0, "unexpected call %s\n", wine_dbgstr_guid(riid));
+ *ppv = NULL;
+ return E_NOINTERFACE;
+}
+
+static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface)
+{
+ return 2;
+}
+
+static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface)
+{
+ return 1;
+}
+
+static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface, REFGUID guidService,
+ REFIID riid, void **ppv)
+{
+ if(IsEqualGUID(&CLSID_MimeEdit, guidService)) {
+ *ppv = NULL;
+ return E_NOINTERFACE;
+ }
+
+ ok(0, "unexpected service %s\n", wine_dbgstr_guid(guidService));
+ return E_FAIL;
+}
+
+static /* const */ IServiceProviderVtbl ServiceProviderVtbl = {
+ ServiceProvider_QueryInterface,
+ ServiceProvider_AddRef,
+ ServiceProvider_Release,
+ ServiceProvider_QueryService
+};
+
+static IServiceProvider service_provider = { &ServiceProviderVtbl };
+
+static HRESULT WINAPI ProtocolSink_QueryInterface(IInternetProtocolSink *iface, REFIID riid, void **ppv)
+{
+ if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocolSink, riid)) {
+ *ppv = iface;
+ return S_OK;
+ }
+
+ if(IsEqualGUID(&IID_IServiceProvider, riid)) {
+ *ppv = &service_provider;
+ return S_OK;
+ }
+
+ *ppv = NULL;
+ ok(0, "unexpected riid %s\n", wine_dbgstr_guid(riid));
+ return E_NOINTERFACE;
+}
+
+static ULONG WINAPI ProtocolSink_AddRef(IInternetProtocolSink *iface)
+{
+ return 2;
+}
+
+static ULONG WINAPI ProtocolSink_Release(IInternetProtocolSink *iface)
+{
+ return 1;
+}
+
+static HRESULT WINAPI ProtocolSink_Switch(IInternetProtocolSink *iface, PROTOCOLDATA *pProtocolData)
+{
+ ok(0, "unexpected call\n");
+ return E_NOTIMPL;
+}
+
+static HRESULT WINAPI ProtocolSink_ReportProgress(IInternetProtocolSink *iface, ULONG ulStatusCode,
+ const WCHAR *szStatusText)
+{
+ switch(ulStatusCode) {
+ case BINDSTATUS_MIMETYPEAVAILABLE:
+ CHECK_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
+ ok(!strcmp_wa(szStatusText, current_binding_test->mime), "status text %s\n", wine_dbgstr_w(szStatusText));
+ return S_OK;
+ case BINDSTATUS_CACHEFILENAMEAVAILABLE:
+ CHECK_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE);
+ return S_OK;
+ default:
+ ok(0, "unexpected call %u %s\n", ulStatusCode, wine_dbgstr_w(szStatusText));
+ }
+
+ return E_NOTIMPL;
+}
+
+static HRESULT WINAPI ProtocolSink_ReportData(IInternetProtocolSink *iface, DWORD grfBSCF, ULONG ulProgress,
+ ULONG ulProgressMax)
+{
+ char buf[1024];
+ DWORD read;
+ HRESULT hres;
+
+ CHECK_EXPECT(ReportData);
+
+ ok(!ulProgress, "ulProgress = %u\n", ulProgress);
+ ok(ulProgress == ulProgressMax, "ulProgress != ulProgressMax\n");
+ ok(grfBSCF == (BSCF_FIRSTDATANOTIFICATION | BSCF_INTERMEDIATEDATANOTIFICATION
+ | BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE | BSCF_AVAILABLEDATASIZEUNKNOWN),
+ "grcf = %08x\n", grfBSCF);
+
+ hres = IInternetProtocol_Read(current_binding_protocol, buf, sizeof(buf), &read);
+ ok(hres == S_OK, "Read failed: %08x\n", hres);
+ buf[read] = 0;
+ ok(!strcmp(buf, current_binding_test->data), "unexpected data: %s\n", buf);
+
+ hres = IInternetProtocol_Read(current_binding_protocol, buf, sizeof(buf), &read);
+ ok(hres == S_FALSE, "Read failed: %08x\n", hres);
+ return S_OK;
+}
+
+static HRESULT WINAPI ProtocolSink_ReportResult(IInternetProtocolSink *iface, HRESULT hrResult, DWORD dwError,
+ LPCWSTR szResult)
+{
+ CHECK_EXPECT(ReportResult);
+ ok(hrResult == S_OK, "hrResult = %08x\n", hrResult);
+ ok(!dwError, "dwError = %u\n", dwError);
+ ok(!szResult, "szResult = %s\n", wine_dbgstr_w(szResult));
+ return S_OK;
+}
+
+static IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = {
+ ProtocolSink_QueryInterface,
+ ProtocolSink_AddRef,
+ ProtocolSink_Release,
+ ProtocolSink_Switch,
+ ProtocolSink_ReportProgress,
+ ProtocolSink_ReportData,
+ ProtocolSink_ReportResult
+};
+
+static IInternetProtocolSink protocol_sink = { &InternetProtocolSinkVtbl };
+
+static void test_mhtml_protocol_binding(const mhtml_binding_test_t *test)
+{
+ char file_name[MAX_PATH+32], *p, urla[INTERNET_MAX_URL_LENGTH];
+ WCHAR test_url[INTERNET_MAX_URL_LENGTH];
+ IInternetProtocol *protocol;
+ IUnknown *unk;
+ HRESULT hres;
+ HANDLE file;
+ DWORD size;
+
+ p = file_name + GetCurrentDirectoryA(sizeof(file_name), file_name);
+ *p++ = '\\';
+ strcpy(p, "winetest.mht");
+
+ file = CreateFileA(file_name, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
+ FILE_ATTRIBUTE_NORMAL, NULL);
+ ok(file != INVALID_HANDLE_VALUE, "CreateFile failed\n");
+
+ WriteFile(file, test->content, strlen(test->content), &size, NULL);
+ CloseHandle(file);
+
+ sprintf(urla, test->url, file_name);
+ MultiByteToWideChar(CP_ACP, 0, urla, -1, test_url, sizeof(test_url)/sizeof(WCHAR));
+
+ hres = CoCreateInstance(&CLSID_IMimeHtmlProtocol, NULL, CLSCTX_INPROC_SERVER, &IID_IInternetProtocol, (void**)&protocol);
+ ok(hres == S_OK, "Could not create protocol handler: %08x\n", hres);
+
+ hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetProtocolEx, (void**)&unk);
+ ok(hres == E_NOINTERFACE, "Could get IInternetProtocolEx\n");
+
+ current_binding_test = test;
+ current_binding_protocol = protocol;
+
+ SET_EXPECT(GetBindInfo);
+ SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
+ SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE);
+ SET_EXPECT(ReportData);
+ SET_EXPECT(ReportResult);
+ hres = IInternetProtocol_Start(protocol, test_url, &protocol_sink, &bind_info, 0, 0);
+ ok(hres == S_OK, "Start failed: %08x\n", hres);
+ CHECK_CALLED(GetBindInfo);
+ CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE);
+ todo_wine CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE);
+ CHECK_CALLED(ReportData);
+ CHECK_CALLED(ReportResult);
+
+ IInternetProtocol_Release(protocol);
+ ok(DeleteFileA("winetest.mht"), "DeleteFile failed: %u\n", GetLastError());
+}
+
+static const struct {
+ const char *base_url;
+ const char *relative_url;
+ const char *expected_result;
+ BOOL todo;
+} combine_tests[] = {
+ {
+ "mhtml:file:///c:/dir/test.mht", "http://test.org",
+ "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org"
+ }, {
+ "mhtml:file:///c:/dir/test.mht", "3D\"http://test.org\"",
+ "mhtml:file:///c:/dir/test.mht!x-usc:3D\"http://test.org\""
+ }, {
+ "mhtml:file:///c:/dir/test.mht", "123abc",
+ "mhtml:file:///c:/dir/test.mht!x-usc:123abc"
+ }, {
+ "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org", "123abc",
+ "mhtml:file:///c:/dir/test.mht!x-usc:123abc"
+ }, {
+ "MhtMl:file:///c:/dir/test.mht!x-usc:http://test.org/dir/dir2/file.html", "../..",
+ "mhtml:file:///c:/dir/test.mht!x-usc:../.."
+ }, {"mhtml:file:///c:/dir/test.mht!x-usc:file:///c:/dir/dir2/file.html", "../..",
+ "mhtml:file:///c:/dir/test.mht!x-usc:../.."
+ }, {
+ "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org", "",
+ "mhtml:file:///c:/dir/test.mht"
+ }, {
+ "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org", "mhtml:file:///d:/file.html",
+ "file:///d:/file.html", TRUE
+ }, {
+ "mhtml:file:///c:/dir/test.mht!x-usc:http://test.org", "mhtml:file:///c:/dir2/test.mht!x-usc:http://test.org",
+ "mhtml:file:///c:/dir2/test.mht!x-usc:http://test.org", TRUE
+ }, {
+ "mhtml:file:///c:/dir/test.mht!http://test.org", "123abc",
+ "mhtml:file:///c:/dir/test.mht!x-usc:123abc"
+ }, {
+ "mhtml:file:///c:/dir/test.mht!http://test.org", "",
+ "mhtml:file:///c:/dir/test.mht"
+ }
+};
+
+static void test_mhtml_protocol_info(void)
+{
+ WCHAR *base_url, *relative_url, combined_url[INTERNET_MAX_URL_LENGTH];
+ IInternetProtocolInfo *protocol_info;
+ DWORD combined_len;
+ unsigned i, exlen;
+ HRESULT hres;
+
+ static const WCHAR http_url[] = {'h','t','t','p',':','/','/','t','e','s','t','.','o','r','g',0};
+
+ hres = CoCreateInstance(&CLSID_IMimeHtmlProtocol, NULL, CLSCTX_INPROC_SERVER,
+ &IID_IInternetProtocolInfo, (void**)&protocol_info);
+ ok(hres == S_OK, "Could not create protocol info: %08x\n", hres);
+
+ for(i = 0; i < sizeof(combine_tests)/sizeof(*combine_tests); i++) {
+ base_url = a2w(combine_tests[i].base_url);
+ relative_url = a2w(combine_tests[i].relative_url);
+
+ combined_len = 0xdeadbeef;
+ hres = IInternetProtocolInfo_CombineUrl(protocol_info, base_url, relative_url, ICU_BROWSER_MODE,
+ combined_url, sizeof(combined_url)/sizeof(WCHAR), &combined_len, 0);
+ todo_wine_if(combine_tests[i].todo)
+ ok(hres == S_OK, "[%u] CombineUrl failed: %08x\n", i, hres);
+ if(SUCCEEDED(hres)) {
+ exlen = strlen(combine_tests[i].expected_result);
+ ok(combined_len == exlen, "[%u] combined len is %u, expected %u\n", i, combined_len, exlen);
+ ok(!strcmp_wa(combined_url, combine_tests[i].expected_result), "[%u] combined URL is %s, expected %s\n",
+ i, wine_dbgstr_w(combined_url), combine_tests[i].expected_result);
+
+ combined_len = 0xdeadbeef;
+ hres = IInternetProtocolInfo_CombineUrl(protocol_info, base_url, relative_url, ICU_BROWSER_MODE,
+ combined_url, exlen, &combined_len, 0);
+ ok(hres == E_FAIL, "[%u] CombineUrl returned: %08x\n", i, hres);
+ ok(!combined_len, "[%u] combined_len = %u\n", i, combined_len);
+ }
+
+ HeapFree(GetProcessHeap(), 0, base_url);
+ HeapFree(GetProcessHeap(), 0, relative_url);
+ }
+
+ hres = IInternetProtocolInfo_CombineUrl(protocol_info, http_url, http_url, ICU_BROWSER_MODE,
+ combined_url, sizeof(combined_url)/sizeof(WCHAR), &combined_len, 0);
+ ok(hres == E_FAIL, "CombineUrl failed: %08x\n", hres);
+
+ IInternetProtocolInfo_Release(protocol_info);
+}
+
+static HRESULT WINAPI outer_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
+{
+ ok(0, "unexpected call\n");
+ return E_NOINTERFACE;
+}
+
+static ULONG WINAPI outer_AddRef(IUnknown *iface)
+{
+ return 2;
+}
+
+static ULONG WINAPI outer_Release(IUnknown *iface)
+{
+ return 1;
+}
+
+static /* const */ IUnknownVtbl outer_vtbl = {
+ outer_QueryInterface,
+ outer_AddRef,
+ outer_Release
+};
+
+static BOOL broken_mhtml_resolver;
+
+static void test_mhtml_protocol(void)
+{
+ IUnknown outer = { &outer_vtbl };
+ IClassFactory *class_factory;
+ IUnknown *unk, *unk2;
+ unsigned i;
+ HRESULT hres;
+
+ /* test class factory */
+ hres = CoGetClassObject(&CLSID_IMimeHtmlProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
+ ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
+
+ hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&unk2);
+ ok(hres == E_NOINTERFACE, "IInternetProtocolInfo supported\n");
+
+ hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&class_factory);
+ ok(hres == S_OK, "Could not get IClassFactory iface: %08x\n", hres);
+ IUnknown_Release(unk);
+
+ hres = IClassFactory_CreateInstance(class_factory, &outer, &IID_IUnknown, (void**)&unk);
+ ok(hres == S_OK, "CreateInstance returned: %08x\n", hres);
+ hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocol, (void**)&unk2);
+ ok(hres == S_OK, "Could not get IInternetProtocol iface: %08x\n", hres);
+ IUnknown_Release(unk2);
+ IUnknown_Release(unk);
+
+ hres = IClassFactory_CreateInstance(class_factory, (IUnknown*)0xdeadbeef, &IID_IInternetProtocol, (void**)&unk2);
+ ok(hres == CLASS_E_NOAGGREGATION, "CreateInstance returned: %08x\n", hres);
+
+ IClassFactory_Release(class_factory);
+
+ if(!broken_mhtml_resolver)
+ test_mhtml_protocol_info();
+
+ for(i = 0; i < sizeof(binding_tests)/sizeof(*binding_tests); i++)
+ test_mhtml_protocol_binding(binding_tests + i);
+}
+
+static void test_MimeOleObjectFromMoniker(void)
+{
+ IMoniker *mon, *new_mon;
+ WCHAR *mhtml_url, *url;
+ IBindCtx *bind_ctx;
+ IUnknown *unk;
+ unsigned i;
+ HRESULT hres;
+
+ static const struct {
+ const char *url;
+ const char *mhtml_url;
+ } tests[] = {
+ {"file:///x:\\dir\\file.mht", "mhtml:file://x:\\dir\\file.mht"},
+ {"file:///x:/dir/file.mht", "mhtml:file://x:\\dir\\file.mht"},
+ {"http://www.winehq.org/index.html?query#hash", "mhtml:http://www.winehq.org/index.html?query#hash"},
+ {"../test.mht", "mhtml:../test.mht"}
+ };
+
+ for(i = 0; i < sizeof(tests)/sizeof(*tests); i++) {
+ url = a2w(tests[i].url);
+ hres = CreateURLMoniker(NULL, url, &mon);
+ ok(hres == S_OK, "CreateURLMoniker failed: %08x\n", hres);
+ HeapFree(GetProcessHeap(), 0, url);
+
+ hres = CreateBindCtx(0, &bind_ctx);
+ ok(hres == S_OK, "CreateBindCtx failed: %08x\n", hres);
+
+ hres = MimeOleObjectFromMoniker(0, mon, bind_ctx, &IID_IUnknown, (void**)&unk, &new_mon);
+ ok(hres == S_OK || broken(!i && hres == INET_E_RESOURCE_NOT_FOUND), "MimeOleObjectFromMoniker failed: %08x\n", hres);
+ IBindCtx_Release(bind_ctx);
+ if(hres == INET_E_RESOURCE_NOT_FOUND) { /* winxp */
+ win_skip("Broken MHTML behaviour found. Skipping some tests.\n");
+ broken_mhtml_resolver = TRUE;
+ return;
+ }
+
+ hres = IMoniker_GetDisplayName(new_mon, NULL, NULL, &mhtml_url);
+ ok(hres == S_OK, "GetDisplayName failed: %08x\n", hres);
+ ok(!strcmp_wa(mhtml_url, tests[i].mhtml_url), "[%d] unexpected mhtml URL: %s\n", i, wine_dbgstr_w(mhtml_url));
+ CoTaskMemFree(mhtml_url);
+
+ IUnknown_Release(unk);
+ IMoniker_Release(new_mon);
+ IMoniker_Release(mon);
+ }
+}
+