2 * Copyright 2005-2009 Jacek Caban for CodeWeavers
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
22 #include <wine/test.h>
32 #define DEFINE_EXPECT(func) \
33 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
35 #define SET_EXPECT(func) \
36 expect_ ## func = TRUE
38 #define CHECK_EXPECT2(func) \
40 ok(expect_ ##func, "unexpected call " #func "\n"); \
41 called_ ## func = TRUE; \
44 #define CHECK_EXPECT(func) \
46 CHECK_EXPECT2(func); \
47 expect_ ## func = FALSE; \
50 #define CHECK_CALLED(func) \
52 ok(called_ ## func, "expected " #func "\n"); \
53 expect_ ## func = called_ ## func = FALSE; \
56 #define CHECK_NOT_CALLED(func) \
58 ok(!called_ ## func, "unexpected " #func "\n"); \
59 expect_ ## func = called_ ## func = FALSE; \
62 #define CLEAR_CALLED(func) \
63 expect_ ## func = called_ ## func = FALSE
65 DEFINE_EXPECT(GetBindInfo
);
66 DEFINE_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
67 DEFINE_EXPECT(ReportProgress_DIRECTBIND
);
68 DEFINE_EXPECT(ReportProgress_RAWMIMETYPE
);
69 DEFINE_EXPECT(ReportProgress_FINDINGRESOURCE
);
70 DEFINE_EXPECT(ReportProgress_CONNECTING
);
71 DEFINE_EXPECT(ReportProgress_SENDINGREQUEST
);
72 DEFINE_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE
);
73 DEFINE_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE
);
74 DEFINE_EXPECT(ReportProgress_PROTOCOLCLASSID
);
75 DEFINE_EXPECT(ReportProgress_COOKIE_SENT
);
76 DEFINE_EXPECT(ReportProgress_REDIRECTING
);
77 DEFINE_EXPECT(ReportProgress_ENCODING
);
78 DEFINE_EXPECT(ReportProgress_ACCEPTRANGES
);
79 DEFINE_EXPECT(ReportProgress_PROXYDETECTING
);
80 DEFINE_EXPECT(ReportProgress_LOADINGMIMEHANDLER
);
81 DEFINE_EXPECT(ReportProgress_DECODING
);
82 DEFINE_EXPECT(ReportData
);
83 DEFINE_EXPECT(ReportData2
);
84 DEFINE_EXPECT(ReportResult
);
85 DEFINE_EXPECT(GetBindString_ACCEPT_MIMES
);
86 DEFINE_EXPECT(GetBindString_USER_AGENT
);
87 DEFINE_EXPECT(GetBindString_POST_COOKIE
);
88 DEFINE_EXPECT(GetBindString_URL
);
89 DEFINE_EXPECT(QueryService_HttpNegotiate
);
90 DEFINE_EXPECT(QueryService_InternetProtocol
);
91 DEFINE_EXPECT(QueryService_HttpSecurity
);
92 DEFINE_EXPECT(QueryInterface_IWinInetInfo
);
93 DEFINE_EXPECT(QueryInterface_IWinInetHttpInfo
);
94 DEFINE_EXPECT(BeginningTransaction
);
95 DEFINE_EXPECT(GetRootSecurityId
);
96 DEFINE_EXPECT(OnResponse
);
97 DEFINE_EXPECT(Switch
);
98 DEFINE_EXPECT(Continue
);
99 DEFINE_EXPECT(CreateInstance
);
100 DEFINE_EXPECT(Start
);
101 DEFINE_EXPECT(Terminate
);
103 DEFINE_EXPECT(Read2
);
104 DEFINE_EXPECT(SetPriority
);
105 DEFINE_EXPECT(LockRequest
);
106 DEFINE_EXPECT(UnlockRequest
);
107 DEFINE_EXPECT(MimeFilter_CreateInstance
);
108 DEFINE_EXPECT(MimeFilter_Start
);
109 DEFINE_EXPECT(MimeFilter_ReportProgress
);
110 DEFINE_EXPECT(MimeFilter_ReportData
);
111 DEFINE_EXPECT(MimeFilter_ReportResult
);
112 DEFINE_EXPECT(MimeFilter_Terminate
);
113 DEFINE_EXPECT(MimeFilter_LockRequest
);
114 DEFINE_EXPECT(MimeFilter_UnlockRequest
);
115 DEFINE_EXPECT(MimeFilter_Read
);
116 DEFINE_EXPECT(MimeFilter_Switch
);
117 DEFINE_EXPECT(MimeFilter_Continue
);
119 static const WCHAR wszIndexHtml
[] = {'i','n','d','e','x','.','h','t','m','l',0};
120 static const WCHAR index_url
[] =
121 {'f','i','l','e',':','i','n','d','e','x','.','h','t','m','l',0};
123 static const WCHAR acc_mimeW
[] = {'*','/','*',0};
124 static const WCHAR user_agentW
[] = {'W','i','n','e',0};
125 static const WCHAR text_htmlW
[] = {'t','e','x','t','/','h','t','m','l',0};
126 static const WCHAR hostW
[] = {'w','w','w','.','w','i','n','e','h','q','.','o','r','g',0};
127 static const WCHAR winehq_ipW
[] = {'2','0','9','.','4','6','.','2','5','.','1','3','4',0};
128 static const WCHAR emptyW
[] = {0};
129 static const WCHAR gzipW
[] = {'g','z','i','p',0};
131 static HRESULT expect_hrResult
;
132 static LPCWSTR file_name
, http_url
, expect_wsz
;
133 static IInternetProtocol
*async_protocol
= NULL
;
134 static BOOL first_data_notif
, http_is_first
, http_post_test
;
135 static int state
= 0, prot_state
, read_report_data
;
136 static DWORD bindf
, ex_priority
, pi
;
137 static IInternetProtocol
*binding_protocol
, *filtered_protocol
;
138 static IInternetBindInfo
*prot_bind_info
;
139 static IInternetProtocolSink
*binding_sink
, *filtered_sink
;
140 static void *expect_pv
;
141 static HANDLE event_complete
, event_complete2
, event_continue
, event_continue_done
;
142 static BOOL binding_test
;
143 static PROTOCOLDATA protocoldata
, *pdata
, continue_protdata
;
144 static DWORD prot_read
, pi
, filter_state
;
145 static BOOL security_problem
;
146 static BOOL async_read_pending
, mimefilter_test
, direct_read
, wait_for_switch
, emulate_prot
, short_read
;
157 static const WCHAR protocol_names
[][10] = {
160 {'h','t','t','p','s',0},
166 static const WCHAR binding_urls
[][130] = {
167 {'f','i','l','e',':','t','e','s','t','.','h','t','m','l',0},
168 {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q','.',
169 'o','r','g','/','s','i','t','e','/','a','b','o','u','t',0},
170 {'h','t','t','p','s',':','/','/','w','w','w','.','c','o','d','e','w','e','a','v','e','r','s',
171 '.','c','o','m','/','t','e','s','t','.','h','t','m','l',0},
172 {'f','t','p',':','/','/','f','t','p','.','w','i','n','e','h','q','.','o','r','g',
173 '/','p','u','b','/','o','t','h','e','r',
174 '/','w','i','n','e','l','o','g','o','.','x','c','f','.','t','a','r','.','b','z','2',0},
175 {'m','k',':','t','e','s','t',0},
176 {'t','e','s','t',':','/','/','f','i','l','e','.','h','t','m','l',0}
179 static const char *debugstr_guid(REFIID riid
)
183 sprintf(buf
, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
184 riid
->Data1
, riid
->Data2
, riid
->Data3
, riid
->Data4
[0],
185 riid
->Data4
[1], riid
->Data4
[2], riid
->Data4
[3], riid
->Data4
[4],
186 riid
->Data4
[5], riid
->Data4
[6], riid
->Data4
[7]);
191 static int strcmp_wa(LPCWSTR strw
, const char *stra
)
194 WideCharToMultiByte(CP_ACP
, 0, strw
, -1, buf
, sizeof(buf
), NULL
, NULL
);
195 return lstrcmpA(stra
, buf
);
198 /* lstrcmpW is not implemented on Win9x */
199 static int strcmp_ww(LPCWSTR strw1
, LPCWSTR strw2
)
201 CHAR stra1
[512], stra2
[512];
202 WideCharToMultiByte(CP_ACP
, 0, strw1
, -1, stra1
, MAX_PATH
, NULL
, NULL
);
203 WideCharToMultiByte(CP_ACP
, 0, strw2
, -1, stra2
, MAX_PATH
, NULL
, NULL
);
204 return lstrcmpA(stra1
, stra2
);
207 static HRESULT WINAPI
HttpSecurity_QueryInterface(IHttpSecurity
*iface
, REFIID riid
, void **ppv
)
209 if(IsEqualGUID(&IID_IUnknown
, riid
)
210 || IsEqualGUID(&IID_IHttpSecurity
, riid
)) {
215 ok(0, "unexpected call\n");
216 return E_NOINTERFACE
;
219 static ULONG WINAPI
HttpSecurity_AddRef(IHttpSecurity
*iface
)
224 static ULONG WINAPI
HttpSecurity_Release(IHttpSecurity
*iface
)
229 static HRESULT WINAPI
HttpSecurity_GetWindow(IHttpSecurity
* iface
, REFGUID rguidReason
, HWND
*phwnd
)
231 trace("HttpSecurity_GetWindow\n");
236 static HRESULT WINAPI
HttpSecurity_OnSecurityProblem(IHttpSecurity
*iface
, DWORD dwProblem
)
238 trace("Security problem: %u\n", dwProblem
);
239 ok(dwProblem
== ERROR_INTERNET_SEC_CERT_REV_FAILED
, "Expected ERROR_INTERNET_SEC_CERT_REV_FAILED got %u\n", dwProblem
);
241 /* Only retry once */
242 if (security_problem
)
245 security_problem
= TRUE
;
246 SET_EXPECT(BeginningTransaction
);
251 static IHttpSecurityVtbl HttpSecurityVtbl
= {
252 HttpSecurity_QueryInterface
,
254 HttpSecurity_Release
,
255 HttpSecurity_GetWindow
,
256 HttpSecurity_OnSecurityProblem
259 static IHttpSecurity http_security
= { &HttpSecurityVtbl
};
261 static HRESULT WINAPI
HttpNegotiate_QueryInterface(IHttpNegotiate2
*iface
, REFIID riid
, void **ppv
)
263 if(IsEqualGUID(&IID_IUnknown
, riid
)
264 || IsEqualGUID(&IID_IHttpNegotiate
, riid
)
265 || IsEqualGUID(&IID_IHttpNegotiate2
, riid
)) {
270 ok(0, "unexpected call\n");
271 return E_NOINTERFACE
;
274 static ULONG WINAPI
HttpNegotiate_AddRef(IHttpNegotiate2
*iface
)
279 static ULONG WINAPI
HttpNegotiate_Release(IHttpNegotiate2
*iface
)
284 static HRESULT WINAPI
HttpNegotiate_BeginningTransaction(IHttpNegotiate2
*iface
, LPCWSTR szURL
,
285 LPCWSTR szHeaders
, DWORD dwReserved
, LPWSTR
*pszAdditionalHeaders
)
289 static const WCHAR wszHeaders
[] =
290 {'C','o','n','t','e','n','t','-','T','y','p','e',':',' ','a','p','p','l','i','c','a','t',
291 'i','o','n','/','x','-','w','w','w','-','f','o','r','m','-','u','r','l','e','n','c','o',
292 'd','e','d','\r','\n',0};
294 CHECK_EXPECT(BeginningTransaction
);
297 ok(!strcmp_ww(szURL
, binding_urls
[tested_protocol
]), "szURL != http_url\n");
299 ok(!strcmp_ww(szURL
, http_url
), "szURL != http_url\n");
300 ok(!dwReserved
, "dwReserved=%d, expected 0\n", dwReserved
);
301 ok(pszAdditionalHeaders
!= NULL
, "pszAdditionalHeaders == NULL\n");
302 if(pszAdditionalHeaders
)
304 ok(*pszAdditionalHeaders
== NULL
, "*pszAdditionalHeaders != NULL\n");
307 addl_headers
= CoTaskMemAlloc(sizeof(wszHeaders
));
310 http_post_test
= FALSE
;
311 skip("Out of memory\n");
312 return E_OUTOFMEMORY
;
314 memcpy(addl_headers
, wszHeaders
, sizeof(wszHeaders
));
315 *pszAdditionalHeaders
= addl_headers
;
322 static HRESULT WINAPI
HttpNegotiate_OnResponse(IHttpNegotiate2
*iface
, DWORD dwResponseCode
,
323 LPCWSTR szResponseHeaders
, LPCWSTR szRequestHeaders
, LPWSTR
*pszAdditionalRequestHeaders
)
325 CHECK_EXPECT(OnResponse
);
327 ok(dwResponseCode
== 200, "dwResponseCode=%d, expected 200\n", dwResponseCode
);
328 ok(szResponseHeaders
!= NULL
, "szResponseHeaders == NULL\n");
329 ok(szRequestHeaders
== NULL
, "szRequestHeaders != NULL\n");
330 ok(pszAdditionalRequestHeaders
== NULL
, "pszAdditionalHeaders != NULL\n");
335 static HRESULT WINAPI
HttpNegotiate_GetRootSecurityId(IHttpNegotiate2
*iface
,
336 BYTE
*pbSecurityId
, DWORD
*pcbSecurityId
, DWORD_PTR dwReserved
)
338 static const BYTE sec_id
[] = {'h','t','t','p',':','t','e','s','t',1,0,0,0};
340 CHECK_EXPECT(GetRootSecurityId
);
342 ok(!dwReserved
, "dwReserved=%ld, expected 0\n", dwReserved
);
343 ok(pbSecurityId
!= NULL
, "pbSecurityId == NULL\n");
344 ok(pcbSecurityId
!= NULL
, "pcbSecurityId == NULL\n");
347 ok(*pcbSecurityId
== 512, "*pcbSecurityId=%d, expected 512\n", *pcbSecurityId
);
348 *pcbSecurityId
= sizeof(sec_id
);
352 memcpy(pbSecurityId
, sec_id
, sizeof(sec_id
));
357 static IHttpNegotiate2Vtbl HttpNegotiateVtbl
= {
358 HttpNegotiate_QueryInterface
,
359 HttpNegotiate_AddRef
,
360 HttpNegotiate_Release
,
361 HttpNegotiate_BeginningTransaction
,
362 HttpNegotiate_OnResponse
,
363 HttpNegotiate_GetRootSecurityId
366 static IHttpNegotiate2 http_negotiate
= { &HttpNegotiateVtbl
};
368 static HRESULT
QueryInterface(REFIID
,void**);
370 static HRESULT WINAPI
ServiceProvider_QueryInterface(IServiceProvider
*iface
, REFIID riid
, void **ppv
)
372 return QueryInterface(riid
, ppv
);
375 static ULONG WINAPI
ServiceProvider_AddRef(IServiceProvider
*iface
)
380 static ULONG WINAPI
ServiceProvider_Release(IServiceProvider
*iface
)
385 static HRESULT WINAPI
ServiceProvider_QueryService(IServiceProvider
*iface
, REFGUID guidService
,
386 REFIID riid
, void **ppv
)
388 if(IsEqualGUID(&IID_IHttpNegotiate
, guidService
) || IsEqualGUID(&IID_IHttpNegotiate2
, riid
)) {
389 CHECK_EXPECT2(QueryService_HttpNegotiate
);
390 return IHttpNegotiate2_QueryInterface(&http_negotiate
, riid
, ppv
);
393 if(IsEqualGUID(&IID_IInternetProtocol
, guidService
)) {
394 ok(IsEqualGUID(&IID_IInternetProtocol
, riid
), "unexpected riid\n");
395 CHECK_EXPECT(QueryService_InternetProtocol
);
396 return E_NOINTERFACE
;
399 if(IsEqualGUID(&IID_IHttpSecurity
, guidService
)) {
400 ok(IsEqualGUID(&IID_IHttpSecurity
, riid
), "unexpected riid\n");
401 CHECK_EXPECT(QueryService_HttpSecurity
);
402 return IHttpSecurity_QueryInterface(&http_security
, riid
, ppv
);
405 ok(0, "unexpected service %s\n", debugstr_guid(guidService
));
409 static const IServiceProviderVtbl ServiceProviderVtbl
= {
410 ServiceProvider_QueryInterface
,
411 ServiceProvider_AddRef
,
412 ServiceProvider_Release
,
413 ServiceProvider_QueryService
416 static IServiceProvider service_provider
= { &ServiceProviderVtbl
};
418 static HRESULT WINAPI
ProtocolSink_QueryInterface(IInternetProtocolSink
*iface
, REFIID riid
, void **ppv
)
420 return QueryInterface(riid
, ppv
);
423 static ULONG WINAPI
ProtocolSink_AddRef(IInternetProtocolSink
*iface
)
428 static ULONG WINAPI
ProtocolSink_Release(IInternetProtocolSink
*iface
)
433 static void call_continue(PROTOCOLDATA
*protocol_data
)
438 if(tested_protocol
== HTTP_TEST
|| tested_protocol
== HTTPS_TEST
|| tested_protocol
== FTP_TEST
) {
440 CLEAR_CALLED(ReportProgress_FINDINGRESOURCE
);
441 CLEAR_CALLED(ReportProgress_CONNECTING
);
442 CLEAR_CALLED(ReportProgress_PROXYDETECTING
);
444 CHECK_NOT_CALLED(ReportProgress_FINDINGRESOURCE
);
445 /* IE7 does call this */
446 CLEAR_CALLED(ReportProgress_CONNECTING
);
449 if(tested_protocol
== FTP_TEST
)
450 todo_wine
CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
451 else if (tested_protocol
!= HTTPS_TEST
)
452 CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
453 if(tested_protocol
== HTTP_TEST
|| tested_protocol
== HTTPS_TEST
) {
454 SET_EXPECT(OnResponse
);
455 if(tested_protocol
== HTTPS_TEST
)
456 SET_EXPECT(ReportProgress_ACCEPTRANGES
);
457 SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
458 if(bindf
& BINDF_NEEDFILE
)
459 SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE
);
463 SET_EXPECT(ReportData
);
464 hres
= IInternetProtocol_Continue(async_protocol
, protocol_data
);
465 ok(hres
== S_OK
, "Continue failed: %08x\n", hres
);
466 if(tested_protocol
== FTP_TEST
)
467 CLEAR_CALLED(ReportData
);
468 else if (! security_problem
)
469 CHECK_CALLED(ReportData
);
472 if (! security_problem
)
475 if(tested_protocol
== HTTP_TEST
|| tested_protocol
== HTTPS_TEST
) {
476 CHECK_CALLED(OnResponse
);
477 if(tested_protocol
== HTTPS_TEST
)
478 CHECK_CALLED(ReportProgress_ACCEPTRANGES
);
479 CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
480 if(bindf
& BINDF_NEEDFILE
)
481 CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE
);
486 security_problem
= FALSE
;
487 SET_EXPECT(ReportProgress_CONNECTING
);
492 static HRESULT WINAPI
ProtocolSink_Switch(IInternetProtocolSink
*iface
, PROTOCOLDATA
*pProtocolData
)
494 if(tested_protocol
== FTP_TEST
)
495 CHECK_EXPECT2(Switch
);
497 CHECK_EXPECT(Switch
);
499 ok(pProtocolData
!= NULL
, "pProtocolData == NULL\n");
501 ok(pProtocolData
!= &protocoldata
, "pProtocolData == &protocoldata\n");
502 ok(pProtocolData
->grfFlags
== protocoldata
.grfFlags
, "grfFlags wrong %x/%x\n",
503 pProtocolData
->grfFlags
, protocoldata
.grfFlags
);
504 ok(pProtocolData
->dwState
== protocoldata
.dwState
, "dwState wrong %x/%x\n",
505 pProtocolData
->dwState
, protocoldata
.dwState
);
506 ok(pProtocolData
->pData
== protocoldata
.pData
, "pData wrong %p/%p\n",
507 pProtocolData
->pData
, protocoldata
.pData
);
508 ok(pProtocolData
->cbData
== protocoldata
.cbData
, "cbData wrong %x/%x\n",
509 pProtocolData
->cbData
, protocoldata
.cbData
);
512 pdata
= pProtocolData
;
515 SetEvent(event_complete
);
516 WaitForSingleObject(event_complete2
, INFINITE
);
519 continue_protdata
= *pProtocolData
;
520 SetEvent(event_continue
);
521 WaitForSingleObject(event_continue_done
, INFINITE
);
523 call_continue(pProtocolData
);
524 SetEvent(event_complete
);
530 static HRESULT WINAPI
ProtocolSink_ReportProgress(IInternetProtocolSink
*iface
, ULONG ulStatusCode
,
531 LPCWSTR szStatusText
)
533 static const WCHAR null_guid
[] = {'{','0','0','0','0','0','0','0','0','-','0','0','0','0','-',
534 '0','0','0','0','-','0','0','0','0','-','0','0','0','0','0','0','0','0','0','0','0','0','}',0};
535 static const WCHAR text_plain
[] = {'t','e','x','t','/','p','l','a','i','n',0};
537 switch(ulStatusCode
) {
538 case BINDSTATUS_MIMETYPEAVAILABLE
:
539 CHECK_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
540 if(tested_protocol
!= FILE_TEST
&& !mimefilter_test
&& (pi
& PI_MIMEVERIFICATION
)) {
541 if(!short_read
|| !direct_read
)
542 CHECK_CALLED(Read
); /* set in Continue */
544 CHECK_CALLED(Read2
); /* set in Read */
546 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
548 if(tested_protocol
== BIND_TEST
)
549 ok(szStatusText
== expect_wsz
, "unexpected szStatusText\n");
550 else if (http_post_test
)
551 ok(lstrlenW(text_plain
) <= lstrlenW(szStatusText
) &&
552 !memcmp(szStatusText
, text_plain
, lstrlenW(text_plain
)*sizeof(WCHAR
)),
553 "szStatusText != text/plain\n");
554 else if(!mimefilter_test
)
555 ok(lstrlenW(text_htmlW
) <= lstrlenW(szStatusText
) &&
556 !memcmp(szStatusText
, text_htmlW
, lstrlenW(text_htmlW
)*sizeof(WCHAR
)),
557 "szStatusText != text/html\n");
560 case BINDSTATUS_DIRECTBIND
:
561 CHECK_EXPECT2(ReportProgress_DIRECTBIND
);
562 ok(szStatusText
== NULL
, "szStatusText != NULL\n");
564 case BINDSTATUS_RAWMIMETYPE
:
565 CHECK_EXPECT2(ReportProgress_RAWMIMETYPE
);
566 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
568 ok(lstrlenW(szStatusText
) < lstrlenW(text_htmlW
) ||
569 !memcmp(szStatusText
, text_htmlW
, lstrlenW(text_htmlW
)*sizeof(WCHAR
)),
570 "szStatusText != text/html\n");
572 case BINDSTATUS_CACHEFILENAMEAVAILABLE
:
573 CHECK_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE
);
574 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
577 ok(!strcmp_ww(szStatusText
, expect_wsz
), "unexpected szStatusText\n");
578 else if(tested_protocol
== FILE_TEST
)
579 ok(!strcmp_ww(szStatusText
, file_name
), "szStatusText = %s\n", wine_dbgstr_w(szStatusText
));
581 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
584 case BINDSTATUS_FINDINGRESOURCE
:
585 CHECK_EXPECT2(ReportProgress_FINDINGRESOURCE
);
586 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
588 case BINDSTATUS_CONNECTING
:
589 CHECK_EXPECT2(ReportProgress_CONNECTING
);
590 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
592 case BINDSTATUS_SENDINGREQUEST
:
593 CHECK_EXPECT2(ReportProgress_SENDINGREQUEST
);
594 if(tested_protocol
== FILE_TEST
) {
595 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
597 ok(!*szStatusText
, "wrong szStatusText\n");
600 case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE
:
601 CHECK_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE
);
602 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
604 ok(!strcmp_ww(szStatusText
, text_htmlW
), "szStatusText != text/html\n");
606 case BINDSTATUS_PROTOCOLCLASSID
:
607 CHECK_EXPECT(ReportProgress_PROTOCOLCLASSID
);
608 ok(szStatusText
!= NULL
, "szStatusText == NULL\n");
609 ok(!strcmp_ww(szStatusText
, null_guid
), "unexpected classid %s\n", wine_dbgstr_w(szStatusText
));
611 case BINDSTATUS_COOKIE_SENT
:
612 CHECK_EXPECT(ReportProgress_COOKIE_SENT
);
613 ok(szStatusText
== NULL
, "szStatusText != NULL\n");
615 case BINDSTATUS_REDIRECTING
:
616 CHECK_EXPECT(ReportProgress_REDIRECTING
);
617 ok(szStatusText
== NULL
, "szStatusText = %s\n", wine_dbgstr_w(szStatusText
));
619 case BINDSTATUS_ENCODING
:
620 CHECK_EXPECT(ReportProgress_ENCODING
);
621 ok(!strcmp_wa(szStatusText
, "gzip"), "szStatusText = %s\n", wine_dbgstr_w(szStatusText
));
623 case BINDSTATUS_ACCEPTRANGES
:
624 CHECK_EXPECT(ReportProgress_ACCEPTRANGES
);
625 ok(!szStatusText
, "szStatusText = %s\n", wine_dbgstr_w(szStatusText
));
627 case BINDSTATUS_PROXYDETECTING
:
628 CHECK_EXPECT(ReportProgress_PROXYDETECTING
);
629 SET_EXPECT(ReportProgress_CONNECTING
);
630 ok(!szStatusText
, "szStatusText = %s\n", wine_dbgstr_w(szStatusText
));
632 case BINDSTATUS_LOADINGMIMEHANDLER
:
633 CHECK_EXPECT(ReportProgress_LOADINGMIMEHANDLER
);
634 ok(!szStatusText
, "szStatusText = %s\n", wine_dbgstr_w(szStatusText
));
636 case BINDSTATUS_DECODING
:
637 CHECK_EXPECT(ReportProgress_DECODING
);
638 ok(!strcmp_ww(szStatusText
, gzipW
), "szStatusText = %s\n", wine_dbgstr_w(szStatusText
));
641 ok(0, "Unexpected status %d\n", ulStatusCode
);
647 static HRESULT WINAPI
ProtocolSink_ReportData(IInternetProtocolSink
*iface
, DWORD grfBSCF
,
648 ULONG ulProgress
, ULONG ulProgressMax
)
652 static int rec_depth
;
655 if(!mimefilter_test
&& tested_protocol
== FILE_TEST
) {
656 CHECK_EXPECT2(ReportData
);
658 ok(ulProgress
== ulProgressMax
, "ulProgress (%d) != ulProgressMax (%d)\n",
659 ulProgress
, ulProgressMax
);
660 ok(ulProgressMax
== 13, "ulProgressMax=%d, expected 13\n", ulProgressMax
);
661 /* BSCF_SKIPDRAINDATAFORFILEURLS added in IE8 */
662 ok((grfBSCF
== (BSCF_FIRSTDATANOTIFICATION
| BSCF_LASTDATANOTIFICATION
)) ||
663 (grfBSCF
== (BSCF_FIRSTDATANOTIFICATION
| BSCF_LASTDATANOTIFICATION
| BSCF_SKIPDRAINDATAFORFILEURLS
)),
664 "grcfBSCF = %08x\n", grfBSCF
);
665 }else if(direct_read
) {
669 if(!read_report_data
&& rec_depth
== 1) {
670 BOOL reported_all_data
= called_ReportData2
;
672 CHECK_EXPECT2(ReportData
);
675 ok(grfBSCF
== (BSCF_FIRSTDATANOTIFICATION
|BSCF_LASTDATANOTIFICATION
|BSCF_DATAFULLYAVAILABLE
)
676 || grfBSCF
== BSCF_FIRSTDATANOTIFICATION
, /* < IE8 */
677 "grcfBSCF = %08x\n", grfBSCF
);
678 CHECK_CALLED(Read
); /* Set in Continue */
679 first_data_notif
= FALSE
;
680 }else if(first_data_notif
) {
681 ok(grfBSCF
== BSCF_FIRSTDATANOTIFICATION
, "grcfBSCF = %08x\n", grfBSCF
);
682 first_data_notif
= FALSE
;
683 }else if(reported_all_data
) {
684 ok(grfBSCF
== (BSCF_LASTDATANOTIFICATION
|BSCF_INTERMEDIATEDATANOTIFICATION
),
685 "grcfBSCF = %08x\n", grfBSCF
);
686 }else if(!direct_read
) {
687 ok(grfBSCF
== BSCF_INTERMEDIATEDATANOTIFICATION
, "grcfBSCF = %08x\n", grfBSCF
);
695 SET_EXPECT(ReportData2
);
696 SET_EXPECT(ReportResult
);
697 hres
= IInternetProtocol_Read(binding_test
? binding_protocol
: async_protocol
, expect_pv
= buf
, sizeof(buf
), &read
);
699 ok(read
, "read == 0\n");
700 if(reported_all_data
) {
701 ok(hres
== S_FALSE
, "Read failed: %08x, expected S_FALSE\n", hres
);
705 if(!reported_all_data
&& called_ReportData2
) {
707 CHECK_CALLED(ReportData2
);
708 CHECK_CALLED(ReportResult
);
709 ok(hres
== S_OK
, "Read failed: %08x\n", hres
);
710 reported_all_data
= TRUE
;
713 CHECK_NOT_CALLED(ReportData2
);
714 CHECK_NOT_CALLED(ReportResult
);
716 }while(hres
== S_OK
);
718 wait_for_switch
= FALSE
;
720 CHECK_EXPECT(ReportData2
);
722 ok(grfBSCF
& BSCF_LASTDATANOTIFICATION
, "grfBSCF = %08x\n", grfBSCF
);
727 hres
= IInternetProtocol_Read(binding_test
? binding_protocol
: async_protocol
, expect_pv
= buf
, sizeof(buf
), &read
);
730 ok(hres
== S_FALSE
, "Read returned: %08x, expected E_FALSE\n", hres
);
731 ok(!read
, "read = %d\n", read
);
733 }else if(!binding_test
&& (tested_protocol
== HTTP_TEST
|| tested_protocol
== HTTPS_TEST
734 || tested_protocol
== FTP_TEST
)) {
735 if(!(grfBSCF
& BSCF_LASTDATANOTIFICATION
) || (grfBSCF
& BSCF_DATAFULLYAVAILABLE
))
736 CHECK_EXPECT(ReportData
);
737 else if (http_post_test
)
738 ok(ulProgress
== 13, "Read %u bytes instead of 13\n", ulProgress
);
740 ok(ulProgress
, "ulProgress == 0\n");
742 if(first_data_notif
) {
743 ok(grfBSCF
== BSCF_FIRSTDATANOTIFICATION
744 || grfBSCF
== (BSCF_LASTDATANOTIFICATION
|BSCF_DATAFULLYAVAILABLE
),
745 "grcfBSCF = %08x\n", grfBSCF
);
746 first_data_notif
= FALSE
;
748 ok(grfBSCF
== BSCF_INTERMEDIATEDATANOTIFICATION
749 || grfBSCF
== (BSCF_LASTDATANOTIFICATION
|BSCF_INTERMEDIATEDATANOTIFICATION
)
750 || broken(grfBSCF
== (BSCF_FIRSTDATANOTIFICATION
|BSCF_LASTDATANOTIFICATION
)),
751 "grcfBSCF = %08x\n", grfBSCF
);
754 if(!(bindf
& BINDF_FROMURLMON
) &&
755 !(grfBSCF
& BSCF_LASTDATANOTIFICATION
)) {
759 CHECK_CALLED(ReportProgress_FINDINGRESOURCE
);
760 CHECK_CALLED(ReportProgress_CONNECTING
);
762 CHECK_NOT_CALLED(ReportProgress_FINDINGRESOURCE
);
763 CHECK_NOT_CALLED(ReportProgress_CONNECTING
);
765 CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
766 CHECK_CALLED(OnResponse
);
767 CHECK_CALLED(ReportProgress_RAWMIMETYPE
);
769 SetEvent(event_complete
);
771 }else if(!read_report_data
) {
776 CHECK_EXPECT(ReportData
);
778 if(tested_protocol
!= BIND_TEST
) {
781 SET_EXPECT(MimeFilter_Read
);
782 else if(rec_depth
> 1)
786 hres
= IInternetProtocol_Read(binding_protocol
, expect_pv
=buf
, sizeof(buf
), &read
);
788 CHECK_CALLED(MimeFilter_Read
);
789 else if(rec_depth
> 1)
793 }while(hres
== S_OK
);
801 static HRESULT WINAPI
ProtocolSink_ReportResult(IInternetProtocolSink
*iface
, HRESULT hrResult
,
802 DWORD dwError
, LPCWSTR szResult
)
804 CHECK_EXPECT(ReportResult
);
806 if(tested_protocol
== FTP_TEST
)
807 ok(hrResult
== E_PENDING
|| hrResult
== S_OK
, "hrResult = %08x, expected E_PENDING or S_OK\n", hrResult
);
809 ok(hrResult
== expect_hrResult
, "hrResult = %08x, expected: %08x\n",
810 hrResult
, expect_hrResult
);
811 if(SUCCEEDED(hrResult
) || tested_protocol
== FTP_TEST
)
812 ok(dwError
== ERROR_SUCCESS
, "dwError = %d, expected ERROR_SUCCESS\n", dwError
);
814 ok(dwError
!= ERROR_SUCCESS
||
815 broken(tested_protocol
== MK_TEST
), /* Win9x, WinME and NT4 */
816 "dwError == ERROR_SUCCESS\n");
817 ok(!szResult
, "szResult != NULL\n");
820 SET_EXPECT(ReportData
); /* checked after main loop */
825 static IInternetProtocolSinkVtbl protocol_sink_vtbl
= {
826 ProtocolSink_QueryInterface
,
828 ProtocolSink_Release
,
830 ProtocolSink_ReportProgress
,
831 ProtocolSink_ReportData
,
832 ProtocolSink_ReportResult
835 static IInternetProtocolSink protocol_sink
= { &protocol_sink_vtbl
};
837 static HRESULT WINAPI
MimeProtocolSink_QueryInterface(IInternetProtocolSink
*iface
, REFIID riid
, void **ppv
)
839 ok(0, "unexpected call\n");
843 static ULONG WINAPI
MimeProtocolSink_AddRef(IInternetProtocolSink
*iface
)
848 static ULONG WINAPI
MimeProtocolSink_Release(IInternetProtocolSink
*iface
)
853 static HRESULT WINAPI
MimeProtocolSink_Switch(IInternetProtocolSink
*iface
, PROTOCOLDATA
*pProtocolData
)
857 CHECK_EXPECT(MimeFilter_Switch
);
860 hres
= IInternetProtocolSink_Switch(filtered_sink
, pProtocolData
);
861 ok(hres
== S_OK
, "Switch failed: %08x\n", hres
);
862 CHECK_CALLED(Switch
);
867 static HRESULT WINAPI
MimeProtocolSink_ReportProgress(IInternetProtocolSink
*iface
, ULONG ulStatusCode
,
868 LPCWSTR szStatusText
)
870 CHECK_EXPECT(MimeFilter_ReportProgress
);
874 static HRESULT WINAPI
MimeProtocolSink_ReportData(IInternetProtocolSink
*iface
, DWORD grfBSCF
,
875 ULONG ulProgress
, ULONG ulProgressMax
)
881 CHECK_EXPECT(MimeFilter_ReportData
);
885 hres
= IInternetProtocol_Read(filtered_protocol
, buf
, sizeof(buf
), &read
);
886 if(tested_protocol
== HTTP_TEST
)
887 ok(hres
== S_OK
|| hres
== E_PENDING
|| hres
== S_FALSE
, "Read failed: %08x\n", hres
);
889 ok(hres
== S_OK
, "Read failed: %08x\n", hres
);
892 SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
893 hres
= IInternetProtocolSink_ReportProgress(filtered_sink
, BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE
, text_htmlW
);
894 ok(hres
== S_OK
, "ReportProgress failed: %08x\n", hres
);
895 CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
897 /* FIXME: test BINDSTATUS_CACHEFILENAMEAVAILABLE */
900 if(!read_report_data
)
901 SET_EXPECT(ReportData
);
902 hres
= IInternetProtocolSink_ReportData(filtered_sink
, grfBSCF
, ulProgress
, ulProgressMax
);
903 ok(hres
== S_OK
, "ReportData failed: %08x\n", hres
);
904 if(!read_report_data
)
905 CHECK_CALLED(ReportData
);
913 static HRESULT WINAPI
MimeProtocolSink_ReportResult(IInternetProtocolSink
*iface
, HRESULT hrResult
,
914 DWORD dwError
, LPCWSTR szResult
)
918 CHECK_EXPECT(MimeFilter_ReportResult
);
920 ok(hrResult
== S_OK
, "hrResult = %08x\n", hrResult
);
921 ok(dwError
== ERROR_SUCCESS
, "dwError = %u\n", dwError
);
922 ok(!szResult
, "szResult = %s\n", wine_dbgstr_w(szResult
));
924 SET_EXPECT(ReportResult
);
925 hres
= IInternetProtocolSink_ReportResult(filtered_sink
, hrResult
, dwError
, szResult
);
926 ok(SUCCEEDED(hres
), "ReportResult failed: %08x\n", hres
);
927 CHECK_CALLED(ReportResult
);
932 static IInternetProtocolSinkVtbl mime_protocol_sink_vtbl
= {
933 MimeProtocolSink_QueryInterface
,
934 MimeProtocolSink_AddRef
,
935 MimeProtocolSink_Release
,
936 MimeProtocolSink_Switch
,
937 MimeProtocolSink_ReportProgress
,
938 MimeProtocolSink_ReportData
,
939 MimeProtocolSink_ReportResult
942 static IInternetProtocolSink mime_protocol_sink
= { &mime_protocol_sink_vtbl
};
944 static HRESULT
QueryInterface(REFIID riid
, void **ppv
)
946 static const IID IID_undocumented
= {0x58DFC7D0,0x5381,0x43E5,{0x9D,0x72,0x4C,0xDD,0xE4,0xCB,0x0F,0x1A}};
950 if(IsEqualGUID(&IID_IUnknown
, riid
) || IsEqualGUID(&IID_IInternetProtocolSink
, riid
))
951 *ppv
= &protocol_sink
;
952 if(IsEqualGUID(&IID_IServiceProvider
, riid
))
953 *ppv
= &service_provider
;
954 if(IsEqualGUID(&IID_IUriContainer
, riid
))
955 return E_NOINTERFACE
; /* TODO */
957 /* NOTE: IE8 queries for undocumented {58DFC7D0-5381-43E5-9D72-4CDDE4CB0F1A} interface. */
958 if(IsEqualGUID(&IID_undocumented
, riid
))
959 return E_NOINTERFACE
;
964 ok(0, "unexpected call %s\n", debugstr_guid(riid
));
965 return E_NOINTERFACE
;
968 static HRESULT WINAPI
BindInfo_QueryInterface(IInternetBindInfo
*iface
, REFIID riid
, void **ppv
)
970 if(IsEqualGUID(&IID_IUnknown
, riid
) || IsEqualGUID(&IID_IInternetBindInfo
, riid
)) {
974 return E_NOINTERFACE
;
977 static ULONG WINAPI
BindInfo_AddRef(IInternetBindInfo
*iface
)
982 static ULONG WINAPI
BindInfo_Release(IInternetBindInfo
*iface
)
987 static HRESULT WINAPI
BindInfo_GetBindInfo(IInternetBindInfo
*iface
, DWORD
*grfBINDF
, BINDINFO
*pbindinfo
)
991 static const CHAR szPostData
[] = "mode=Test";
993 CHECK_EXPECT(GetBindInfo
);
995 ok(grfBINDF
!= NULL
, "grfBINDF == NULL\n");
996 ok(pbindinfo
!= NULL
, "pbindinfo == NULL\n");
997 ok(pbindinfo
->cbSize
== sizeof(BINDINFO
), "wrong size of pbindinfo: %d\n", pbindinfo
->cbSize
);
1001 *grfBINDF
|= BINDF_FROMURLMON
;
1002 cbSize
= pbindinfo
->cbSize
;
1003 memset(pbindinfo
, 0, cbSize
);
1004 pbindinfo
->cbSize
= cbSize
;
1008 /* Must be GMEM_FIXED, GMEM_MOVABLE does not work properly
1009 * with urlmon on native (Win98 and WinXP) */
1010 U(pbindinfo
->stgmedData
).hGlobal
= GlobalAlloc(GPTR
, sizeof(szPostData
));
1011 if (!U(pbindinfo
->stgmedData
).hGlobal
)
1013 http_post_test
= FALSE
;
1014 skip("Out of memory\n");
1015 return E_OUTOFMEMORY
;
1017 lstrcpy((LPSTR
)U(pbindinfo
->stgmedData
).hGlobal
, szPostData
);
1018 pbindinfo
->cbstgmedData
= sizeof(szPostData
)-1;
1019 pbindinfo
->dwBindVerb
= BINDVERB_POST
;
1020 pbindinfo
->stgmedData
.tymed
= TYMED_HGLOBAL
;
1026 static HRESULT WINAPI
BindInfo_GetBindString(IInternetBindInfo
*iface
, ULONG ulStringType
,
1027 LPOLESTR
*ppwzStr
, ULONG cEl
, ULONG
*pcElFetched
)
1029 ok(ppwzStr
!= NULL
, "ppwzStr == NULL\n");
1030 ok(pcElFetched
!= NULL
, "pcElFetched == NULL\n");
1032 switch(ulStringType
) {
1033 case BINDSTRING_ACCEPT_MIMES
:
1034 CHECK_EXPECT(GetBindString_ACCEPT_MIMES
);
1035 ok(cEl
== 256, "cEl=%d, expected 256\n", cEl
);
1037 ok(*pcElFetched
== 256, "*pcElFetched=%d, expected 256\n", *pcElFetched
);
1041 *ppwzStr
= CoTaskMemAlloc(sizeof(acc_mimeW
));
1042 memcpy(*ppwzStr
, acc_mimeW
, sizeof(acc_mimeW
));
1045 case BINDSTRING_USER_AGENT
:
1046 CHECK_EXPECT(GetBindString_USER_AGENT
);
1047 ok(cEl
== 1, "cEl=%d, expected 1\n", cEl
);
1049 ok(*pcElFetched
== 0, "*pcElFetch=%d, expectd 0\n", *pcElFetched
);
1053 *ppwzStr
= CoTaskMemAlloc(sizeof(user_agentW
));
1054 memcpy(*ppwzStr
, user_agentW
, sizeof(user_agentW
));
1057 case BINDSTRING_POST_COOKIE
:
1058 CHECK_EXPECT(GetBindString_POST_COOKIE
);
1059 ok(cEl
== 1, "cEl=%d, expected 1\n", cEl
);
1061 ok(*pcElFetched
== 0, "*pcElFetch=%d, expectd 0\n", *pcElFetched
);
1063 case BINDSTRING_URL
: {
1066 CHECK_EXPECT(GetBindString_URL
);
1067 ok(cEl
== 1, "cEl=%d, expected 1\n", cEl
);
1068 ok(*pcElFetched
== 0, "*pcElFetch=%d, expectd 0\n", *pcElFetched
);
1071 size
= (lstrlenW(binding_urls
[tested_protocol
])+1)*sizeof(WCHAR
);
1072 *ppwzStr
= CoTaskMemAlloc(size
);
1073 memcpy(*ppwzStr
, binding_urls
[tested_protocol
], size
);
1077 ok(0, "unexpected call\n");
1083 static IInternetBindInfoVtbl bind_info_vtbl
= {
1084 BindInfo_QueryInterface
,
1087 BindInfo_GetBindInfo
,
1088 BindInfo_GetBindString
1091 static IInternetBindInfo bind_info
= { &bind_info_vtbl
};
1093 static HRESULT WINAPI
InternetPriority_QueryInterface(IInternetPriority
*iface
,
1094 REFIID riid
, void **ppv
)
1096 ok(0, "unexpected call\n");
1097 return E_NOINTERFACE
;
1100 static ULONG WINAPI
InternetPriority_AddRef(IInternetPriority
*iface
)
1105 static ULONG WINAPI
InternetPriority_Release(IInternetPriority
*iface
)
1110 static HRESULT WINAPI
InternetPriority_SetPriority(IInternetPriority
*iface
, LONG nPriority
)
1112 CHECK_EXPECT(SetPriority
);
1113 ok(nPriority
== ex_priority
, "nPriority=%d\n", nPriority
);
1117 static HRESULT WINAPI
InternetPriority_GetPriority(IInternetPriority
*iface
, LONG
*pnPriority
)
1119 ok(0, "unexpected call\n");
1124 static const IInternetPriorityVtbl InternetPriorityVtbl
= {
1125 InternetPriority_QueryInterface
,
1126 InternetPriority_AddRef
,
1127 InternetPriority_Release
,
1128 InternetPriority_SetPriority
,
1129 InternetPriority_GetPriority
1132 static IInternetPriority InternetPriority
= { &InternetPriorityVtbl
};
1134 static ULONG WINAPI
Protocol_AddRef(IInternetProtocol
*iface
)
1139 static ULONG WINAPI
Protocol_Release(IInternetProtocol
*iface
)
1144 static HRESULT WINAPI
Protocol_Abort(IInternetProtocol
*iface
, HRESULT hrReason
,
1147 ok(0, "unexpected call\n");
1151 static HRESULT WINAPI
Protocol_Suspend(IInternetProtocol
*iface
)
1153 ok(0, "unexpected call\n");
1157 static HRESULT WINAPI
Protocol_Resume(IInternetProtocol
*iface
)
1159 ok(0, "unexpected call\n");
1163 static HRESULT WINAPI
Protocol_Seek(IInternetProtocol
*iface
,
1164 LARGE_INTEGER dlibMove
, DWORD dwOrigin
, ULARGE_INTEGER
*plibNewPosition
)
1166 ok(0, "unexpected call\n");
1170 static HRESULT WINAPI
ProtocolEmul_QueryInterface(IInternetProtocol
*iface
, REFIID riid
, void **ppv
)
1172 if(IsEqualGUID(&IID_IUnknown
, riid
) || IsEqualGUID(&IID_IInternetProtocol
, riid
)) {
1177 if(IsEqualGUID(&IID_IInternetProtocolEx
, riid
)) {
1178 trace("IID_IInternetProtocolEx not supported\n");
1180 return E_NOINTERFACE
;
1183 if(IsEqualGUID(&IID_IInternetPriority
, riid
)) {
1184 *ppv
= &InternetPriority
;
1188 if(IsEqualGUID(&IID_IWinInetInfo
, riid
)) {
1189 CHECK_EXPECT(QueryInterface_IWinInetInfo
);
1191 return E_NOINTERFACE
;
1194 if(IsEqualGUID(&IID_IWinInetHttpInfo
, riid
)) {
1195 CHECK_EXPECT(QueryInterface_IWinInetHttpInfo
);
1197 return E_NOINTERFACE
;
1200 ok(0, "unexpected riid %s\n", debugstr_guid(riid
));
1202 return E_NOINTERFACE
;
1205 static DWORD WINAPI
thread_proc(PVOID arg
)
1209 memset(&protocoldata
, -1, sizeof(protocoldata
));
1213 SET_EXPECT(ReportProgress_FINDINGRESOURCE
);
1214 hres
= IInternetProtocolSink_ReportProgress(binding_sink
,
1215 BINDSTATUS_FINDINGRESOURCE
, hostW
);
1216 CHECK_CALLED(ReportProgress_FINDINGRESOURCE
);
1217 ok(hres
== S_OK
, "ReportProgress failed: %08x\n", hres
);
1219 SET_EXPECT(ReportProgress_CONNECTING
);
1220 hres
= IInternetProtocolSink_ReportProgress(binding_sink
,
1221 BINDSTATUS_CONNECTING
, winehq_ipW
);
1222 CHECK_CALLED(ReportProgress_CONNECTING
);
1223 ok(hres
== S_OK
, "ReportProgress failed: %08x\n", hres
);
1225 SET_EXPECT(ReportProgress_SENDINGREQUEST
);
1226 hres
= IInternetProtocolSink_ReportProgress(binding_sink
,
1227 BINDSTATUS_SENDINGREQUEST
, NULL
);
1228 CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
1229 ok(hres
== S_OK
, "ReportProgress failed: %08x\n", hres
);
1233 hres
= IInternetProtocolSink_Switch(binding_sink
, &protocoldata
);
1234 CHECK_CALLED(Switch
);
1235 ok(hres
== S_OK
, "Switch failed: %08x\n", hres
);
1240 SET_EXPECT(MimeFilter_Switch
);
1243 hres
= IInternetProtocolSink_Switch(binding_sink
, &protocoldata
);
1244 ok(hres
== S_OK
, "Switch failed: %08x\n", hres
);
1246 CHECK_CALLED(MimeFilter_Switch
);
1248 CHECK_CALLED(Switch
);
1252 SET_EXPECT(MimeFilter_Switch
);
1255 hres
= IInternetProtocolSink_Switch(binding_sink
, &protocoldata
);
1256 ok(hres
== S_OK
, "Switch failed: %08x\n", hres
);
1258 CHECK_CALLED(MimeFilter_Switch
);
1260 CHECK_CALLED(Switch
);
1264 SET_EXPECT(MimeFilter_Switch
);
1267 hres
= IInternetProtocolSink_Switch(binding_sink
, &protocoldata
);
1268 ok(hres
== S_OK
, "Switch failed: %08x\n", hres
);
1270 CHECK_CALLED(MimeFilter_Switch
);
1272 CHECK_CALLED(Switch
);
1275 SetEvent(event_complete
);
1280 static HRESULT WINAPI
ProtocolEmul_Start(IInternetProtocol
*iface
, LPCWSTR szUrl
,
1281 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
1282 DWORD grfPI
, HANDLE_PTR dwReserved
)
1284 BINDINFO bindinfo
, exp_bindinfo
;
1288 CHECK_EXPECT(Start
);
1290 ok(pOIProtSink
!= NULL
, "pOIProtSink == NULL\n");
1291 ok(pOIBindInfo
!= NULL
, "pOIBindInfo == NULL\n");
1292 ok(pOIProtSink
!= &protocol_sink
, "unexpected pOIProtSink\n");
1293 ok(pOIBindInfo
!= &bind_info
, "unexpected pOIBindInfo\n");
1294 ok(!grfPI
, "grfPI = %x\n", grfPI
);
1295 ok(!dwReserved
, "dwReserved = %lx\n", dwReserved
);
1298 ok(pOIProtSink
== binding_sink
, "pOIProtSink != binding_sink\n");
1300 memset(&bindinfo
, 0, sizeof(bindinfo
));
1301 bindinfo
.cbSize
= sizeof(bindinfo
);
1302 memcpy(&exp_bindinfo
, &bindinfo
, sizeof(bindinfo
));
1303 SET_EXPECT(GetBindInfo
);
1304 hres
= IInternetBindInfo_GetBindInfo(pOIBindInfo
, &cbindf
, &bindinfo
);
1305 ok(hres
== S_OK
, "GetBindInfo failed: %08x\n", hres
);
1306 CHECK_CALLED(GetBindInfo
);
1307 ok(cbindf
== (bindf
|BINDF_FROMURLMON
), "bindf = %x, expected %x\n",
1308 cbindf
, (bindf
|BINDF_FROMURLMON
));
1309 ok(!memcmp(&exp_bindinfo
, &bindinfo
, sizeof(bindinfo
)), "unexpected bindinfo\n");
1310 ReleaseBindInfo(&bindinfo
);
1312 SET_EXPECT(ReportProgress_SENDINGREQUEST
);
1313 hres
= IInternetProtocolSink_ReportProgress(pOIProtSink
, BINDSTATUS_SENDINGREQUEST
, emptyW
);
1314 ok(hres
== S_OK
, "ReportProgress(BINDSTATUS_SENDINGREQUEST) failed: %08x\n", hres
);
1315 CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
1317 if(tested_protocol
== HTTP_TEST
|| tested_protocol
== HTTPS_TEST
) {
1318 IServiceProvider
*service_provider
;
1319 IHttpNegotiate
*http_negotiate
;
1320 IHttpNegotiate2
*http_negotiate2
;
1321 LPWSTR ua
= (LPWSTR
)0xdeadbeef, accept_mimes
[256];
1322 LPWSTR additional_headers
= NULL
;
1324 DWORD fetched
= 0, size
= 100;
1327 SET_EXPECT(GetBindString_USER_AGENT
);
1328 hres
= IInternetBindInfo_GetBindString(pOIBindInfo
, BINDSTRING_USER_AGENT
,
1330 CHECK_CALLED(GetBindString_USER_AGENT
);
1331 ok(hres
== S_OK
, "GetBindString(BINDSTRING_USER_AGETNT) failed: %08x\n", hres
);
1332 ok(fetched
== 1, "fetched = %d, expected 254\n", fetched
);
1333 ok(ua
!= NULL
, "ua = %p\n", ua
);
1334 ok(!strcmp_ww(ua
, user_agentW
), "unexpected user agent %s\n", wine_dbgstr_w(ua
));
1338 SET_EXPECT(GetBindString_ACCEPT_MIMES
);
1339 hres
= IInternetBindInfo_GetBindString(pOIBindInfo
, BINDSTRING_ACCEPT_MIMES
,
1340 accept_mimes
, 256, &fetched
);
1341 CHECK_CALLED(GetBindString_ACCEPT_MIMES
);
1344 "GetBindString(BINDSTRING_ACCEPT_MIMES) failed: %08x\n", hres
);
1345 ok(fetched
== 1, "fetched = %d, expected 1\n", fetched
);
1346 ok(!strcmp_ww(acc_mimeW
, accept_mimes
[0]), "unexpected mimes %s\n", wine_dbgstr_w(accept_mimes
[0]));
1348 hres
= IInternetBindInfo_QueryInterface(pOIBindInfo
, &IID_IServiceProvider
,
1349 (void**)&service_provider
);
1350 ok(hres
== S_OK
, "QueryInterface failed: %08x\n", hres
);
1352 SET_EXPECT(QueryService_HttpNegotiate
);
1353 hres
= IServiceProvider_QueryService(service_provider
, &IID_IHttpNegotiate
,
1354 &IID_IHttpNegotiate
, (void**)&http_negotiate
);
1355 CHECK_CALLED(QueryService_HttpNegotiate
);
1356 ok(hres
== S_OK
, "QueryService failed: %08x\n", hres
);
1358 SET_EXPECT(BeginningTransaction
);
1359 hres
= IHttpNegotiate_BeginningTransaction(http_negotiate
, binding_urls
[tested_protocol
],
1360 NULL
, 0, &additional_headers
);
1361 CHECK_CALLED(BeginningTransaction
);
1362 IHttpNegotiate_Release(http_negotiate
);
1363 ok(hres
== S_OK
, "BeginningTransction failed: %08x\n", hres
);
1364 ok(additional_headers
== NULL
, "additional_headers=%p\n", additional_headers
);
1366 SET_EXPECT(QueryService_HttpNegotiate
);
1367 hres
= IServiceProvider_QueryService(service_provider
, &IID_IHttpNegotiate2
,
1368 &IID_IHttpNegotiate2
, (void**)&http_negotiate2
);
1369 CHECK_CALLED(QueryService_HttpNegotiate
);
1370 ok(hres
== S_OK
, "QueryService failed: %08x\n", hres
);
1373 SET_EXPECT(GetRootSecurityId
);
1374 hres
= IHttpNegotiate2_GetRootSecurityId(http_negotiate2
, sec_id
, &size
, 0);
1375 CHECK_CALLED(GetRootSecurityId
);
1376 IHttpNegotiate2_Release(http_negotiate2
);
1377 ok(hres
== E_FAIL
, "GetRootSecurityId failed: %08x, expected E_FAIL\n", hres
);
1378 ok(size
== 13, "size=%d\n", size
);
1380 IServiceProvider_Release(service_provider
);
1382 CreateThread(NULL
, 0, thread_proc
, NULL
, 0, &tid
);
1387 SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE
);
1388 hres
= IInternetProtocolSink_ReportProgress(pOIProtSink
,
1389 BINDSTATUS_CACHEFILENAMEAVAILABLE
, expect_wsz
= emptyW
);
1390 ok(hres
== S_OK
, "ReportProgress(BINDSTATUS_CACHEFILENAMEAVAILABLE) failed: %08x\n", hres
);
1391 CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE
);
1393 if(mimefilter_test
) {
1394 SET_EXPECT(MimeFilter_CreateInstance
);
1395 SET_EXPECT(MimeFilter_Start
);
1396 SET_EXPECT(ReportProgress_LOADINGMIMEHANDLER
);
1398 SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
1399 hres
= IInternetProtocolSink_ReportProgress(pOIProtSink
, BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE
,
1400 mimefilter_test
? gzipW
: (expect_wsz
= text_htmlW
));
1402 "ReportProgress(BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE) failed: %08x\n", hres
);
1403 if(mimefilter_test
) {
1404 CHECK_CALLED(MimeFilter_CreateInstance
);
1405 CHECK_CALLED(MimeFilter_Start
);
1406 CHECK_CALLED(ReportProgress_LOADINGMIMEHANDLER
);
1407 todo_wine
CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
1409 CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
1413 SET_EXPECT(MimeFilter_ReportData
);
1415 SET_EXPECT(ReportData
);
1416 hres
= IInternetProtocolSink_ReportData(pOIProtSink
,
1417 BSCF_FIRSTDATANOTIFICATION
| BSCF_LASTDATANOTIFICATION
, 13, 13);
1418 ok(hres
== S_OK
, "ReportData failed: %08x\n", hres
);
1420 CHECK_CALLED(MimeFilter_ReportData
);
1422 CHECK_CALLED(ReportData
);
1424 if(tested_protocol
== BIND_TEST
) {
1425 hres
= IInternetProtocol_Terminate(binding_protocol
, 0);
1426 ok(hres
== E_FAIL
, "Termiante failed: %08x\n", hres
);
1430 SET_EXPECT(MimeFilter_ReportResult
);
1432 SET_EXPECT(ReportResult
);
1433 hres
= IInternetProtocolSink_ReportResult(pOIProtSink
, S_OK
, 0, NULL
);
1434 ok(hres
== S_OK
, "ReportResult failed: %08x\n", hres
);
1436 CHECK_CALLED(MimeFilter_ReportResult
);
1438 CHECK_CALLED(ReportResult
);
1443 static HRESULT WINAPI
ProtocolEmul_Continue(IInternetProtocol
*iface
,
1444 PROTOCOLDATA
*pProtocolData
)
1449 CHECK_EXPECT(Continue
);
1451 ok(pProtocolData
!= NULL
, "pProtocolData == NULL\n");
1452 if(!pProtocolData
|| tested_protocol
== BIND_TEST
)
1455 ok(pProtocolData
!= &protocoldata
, "pProtocolData == &protocoldata\n");
1456 ok(pProtocolData
->grfFlags
== protocoldata
.grfFlags
, "grfFlags wrong %x/%x\n",
1457 pProtocolData
->grfFlags
, protocoldata
.grfFlags
);
1458 ok(pProtocolData
->dwState
== protocoldata
.dwState
, "dwState wrong %x/%x\n",
1459 pProtocolData
->dwState
, protocoldata
.dwState
);
1460 ok(pProtocolData
->pData
== protocoldata
.pData
, "pData wrong %p/%p\n",
1461 pProtocolData
->pData
, protocoldata
.pData
);
1462 ok(pProtocolData
->cbData
== protocoldata
.cbData
, "cbData wrong %x/%x\n",
1463 pProtocolData
->cbData
, protocoldata
.cbData
);
1466 switch(prot_state
) {
1468 IServiceProvider
*service_provider
;
1469 IHttpNegotiate
*http_negotiate
;
1470 static WCHAR header
[] = {'?',0};
1472 hres
= IInternetProtocolSink_QueryInterface(binding_sink
, &IID_IServiceProvider
,
1473 (void**)&service_provider
);
1474 ok(hres
== S_OK
, "Could not get IServiceProvicder\n");
1476 SET_EXPECT(QueryService_HttpNegotiate
);
1477 hres
= IServiceProvider_QueryService(service_provider
, &IID_IHttpNegotiate
,
1478 &IID_IHttpNegotiate
, (void**)&http_negotiate
);
1479 IServiceProvider_Release(service_provider
);
1480 CHECK_CALLED(QueryService_HttpNegotiate
);
1481 ok(hres
== S_OK
, "Could not get IHttpNegotiate\n");
1483 SET_EXPECT(OnResponse
);
1484 hres
= IHttpNegotiate_OnResponse(http_negotiate
, 200, header
, NULL
, NULL
);
1485 IHttpNegotiate_Release(http_negotiate
);
1486 CHECK_CALLED(OnResponse
);
1487 IHttpNegotiate_Release(http_negotiate
);
1488 ok(hres
== S_OK
, "OnResponse failed: %08x\n", hres
);
1490 if(mimefilter_test
) {
1491 SET_EXPECT(MimeFilter_CreateInstance
);
1492 SET_EXPECT(MimeFilter_Start
);
1493 SET_EXPECT(ReportProgress_LOADINGMIMEHANDLER
);
1494 }else if(!(pi
& PI_MIMEVERIFICATION
)) {
1495 SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
1497 hres
= IInternetProtocolSink_ReportProgress(binding_sink
,
1498 BINDSTATUS_MIMETYPEAVAILABLE
, mimefilter_test
? gzipW
: text_htmlW
);
1499 if(mimefilter_test
) {
1500 CHECK_CALLED(MimeFilter_CreateInstance
);
1501 CHECK_CALLED(MimeFilter_Start
);
1502 CHECK_CALLED(ReportProgress_LOADINGMIMEHANDLER
);
1503 }else if(!(pi
& PI_MIMEVERIFICATION
)) {
1504 CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
1507 "ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE) failed: %08x\n", hres
);
1509 bscf
|= BSCF_FIRSTDATANOTIFICATION
;
1514 bscf
= BSCF_INTERMEDIATEDATANOTIFICATION
;
1519 if(mimefilter_test
) {
1520 SET_EXPECT(MimeFilter_ReportData
);
1521 }else if(pi
& PI_MIMEVERIFICATION
) {
1523 SET_EXPECT(Read
); /* checked in ReportData for short_read */
1525 if(!mimefilter_test
)
1526 SET_EXPECT(Read
); /* checked in BINDSTATUS_MIMETYPEAVAILABLE or ReportData */
1527 SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
1530 SET_EXPECT(ReportData
);
1532 SET_EXPECT(ReportData
);
1535 hres
= IInternetProtocolSink_ReportData(binding_sink
, bscf
, pr
, 400);
1536 ok(hres
== S_OK
, "ReportData failed: %08x\n", hres
);
1538 if(mimefilter_test
) {
1539 SET_EXPECT(MimeFilter_ReportData
);
1540 }else if(pi
& PI_MIMEVERIFICATION
) {
1541 if(!short_read
&& pr
< 200)
1544 CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
1547 CHECK_CALLED(ReportData
);
1556 static HRESULT WINAPI
ProtocolEmul_Terminate(IInternetProtocol
*iface
, DWORD dwOptions
)
1558 CHECK_EXPECT(Terminate
);
1559 ok(!dwOptions
, "dwOptions=%d\n", dwOptions
);
1563 static HRESULT WINAPI
ProtocolEmul_Read(IInternetProtocol
*iface
, void *pv
,
1564 ULONG cb
, ULONG
*pcbRead
)
1566 if(read_report_data
)
1567 CHECK_EXPECT2(Read2
);
1569 if(mimefilter_test
|| short_read
) {
1570 if(!read_report_data
)
1571 CHECK_EXPECT2(Read
);
1572 }else if((pi
& PI_MIMEVERIFICATION
)) {
1573 if(!read_report_data
)
1574 CHECK_EXPECT2(Read
);
1576 if(prot_read
< 300) {
1577 ok(pv
!= expect_pv
, "pv == expect_pv\n");
1579 ok(cb
== 2048-prot_read
, "cb=%d\n", cb
);
1581 ok(cb
== 700, "cb=%d\n", cb
);
1583 ok(expect_pv
<= pv
&& (BYTE
*)pv
< (BYTE
*)expect_pv
+ cb
, "pv != expect_pv\n");
1586 if(!read_report_data
)
1589 ok(pv
== expect_pv
, "pv != expect_pv\n");
1590 ok(cb
== 1000, "cb=%d\n", cb
);
1591 ok(!*pcbRead
, "*pcbRead = %d\n", *pcbRead
);
1593 ok(pcbRead
!= NULL
, "pcbRead == NULL\n");
1595 if(prot_state
== 3 || (short_read
&& prot_state
!= 4)) {
1600 SET_EXPECT(Read2
); /* checked in BINDSTATUS_MIMETYPEAVAILABLE */
1601 SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
1604 SET_EXPECT(MimeFilter_ReportData
);
1605 else if(direct_read
)
1606 SET_EXPECT(ReportData2
);
1608 hres
= IInternetProtocolSink_ReportData(binding_sink
,
1609 BSCF_LASTDATANOTIFICATION
|BSCF_INTERMEDIATEDATANOTIFICATION
, 0, 0);
1611 ok(hres
== S_OK
, "ReportData failed: %08x\n", hres
);
1613 CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
1615 CHECK_CALLED(MimeFilter_ReportData
);
1616 else if(direct_read
)
1617 CHECK_CALLED(ReportData2
);
1620 SET_EXPECT(MimeFilter_ReportResult
);
1622 SET_EXPECT(ReportResult
);
1623 hres
= IInternetProtocolSink_ReportResult(binding_sink
, S_OK
, ERROR_SUCCESS
, NULL
);
1624 ok(hres
== S_OK
, "ReportResult failed: %08x\n", hres
);
1626 CHECK_CALLED(MimeFilter_ReportResult
);
1628 CHECK_CALLED(ReportResult
);
1630 memset(pv
, 'x', 100);
1631 prot_read
+= *pcbRead
= 100;
1633 }if(prot_state
== 4) {
1638 if((async_read_pending
= !async_read_pending
)) {
1640 return tested_protocol
== HTTP_TEST
|| tested_protocol
== HTTPS_TEST
? E_PENDING
: S_FALSE
;
1643 memset(pv
, 'x', 100);
1644 prot_read
+= *pcbRead
= 100;
1648 static HRESULT WINAPI
ProtocolEmul_LockRequest(IInternetProtocol
*iface
, DWORD dwOptions
)
1650 CHECK_EXPECT(LockRequest
);
1651 ok(dwOptions
== 0, "dwOptions=%x\n", dwOptions
);
1655 static HRESULT WINAPI
ProtocolEmul_UnlockRequest(IInternetProtocol
*iface
)
1657 CHECK_EXPECT(UnlockRequest
);
1661 static const IInternetProtocolVtbl ProtocolVtbl
= {
1662 ProtocolEmul_QueryInterface
,
1666 ProtocolEmul_Continue
,
1668 ProtocolEmul_Terminate
,
1673 ProtocolEmul_LockRequest
,
1674 ProtocolEmul_UnlockRequest
1677 static IInternetProtocol Protocol
= { &ProtocolVtbl
};
1679 static HRESULT WINAPI
MimeProtocol_QueryInterface(IInternetProtocol
*iface
, REFIID riid
, void **ppv
)
1681 if(IsEqualGUID(&IID_IUnknown
, riid
) || IsEqualGUID(&IID_IInternetProtocol
, riid
)) {
1686 if(IsEqualGUID(&IID_IInternetProtocolSink
, riid
)) {
1687 *ppv
= &mime_protocol_sink
;
1691 ok(0, "unexpected riid %s\n", debugstr_guid(riid
));
1693 return E_NOINTERFACE
;
1696 static HRESULT WINAPI
MimeProtocol_Start(IInternetProtocol
*iface
, LPCWSTR szUrl
,
1697 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
1698 DWORD grfPI
, HANDLE_PTR dwReserved
)
1700 PROTOCOLFILTERDATA
*data
;
1701 LPOLESTR url_str
= NULL
;
1707 CHECK_EXPECT(MimeFilter_Start
);
1709 ok(!strcmp_ww(szUrl
, gzipW
), "wrong url %s\n", wine_dbgstr_w(szUrl
));
1710 ok(grfPI
== (PI_FILTER_MODE
|PI_FORCE_ASYNC
), "grfPI=%x, expected PI_FILTER_MODE|PI_FORCE_ASYNC\n", grfPI
);
1711 ok(dwReserved
, "dwReserved == 0\n");
1712 ok(pOIProtSink
!= NULL
, "pOIProtSink == NULL\n");
1713 ok(pOIBindInfo
!= NULL
, "pOIBindInfo == NULL\n");
1716 ok(pOIProtSink
!= binding_sink
, "pOIProtSink == protocol_sink\n");
1717 ok(pOIBindInfo
== prot_bind_info
, "pOIBindInfo != bind_info\n");
1719 ok(pOIProtSink
== &protocol_sink
, "pOIProtSink != protocol_sink\n");
1720 ok(pOIBindInfo
== &bind_info
, "pOIBindInfo != bind_info\n");
1723 data
= (void*)dwReserved
;
1724 ok(data
->cbSize
== sizeof(*data
), "data->cbSize = %d\n", data
->cbSize
);
1725 ok(!data
->pProtocolSink
, "data->pProtocolSink != NULL\n");
1726 ok(data
->pProtocol
!= NULL
, "data->pProtocol == NULL\n");
1727 ok(!data
->pUnk
, "data->pUnk != NULL\n");
1728 ok(!data
->dwFilterFlags
, "data->dwProtocolFlags = %x\n", data
->dwFilterFlags
);
1730 IInternetProtocolSink
*prot_sink
;
1732 IInternetProtocol_QueryInterface(data
->pProtocol
, &IID_IInternetProtocolSink
, (void**)&prot_sink
);
1733 ok(prot_sink
== pOIProtSink
, "QI(data->pProtocol, IID_IInternetProtocolSink) != pOIProtSink\n");
1734 IInternetProtocolSink_Release(prot_sink
);
1736 ok(data
->pProtocol
!= binding_protocol
, "data->pProtocol == binding_protocol\n");
1738 filtered_protocol
= data
->pProtocol
;
1739 IInternetProtocol_AddRef(filtered_protocol
);
1741 IInternetProtocol
*prot
;
1743 IInternetProtocol_QueryInterface(data
->pProtocol
, &IID_IInternetProtocol
, (void**)&prot
);
1744 ok(prot
== async_protocol
, "QI(data->pProtocol, IID_IInternetProtocol) != async_protocol\n");
1745 IInternetProtocol_Release(prot
);
1747 ok(data
->pProtocol
!= async_protocol
, "data->pProtocol == async_protocol\n");
1750 filtered_sink
= pOIProtSink
;
1752 SET_EXPECT(ReportProgress_DECODING
);
1753 hres
= IInternetProtocolSink_ReportProgress(pOIProtSink
, BINDSTATUS_DECODING
, gzipW
);
1754 ok(hres
== S_OK
, "ReportProgress(BINDSTATUS_DECODING) failed: %08x\n", hres
);
1755 CHECK_CALLED(ReportProgress_DECODING
);
1757 SET_EXPECT(GetBindInfo
);
1758 memset(&bindinfo
, 0, sizeof(bindinfo
));
1759 bindinfo
.cbSize
= sizeof(bindinfo
);
1760 hres
= IInternetBindInfo_GetBindInfo(pOIBindInfo
, &cbindf
, &bindinfo
);
1761 ok(hres
== S_OK
, "GetBindInfo failed: %08x\n", hres
);
1762 ok(cbindf
== (bindf
|BINDF_FROMURLMON
), "cbindf = %x, expected %x\n", cbindf
, bindf
);
1763 CHECK_CALLED(GetBindInfo
);
1765 SET_EXPECT(GetBindString_URL
);
1766 hres
= IInternetBindInfo_GetBindString(pOIBindInfo
, BINDSTRING_URL
, &url_str
, 1, &fetched
);
1767 ok(hres
== S_OK
, "GetBindString(BINDSTRING_URL) failed: %08x\n", hres
);
1768 ok(fetched
== 1, "fetched = %d\n", fetched
);
1769 ok(!strcmp_ww(url_str
, binding_urls
[tested_protocol
]), "wrong url_str %s\n", wine_dbgstr_w(url_str
));
1770 CoTaskMemFree(url_str
);
1771 CHECK_CALLED(GetBindString_URL
);
1776 static HRESULT WINAPI
Protocol_Continue(IInternetProtocol
*iface
,
1777 PROTOCOLDATA
*pProtocolData
)
1779 CHECK_EXPECT(MimeFilter_Continue
);
1783 static HRESULT WINAPI
MimeProtocol_Terminate(IInternetProtocol
*iface
, DWORD dwOptions
)
1787 CHECK_EXPECT(MimeFilter_Terminate
);
1789 ok(!dwOptions
, "dwOptions = %x\n", dwOptions
);
1791 SET_EXPECT(Terminate
);
1792 hres
= IInternetProtocol_Terminate(filtered_protocol
, dwOptions
);
1793 ok(hres
== S_OK
, "Terminate failed: %08x\n", hres
);
1794 CHECK_CALLED(Terminate
);
1799 static HRESULT WINAPI
MimeProtocol_Read(IInternetProtocol
*iface
, void *pv
,
1800 ULONG cb
, ULONG
*pcbRead
)
1806 CHECK_EXPECT(MimeFilter_Read
);
1808 ok(pv
!= NULL
, "pv == NULL\n");
1809 ok(cb
!= 0, "cb == 0\n");
1810 ok(pcbRead
!= NULL
, "pcbRead == NULL\n");
1812 if(read_report_data
)
1816 hres
= IInternetProtocol_Read(filtered_protocol
, buf
, sizeof(buf
), &read
);
1817 ok(hres
== S_OK
|| hres
== S_FALSE
|| hres
== E_PENDING
, "Read failed: %08x\n", hres
);
1818 if(read_report_data
)
1819 CHECK_CALLED(Read2
);
1824 ok(*pcbRead
== 0, "*pcbRead=%d, expected 0\n", *pcbRead
);
1828 memset(pv
, 'x', read
);
1832 static HRESULT WINAPI
MimeProtocol_LockRequest(IInternetProtocol
*iface
, DWORD dwOptions
)
1836 CHECK_EXPECT(MimeFilter_LockRequest
);
1838 ok(!dwOptions
, "dwOptions = %x\n", dwOptions
);
1840 SET_EXPECT(LockRequest
);
1841 hres
= IInternetProtocol_LockRequest(filtered_protocol
, dwOptions
);
1842 ok(hres
== S_OK
, "LockRequest failed: %08x\n", hres
);
1843 CHECK_CALLED(LockRequest
);
1848 static HRESULT WINAPI
MimeProtocol_UnlockRequest(IInternetProtocol
*iface
)
1852 CHECK_EXPECT(MimeFilter_UnlockRequest
);
1854 SET_EXPECT(UnlockRequest
);
1855 hres
= IInternetProtocol_UnlockRequest(filtered_protocol
);
1856 ok(hres
== S_OK
, "UnlockRequest failed: %08x\n", hres
);
1857 CHECK_CALLED(UnlockRequest
);
1862 static const IInternetProtocolVtbl MimeProtocolVtbl
= {
1863 MimeProtocol_QueryInterface
,
1869 MimeProtocol_Terminate
,
1874 MimeProtocol_LockRequest
,
1875 MimeProtocol_UnlockRequest
1878 static IInternetProtocol MimeProtocol
= { &MimeProtocolVtbl
};
1880 static HRESULT WINAPI
ClassFactory_QueryInterface(IClassFactory
*iface
, REFIID riid
, void **ppv
)
1882 ok(0, "unexpected call\n");
1883 return E_NOINTERFACE
;
1886 static ULONG WINAPI
ClassFactory_AddRef(IClassFactory
*iface
)
1891 static ULONG WINAPI
ClassFactory_Release(IClassFactory
*iface
)
1896 static HRESULT WINAPI
ClassFactory_CreateInstance(IClassFactory
*iface
, IUnknown
*pOuter
,
1897 REFIID riid
, void **ppv
)
1899 CHECK_EXPECT(CreateInstance
);
1901 ok(pOuter
== (IUnknown
*)prot_bind_info
, "pOuter != protocol_unk\n");
1902 ok(IsEqualGUID(&IID_IUnknown
, riid
), "unexpected riid %s\n", debugstr_guid(riid
));
1903 ok(ppv
!= NULL
, "ppv == NULL\n");
1909 static HRESULT WINAPI
ClassFactory_LockServer(IClassFactory
*iface
, BOOL dolock
)
1911 ok(0, "unexpected call\n");
1915 static const IClassFactoryVtbl ClassFactoryVtbl
= {
1916 ClassFactory_QueryInterface
,
1917 ClassFactory_AddRef
,
1918 ClassFactory_Release
,
1919 ClassFactory_CreateInstance
,
1920 ClassFactory_LockServer
1923 static IClassFactory ClassFactory
= { &ClassFactoryVtbl
};
1925 static HRESULT WINAPI
MimeFilter_CreateInstance(IClassFactory
*iface
, IUnknown
*outer
, REFIID riid
, void **ppv
)
1927 CHECK_EXPECT(MimeFilter_CreateInstance
);
1929 ok(!outer
, "outer = %p\n", outer
);
1930 ok(IsEqualGUID(&IID_IInternetProtocol
, riid
), "unexpected riid %s\n", debugstr_guid(riid
));
1932 *ppv
= &MimeProtocol
;
1936 static const IClassFactoryVtbl MimeFilterCFVtbl
= {
1937 ClassFactory_QueryInterface
,
1938 ClassFactory_AddRef
,
1939 ClassFactory_Release
,
1940 MimeFilter_CreateInstance
,
1941 ClassFactory_LockServer
1944 static IClassFactory mimefilter_cf
= { &MimeFilterCFVtbl
};
1946 #define TEST_BINDING 0x01
1947 #define TEST_FILTER 0x02
1948 #define TEST_FIRST_HTTP 0x04
1949 #define TEST_DIRECT_READ 0x08
1950 #define TEST_POST 0x10
1951 #define TEST_EMULATEPROT 0x20
1952 #define TEST_SHORT_READ 0x40
1954 static void init_test(int prot
, DWORD flags
)
1956 tested_protocol
= prot
;
1957 binding_test
= (flags
& TEST_BINDING
) != 0;
1958 first_data_notif
= TRUE
;
1961 async_read_pending
= TRUE
;
1962 mimefilter_test
= (flags
& TEST_FILTER
) != 0;
1964 ResetEvent(event_complete
);
1965 ResetEvent(event_complete2
);
1966 ResetEvent(event_continue
);
1967 ResetEvent(event_continue_done
);
1968 async_protocol
= binding_protocol
= filtered_protocol
= NULL
;
1969 filtered_sink
= NULL
;
1970 http_is_first
= (flags
& TEST_FIRST_HTTP
) != 0;
1971 first_data_notif
= TRUE
;
1973 direct_read
= (flags
& TEST_DIRECT_READ
) != 0;
1974 http_post_test
= (flags
& TEST_POST
) != 0;
1975 emulate_prot
= (flags
& TEST_EMULATEPROT
) != 0;
1976 wait_for_switch
= TRUE
;
1977 short_read
= (flags
& TEST_SHORT_READ
) != 0;
1980 static void test_priority(IInternetProtocol
*protocol
)
1982 IInternetPriority
*priority
;
1986 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IInternetPriority
,
1988 ok(hres
== S_OK
, "QueryInterface(IID_IInternetPriority) failed: %08x\n", hres
);
1992 hres
= IInternetPriority_GetPriority(priority
, &pr
);
1993 ok(hres
== S_OK
, "GetPriority failed: %08x\n", hres
);
1994 ok(pr
== 0, "pr=%d, expected 0\n", pr
);
1996 hres
= IInternetPriority_SetPriority(priority
, 1);
1997 ok(hres
== S_OK
, "SetPriority failed: %08x\n", hres
);
1999 hres
= IInternetPriority_GetPriority(priority
, &pr
);
2000 ok(hres
== S_OK
, "GetPriority failed: %08x\n", hres
);
2001 ok(pr
== 1, "pr=%d, expected 1\n", pr
);
2003 IInternetPriority_Release(priority
);
2006 static BOOL
file_protocol_start(IInternetProtocol
*protocol
, LPCWSTR url
, BOOL is_first
)
2010 SET_EXPECT(GetBindInfo
);
2011 if(!(bindf
& BINDF_FROMURLMON
))
2012 SET_EXPECT(ReportProgress_DIRECTBIND
);
2014 SET_EXPECT(ReportProgress_SENDINGREQUEST
);
2015 SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE
);
2016 if(bindf
& BINDF_FROMURLMON
)
2017 SET_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE
);
2019 SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
2021 SET_EXPECT(ReportData
);
2023 SET_EXPECT(ReportResult
);
2025 expect_hrResult
= S_OK
;
2027 hres
= IInternetProtocol_Start(protocol
, url
, &protocol_sink
, &bind_info
, 0, 0);
2028 if(hres
== INET_E_RESOURCE_NOT_FOUND
) {
2029 win_skip("Start failed\n");
2032 ok(hres
== S_OK
, "Start failed: %08x\n", hres
);
2034 CHECK_CALLED(GetBindInfo
);
2035 if(!(bindf
& BINDF_FROMURLMON
))
2036 CHECK_CALLED(ReportProgress_DIRECTBIND
);
2038 CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
2039 CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE
);
2040 if(bindf
& BINDF_FROMURLMON
)
2041 CHECK_CALLED(ReportProgress_VERIFIEDMIMETYPEAVAILABLE
);
2043 CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
2045 CHECK_CALLED(ReportData
);
2047 CHECK_CALLED(ReportResult
);
2052 static void test_file_protocol_url(LPCWSTR url
)
2054 IInternetProtocolInfo
*protocol_info
;
2056 IClassFactory
*factory
;
2059 hres
= CoGetClassObject(&CLSID_FileProtocol
, CLSCTX_INPROC_SERVER
, NULL
,
2060 &IID_IUnknown
, (void**)&unk
);
2061 ok(hres
== S_OK
, "CoGetClassObject failed: %08x\n", hres
);
2065 hres
= IUnknown_QueryInterface(unk
, &IID_IInternetProtocolInfo
, (void**)&protocol_info
);
2066 ok(hres
== E_NOINTERFACE
,
2067 "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres
);
2069 hres
= IUnknown_QueryInterface(unk
, &IID_IClassFactory
, (void**)&factory
);
2070 ok(hres
== S_OK
, "Could not get IClassFactory interface\n");
2071 if(SUCCEEDED(hres
)) {
2072 IInternetProtocol
*protocol
;
2075 hres
= IClassFactory_CreateInstance(factory
, NULL
, &IID_IInternetProtocol
, (void**)&protocol
);
2076 ok(hres
== S_OK
, "Could not get IInternetProtocol: %08x\n", hres
);
2078 if(SUCCEEDED(hres
)) {
2079 if(file_protocol_start(protocol
, url
, TRUE
)) {
2080 hres
= IInternetProtocol_Read(protocol
, buf
, 2, &cb
);
2081 ok(hres
== S_OK
, "Read failed: %08x\n", hres
);
2082 ok(cb
== 2, "cb=%u expected 2\n", cb
);
2083 hres
= IInternetProtocol_Read(protocol
, buf
, sizeof(buf
), &cb
);
2084 ok(hres
== S_FALSE
, "Read failed: %08x\n", hres
);
2085 hres
= IInternetProtocol_Read(protocol
, buf
, sizeof(buf
), &cb
);
2086 ok(hres
== S_FALSE
, "Read failed: %08x expected S_FALSE\n", hres
);
2087 ok(cb
== 0, "cb=%u expected 0\n", cb
);
2088 hres
= IInternetProtocol_UnlockRequest(protocol
);
2089 ok(hres
== S_OK
, "UnlockRequest failed: %08x\n", hres
);
2092 if(file_protocol_start(protocol
, url
, FALSE
)) {
2093 hres
= IInternetProtocol_Read(protocol
, buf
, 2, &cb
);
2094 ok(hres
== S_FALSE
, "Read failed: %08x\n", hres
);
2095 hres
= IInternetProtocol_LockRequest(protocol
, 0);
2096 ok(hres
== S_OK
, "LockRequest failed: %08x\n", hres
);
2097 hres
= IInternetProtocol_UnlockRequest(protocol
);
2098 ok(hres
== S_OK
, "UnlockRequest failed: %08x\n", hres
);
2101 IInternetProtocol_Release(protocol
);
2104 hres
= IClassFactory_CreateInstance(factory
, NULL
, &IID_IInternetProtocol
, (void**)&protocol
);
2105 ok(hres
== S_OK
, "Could not get IInternetProtocol: %08x\n", hres
);
2107 if(SUCCEEDED(hres
)) {
2108 if(file_protocol_start(protocol
, url
, TRUE
)) {
2109 hres
= IInternetProtocol_LockRequest(protocol
, 0);
2110 ok(hres
== S_OK
, "LockRequest failed: %08x\n", hres
);
2111 hres
= IInternetProtocol_Terminate(protocol
, 0);
2112 ok(hres
== S_OK
, "Terminate failed: %08x\n", hres
);
2113 hres
= IInternetProtocol_Read(protocol
, buf
, 2, &cb
);
2114 ok(hres
== S_OK
, "Read failed: %08x\n\n", hres
);
2115 hres
= IInternetProtocol_UnlockRequest(protocol
);
2116 ok(hres
== S_OK
, "UnlockRequest failed: %08x\n", hres
);
2117 hres
= IInternetProtocol_Read(protocol
, buf
, 2, &cb
);
2118 ok(hres
== S_OK
, "Read failed: %08x\n", hres
);
2119 hres
= IInternetProtocol_Terminate(protocol
, 0);
2120 ok(hres
== S_OK
, "Terminate failed: %08x\n", hres
);
2123 IInternetProtocol_Release(protocol
);
2126 hres
= IClassFactory_CreateInstance(factory
, NULL
, &IID_IInternetProtocol
, (void**)&protocol
);
2127 ok(hres
== S_OK
, "Could not get IInternetProtocol: %08x\n", hres
);
2129 if(SUCCEEDED(hres
)) {
2130 if(file_protocol_start(protocol
, url
, TRUE
)) {
2131 hres
= IInternetProtocol_Terminate(protocol
, 0);
2132 ok(hres
== S_OK
, "Terminate failed: %08x\n", hres
);
2133 hres
= IInternetProtocol_Read(protocol
, buf
, 2, &cb
);
2134 ok(hres
== S_OK
, "Read failed: %08x\n", hres
);
2135 ok(cb
== 2, "cb=%u expected 2\n", cb
);
2138 IInternetProtocol_Release(protocol
);
2141 IClassFactory_Release(factory
);
2144 IUnknown_Release(unk
);
2147 static void test_file_protocol_fail(void)
2149 IInternetProtocol
*protocol
;
2152 static const WCHAR index_url2
[] =
2153 {'f','i','l','e',':','/','/','i','n','d','e','x','.','h','t','m','l',0};
2155 hres
= CoCreateInstance(&CLSID_FileProtocol
, NULL
, CLSCTX_INPROC_SERVER
|CLSCTX_INPROC_HANDLER
,
2156 &IID_IInternetProtocol
, (void**)&protocol
);
2157 ok(hres
== S_OK
, "CoCreateInstance failed: %08x\n", hres
);
2161 SET_EXPECT(GetBindInfo
);
2162 expect_hrResult
= MK_E_SYNTAX
;
2163 hres
= IInternetProtocol_Start(protocol
, wszIndexHtml
, &protocol_sink
, &bind_info
, 0, 0);
2164 ok(hres
== MK_E_SYNTAX
||
2165 hres
== E_INVALIDARG
,
2166 "Start failed: %08x, expected MK_E_SYNTAX or E_INVALIDARG\n", hres
);
2167 CLEAR_CALLED(GetBindInfo
); /* GetBindInfo not called in IE7 */
2169 SET_EXPECT(GetBindInfo
);
2170 if(!(bindf
& BINDF_FROMURLMON
))
2171 SET_EXPECT(ReportProgress_DIRECTBIND
);
2172 SET_EXPECT(ReportProgress_SENDINGREQUEST
);
2173 SET_EXPECT(ReportResult
);
2174 expect_hrResult
= INET_E_RESOURCE_NOT_FOUND
;
2175 hres
= IInternetProtocol_Start(protocol
, index_url
, &protocol_sink
, &bind_info
, 0, 0);
2176 ok(hres
== INET_E_RESOURCE_NOT_FOUND
,
2177 "Start failed: %08x expected INET_E_RESOURCE_NOT_FOUND\n", hres
);
2178 CHECK_CALLED(GetBindInfo
);
2179 if(!(bindf
& BINDF_FROMURLMON
))
2180 CHECK_CALLED(ReportProgress_DIRECTBIND
);
2181 CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
2182 CHECK_CALLED(ReportResult
);
2184 IInternetProtocol_Release(protocol
);
2186 hres
= CoCreateInstance(&CLSID_FileProtocol
, NULL
, CLSCTX_INPROC_SERVER
|CLSCTX_INPROC_HANDLER
,
2187 &IID_IInternetProtocol
, (void**)&protocol
);
2188 ok(hres
== S_OK
, "CoCreateInstance failed: %08x\n", hres
);
2192 SET_EXPECT(GetBindInfo
);
2193 if(!(bindf
& BINDF_FROMURLMON
))
2194 SET_EXPECT(ReportProgress_DIRECTBIND
);
2195 SET_EXPECT(ReportProgress_SENDINGREQUEST
);
2196 SET_EXPECT(ReportResult
);
2197 expect_hrResult
= INET_E_RESOURCE_NOT_FOUND
;
2199 hres
= IInternetProtocol_Start(protocol
, index_url2
, &protocol_sink
, &bind_info
, 0, 0);
2200 ok(hres
== INET_E_RESOURCE_NOT_FOUND
,
2201 "Start failed: %08x, expected INET_E_RESOURCE_NOT_FOUND\n", hres
);
2202 CHECK_CALLED(GetBindInfo
);
2203 if(!(bindf
& BINDF_FROMURLMON
))
2204 CHECK_CALLED(ReportProgress_DIRECTBIND
);
2205 CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
2206 CHECK_CALLED(ReportResult
);
2208 SET_EXPECT(GetBindInfo
);
2209 hres
= IInternetProtocol_Start(protocol
, NULL
, &protocol_sink
, &bind_info
, 0, 0);
2210 ok(hres
== E_INVALIDARG
, "Start failed: %08x, expected E_INVALIDARG\n", hres
);
2211 CLEAR_CALLED(GetBindInfo
); /* GetBindInfo not called in IE7 */
2213 SET_EXPECT(GetBindInfo
);
2214 hres
= IInternetProtocol_Start(protocol
, emptyW
, &protocol_sink
, &bind_info
, 0, 0);
2215 ok(hres
== E_INVALIDARG
, "Start failed: %08x, expected E_INVALIDARG\n", hres
);
2216 CLEAR_CALLED(GetBindInfo
); /* GetBindInfo not called in IE7 */
2218 IInternetProtocol_Release(protocol
);
2221 static void test_file_protocol(void) {
2222 WCHAR buf
[INTERNET_MAX_URL_LENGTH
], file_name_buf
[MAX_PATH
];
2227 static const WCHAR wszFile
[] = {'f','i','l','e',':',0};
2228 static const WCHAR wszFile2
[] = {'f','i','l','e',':','/','/',0};
2229 static const WCHAR wszFile3
[] = {'f','i','l','e',':','/','/','/',0};
2230 static const char html_doc
[] = "<HTML></HTML>";
2232 trace("Testing file protocol...\n");
2233 init_test(FILE_TEST
, 0);
2235 SetLastError(0xdeadbeef);
2236 file
= CreateFileW(wszIndexHtml
, GENERIC_WRITE
, 0, NULL
, CREATE_ALWAYS
,
2237 FILE_ATTRIBUTE_NORMAL
, NULL
);
2238 if(!file
&& GetLastError() == ERROR_CALL_NOT_IMPLEMENTED
)
2240 win_skip("Detected Win9x or WinMe\n");
2243 ok(file
!= INVALID_HANDLE_VALUE
, "CreateFile failed\n");
2244 if(file
== INVALID_HANDLE_VALUE
)
2246 WriteFile(file
, html_doc
, sizeof(html_doc
)-1, &size
, NULL
);
2249 file_name
= wszIndexHtml
;
2251 test_file_protocol_url(index_url
);
2252 bindf
= BINDF_FROMURLMON
;
2253 test_file_protocol_url(index_url
);
2254 bindf
= BINDF_FROMURLMON
| BINDF_NEEDFILE
;
2255 test_file_protocol_url(index_url
);
2257 memcpy(buf
, wszFile
, sizeof(wszFile
));
2258 len
= sizeof(wszFile
)/sizeof(WCHAR
)-1;
2259 len
+= GetCurrentDirectoryW(sizeof(buf
)/sizeof(WCHAR
)-len
, buf
+len
);
2261 memcpy(buf
+len
, wszIndexHtml
, sizeof(wszIndexHtml
));
2263 file_name
= buf
+ sizeof(wszFile
)/sizeof(WCHAR
)-1;
2265 test_file_protocol_url(buf
);
2266 bindf
= BINDF_FROMURLMON
;
2267 test_file_protocol_url(buf
);
2269 memcpy(buf
, wszFile2
, sizeof(wszFile2
));
2270 len
= GetCurrentDirectoryW(sizeof(file_name_buf
)/sizeof(WCHAR
), file_name_buf
);
2271 file_name_buf
[len
++] = '\\';
2272 memcpy(file_name_buf
+len
, wszIndexHtml
, sizeof(wszIndexHtml
));
2273 lstrcpyW(buf
+sizeof(wszFile2
)/sizeof(WCHAR
)-1, file_name_buf
);
2274 file_name
= file_name_buf
;
2276 test_file_protocol_url(buf
);
2277 bindf
= BINDF_FROMURLMON
;
2278 test_file_protocol_url(buf
);
2280 buf
[sizeof(wszFile2
)/sizeof(WCHAR
)] = '|';
2281 test_file_protocol_url(buf
);
2283 memcpy(buf
, wszFile3
, sizeof(wszFile3
));
2284 len
= sizeof(wszFile3
)/sizeof(WCHAR
)-1;
2285 len
+= GetCurrentDirectoryW(sizeof(buf
)/sizeof(WCHAR
)-len
, buf
+len
);
2287 memcpy(buf
+len
, wszIndexHtml
, sizeof(wszIndexHtml
));
2289 file_name
= buf
+ sizeof(wszFile3
)/sizeof(WCHAR
)-1;
2291 test_file_protocol_url(buf
);
2292 bindf
= BINDF_FROMURLMON
;
2293 test_file_protocol_url(buf
);
2295 DeleteFileW(wszIndexHtml
);
2298 test_file_protocol_fail();
2299 bindf
= BINDF_FROMURLMON
;
2300 test_file_protocol_fail();
2303 static BOOL
http_protocol_start(LPCWSTR url
)
2305 static BOOL got_user_agent
= FALSE
;
2308 SET_EXPECT(GetBindInfo
);
2309 if (!(bindf
& BINDF_FROMURLMON
))
2310 SET_EXPECT(ReportProgress_DIRECTBIND
);
2312 SET_EXPECT(GetBindString_USER_AGENT
);
2313 SET_EXPECT(GetBindString_ACCEPT_MIMES
);
2314 SET_EXPECT(QueryService_HttpNegotiate
);
2315 SET_EXPECT(BeginningTransaction
);
2316 SET_EXPECT(GetRootSecurityId
);
2318 SET_EXPECT(GetBindString_POST_COOKIE
);
2320 hres
= IInternetProtocol_Start(async_protocol
, url
, &protocol_sink
, &bind_info
, 0, 0);
2321 ok(hres
== S_OK
, "Start failed: %08x\n", hres
);
2325 CHECK_CALLED(GetBindInfo
);
2326 if (!(bindf
& BINDF_FROMURLMON
))
2327 CHECK_CALLED(ReportProgress_DIRECTBIND
);
2328 if (!got_user_agent
)
2330 CHECK_CALLED(GetBindString_USER_AGENT
);
2331 got_user_agent
= TRUE
;
2333 CHECK_CALLED(GetBindString_ACCEPT_MIMES
);
2334 CHECK_CALLED(QueryService_HttpNegotiate
);
2335 CHECK_CALLED(BeginningTransaction
);
2336 /* GetRootSecurityId called on WinXP but not on Win98 */
2337 CLEAR_CALLED(GetRootSecurityId
);
2339 CHECK_CALLED(GetBindString_POST_COOKIE
);
2344 static void test_protocol_terminate(IInternetProtocol
*protocol
)
2350 hres
= IInternetProtocol_LockRequest(protocol
, 0);
2351 ok(hres
== S_OK
, "LockRequest failed: %08x\n", hres
);
2353 hres
= IInternetProtocol_Read(protocol
, buf
, 1, &cb
);
2354 ok(hres
== S_FALSE
, "Read failed: %08x\n", hres
);
2356 hres
= IInternetProtocol_Terminate(protocol
, 0);
2357 ok(hres
== S_OK
, "Terminate failed: %08x\n", hres
);
2359 /* This wait is to give the internet handles being freed in Terminate
2360 * enough time to actually terminate in all cases. Internet handles
2361 * terminate asynchronously and native reuses the main InternetOpen
2362 * handle. The only case in which this seems to be necessary is on
2363 * wine with native wininet and urlmon, resulting in the next time
2364 * test_http_protocol_url being called the first data notification actually
2365 * being an extra last data notification from the previous connection
2366 * about once out of every ten times. */
2369 hres
= IInternetProtocol_UnlockRequest(protocol
);
2370 ok(hres
== S_OK
, "UnlockRequest failed: %08x\n", hres
);
2373 static void test_http_info(IInternetProtocol
*protocol
)
2375 IWinInetHttpInfo
*info
;
2378 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IWinInetHttpInfo
, (void**)&info
);
2379 ok(hres
== S_OK
, "Could not get IWinInterHttpInfo iface: %08x\n", hres
);
2383 IWinInetHttpInfo_Release(info
);
2386 /* is_first refers to whether this is the first call to this function
2388 static void test_http_protocol_url(LPCWSTR url
, int prot
, DWORD flags
)
2390 IInternetProtocolInfo
*protocol_info
;
2391 IClassFactory
*factory
;
2395 init_test(prot
, flags
);
2398 hres
= CoGetClassObject(prot
== HTTPS_TEST
? &CLSID_HttpSProtocol
: &CLSID_HttpProtocol
,
2399 CLSCTX_INPROC_SERVER
, NULL
, &IID_IUnknown
, (void**)&unk
);
2400 ok(hres
== S_OK
, "CoGetClassObject failed: %08x\n", hres
);
2404 hres
= IUnknown_QueryInterface(unk
, &IID_IInternetProtocolInfo
, (void**)&protocol_info
);
2405 ok(hres
== E_NOINTERFACE
,
2406 "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n",
2409 hres
= IUnknown_QueryInterface(unk
, &IID_IClassFactory
, (void**)&factory
);
2410 ok(hres
== S_OK
, "Could not get IClassFactory interface\n");
2411 IUnknown_Release(unk
);
2415 hres
= IClassFactory_CreateInstance(factory
, NULL
, &IID_IInternetProtocol
,
2416 (void**)&async_protocol
);
2417 ok(hres
== S_OK
, "Could not get IInternetProtocol: %08x\n", hres
);
2418 if(SUCCEEDED(hres
)) {
2423 test_priority(async_protocol
);
2424 test_http_info(async_protocol
);
2426 SET_EXPECT(ReportProgress_FINDINGRESOURCE
);
2427 SET_EXPECT(ReportProgress_CONNECTING
);
2428 SET_EXPECT(ReportProgress_SENDINGREQUEST
);
2429 SET_EXPECT(ReportProgress_PROXYDETECTING
);
2430 if(prot
== HTTP_TEST
)
2431 SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE
);
2433 SET_EXPECT(QueryService_HttpSecurity
);
2434 if(!(bindf
& BINDF_FROMURLMON
)) {
2435 SET_EXPECT(OnResponse
);
2436 SET_EXPECT(ReportProgress_RAWMIMETYPE
);
2437 SET_EXPECT(ReportData
);
2442 if(!http_protocol_start(url
))
2446 SET_EXPECT(ReportResult
);
2447 expect_hrResult
= S_OK
;
2450 while(wait_for_switch
) {
2452 WaitForSingleObject(event_continue
, INFINITE
);
2453 CHECK_CALLED(Switch
);
2454 call_continue(&continue_protdata
);
2455 SetEvent(event_continue_done
);
2458 hres
= IInternetProtocol_Read(async_protocol
, buf
, 1, &cb
);
2459 ok((hres
== E_PENDING
&& cb
==0) ||
2460 (hres
== S_OK
&& cb
==1), "Read failed: %08x (%d bytes)\n", hres
, cb
);
2462 WaitForSingleObject(event_complete
, INFINITE
);
2463 if(bindf
& BINDF_FROMURLMON
)
2464 CHECK_CALLED(Switch
);
2466 CHECK_CALLED(ReportData
);
2467 if(prot
== HTTPS_TEST
)
2468 CLEAR_CALLED(QueryService_HttpSecurity
);
2471 if(bindf
& BINDF_FROMURLMON
)
2474 SET_EXPECT(ReportData
);
2475 hres
= IInternetProtocol_Read(async_protocol
, buf
, sizeof(buf
), &cb
);
2476 if(hres
== E_PENDING
) {
2477 hres
= IInternetProtocol_Read(async_protocol
, buf
, 1, &cb
);
2478 ok((hres
== E_PENDING
&& cb
==0) ||
2479 (hres
== S_OK
&& cb
==1), "Read failed: %08x (%d bytes)\n", hres
, cb
);
2480 WaitForSingleObject(event_complete
, INFINITE
);
2481 if(bindf
& BINDF_FROMURLMON
)
2482 CHECK_CALLED(Switch
);
2484 CHECK_CALLED(ReportData
);
2486 if(bindf
& BINDF_FROMURLMON
)
2487 CHECK_NOT_CALLED(Switch
);
2489 CHECK_NOT_CALLED(ReportData
);
2493 ok(hres
== S_FALSE
, "Read failed: %08x\n", hres
);
2494 CHECK_CALLED(ReportResult
);
2496 if(prot
== HTTPS_TEST
)
2497 CLEAR_CALLED(ReportProgress_SENDINGREQUEST
);
2499 test_protocol_terminate(async_protocol
);
2500 ref
= IInternetProtocol_Release(async_protocol
);
2501 ok(!ref
, "ref=%x\n", hres
);
2504 IClassFactory_Release(factory
);
2507 static void test_http_protocol(void)
2509 static const WCHAR winehq_url
[] =
2510 {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q','.',
2511 'o','r','g','/','s','i','t','e','/','a','b','o','u','t',0};
2512 static const WCHAR posttest_url
[] =
2513 {'h','t','t','p',':','/','/','c','r','o','s','s','o','v','e','r','.',
2514 'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m','/',
2515 'p','o','s','t','t','e','s','t','.','p','h','p',0};
2517 trace("Testing http protocol (not from urlmon)...\n");
2518 bindf
= BINDF_ASYNCHRONOUS
| BINDF_ASYNCSTORAGE
| BINDF_PULLDATA
;
2519 test_http_protocol_url(winehq_url
, HTTP_TEST
, TEST_FIRST_HTTP
);
2521 trace("Testing http protocol (from urlmon)...\n");
2522 bindf
= BINDF_ASYNCHRONOUS
| BINDF_ASYNCSTORAGE
| BINDF_PULLDATA
| BINDF_FROMURLMON
;
2523 test_http_protocol_url(winehq_url
, HTTP_TEST
, 0);
2525 trace("Testing http protocol (to file)...\n");
2526 bindf
= BINDF_ASYNCHRONOUS
| BINDF_ASYNCSTORAGE
| BINDF_PULLDATA
| BINDF_FROMURLMON
| BINDF_NEEDFILE
;
2527 test_http_protocol_url(winehq_url
, HTTP_TEST
, 0);
2529 trace("Testing http protocol (post data)...\n");
2530 /* Without this flag we get a ReportProgress_CACHEFILENAMEAVAILABLE
2531 * notification with BINDVERB_POST */
2532 bindf
= BINDF_ASYNCHRONOUS
| BINDF_ASYNCSTORAGE
| BINDF_PULLDATA
| BINDF_FROMURLMON
| BINDF_NOWRITECACHE
;
2533 test_http_protocol_url(posttest_url
, HTTP_TEST
, TEST_FIRST_HTTP
|TEST_POST
);
2535 trace("Testing http protocol (direct read)...\n");
2536 bindf
= BINDF_ASYNCHRONOUS
| BINDF_ASYNCSTORAGE
| BINDF_PULLDATA
| BINDF_FROMURLMON
;
2537 test_http_protocol_url(winehq_url
, HTTP_TEST
, TEST_DIRECT_READ
);
2540 static void test_https_protocol(void)
2542 static const WCHAR codeweavers_url
[] =
2543 {'h','t','t','p','s',':','/','/','w','w','w','.','c','o','d','e','w','e','a','v','e','r','s',
2544 '.','c','o','m','/','t','e','s','t','.','h','t','m','l',0};
2546 trace("Testing https protocol (from urlmon)...\n");
2547 bindf
= BINDF_ASYNCHRONOUS
| BINDF_ASYNCSTORAGE
| BINDF_PULLDATA
| BINDF_FROMURLMON
| BINDF_NOWRITECACHE
;
2548 test_http_protocol_url(codeweavers_url
, HTTPS_TEST
, TEST_FIRST_HTTP
);
2552 static void test_ftp_protocol(void)
2554 IInternetProtocolInfo
*protocol_info
;
2555 IClassFactory
*factory
;
2562 static const WCHAR ftp_urlW
[] = {'f','t','p',':','/','/','f','t','p','.','w','i','n','e','h','q','.','o','r','g',
2563 '/','p','u','b','/','o','t','h','e','r','/',
2564 'w','i','n','e','l','o','g','o','.','x','c','f','.','t','a','r','.','b','z','2',0};
2566 trace("Testing ftp protocol...\n");
2568 bindf
= BINDF_ASYNCHRONOUS
| BINDF_ASYNCSTORAGE
| BINDF_PULLDATA
| BINDF_FROMURLMON
| BINDF_NOWRITECACHE
;
2570 tested_protocol
= FTP_TEST
;
2571 first_data_notif
= TRUE
;
2572 expect_hrResult
= E_PENDING
;
2574 hres
= CoGetClassObject(&CLSID_FtpProtocol
, CLSCTX_INPROC_SERVER
, NULL
, &IID_IUnknown
, (void**)&unk
);
2575 ok(hres
== S_OK
, "CoGetClassObject failed: %08x\n", hres
);
2579 hres
= IUnknown_QueryInterface(unk
, &IID_IInternetProtocolInfo
, (void**)&protocol_info
);
2580 ok(hres
== E_NOINTERFACE
, "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres
);
2582 hres
= IUnknown_QueryInterface(unk
, &IID_IClassFactory
, (void**)&factory
);
2583 ok(hres
== S_OK
, "Could not get IClassFactory interface\n");
2584 IUnknown_Release(unk
);
2588 hres
= IClassFactory_CreateInstance(factory
, NULL
, &IID_IInternetProtocol
,
2589 (void**)&async_protocol
);
2590 IClassFactory_Release(factory
);
2591 ok(hres
== S_OK
, "Could not get IInternetProtocol: %08x\n", hres
);
2593 test_priority(async_protocol
);
2594 test_http_info(async_protocol
);
2596 SET_EXPECT(GetBindInfo
);
2597 SET_EXPECT(ReportProgress_FINDINGRESOURCE
);
2598 SET_EXPECT(ReportProgress_CONNECTING
);
2599 SET_EXPECT(ReportProgress_SENDINGREQUEST
);
2602 hres
= IInternetProtocol_Start(async_protocol
, ftp_urlW
, &protocol_sink
, &bind_info
, 0, 0);
2603 ok(hres
== S_OK
, "Start failed: %08x\n", hres
);
2604 CHECK_CALLED(GetBindInfo
);
2606 SET_EXPECT(ReportResult
);
2608 hres
= IInternetProtocol_Read(async_protocol
, buf
, 1, &cb
);
2609 ok((hres
== E_PENDING
&& cb
==0) ||
2610 (hres
== S_OK
&& cb
==1), "Read failed: %08x (%d bytes)\n", hres
, cb
);
2612 WaitForSingleObject(event_complete
, INFINITE
);
2616 hres
= IInternetProtocol_Read(async_protocol
, buf
, sizeof(buf
), &cb
);
2617 if(hres
== E_PENDING
)
2618 WaitForSingleObject(event_complete
, INFINITE
);
2623 ok(hres
== S_FALSE
, "Read failed: %08x\n", hres
);
2624 CHECK_CALLED(ReportResult
);
2625 CHECK_CALLED(Switch
);
2627 test_protocol_terminate(async_protocol
);
2629 ref
= IInternetProtocol_Release(async_protocol
);
2630 ok(!ref
, "ref=%d\n", ref
);
2633 static void test_gopher_protocol(void)
2635 IInternetProtocolInfo
*protocol_info
;
2636 IClassFactory
*factory
;
2640 trace("Testing gopher protocol...\n");
2642 hres
= CoGetClassObject(&CLSID_GopherProtocol
, CLSCTX_INPROC_SERVER
, NULL
, &IID_IUnknown
, (void**)&unk
);
2644 hres
== REGDB_E_CLASSNOTREG
, /* Gopher protocol has been removed as of Vista */
2645 "CoGetClassObject failed: %08x\n", hres
);
2649 hres
= IUnknown_QueryInterface(unk
, &IID_IInternetProtocolInfo
, (void**)&protocol_info
);
2650 ok(hres
== E_NOINTERFACE
, "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres
);
2652 hres
= IUnknown_QueryInterface(unk
, &IID_IClassFactory
, (void**)&factory
);
2653 ok(hres
== S_OK
, "Could not get IClassFactory interface\n");
2654 IUnknown_Release(unk
);
2658 hres
= IClassFactory_CreateInstance(factory
, NULL
, &IID_IInternetProtocol
,
2659 (void**)&async_protocol
);
2660 IClassFactory_Release(factory
);
2661 ok(hres
== S_OK
, "Could not get IInternetProtocol: %08x\n", hres
);
2663 test_priority(async_protocol
);
2665 IInternetProtocol_Release(async_protocol
);
2668 static void test_mk_protocol(void)
2670 IInternetProtocolInfo
*protocol_info
;
2671 IInternetProtocol
*protocol
;
2672 IClassFactory
*factory
;
2676 static const WCHAR wrong_url1
[] = {'t','e','s','t',':','@','M','S','I','T','S','t','o','r','e',
2677 ':',':','/','t','e','s','t','.','h','t','m','l',0};
2678 static const WCHAR wrong_url2
[] = {'m','k',':','/','t','e','s','t','.','h','t','m','l',0};
2680 trace("Testing mk protocol...\n");
2681 init_test(MK_TEST
, 0);
2683 hres
= CoGetClassObject(&CLSID_MkProtocol
, CLSCTX_INPROC_SERVER
, NULL
,
2684 &IID_IUnknown
, (void**)&unk
);
2685 ok(hres
== S_OK
, "CoGetClassObject failed: %08x\n", hres
);
2687 hres
= IUnknown_QueryInterface(unk
, &IID_IInternetProtocolInfo
, (void**)&protocol_info
);
2688 ok(hres
== E_NOINTERFACE
,
2689 "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n",
2692 hres
= IUnknown_QueryInterface(unk
, &IID_IClassFactory
, (void**)&factory
);
2693 ok(hres
== S_OK
, "Could not get IClassFactory interface\n");
2694 IUnknown_Release(unk
);
2698 hres
= IClassFactory_CreateInstance(factory
, NULL
, &IID_IInternetProtocol
,
2700 IClassFactory_Release(factory
);
2701 ok(hres
== S_OK
, "Could not get IInternetProtocol: %08x\n", hres
);
2703 SET_EXPECT(GetBindInfo
);
2704 hres
= IInternetProtocol_Start(protocol
, wrong_url1
, &protocol_sink
, &bind_info
, 0, 0);
2705 ok(hres
== MK_E_SYNTAX
|| hres
== INET_E_INVALID_URL
,
2706 "Start failed: %08x, expected MK_E_SYNTAX or INET_E_INVALID_URL\n", hres
);
2707 CLEAR_CALLED(GetBindInfo
);
2709 SET_EXPECT(GetBindInfo
);
2710 SET_EXPECT(ReportProgress_DIRECTBIND
);
2711 SET_EXPECT(ReportProgress_SENDINGREQUEST
);
2712 SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE
);
2713 SET_EXPECT(ReportResult
);
2714 expect_hrResult
= INET_E_RESOURCE_NOT_FOUND
;
2716 hres
= IInternetProtocol_Start(protocol
, wrong_url2
, &protocol_sink
, &bind_info
, 0, 0);
2717 ok(hres
== INET_E_RESOURCE_NOT_FOUND
||
2718 hres
== INET_E_INVALID_URL
, /* win2k3 */
2719 "Start failed: %08x, expected INET_E_RESOURCE_NOT_FOUND or INET_E_INVALID_URL\n", hres
);
2721 if (hres
== INET_E_RESOURCE_NOT_FOUND
) {
2722 CHECK_CALLED(GetBindInfo
);
2723 CLEAR_CALLED(ReportProgress_DIRECTBIND
);
2724 CHECK_CALLED(ReportProgress_SENDINGREQUEST
);
2725 CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
2726 CHECK_CALLED(ReportResult
);
2728 CLEAR_CALLED(GetBindInfo
);
2729 CLEAR_CALLED(ReportProgress_DIRECTBIND
);
2730 CLEAR_CALLED(ReportProgress_SENDINGREQUEST
);
2731 CLEAR_CALLED(ReportProgress_MIMETYPEAVAILABLE
);
2732 CLEAR_CALLED(ReportResult
);
2735 IInternetProtocol_Release(protocol
);
2738 static void test_CreateBinding(void)
2740 IInternetProtocol
*protocol
;
2741 IInternetPriority
*priority
;
2742 IInternetSession
*session
;
2743 IWinInetHttpInfo
*http_info
;
2744 IWinInetInfo
*inet_info
;
2750 static const WCHAR test_url
[] =
2751 {'t','e','s','t',':','/','/','f','i','l','e','.','h','t','m','l',0};
2752 static const WCHAR wsz_test
[] = {'t','e','s','t',0};
2754 trace("Testing CreateBinding...\n");
2755 init_test(BIND_TEST
, TEST_BINDING
);
2757 hres
= CoInternetGetSession(0, &session
, 0);
2758 ok(hres
== S_OK
, "CoInternetGetSession failed: %08x\n", hres
);
2760 hres
= IInternetSession_RegisterNameSpace(session
, &ClassFactory
, &IID_NULL
, wsz_test
, 0, NULL
, 0);
2761 ok(hres
== S_OK
, "RegisterNameSpace failed: %08x\n", hres
);
2763 hres
= IInternetSession_CreateBinding(session
, NULL
, test_url
, NULL
, NULL
, &protocol
, 0);
2764 binding_protocol
= protocol
;
2765 ok(hres
== S_OK
, "CreateBinding failed: %08x\n", hres
);
2766 ok(protocol
!= NULL
, "protocol == NULL\n");
2768 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IInternetBindInfo
, (void**)&prot_bind_info
);
2769 ok(hres
== S_OK
, "QueryInterface(IID_IInternetBindInfo) failed: %08x\n", hres
);
2771 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IInternetProtocolSink
, (void**)&binding_sink
);
2772 ok(hres
== S_OK
, "Could not get IInternetProtocolSink: %08x\n", hres
);
2774 hres
= IInternetProtocol_Start(protocol
, test_url
, NULL
, &bind_info
, 0, 0);
2775 ok(hres
== E_INVALIDARG
, "Start failed: %08x, expected E_INVALIDARG\n", hres
);
2776 hres
= IInternetProtocol_Start(protocol
, test_url
, &protocol_sink
, NULL
, 0, 0);
2777 ok(hres
== E_INVALIDARG
, "Start failed: %08x, expected E_INVALIDARG\n", hres
);
2778 hres
= IInternetProtocol_Start(protocol
, NULL
, &protocol_sink
, &bind_info
, 0, 0);
2779 ok(hres
== E_INVALIDARG
, "Start failed: %08x, expected E_INVALIDARG\n", hres
);
2781 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IInternetPriority
, (void**)&priority
);
2782 ok(hres
== S_OK
, "QueryInterface(IID_IInternetPriority) failed: %08x\n", hres
);
2785 hres
= IInternetPriority_GetPriority(priority
, &p
);
2786 ok(hres
== S_OK
, "GetPriority failed: %08x\n", hres
);
2787 ok(!p
, "p=%d\n", p
);
2790 hres
= IInternetPriority_SetPriority(priority
, 100);
2791 ok(hres
== S_OK
, "SetPriority failed: %08x\n", hres
);
2794 hres
= IInternetPriority_GetPriority(priority
, &p
);
2795 ok(hres
== S_OK
, "GetPriority failed: %08x\n", hres
);
2796 ok(p
== 100, "p=%d\n", p
);
2798 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IWinInetInfo
, (void**)&inet_info
);
2799 ok(hres
== E_NOINTERFACE
, "Could not get IWinInetInfo protocol: %08x\n", hres
);
2801 SET_EXPECT(QueryService_InternetProtocol
);
2802 SET_EXPECT(CreateInstance
);
2803 SET_EXPECT(ReportProgress_PROTOCOLCLASSID
);
2804 SET_EXPECT(SetPriority
);
2807 expect_hrResult
= S_OK
;
2808 hres
= IInternetProtocol_Start(protocol
, test_url
, &protocol_sink
, &bind_info
, 0, 0);
2809 ok(hres
== S_OK
, "Start failed: %08x\n", hres
);
2811 CHECK_CALLED(QueryService_InternetProtocol
);
2812 CHECK_CALLED(CreateInstance
);
2813 CHECK_CALLED(ReportProgress_PROTOCOLCLASSID
);
2814 CHECK_CALLED(SetPriority
);
2815 CHECK_CALLED(Start
);
2817 SET_EXPECT(QueryInterface_IWinInetInfo
);
2818 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IWinInetInfo
, (void**)&inet_info
);
2819 ok(hres
== E_NOINTERFACE
, "Could not get IWinInetInfo protocol: %08x\n", hres
);
2820 CHECK_CALLED(QueryInterface_IWinInetInfo
);
2822 SET_EXPECT(QueryInterface_IWinInetInfo
);
2823 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IWinInetInfo
, (void**)&inet_info
);
2824 ok(hres
== E_NOINTERFACE
, "Could not get IWinInetInfo protocol: %08x\n", hres
);
2825 CHECK_CALLED(QueryInterface_IWinInetInfo
);
2827 SET_EXPECT(QueryInterface_IWinInetHttpInfo
);
2828 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IWinInetHttpInfo
, (void**)&http_info
);
2829 ok(hres
== E_NOINTERFACE
, "Could not get IWinInetInfo protocol: %08x\n", hres
);
2830 CHECK_CALLED(QueryInterface_IWinInetHttpInfo
);
2834 hres
= IInternetProtocol_Read(protocol
, expect_pv
= buf
, sizeof(buf
), &read
);
2835 ok(hres
== S_OK
, "Read failed: %08x\n", hres
);
2836 ok(read
== 100, "read = %d\n", read
);
2841 hres
= IInternetProtocol_Read(protocol
, expect_pv
= buf
, sizeof(buf
), &read
);
2842 ok(hres
== S_FALSE
, "Read failed: %08x\n", hres
);
2843 ok(!read
, "read = %d\n", read
);
2847 hres
= IInternetPriority_GetPriority(priority
, &p
);
2848 ok(hres
== S_OK
, "GetPriority failed: %08x\n", hres
);
2849 ok(p
== 100, "p=%d\n", p
);
2851 hres
= IInternetPriority_SetPriority(priority
, 101);
2852 ok(hres
== S_OK
, "SetPriority failed: %08x\n", hres
);
2854 SET_EXPECT(Terminate
);
2855 hres
= IInternetProtocol_Terminate(protocol
, 0xdeadbeef);
2856 ok(hres
== S_OK
, "Terminate failed: %08x\n", hres
);
2857 CHECK_CALLED(Terminate
);
2859 SET_EXPECT(Continue
);
2860 hres
= IInternetProtocolSink_Switch(binding_sink
, &protocoldata
);
2861 ok(hres
== S_OK
, "Switch failed: %08x\n", hres
);
2862 CHECK_CALLED(Continue
);
2864 hres
= IInternetProtocolSink_ReportProgress(binding_sink
,
2865 BINDSTATUS_CACHEFILENAMEAVAILABLE
, expect_wsz
= emptyW
);
2866 ok(hres
== S_OK
, "ReportProgress(BINDSTATUS_CACHEFILENAMEAVAILABLE) failed: %08x\n", hres
);
2868 hres
= IInternetProtocolSink_ReportResult(binding_sink
, S_OK
, ERROR_SUCCESS
, NULL
);
2869 ok(hres
== E_FAIL
, "ReportResult failed: %08x, expected E_FAIL\n", hres
);
2871 hres
= IInternetProtocolSink_ReportData(binding_sink
, 0, 0, 0);
2872 ok(hres
== S_OK
, "ReportData failed: %08x\n", hres
);
2874 IInternetProtocolSink_Release(binding_sink
);
2875 IInternetPriority_Release(priority
);
2876 IInternetBindInfo_Release(prot_bind_info
);
2877 IInternetProtocol_Release(protocol
);
2879 hres
= IInternetSession_UnregisterNameSpace(session
, &ClassFactory
, wsz_test
);
2880 ok(hres
== S_OK
, "UnregisterNameSpace failed: %08x\n", hres
);
2882 IInternetSession_Release(session
);
2885 static void test_binding(int prot
, DWORD grf_pi
, DWORD test_flags
)
2887 IInternetProtocol
*protocol
;
2888 IInternetSession
*session
;
2894 init_test(prot
, test_flags
|TEST_BINDING
);
2896 hres
= CoInternetGetSession(0, &session
, 0);
2897 ok(hres
== S_OK
, "CoInternetGetSession failed: %08x\n", hres
);
2899 if(test_flags
& TEST_EMULATEPROT
) {
2900 hres
= IInternetSession_RegisterNameSpace(session
, &ClassFactory
, &IID_NULL
, protocol_names
[prot
], 0, NULL
, 0);
2901 ok(hres
== S_OK
, "RegisterNameSpace failed: %08x\n", hres
);
2904 hres
= IInternetSession_CreateBinding(session
, NULL
, binding_urls
[prot
], NULL
, NULL
, &protocol
, 0);
2905 binding_protocol
= protocol
;
2906 ok(hres
== S_OK
, "CreateBinding failed: %08x\n", hres
);
2907 ok(protocol
!= NULL
, "protocol == NULL\n");
2909 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IInternetBindInfo
, (void**)&prot_bind_info
);
2910 ok(hres
== S_OK
, "QueryInterface(IID_IInternetBindInfo) failed: %08x\n", hres
);
2912 hres
= IInternetProtocol_QueryInterface(protocol
, &IID_IInternetProtocolSink
, (void**)&binding_sink
);
2913 ok(hres
== S_OK
, "QueryInterface(IID_IInternetProtocolSink) failed: %08x\n", hres
);
2916 SET_EXPECT(QueryService_InternetProtocol
);
2917 SET_EXPECT(CreateInstance
);
2918 SET_EXPECT(ReportProgress_PROTOCOLCLASSID
);
2919 SET_EXPECT(SetPriority
);
2922 expect_hrResult
= S_OK
;
2923 hres
= IInternetProtocol_Start(protocol
, binding_urls
[prot
], &protocol_sink
, &bind_info
, pi
, 0);
2924 ok(hres
== S_OK
, "Start failed: %08x\n", hres
);
2926 CHECK_CALLED(QueryService_InternetProtocol
);
2927 CHECK_CALLED(CreateInstance
);
2928 CHECK_CALLED(ReportProgress_PROTOCOLCLASSID
);
2929 CHECK_CALLED(SetPriority
);
2930 CHECK_CALLED(Start
);
2932 if(prot
== HTTP_TEST
|| prot
== HTTPS_TEST
) {
2933 while(prot_state
< 4) {
2934 WaitForSingleObject(event_complete
, INFINITE
);
2935 if(mimefilter_test
&& filtered_protocol
) {
2936 SET_EXPECT(Continue
);
2937 IInternetProtocol_Continue(filtered_protocol
, pdata
);
2938 CHECK_CALLED(Continue
);
2940 SET_EXPECT(Continue
);
2941 IInternetProtocol_Continue(protocol
, pdata
);
2942 CHECK_CALLED(Continue
);
2944 SetEvent(event_complete2
);
2947 CHECK_CALLED(ReportData
); /* Set in ReportResult */
2948 WaitForSingleObject(event_complete
, INFINITE
);
2951 SET_EXPECT(MimeFilter_LockRequest
);
2953 SET_EXPECT(LockRequest
);
2954 hres
= IInternetProtocol_LockRequest(protocol
, 0);
2955 ok(hres
== S_OK
, "LockRequest failed: %08x\n", hres
);
2957 CHECK_CALLED(MimeFilter_LockRequest
);
2959 CHECK_CALLED(LockRequest
);
2962 SET_EXPECT(MimeFilter_UnlockRequest
);
2964 SET_EXPECT(UnlockRequest
);
2965 hres
= IInternetProtocol_UnlockRequest(protocol
);
2966 ok(hres
== S_OK
, "UnlockRequest failed: %08x\n", hres
);
2968 CHECK_CALLED(MimeFilter_UnlockRequest
);
2970 CHECK_CALLED(UnlockRequest
);
2974 SET_EXPECT(MimeFilter_Terminate
);
2976 SET_EXPECT(Terminate
);
2977 hres
= IInternetProtocol_Terminate(protocol
, 0);
2978 ok(hres
== S_OK
, "Terminate failed: %08x\n", hres
);
2980 CLEAR_CALLED(MimeFilter_Terminate
);
2982 CHECK_CALLED(Terminate
);
2984 if(filtered_protocol
)
2985 IInternetProtocol_Release(filtered_protocol
);
2986 IInternetBindInfo_Release(prot_bind_info
);
2987 IInternetProtocolSink_Release(binding_sink
);
2988 ref
= IInternetProtocol_Release(protocol
);
2989 ok(!ref
, "ref=%u, expected 0\n", ref
);
2991 if(test_flags
& TEST_EMULATEPROT
) {
2992 hres
= IInternetSession_UnregisterNameSpace(session
, &ClassFactory
, protocol_names
[prot
]);
2993 ok(hres
== S_OK
, "UnregisterNameSpace failed: %08x\n", hres
);
2996 IInternetSession_Release(session
);
2999 static void register_filter(void)
3001 IInternetSession
*session
;
3004 static const WCHAR gzipW
[] = {'g','z','i','p',0};
3006 CoInternetGetSession(0, &session
, 0);
3008 hres
= IInternetSession_RegisterMimeFilter(session
, &mimefilter_cf
, &IID_IInternetProtocol
, gzipW
);
3009 ok(hres
== S_OK
, "RegisterMimeFilter failed: %08x\n", hres
);
3011 IInternetSession_Release(session
);
3014 START_TEST(protocol
)
3016 OleInitialize(NULL
);
3018 event_complete
= CreateEvent(NULL
, FALSE
, FALSE
, NULL
);
3019 event_complete2
= CreateEvent(NULL
, FALSE
, FALSE
, NULL
);
3020 event_continue
= CreateEvent(NULL
, FALSE
, FALSE
, NULL
);
3021 event_continue_done
= CreateEvent(NULL
, FALSE
, FALSE
, NULL
);
3025 test_file_protocol();
3026 test_http_protocol();
3027 test_https_protocol();
3028 test_ftp_protocol();
3029 test_gopher_protocol();
3031 test_CreateBinding();
3033 bindf
&= ~BINDF_FROMURLMON
;
3034 trace("Testing file binding (mime verification, emulate prot)...\n");
3035 test_binding(FILE_TEST
, PI_MIMEVERIFICATION
, TEST_EMULATEPROT
);
3036 trace("Testing http binding (mime verification, emulate prot)...\n");
3037 test_binding(HTTP_TEST
, PI_MIMEVERIFICATION
, TEST_EMULATEPROT
);
3038 trace("Testing http binding (mime verification, emulate prot, short read, direct read)...\n");
3039 test_binding(HTTP_TEST
, PI_MIMEVERIFICATION
, TEST_EMULATEPROT
|TEST_SHORT_READ
|TEST_DIRECT_READ
);
3040 trace("Testing file binding (mime verification, emulate prot, mime filter)...\n");
3041 test_binding(FILE_TEST
, PI_MIMEVERIFICATION
, TEST_EMULATEPROT
|TEST_FILTER
);
3042 trace("Testing http binding (mime verification, emulate prot, mime filter)...\n");
3043 test_binding(HTTP_TEST
, PI_MIMEVERIFICATION
, TEST_EMULATEPROT
|TEST_FILTER
);
3044 trace("Testing http binding (mime verification, emulate prot, direct read)...\n");
3045 test_binding(HTTP_TEST
, PI_MIMEVERIFICATION
, TEST_EMULATEPROT
|TEST_DIRECT_READ
);
3047 CloseHandle(event_complete
);
3048 CloseHandle(event_complete2
);
3049 CloseHandle(event_continue
);
3050 CloseHandle(event_continue_done
);