mshtml: Add stub for QUERY_IS_CACHED_AND_USABLE_OFFLINE.
[wine.git] / dlls / mshtml / tests / protocol.c
blob5825015093f5794f90d57896b8e2b6017b572b71
1 /*
2 * Copyright 2005 Jacek Caban
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 #define COBJMACROS
21 #include <wine/test.h>
22 #include <stdarg.h>
23 #include <stdio.h>
25 #include "windef.h"
26 #include "winbase.h"
27 #include "ole2.h"
28 #include "urlmon.h"
29 #include "shlwapi.h"
31 #include "initguid.h"
33 #define DEFINE_EXPECT(func) \
34 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
36 #define SET_EXPECT(func) \
37 expect_ ## func = TRUE
39 #define CHECK_EXPECT(func) \
40 do { \
41 ok(expect_ ##func, "unexpected call " #func "\n"); \
42 expect_ ## func = FALSE; \
43 called_ ## func = TRUE; \
44 }while(0)
46 #define CHECK_EXPECT2(func) \
47 do { \
48 ok(expect_ ##func, "unexpected call " #func "\n"); \
49 called_ ## func = TRUE; \
50 }while(0)
52 #define CHECK_CALLED(func) \
53 do { \
54 ok(called_ ## func, "expected " #func "\n"); \
55 expect_ ## func = called_ ## func = FALSE; \
56 }while(0)
58 DEFINE_GUID(CLSID_ResProtocol, 0x3050F3BC, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
59 DEFINE_GUID(CLSID_AboutProtocol, 0x3050F406, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
60 DEFINE_GUID(CLSID_JSProtocol, 0x3050F3B2, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
62 DEFINE_EXPECT(GetBindInfo);
63 DEFINE_EXPECT(ReportProgress);
64 DEFINE_EXPECT(ReportData);
65 DEFINE_EXPECT(ReportResult);
67 static HRESULT expect_hrResult;
68 static BOOL expect_hr_win32err = FALSE;
69 static DWORD bindf;
71 static const WCHAR about_blank_url[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
72 static const WCHAR about_test_url[] = {'a','b','o','u','t',':','t','e','s','t',0};
73 static const WCHAR about_res_url[] = {'r','e','s',':','b','l','a','n','k',0};
74 static const WCHAR javascript_test_url[] = {'j','a','v','a','s','c','r','i','p','t',':','t','e','s','t','(',')',0};
76 static const char *debugstr_guid(REFIID riid)
78 static char buf[50];
80 sprintf(buf, "{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
81 riid->Data1, riid->Data2, riid->Data3, riid->Data4[0],
82 riid->Data4[1], riid->Data4[2], riid->Data4[3], riid->Data4[4],
83 riid->Data4[5], riid->Data4[6], riid->Data4[7]);
85 return buf;
88 static HRESULT WINAPI ProtocolSink_QueryInterface(IInternetProtocolSink *iface, REFIID riid, void **ppv)
90 if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocolSink, riid)) {
91 *ppv = iface;
92 return S_OK;
95 *ppv = NULL;
96 ok(0, "unexpected riid %s\n", debugstr_guid(riid));
97 return E_NOINTERFACE;
100 static ULONG WINAPI ProtocolSink_AddRef(IInternetProtocolSink *iface)
102 return 2;
105 static ULONG WINAPI ProtocolSink_Release(IInternetProtocolSink *iface)
107 return 1;
110 static HRESULT WINAPI ProtocolSink_Switch(IInternetProtocolSink *iface, PROTOCOLDATA *pProtocolData)
112 ok(0, "unexpected call\n");
113 return E_NOTIMPL;
116 static HRESULT WINAPI ProtocolSink_ReportProgress(IInternetProtocolSink *iface, ULONG ulStatusCode,
117 LPCWSTR szStatusText)
119 static const WCHAR text_html[] = {'t','e','x','t','/','h','t','m','l',0};
121 CHECK_EXPECT(ReportProgress);
123 ok(ulStatusCode == BINDSTATUS_MIMETYPEAVAILABLE
124 || ulStatusCode == BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE,
125 "ulStatusCode=%d\n", ulStatusCode);
126 ok(!lstrcmpW(szStatusText, text_html), "szStatusText != text/html\n");
128 return S_OK;
131 static HRESULT WINAPI ProtocolSink_ReportData(IInternetProtocolSink *iface, DWORD grfBSCF, ULONG ulProgress,
132 ULONG ulProgressMax)
134 CHECK_EXPECT(ReportData);
136 ok(ulProgress == ulProgressMax, "ulProgress != ulProgressMax\n");
137 ok(grfBSCF == (BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE),
138 "grcf = %08x\n", grfBSCF);
140 return S_OK;
143 static HRESULT WINAPI ProtocolSink_ReportResult(IInternetProtocolSink *iface, HRESULT hrResult, DWORD dwError,
144 LPCWSTR szResult)
146 CHECK_EXPECT(ReportResult);
148 if(expect_hr_win32err)
149 ok((hrResult&0xffff0000) == ((FACILITY_WIN32 << 16)|0x80000000) || expect_hrResult,
150 "expected win32 err or %08x got: %08x\n", expect_hrResult, hrResult);
151 else
152 ok(hrResult == expect_hrResult || ((expect_hrResult == E_INVALIDARG ||
153 expect_hrResult == HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND)) &&
154 hrResult == MK_E_SYNTAX), "expected: %08x got: %08x\n", expect_hrResult, hrResult);
155 ok(dwError == 0, "dwError = %d\n", dwError);
156 ok(!szResult, "szResult != NULL\n");
158 return S_OK;
161 static IInternetProtocolSinkVtbl protocol_sink_vtbl = {
162 ProtocolSink_QueryInterface,
163 ProtocolSink_AddRef,
164 ProtocolSink_Release,
165 ProtocolSink_Switch,
166 ProtocolSink_ReportProgress,
167 ProtocolSink_ReportData,
168 ProtocolSink_ReportResult
171 static IInternetProtocolSink protocol_sink = {
172 &protocol_sink_vtbl
175 static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface, REFIID riid, void **ppv)
177 if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetBindInfo, riid)) {
178 *ppv = iface;
179 return S_OK;
182 *ppv = NULL;
183 ok(0, "unexpected riid %s\n", debugstr_guid(riid));
184 return E_NOINTERFACE;
187 static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface)
189 return 2;
192 static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface)
194 return 1;
197 static HRESULT WINAPI BindInfo_GetBindInfo(IInternetBindInfo *iface, DWORD *grfBINDF, BINDINFO *pbindinfo)
199 CHECK_EXPECT(GetBindInfo);
201 ok(grfBINDF != NULL, "grfBINDF == NULL\n");
202 ok(pbindinfo != NULL, "pbindinfo == NULL\n");
203 ok(pbindinfo->cbSize == sizeof(BINDINFO), "wrong size of pbindinfo: %d\n", pbindinfo->cbSize);
205 *grfBINDF = bindf;
206 return S_OK;
209 static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface, ULONG ulStringType, LPOLESTR *ppwzStr,
210 ULONG cEl, ULONG *pcElFetched)
212 ok(0, "unexpected call\n");
213 return E_NOTIMPL;
216 static IInternetBindInfoVtbl bind_info_vtbl = {
217 BindInfo_QueryInterface,
218 BindInfo_AddRef,
219 BindInfo_Release,
220 BindInfo_GetBindInfo,
221 BindInfo_GetBindString
224 static IInternetBindInfo bind_info = {
225 &bind_info_vtbl
228 static void test_protocol_fail(IInternetProtocol *protocol, LPCWSTR url, HRESULT expected_hres,
229 BOOL expect_win32err)
231 HRESULT hres;
233 SET_EXPECT(GetBindInfo);
234 SET_EXPECT(ReportResult);
236 expect_hrResult = expected_hres;
237 expect_hr_win32err = expect_win32err;
238 hres = IInternetProtocol_Start(protocol, url, &protocol_sink, &bind_info, 0, 0);
239 if(expect_win32err)
240 ok((hres&0xffff0000) == ((FACILITY_WIN32 << 16)|0x80000000) || hres == expect_hrResult,
241 "expected win32 err or %08x got: %08x\n", expected_hres, hres);
242 else
243 ok(hres == expected_hres || ((expected_hres == E_INVALIDARG ||
244 expected_hres == HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND)) && hres == MK_E_SYNTAX),
245 "expected: %08x got: %08x\n", expected_hres, hres);
247 CHECK_CALLED(GetBindInfo);
248 CHECK_CALLED(ReportResult);
251 static void protocol_start(IInternetProtocol *protocol, LPCWSTR url)
253 HRESULT hres;
255 SET_EXPECT(GetBindInfo);
256 SET_EXPECT(ReportResult);
257 SET_EXPECT(ReportProgress);
258 SET_EXPECT(ReportData);
259 expect_hrResult = S_OK;
260 expect_hr_win32err = FALSE;
262 hres = IInternetProtocol_Start(protocol, url, &protocol_sink, &bind_info, 0, 0);
263 ok(hres == S_OK, "Start failed: %08x\n", hres);
265 CHECK_CALLED(GetBindInfo);
266 CHECK_CALLED(ReportProgress);
267 CHECK_CALLED(ReportData);
268 CHECK_CALLED(ReportResult);
271 static void res_sec_url_cmp(LPCWSTR url, DWORD size, LPCWSTR file)
273 WCHAR buf[MAX_PATH];
274 DWORD len;
276 static const WCHAR fileW[] = {'f','i','l','e',':','/','/'};
278 if(size < sizeof(fileW)/sizeof(WCHAR) || memcmp(url, fileW, sizeof(fileW))) {
279 ok(0, "wrong URL protocol\n");
280 return;
283 SetLastError(0xdeadbeef);
284 len = SearchPathW(NULL, file, NULL, sizeof(buf)/sizeof(WCHAR), buf, NULL);
285 if(!len) {
286 if (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
287 win_skip("SearchPathW is not implemented\n");
288 else
289 ok(0, "SearchPath failed: %u\n", GetLastError());
290 return;
293 len += sizeof(fileW)/sizeof(WCHAR)+1;
294 ok(len == size, "wrong size %u, expected %u\n", size, len);
295 ok(!lstrcmpW(url + sizeof(fileW)/sizeof(WCHAR), buf), "wrong file part %s\n", wine_dbgstr_w(url));
298 static void test_res_protocol(void)
300 IInternetProtocolInfo *protocol_info;
301 IUnknown *unk;
302 IClassFactory *factory;
303 HRESULT hres;
305 static const WCHAR blank_url[] =
306 {'r','e','s',':','/','/','m','s','h','t','m','l','.','d','l','l','/','b','l','a','n','k','.','h','t','m',0};
307 static const WCHAR test_part_url[] = {'r','e','s',':','/','/','C','S','S','/','t','e','s','t',0};
308 static const WCHAR wrong_url1[] =
309 {'m','s','h','t','m','l','.','d','l','l','/','b','l','a','n','k','.','m','t','h',0};
310 static const WCHAR wrong_url2[] =
311 {'r','e','s',':','/','/','m','s','h','t','m','l','.','d','l','l',0};
312 static const WCHAR wrong_url3[] =
313 {'r','e','s',':','/','/','m','s','h','t','m','l','.','d','l','l','/','x','x','.','h','t','m',0};
314 static const WCHAR wrong_url4[] =
315 {'r','e','s',':','/','/','x','x','.','d','l','l','/','b','l','a','n','k','.','h','t','m',0};
316 static const WCHAR wrong_url5[] =
317 {'r','e','s',':','/','/','s','h','t','m','l','.','d','l','l','/','b','l','a','n','k','.','h','t','m',0};
318 static const WCHAR wrong_url6[] =
319 {'r','e','s',':','/','/','c',':','\\','d','i','r','\\','f','i','l','e','.','d','l','l','/','b','l','a','n','k','.','h','t','m',0};
320 static const WCHAR mshtml_dllW[] = {'m','s','h','t','m','l','.','d','l','l',0};
322 hres = CoGetClassObject(&CLSID_ResProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
323 ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
324 if(FAILED(hres))
325 return;
327 hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
328 ok(hres == S_OK, "Could not get IInternetProtocolInfo interface: %08x\n", hres);
329 if(SUCCEEDED(hres)) {
330 WCHAR buf[128];
331 DWORD size, expected_size;
332 int i;
334 for(i = PARSE_CANONICALIZE; i <= PARSE_UNESCAPE; i++) {
335 if(i != PARSE_SECURITY_URL && i != PARSE_DOMAIN) {
336 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, i, 0, buf,
337 sizeof(buf)/sizeof(buf[0]), &size, 0);
338 ok(hres == INET_E_DEFAULT_ACTION,
339 "[%d] failed: %08x, expected INET_E_DEFAULT_ACTION\n", i, hres);
343 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_SECURITY_URL, 0, buf,
344 sizeof(buf)/sizeof(buf[0]), &size, 0);
345 ok(hres == S_OK, "ParseUrl failed: %08x\n", hres);
346 res_sec_url_cmp(buf, size, mshtml_dllW);
347 ok(size == lstrlenW(buf)+1, "size = %d\n", size);
348 expected_size = size;
350 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_SECURITY_URL, 0, buf,
351 expected_size, &size, 0);
352 ok(hres == S_OK, "ParseUrl failed: %08x\n", hres);
353 res_sec_url_cmp(buf, size, mshtml_dllW);
354 ok(size == expected_size, "size = %d\n", size);
356 size = 0;
357 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_SECURITY_URL, 0, buf,
358 3, &size, 0);
359 ok(hres == S_FALSE, "ParseUrl failed: %08x, expected S_FALSE\n", hres);
360 ok(size == expected_size, "size = %d\n", size);
362 hres = IInternetProtocolInfo_ParseUrl(protocol_info, wrong_url1, PARSE_SECURITY_URL, 0, buf,
363 sizeof(buf)/sizeof(buf[0]), &size, 0);
364 ok(hres == MK_E_SYNTAX || hres == E_INVALIDARG,
365 "ParseUrl failed: %08x, expected MK_E_SYNTAX\n", hres);
367 hres = IInternetProtocolInfo_ParseUrl(protocol_info, wrong_url5, PARSE_SECURITY_URL, 0, buf,
368 sizeof(buf)/sizeof(buf[0]), &size, 0);
369 ok(hres == MK_E_SYNTAX, "ParseUrl failed: %08x, expected MK_E_SYNTAX\n", hres);
371 hres = IInternetProtocolInfo_ParseUrl(protocol_info, wrong_url6, PARSE_SECURITY_URL, 0, buf,
372 sizeof(buf)/sizeof(buf[0]), &size, 0);
373 ok(hres == MK_E_SYNTAX, "ParseUrl failed: %08x, expected MK_E_SYNTAX\n", hres);
375 size = 0xdeadbeef;
376 buf[0] = '?';
377 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_DOMAIN, 0, buf,
378 sizeof(buf)/sizeof(buf[0]), &size, 0);
379 ok(hres == S_OK || hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
380 ok(buf[0] == '?', "buf changed\n");
381 ok(size == sizeof(blank_url)/sizeof(WCHAR) ||
382 size == sizeof(buf)/sizeof(buf[0]), /* IE8 */
383 "size=%d\n", size);
385 size = 0xdeadbeef;
386 hres = IInternetProtocolInfo_ParseUrl(protocol_info, wrong_url1, PARSE_DOMAIN, 0, buf,
387 sizeof(buf)/sizeof(buf[0]), &size, 0);
388 ok(hres == S_OK || hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
389 ok(buf[0] == '?', "buf changed\n");
390 ok(size == sizeof(wrong_url1)/sizeof(WCHAR) ||
391 size == sizeof(buf)/sizeof(buf[0]), /* IE8 */
392 "size=%d\n", size);
394 if (0)
396 /* Crashes on windows */
397 size = 0xdeadbeef;
398 buf[0] = '?';
399 hres = IInternetProtocolInfo_ParseUrl(protocol_info, NULL, PARSE_DOMAIN, 0, buf,
400 sizeof(buf)/sizeof(buf[0]), &size, 0);
401 ok(hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
402 ok(buf[0] == '?', "buf changed\n");
403 ok(size == 1, "size=%u, expected 1\n", size);
405 buf[0] = '?';
406 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_DOMAIN, 0, buf,
407 sizeof(buf)/sizeof(buf[0]), NULL, 0);
408 ok(hres == E_POINTER, "ParseUrl failed: %08x\n", hres);
409 ok(buf[0] == '?', "buf changed\n");
411 buf[0] = '?';
412 hres = IInternetProtocolInfo_ParseUrl(protocol_info, NULL, PARSE_DOMAIN, 0, buf,
413 sizeof(buf)/sizeof(buf[0]), NULL, 0);
414 ok(hres == E_POINTER, "ParseUrl failed: %08x\n", hres);
415 ok(buf[0] == '?', "buf changed\n");
418 buf[0] = '?';
419 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_UNESCAPE+1, 0, buf,
420 sizeof(buf)/sizeof(buf[0]), &size, 0);
421 ok(hres == INET_E_DEFAULT_ACTION,
422 "ParseUrl failed: %08x, expected INET_E_DEFAULT_ACTION\n", hres);
423 ok(buf[0] == '?', "buf changed\n");
425 size = 0xdeadbeef;
426 hres = IInternetProtocolInfo_CombineUrl(protocol_info, blank_url, test_part_url,
427 0, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
428 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
429 ok(size == 0xdeadbeef, "size=%d\n", size);
431 size = 0xdeadbeef;
432 hres = IInternetProtocolInfo_CombineUrl(protocol_info, blank_url, test_part_url,
433 URL_FILE_USE_PATHURL, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
434 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
435 ok(size == 0xdeadbeef, "size=%d\n", size);
437 size = 0xdeadbeef;
438 hres = IInternetProtocolInfo_CombineUrl(protocol_info, NULL, NULL,
439 URL_FILE_USE_PATHURL, NULL, 0xdeadbeef, NULL, 0);
440 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
441 ok(size == 0xdeadbeef, "size=%d\n", size);
443 hres = IInternetProtocolInfo_CompareUrl(protocol_info, blank_url, blank_url, 0);
444 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
446 hres = IInternetProtocolInfo_CompareUrl(protocol_info, NULL, NULL, 0xdeadbeef);
447 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
449 for(i=0; i<30; i++) {
450 if(i == QUERY_USES_NETWORK || i == QUERY_IS_SECURE || i == QUERY_IS_SAFE)
451 continue;
453 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, i, 0,
454 buf, sizeof(buf), &size, 0);
455 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER,
456 "QueryInfo(%d) returned: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER\n", i, hres);
459 size = 0xdeadbeef;
460 memset(buf, '?', sizeof(buf));
461 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, QUERY_USES_NETWORK, 0,
462 buf, sizeof(buf), &size, 0);
463 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
464 ok(size == sizeof(DWORD), "size=%d\n", size);
465 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
467 memset(buf, '?', sizeof(buf));
468 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, QUERY_USES_NETWORK, 0,
469 buf, sizeof(buf), NULL, 0);
470 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
471 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
473 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, QUERY_USES_NETWORK, 0,
474 buf, 3, &size, 0);
475 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
477 size = 0xdeadbeef;
478 memset(buf, '?', sizeof(buf));
479 hres = IInternetProtocolInfo_QueryInfo(protocol_info, NULL, QUERY_USES_NETWORK, 0,
480 buf, sizeof(buf), &size, 0);
481 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
482 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
483 ok(size == sizeof(DWORD), "size=%d\n", size);
484 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
486 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, QUERY_USES_NETWORK, 0,
487 NULL, sizeof(buf), &size, 0);
488 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
490 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, 60, 0,
491 NULL, sizeof(buf), &size, 0);
492 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER,
493 "QueryInfo failed: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER\n", hres);
495 IInternetProtocolInfo_Release(protocol_info);
498 hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
499 ok(hres == S_OK, "Could not get IClassFactory interface\n");
500 if(SUCCEEDED(hres)) {
501 IInternetProtocol *protocol;
502 BYTE buf[512];
503 ULONG cb;
504 hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol);
505 ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
507 if(SUCCEEDED(hres)) {
508 IInternetPriority *priority;
510 hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority);
511 ok(hres == E_NOINTERFACE,
512 "QueryInterface(IInternetPriority) returned %08x, expected E_NOINTEFACE\n", hres);
514 test_protocol_fail(protocol, wrong_url1, E_INVALIDARG, FALSE);
515 test_protocol_fail(protocol, wrong_url2,
516 HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND), FALSE);
517 test_protocol_fail(protocol, wrong_url3, E_FAIL, TRUE);
518 test_protocol_fail(protocol, wrong_url4, E_FAIL, TRUE);
520 cb = 0xdeadbeef;
521 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
522 ok(hres == E_FAIL, "Read returned %08x expected E_FAIL\n", hres);
523 ok(cb == 0xdeadbeef, "cb=%u expected 0xdeadbeef\n", cb);
525 protocol_start(protocol, blank_url);
526 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
527 ok(hres == S_OK, "Read failed: %08x\n", hres);
528 ok(cb == 2, "cb=%u expected 2\n", cb);
529 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
530 ok(hres == S_OK, "Read failed: %08x\n", hres);
531 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
532 ok(hres == S_FALSE, "Read failed: %08x expected S_FALSE\n", hres);
533 ok(cb == 0, "cb=%u expected 0\n", cb);
534 hres = IInternetProtocol_UnlockRequest(protocol);
535 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
537 protocol_start(protocol, blank_url);
538 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
539 ok(hres == S_OK, "Read failed: %08x\n", hres);
540 hres = IInternetProtocol_LockRequest(protocol, 0);
541 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
542 hres = IInternetProtocol_UnlockRequest(protocol);
543 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
544 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
545 ok(hres == S_OK, "Read failed: %08x\n", hres);
547 protocol_start(protocol, blank_url);
548 hres = IInternetProtocol_LockRequest(protocol, 0);
549 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
550 hres = IInternetProtocol_Terminate(protocol, 0);
551 ok(hres == S_OK, "Terminate failed: %08x\n", hres);
552 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
553 ok(hres == S_OK, "Read failed: %08x\n\n", hres);
554 hres = IInternetProtocol_UnlockRequest(protocol);
555 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
556 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
557 ok(hres == S_OK, "Read failed: %08x\n", hres);
558 hres = IInternetProtocol_Terminate(protocol, 0);
559 ok(hres == S_OK, "Terminate failed: %08x\n", hres);
560 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
561 ok(hres == S_OK, "Read failed: %08x\n", hres);
562 ok(cb == 2, "cb=%u expected 2\n", cb);
564 protocol_start(protocol, blank_url);
565 hres = IInternetProtocol_LockRequest(protocol, 0);
566 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
567 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
568 ok(hres == S_OK, "Read failed: %08x\n", hres);
569 protocol_start(protocol, blank_url);
570 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
571 ok(hres == S_OK, "Read failed: %08x\n", hres);
572 hres = IInternetProtocol_Terminate(protocol, 0);
573 ok(hres == S_OK, "Terminate failed: %08x\n", hres);
575 IInternetProtocol_Release(protocol);
578 IClassFactory_Release(factory);
581 IUnknown_Release(unk);
584 static void do_test_about_protocol(IClassFactory *factory, DWORD bf)
586 IInternetProtocol *protocol;
587 IInternetPriority *priority;
588 BYTE buf[512];
589 ULONG cb;
590 HRESULT hres;
592 static const WCHAR blank_html[] = {0xfeff,'<','H','T','M','L','>','<','/','H','T','M','L','>',0};
593 static const WCHAR test_html[] =
594 {0xfeff,'<','H','T','M','L','>','t','e','s','t','<','/','H','T','M','L','>',0};
596 bindf = bf;
598 hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol);
599 ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
600 if(FAILED(hres))
601 return;
603 hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority);
604 ok(hres == E_NOINTERFACE,
605 "QueryInterface(IInternetPriority) returned %08x, expected E_NOINTEFACE\n", hres);
607 protocol_start(protocol, about_blank_url);
608 hres = IInternetProtocol_LockRequest(protocol, 0);
609 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
610 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
611 ok(hres == S_OK, "Read failed: %08x\n", hres);
612 ok(cb == sizeof(blank_html), "cb=%d\n", cb);
613 ok(!memcmp(buf, blank_html, cb), "Readed wrong data\n");
614 hres = IInternetProtocol_UnlockRequest(protocol);
615 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
617 protocol_start(protocol, about_test_url);
618 hres = IInternetProtocol_LockRequest(protocol, 0);
619 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
620 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
621 ok(hres == S_OK, "Read failed: %08x\n", hres);
622 ok(cb == sizeof(test_html), "cb=%d\n", cb);
623 ok(!memcmp(buf, test_html, cb), "Readed wrong data\n");
624 hres = IInternetProtocol_UnlockRequest(protocol);
625 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
627 protocol_start(protocol, about_res_url);
628 hres = IInternetProtocol_LockRequest(protocol, 0);
629 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
630 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
631 ok(hres == S_OK, "Read failed: %08x\n", hres);
632 ok(cb == sizeof(blank_html), "cb=%d\n", cb);
633 ok(!memcmp(buf, blank_html, cb), "Readed wrong data\n");
634 hres = IInternetProtocol_UnlockRequest(protocol);
635 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
637 IInternetProtocol_Release(protocol);
640 static void test_about_protocol(void)
642 IInternetProtocolInfo *protocol_info;
643 IUnknown *unk;
644 IClassFactory *factory;
645 HRESULT hres;
647 hres = CoGetClassObject(&CLSID_AboutProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
648 ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
649 if(FAILED(hres))
650 return;
652 hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
653 ok(hres == S_OK, "Could not get IInternetProtocolInfo interface: %08x\n", hres);
654 if(SUCCEEDED(hres)) {
655 WCHAR buf[128];
656 DWORD size;
657 int i;
659 for(i = PARSE_CANONICALIZE; i <= PARSE_UNESCAPE; i++) {
660 if(i != PARSE_SECURITY_URL && i != PARSE_DOMAIN) {
661 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, i, 0, buf,
662 sizeof(buf)/sizeof(buf[0]), &size, 0);
663 ok(hres == INET_E_DEFAULT_ACTION,
664 "[%d] failed: %08x, expected INET_E_DEFAULT_ACTION\n", i, hres);
668 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_SECURITY_URL, 0, buf,
669 sizeof(buf)/sizeof(buf[0]), &size, 0);
670 ok(hres == S_OK, "ParseUrl failed: %08x\n", hres);
671 ok(!lstrcmpW(about_blank_url, buf), "buf != blank_url\n");
673 size = 0xdeadbeef;
674 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_SECURITY_URL, 0, buf,
675 3, &size, 0);
676 ok(hres == S_FALSE, "ParseUrl failed: %08x, expected S_FALSE\n", hres);
677 ok(size == 12, "size = %d\n", size);
679 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_test_url, PARSE_SECURITY_URL, 0, buf,
680 sizeof(buf)/sizeof(buf[0]), &size, 0);
681 ok(hres == S_OK, "ParseUrl failed: %08x\n", hres);
682 ok(!lstrcmpW(about_test_url, buf), "buf != test_url\n");
683 ok(size == 11, "size = %d\n", size);
685 size = 0xdeadbeef;
686 buf[0] = '?';
687 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_DOMAIN, 0, buf,
688 sizeof(buf)/sizeof(buf[0]), &size, 0);
689 ok(hres == S_OK || hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
690 ok(buf[0] == '?', "buf changed\n");
691 ok(size == sizeof(about_blank_url)/sizeof(WCHAR) ||
692 size == sizeof(buf)/sizeof(buf[0]), /* IE8 */
693 "size=%d\n", size);
695 if (0)
697 /* Crashes on windows */
698 size = 0xdeadbeef;
699 buf[0] = '?';
700 hres = IInternetProtocolInfo_ParseUrl(protocol_info, NULL, PARSE_DOMAIN, 0, buf,
701 sizeof(buf)/sizeof(buf[0]), &size, 0);
702 ok(hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
703 ok(buf[0] == '?', "buf changed\n");
704 ok(size == 1, "size=%u, expected 1\n", size);
706 buf[0] = '?';
707 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_DOMAIN, 0, buf,
708 sizeof(buf)/sizeof(buf[0]), NULL, 0);
709 ok(hres == E_POINTER, "ParseUrl failed: %08x\n", hres);
710 ok(buf[0] == '?', "buf changed\n");
712 buf[0] = '?';
713 hres = IInternetProtocolInfo_ParseUrl(protocol_info, NULL, PARSE_DOMAIN, 0, buf,
714 sizeof(buf)/sizeof(buf[0]), NULL, 0);
715 ok(hres == E_POINTER, "ParseUrl failed: %08x\n", hres);
716 ok(buf[0] == '?', "buf changed\n");
719 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_UNESCAPE+1, 0, buf,
720 sizeof(buf)/sizeof(buf[0]), &size, 0);
721 ok(hres == INET_E_DEFAULT_ACTION,
722 "ParseUrl failed: %08x, expected INET_E_DEFAULT_ACTION\n", hres);
724 size = 0xdeadbeef;
725 hres = IInternetProtocolInfo_CombineUrl(protocol_info, about_blank_url, about_test_url,
726 0, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
727 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
728 ok(size == 0xdeadbeef, "size=%d\n", size);
730 size = 0xdeadbeef;
731 hres = IInternetProtocolInfo_CombineUrl(protocol_info, about_blank_url, about_test_url,
732 URL_FILE_USE_PATHURL, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
733 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
734 ok(size == 0xdeadbeef, "size=%d\n", size);
736 size = 0xdeadbeef;
737 hres = IInternetProtocolInfo_CombineUrl(protocol_info, NULL, NULL,
738 URL_FILE_USE_PATHURL, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
739 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
740 ok(size == 0xdeadbeef, "size=%d\n", size);
742 hres = IInternetProtocolInfo_CompareUrl(protocol_info, about_blank_url, about_blank_url, 0);
743 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
745 hres = IInternetProtocolInfo_CompareUrl(protocol_info, NULL, NULL, 0xdeadbeef);
746 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
748 for(i=0; i<30; i++) {
749 switch(i) {
750 case QUERY_CAN_NAVIGATE:
751 case QUERY_USES_NETWORK:
752 case QUERY_IS_CACHED:
753 case QUERY_IS_INSTALLEDENTRY:
754 case QUERY_IS_CACHED_OR_MAPPED:
755 case QUERY_IS_SECURE:
756 case QUERY_IS_SAFE:
757 case QUERY_USES_HISTORYFOLDER:
758 case QUERY_IS_CACHED_AND_USABLE_OFFLINE:
759 break;
760 default:
761 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, i, 0,
762 buf, sizeof(buf), &size, 0);
763 ok(hres == E_FAIL, "QueryInfo(%d) returned: %08x, expected E_FAIL\n", i, hres);
767 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_CAN_NAVIGATE, 0,
768 buf, sizeof(buf), &size, 0);
769 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER ||
770 hres == E_FAIL, /* win2k */
771 "QueryInfo returned: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER or E_FAIL\n", hres);
773 size = 0xdeadbeef;
774 memset(buf, '?', sizeof(buf));
775 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_USES_NETWORK, 0,
776 buf, sizeof(buf), &size, 0);
777 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
778 ok(size == sizeof(DWORD), "size=%d\n", size);
779 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
781 memset(buf, '?', sizeof(buf));
782 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_USES_NETWORK, 0,
783 buf, sizeof(buf), NULL, 0);
784 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
785 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
787 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_USES_NETWORK, 0,
788 buf, 3, &size, 0);
789 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
791 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_USES_NETWORK, 0,
792 NULL, sizeof(buf), &size, 0);
793 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
795 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, 60, 0,
796 NULL, sizeof(buf), &size, 0);
797 ok(hres == E_FAIL, "QueryInfo failed: %08x, expected E_FAIL\n", hres);
799 IInternetProtocolInfo_Release(protocol_info);
802 hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
803 ok(hres == S_OK, "Could not get IClassFactory interface\n");
804 if(SUCCEEDED(hres)) {
805 do_test_about_protocol(factory, 0);
806 do_test_about_protocol(factory,
807 BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NEEDFILE);
809 IClassFactory_Release(factory);
812 IUnknown_Release(unk);
815 static void test_javascript_protocol(void)
817 IInternetProtocolInfo *protocol_info;
818 IUnknown *unk;
819 IClassFactory *factory;
820 HRESULT hres;
822 hres = CoGetClassObject(&CLSID_JSProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
823 ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
824 if(FAILED(hres))
825 return;
827 hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
828 ok(hres == S_OK, "Could not get IInternetProtocolInfo interface: %08x\n", hres);
829 if(SUCCEEDED(hres)) {
830 WCHAR buf[128];
831 DWORD size;
832 int i;
834 for(i = PARSE_CANONICALIZE; i <= PARSE_UNESCAPE; i++) {
835 if(i != PARSE_SECURITY_URL && i != PARSE_DOMAIN) {
836 hres = IInternetProtocolInfo_ParseUrl(protocol_info, javascript_test_url, i, 0, buf,
837 sizeof(buf)/sizeof(buf[0]), &size, 0);
838 ok(hres == INET_E_DEFAULT_ACTION,
839 "[%d] failed: %08x, expected INET_E_DEFAULT_ACTION\n", i, hres);
843 hres = IInternetProtocolInfo_ParseUrl(protocol_info, javascript_test_url, PARSE_UNESCAPE+1, 0, buf,
844 sizeof(buf)/sizeof(buf[0]), &size, 0);
845 ok(hres == INET_E_DEFAULT_ACTION,
846 "ParseUrl failed: %08x, expected INET_E_DEFAULT_ACTION\n", hres);
848 size = 0xdeadbeef;
849 hres = IInternetProtocolInfo_CombineUrl(protocol_info, javascript_test_url, javascript_test_url,
850 0, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
851 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
852 ok(size == 0xdeadbeef, "size=%d\n", size);
854 hres = IInternetProtocolInfo_CompareUrl(protocol_info, javascript_test_url, javascript_test_url, 0);
855 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
857 for(i=0; i<30; i++) {
858 switch(i) {
859 case QUERY_USES_NETWORK:
860 case QUERY_IS_SECURE:
861 break;
862 default:
863 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, i, 0,
864 buf, sizeof(buf), &size, 0);
865 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER,
866 "QueryInfo(%d) returned: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER\n", i, hres);
871 memset(buf, '?', sizeof(buf));
872 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, QUERY_USES_NETWORK, 0,
873 buf, sizeof(buf), &size, 0);
874 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
875 ok(size == sizeof(DWORD), "size=%d\n", size);
876 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
878 memset(buf, '?', sizeof(buf));
879 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, QUERY_USES_NETWORK, 0,
880 buf, sizeof(buf), NULL, 0);
881 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
882 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
884 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, QUERY_USES_NETWORK, 0,
885 buf, 3, &size, 0);
886 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
888 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, QUERY_USES_NETWORK, 0,
889 NULL, sizeof(buf), &size, 0);
890 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
892 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, 60, 0,
893 NULL, sizeof(buf), &size, 0);
894 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER,
895 "QueryInfo failed: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER\n", hres);
897 /* FIXME: test QUERY_IS_SECURE */
899 IInternetProtocolInfo_Release(protocol_info);
902 hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
903 ok(hres == S_OK, "Could not get IClassFactory interface\n");
904 if(SUCCEEDED(hres))
905 IClassFactory_Release(factory);
907 IUnknown_Release(unk);
910 START_TEST(protocol)
912 OleInitialize(NULL);
914 test_res_protocol();
915 test_about_protocol();
916 test_javascript_protocol();
918 OleUninitialize();