/* * Copyright 2005-2011 Jacek Caban for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #define COBJMACROS #define CONST_VTABLE #include <wine/test.h> #include <stdarg.h> #include <stdio.h> #include "windef.h" #include "winbase.h" #include "ole2.h" #include "urlmon.h" #include "wininet.h" static HRESULT (WINAPI *pCoInternetGetSession)(DWORD, IInternetSession **, DWORD); static HRESULT (WINAPI *pReleaseBindInfo)(BINDINFO*); static HRESULT (WINAPI *pCreateUri)(LPCWSTR, DWORD, DWORD_PTR, IUri**); #define DEFINE_EXPECT(func) \ static BOOL expect_ ## func = FALSE, called_ ## func = FALSE #define SET_EXPECT(func) \ expect_ ## func = TRUE #define CHECK_EXPECT2(func) \ do { \ ok(expect_ ##func, "unexpected call " #func "\n"); \ called_ ## func = TRUE; \ }while(0) #define CHECK_EXPECT(func) \ do { \ CHECK_EXPECT2(func); \ expect_ ## func = FALSE; \ }while(0) #define CHECK_CALLED(func) \ do { \ ok(called_ ## func, "expected " #func "\n"); \ expect_ ## func = called_ ## func = FALSE; \ }while(0) #define CHECK_NOT_CALLED(func) \ do { \ ok(!called_ ## func, "unexpected " #func "\n"); \ expect_ ## func = called_ ## func = FALSE; \ }while(0) #define CLEAR_CALLED(func) \ expect_ ## func = called_ ## func = FALSE DEFINE_EXPECT(GetBindInfo); DEFINE_EXPECT(ReportProgress_MIMETYPEAVAILABLE); DEFINE_EXPECT(ReportProgress_DIRECTBIND); DEFINE_EXPECT(ReportProgress_RAWMIMETYPE); DEFINE_EXPECT(ReportProgress_FINDINGRESOURCE); DEFINE_EXPECT(ReportProgress_CONNECTING); DEFINE_EXPECT(ReportProgress_SENDINGREQUEST); DEFINE_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); DEFINE_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE); DEFINE_EXPECT(ReportProgress_PROTOCOLCLASSID); DEFINE_EXPECT(ReportProgress_COOKIE_SENT); DEFINE_EXPECT(ReportProgress_REDIRECTING); DEFINE_EXPECT(ReportProgress_ENCODING); DEFINE_EXPECT(ReportProgress_ACCEPTRANGES); DEFINE_EXPECT(ReportProgress_PROXYDETECTING); DEFINE_EXPECT(ReportProgress_LOADINGMIMEHANDLER); DEFINE_EXPECT(ReportProgress_DECODING); DEFINE_EXPECT(ReportData); DEFINE_EXPECT(ReportData2); DEFINE_EXPECT(ReportResult); DEFINE_EXPECT(GetBindString_ACCEPT_MIMES); DEFINE_EXPECT(GetBindString_USER_AGENT); DEFINE_EXPECT(GetBindString_POST_COOKIE); DEFINE_EXPECT(GetBindString_URL); DEFINE_EXPECT(GetBindString_ROOTDOC_URL); DEFINE_EXPECT(QueryService_HttpNegotiate); DEFINE_EXPECT(QueryService_InternetProtocol); DEFINE_EXPECT(QueryService_HttpSecurity); DEFINE_EXPECT(QueryInterface_IWinInetInfo); DEFINE_EXPECT(QueryInterface_IWinInetHttpInfo); DEFINE_EXPECT(BeginningTransaction); DEFINE_EXPECT(GetRootSecurityId); DEFINE_EXPECT(OnResponse); DEFINE_EXPECT(Switch); DEFINE_EXPECT(Continue); DEFINE_EXPECT(CreateInstance); DEFINE_EXPECT(Start); DEFINE_EXPECT(StartEx); DEFINE_EXPECT(Terminate); DEFINE_EXPECT(Read); DEFINE_EXPECT(Read2); DEFINE_EXPECT(SetPriority); DEFINE_EXPECT(LockRequest); DEFINE_EXPECT(UnlockRequest); DEFINE_EXPECT(Abort); DEFINE_EXPECT(MimeFilter_CreateInstance); DEFINE_EXPECT(MimeFilter_Start); DEFINE_EXPECT(MimeFilter_ReportData); DEFINE_EXPECT(MimeFilter_ReportResult); DEFINE_EXPECT(MimeFilter_Terminate); DEFINE_EXPECT(MimeFilter_LockRequest); DEFINE_EXPECT(MimeFilter_UnlockRequest); DEFINE_EXPECT(MimeFilter_Read); DEFINE_EXPECT(MimeFilter_Switch); DEFINE_EXPECT(MimeFilter_Continue); DEFINE_EXPECT(Stream_Seek); DEFINE_EXPECT(Stream_Read); static const WCHAR wszIndexHtml[] = {'i','n','d','e','x','.','h','t','m','l',0}; static const WCHAR index_url[] = {'f','i','l','e',':','i','n','d','e','x','.','h','t','m','l',0}; static const WCHAR acc_mimeW[] = {'*','/','*',0}; static const WCHAR user_agentW[] = {'W','i','n','e',0}; static const WCHAR text_htmlW[] = {'t','e','x','t','/','h','t','m','l',0}; static const WCHAR hostW[] = {'w','w','w','.','w','i','n','e','h','q','.','o','r','g',0}; static const WCHAR winehq_ipW[] = {'2','0','9','.','4','6','.','2','5','.','1','3','4',0}; static const WCHAR emptyW[] = {0}; static const WCHAR pjpegW[] = {'i','m','a','g','e','/','p','j','p','e','g',0}; static const WCHAR gifW[] = {'i','m','a','g','e','/','g','i','f',0}; static HRESULT expect_hrResult; static LPCWSTR file_name, http_url, expect_wsz; static IInternetProtocol *async_protocol = NULL; static BOOL first_data_notif, http_is_first, test_redirect; static int prot_state, read_report_data, post_stream_read; static DWORD bindf, ex_priority , pi; static IInternetProtocol *binding_protocol, *filtered_protocol; static IInternetBindInfo *prot_bind_info; static IInternetProtocolSink *binding_sink, *filtered_sink; static void *expect_pv; static HANDLE event_complete, event_complete2, event_continue, event_continue_done; static BOOL binding_test; static PROTOCOLDATA protocoldata, *pdata, continue_protdata; static DWORD prot_read, filter_state, http_post_test, thread_id; static BOOL security_problem, test_async_req, impl_protex; static BOOL async_read_pending, mimefilter_test, direct_read, wait_for_switch, emulate_prot, short_read, test_abort; static BOOL empty_file, no_mime, bind_from_cache, file_with_hash; enum { STATE_CONNECTING, STATE_SENDINGREQUEST, STATE_STARTDOWNLOADING, STATE_DOWNLOADING } state; static enum { FILE_TEST, HTTP_TEST, HTTPS_TEST, FTP_TEST, MK_TEST, ITS_TEST, BIND_TEST } tested_protocol; static const WCHAR protocol_names[][10] = { {'f','i','l','e',0}, {'h','t','t','p',0}, {'h','t','t','p','s',0}, {'f','t','p',0}, {'m','k',0}, {'i','t','s',0}, {'t','e','s','t',0} }; static const WCHAR binding_urls[][130] = { {'f','i','l','e',':','t','e','s','t','.','h','t','m','l',0}, {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q','.', 'o','r','g','/','s','i','t','e','/','a','b','o','u','t',0}, {'h','t','t','p','s',':','/','/','w','w','w','.','c','o','d','e','w','e','a','v','e','r','s', '.','c','o','m','/','t','e','s','t','.','h','t','m','l',0}, {'f','t','p',':','/','/','f','t','p','.','w','i','n','e','h','q','.','o','r','g', '/','p','u','b','/','o','t','h','e','r', '/','w','i','n','e','l','o','g','o','.','x','c','f','.','t','a','r','.','b','z','2',0}, {'m','k',':','t','e','s','t',0}, {'i','t','s',':','t','e','s','t','.','c','h','m',':',':','/','b','l','a','n','k','.','h','t','m','l',0}, {'t','e','s','t',':','/','/','f','i','l','e','.','h','t','m','l',0} }; static const CHAR post_data[] = "mode=Test"; static int strcmp_wa(LPCWSTR strw, const char *stra) { CHAR buf[512]; WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), NULL, NULL); return lstrcmpA(stra, buf); } static const char *w2a(LPCWSTR str) { static char buf[INTERNET_MAX_URL_LENGTH]; WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL); return buf; } static HRESULT WINAPI HttpSecurity_QueryInterface(IHttpSecurity *iface, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IHttpSecurity, riid)) { *ppv = iface; return S_OK; } ok(0, "unexpected call\n"); return E_NOINTERFACE; } static ULONG WINAPI HttpSecurity_AddRef(IHttpSecurity *iface) { return 2; } static ULONG WINAPI HttpSecurity_Release(IHttpSecurity *iface) { return 1; } static HRESULT WINAPI HttpSecurity_GetWindow(IHttpSecurity* iface, REFGUID rguidReason, HWND *phwnd) { trace("HttpSecurity_GetWindow\n"); return S_FALSE; } static HRESULT WINAPI HttpSecurity_OnSecurityProblem(IHttpSecurity *iface, DWORD dwProblem) { trace("Security problem: %u\n", dwProblem); ok(dwProblem == ERROR_INTERNET_SEC_CERT_REV_FAILED, "Expected ERROR_INTERNET_SEC_CERT_REV_FAILED got %u\n", dwProblem); /* Only retry once */ if (security_problem) return E_ABORT; security_problem = TRUE; SET_EXPECT(BeginningTransaction); return RPC_E_RETRY; } static IHttpSecurityVtbl HttpSecurityVtbl = { HttpSecurity_QueryInterface, HttpSecurity_AddRef, HttpSecurity_Release, HttpSecurity_GetWindow, HttpSecurity_OnSecurityProblem }; static IHttpSecurity http_security = { &HttpSecurityVtbl }; static HRESULT WINAPI HttpNegotiate_QueryInterface(IHttpNegotiate2 *iface, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IHttpNegotiate, riid) || IsEqualGUID(&IID_IHttpNegotiate2, riid)) { *ppv = iface; return S_OK; } ok(0, "unexpected call\n"); return E_NOINTERFACE; } static ULONG WINAPI HttpNegotiate_AddRef(IHttpNegotiate2 *iface) { return 2; } static ULONG WINAPI HttpNegotiate_Release(IHttpNegotiate2 *iface) { return 1; } static HRESULT WINAPI HttpNegotiate_BeginningTransaction(IHttpNegotiate2 *iface, LPCWSTR szURL, LPCWSTR szHeaders, DWORD dwReserved, LPWSTR *pszAdditionalHeaders) { LPWSTR addl_headers; static const WCHAR wszHeaders[] = {'C','o','n','t','e','n','t','-','T','y','p','e',':',' ','a','p','p','l','i','c','a','t', 'i','o','n','/','x','-','w','w','w','-','f','o','r','m','-','u','r','l','e','n','c','o', 'd','e','d','\r','\n',0}; CHECK_EXPECT(BeginningTransaction); if(binding_test) ok(!lstrcmpW(szURL, binding_urls[tested_protocol]), "szURL != http_url\n"); else ok(!lstrcmpW(szURL, http_url), "szURL != http_url\n"); ok(!dwReserved, "dwReserved=%d, expected 0\n", dwReserved); ok(pszAdditionalHeaders != NULL, "pszAdditionalHeaders == NULL\n"); if(pszAdditionalHeaders) { ok(*pszAdditionalHeaders == NULL, "*pszAdditionalHeaders != NULL\n"); if (http_post_test) { addl_headers = CoTaskMemAlloc(sizeof(wszHeaders)); memcpy(addl_headers, wszHeaders, sizeof(wszHeaders)); *pszAdditionalHeaders = addl_headers; } } return S_OK; } static HRESULT WINAPI HttpNegotiate_OnResponse(IHttpNegotiate2 *iface, DWORD dwResponseCode, LPCWSTR szResponseHeaders, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders) { CHECK_EXPECT(OnResponse); ok(dwResponseCode == 200, "dwResponseCode=%d, expected 200\n", dwResponseCode); ok(szResponseHeaders != NULL, "szResponseHeaders == NULL\n"); ok(szRequestHeaders == NULL, "szRequestHeaders != NULL\n"); ok(pszAdditionalRequestHeaders == NULL, "pszAdditionalHeaders != NULL\n"); return S_OK; } static HRESULT WINAPI HttpNegotiate_GetRootSecurityId(IHttpNegotiate2 *iface, BYTE *pbSecurityId, DWORD *pcbSecurityId, DWORD_PTR dwReserved) { static const BYTE sec_id[] = {'h','t','t','p',':','t','e','s','t',1,0,0,0}; CHECK_EXPECT(GetRootSecurityId); ok(!dwReserved, "dwReserved=%ld, expected 0\n", dwReserved); ok(pbSecurityId != NULL, "pbSecurityId == NULL\n"); ok(pcbSecurityId != NULL, "pcbSecurityId == NULL\n"); if(pcbSecurityId) { ok(*pcbSecurityId == 512, "*pcbSecurityId=%d, expected 512\n", *pcbSecurityId); *pcbSecurityId = sizeof(sec_id); } if(pbSecurityId) memcpy(pbSecurityId, sec_id, sizeof(sec_id)); return E_FAIL; } static IHttpNegotiate2Vtbl HttpNegotiateVtbl = { HttpNegotiate_QueryInterface, HttpNegotiate_AddRef, HttpNegotiate_Release, HttpNegotiate_BeginningTransaction, HttpNegotiate_OnResponse, HttpNegotiate_GetRootSecurityId }; static IHttpNegotiate2 http_negotiate = { &HttpNegotiateVtbl }; static HRESULT QueryInterface(REFIID,void**); static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv) { return QueryInterface(riid, ppv); } static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface) { return 2; } static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface) { return 1; } static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface, REFGUID guidService, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IHttpNegotiate, guidService) || IsEqualGUID(&IID_IHttpNegotiate2, riid)) { CHECK_EXPECT2(QueryService_HttpNegotiate); return IHttpNegotiate2_QueryInterface(&http_negotiate, riid, ppv); } if(IsEqualGUID(&IID_IInternetProtocol, guidService)) { ok(IsEqualGUID(&IID_IInternetProtocol, riid), "unexpected riid\n"); CHECK_EXPECT(QueryService_InternetProtocol); return E_NOINTERFACE; } if(IsEqualGUID(&IID_IHttpSecurity, guidService)) { ok(IsEqualGUID(&IID_IHttpSecurity, riid), "unexpected riid\n"); CHECK_EXPECT(QueryService_HttpSecurity); return IHttpSecurity_QueryInterface(&http_security, riid, ppv); } if(IsEqualGUID(&IID_IGetBindHandle, guidService)) { trace("QueryService(IID_IGetBindHandle)\n"); *ppv = NULL; return E_NOINTERFACE; } if(IsEqualGUID(&IID_IWindowForBindingUI, guidService)) { trace("QueryService(IID_IWindowForBindingUI)\n"); *ppv = NULL; return E_NOINTERFACE; } ok(0, "unexpected service %s\n", wine_dbgstr_guid(guidService)); return E_FAIL; } static const IServiceProviderVtbl ServiceProviderVtbl = { ServiceProvider_QueryInterface, ServiceProvider_AddRef, ServiceProvider_Release, ServiceProvider_QueryService }; static IServiceProvider service_provider = { &ServiceProviderVtbl }; static HRESULT WINAPI Stream_QueryInterface(IStream *iface, REFIID riid, void **ppv) { static const IID IID_strm_unknown = {0x2f68429a,0x199a,0x4043,{0x93,0x11,0xf2,0xfe,0x7c,0x13,0xcc,0xb9}}; if(!IsEqualGUID(&IID_strm_unknown, riid)) /* IE11 */ ok(0, "unexpected call %s\n", wine_dbgstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; } static ULONG WINAPI Stream_AddRef(IStream *iface) { return 2; } static ULONG WINAPI Stream_Release(IStream *iface) { return 1; } static HRESULT WINAPI Stream_Read(IStream *iface, void *pv, ULONG cb, ULONG *pcbRead) { CHECK_EXPECT2(Stream_Read); ok(GetCurrentThreadId() != thread_id, "wrong thread %d\n", GetCurrentThreadId()); ok(pv != NULL, "pv == NULL\n"); ok(cb == 0x20000 || broken(cb == 0x2000), "cb = %d\n", cb); ok(pcbRead != NULL, "pcbRead == NULL\n"); if(post_stream_read) { *pcbRead = 0; return S_FALSE; } memcpy(pv, post_data, sizeof(post_data)-1); post_stream_read += *pcbRead = sizeof(post_data)-1; return S_OK; } static HRESULT WINAPI Stream_Write(IStream *iface, const void *pv, ULONG cb, ULONG *pcbWritten) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_Seek(IStream *iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition) { CHECK_EXPECT(Stream_Seek); ok(!dlibMove.QuadPart, "dlibMove != 0\n"); ok(dwOrigin == STREAM_SEEK_SET, "dwOrigin = %d\n", dwOrigin); ok(!plibNewPosition, "plibNewPosition == NULL\n"); return S_OK; } static HRESULT WINAPI Stream_SetSize(IStream *iface, ULARGE_INTEGER libNewSize) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_CopyTo(IStream *iface, IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_Commit(IStream *iface, DWORD grfCommitFlags) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_Revert(IStream *iface) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_LockRegion(IStream *iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_UnlockRegion(IStream *iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_Stat(IStream *iface, STATSTG *pstatstg, DWORD dwStatFlag) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Stream_Clone(IStream *iface, IStream **ppstm) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static const IStreamVtbl StreamVtbl = { Stream_QueryInterface, Stream_AddRef, Stream_Release, Stream_Read, Stream_Write, Stream_Seek, Stream_SetSize, Stream_CopyTo, Stream_Commit, Stream_Revert, Stream_LockRegion, Stream_UnlockRegion, Stream_Stat, Stream_Clone }; static IStream Stream = { &StreamVtbl }; static HRESULT WINAPI ProtocolSink_QueryInterface(IInternetProtocolSink *iface, REFIID riid, void **ppv) { return QueryInterface(riid, ppv); } static ULONG WINAPI ProtocolSink_AddRef(IInternetProtocolSink *iface) { return 2; } static ULONG WINAPI ProtocolSink_Release(IInternetProtocolSink *iface) { return 1; } static void call_continue(PROTOCOLDATA *protocol_data) { HRESULT hres; if (winetest_debug > 1) trace("continue in state %d\n", state); if(state == STATE_CONNECTING) { if(tested_protocol == HTTP_TEST || tested_protocol == HTTPS_TEST || tested_protocol == FTP_TEST) { if (http_is_first){ CLEAR_CALLED(ReportProgress_FINDINGRESOURCE); CLEAR_CALLED(ReportProgress_PROXYDETECTING); } CLEAR_CALLED(ReportProgress_CONNECTING); } if(tested_protocol == FTP_TEST) todo_wine CHECK_CALLED(ReportProgress_SENDINGREQUEST); else if (tested_protocol != HTTPS_TEST) CHECK_CALLED(ReportProgress_SENDINGREQUEST); if(test_redirect) CHECK_CALLED(ReportProgress_REDIRECTING); state = test_async_req ? STATE_SENDINGREQUEST : STATE_STARTDOWNLOADING; } switch(state) { case STATE_SENDINGREQUEST: SET_EXPECT(Stream_Read); SET_EXPECT(ReportProgress_SENDINGREQUEST); break; case STATE_STARTDOWNLOADING: if(tested_protocol == HTTP_TEST || tested_protocol == HTTPS_TEST) { SET_EXPECT(OnResponse); if(tested_protocol == HTTPS_TEST || test_redirect || test_abort || empty_file) SET_EXPECT(ReportProgress_ACCEPTRANGES); SET_EXPECT(ReportProgress_ENCODING); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); if(bindf & BINDF_NEEDFILE) SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); } default: break; } if(state != STATE_SENDINGREQUEST) SET_EXPECT(ReportData); hres = IInternetProtocol_Continue(async_protocol, protocol_data); ok(hres == S_OK, "Continue failed: %08x\n", hres); if(tested_protocol == FTP_TEST || security_problem) CLEAR_CALLED(ReportData); else if(state != STATE_SENDINGREQUEST) CHECK_CALLED(ReportData); switch(state) { case STATE_SENDINGREQUEST: CHECK_CALLED(Stream_Read); CHECK_CALLED(ReportProgress_SENDINGREQUEST); state = STATE_STARTDOWNLOADING; break; case STATE_STARTDOWNLOADING: if (! security_problem) { state = STATE_DOWNLOADING; if(tested_protocol == HTTP_TEST || tested_protocol == HTTPS_TEST) { CHECK_CALLED(OnResponse); if(tested_protocol == HTTPS_TEST || empty_file) CHECK_CALLED(ReportProgress_ACCEPTRANGES); else if(test_redirect || test_abort) CLEAR_CALLED(ReportProgress_ACCEPTRANGES); CLEAR_CALLED(ReportProgress_ENCODING); CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); if(bindf & BINDF_NEEDFILE) CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE); } } else { security_problem = FALSE; SET_EXPECT(ReportProgress_CONNECTING); } default: break; } } static HRESULT WINAPI ProtocolSink_Switch(IInternetProtocolSink *iface, PROTOCOLDATA *pProtocolData) { if(tested_protocol == FTP_TEST) CHECK_EXPECT2(Switch); else CHECK_EXPECT(Switch); ok(pProtocolData != NULL, "pProtocolData == NULL\n"); if(binding_test) { ok(pProtocolData != &protocoldata, "pProtocolData == &protocoldata\n"); ok(pProtocolData->grfFlags == protocoldata.grfFlags, "grfFlags wrong %x/%x\n", pProtocolData->grfFlags, protocoldata.grfFlags ); ok(pProtocolData->dwState == protocoldata.dwState, "dwState wrong %x/%x\n", pProtocolData->dwState, protocoldata.dwState ); ok(pProtocolData->pData == protocoldata.pData, "pData wrong %p/%p\n", pProtocolData->pData, protocoldata.pData ); ok(pProtocolData->cbData == protocoldata.cbData, "cbData wrong %x/%x\n", pProtocolData->cbData, protocoldata.cbData ); } pdata = pProtocolData; if(binding_test) { SetEvent(event_complete); ok( WaitForSingleObject(event_complete2, 90000) == WAIT_OBJECT_0, "wait timed out\n" ); return S_OK; }if(direct_read) { continue_protdata = *pProtocolData; SetEvent(event_continue); ok( WaitForSingleObject(event_continue_done, 90000) == WAIT_OBJECT_0, "wait timed out\n" ); }else { call_continue(pProtocolData); SetEvent(event_complete); } return S_OK; } static const char *status_names[] = { "0", "FINDINGRESOURCE", "CONNECTING", "REDIRECTING", "BEGINDOWNLOADDATA", "DOWNLOADINGDATA", "ENDDOWNLOADDATA", "BEGINDOWNLOADCOMPONENTS", "INSTALLINGCOMPONENTS", "ENDDOWNLOADCOMPONENTS", "USINGCACHEDCOPY", "SENDINGREQUEST", "CLASSIDAVAILABLE", "MIMETYPEAVAILABLE", "CACHEFILENAMEAVAILABLE", "BEGINSYNCOPERATION", "ENDSYNCOPERATION", "BEGINUPLOADDATA", "UPLOADINGDATA", "ENDUPLOADINGDATA", "PROTOCOLCLASSID", "ENCODING", "VERIFIEDMIMETYPEAVAILABLE", "CLASSINSTALLLOCATION", "DECODING", "LOADINGMIMEHANDLER", "CONTENTDISPOSITIONATTACH", "FILTERREPORTMIMETYPE", "CLSIDCANINSTANTIATE", "IUNKNOWNAVAILABLE", "DIRECTBIND", "RAWMIMETYPE", "PROXYDETECTING", "ACCEPTRANGES", "COOKIE_SENT", "COMPACT_POLICY_RECEIVED", "COOKIE_SUPPRESSED", "COOKIE_STATE_UNKNOWN", "COOKIE_STATE_ACCEPT", "COOKIE_STATE_REJECT", "COOKIE_STATE_PROMPT", "COOKIE_STATE_LEASH", "COOKIE_STATE_DOWNGRADE", "POLICY_HREF", "P3P_HEADER", "SESSION_COOKIE_RECEIVED", "PERSISTENT_COOKIE_RECEIVED", "SESSION_COOKIES_ALLOWED", "CACHECONTROL", "CONTENTDISPOSITIONFILENAME", "MIMETEXTPLAINMISMATCH", "PUBLISHERAVAILABLE", "DISPLAYNAMEAVAILABLE" }; static HRESULT WINAPI ProtocolSink_ReportProgress(IInternetProtocolSink *iface, ULONG ulStatusCode, LPCWSTR szStatusText) { static const WCHAR null_guid[] = {'{','0','0','0','0','0','0','0','0','-','0','0','0','0','-', '0','0','0','0','-','0','0','0','0','-','0','0','0','0','0','0','0','0','0','0','0','0','}',0}; static const WCHAR text_plain[] = {'t','e','x','t','/','p','l','a','i','n',0}; if (winetest_debug > 1) { if (ulStatusCode < sizeof(status_names)/sizeof(status_names[0])) trace( "progress: %s %s\n", status_names[ulStatusCode], wine_dbgstr_w(szStatusText) ); else trace( "progress: %u %s\n", ulStatusCode, wine_dbgstr_w(szStatusText) ); } switch(ulStatusCode) { case BINDSTATUS_MIMETYPEAVAILABLE: CHECK_EXPECT2(ReportProgress_MIMETYPEAVAILABLE); if(tested_protocol != FILE_TEST && tested_protocol != ITS_TEST && !mimefilter_test && (pi & PI_MIMEVERIFICATION)) { if(!short_read || !direct_read) CHECK_CALLED(Read); /* set in Continue */ else if(short_read) CHECK_CALLED(Read2); /* set in Read */ } ok(szStatusText != NULL, "szStatusText == NULL\n"); if(szStatusText) { if(tested_protocol == BIND_TEST) ok(!lstrcmpW(szStatusText, expect_wsz), "unexpected szStatusText %s\n", wine_dbgstr_w(szStatusText)); else if (http_post_test) ok(lstrlenW(text_plain) <= lstrlenW(szStatusText) && !memcmp(szStatusText, text_plain, lstrlenW(text_plain)*sizeof(WCHAR)), "szStatusText != text/plain\n"); else if(empty_file) ok(!strcmp_wa(szStatusText, "application/javascript"), "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); else if((pi & PI_MIMEVERIFICATION) && emulate_prot && !mimefilter_test && tested_protocol==HTTP_TEST && !short_read) ok(lstrlenW(gifW) <= lstrlenW(szStatusText) && !memcmp(szStatusText, gifW, lstrlenW(gifW)*sizeof(WCHAR)), "szStatusText != image/gif\n"); else if(!mimefilter_test) ok(lstrlenW(text_htmlW) <= lstrlenW(szStatusText) && !memcmp(szStatusText, text_htmlW, lstrlenW(text_htmlW)*sizeof(WCHAR)), "szStatusText != text/html\n"); } break; case BINDSTATUS_DIRECTBIND: CHECK_EXPECT2(ReportProgress_DIRECTBIND); ok(szStatusText == NULL, "szStatusText != NULL\n"); break; case BINDSTATUS_RAWMIMETYPE: CHECK_EXPECT2(ReportProgress_RAWMIMETYPE); ok(szStatusText != NULL, "szStatusText == NULL\n"); if(szStatusText) ok(lstrlenW(szStatusText) < lstrlenW(text_htmlW) || !memcmp(szStatusText, text_htmlW, lstrlenW(text_htmlW)*sizeof(WCHAR)), "szStatusText != text/html\n"); break; case BINDSTATUS_CACHEFILENAMEAVAILABLE: CHECK_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); ok(szStatusText != NULL, "szStatusText == NULL\n"); if(szStatusText) { if(binding_test) ok(!lstrcmpW(szStatusText, expect_wsz), "unexpected szStatusText\n"); else if(tested_protocol == FILE_TEST) ok(!lstrcmpW(szStatusText, file_name), "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); else ok(szStatusText != NULL, "szStatusText == NULL\n"); } break; case BINDSTATUS_FINDINGRESOURCE: CHECK_EXPECT2(ReportProgress_FINDINGRESOURCE); ok(szStatusText != NULL, "szStatusText == NULL\n"); break; case BINDSTATUS_CONNECTING: CHECK_EXPECT2(ReportProgress_CONNECTING); ok(szStatusText != NULL, "szStatusText == NULL\n"); break; case BINDSTATUS_SENDINGREQUEST: CHECK_EXPECT2(ReportProgress_SENDINGREQUEST); if(tested_protocol == FILE_TEST || tested_protocol == ITS_TEST) { ok(szStatusText != NULL, "szStatusText == NULL\n"); if(szStatusText) ok(!*szStatusText, "wrong szStatusText\n"); } break; case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE: CHECK_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE); ok(szStatusText != NULL, "szStatusText == NULL\n"); if(szStatusText) ok(!strcmp_wa(szStatusText, "text/html"), "szStatusText != text/html\n"); break; case BINDSTATUS_PROTOCOLCLASSID: CHECK_EXPECT(ReportProgress_PROTOCOLCLASSID); ok(szStatusText != NULL, "szStatusText == NULL\n"); ok(!lstrcmpW(szStatusText, null_guid), "unexpected classid %s\n", wine_dbgstr_w(szStatusText)); break; case BINDSTATUS_COOKIE_SENT: CHECK_EXPECT2(ReportProgress_COOKIE_SENT); ok(szStatusText == NULL, "szStatusText != NULL\n"); break; case BINDSTATUS_REDIRECTING: CHECK_EXPECT(ReportProgress_REDIRECTING); if(test_redirect) ok(!strcmp_wa(szStatusText, "http://test.winehq.org/tests/hello.html"), "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); else ok(szStatusText == NULL, "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); break; case BINDSTATUS_ENCODING: CHECK_EXPECT(ReportProgress_ENCODING); ok(!strcmp_wa(szStatusText, "gzip"), "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); break; case BINDSTATUS_ACCEPTRANGES: CHECK_EXPECT(ReportProgress_ACCEPTRANGES); ok(!szStatusText, "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); break; case BINDSTATUS_PROXYDETECTING: if(!called_ReportProgress_PROXYDETECTING) SET_EXPECT(ReportProgress_CONNECTING); CHECK_EXPECT2(ReportProgress_PROXYDETECTING); ok(!szStatusText, "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); break; case BINDSTATUS_LOADINGMIMEHANDLER: CHECK_EXPECT(ReportProgress_LOADINGMIMEHANDLER); ok(!szStatusText, "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); break; case BINDSTATUS_DECODING: CHECK_EXPECT(ReportProgress_DECODING); ok(!lstrcmpW(szStatusText, pjpegW), "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); break; case BINDSTATUS_RESERVED_7: trace("BINDSTATUS_RESERVED_7\n"); break; case BINDSTATUS_RESERVED_8: trace("BINDSTATUS_RESERVED_8\n"); break; default: ok(0, "Unexpected status %d (%d)\n", ulStatusCode, ulStatusCode-BINDSTATUS_LAST); }; return S_OK; } static HRESULT WINAPI ProtocolSink_ReportData(IInternetProtocolSink *iface, DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax) { HRESULT hres; static int rec_depth; rec_depth++; if(!mimefilter_test && (tested_protocol == FILE_TEST || tested_protocol == ITS_TEST)) { CHECK_EXPECT2(ReportData); ok(ulProgress == ulProgressMax, "ulProgress (%d) != ulProgressMax (%d)\n", ulProgress, ulProgressMax); if(!file_with_hash) ok(ulProgressMax == 13, "ulProgressMax=%d, expected 13\n", ulProgressMax); /* BSCF_SKIPDRAINDATAFORFILEURLS added in IE8 */ if(tested_protocol == FILE_TEST) ok((grfBSCF == (BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION)) || (grfBSCF == (BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION | BSCF_SKIPDRAINDATAFORFILEURLS)), "grcfBSCF = %08x\n", grfBSCF); else ok(grfBSCF == (BSCF_FIRSTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE), "grcfBSCF = %08x\n", grfBSCF); }else if(bind_from_cache) { CHECK_EXPECT(ReportData); ok(grfBSCF == (BSCF_LASTDATANOTIFICATION|BSCF_DATAFULLYAVAILABLE), "grcfBSCF = %08x\n", grfBSCF); ok(ulProgress == 1000, "ulProgress = %u\n", ulProgress); ok(!ulProgressMax, "ulProgressMax = %u\n", ulProgressMax); }else if(direct_read) { BYTE buf[14096]; ULONG read; if(!read_report_data && rec_depth == 1) { BOOL reported_all_data = called_ReportData2; CHECK_EXPECT2(ReportData); if(short_read) { ok(grfBSCF == (BSCF_FIRSTDATANOTIFICATION|BSCF_LASTDATANOTIFICATION|BSCF_DATAFULLYAVAILABLE) || grfBSCF == BSCF_FIRSTDATANOTIFICATION, /* < IE8 */ "grcfBSCF = %08x\n", grfBSCF); CHECK_CALLED(Read); /* Set in Continue */ first_data_notif = FALSE; }else if(first_data_notif) { ok(grfBSCF == BSCF_FIRSTDATANOTIFICATION, "grcfBSCF = %08x\n", grfBSCF); first_data_notif = FALSE; }else if(reported_all_data) { ok(grfBSCF == (BSCF_LASTDATANOTIFICATION|BSCF_INTERMEDIATEDATANOTIFICATION), "grcfBSCF = %08x\n", grfBSCF); }else if(!direct_read) { ok(grfBSCF == BSCF_INTERMEDIATEDATANOTIFICATION, "grcfBSCF = %08x\n", grfBSCF); } do { read = 0; if(emulate_prot) SET_EXPECT(Read); else SET_EXPECT(ReportData2); SET_EXPECT(ReportResult); if(!emulate_prot) SET_EXPECT(Switch); hres = IInternetProtocol_Read(binding_test ? binding_protocol : async_protocol, expect_pv = buf, sizeof(buf), &read); ok(hres == E_PENDING || hres == S_FALSE || hres == S_OK, "Read failed: %08x\n", hres); if(hres == S_OK) ok(read, "read == 0\n"); if(reported_all_data) ok(hres == S_FALSE, "Read failed: %08x, expected S_FALSE\n", hres); if(!emulate_prot && hres != E_PENDING) CHECK_NOT_CALLED(Switch); /* otherwise checked in wait_for_switch loop */ if(emulate_prot) CHECK_CALLED(Read); if(!reported_all_data && called_ReportData2) { if(!emulate_prot) CHECK_CALLED(ReportData2); CHECK_CALLED(ReportResult); reported_all_data = TRUE; }else { if(!emulate_prot) CHECK_NOT_CALLED(ReportData2); CHECK_NOT_CALLED(ReportResult); } }while(hres == S_OK); if(hres == S_FALSE) wait_for_switch = FALSE; }else { CHECK_EXPECT(ReportData2); ok(grfBSCF & BSCF_LASTDATANOTIFICATION, "grfBSCF = %08x\n", grfBSCF); read = 0xdeadbeef; if(emulate_prot) SET_EXPECT(Read2); hres = IInternetProtocol_Read(binding_test ? binding_protocol : async_protocol, expect_pv = buf, sizeof(buf), &read); if(emulate_prot) CHECK_CALLED(Read2); ok(hres == S_FALSE, "Read returned: %08x, expected E_FALSE\n", hres); ok(!read, "read = %d\n", read); } }else if(!binding_test && (tested_protocol == HTTP_TEST || tested_protocol == HTTPS_TEST || tested_protocol == FTP_TEST)) { if(empty_file) CHECK_EXPECT2(ReportData); else if(!(grfBSCF & BSCF_LASTDATANOTIFICATION) || (grfBSCF & BSCF_DATAFULLYAVAILABLE)) CHECK_EXPECT(ReportData); else if (http_post_test) ok(ulProgress == 13, "Read %u bytes instead of 13\n", ulProgress); if(empty_file) { ok(!ulProgress, "ulProgress = %d\n", ulProgress); ok(!ulProgressMax, "ulProgressMax = %d\n", ulProgressMax); }else { ok(ulProgress, "ulProgress == 0\n"); } if(empty_file) { ok(grfBSCF == (BSCF_FIRSTDATANOTIFICATION|BSCF_LASTDATANOTIFICATION), "grcfBSCF = %08x\n", grfBSCF); first_data_notif = FALSE; }else if(first_data_notif) { ok(grfBSCF == BSCF_FIRSTDATANOTIFICATION || grfBSCF == (BSCF_LASTDATANOTIFICATION|BSCF_DATAFULLYAVAILABLE), "grcfBSCF = %08x\n", grfBSCF); first_data_notif = FALSE; } else { ok(grfBSCF == BSCF_INTERMEDIATEDATANOTIFICATION || grfBSCF == (BSCF_LASTDATANOTIFICATION|BSCF_INTERMEDIATEDATANOTIFICATION) || broken(grfBSCF == (BSCF_FIRSTDATANOTIFICATION|BSCF_LASTDATANOTIFICATION)), "grcfBSCF = %08x\n", grfBSCF); } if(!(bindf & BINDF_FROMURLMON) && !(grfBSCF & BSCF_LASTDATANOTIFICATION)) { if(state == STATE_CONNECTING) { state = STATE_DOWNLOADING; if(http_is_first) { CHECK_CALLED(ReportProgress_FINDINGRESOURCE); CHECK_CALLED(ReportProgress_CONNECTING); } CHECK_CALLED(ReportProgress_SENDINGREQUEST); CHECK_CALLED(OnResponse); CHECK_CALLED(ReportProgress_RAWMIMETYPE); } SetEvent(event_complete); } }else if(!read_report_data) { BYTE buf[1000]; ULONG read; HRESULT hres; CHECK_EXPECT(ReportData); if(tested_protocol != BIND_TEST) { do { if(mimefilter_test) SET_EXPECT(MimeFilter_Read); else if(rec_depth > 1) SET_EXPECT(Read2); else SET_EXPECT(Read); hres = IInternetProtocol_Read(binding_protocol, expect_pv=buf, sizeof(buf), &read); if(mimefilter_test) CHECK_CALLED(MimeFilter_Read); else if(rec_depth > 1) CHECK_CALLED(Read2); else CHECK_CALLED(Read); }while(hres == S_OK); } } rec_depth--; return S_OK; } static HRESULT WINAPI ProtocolSink_ReportResult(IInternetProtocolSink *iface, HRESULT hrResult, DWORD dwError, LPCWSTR szResult) { CHECK_EXPECT(ReportResult); if(tested_protocol == FTP_TEST) ok(hrResult == E_PENDING || hrResult == S_OK, "hrResult = %08x, expected E_PENDING or S_OK\n", hrResult); else ok(hrResult == expect_hrResult, "hrResult = %08x, expected: %08x\n", hrResult, expect_hrResult); if(SUCCEEDED(hrResult) || tested_protocol == FTP_TEST || test_abort) ok(dwError == ERROR_SUCCESS, "dwError = %d, expected ERROR_SUCCESS\n", dwError); else ok(dwError != ERROR_SUCCESS || broken(tested_protocol == MK_TEST), /* WinME and NT4 */ "dwError == ERROR_SUCCESS\n"); ok(!szResult, "szResult != NULL\n"); if(direct_read) SET_EXPECT(ReportData); /* checked after main loop */ return S_OK; } static IInternetProtocolSinkVtbl protocol_sink_vtbl = { ProtocolSink_QueryInterface, ProtocolSink_AddRef, ProtocolSink_Release, ProtocolSink_Switch, ProtocolSink_ReportProgress, ProtocolSink_ReportData, ProtocolSink_ReportResult }; static IInternetProtocolSink protocol_sink = { &protocol_sink_vtbl }; static HRESULT WINAPI MimeProtocolSink_QueryInterface(IInternetProtocolSink *iface, REFIID riid, void **ppv) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static ULONG WINAPI MimeProtocolSink_AddRef(IInternetProtocolSink *iface) { return 2; } static ULONG WINAPI MimeProtocolSink_Release(IInternetProtocolSink *iface) { return 1; } static HRESULT WINAPI MimeProtocolSink_Switch(IInternetProtocolSink *iface, PROTOCOLDATA *pProtocolData) { HRESULT hres; CHECK_EXPECT(MimeFilter_Switch); SET_EXPECT(Switch); hres = IInternetProtocolSink_Switch(filtered_sink, pProtocolData); ok(hres == S_OK, "Switch failed: %08x\n", hres); CHECK_CALLED(Switch); return S_OK; } static HRESULT WINAPI MimeProtocolSink_ReportProgress(IInternetProtocolSink *iface, ULONG ulStatusCode, LPCWSTR szStatusText) { switch(ulStatusCode) { case BINDSTATUS_LOADINGMIMEHANDLER: /* * IE9 for some reason (bug?) calls this on mime handler's protocol sink instead of the * main protocol sink. We check ReportProgress_LOADINGMIMEHANDLER both here and in * ProtocolSink_ReportProgress to workaround it. */ CHECK_EXPECT(ReportProgress_LOADINGMIMEHANDLER); ok(!szStatusText, "szStatusText = %s\n", wine_dbgstr_w(szStatusText)); break; default: ok(0, "Unexpected status code %d\n", ulStatusCode); } return S_OK; } static HRESULT WINAPI MimeProtocolSink_ReportData(IInternetProtocolSink *iface, DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax) { DWORD read = 0; BYTE buf[8192]; HRESULT hres; BOOL report_mime = FALSE; CHECK_EXPECT(MimeFilter_ReportData); if(!filter_state && !no_mime) { SET_EXPECT(Read); hres = IInternetProtocol_Read(filtered_protocol, buf, sizeof(buf), &read); if(tested_protocol == HTTP_TEST) ok(hres == S_OK || hres == E_PENDING || hres == S_FALSE, "Read failed: %08x\n", hres); else ok(hres == S_OK, "Read failed: %08x\n", hres); CHECK_CALLED(Read); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); hres = IInternetProtocolSink_ReportProgress(filtered_sink, BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE, text_htmlW); ok(hres == S_OK, "ReportProgress failed: %08x\n", hres); CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); hres = IInternetProtocolSink_ReportProgress(filtered_sink, BINDSTATUS_MIMETYPEAVAILABLE, text_htmlW); ok(hres == S_OK, "ReportProgress failed: %08x\n", hres); CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); /* FIXME: test BINDSTATUS_CACHEFILENAMEAVAILABLE */ } if(no_mime && prot_read<200) { SET_EXPECT(Read); }else if(no_mime && prot_read<300) { report_mime = TRUE; SET_EXPECT(Read); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); SET_EXPECT(ReportData); }else if(!read_report_data) { SET_EXPECT(ReportData); } hres = IInternetProtocolSink_ReportData(filtered_sink, grfBSCF, ulProgress, ulProgressMax); ok(hres == S_OK, "ReportData failed: %08x\n", hres); if(no_mime && prot_read<=200) { CHECK_CALLED(Read); }else if(report_mime) { CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); CHECK_CALLED(ReportData); }else if(!read_report_data) { CHECK_CALLED(ReportData); } if(!filter_state) filter_state = 1; return S_OK; } static HRESULT WINAPI MimeProtocolSink_ReportResult(IInternetProtocolSink *iface, HRESULT hrResult, DWORD dwError, LPCWSTR szResult) { HRESULT hres; CHECK_EXPECT(MimeFilter_ReportResult); ok(hrResult == S_OK, "hrResult = %08x\n", hrResult); ok(dwError == ERROR_SUCCESS, "dwError = %u\n", dwError); ok(!szResult, "szResult = %s\n", wine_dbgstr_w(szResult)); SET_EXPECT(ReportResult); hres = IInternetProtocolSink_ReportResult(filtered_sink, hrResult, dwError, szResult); ok(SUCCEEDED(hres), "ReportResult failed: %08x\n", hres); CHECK_CALLED(ReportResult); return S_OK; } static IInternetProtocolSinkVtbl mime_protocol_sink_vtbl = { MimeProtocolSink_QueryInterface, MimeProtocolSink_AddRef, MimeProtocolSink_Release, MimeProtocolSink_Switch, MimeProtocolSink_ReportProgress, MimeProtocolSink_ReportData, MimeProtocolSink_ReportResult }; static IInternetProtocolSink mime_protocol_sink = { &mime_protocol_sink_vtbl }; static HRESULT QueryInterface(REFIID riid, void **ppv) { static const IID IID_undocumented = {0x58DFC7D0,0x5381,0x43E5,{0x9D,0x72,0x4C,0xDD,0xE4,0xCB,0x0F,0x1A}}; static const IID IID_undocumentedIE10 = {0xc28722e5,0xbc1a,0x4c55,{0xa6,0x8d,0x33,0x21,0x9f,0x69,0x89,0x10}}; *ppv = NULL; if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocolSink, riid)) *ppv = &protocol_sink; if(IsEqualGUID(&IID_IServiceProvider, riid)) *ppv = &service_provider; if(IsEqualGUID(&IID_IUriContainer, riid)) return E_NOINTERFACE; /* TODO */ /* NOTE: IE8 queries for undocumented {58DFC7D0-5381-43E5-9D72-4CDDE4CB0F1A} interface. */ if(IsEqualGUID(&IID_undocumented, riid)) return E_NOINTERFACE; /* NOTE: IE10 queries for undocumented {c28722e5-bc1a-4c55-a68d-33219f698910} interface. */ if(IsEqualGUID(&IID_undocumentedIE10, riid)) return E_NOINTERFACE; if(*ppv) return S_OK; ok(0, "unexpected call %s\n", wine_dbgstr_guid(riid)); return E_NOINTERFACE; } static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetBindInfo, riid)) { *ppv = iface; return S_OK; } return E_NOINTERFACE; } static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface) { return 2; } static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface) { return 1; } static HRESULT WINAPI BindInfo_GetBindInfo(IInternetBindInfo *iface, DWORD *grfBINDF, BINDINFO *pbindinfo) { DWORD cbSize; CHECK_EXPECT(GetBindInfo); ok(grfBINDF != NULL, "grfBINDF == NULL\n"); ok(pbindinfo != NULL, "pbindinfo == NULL\n"); ok(pbindinfo->cbSize == sizeof(BINDINFO), "wrong size of pbindinfo: %d\n", pbindinfo->cbSize); *grfBINDF = bindf; if(binding_test) *grfBINDF |= BINDF_FROMURLMON; cbSize = pbindinfo->cbSize; memset(pbindinfo, 0, cbSize); pbindinfo->cbSize = cbSize; if(http_post_test) { pbindinfo->cbstgmedData = sizeof(post_data)-1; pbindinfo->dwBindVerb = BINDVERB_POST; pbindinfo->stgmedData.tymed = http_post_test; if(http_post_test == TYMED_HGLOBAL) { HGLOBAL data; /* Must be GMEM_FIXED, GMEM_MOVABLE does not work properly */ data = GlobalAlloc(GPTR, sizeof(post_data)); memcpy(data, post_data, sizeof(post_data)); U(pbindinfo->stgmedData).hGlobal = data; }else { IStream *post_stream; HGLOBAL data; HRESULT hres; if(0) { /* Must be GMEM_FIXED, GMEM_MOVABLE does not work properly */ data = GlobalAlloc(GPTR, sizeof(post_data)); memcpy(data, post_data, sizeof(post_data)); U(pbindinfo->stgmedData).hGlobal = data; hres = CreateStreamOnHGlobal(data, FALSE, &post_stream); ok(hres == S_OK, "CreateStreamOnHGlobal failed: %08x\n", hres); U(pbindinfo->stgmedData).pstm =post_stream;/* &Stream; */ } U(pbindinfo->stgmedData).pstm = &Stream; } } return S_OK; } static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface, ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched) { ok(ppwzStr != NULL, "ppwzStr == NULL\n"); ok(pcElFetched != NULL, "pcElFetched == NULL\n"); switch(ulStringType) { case BINDSTRING_ACCEPT_MIMES: CHECK_EXPECT(GetBindString_ACCEPT_MIMES); ok(cEl == 256, "cEl=%d, expected 256\n", cEl); if(pcElFetched) { ok(*pcElFetched == 256, "*pcElFetched=%d, expected 256\n", *pcElFetched); *pcElFetched = 1; } if(ppwzStr) { *ppwzStr = CoTaskMemAlloc(sizeof(acc_mimeW)); memcpy(*ppwzStr, acc_mimeW, sizeof(acc_mimeW)); } return S_OK; case BINDSTRING_USER_AGENT: CHECK_EXPECT(GetBindString_USER_AGENT); ok(cEl == 1, "cEl=%d, expected 1\n", cEl); if(pcElFetched) { ok(*pcElFetched == 0, "*pcElFetch=%d, expected 0\n", *pcElFetched); *pcElFetched = 1; } if(ppwzStr) { *ppwzStr = CoTaskMemAlloc(sizeof(user_agentW)); memcpy(*ppwzStr, user_agentW, sizeof(user_agentW)); } return S_OK; case BINDSTRING_POST_COOKIE: CHECK_EXPECT(GetBindString_POST_COOKIE); ok(cEl == 1, "cEl=%d, expected 1\n", cEl); if(pcElFetched) ok(*pcElFetched == 0, "*pcElFetch=%d, expected 0\n", *pcElFetched); return S_OK; case BINDSTRING_URL: { DWORD size; CHECK_EXPECT(GetBindString_URL); ok(cEl == 1, "cEl=%d, expected 1\n", cEl); ok(*pcElFetched == 0, "*pcElFetch=%d, expected 0\n", *pcElFetched); *pcElFetched = 1; size = (lstrlenW(binding_urls[tested_protocol])+1)*sizeof(WCHAR); *ppwzStr = CoTaskMemAlloc(size); memcpy(*ppwzStr, binding_urls[tested_protocol], size); return S_OK; } case BINDSTRING_ROOTDOC_URL: CHECK_EXPECT(GetBindString_ROOTDOC_URL); ok(cEl == 1, "cEl=%d, expected 1\n", cEl); return E_NOTIMPL; case BINDSTRING_ENTERPRISE_ID: ok(cEl == 1, "cEl=%d, expected 1\n", cEl); return E_NOTIMPL; default: ok(0, "unexpected ulStringType %d\n", ulStringType); } return E_NOTIMPL; } static IInternetBindInfoVtbl bind_info_vtbl = { BindInfo_QueryInterface, BindInfo_AddRef, BindInfo_Release, BindInfo_GetBindInfo, BindInfo_GetBindString }; static IInternetBindInfo bind_info = { &bind_info_vtbl }; static HRESULT WINAPI InternetPriority_QueryInterface(IInternetPriority *iface, REFIID riid, void **ppv) { ok(0, "unexpected call\n"); return E_NOINTERFACE; } static ULONG WINAPI InternetPriority_AddRef(IInternetPriority *iface) { return 2; } static ULONG WINAPI InternetPriority_Release(IInternetPriority *iface) { return 1; } static HRESULT WINAPI InternetPriority_SetPriority(IInternetPriority *iface, LONG nPriority) { CHECK_EXPECT(SetPriority); ok(nPriority == ex_priority, "nPriority=%d\n", nPriority); return S_OK; } static HRESULT WINAPI InternetPriority_GetPriority(IInternetPriority *iface, LONG *pnPriority) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static const IInternetPriorityVtbl InternetPriorityVtbl = { InternetPriority_QueryInterface, InternetPriority_AddRef, InternetPriority_Release, InternetPriority_SetPriority, InternetPriority_GetPriority }; static IInternetPriority InternetPriority = { &InternetPriorityVtbl }; static ULONG WINAPI Protocol_AddRef(IInternetProtocolEx *iface) { return 2; } static ULONG WINAPI Protocol_Release(IInternetProtocolEx *iface) { return 1; } static HRESULT WINAPI Protocol_Abort(IInternetProtocolEx *iface, HRESULT hrReason, DWORD dwOptions) { HRESULT hres; CHECK_EXPECT(Abort); SET_EXPECT(ReportResult); hres = IInternetProtocolSink_ReportResult(binding_sink, S_OK, ERROR_SUCCESS, NULL); ok(hres == S_OK, "ReportResult failed: %08x\n", hres); CHECK_CALLED(ReportResult); return S_OK; } static HRESULT WINAPI Protocol_Suspend(IInternetProtocolEx *iface) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Protocol_Resume(IInternetProtocolEx *iface) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI Protocol_Seek(IInternetProtocolEx *iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI ProtocolEmul_QueryInterface(IInternetProtocolEx *iface, REFIID riid, void **ppv) { static const IID unknown_iid = {0x7daf9908,0x8415,0x4005,{0x95,0xae, 0xbd,0x27,0xf6,0xe3,0xdc,0x00}}; if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocol, riid)) { *ppv = iface; return S_OK; } if(IsEqualGUID(&IID_IInternetProtocolEx, riid)) { if(impl_protex) { *ppv = iface; return S_OK; } *ppv = NULL; return E_NOINTERFACE; } if(IsEqualGUID(&IID_IInternetPriority, riid)) { *ppv = &InternetPriority; return S_OK; } if(IsEqualGUID(&IID_IWinInetInfo, riid)) { CHECK_EXPECT(QueryInterface_IWinInetInfo); *ppv = NULL; return E_NOINTERFACE; } if(IsEqualGUID(&IID_IWinInetHttpInfo, riid)) { CHECK_EXPECT(QueryInterface_IWinInetHttpInfo); *ppv = NULL; return E_NOINTERFACE; } if(!IsEqualGUID(riid, &unknown_iid)) /* IE10 */ ok(0, "unexpected riid %s\n", wine_dbgstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; } static DWORD WINAPI thread_proc(PVOID arg) { HRESULT hres; memset(&protocoldata, -1, sizeof(protocoldata)); prot_state = 0; SET_EXPECT(ReportProgress_FINDINGRESOURCE); hres = IInternetProtocolSink_ReportProgress(binding_sink, BINDSTATUS_FINDINGRESOURCE, hostW); CHECK_CALLED(ReportProgress_FINDINGRESOURCE); ok(hres == S_OK, "ReportProgress failed: %08x\n", hres); SET_EXPECT(ReportProgress_CONNECTING); hres = IInternetProtocolSink_ReportProgress(binding_sink, BINDSTATUS_CONNECTING, winehq_ipW); CHECK_CALLED(ReportProgress_CONNECTING); ok(hres == S_OK, "ReportProgress failed: %08x\n", hres); SET_EXPECT(ReportProgress_SENDINGREQUEST); hres = IInternetProtocolSink_ReportProgress(binding_sink, BINDSTATUS_SENDINGREQUEST, NULL); CHECK_CALLED(ReportProgress_SENDINGREQUEST); ok(hres == S_OK, "ReportProgress failed: %08x\n", hres); prot_state = 1; SET_EXPECT(Switch); hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata); CHECK_CALLED(Switch); ok(hres == S_OK, "Switch failed: %08x\n", hres); if(!short_read) { prot_state = 2; if(mimefilter_test) SET_EXPECT(MimeFilter_Switch); else SET_EXPECT(Switch); hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata); ok(hres == S_OK, "Switch failed: %08x\n", hres); if(mimefilter_test) CHECK_CALLED(MimeFilter_Switch); else CHECK_CALLED(Switch); if(test_abort) { SetEvent(event_complete); return 0; } prot_state = 2; if(mimefilter_test) SET_EXPECT(MimeFilter_Switch); else SET_EXPECT(Switch); hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata); ok(hres == S_OK, "Switch failed: %08x\n", hres); if(mimefilter_test) CHECK_CALLED(MimeFilter_Switch); else CHECK_CALLED(Switch); prot_state = 3; if(mimefilter_test) SET_EXPECT(MimeFilter_Switch); else SET_EXPECT(Switch); hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata); ok(hres == S_OK, "Switch failed: %08x\n", hres); if(mimefilter_test) CHECK_CALLED(MimeFilter_Switch); else CHECK_CALLED(Switch); } SetEvent(event_complete); return 0; } static void protocol_start(IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo, DWORD pi) { BINDINFO bindinfo, exp_bindinfo; DWORD cbindf = 0; HRESULT hres; ok(pOIProtSink != NULL, "pOIProtSink == NULL\n"); ok(pOIBindInfo != NULL, "pOIBindInfo == NULL\n"); ok(pOIProtSink != &protocol_sink, "unexpected pOIProtSink\n"); ok(pOIBindInfo != &bind_info, "unexpected pOIBindInfo\n"); ok(!pi, "pi = %x\n", pi); if(binding_test) ok(pOIProtSink == binding_sink, "pOIProtSink != binding_sink\n"); memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); memcpy(&exp_bindinfo, &bindinfo, sizeof(bindinfo)); SET_EXPECT(GetBindInfo); hres = IInternetBindInfo_GetBindInfo(pOIBindInfo, &cbindf, &bindinfo); ok(hres == S_OK, "GetBindInfo failed: %08x\n", hres); CHECK_CALLED(GetBindInfo); ok(cbindf == (bindf|BINDF_FROMURLMON), "bindf = %x, expected %x\n", cbindf, (bindf|BINDF_FROMURLMON)); ok(!memcmp(&exp_bindinfo, &bindinfo, sizeof(bindinfo)), "unexpected bindinfo\n"); pReleaseBindInfo(&bindinfo); SET_EXPECT(ReportProgress_SENDINGREQUEST); hres = IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_SENDINGREQUEST, emptyW); ok(hres == S_OK, "ReportProgress(BINDSTATUS_SENDINGREQUEST) failed: %08x\n", hres); CHECK_CALLED(ReportProgress_SENDINGREQUEST); if(tested_protocol == HTTP_TEST || tested_protocol == HTTPS_TEST) { IServiceProvider *service_provider; IHttpNegotiate *http_negotiate; IHttpNegotiate2 *http_negotiate2; LPWSTR ua = (LPWSTR)0xdeadbeef, accept_mimes[256]; LPWSTR additional_headers = NULL; BYTE sec_id[100]; DWORD fetched = 0, size = 100; DWORD tid; SET_EXPECT(GetBindString_USER_AGENT); hres = IInternetBindInfo_GetBindString(pOIBindInfo, BINDSTRING_USER_AGENT, &ua, 1, &fetched); CHECK_CALLED(GetBindString_USER_AGENT); ok(hres == S_OK, "GetBindString(BINDSTRING_USER_AGETNT) failed: %08x\n", hres); ok(fetched == 1, "fetched = %d, expected 254\n", fetched); ok(ua != NULL, "ua = %p\n", ua); ok(!lstrcmpW(ua, user_agentW), "unexpected user agent %s\n", wine_dbgstr_w(ua)); CoTaskMemFree(ua); fetched = 256; SET_EXPECT(GetBindString_ACCEPT_MIMES); hres = IInternetBindInfo_GetBindString(pOIBindInfo, BINDSTRING_ACCEPT_MIMES, accept_mimes, 256, &fetched); CHECK_CALLED(GetBindString_ACCEPT_MIMES); ok(hres == S_OK, "GetBindString(BINDSTRING_ACCEPT_MIMES) failed: %08x\n", hres); ok(fetched == 1, "fetched = %d, expected 1\n", fetched); ok(!lstrcmpW(acc_mimeW, accept_mimes[0]), "unexpected mimes %s\n", wine_dbgstr_w(accept_mimes[0])); CoTaskMemFree(accept_mimes[0]); hres = IInternetBindInfo_QueryInterface(pOIBindInfo, &IID_IServiceProvider, (void**)&service_provider); ok(hres == S_OK, "QueryInterface failed: %08x\n", hres); SET_EXPECT(QueryService_HttpNegotiate); hres = IServiceProvider_QueryService(service_provider, &IID_IHttpNegotiate, &IID_IHttpNegotiate, (void**)&http_negotiate); CHECK_CALLED(QueryService_HttpNegotiate); ok(hres == S_OK, "QueryService failed: %08x\n", hres); SET_EXPECT(BeginningTransaction); hres = IHttpNegotiate_BeginningTransaction(http_negotiate, binding_urls[tested_protocol], NULL, 0, &additional_headers); CHECK_CALLED(BeginningTransaction); IHttpNegotiate_Release(http_negotiate); ok(hres == S_OK, "BeginningTransction failed: %08x\n", hres); ok(additional_headers == NULL, "additional_headers=%p\n", additional_headers); SET_EXPECT(QueryService_HttpNegotiate); hres = IServiceProvider_QueryService(service_provider, &IID_IHttpNegotiate2, &IID_IHttpNegotiate2, (void**)&http_negotiate2); CHECK_CALLED(QueryService_HttpNegotiate); ok(hres == S_OK, "QueryService failed: %08x\n", hres); size = 512; SET_EXPECT(GetRootSecurityId); hres = IHttpNegotiate2_GetRootSecurityId(http_negotiate2, sec_id, &size, 0); CHECK_CALLED(GetRootSecurityId); IHttpNegotiate2_Release(http_negotiate2); ok(hres == E_FAIL, "GetRootSecurityId failed: %08x, expected E_FAIL\n", hres); ok(size == 13, "size=%d\n", size); IServiceProvider_Release(service_provider); CreateThread(NULL, 0, thread_proc, NULL, 0, &tid); return; } SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); hres = IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_CACHEFILENAMEAVAILABLE, expect_wsz = emptyW); ok(hres == S_OK, "ReportProgress(BINDSTATUS_CACHEFILENAMEAVAILABLE) failed: %08x\n", hres); CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE); if(mimefilter_test) { SET_EXPECT(MimeFilter_CreateInstance); SET_EXPECT(MimeFilter_Start); SET_EXPECT(ReportProgress_LOADINGMIMEHANDLER); } SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); hres = IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE, mimefilter_test ? pjpegW : (expect_wsz = text_htmlW)); ok(hres == S_OK, "ReportProgress(BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE) failed: %08x\n", hres); if(mimefilter_test) { CHECK_CALLED(MimeFilter_CreateInstance); CHECK_CALLED(MimeFilter_Start); CHECK_CALLED(ReportProgress_LOADINGMIMEHANDLER); CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); }else { CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); } if(mimefilter_test) SET_EXPECT(MimeFilter_ReportData); else SET_EXPECT(ReportData); hres = IInternetProtocolSink_ReportData(pOIProtSink, BSCF_FIRSTDATANOTIFICATION | (tested_protocol == ITS_TEST ? BSCF_DATAFULLYAVAILABLE : BSCF_LASTDATANOTIFICATION), 13, 13); ok(hres == S_OK, "ReportData failed: %08x\n", hres); if(mimefilter_test) CHECK_CALLED(MimeFilter_ReportData); else CHECK_CALLED(ReportData); if(tested_protocol == ITS_TEST) { SET_EXPECT(ReportData); hres = IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_BEGINDOWNLOADDATA, NULL); ok(hres == S_OK, "ReportProgress(BINDSTATUS_BEGINDOWNLOADDATA) failed: %08x\n", hres); CHECK_CALLED(ReportData); } if(tested_protocol == BIND_TEST) { hres = IInternetProtocol_Terminate(binding_protocol, 0); ok(hres == E_FAIL, "Termiante failed: %08x\n", hres); } if(mimefilter_test) SET_EXPECT(MimeFilter_ReportResult); else SET_EXPECT(ReportResult); hres = IInternetProtocolSink_ReportResult(pOIProtSink, S_OK, 0, NULL); ok(hres == S_OK, "ReportResult failed: %08x\n", hres); if(mimefilter_test) CHECK_CALLED(MimeFilter_ReportResult); else CHECK_CALLED(ReportResult); } static HRESULT WINAPI ProtocolEmul_Start(IInternetProtocolEx *iface, LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) { CHECK_EXPECT(Start); ok(!dwReserved, "dwReserved = %lx\n", dwReserved); protocol_start(pOIProtSink, pOIBindInfo, grfPI); return S_OK; } static HRESULT WINAPI ProtocolEmul_Continue(IInternetProtocolEx *iface, PROTOCOLDATA *pProtocolData) { DWORD bscf = 0, pr; HRESULT hres; CHECK_EXPECT(Continue); ok(pProtocolData != NULL, "pProtocolData == NULL\n"); if(!pProtocolData || tested_protocol == BIND_TEST) return S_OK; if(binding_test) { ok(pProtocolData != &protocoldata, "pProtocolData == &protocoldata\n"); ok(pProtocolData->grfFlags == protocoldata.grfFlags, "grfFlags wrong %x/%x\n", pProtocolData->grfFlags, protocoldata.grfFlags ); ok(pProtocolData->dwState == protocoldata.dwState, "dwState wrong %x/%x\n", pProtocolData->dwState, protocoldata.dwState ); ok(pProtocolData->pData == protocoldata.pData, "pData wrong %p/%p\n", pProtocolData->pData, protocoldata.pData ); ok(pProtocolData->cbData == protocoldata.cbData, "cbData wrong %x/%x\n", pProtocolData->cbData, protocoldata.cbData ); } switch(prot_state) { case 1: { IServiceProvider *service_provider; IHttpNegotiate *http_negotiate; static const WCHAR header[] = {'?',0}; hres = IInternetProtocolSink_QueryInterface(binding_sink, &IID_IServiceProvider, (void**)&service_provider); ok(hres == S_OK, "Could not get IServiceProvicder\n"); SET_EXPECT(QueryService_HttpNegotiate); hres = IServiceProvider_QueryService(service_provider, &IID_IHttpNegotiate, &IID_IHttpNegotiate, (void**)&http_negotiate); IServiceProvider_Release(service_provider); CHECK_CALLED(QueryService_HttpNegotiate); ok(hres == S_OK, "Could not get IHttpNegotiate\n"); SET_EXPECT(OnResponse); hres = IHttpNegotiate_OnResponse(http_negotiate, 200, header, NULL, NULL); IHttpNegotiate_Release(http_negotiate); CHECK_CALLED(OnResponse); IHttpNegotiate_Release(http_negotiate); ok(hres == S_OK, "OnResponse failed: %08x\n", hres); if(mimefilter_test) { SET_EXPECT(MimeFilter_CreateInstance); SET_EXPECT(MimeFilter_Start); SET_EXPECT(ReportProgress_LOADINGMIMEHANDLER); }else if(!(pi & PI_MIMEVERIFICATION)) { SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); } hres = IInternetProtocolSink_ReportProgress(binding_sink, BINDSTATUS_MIMETYPEAVAILABLE, mimefilter_test ? pjpegW : text_htmlW); if(mimefilter_test) { CHECK_CALLED(MimeFilter_CreateInstance); CHECK_CALLED(MimeFilter_Start); CHECK_CALLED(ReportProgress_LOADINGMIMEHANDLER); }else if(!(pi & PI_MIMEVERIFICATION)) { CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); } ok(hres == S_OK, "ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE) failed: %08x\n", hres); bscf |= BSCF_FIRSTDATANOTIFICATION; break; } case 2: case 3: bscf = BSCF_INTERMEDIATEDATANOTIFICATION; break; } pr = prot_read; if(mimefilter_test) SET_EXPECT(MimeFilter_ReportData); if((!mimefilter_test || no_mime) && (pi & PI_MIMEVERIFICATION)) { if(pr < 200) SET_EXPECT(Read); /* checked in ReportData for short_read */ if(pr == 200) { if(!mimefilter_test) SET_EXPECT(Read); /* checked in BINDSTATUS_MIMETYPEAVAILABLE or ReportData */ SET_EXPECT(GetBindInfo); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); } if(pr >= 200) SET_EXPECT(ReportData); }else { SET_EXPECT(ReportData); } hres = IInternetProtocolSink_ReportData(binding_sink, bscf, pr, 400); ok(hres == S_OK, "ReportData failed: %08x\n", hres); if(mimefilter_test) { SET_EXPECT(MimeFilter_ReportData); }else if(pi & PI_MIMEVERIFICATION) { if(!short_read && pr < 200) CHECK_CALLED(Read); if(pr == 200) { CLEAR_CALLED(GetBindInfo); /* IE9 */ CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); } }else { CHECK_CALLED(ReportData); } if(prot_state == 3) prot_state = 4; return S_OK; } static HRESULT WINAPI ProtocolEmul_Terminate(IInternetProtocolEx *iface, DWORD dwOptions) { CHECK_EXPECT(Terminate); ok(!dwOptions, "dwOptions=%d\n", dwOptions); return S_OK; } static HRESULT WINAPI ProtocolEmul_Read(IInternetProtocolEx *iface, void *pv, ULONG cb, ULONG *pcbRead) { if(read_report_data) CHECK_EXPECT2(Read2); if(mimefilter_test || short_read) { if(!read_report_data) CHECK_EXPECT2(Read); }else if((pi & PI_MIMEVERIFICATION)) { if(!read_report_data) CHECK_EXPECT2(Read); if(prot_read < 300) { ok(pv != expect_pv, "pv == expect_pv\n"); if(prot_read < 300) ok(cb == 2048-prot_read, "cb=%d\n", cb); else ok(cb == 700, "cb=%d\n", cb); }else { ok(expect_pv <= pv && (BYTE*)pv < (BYTE*)expect_pv + cb, "pv != expect_pv\n"); } }else { if(!read_report_data) CHECK_EXPECT(Read); ok(pv == expect_pv, "pv != expect_pv\n"); ok(cb == 1000, "cb=%d\n", cb); ok(!*pcbRead, "*pcbRead = %d\n", *pcbRead); } ok(pcbRead != NULL, "pcbRead == NULL\n"); if(prot_state == 3 || (short_read && prot_state != 4)) { HRESULT hres; prot_state = 4; if(short_read) { SET_EXPECT(Read2); /* checked in BINDSTATUS_MIMETYPEAVAILABLE */ SET_EXPECT(GetBindInfo); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); } if(mimefilter_test) SET_EXPECT(MimeFilter_ReportData); else if(direct_read) SET_EXPECT(ReportData2); read_report_data++; hres = IInternetProtocolSink_ReportData(binding_sink, BSCF_LASTDATANOTIFICATION|BSCF_INTERMEDIATEDATANOTIFICATION, 0, 0); read_report_data--; ok(hres == S_OK, "ReportData failed: %08x\n", hres); if(short_read) { CLEAR_CALLED(GetBindInfo); /* IE9 */ CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); } if(mimefilter_test) CHECK_CALLED(MimeFilter_ReportData); else if(direct_read) CHECK_CALLED(ReportData2); if(mimefilter_test) SET_EXPECT(MimeFilter_ReportResult); else SET_EXPECT(ReportResult); hres = IInternetProtocolSink_ReportResult(binding_sink, S_OK, ERROR_SUCCESS, NULL); ok(hres == S_OK, "ReportResult failed: %08x\n", hres); if(mimefilter_test) CHECK_CALLED(MimeFilter_ReportResult); else CHECK_CALLED(ReportResult); if(cb > 100) cb = 100; memset(pv, 'x', cb); if(cb>6) memcpy(pv, "gif87a", 6); prot_read += *pcbRead = cb; return S_OK; } if(prot_state == 4) { *pcbRead = 0; return S_FALSE; } if((async_read_pending = !async_read_pending)) { *pcbRead = 0; return tested_protocol == HTTP_TEST || tested_protocol == HTTPS_TEST ? E_PENDING : S_FALSE; } if(cb > 100) cb = 100; memset(pv, 'x', cb); if(cb>6) memcpy(pv, "gif87a", 6); prot_read += *pcbRead = cb; return S_OK; } static HRESULT WINAPI ProtocolEmul_LockRequest(IInternetProtocolEx *iface, DWORD dwOptions) { CHECK_EXPECT(LockRequest); ok(dwOptions == 0, "dwOptions=%x\n", dwOptions); return S_OK; } static HRESULT WINAPI ProtocolEmul_UnlockRequest(IInternetProtocolEx *iface) { CHECK_EXPECT(UnlockRequest); return S_OK; } static HRESULT WINAPI ProtocolEmul_StartEx(IInternetProtocolEx *iface, IUri *pUri, IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE *dwReserved) { CHECK_EXPECT(StartEx); ok(!dwReserved, "dwReserved = %p\n", dwReserved); protocol_start(pOIProtSink, pOIBindInfo, grfPI); return S_OK; } static const IInternetProtocolExVtbl ProtocolVtbl = { ProtocolEmul_QueryInterface, Protocol_AddRef, Protocol_Release, ProtocolEmul_Start, ProtocolEmul_Continue, Protocol_Abort, ProtocolEmul_Terminate, Protocol_Suspend, Protocol_Resume, ProtocolEmul_Read, Protocol_Seek, ProtocolEmul_LockRequest, ProtocolEmul_UnlockRequest, ProtocolEmul_StartEx }; static IInternetProtocolEx Protocol = { &ProtocolVtbl }; static HRESULT WINAPI MimeProtocol_QueryInterface(IInternetProtocolEx *iface, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocol, riid)) { *ppv = iface; return S_OK; } if(IsEqualGUID(&IID_IInternetProtocolSink, riid)) { *ppv = &mime_protocol_sink; return S_OK; } ok(0, "unexpected riid %s\n", wine_dbgstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; } static HRESULT WINAPI MimeProtocol_Start(IInternetProtocolEx *iface, LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) { PROTOCOLFILTERDATA *data; LPOLESTR url_str = NULL; DWORD fetched = 0; BINDINFO bindinfo; DWORD cbindf = 0; HRESULT hres; CHECK_EXPECT(MimeFilter_Start); ok(!lstrcmpW(szUrl, pjpegW), "wrong url %s\n", wine_dbgstr_w(szUrl)); ok(grfPI == (PI_FILTER_MODE|PI_FORCE_ASYNC), "grfPI=%x, expected PI_FILTER_MODE|PI_FORCE_ASYNC\n", grfPI); ok(dwReserved, "dwReserved == 0\n"); ok(pOIProtSink != NULL, "pOIProtSink == NULL\n"); ok(pOIBindInfo != NULL, "pOIBindInfo == NULL\n"); if(binding_test) { ok(pOIProtSink != binding_sink, "pOIProtSink == protocol_sink\n"); ok(pOIBindInfo == prot_bind_info, "pOIBindInfo != bind_info\n"); }else { ok(pOIProtSink == &protocol_sink, "pOIProtSink != protocol_sink\n"); ok(pOIBindInfo == &bind_info, "pOIBindInfo != bind_info\n"); } data = (void*)dwReserved; ok(data->cbSize == sizeof(*data), "data->cbSize = %d\n", data->cbSize); ok(!data->pProtocolSink, "data->pProtocolSink != NULL\n"); ok(data->pProtocol != NULL, "data->pProtocol == NULL\n"); ok(!data->pUnk, "data->pUnk != NULL\n"); ok(!data->dwFilterFlags, "data->dwProtocolFlags = %x\n", data->dwFilterFlags); if(binding_test) { IInternetProtocolSink *prot_sink; IInternetProtocol_QueryInterface(data->pProtocol, &IID_IInternetProtocolSink, (void**)&prot_sink); ok(prot_sink == pOIProtSink, "QI(data->pProtocol, IID_IInternetProtocolSink) != pOIProtSink\n"); IInternetProtocolSink_Release(prot_sink); ok(data->pProtocol != binding_protocol, "data->pProtocol == binding_protocol\n"); filtered_protocol = data->pProtocol; IInternetProtocol_AddRef(filtered_protocol); }else { IInternetProtocol *prot; IInternetProtocol_QueryInterface(data->pProtocol, &IID_IInternetProtocol, (void**)&prot); ok(prot == async_protocol, "QI(data->pProtocol, IID_IInternetProtocol) != async_protocol\n"); IInternetProtocol_Release(prot); ok(data->pProtocol != async_protocol, "data->pProtocol == async_protocol\n"); } filtered_sink = pOIProtSink; SET_EXPECT(ReportProgress_DECODING); hres = IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_DECODING, pjpegW); ok(hres == S_OK, "ReportProgress(BINDSTATUS_DECODING) failed: %08x\n", hres); CHECK_CALLED(ReportProgress_DECODING); SET_EXPECT(GetBindInfo); memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); hres = IInternetBindInfo_GetBindInfo(pOIBindInfo, &cbindf, &bindinfo); ok(hres == S_OK, "GetBindInfo failed: %08x\n", hres); ok(cbindf == (bindf|BINDF_FROMURLMON), "cbindf = %x, expected %x\n", cbindf, bindf); CHECK_CALLED(GetBindInfo); SET_EXPECT(GetBindString_URL); hres = IInternetBindInfo_GetBindString(pOIBindInfo, BINDSTRING_URL, &url_str, 1, &fetched); ok(hres == S_OK, "GetBindString(BINDSTRING_URL) failed: %08x\n", hres); ok(fetched == 1, "fetched = %d\n", fetched); ok(!lstrcmpW(url_str, binding_urls[tested_protocol]), "wrong url_str %s\n", wine_dbgstr_w(url_str)); CoTaskMemFree(url_str); CHECK_CALLED(GetBindString_URL); return S_OK; } static HRESULT WINAPI Protocol_Continue(IInternetProtocolEx *iface, PROTOCOLDATA *pProtocolData) { CHECK_EXPECT(MimeFilter_Continue); return E_NOTIMPL; } static HRESULT WINAPI MimeProtocol_Terminate(IInternetProtocolEx *iface, DWORD dwOptions) { HRESULT hres; CHECK_EXPECT(MimeFilter_Terminate); ok(!dwOptions, "dwOptions = %x\n", dwOptions); SET_EXPECT(Terminate); hres = IInternetProtocol_Terminate(filtered_protocol, dwOptions); ok(hres == S_OK, "Terminate failed: %08x\n", hres); CHECK_CALLED(Terminate); return S_OK; } static HRESULT WINAPI MimeProtocol_Read(IInternetProtocolEx *iface, void *pv, ULONG cb, ULONG *pcbRead) { BYTE buf[2096]; DWORD read = 0; HRESULT hres; CHECK_EXPECT(MimeFilter_Read); ok(pv != NULL, "pv == NULL\n"); ok(cb != 0, "cb == 0\n"); ok(pcbRead != NULL, "pcbRead == NULL\n"); if(read_report_data) SET_EXPECT(Read2); else SET_EXPECT(Read); hres = IInternetProtocol_Read(filtered_protocol, buf, sizeof(buf), &read); ok(hres == S_OK || hres == S_FALSE || hres == E_PENDING, "Read failed: %08x\n", hres); if(read_report_data) CHECK_CALLED(Read2); else CHECK_CALLED(Read); if(pcbRead) { ok(*pcbRead == 0, "*pcbRead=%d, expected 0\n", *pcbRead); *pcbRead = read; } memset(pv, 'x', read); return hres; } static HRESULT WINAPI MimeProtocol_LockRequest(IInternetProtocolEx *iface, DWORD dwOptions) { HRESULT hres; CHECK_EXPECT(MimeFilter_LockRequest); ok(!dwOptions, "dwOptions = %x\n", dwOptions); SET_EXPECT(LockRequest); hres = IInternetProtocol_LockRequest(filtered_protocol, dwOptions); ok(hres == S_OK, "LockRequest failed: %08x\n", hres); CHECK_CALLED(LockRequest); return S_OK; } static HRESULT WINAPI MimeProtocol_UnlockRequest(IInternetProtocolEx *iface) { HRESULT hres; CHECK_EXPECT(MimeFilter_UnlockRequest); SET_EXPECT(UnlockRequest); hres = IInternetProtocol_UnlockRequest(filtered_protocol); ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres); CHECK_CALLED(UnlockRequest); return S_OK; } static const IInternetProtocolExVtbl MimeProtocolVtbl = { MimeProtocol_QueryInterface, Protocol_AddRef, Protocol_Release, MimeProtocol_Start, Protocol_Continue, Protocol_Abort, MimeProtocol_Terminate, Protocol_Suspend, Protocol_Resume, MimeProtocol_Read, Protocol_Seek, MimeProtocol_LockRequest, MimeProtocol_UnlockRequest }; static IInternetProtocolEx MimeProtocol = { &MimeProtocolVtbl }; static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv) { ok(0, "unexpected call\n"); return E_NOINTERFACE; } static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface) { return 2; } static ULONG WINAPI ClassFactory_Release(IClassFactory *iface) { return 1; } static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pOuter, REFIID riid, void **ppv) { CHECK_EXPECT(CreateInstance); ok(pOuter == (IUnknown*)prot_bind_info, "pOuter != protocol_unk\n"); ok(IsEqualGUID(&IID_IUnknown, riid), "unexpected riid %s\n", wine_dbgstr_guid(riid)); ok(ppv != NULL, "ppv == NULL\n"); *ppv = &Protocol; return S_OK; } static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock) { ok(0, "unexpected call\n"); return S_OK; } static const IClassFactoryVtbl ClassFactoryVtbl = { ClassFactory_QueryInterface, ClassFactory_AddRef, ClassFactory_Release, ClassFactory_CreateInstance, ClassFactory_LockServer }; static IClassFactory ClassFactory = { &ClassFactoryVtbl }; static HRESULT WINAPI MimeFilter_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID riid, void **ppv) { CHECK_EXPECT(MimeFilter_CreateInstance); ok(!outer, "outer = %p\n", outer); ok(IsEqualGUID(&IID_IInternetProtocol, riid), "unexpected riid %s\n", wine_dbgstr_guid(riid)); *ppv = &MimeProtocol; return S_OK; } static const IClassFactoryVtbl MimeFilterCFVtbl = { ClassFactory_QueryInterface, ClassFactory_AddRef, ClassFactory_Release, MimeFilter_CreateInstance, ClassFactory_LockServer }; static IClassFactory mimefilter_cf = { &MimeFilterCFVtbl }; #define TEST_BINDING 0x0001 #define TEST_FILTER 0x0002 #define TEST_FIRST_HTTP 0x0004 #define TEST_DIRECT_READ 0x0008 #define TEST_POST 0x0010 #define TEST_EMULATEPROT 0x0020 #define TEST_SHORT_READ 0x0040 #define TEST_REDIRECT 0x0080 #define TEST_ABORT 0x0100 #define TEST_ASYNCREQ 0x0200 #define TEST_USEIURI 0x0400 #define TEST_IMPLPROTEX 0x0800 #define TEST_EMPTY 0x1000 #define TEST_NOMIME 0x2000 #define TEST_FROMCACHE 0x4000 static void register_filter(BOOL do_register) { IInternetSession *session; HRESULT hres; hres = pCoInternetGetSession(0, &session, 0); ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres); if(do_register) { hres = IInternetSession_RegisterMimeFilter(session, &mimefilter_cf, &IID_IInternetProtocol, pjpegW); ok(hres == S_OK, "RegisterMimeFilter failed: %08x\n", hres); hres = IInternetSession_RegisterMimeFilter(session, &mimefilter_cf, &IID_IInternetProtocol, gifW); ok(hres == S_OK, "RegisterMimeFilter failed: %08x\n", hres); }else { hres = IInternetSession_UnregisterMimeFilter(session, &mimefilter_cf, pjpegW); ok(hres == S_OK, "RegisterMimeFilter failed: %08x\n", hres); hres = IInternetSession_UnregisterMimeFilter(session, &mimefilter_cf, gifW); ok(hres == S_OK, "RegisterMimeFilter failed: %08x\n", hres); } IInternetSession_Release(session); } static void init_test(int prot, DWORD flags) { tested_protocol = prot; binding_test = (flags & TEST_BINDING) != 0; first_data_notif = TRUE; prot_read = 0; prot_state = 0; async_read_pending = TRUE; mimefilter_test = (flags & TEST_FILTER) != 0; no_mime = (flags & TEST_NOMIME) != 0; filter_state = 0; post_stream_read = 0; ResetEvent(event_complete); ResetEvent(event_complete2); ResetEvent(event_continue); ResetEvent(event_continue_done); async_protocol = binding_protocol = filtered_protocol = NULL; filtered_sink = NULL; http_is_first = (flags & TEST_FIRST_HTTP) != 0; first_data_notif = TRUE; state = STATE_CONNECTING; test_async_req = (flags & TEST_ASYNCREQ) != 0; direct_read = (flags & TEST_DIRECT_READ) != 0; emulate_prot = (flags & TEST_EMULATEPROT) != 0; wait_for_switch = TRUE; short_read = (flags & TEST_SHORT_READ) != 0; http_post_test = TYMED_NULL; test_redirect = (flags & TEST_REDIRECT) != 0; test_abort = (flags & TEST_ABORT) != 0; impl_protex = (flags & TEST_IMPLPROTEX) != 0; empty_file = (flags & TEST_EMPTY) != 0; bind_from_cache = (flags & TEST_FROMCACHE) != 0; file_with_hash = FALSE; register_filter(mimefilter_test); } static void test_priority(IInternetProtocol *protocol) { IInternetPriority *priority; LONG pr; HRESULT hres; hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority); ok(hres == S_OK, "QueryInterface(IID_IInternetPriority) failed: %08x\n", hres); if(FAILED(hres)) return; hres = IInternetPriority_GetPriority(priority, &pr); ok(hres == S_OK, "GetPriority failed: %08x\n", hres); ok(pr == 0, "pr=%d, expected 0\n", pr); hres = IInternetPriority_SetPriority(priority, 1); ok(hres == S_OK, "SetPriority failed: %08x\n", hres); hres = IInternetPriority_GetPriority(priority, &pr); ok(hres == S_OK, "GetPriority failed: %08x\n", hres); ok(pr == 1, "pr=%d, expected 1\n", pr); IInternetPriority_Release(priority); } static void test_early_abort(const CLSID *clsid) { IInternetProtocol *protocol; HRESULT hres; hres = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER, &IID_IInternetProtocol, (void**)&protocol); ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres); hres = IInternetProtocol_Abort(protocol, E_ABORT, 0); ok(hres == S_OK, "Abort failed: %08x\n", hres); hres = IInternetProtocol_Abort(protocol, E_FAIL, 0); ok(hres == S_OK, "Abort failed: %08x\n", hres); IInternetProtocol_Release(protocol); } static BOOL file_protocol_start(IInternetProtocol *protocol, LPCWSTR url, IInternetProtocolEx *protocolex, IUri *uri, BOOL is_first) { HRESULT hres; SET_EXPECT(GetBindInfo); if(!(bindf & BINDF_FROMURLMON)) SET_EXPECT(ReportProgress_DIRECTBIND); if(is_first) { SET_EXPECT(ReportProgress_SENDINGREQUEST); SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); if(bindf & BINDF_FROMURLMON) SET_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE); else SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); } SET_EXPECT(ReportData); if(is_first) SET_EXPECT(ReportResult); expect_hrResult = S_OK; if(protocolex) { hres = IInternetProtocolEx_StartEx(protocolex, uri, &protocol_sink, &bind_info, 0, 0); ok(hres == S_OK, "StartEx failed: %08x\n", hres); }else { hres = IInternetProtocol_Start(protocol, url, &protocol_sink, &bind_info, 0, 0); if(hres == INET_E_RESOURCE_NOT_FOUND) { win_skip("Start failed\n"); return FALSE; } ok(hres == S_OK, "Start failed: %08x\n", hres); } CHECK_CALLED(GetBindInfo); if(!(bindf & BINDF_FROMURLMON)) CLEAR_CALLED(ReportProgress_DIRECTBIND); /* Not called by IE10 */ if(is_first) { CHECK_CALLED(ReportProgress_SENDINGREQUEST); CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE); if(bindf & BINDF_FROMURLMON) CHECK_CALLED(ReportProgress_VERIFIEDMIMETYPEAVAILABLE); else CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); } CHECK_CALLED(ReportData); if(is_first) CHECK_CALLED(ReportResult); return TRUE; } static void test_file_protocol_url(LPCWSTR url) { IInternetProtocolInfo *protocol_info; IUnknown *unk; IClassFactory *factory; IInternetProtocol *protocol; BYTE buf[512]; ULONG cb; HRESULT hres; hres = CoGetClassObject(&CLSID_FileProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk); ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres); if(FAILED(hres)) return; hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info); ok(hres == E_NOINTERFACE, "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres); hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory); ok(hres == S_OK, "Could not get IClassFactory interface\n"); IUnknown_Release(unk); if(FAILED(hres)) return; hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol); ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres); if(SUCCEEDED(hres)) { if(file_protocol_start(protocol, url, NULL, NULL, TRUE)) { hres = IInternetProtocol_Read(protocol, buf, 2, &cb); ok(hres == S_OK, "Read failed: %08x\n", hres); ok(cb == 2, "cb=%u expected 2\n", cb); buf[2] = 0; ok(!memcmp(buf, file_with_hash ? "XX" : "<H", 2), "Unexpected data %s\n", buf); hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb); ok(hres == S_FALSE, "Read failed: %08x\n", hres); hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb); ok(hres == S_FALSE, "Read failed: %08x expected S_FALSE\n", hres); ok(cb == 0, "cb=%u expected 0\n", cb); hres = IInternetProtocol_UnlockRequest(protocol); ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres); } if(file_protocol_start(protocol, url, NULL, NULL, FALSE)) { hres = IInternetProtocol_Read(protocol, buf, 2, &cb); ok(hres == S_FALSE, "Read failed: %08x\n", hres); hres = IInternetProtocol_LockRequest(protocol, 0); ok(hres == S_OK, "LockRequest failed: %08x\n", hres); hres = IInternetProtocol_UnlockRequest(protocol); ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres); } IInternetProtocol_Release(protocol); } hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol); ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres); if(SUCCEEDED(hres)) { if(file_protocol_start(protocol, url, NULL, NULL, TRUE)) { hres = IInternetProtocol_LockRequest(protocol, 0); ok(hres == S_OK, "LockRequest failed: %08x\n", hres); hres = IInternetProtocol_Terminate(protocol, 0); ok(hres == S_OK, "Terminate failed: %08x\n", hres); hres = IInternetProtocol_Read(protocol, buf, 2, &cb); ok(hres == S_OK, "Read failed: %08x\n\n", hres); hres = IInternetProtocol_UnlockRequest(protocol); ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres); hres = IInternetProtocol_Read(protocol, buf, 2, &cb); if(file_with_hash) /* FIXME: An effect of UnlockRequest call? */ todo_wine ok(hres == S_OK, "Read failed: %08x\n", hres); else ok(hres == S_OK, "Read failed: %08x\n", hres); hres = IInternetProtocol_Terminate(protocol, 0); ok(hres == S_OK, "Terminate failed: %08x\n", hres); } IInternetProtocol_Release(protocol); } hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol); ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres); if(SUCCEEDED(hres)) { if(file_protocol_start(protocol, url, NULL, NULL, TRUE)) { hres = IInternetProtocol_Terminate(protocol, 0); ok(hres == S_OK, "Terminate failed: %08x\n", hres); hres = IInternetProtocol_Read(protocol, buf, 2, &cb); ok(hres == S_OK, "Read failed: %08x\n", hres); ok(cb == 2, "cb=%u expected 2\n", cb); } IInternetProtocol_Release(protocol); } if(pCreateUri) { IInternetProtocolEx *protocolex; IUri *uri; hres = pCreateUri(url, Uri_CREATE_FILE_USE_DOS_PATH, 0, &uri); ok(hres == S_OK, "CreateUri failed: %08x\n", hres); hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocolEx, (void**)&protocolex); ok(hres == S_OK, "Could not get IInternetProtocolEx: %08x\n", hres); if(file_protocol_start(NULL, NULL, protocolex, uri, TRUE)) { hres = IInternetProtocolEx_Read(protocolex, buf, 2, &cb); ok(hres == S_OK, "Read failed: %08x\n", hres); hres = IInternetProtocolEx_LockRequest(protocolex, 0); ok(hres == S_OK, "LockRequest failed: %08x\n", hres); hres = IInternetProtocolEx_UnlockRequest(protocolex); ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres); } IUri_Release(uri); IInternetProtocolEx_Release(protocolex); hres = pCreateUri(url, 0, 0, &uri); ok(hres == S_OK, "CreateUri failed: %08x\n", hres); hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocolEx, (void**)&protocolex); ok(hres == S_OK, "Could not get IInternetProtocolEx: %08x\n", hres); if(file_protocol_start(NULL, NULL, protocolex, uri, TRUE)) { hres = IInternetProtocolEx_Read(protocolex, buf, 2, &cb); ok(hres == S_OK, "Read failed: %08x\n", hres); hres = IInternetProtocolEx_LockRequest(protocolex, 0); ok(hres == S_OK, "LockRequest failed: %08x\n", hres); hres = IInternetProtocolEx_UnlockRequest(protocolex); ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres); } IUri_Release(uri); IInternetProtocolEx_Release(protocolex); }else { win_skip("Skipping file protocol StartEx tests\n"); } IClassFactory_Release(factory); } static void test_file_protocol_fail(void) { IInternetProtocol *protocol; HRESULT hres; static const WCHAR index_url2[] = {'f','i','l','e',':','/','/','i','n','d','e','x','.','h','t','m','l',0}; hres = CoCreateInstance(&CLSID_FileProtocol, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER, &IID_IInternetProtocol, (void**)&protocol); ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres); if(FAILED(hres)) return; SET_EXPECT(GetBindInfo); expect_hrResult = MK_E_SYNTAX; hres = IInternetProtocol_Start(protocol, wszIndexHtml, &protocol_sink, &bind_info, 0, 0); ok(hres == MK_E_SYNTAX || hres == E_INVALIDARG, "Start failed: %08x, expected MK_E_SYNTAX or E_INVALIDARG\n", hres); CLEAR_CALLED(GetBindInfo); /* GetBindInfo not called in IE7 */ SET_EXPECT(GetBindInfo); if(!(bindf & BINDF_FROMURLMON)) SET_EXPECT(ReportProgress_DIRECTBIND); SET_EXPECT(ReportProgress_SENDINGREQUEST); SET_EXPECT(ReportResult); expect_hrResult = INET_E_RESOURCE_NOT_FOUND; hres = IInternetProtocol_Start(protocol, index_url, &protocol_sink, &bind_info, 0, 0); ok(hres == INET_E_RESOURCE_NOT_FOUND, "Start failed: %08x expected INET_E_RESOURCE_NOT_FOUND\n", hres); CHECK_CALLED(GetBindInfo); if(!(bindf & BINDF_FROMURLMON)) CHECK_CALLED(ReportProgress_DIRECTBIND); CHECK_CALLED(ReportProgress_SENDINGREQUEST); CHECK_CALLED(ReportResult); IInternetProtocol_Release(protocol); hres = CoCreateInstance(&CLSID_FileProtocol, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER, &IID_IInternetProtocol, (void**)&protocol); ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres); if(FAILED(hres)) return; SET_EXPECT(GetBindInfo); if(!(bindf & BINDF_FROMURLMON)) SET_EXPECT(ReportProgress_DIRECTBIND); SET_EXPECT(ReportProgress_SENDINGREQUEST); SET_EXPECT(ReportResult); expect_hrResult = INET_E_RESOURCE_NOT_FOUND; hres = IInternetProtocol_Start(protocol, index_url2, &protocol_sink, &bind_info, 0, 0); ok(hres == INET_E_RESOURCE_NOT_FOUND, "Start failed: %08x, expected INET_E_RESOURCE_NOT_FOUND\n", hres); CHECK_CALLED(GetBindInfo); if(!(bindf & BINDF_FROMURLMON)) CHECK_CALLED(ReportProgress_DIRECTBIND); CHECK_CALLED(ReportProgress_SENDINGREQUEST); CHECK_CALLED(ReportResult); SET_EXPECT(GetBindInfo); hres = IInternetProtocol_Start(protocol, NULL, &protocol_sink, &bind_info, 0, 0); ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres); CLEAR_CALLED(GetBindInfo); /* GetBindInfo not called in IE7 */ SET_EXPECT(GetBindInfo); hres = IInternetProtocol_Start(protocol, emptyW, &protocol_sink, &bind_info, 0, 0); ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres); CLEAR_CALLED(GetBindInfo); /* GetBindInfo not called in IE7 */ IInternetProtocol_Release(protocol); } static void test_file_protocol(void) { WCHAR buf[INTERNET_MAX_URL_LENGTH], file_name_buf[MAX_PATH]; DWORD size; ULONG len; HANDLE file; static const WCHAR wszFile[] = {'f','i','l','e',':',0}; static const WCHAR wszFile2[] = {'f','i','l','e',':','/','/',0}; static const WCHAR wszFile3[] = {'f','i','l','e',':','/','/','/',0}; static const WCHAR wszFile4[] = {'f','i','l','e',':','\\','\\',0}; static const char html_doc[] = "<HTML></HTML>"; static const WCHAR fragmentW[] = {'#','f','r','a','g',0}; trace("Testing file protocol...\n"); init_test(FILE_TEST, 0); SetLastError(0xdeadbeef); file = CreateFileW(wszIndexHtml, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); ok(file != INVALID_HANDLE_VALUE, "CreateFile failed\n"); if(file == INVALID_HANDLE_VALUE) return; WriteFile(file, html_doc, sizeof(html_doc)-1, &size, NULL); CloseHandle(file); file_name = wszIndexHtml; bindf = 0; test_file_protocol_url(index_url); bindf = BINDF_FROMURLMON; test_file_protocol_url(index_url); bindf = BINDF_FROMURLMON | BINDF_NEEDFILE; test_file_protocol_url(index_url); memcpy(buf, wszFile, sizeof(wszFile)); len = sizeof(wszFile)/sizeof(WCHAR)-1; len += GetCurrentDirectoryW(sizeof(buf)/sizeof(WCHAR)-len, buf+len); buf[len++] = '\\'; memcpy(buf+len, wszIndexHtml, sizeof(wszIndexHtml)); file_name = buf + sizeof(wszFile)/sizeof(WCHAR)-1; bindf = 0; test_file_protocol_url(buf); bindf = BINDF_FROMURLMON; test_file_protocol_url(buf); memcpy(buf, wszFile2, sizeof(wszFile2)); len = GetCurrentDirectoryW(sizeof(file_name_buf)/sizeof(WCHAR), file_name_buf); file_name_buf[len++] = '\\'; memcpy(file_name_buf+len, wszIndexHtml, sizeof(wszIndexHtml)); lstrcpyW(buf+sizeof(wszFile2)/sizeof(WCHAR)-1, file_name_buf); file_name = file_name_buf; bindf = 0; test_file_protocol_url(buf); bindf = BINDF_FROMURLMON; test_file_protocol_url(buf); buf[sizeof(wszFile2)/sizeof(WCHAR)] = '|'; test_file_protocol_url(buf); memcpy(buf, wszFile3, sizeof(wszFile3)); len = sizeof(wszFile3)/sizeof(WCHAR)-1; len += GetCurrentDirectoryW(sizeof(buf)/sizeof(WCHAR)-len, buf+len); buf[len++] = '\\'; memcpy(buf+len, wszIndexHtml, sizeof(wszIndexHtml)); file_name = buf + sizeof(wszFile3)/sizeof(WCHAR)-1; bindf = 0; test_file_protocol_url(buf); bindf = BINDF_FROMURLMON; test_file_protocol_url(buf); memcpy(buf, wszFile4, sizeof(wszFile4)); len = GetCurrentDirectoryW(sizeof(file_name_buf)/sizeof(WCHAR), file_name_buf); file_name_buf[len++] = '\\'; memcpy(file_name_buf+len, wszIndexHtml, sizeof(wszIndexHtml)); lstrcpyW(buf+sizeof(wszFile4)/sizeof(WCHAR)-1, file_name_buf); file_name = file_name_buf; bindf = 0; test_file_protocol_url(buf); bindf = BINDF_FROMURLMON; test_file_protocol_url(buf); buf[sizeof(wszFile4)/sizeof(WCHAR)] = '|'; test_file_protocol_url(buf); /* Fragment part of URL is skipped if the file doesn't exist. */ lstrcatW(buf, fragmentW); test_file_protocol_url(buf); /* Fragment part is considered a part of the file name, if the file exsists. */ len = lstrlenW(file_name_buf); lstrcpyW(file_name_buf+len, fragmentW); file = CreateFileW(wszIndexHtml, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); ok(file != INVALID_HANDLE_VALUE, "CreateFile failed\n"); WriteFile(file, "XXX", 3, &size, NULL); CloseHandle(file); file_name_buf[len] = 0; file_with_hash = TRUE; test_file_protocol_url(buf); DeleteFileW(wszIndexHtml); DeleteFileW(file_name_buf); bindf = 0; test_file_protocol_fail(); bindf = BINDF_FROMURLMON; test_file_protocol_fail(); } static void create_cache_entry(const WCHAR *urlw) { FILETIME now, tomorrow, yesterday; char file_path[MAX_PATH]; BYTE content[1000]; ULARGE_INTEGER li; const char *url; HANDLE file; DWORD size; unsigned i; BOOL res; BYTE cache_headers[] = "HTTP/1.1 200 OK\r\n\r\n"; trace("Testing cache read...\n"); url = w2a(urlw); for(i = 0; i < sizeof(content); i++) content[i] = '0' + (i%10); GetSystemTimeAsFileTime(&now); li.u.HighPart = now.dwHighDateTime; li.u.LowPart = now.dwLowDateTime; li.QuadPart += (LONGLONG)10000000 * 3600 * 24; tomorrow.dwHighDateTime = li.u.HighPart; tomorrow.dwLowDateTime = li.u.LowPart; li.QuadPart -= (LONGLONG)10000000 * 3600 * 24 * 2; yesterday.dwHighDateTime = li.u.HighPart; yesterday.dwLowDateTime = li.u.LowPart; res = CreateUrlCacheEntryA(url, sizeof(content), "", file_path, 0); ok(res, "CreateUrlCacheEntryA failed: %u\n", GetLastError()); file = CreateFileA(file_path, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); ok(file != INVALID_HANDLE_VALUE, "CreateFile failed\n"); WriteFile(file, content, sizeof(content), &size, NULL); CloseHandle(file); res = CommitUrlCacheEntryA(url, file_path, tomorrow, yesterday, NORMAL_CACHE_ENTRY, cache_headers, sizeof(cache_headers)-1, "", 0); ok(res, "CommitUrlCacheEntryA failed: %u\n", GetLastError()); } static BOOL http_protocol_start(LPCWSTR url, BOOL use_iuri) { static BOOL got_user_agent = FALSE; IUri *uri = NULL; HRESULT hres; if(use_iuri && pCreateUri) { hres = pCreateUri(url, 0, 0, &uri); ok(hres == S_OK, "CreateUri failed: %08x\n", hres); } SET_EXPECT(GetBindInfo); if (!(bindf & BINDF_FROMURLMON)) SET_EXPECT(ReportProgress_DIRECTBIND); if(!got_user_agent) SET_EXPECT(GetBindString_USER_AGENT); SET_EXPECT(GetBindString_ROOTDOC_URL); SET_EXPECT(GetBindString_ACCEPT_MIMES); SET_EXPECT(QueryService_HttpNegotiate); SET_EXPECT(BeginningTransaction); SET_EXPECT(GetRootSecurityId); if(http_post_test) { SET_EXPECT(GetBindString_POST_COOKIE); if(http_post_test == TYMED_ISTREAM) SET_EXPECT(Stream_Seek); } if(bind_from_cache) { SET_EXPECT(OnResponse); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); SET_EXPECT(ReportData); } if(uri) { IInternetProtocolEx *protocolex; hres = IInternetProtocol_QueryInterface(async_protocol, &IID_IInternetProtocolEx, (void**)&protocolex); ok(hres == S_OK, "Could not get IInternetProtocolEx iface: %08x\n", hres); hres = IInternetProtocolEx_StartEx(protocolex, uri, &protocol_sink, &bind_info, 0, 0); ok(hres == S_OK, "Start failed: %08x\n", hres); IInternetProtocolEx_Release(protocolex); IUri_Release(uri); }else { hres = IInternetProtocol_Start(async_protocol, url, &protocol_sink, &bind_info, 0, 0); ok(hres == S_OK, "Start failed: %08x\n", hres); } if(FAILED(hres)) return FALSE; CHECK_CALLED(GetBindInfo); if (!(bindf & BINDF_FROMURLMON)) CHECK_CALLED(ReportProgress_DIRECTBIND); if (!got_user_agent) { CHECK_CALLED(GetBindString_USER_AGENT); got_user_agent = TRUE; } CLEAR_CALLED(GetBindString_ROOTDOC_URL); /* New in IE11 */ CHECK_CALLED(GetBindString_ACCEPT_MIMES); CHECK_CALLED(QueryService_HttpNegotiate); CHECK_CALLED(BeginningTransaction); /* GetRootSecurityId called on WinXP but not on Win98 */ CLEAR_CALLED(GetRootSecurityId); if(http_post_test) { CHECK_CALLED(GetBindString_POST_COOKIE); if(http_post_test == TYMED_ISTREAM) CHECK_CALLED(Stream_Seek); } if(bind_from_cache) { CHECK_CALLED(OnResponse); CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); CHECK_CALLED(ReportData); } return TRUE; } static void test_protocol_terminate(IInternetProtocol *protocol) { BYTE buf[3600]; DWORD cb; HRESULT hres; hres = IInternetProtocol_LockRequest(protocol, 0); ok(hres == S_OK, "LockRequest failed: %08x\n", hres); hres = IInternetProtocol_Read(protocol, buf, 1, &cb); ok(hres == (test_abort ? S_OK : S_FALSE), "Read failed: %08x\n", hres); hres = IInternetProtocol_Terminate(protocol, 0); ok(hres == S_OK, "Terminate failed: %08x\n", hres); /* This wait is to give the internet handles being freed in Terminate * enough time to actually terminate in all cases. Internet handles * terminate asynchronously and native reuses the main InternetOpen * handle. The only case in which this seems to be necessary is on * wine with native wininet and urlmon, resulting in the next time * test_http_protocol_url being called the first data notification actually * being an extra last data notification from the previous connection * about once out of every ten times. */ Sleep(100); hres = IInternetProtocol_UnlockRequest(protocol); ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres); } static void test_http_info(IInternetProtocol *protocol) { IWinInetHttpInfo *info; HRESULT hres; hres = IInternetProtocol_QueryInterface(protocol, &IID_IWinInetHttpInfo, (void**)&info); ok(hres == S_OK, "Could not get IWinInterHttpInfo iface: %08x\n", hres); /* TODO */ IWinInetHttpInfo_Release(info); } /* is_first refers to whether this is the first call to this function * _for this url_ */ static void test_http_protocol_url(LPCWSTR url, int prot, DWORD flags, DWORD tymed) { IInternetProtocolInfo *protocol_info; IClassFactory *factory; IUnknown *unk; HRESULT hres; init_test(prot, flags); http_url = url; http_post_test = tymed; if(flags & TEST_FROMCACHE) create_cache_entry(url); hres = CoGetClassObject(prot == HTTPS_TEST ? &CLSID_HttpSProtocol : &CLSID_HttpProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk); ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres); if(FAILED(hres)) return; hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info); ok(hres == E_NOINTERFACE, "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres); hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory); ok(hres == S_OK, "Could not get IClassFactory interface\n"); IUnknown_Release(unk); if(FAILED(hres)) return; hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&async_protocol); ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres); if(SUCCEEDED(hres)) { BYTE buf[3600]; DWORD cb; ULONG ref; test_priority(async_protocol); test_http_info(async_protocol); SET_EXPECT(ReportProgress_COOKIE_SENT); if(http_is_first) { SET_EXPECT(ReportProgress_FINDINGRESOURCE); SET_EXPECT(ReportProgress_CONNECTING); } SET_EXPECT(ReportProgress_SENDINGREQUEST); if(test_redirect) SET_EXPECT(ReportProgress_REDIRECTING); SET_EXPECT(ReportProgress_PROXYDETECTING); if(prot == HTTP_TEST) SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE); else SET_EXPECT(QueryService_HttpSecurity); if(!(bindf & BINDF_FROMURLMON)) { SET_EXPECT(OnResponse); SET_EXPECT(ReportProgress_RAWMIMETYPE); SET_EXPECT(ReportData); } else { SET_EXPECT(Switch); } if(!http_protocol_start(url, (flags & TEST_USEIURI) != 0)) { IInternetProtocol_Abort(async_protocol, E_ABORT, 0); IInternetProtocol_Release(async_protocol); return; } if(!direct_read && !test_abort && !bind_from_cache) SET_EXPECT(ReportResult); expect_hrResult = test_abort ? E_ABORT : S_OK; if(direct_read) { SET_EXPECT(Switch); while(wait_for_switch) { ok( WaitForSingleObject(event_continue, 90000) == WAIT_OBJECT_0, "wait timed out\n" ); CHECK_CALLED(Switch); /* Set in ReportData */ call_continue(&continue_protdata); SetEvent(event_continue_done); } }else if(bind_from_cache) { BYTE buf[1500]; hres = IInternetProtocol_Read(async_protocol, buf, 100, &cb); ok(hres == S_OK && cb == 100, "Read failed: %08x (%d bytes)\n", hres, cb); SET_EXPECT(ReportResult); hres = IInternetProtocol_Read(async_protocol, buf, sizeof(buf), &cb); ok(hres == S_OK && cb == 900, "Read failed: %08x (%d bytes)\n", hres, cb); CHECK_CALLED(ReportResult); hres = IInternetProtocol_Read(async_protocol, buf, sizeof(buf), &cb); ok(hres == S_FALSE && !cb, "Read failed: %08x (%d bytes)\n", hres, cb); }else { hres = IInternetProtocol_Read(async_protocol, buf, 1, &cb); ok((hres == E_PENDING && cb==0) || (hres == S_OK && cb==1), "Read failed: %08x (%d bytes)\n", hres, cb); ok( WaitForSingleObject(event_complete, 90000) == WAIT_OBJECT_0, "wait timed out\n" ); if(bindf & BINDF_FROMURLMON) CHECK_CALLED(Switch); else CHECK_CALLED(ReportData); if(prot == HTTPS_TEST) CLEAR_CALLED(QueryService_HttpSecurity); while(1) { if(bindf & BINDF_FROMURLMON) SET_EXPECT(Switch); else SET_EXPECT(ReportData); hres = IInternetProtocol_Read(async_protocol, buf, sizeof(buf), &cb); if(hres == E_PENDING) { hres = IInternetProtocol_Read(async_protocol, buf, 1, &cb); ok((hres == E_PENDING && cb==0) || (hres == S_OK && cb==1), "Read failed: %08x (%d bytes)\n", hres, cb); ok( WaitForSingleObject(event_complete, 90000) == WAIT_OBJECT_0, "wait timed out\n" ); if(bindf & BINDF_FROMURLMON) CHECK_CALLED(Switch); else CHECK_CALLED(ReportData); if(test_abort) { HRESULT hres; SET_EXPECT(ReportResult); hres = IInternetProtocol_Abort(async_protocol, E_ABORT, 0); ok(hres == S_OK, "Abort failed: %08x\n", hres); CHECK_CALLED(ReportResult); hres = IInternetProtocol_Abort(async_protocol, E_ABORT, 0); ok(hres == INET_E_RESULT_DISPATCHED || hres == S_OK /* IE10 */, "Abort failed: %08x\n", hres); break; } }else { if(bindf & BINDF_FROMURLMON) CHECK_NOT_CALLED(Switch); else CHECK_NOT_CALLED(ReportData); if(cb == 0) break; } } if(!test_abort) { ok(hres == S_FALSE, "Read failed: %08x\n", hres); CHECK_CALLED(ReportResult); } } if(prot == HTTPS_TEST) CLEAR_CALLED(ReportProgress_SENDINGREQUEST); if (prot == HTTP_TEST || prot == HTTPS_TEST) CLEAR_CALLED(ReportProgress_COOKIE_SENT); hres = IInternetProtocol_Abort(async_protocol, E_ABORT, 0); ok(hres == INET_E_RESULT_DISPATCHED || hres == S_OK /* IE10 */, "Abort failed: %08x\n", hres); test_protocol_terminate(async_protocol); hres = IInternetProtocol_Abort(async_protocol, E_ABORT, 0); ok(hres == S_OK, "Abort failed: %08x\n", hres); ref = IInternetProtocol_Release(async_protocol); ok(!ref, "ref=%x\n", ref); } IClassFactory_Release(factory); if(flags & TEST_FROMCACHE) { BOOL res; res = DeleteUrlCacheEntryW(url); ok(res, "DeleteUrlCacheEntryA failed: %u\n", GetLastError()); } } static void test_http_protocol(void) { static const WCHAR posttest_url[] = {'h','t','t','p',':','/','/','t','e','s','t','.','w','i','n','e','h','q','.','o','r','g','/', 't','e','s','t','s','/','p','o','s','t','.','p','h','p',0}; static const WCHAR redirect_url[] = {'h','t','t','p',':','/','/','t','e','s','t','.','w','i','n','e','h','q','.','o','r','g','/', 't','e','s','t','s','/','r','e','d','i','r','e','c','t',0}; static const WCHAR winetest_url[] = {'h','t','t','p',':','/','/','t','e','s','t','.','w','i','n','e','h','q','.','o','r','g','/', 't','e','s','t','s','/','d','a','t','a','.','p','h','p',0}; static const WCHAR empty_url[] = {'h','t','t','p',':','/','/','t','e','s','t','.','w','i','n','e','h','q','.','o','r','g','/', 't','e','s','t','s','/','e','m','p','t','y','.','j','s',0}; static const WCHAR cache_only_url[] = {'h','t','t','p',':','/','/','t','e','s','t','.','w','i','n','e','h','q','.','o','r','g','/', 't','e','s','t','s','/','c','a','c','h','e','-','o','n','l','y',0}; trace("Testing http protocol (not from urlmon)...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; test_http_protocol_url(winetest_url, HTTP_TEST, TEST_FIRST_HTTP, TYMED_NULL); trace("Testing http protocol (from urlmon)...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON; test_http_protocol_url(winetest_url, HTTP_TEST, 0, TYMED_NULL); trace("Testing http protocol (to file)...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NEEDFILE; test_http_protocol_url(winetest_url, HTTP_TEST, 0, TYMED_NULL); trace("Testing http protocol (post data)...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON; test_http_protocol_url(posttest_url, HTTP_TEST, TEST_FIRST_HTTP|TEST_POST, TYMED_HGLOBAL); trace("Testing http protocol (post data stream)...\n"); test_http_protocol_url(posttest_url, HTTP_TEST, TEST_FIRST_HTTP|TEST_POST|TEST_ASYNCREQ, TYMED_ISTREAM); trace("Testing http protocol (direct read)...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON; test_http_protocol_url(winetest_url, HTTP_TEST, TEST_DIRECT_READ|TEST_USEIURI, TYMED_NULL); trace("Testing http protocol (redirected)...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NOWRITECACHE; test_http_protocol_url(redirect_url, HTTP_TEST, TEST_REDIRECT, TYMED_NULL); trace("Testing http protocol empty file...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NOWRITECACHE; test_http_protocol_url(empty_url, HTTP_TEST, TEST_EMPTY, TYMED_NULL); /* This is a bit ugly. We unconditionally disable this test on Wine. This won't work until we have * support for reading from cache via HTTP layer in wininet. Until then, Wine will fail badly, affecting * other, unrelated, tests. Working around it is not worth the trouble, we may simply make sure those * tests work on Windows and have them around for the future. */ if(broken(1)) { trace("Testing http protocol (from cache)...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON; test_http_protocol_url(cache_only_url, HTTP_TEST, TEST_FROMCACHE, TYMED_NULL); } trace("Testing http protocol abort...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NOWRITECACHE; test_http_protocol_url(winetest_url, HTTP_TEST, TEST_ABORT, TYMED_NULL); test_early_abort(&CLSID_HttpProtocol); test_early_abort(&CLSID_HttpSProtocol); } static void test_https_protocol(void) { static const WCHAR https_winehq_url[] = {'h','t','t','p','s',':','/','/','t','e','s','t','.','w','i','n','e','h','q','.','o','r','g','/', 't','e','s','t','s','/','h','e','l','l','o','.','h','t','m','l',0}; trace("Testing https protocol (from urlmon)...\n"); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NOWRITECACHE; test_http_protocol_url(https_winehq_url, HTTPS_TEST, TEST_FIRST_HTTP, TYMED_NULL); } static void test_ftp_protocol(void) { IInternetProtocolInfo *protocol_info; IClassFactory *factory; IUnknown *unk; BYTE buf[4096]; ULONG ref; DWORD cb; HRESULT hres; static const WCHAR ftp_urlW[] = {'f','t','p',':','/','/','f','t','p','.','w','i','n','e','h','q','.','o','r','g', '/','p','u','b','/','o','t','h','e','r','/', 'w','i','n','e','l','o','g','o','.','x','c','f','.','t','a','r','.','b','z','2',0}; trace("Testing ftp protocol...\n"); init_test(FTP_TEST, 0); bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NOWRITECACHE; state = STATE_STARTDOWNLOADING; expect_hrResult = E_PENDING; hres = CoGetClassObject(&CLSID_FtpProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk); ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres); if(FAILED(hres)) return; hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info); ok(hres == E_NOINTERFACE, "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres); hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory); ok(hres == S_OK, "Could not get IClassFactory interface\n"); IUnknown_Release(unk); if(FAILED(hres)) return; hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&async_protocol); IClassFactory_Release(factory); ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres); test_priority(async_protocol); test_http_info(async_protocol); SET_EXPECT(GetBindInfo); SET_EXPECT(ReportProgress_FINDINGRESOURCE); SET_EXPECT(ReportProgress_CONNECTING); SET_EXPECT(ReportProgress_SENDINGREQUEST); SET_EXPECT(Switch); hres = IInternetProtocol_Start(async_protocol, ftp_urlW, &protocol_sink, &bind_info, 0, 0); ok(hres == S_OK, "Start failed: %08x\n", hres); CHECK_CALLED(GetBindInfo); SET_EXPECT(ReportResult); hres = IInternetProtocol_Read(async_protocol, buf, 1, &cb); ok((hres == E_PENDING && cb==0) || (hres == S_OK && cb==1), "Read failed: %08x (%d bytes)\n", hres, cb); ok( WaitForSingleObject(event_complete, 90000) == WAIT_OBJECT_0, "wait timed out\n" ); while(1) { hres = IInternetProtocol_Read(async_protocol, buf, sizeof(buf), &cb); if(hres == E_PENDING) { DWORD ret = WaitForSingleObject(event_complete, 90000); ok( ret == WAIT_OBJECT_0, "wait timed out\n" ); if (ret != WAIT_OBJECT_0) break; } else if(cb == 0) break; } ok(hres == S_FALSE, "Read failed: %08x\n", hres); CHECK_CALLED(ReportResult); CHECK_CALLED(Switch); test_protocol_terminate(async_protocol); if(pCreateUri) { IInternetProtocolEx *protocolex; hres = IInternetProtocol_QueryInterface(async_protocol, &IID_IInternetProtocolEx, (void**)&protocolex); ok(hres == S_OK, "Could not get IInternetProtocolEx iface: %08x\n", hres); IInternetProtocolEx_Release(protocolex); } ref = IInternetProtocol_Release(async_protocol); ok(!ref, "ref=%d\n", ref); test_early_abort(&CLSID_FtpProtocol); } static void test_gopher_protocol(void) { IInternetProtocolInfo *protocol_info; IClassFactory *factory; IUnknown *unk; HRESULT hres; trace("Testing gopher protocol...\n"); hres = CoGetClassObject(&CLSID_GopherProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk); ok(hres == S_OK || broken(hres == REGDB_E_CLASSNOTREG || hres == CLASS_E_CLASSNOTAVAILABLE), /* Gopher protocol has been removed as of Vista */ "CoGetClassObject failed: %08x\n", hres); if(FAILED(hres)) return; hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info); ok(hres == E_NOINTERFACE, "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres); hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory); ok(hres == S_OK, "Could not get IClassFactory interface\n"); IUnknown_Release(unk); if(FAILED(hres)) return; hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&async_protocol); IClassFactory_Release(factory); ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres); test_priority(async_protocol); IInternetProtocol_Release(async_protocol); test_early_abort(&CLSID_GopherProtocol); } static void test_mk_protocol(void) { IInternetProtocolInfo *protocol_info; IInternetProtocol *protocol; IClassFactory *factory; IUnknown *unk; HRESULT hres; static const WCHAR wrong_url1[] = {'t','e','s','t',':','@','M','S','I','T','S','t','o','r','e', ':',':','/','t','e','s','t','.','h','t','m','l',0}; static const WCHAR wrong_url2[] = {'m','k',':','/','t','e','s','t','.','h','t','m','l',0}; trace("Testing mk protocol...\n"); init_test(MK_TEST, 0); hres = CoGetClassObject(&CLSID_MkProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk); ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres); hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info); ok(hres == E_NOINTERFACE, "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres); hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory); ok(hres == S_OK, "Could not get IClassFactory interface\n"); IUnknown_Release(unk); if(FAILED(hres)) return; hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol); IClassFactory_Release(factory); ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres); SET_EXPECT(GetBindInfo); hres = IInternetProtocol_Start(protocol, wrong_url1, &protocol_sink, &bind_info, 0, 0); ok(hres == MK_E_SYNTAX || hres == INET_E_INVALID_URL, "Start failed: %08x, expected MK_E_SYNTAX or INET_E_INVALID_URL\n", hres); CLEAR_CALLED(GetBindInfo); SET_EXPECT(GetBindInfo); SET_EXPECT(ReportProgress_DIRECTBIND); SET_EXPECT(ReportProgress_SENDINGREQUEST); SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE); SET_EXPECT(ReportResult); expect_hrResult = INET_E_RESOURCE_NOT_FOUND; hres = IInternetProtocol_Start(protocol, wrong_url2, &protocol_sink, &bind_info, 0, 0); ok(hres == INET_E_RESOURCE_NOT_FOUND || hres == INET_E_INVALID_URL, /* win2k3 */ "Start failed: %08x, expected INET_E_RESOURCE_NOT_FOUND or INET_E_INVALID_URL\n", hres); if (hres == INET_E_RESOURCE_NOT_FOUND) { CHECK_CALLED(GetBindInfo); CLEAR_CALLED(ReportProgress_DIRECTBIND); CHECK_CALLED(ReportProgress_SENDINGREQUEST); CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE); CHECK_CALLED(ReportResult); }else { CLEAR_CALLED(GetBindInfo); CLEAR_CALLED(ReportProgress_DIRECTBIND); CLEAR_CALLED(ReportProgress_SENDINGREQUEST); CLEAR_CALLED(ReportProgress_MIMETYPEAVAILABLE); CLEAR_CALLED(ReportResult); } IInternetProtocol_Release(protocol); } static void test_CreateBinding(void) { IInternetProtocol *protocol; IInternetPriority *priority; IInternetSession *session; IWinInetHttpInfo *http_info; IWinInetInfo *inet_info; LONG p; BYTE buf[1000]; DWORD read; HRESULT hres; static const WCHAR test_url[] = {'t','e','s','t',':','/','/','f','i','l','e','.','h','t','m','l',0}; static const WCHAR wsz_test[] = {'t','e','s','t',0}; trace("Testing CreateBinding...\n"); init_test(BIND_TEST, TEST_BINDING); hres = pCoInternetGetSession(0, &session, 0); ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres); hres = IInternetSession_RegisterNameSpace(session, &ClassFactory, &IID_NULL, wsz_test, 0, NULL, 0); ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres); hres = IInternetSession_CreateBinding(session, NULL, test_url, NULL, NULL, &protocol, 0); binding_protocol = protocol; ok(hres == S_OK, "CreateBinding failed: %08x\n", hres); ok(protocol != NULL, "protocol == NULL\n"); hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetBindInfo, (void**)&prot_bind_info); ok(hres == S_OK, "QueryInterface(IID_IInternetBindInfo) failed: %08x\n", hres); hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetProtocolSink, (void**)&binding_sink); ok(hres == S_OK, "Could not get IInternetProtocolSink: %08x\n", hres); hres = IInternetProtocol_Start(protocol, test_url, NULL, &bind_info, 0, 0); ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres); hres = IInternetProtocol_Start(protocol, test_url, &protocol_sink, NULL, 0, 0); ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres); hres = IInternetProtocol_Start(protocol, NULL, &protocol_sink, &bind_info, 0, 0); ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres); hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority); ok(hres == S_OK, "QueryInterface(IID_IInternetPriority) failed: %08x\n", hres); p = 0xdeadbeef; hres = IInternetPriority_GetPriority(priority, &p); ok(hres == S_OK, "GetPriority failed: %08x\n", hres); ok(!p, "p=%d\n", p); ex_priority = 100; hres = IInternetPriority_SetPriority(priority, 100); ok(hres == S_OK, "SetPriority failed: %08x\n", hres); p = 0xdeadbeef; hres = IInternetPriority_GetPriority(priority, &p); ok(hres == S_OK, "GetPriority failed: %08x\n", hres); ok(p == 100, "p=%d\n", p); hres = IInternetProtocol_QueryInterface(protocol, &IID_IWinInetInfo, (void**)&inet_info); ok(hres == E_NOINTERFACE, "Could not get IWinInetInfo protocol: %08x\n", hres); SET_EXPECT(QueryService_InternetProtocol); SET_EXPECT(CreateInstance); SET_EXPECT(ReportProgress_PROTOCOLCLASSID); SET_EXPECT(SetPriority); SET_EXPECT(Start); expect_hrResult = S_OK; hres = IInternetProtocol_Start(protocol, test_url, &protocol_sink, &bind_info, 0, 0); ok(hres == S_OK, "Start failed: %08x\n", hres); CHECK_CALLED(QueryService_InternetProtocol); CHECK_CALLED(CreateInstance); CHECK_CALLED(ReportProgress_PROTOCOLCLASSID); CHECK_CALLED(SetPriority); CHECK_CALLED(Start); SET_EXPECT(QueryInterface_IWinInetInfo); hres = IInternetProtocol_QueryInterface(protocol, &IID_IWinInetInfo, (void**)&inet_info); ok(hres == E_NOINTERFACE, "Could not get IWinInetInfo protocol: %08x\n", hres); CHECK_CALLED(QueryInterface_IWinInetInfo); SET_EXPECT(QueryInterface_IWinInetInfo); hres = IInternetProtocol_QueryInterface(protocol, &IID_IWinInetInfo, (void**)&inet_info); ok(hres == E_NOINTERFACE, "Could not get IWinInetInfo protocol: %08x\n", hres); CHECK_CALLED(QueryInterface_IWinInetInfo); SET_EXPECT(QueryInterface_IWinInetHttpInfo); hres = IInternetProtocol_QueryInterface(protocol, &IID_IWinInetHttpInfo, (void**)&http_info); ok(hres == E_NOINTERFACE, "Could not get IWinInetInfo protocol: %08x\n", hres); CHECK_CALLED(QueryInterface_IWinInetHttpInfo); SET_EXPECT(Read); read = 0xdeadbeef; hres = IInternetProtocol_Read(protocol, expect_pv = buf, sizeof(buf), &read); ok(hres == S_OK, "Read failed: %08x\n", hres); ok(read == 100, "read = %d\n", read); CHECK_CALLED(Read); SET_EXPECT(Read); read = 0xdeadbeef; hres = IInternetProtocol_Read(protocol, expect_pv = buf, sizeof(buf), &read); ok(hres == S_FALSE, "Read failed: %08x\n", hres); ok(!read, "read = %d\n", read); CHECK_CALLED(Read); p = 0xdeadbeef; hres = IInternetPriority_GetPriority(priority, &p); ok(hres == S_OK, "GetPriority failed: %08x\n", hres); ok(p == 100, "p=%d\n", p); hres = IInternetPriority_SetPriority(priority, 101); ok(hres == S_OK, "SetPriority failed: %08x\n", hres); SET_EXPECT(Terminate); hres = IInternetProtocol_Terminate(protocol, 0xdeadbeef); ok(hres == S_OK, "Terminate failed: %08x\n", hres); CHECK_CALLED(Terminate); SET_EXPECT(Continue); hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata); ok(hres == S_OK, "Switch failed: %08x\n", hres); CHECK_CALLED(Continue); hres = IInternetProtocolSink_ReportProgress(binding_sink, BINDSTATUS_CACHEFILENAMEAVAILABLE, expect_wsz = emptyW); ok(hres == S_OK, "ReportProgress(BINDSTATUS_CACHEFILENAMEAVAILABLE) failed: %08x\n", hres); hres = IInternetProtocolSink_ReportResult(binding_sink, S_OK, ERROR_SUCCESS, NULL); ok(hres == E_FAIL, "ReportResult failed: %08x, expected E_FAIL\n", hres); hres = IInternetProtocolSink_ReportData(binding_sink, 0, 0, 0); ok(hres == S_OK, "ReportData failed: %08x\n", hres); IInternetProtocolSink_Release(binding_sink); IInternetPriority_Release(priority); IInternetBindInfo_Release(prot_bind_info); IInternetProtocol_Release(protocol); hres = IInternetSession_CreateBinding(session, NULL, test_url, NULL, NULL, &protocol, 0); ok(hres == S_OK, "CreateBinding failed: %08x\n", hres); ok(protocol != NULL, "protocol == NULL\n"); hres = IInternetProtocol_Abort(protocol, E_ABORT, 0); ok(hres == S_OK, "Abort failed: %08x\n", hres); hres = IInternetProtocol_Abort(protocol, E_FAIL, 0); ok(hres == S_OK, "Abort failed: %08x\n", hres); IInternetProtocol_Release(protocol); hres = IInternetSession_UnregisterNameSpace(session, &ClassFactory, wsz_test); ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres); hres = IInternetSession_CreateBinding(session, NULL, test_url, NULL, NULL, &protocol, 0); ok(hres == S_OK, "CreateBinding failed: %08x\n", hres); ok(protocol != NULL, "protocol == NULL\n"); SET_EXPECT(QueryService_InternetProtocol); hres = IInternetProtocol_Start(protocol, test_url, &protocol_sink, &bind_info, 0, 0); ok(hres == MK_E_SYNTAX, "Start failed: %08x, expected MK_E_SYNTAX\n", hres); CHECK_CALLED(QueryService_InternetProtocol); IInternetProtocol_Release(protocol); IInternetSession_Release(session); } static void test_binding(int prot, DWORD grf_pi, DWORD test_flags) { IInternetProtocolEx *protocolex = NULL; IInternetProtocol *protocol; IInternetSession *session; IUri *uri = NULL; ULONG ref; HRESULT hres; pi = grf_pi; init_test(prot, test_flags|TEST_BINDING); hres = pCoInternetGetSession(0, &session, 0); ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres); if(test_flags & TEST_EMULATEPROT) { hres = IInternetSession_RegisterNameSpace(session, &ClassFactory, &IID_NULL, protocol_names[prot], 0, NULL, 0); ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres); } hres = IInternetSession_CreateBinding(session, NULL, binding_urls[prot], NULL, NULL, &protocol, 0); binding_protocol = protocol; ok(hres == S_OK, "CreateBinding failed: %08x\n", hres); ok(protocol != NULL, "protocol == NULL\n"); hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetBindInfo, (void**)&prot_bind_info); ok(hres == S_OK, "QueryInterface(IID_IInternetBindInfo) failed: %08x\n", hres); hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetProtocolSink, (void**)&binding_sink); ok(hres == S_OK, "QueryInterface(IID_IInternetProtocolSink) failed: %08x\n", hres); if(test_flags & TEST_USEIURI) { hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetProtocolEx, (void**)&protocolex); ok(hres == S_OK, "Could not get IInternetProtocolEx iface: %08x\n", hres); hres = pCreateUri(binding_urls[prot], Uri_CREATE_FILE_USE_DOS_PATH, 0, &uri); ok(hres == S_OK, "CreateUri failed: %08x\n", hres); } ex_priority = 0; SET_EXPECT(QueryService_InternetProtocol); SET_EXPECT(CreateInstance); SET_EXPECT(ReportProgress_PROTOCOLCLASSID); SET_EXPECT(SetPriority); if(impl_protex) SET_EXPECT(StartEx); else SET_EXPECT(Start); expect_hrResult = S_OK; if(protocolex) { hres = IInternetProtocolEx_StartEx(protocolex, uri, &protocol_sink, &bind_info, pi, 0); ok(hres == S_OK, "StartEx failed: %08x\n", hres); }else { hres = IInternetProtocol_Start(protocol, binding_urls[prot], &protocol_sink, &bind_info, pi, 0); ok(hres == S_OK, "Start failed: %08x\n", hres); } CHECK_CALLED(QueryService_InternetProtocol); CHECK_CALLED(CreateInstance); CHECK_CALLED(ReportProgress_PROTOCOLCLASSID); CLEAR_CALLED(SetPriority); /* IE11 does not call it. */ if(impl_protex) CHECK_CALLED(StartEx); else CHECK_CALLED(Start); if(protocolex) IInternetProtocolEx_Release(protocolex); if(uri) IUri_Release(uri); if(prot == HTTP_TEST || prot == HTTPS_TEST) { while(prot_state < 4) { ok( WaitForSingleObject(event_complete, 90000) == WAIT_OBJECT_0, "wait timed out\n" ); if(mimefilter_test && filtered_protocol) { SET_EXPECT(Continue); IInternetProtocol_Continue(filtered_protocol, pdata); CHECK_CALLED(Continue); }else { SET_EXPECT(Continue); IInternetProtocol_Continue(protocol, pdata); CHECK_CALLED(Continue); } if(test_abort && prot_state == 2) { SET_EXPECT(Abort); hres = IInternetProtocol_Abort(protocol, E_ABORT, 0); ok(hres == S_OK, "Abort failed: %08x\n", hres); CHECK_CALLED(Abort); hres = IInternetProtocol_Abort(protocol, E_ABORT, 0); ok(hres == S_OK, "Abort failed: %08x\n", hres); SetEvent(event_complete2); break; } SetEvent(event_complete2); } if(direct_read) CHECK_CALLED(ReportData); /* Set in ReportResult */ ok( WaitForSingleObject(event_complete, 90000) == WAIT_OBJECT_0, "wait timed out\n" ); }else { if(mimefilter_test) SET_EXPECT(MimeFilter_LockRequest); else SET_EXPECT(LockRequest); hres = IInternetProtocol_LockRequest(protocol, 0); ok(hres == S_OK, "LockRequest failed: %08x\n", hres); if(mimefilter_test) CHECK_CALLED(MimeFilter_LockRequest); else CHECK_CALLED(LockRequest); if(mimefilter_test) SET_EXPECT(MimeFilter_UnlockRequest); else SET_EXPECT(UnlockRequest); hres = IInternetProtocol_UnlockRequest(protocol); ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres); if(mimefilter_test) CHECK_CALLED(MimeFilter_UnlockRequest); else CHECK_CALLED(UnlockRequest); } if(mimefilter_test) SET_EXPECT(MimeFilter_Terminate); else SET_EXPECT(Terminate); hres = IInternetProtocol_Terminate(protocol, 0); ok(hres == S_OK, "Terminate failed: %08x\n", hres); if(mimefilter_test) CLEAR_CALLED(MimeFilter_Terminate); else CHECK_CALLED(Terminate); if(filtered_protocol) IInternetProtocol_Release(filtered_protocol); IInternetBindInfo_Release(prot_bind_info); IInternetProtocolSink_Release(binding_sink); ref = IInternetProtocol_Release(protocol); ok(!ref, "ref=%u, expected 0\n", ref); if(test_flags & TEST_EMULATEPROT) { hres = IInternetSession_UnregisterNameSpace(session, &ClassFactory, protocol_names[prot]); ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres); } IInternetSession_Release(session); } START_TEST(protocol) { HMODULE hurlmon; hurlmon = GetModuleHandleA("urlmon.dll"); pCoInternetGetSession = (void*) GetProcAddress(hurlmon, "CoInternetGetSession"); pReleaseBindInfo = (void*) GetProcAddress(hurlmon, "ReleaseBindInfo"); pCreateUri = (void*) GetProcAddress(hurlmon, "CreateUri"); if(!GetProcAddress(hurlmon, "CompareSecurityIds")) { win_skip("Various needed functions not present, too old IE\n"); return; } if(!pCreateUri) win_skip("CreateUri not supported\n"); OleInitialize(NULL); event_complete = CreateEventW(NULL, FALSE, FALSE, NULL); event_complete2 = CreateEventW(NULL, FALSE, FALSE, NULL); event_continue = CreateEventW(NULL, FALSE, FALSE, NULL); event_continue_done = CreateEventW(NULL, FALSE, FALSE, NULL); thread_id = GetCurrentThreadId(); test_file_protocol(); test_http_protocol(); if(pCreateUri) test_https_protocol(); else win_skip("Skipping https tests on too old platform\n"); test_ftp_protocol(); test_gopher_protocol(); test_mk_protocol(); test_CreateBinding(); bindf &= ~BINDF_FROMURLMON; trace("Testing file binding (mime verification, emulate prot)...\n"); test_binding(FILE_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT); trace("Testing http binding (mime verification, emulate prot)...\n"); test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT); trace("Testing its binding (mime verification, emulate prot)...\n"); test_binding(ITS_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT); trace("Testing http binding (mime verification, emulate prot, short read, direct read)...\n"); test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_SHORT_READ|TEST_DIRECT_READ); trace("Testing file binding (mime verification, emulate prot, mime filter)...\n"); test_binding(FILE_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_FILTER); trace("Testing http binding (mime verification, emulate prot, mime filter)...\n"); test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_FILTER); trace("Testing http binding (mime verification, emulate prot, mime filter, no mime)...\n"); test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_FILTER|TEST_NOMIME); trace("Testing http binding (mime verification, emulate prot, direct read)...\n"); test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_DIRECT_READ); trace("Testing http binding (mime verification, emulate prot, abort)...\n"); test_binding(HTTP_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_ABORT); if(pCreateUri) { trace("Testing file binding (use IUri, mime verification, emulate prot)...\n"); test_binding(FILE_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_USEIURI); trace("Testing file binding (use IUri, impl StartEx, mime verification, emulate prot)...\n"); test_binding(FILE_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_USEIURI|TEST_IMPLPROTEX); trace("Testing file binding (impl StartEx, mime verification, emulate prot)...\n"); test_binding(FILE_TEST, PI_MIMEVERIFICATION, TEST_EMULATEPROT|TEST_IMPLPROTEX); } CloseHandle(event_complete); CloseHandle(event_complete2); CloseHandle(event_continue); CloseHandle(event_continue_done); OleUninitialize(); }