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
21 #include <wine/test.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) \
41 ok(expect_ ##func, "unexpected call " #func "\n"); \
42 expect_ ## func = FALSE; \
43 called_ ## func = TRUE; \
46 #define CHECK_EXPECT2(func) \
48 ok(expect_ ##func, "unexpected call " #func "\n"); \
49 called_ ## func = TRUE; \
52 #define CHECK_CALLED(func) \
54 ok(called_ ## func, "expected " #func "\n"); \
55 expect_ ## func = called_ ## func = FALSE; \
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;
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)
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]);
88 static HRESULT WINAPI ProtocolSink_QueryInterface(IInternetProtocolSink *iface, REFIID riid, void **ppv)
90 if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocolSink, riid)) {
96 ok(0, "unexpected riid %s\n", debugstr_guid(riid));
100 static ULONG WINAPI ProtocolSink_AddRef(IInternetProtocolSink *iface)
105 static ULONG WINAPI ProtocolSink_Release(IInternetProtocolSink *iface)
110 static HRESULT WINAPI ProtocolSink_Switch(IInternetProtocolSink *iface, PROTOCOLDATA *pProtocolData)
112 ok(0, "unexpected call\n");
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");
131 static HRESULT WINAPI ProtocolSink_ReportData(IInternetProtocolSink *iface, DWORD grfBSCF, ULONG ulProgress,
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);
143 static HRESULT WINAPI ProtocolSink_ReportResult(IInternetProtocolSink *iface, HRESULT hrResult, DWORD dwError,
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);
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");
161 static IInternetProtocolSinkVtbl protocol_sink_vtbl = {
162 ProtocolSink_QueryInterface,
164 ProtocolSink_Release,
166 ProtocolSink_ReportProgress,
167 ProtocolSink_ReportData,
168 ProtocolSink_ReportResult
171 static IInternetProtocolSink protocol_sink = {
175 static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface, REFIID riid, void **ppv)
177 if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetBindInfo, riid)) {
183 ok(0, "unexpected riid %s\n", debugstr_guid(riid));
184 return E_NOINTERFACE;
187 static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface)
192 static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface)
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);
209 static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface, ULONG ulStringType, LPOLESTR *ppwzStr,
210 ULONG cEl, ULONG *pcElFetched)
212 ok(0, "unexpected call\n");
216 static IInternetBindInfoVtbl bind_info_vtbl = {
217 BindInfo_QueryInterface,
220 BindInfo_GetBindInfo,
221 BindInfo_GetBindString
224 static IInternetBindInfo bind_info = {
228 static void test_protocol_fail(IInternetProtocol *protocol, LPCWSTR url, HRESULT expected_hres,
229 BOOL expect_win32err)
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);
240 ok((hres&0xffff0000) == ((FACILITY_WIN32 << 16)|0x80000000) || hres == expect_hrResult,
241 "expected win32 err or %08x got: %08x\n", expected_hres, hres);
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)
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)
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");
283 SetLastError(0xdeadbeef);
284 len = SearchPathW(NULL, file, NULL, sizeof(buf)/sizeof(WCHAR), buf, NULL);
286 if (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
287 win_skip("SearchPathW is not implemented\n");
289 ok(0, "SearchPath failed: %u\n", GetLastError());
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;
302 IClassFactory *factory;
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 mshtml_dllW[] = {'m','s','h','t','m','l','.','d','l','l',0};
320 hres = CoGetClassObject(&CLSID_ResProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
321 ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
325 hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
326 ok(hres == S_OK, "Could not get IInternetProtocolInfo interface: %08x\n", hres);
327 if(SUCCEEDED(hres)) {
329 DWORD size, expected_size;
332 for(i = PARSE_CANONICALIZE; i <= PARSE_UNESCAPE; i++) {
333 if(i != PARSE_SECURITY_URL && i != PARSE_DOMAIN) {
334 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, i, 0, buf,
335 sizeof(buf)/sizeof(buf[0]), &size, 0);
336 ok(hres == INET_E_DEFAULT_ACTION,
337 "[%d] failed: %08x, expected INET_E_DEFAULT_ACTION\n", i, hres);
341 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_SECURITY_URL, 0, buf,
342 sizeof(buf)/sizeof(buf[0]), &size, 0);
343 ok(hres == S_OK, "ParseUrl failed: %08x\n", hres);
344 res_sec_url_cmp(buf, size, mshtml_dllW);
345 ok(size == lstrlenW(buf)+1, "size = %d\n", size);
346 expected_size = size;
348 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_SECURITY_URL, 0, buf,
349 expected_size, &size, 0);
350 ok(hres == S_OK, "ParseUrl failed: %08x\n", hres);
351 res_sec_url_cmp(buf, size, mshtml_dllW);
352 ok(size == expected_size, "size = %d\n", size);
355 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_SECURITY_URL, 0, buf,
357 ok(hres == S_FALSE, "ParseUrl failed: %08x, expected S_FALSE\n", hres);
358 ok(size == expected_size, "size = %d\n", size);
360 hres = IInternetProtocolInfo_ParseUrl(protocol_info, wrong_url1, PARSE_SECURITY_URL, 0, buf,
361 sizeof(buf)/sizeof(buf[0]), &size, 0);
362 ok(hres == MK_E_SYNTAX || hres == E_INVALIDARG,
363 "ParseUrl failed: %08x, expected MK_E_SYNTAX\n", hres);
365 hres = IInternetProtocolInfo_ParseUrl(protocol_info, wrong_url5, PARSE_SECURITY_URL, 0, buf,
366 sizeof(buf)/sizeof(buf[0]), &size, 0);
367 ok(hres == MK_E_SYNTAX, "ParseUrl failed: %08x, expected MK_E_SYNTAX\n", hres);
371 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_DOMAIN, 0, buf,
372 sizeof(buf)/sizeof(buf[0]), &size, 0);
373 ok(hres == S_OK || hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
374 ok(buf[0] == '?', "buf changed\n");
375 ok(size == sizeof(blank_url)/sizeof(WCHAR) ||
376 size == sizeof(buf)/sizeof(buf[0]), /* IE8 */
380 hres = IInternetProtocolInfo_ParseUrl(protocol_info, wrong_url1, PARSE_DOMAIN, 0, buf,
381 sizeof(buf)/sizeof(buf[0]), &size, 0);
382 ok(hres == S_OK || hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
383 ok(buf[0] == '?', "buf changed\n");
384 ok(size == sizeof(wrong_url1)/sizeof(WCHAR) ||
385 size == sizeof(buf)/sizeof(buf[0]), /* IE8 */
390 /* Crashes on win9x */
393 hres = IInternetProtocolInfo_ParseUrl(protocol_info, NULL, PARSE_DOMAIN, 0, buf,
394 sizeof(buf)/sizeof(buf[0]), &size, 0);
395 ok(hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
396 ok(buf[0] == '?', "buf changed\n");
397 ok(size == 1, "size=%u, expected 1\n", size);
400 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_DOMAIN, 0, buf,
401 sizeof(buf)/sizeof(buf[0]), NULL, 0);
402 ok(hres == E_POINTER, "ParseUrl failed: %08x\n", hres);
403 ok(buf[0] == '?', "buf changed\n");
406 hres = IInternetProtocolInfo_ParseUrl(protocol_info, NULL, 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");
413 hres = IInternetProtocolInfo_ParseUrl(protocol_info, blank_url, PARSE_UNESCAPE+1, 0, buf,
414 sizeof(buf)/sizeof(buf[0]), &size, 0);
415 ok(hres == INET_E_DEFAULT_ACTION,
416 "ParseUrl failed: %08x, expected INET_E_DEFAULT_ACTION\n", hres);
417 ok(buf[0] == '?', "buf changed\n");
420 hres = IInternetProtocolInfo_CombineUrl(protocol_info, blank_url, test_part_url,
421 0, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
422 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
423 ok(size == 0xdeadbeef, "size=%d\n", size);
426 hres = IInternetProtocolInfo_CombineUrl(protocol_info, blank_url, test_part_url,
427 URL_FILE_USE_PATHURL, 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);
432 hres = IInternetProtocolInfo_CombineUrl(protocol_info, NULL, NULL,
433 URL_FILE_USE_PATHURL, NULL, 0xdeadbeef, NULL, 0);
434 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
435 ok(size == 0xdeadbeef, "size=%d\n", size);
437 hres = IInternetProtocolInfo_CompareUrl(protocol_info, blank_url, blank_url, 0);
438 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
440 hres = IInternetProtocolInfo_CompareUrl(protocol_info, NULL, NULL, 0xdeadbeef);
441 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
443 for(i=0; i<30; i++) {
444 if(i == QUERY_USES_NETWORK || i == QUERY_IS_SECURE || i == QUERY_IS_SAFE)
447 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, i, 0,
448 buf, sizeof(buf), &size, 0);
449 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER,
450 "QueryInfo(%d) returned: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER\n", i, hres);
454 memset(buf, '?', sizeof(buf));
455 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, QUERY_USES_NETWORK, 0,
456 buf, sizeof(buf), &size, 0);
457 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
458 ok(size == sizeof(DWORD), "size=%d\n", size);
459 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
461 memset(buf, '?', sizeof(buf));
462 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, QUERY_USES_NETWORK, 0,
463 buf, sizeof(buf), NULL, 0);
464 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
465 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
467 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, QUERY_USES_NETWORK, 0,
469 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
472 memset(buf, '?', sizeof(buf));
473 hres = IInternetProtocolInfo_QueryInfo(protocol_info, NULL, QUERY_USES_NETWORK, 0,
474 buf, sizeof(buf), &size, 0);
475 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
476 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
477 ok(size == sizeof(DWORD), "size=%d\n", size);
478 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
480 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, QUERY_USES_NETWORK, 0,
481 NULL, sizeof(buf), &size, 0);
482 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
484 hres = IInternetProtocolInfo_QueryInfo(protocol_info, blank_url, 60, 0,
485 NULL, sizeof(buf), &size, 0);
486 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER,
487 "QueryInfo failed: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER\n", hres);
489 IInternetProtocolInfo_Release(protocol_info);
492 hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
493 ok(hres == S_OK, "Could not get IClassFactory interface\n");
494 if(SUCCEEDED(hres)) {
495 IInternetProtocol *protocol;
498 hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol);
499 ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
501 if(SUCCEEDED(hres)) {
502 IInternetPriority *priority;
504 hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority);
505 ok(hres == E_NOINTERFACE,
506 "QueryInterface(IInternetPriority) returned %08x, expected E_NOINTEFACE\n", hres);
508 test_protocol_fail(protocol, wrong_url1, E_INVALIDARG, FALSE);
509 test_protocol_fail(protocol, wrong_url2,
510 HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND), FALSE);
511 test_protocol_fail(protocol, wrong_url3, E_FAIL, TRUE);
512 test_protocol_fail(protocol, wrong_url4, E_FAIL, TRUE);
515 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
516 ok(hres == E_FAIL, "Read returned %08x expected E_FAIL\n", hres);
517 ok(cb == 0xdeadbeef, "cb=%u expected 0xdeadbeef\n", cb);
519 protocol_start(protocol, blank_url);
520 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
521 ok(hres == S_OK, "Read failed: %08x\n", hres);
522 ok(cb == 2, "cb=%u expected 2\n", cb);
523 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
524 ok(hres == S_OK, "Read failed: %08x\n", hres);
525 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
526 ok(hres == S_FALSE, "Read failed: %08x expected S_FALSE\n", hres);
527 ok(cb == 0, "cb=%u expected 0\n", cb);
528 hres = IInternetProtocol_UnlockRequest(protocol);
529 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
531 protocol_start(protocol, blank_url);
532 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
533 ok(hres == S_OK, "Read failed: %08x\n", hres);
534 hres = IInternetProtocol_LockRequest(protocol, 0);
535 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
536 hres = IInternetProtocol_UnlockRequest(protocol);
537 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
538 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
539 ok(hres == S_OK, "Read failed: %08x\n", hres);
541 protocol_start(protocol, blank_url);
542 hres = IInternetProtocol_LockRequest(protocol, 0);
543 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
544 hres = IInternetProtocol_Terminate(protocol, 0);
545 ok(hres == S_OK, "Terminate failed: %08x\n", hres);
546 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
547 ok(hres == S_OK, "Read failed: %08x\n\n", hres);
548 hres = IInternetProtocol_UnlockRequest(protocol);
549 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
550 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
551 ok(hres == S_OK, "Read failed: %08x\n", hres);
552 hres = IInternetProtocol_Terminate(protocol, 0);
553 ok(hres == S_OK, "Terminate failed: %08x\n", hres);
554 hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
555 ok(hres == S_OK, "Read failed: %08x\n", hres);
556 ok(cb == 2, "cb=%u expected 2\n", cb);
558 protocol_start(protocol, blank_url);
559 hres = IInternetProtocol_LockRequest(protocol, 0);
560 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
561 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
562 ok(hres == S_OK, "Read failed: %08x\n", hres);
563 protocol_start(protocol, blank_url);
564 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
565 ok(hres == S_OK, "Read failed: %08x\n", hres);
566 hres = IInternetProtocol_Terminate(protocol, 0);
567 ok(hres == S_OK, "Terminate failed: %08x\n", hres);
569 IInternetProtocol_Release(protocol);
572 IClassFactory_Release(factory);
575 IUnknown_Release(unk);
578 static void do_test_about_protocol(IClassFactory *factory, DWORD bf)
580 IInternetProtocol *protocol;
581 IInternetPriority *priority;
586 static const WCHAR blank_html[] = {0xfeff,'<','H','T','M','L','>','<','/','H','T','M','L','>',0};
587 static const WCHAR test_html[] =
588 {0xfeff,'<','H','T','M','L','>','t','e','s','t','<','/','H','T','M','L','>',0};
592 hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol);
593 ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
597 hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority);
598 ok(hres == E_NOINTERFACE,
599 "QueryInterface(IInternetPriority) returned %08x, expected E_NOINTEFACE\n", hres);
601 protocol_start(protocol, about_blank_url);
602 hres = IInternetProtocol_LockRequest(protocol, 0);
603 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
604 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
605 ok(hres == S_OK, "Read failed: %08x\n", hres);
606 ok(cb == sizeof(blank_html), "cb=%d\n", cb);
607 ok(!memcmp(buf, blank_html, cb), "Readed wrong data\n");
608 hres = IInternetProtocol_UnlockRequest(protocol);
609 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
611 protocol_start(protocol, about_test_url);
612 hres = IInternetProtocol_LockRequest(protocol, 0);
613 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
614 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
615 ok(hres == S_OK, "Read failed: %08x\n", hres);
616 ok(cb == sizeof(test_html), "cb=%d\n", cb);
617 ok(!memcmp(buf, test_html, cb), "Readed wrong data\n");
618 hres = IInternetProtocol_UnlockRequest(protocol);
619 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
621 protocol_start(protocol, about_res_url);
622 hres = IInternetProtocol_LockRequest(protocol, 0);
623 ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
624 hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
625 ok(hres == S_OK, "Read failed: %08x\n", hres);
626 ok(cb == sizeof(blank_html), "cb=%d\n", cb);
627 ok(!memcmp(buf, blank_html, cb), "Readed wrong data\n");
628 hres = IInternetProtocol_UnlockRequest(protocol);
629 ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
631 IInternetProtocol_Release(protocol);
634 static void test_about_protocol(void)
636 IInternetProtocolInfo *protocol_info;
638 IClassFactory *factory;
641 hres = CoGetClassObject(&CLSID_AboutProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
642 ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
646 hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
647 ok(hres == S_OK, "Could not get IInternetProtocolInfo interface: %08x\n", hres);
648 if(SUCCEEDED(hres)) {
653 for(i = PARSE_CANONICALIZE; i <= PARSE_UNESCAPE; i++) {
654 if(i != PARSE_SECURITY_URL && i != PARSE_DOMAIN) {
655 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, i, 0, buf,
656 sizeof(buf)/sizeof(buf[0]), &size, 0);
657 ok(hres == INET_E_DEFAULT_ACTION,
658 "[%d] failed: %08x, expected INET_E_DEFAULT_ACTION\n", i, hres);
662 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_SECURITY_URL, 0, buf,
663 sizeof(buf)/sizeof(buf[0]), &size, 0);
664 ok(hres == S_OK, "ParseUrl failed: %08x\n", hres);
665 ok(!lstrcmpW(about_blank_url, buf), "buf != blank_url\n");
668 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_SECURITY_URL, 0, buf,
670 ok(hres == S_FALSE, "ParseUrl failed: %08x, expected S_FALSE\n", hres);
671 ok(size == 12, "size = %d\n", size);
673 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_test_url, PARSE_SECURITY_URL, 0, buf,
674 sizeof(buf)/sizeof(buf[0]), &size, 0);
675 ok(hres == S_OK, "ParseUrl failed: %08x\n", hres);
676 ok(!lstrcmpW(about_test_url, buf), "buf != test_url\n");
677 ok(size == 11, "size = %d\n", size);
681 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_DOMAIN, 0, buf,
682 sizeof(buf)/sizeof(buf[0]), &size, 0);
683 ok(hres == S_OK || hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
684 ok(buf[0] == '?', "buf changed\n");
685 ok(size == sizeof(about_blank_url)/sizeof(WCHAR) ||
686 size == sizeof(buf)/sizeof(buf[0]), /* IE8 */
691 /* Crashes on win9x */
694 hres = IInternetProtocolInfo_ParseUrl(protocol_info, NULL, PARSE_DOMAIN, 0, buf,
695 sizeof(buf)/sizeof(buf[0]), &size, 0);
696 ok(hres == E_FAIL, "ParseUrl failed: %08x\n", hres);
697 ok(buf[0] == '?', "buf changed\n");
698 ok(size == 1, "size=%u, expected 1\n", size);
701 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_DOMAIN, 0, buf,
702 sizeof(buf)/sizeof(buf[0]), NULL, 0);
703 ok(hres == E_POINTER, "ParseUrl failed: %08x\n", hres);
704 ok(buf[0] == '?', "buf changed\n");
707 hres = IInternetProtocolInfo_ParseUrl(protocol_info, NULL, 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");
713 hres = IInternetProtocolInfo_ParseUrl(protocol_info, about_blank_url, PARSE_UNESCAPE+1, 0, buf,
714 sizeof(buf)/sizeof(buf[0]), &size, 0);
715 ok(hres == INET_E_DEFAULT_ACTION,
716 "ParseUrl failed: %08x, expected INET_E_DEFAULT_ACTION\n", hres);
719 hres = IInternetProtocolInfo_CombineUrl(protocol_info, about_blank_url, about_test_url,
720 0, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
721 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
722 ok(size == 0xdeadbeef, "size=%d\n", size);
725 hres = IInternetProtocolInfo_CombineUrl(protocol_info, about_blank_url, about_test_url,
726 URL_FILE_USE_PATHURL, 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);
731 hres = IInternetProtocolInfo_CombineUrl(protocol_info, NULL, NULL,
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 hres = IInternetProtocolInfo_CompareUrl(protocol_info, about_blank_url, about_blank_url, 0);
737 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
739 hres = IInternetProtocolInfo_CompareUrl(protocol_info, NULL, NULL, 0xdeadbeef);
740 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
742 for(i=0; i<30; i++) {
744 case QUERY_CAN_NAVIGATE:
745 case QUERY_USES_NETWORK:
746 case QUERY_IS_CACHED:
747 case QUERY_IS_INSTALLEDENTRY:
748 case QUERY_IS_CACHED_OR_MAPPED:
749 case QUERY_IS_SECURE:
751 case QUERY_USES_HISTORYFOLDER:
754 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, i, 0,
755 buf, sizeof(buf), &size, 0);
756 ok(hres == E_FAIL, "QueryInfo(%d) returned: %08x, expected E_FAIL\n", i, hres);
760 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_CAN_NAVIGATE, 0,
761 buf, sizeof(buf), &size, 0);
762 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER ||
763 hres == E_FAIL, /* win2k */
764 "QueryInfo returned: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER or E_FAIL\n", hres);
767 memset(buf, '?', sizeof(buf));
768 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_USES_NETWORK, 0,
769 buf, sizeof(buf), &size, 0);
770 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
771 ok(size == sizeof(DWORD), "size=%d\n", size);
772 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
774 memset(buf, '?', sizeof(buf));
775 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_USES_NETWORK, 0,
776 buf, sizeof(buf), NULL, 0);
777 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
778 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
780 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_USES_NETWORK, 0,
782 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
784 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, QUERY_USES_NETWORK, 0,
785 NULL, sizeof(buf), &size, 0);
786 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
788 hres = IInternetProtocolInfo_QueryInfo(protocol_info, about_blank_url, 60, 0,
789 NULL, sizeof(buf), &size, 0);
790 ok(hres == E_FAIL, "QueryInfo failed: %08x, expected E_FAIL\n", hres);
792 IInternetProtocolInfo_Release(protocol_info);
795 hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
796 ok(hres == S_OK, "Could not get IClassFactory interface\n");
797 if(SUCCEEDED(hres)) {
798 do_test_about_protocol(factory, 0);
799 do_test_about_protocol(factory,
800 BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NEEDFILE);
802 IClassFactory_Release(factory);
805 IUnknown_Release(unk);
808 static void test_javascript_protocol(void)
810 IInternetProtocolInfo *protocol_info;
812 IClassFactory *factory;
815 hres = CoGetClassObject(&CLSID_JSProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
816 ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
820 hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
821 ok(hres == S_OK, "Could not get IInternetProtocolInfo interface: %08x\n", hres);
822 if(SUCCEEDED(hres)) {
827 for(i = PARSE_CANONICALIZE; i <= PARSE_UNESCAPE; i++) {
828 if(i != PARSE_SECURITY_URL && i != PARSE_DOMAIN) {
829 hres = IInternetProtocolInfo_ParseUrl(protocol_info, javascript_test_url, i, 0, buf,
830 sizeof(buf)/sizeof(buf[0]), &size, 0);
831 ok(hres == INET_E_DEFAULT_ACTION,
832 "[%d] failed: %08x, expected INET_E_DEFAULT_ACTION\n", i, hres);
836 hres = IInternetProtocolInfo_ParseUrl(protocol_info, javascript_test_url, PARSE_UNESCAPE+1, 0, buf,
837 sizeof(buf)/sizeof(buf[0]), &size, 0);
838 ok(hres == INET_E_DEFAULT_ACTION,
839 "ParseUrl failed: %08x, expected INET_E_DEFAULT_ACTION\n", hres);
842 hres = IInternetProtocolInfo_CombineUrl(protocol_info, javascript_test_url, javascript_test_url,
843 0, buf, sizeof(buf)/sizeof(buf[0]), &size, 0);
844 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER, "CombineUrl failed: %08x\n", hres);
845 ok(size == 0xdeadbeef, "size=%d\n", size);
847 hres = IInternetProtocolInfo_CompareUrl(protocol_info, javascript_test_url, javascript_test_url, 0);
848 ok(hres == E_NOTIMPL, "CompareUrl failed: %08x\n", hres);
850 for(i=0; i<30; i++) {
852 case QUERY_USES_NETWORK:
853 case QUERY_IS_SECURE:
856 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, i, 0,
857 buf, sizeof(buf), &size, 0);
858 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER,
859 "QueryInfo(%d) returned: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER\n", i, hres);
864 memset(buf, '?', sizeof(buf));
865 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, QUERY_USES_NETWORK, 0,
866 buf, sizeof(buf), &size, 0);
867 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
868 ok(size == sizeof(DWORD), "size=%d\n", size);
869 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
871 memset(buf, '?', sizeof(buf));
872 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, QUERY_USES_NETWORK, 0,
873 buf, sizeof(buf), NULL, 0);
874 ok(hres == S_OK, "QueryInfo(QUERY_USES_NETWORK) failed: %08x\n", hres);
875 ok(!*(DWORD*)buf, "buf=%d\n", *(DWORD*)buf);
877 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, QUERY_USES_NETWORK, 0,
879 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
881 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, QUERY_USES_NETWORK, 0,
882 NULL, sizeof(buf), &size, 0);
883 ok(hres == E_FAIL, "QueryInfo(QUERY_USES_NETWORK) failed: %08x, expected E_FAIL\n", hres);
885 hres = IInternetProtocolInfo_QueryInfo(protocol_info, javascript_test_url, 60, 0,
886 NULL, sizeof(buf), &size, 0);
887 ok(hres == INET_E_USE_DEFAULT_PROTOCOLHANDLER,
888 "QueryInfo failed: %08x, expected INET_E_USE_DEFAULT_PROTOCOLHANDLER\n", hres);
890 /* FIXME: test QUERY_IS_SECURE */
892 IInternetProtocolInfo_Release(protocol_info);
895 hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
896 ok(hres == S_OK, "Could not get IClassFactory interface\n");
898 IClassFactory_Release(factory);
900 IUnknown_Release(unk);
908 test_about_protocol();
909 test_javascript_protocol();