/* * Copyright 2005-2006 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 #define NONAMELESSUNION #include #include #include #include #include "windef.h" #include "winbase.h" #include "ole2.h" #include "urlmon.h" #include "initguid.h" #include "wine/heap.h" DEFINE_GUID(CLSID_AboutProtocol, 0x3050F406, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B); #define DEFINE_EXPECT(func) \ static BOOL expect_ ## func = FALSE, called_ ## func = FALSE #define SET_EXPECT(func) \ expect_ ## func = TRUE #define CHECK_EXPECT(func) \ do { \ ok(expect_ ##func, "unexpected call " #func "\n"); \ expect_ ## func = FALSE; \ called_ ## func = TRUE; \ }while(0) #define CHECK_EXPECT2(func) \ do { \ ok(expect_ ##func, "unexpected call " #func "\n"); \ called_ ## func = TRUE; \ }while(0) #define CHECK_CALLED(func) \ do { \ ok(called_ ## func, "expected " #func "\n"); \ expect_ ## func = called_ ## func = FALSE; \ }while(0) DEFINE_EXPECT(ParseUrl); DEFINE_EXPECT(ParseUrl_ENCODE); DEFINE_EXPECT(ParseUrl_UNESCAPE); DEFINE_EXPECT(QI_IInternetProtocolInfo); DEFINE_EXPECT(CreateInstance); DEFINE_EXPECT(unk_Release); static HRESULT (WINAPI *pCoInternetCompareUrl)(LPCWSTR, LPCWSTR, DWORD); static HRESULT (WINAPI *pCoInternetGetSecurityUrl)(LPCWSTR, LPWSTR*, PSUACTION, DWORD); static HRESULT (WINAPI *pCoInternetGetSession)(DWORD, IInternetSession **, DWORD); static HRESULT (WINAPI *pCoInternetParseUrl)(LPCWSTR, PARSEACTION, DWORD, LPWSTR, DWORD, DWORD *, DWORD); static HRESULT (WINAPI *pCoInternetQueryInfo)(LPCWSTR, QUERYOPTION, DWORD, LPVOID, DWORD, DWORD *, DWORD); static HRESULT (WINAPI *pCopyStgMedium)(const STGMEDIUM *, STGMEDIUM *); static HRESULT (WINAPI *pCopyBindInfo)(const BINDINFO *, BINDINFO *); static HRESULT (WINAPI *pFindMimeFromData)(LPBC, LPCWSTR, LPVOID, DWORD, LPCWSTR, DWORD, LPWSTR*, DWORD); static HRESULT (WINAPI *pObtainUserAgentString)(DWORD, LPSTR, DWORD*); static HRESULT (WINAPI *pReleaseBindInfo)(BINDINFO*); static HRESULT (WINAPI *pUrlMkGetSessionOption)(DWORD, LPVOID, DWORD, DWORD *, DWORD); static HRESULT (WINAPI *pCompareSecurityIds)(BYTE*,DWORD,BYTE*,DWORD,DWORD); static HRESULT (WINAPI *pCoInternetIsFeatureEnabled)(INTERNETFEATURELIST,DWORD); static HRESULT (WINAPI *pCoInternetSetFeatureEnabled)(INTERNETFEATURELIST,DWORD,BOOL); static HRESULT (WINAPI *pIEInstallScope)(DWORD*); static WCHAR *a2co(const char *str) { WCHAR *ret; int len; if(!str) return NULL; len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0); ret = CoTaskMemAlloc(len*sizeof(WCHAR)); MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len); return ret; } static void test_CreateFormatEnum(void) { IEnumFORMATETC *fenum = NULL, *fenum2 = NULL; FORMATETC fetc[5]; ULONG ul; HRESULT hres; static DVTARGETDEVICE dev = {sizeof(dev),0,0,0,0,{0}}; static FORMATETC formatetc[] = { {0,&dev,0,0,0}, {0,&dev,0,1,0}, {0,NULL,0,2,0}, {0,NULL,0,3,0}, {0,NULL,0,4,0} }; hres = CreateFormatEnumerator(0, formatetc, &fenum); ok(hres == E_FAIL, "CreateFormatEnumerator failed: %08x, expected E_FAIL\n", hres); hres = CreateFormatEnumerator(0, formatetc, NULL); ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres); hres = CreateFormatEnumerator(5, formatetc, NULL); ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres); hres = CreateFormatEnumerator(5, formatetc, &fenum); ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres); if(FAILED(hres)) return; hres = IEnumFORMATETC_Next(fenum, 2, NULL, &ul); ok(hres == E_INVALIDARG, "Next failed: %08x, expected E_INVALIDARG\n", hres); ul = 100; hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul); ok(hres == S_OK, "Next failed: %08x\n", hres); ok(ul == 0, "ul=%d, expected 0\n", ul); hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul); ok(hres == S_OK, "Next failed: %08x\n", hres); ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex); ok(fetc[1].lindex == 1, "fetc[1].lindex=%d, expected 1\n", fetc[1].lindex); ok(fetc[0].ptd == &dev, "fetc[0].ptd=%p, expected %p\n", fetc[0].ptd, &dev); ok(ul == 2, "ul=%d, expected 2\n", ul); hres = IEnumFORMATETC_Skip(fenum, 1); ok(hres == S_OK, "Skip failed: %08x\n", hres); hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul); ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres); ok(fetc[0].lindex == 3, "fetc[0].lindex=%d, expected 3\n", fetc[0].lindex); ok(fetc[1].lindex == 4, "fetc[1].lindex=%d, expected 4\n", fetc[1].lindex); ok(fetc[0].ptd == NULL, "fetc[0].ptd=%p, expected NULL\n", fetc[0].ptd); ok(ul == 2, "ul=%d, expected 2\n", ul); hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul); ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres); ok(ul == 0, "ul=%d, expected 0\n", ul); ul = 100; hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul); ok(hres == S_OK, "Next failed: %08x\n", hres); ok(ul == 0, "ul=%d, expected 0\n", ul); hres = IEnumFORMATETC_Skip(fenum, 3); ok(hres == S_FALSE, "Skip failed: %08x, expected S_FALSE\n", hres); hres = IEnumFORMATETC_Reset(fenum); ok(hres == S_OK, "Reset failed: %08x\n", hres); hres = IEnumFORMATETC_Next(fenum, 5, fetc, NULL); ok(hres == S_OK, "Next failed: %08x\n", hres); ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex); hres = IEnumFORMATETC_Reset(fenum); ok(hres == S_OK, "Reset failed: %08x\n", hres); hres = IEnumFORMATETC_Skip(fenum, 2); ok(hres == S_OK, "Skip failed: %08x\n", hres); hres = IEnumFORMATETC_Clone(fenum, NULL); ok(hres == E_INVALIDARG, "Clone failed: %08x, expected E_INVALIDARG\n", hres); hres = IEnumFORMATETC_Clone(fenum, &fenum2); ok(hres == S_OK, "Clone failed: %08x\n", hres); if(SUCCEEDED(hres)) { ok(fenum != fenum2, "fenum == fenum2\n"); hres = IEnumFORMATETC_Next(fenum2, 2, fetc, &ul); ok(hres == S_OK, "Next failed: %08x\n", hres); ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex); IEnumFORMATETC_Release(fenum2); } hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul); ok(hres == S_OK, "Next failed: %08x\n", hres); ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex); hres = IEnumFORMATETC_Skip(fenum, 1); ok(hres == S_OK, "Skip failed: %08x\n", hres); IEnumFORMATETC_Release(fenum); } static void test_RegisterFormatEnumerator(void) { IBindCtx *bctx = NULL; IEnumFORMATETC *format = NULL, *format2 = NULL; IUnknown *unk = NULL; HRESULT hres; static FORMATETC formatetc = {0,NULL,0,0,0}; static WCHAR wszEnumFORMATETC[] = {'_','E','n','u','m','F','O','R','M','A','T','E','T','C','_',0}; CreateBindCtx(0, &bctx); hres = CreateFormatEnumerator(1, &formatetc, &format); ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres); if(FAILED(hres)) return; hres = RegisterFormatEnumerator(NULL, format, 0); ok(hres == E_INVALIDARG, "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres); hres = RegisterFormatEnumerator(bctx, NULL, 0); ok(hres == E_INVALIDARG, "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres); hres = RegisterFormatEnumerator(bctx, format, 0); ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres); hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk); ok(hres == S_OK, "GetObjectParam failed: %08x\n", hres); ok(unk == (IUnknown*)format, "unk != format\n"); hres = RevokeFormatEnumerator(NULL, format); ok(hres == E_INVALIDARG, "RevokeFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres); hres = RevokeFormatEnumerator(bctx, format); ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres); hres = RevokeFormatEnumerator(bctx, format); ok(hres == E_FAIL, "RevokeFormatEnumerator failed: %08x, expected E_FAIL\n", hres); hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk); ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres); hres = RegisterFormatEnumerator(bctx, format, 0); ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres); hres = CreateFormatEnumerator(1, &formatetc, &format2); ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres); if(SUCCEEDED(hres)) { hres = RevokeFormatEnumerator(bctx, format); ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres); IEnumFORMATETC_Release(format2); } hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk); ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres); IEnumFORMATETC_Release(format); hres = RegisterFormatEnumerator(bctx, format, 0); ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres); hres = RevokeFormatEnumerator(bctx, NULL); ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres); hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk); ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres); IEnumFORMATETC_Release(format); IBindCtx_Release(bctx); } static const WCHAR url1[] = {'r','e','s',':','/','/','m','s','h','t','m','l','.','d','l','l', '/','b','l','a','n','k','.','h','t','m',0}; static const WCHAR url2[] = {'i','n','d','e','x','.','h','t','m',0}; static const WCHAR url3[] = {'f','i','l','e',':','/','/','c',':','\\','I','n','d','e','x','.','h','t','m',0}; static const WCHAR url4[] = {'f','i','l','e',':','s','o','m','e','%','2','0','f','i','l','e', '%','2','e','j','p','g',0}; static const WCHAR url5[] = {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q', '.','o','r','g',0}; static const WCHAR url6[] = {'a','b','o','u','t',':','b','l','a','n','k',0}; static const WCHAR url7[] = {'f','t','p',':','/','/','w','i','n','e','h','q','.','o','r','g','/', 'f','i','l','e','.','t','e','s','t',0}; static const WCHAR url8[] = {'t','e','s','t',':','1','2','3','a','b','c',0}; static const WCHAR url9[] = {'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}; static const WCHAR url10[] = {'h','t','t','p',':','/','/','g','o','o','g','l','e','.','*','.', 'c','o','m',0}; static const WCHAR url4e[] = {'f','i','l','e',':','s','o','m','e',' ','f','i','l','e', '.','j','p','g',0}; static const WCHAR path3[] = {'c',':','\\','I','n','d','e','x','.','h','t','m',0}; static const WCHAR path4[] = {'s','o','m','e',' ','f','i','l','e','.','j','p','g',0}; static const WCHAR wszRes[] = {'r','e','s',0}; static const WCHAR wszFile[] = {'f','i','l','e',0}; static const WCHAR wszHttp[] = {'h','t','t','p',0}; static const WCHAR wszAbout[] = {'a','b','o','u','t',0}; static const WCHAR wszEmpty[] = {0}; static const WCHAR wszGoogle[] = {'g','o','o','g','l','e','.','*','.','c','o','m',0}; static const WCHAR wszWineHQ[] = {'w','w','w','.','w','i','n','e','h','q','.','o','r','g',0}; static const WCHAR wszHttpWineHQ[] = {'h','t','t','p',':','/','/','w','w','w','.', 'w','i','n','e','h','q','.','o','r','g',0}; static const WCHAR wszHttpGoogle[] = {'h','t','t','p',':','/','/','g','o','o','g','l','e', '.','*','.','c','o','m',0}; struct parse_test { LPCWSTR url; HRESULT secur_hres; LPCWSTR encoded_url; HRESULT path_hres; LPCWSTR path; LPCWSTR schema; LPCWSTR domain; HRESULT domain_hres; LPCWSTR rootdocument; HRESULT rootdocument_hres; }; static const struct parse_test parse_tests[] = { {url1, S_OK, url1, E_INVALIDARG, NULL, wszRes, NULL, E_FAIL, NULL, E_FAIL}, {url2, E_FAIL, url2, E_INVALIDARG, NULL, wszEmpty, NULL, E_FAIL, NULL, E_FAIL}, {url3, E_FAIL, url3, S_OK, path3, wszFile, wszEmpty, S_OK, NULL, E_FAIL}, {url4, E_FAIL, url4e, S_OK, path4, wszFile, wszEmpty, S_OK, NULL, E_FAIL}, {url5, E_FAIL, url5, E_INVALIDARG, NULL, wszHttp, wszWineHQ, S_OK, wszHttpWineHQ, S_OK}, {url6, S_OK, url6, E_INVALIDARG, NULL, wszAbout, NULL, E_FAIL, NULL, E_FAIL}, {url10, E_FAIL, url10, E_INVALIDARG,NULL, wszHttp, wszGoogle, S_OK, wszHttpGoogle, S_OK} }; static void test_CoInternetParseUrl(void) { HRESULT hres; DWORD size; int i; static WCHAR buf[4096]; memset(buf, 0xf0, sizeof(buf)); hres = pCoInternetParseUrl(parse_tests[0].url, PARSE_SCHEMA, 0, buf, 3, &size, 0); ok(hres == E_POINTER, "schema failed: %08x, expected E_POINTER\n", hres); for(i = 0; i < ARRAY_SIZE(parse_tests); i++) { memset(buf, 0xf0, sizeof(buf)); hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_SECURITY_URL, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == parse_tests[i].secur_hres, "[%d] security url failed: %08x, expected %08x\n", i, hres, parse_tests[i].secur_hres); memset(buf, 0xf0, sizeof(buf)); hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_ENCODE, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == S_OK, "[%d] encoding failed: %08x\n", i, hres); ok(size == lstrlenW(parse_tests[i].encoded_url), "[%d] wrong size\n", i); ok(!lstrcmpW(parse_tests[i].encoded_url, buf), "[%d] wrong encoded url\n", i); memset(buf, 0xf0, sizeof(buf)); hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_UNESCAPE, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == S_OK, "[%d] encoding failed: %08x\n", i, hres); ok(size == lstrlenW(parse_tests[i].encoded_url), "[%d] wrong size\n", i); ok(!lstrcmpW(parse_tests[i].encoded_url, buf), "[%d] wrong encoded url\n", i); memset(buf, 0xf0, sizeof(buf)); hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_PATH_FROM_URL, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == parse_tests[i].path_hres, "[%d] path failed: %08x, expected %08x\n", i, hres, parse_tests[i].path_hres); if(parse_tests[i].path) { ok(size == lstrlenW(parse_tests[i].path), "[%d] wrong size\n", i); ok(!lstrcmpW(parse_tests[i].path, buf), "[%d] wrong path\n", i); } memset(buf, 0xf0, sizeof(buf)); hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_SCHEMA, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == S_OK, "[%d] schema failed: %08x\n", i, hres); ok(size == lstrlenW(parse_tests[i].schema), "[%d] wrong size\n", i); ok(!lstrcmpW(parse_tests[i].schema, buf), "[%d] wrong schema\n", i); if(memcmp(parse_tests[i].url, wszRes, 3*sizeof(WCHAR)) && memcmp(parse_tests[i].url, wszAbout, 5*sizeof(WCHAR))) { memset(buf, 0xf0, sizeof(buf)); hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_DOMAIN, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == parse_tests[i].domain_hres, "[%d] domain failed: %08x\n", i, hres); if(parse_tests[i].domain) ok(!lstrcmpW(parse_tests[i].domain, buf), "[%d] wrong domain, received %s\n", i, wine_dbgstr_w(buf)); } memset(buf, 0xf0, sizeof(buf)); hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_ROOTDOCUMENT, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == parse_tests[i].rootdocument_hres, "[%d] rootdocument failed: %08x\n", i, hres); if(parse_tests[i].rootdocument) ok(!lstrcmpW(parse_tests[i].rootdocument, buf), "[%d] wrong rootdocument, received %s\n", i, wine_dbgstr_w(buf)); } } static void test_CoInternetCompareUrl(void) { HRESULT hres; hres = pCoInternetCompareUrl(url1, url1, 0); ok(hres == S_OK, "CoInternetCompareUrl failed: %08x\n", hres); hres = pCoInternetCompareUrl(url1, url3, 0); ok(hres == S_FALSE, "CoInternetCompareUrl failed: %08x\n", hres); hres = pCoInternetCompareUrl(url3, url1, 0); ok(hres == S_FALSE, "CoInternetCompareUrl failed: %08x\n", hres); } static const struct { LPCWSTR url; DWORD uses_net; } query_info_tests[] = { {url1, 0}, {url2, 0}, {url3, 0}, {url4, 0}, {url5, 0}, {url6, 0}, {url7, 0}, {url8, 0} }; static void test_CoInternetQueryInfo(void) { BYTE buf[100]; DWORD cb, i; HRESULT hres; for(i = 0; i < ARRAY_SIZE(query_info_tests); i++) { cb = 0xdeadbeef; memset(buf, '?', sizeof(buf)); hres = pCoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, sizeof(buf), &cb, 0); ok(hres == S_OK, "[%d] CoInternetQueryInfo failed: %08x\n", i, hres); ok(cb == sizeof(DWORD), "[%d] cb = %d\n", i, cb); ok(*(DWORD*)buf == query_info_tests[i].uses_net, "[%d] ret %x, expected %x\n", i, *(DWORD*)buf, query_info_tests[i].uses_net); hres = pCoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, 3, &cb, 0); ok(hres == E_FAIL, "[%d] CoInternetQueryInfo failed: %08x, expected E_FAIL\n", i, hres); hres = pCoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, NULL, sizeof(buf), &cb, 0); ok(hres == E_FAIL, "[%d] CoInternetQueryInfo failed: %08x, expected E_FAIL\n", i, hres); memset(buf, '?', sizeof(buf)); hres = pCoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, sizeof(buf), NULL, 0); ok(hres == S_OK, "[%d] CoInternetQueryInfo failed: %08x\n", i, hres); ok(*(DWORD*)buf == query_info_tests[i].uses_net, "[%d] ret %x, expected %x\n", i, *(DWORD*)buf, query_info_tests[i].uses_net); } } static const struct { const WCHAR *url; const WCHAR *mime; HRESULT hres; BOOL broken_failure; const WCHAR *broken_mime; } mime_tests[] = { {L"res://mshtml.dll/blank.htm", L"text/html", S_OK}, {L"index.htm", L"text/html", S_OK}, {L"file://c:\\Index.htm", L"text/html", S_OK}, {L"file://c:\\Index.htm?q=test", L"text/html", S_OK, TRUE}, {L"file://c:\\Index.htm#hash_part", L"text/html", S_OK, TRUE}, {L"file://c:\\Index.htm#hash_part.txt", L"text/html", S_OK, FALSE, L"text/plain"}, {L"file://some%20file%2ejpg", NULL, E_FAIL}, {L"http://www.winehq.org", NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)}, {L"about:blank", NULL, E_FAIL}, {L"ftp://winehq.org/file.test", NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)} }; static BYTE data1[] = "test data\n"; static BYTE data2[] = {31,'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0}; static BYTE data3[] = {0,0,0}; static BYTE data4[] = {'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0,0}; static BYTE data5[] = {0xa,0xa,0xa,'x',32,'x',0}; static BYTE data6[] = {0xfa,0xfa,0xfa,0xfa,'\n','\r','\t','x','x','x',1}; static BYTE data7[] = "blahblah"; static BYTE data8[] = {'t','e','s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0}; static BYTE data9[] = {'t','e',0,'s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0}; static BYTE data10[] = "blahblah"; static BYTE data11[] = "blahblahblah"; static BYTE data12[] = "blah'}; static BYTE data19[] = {'G','I','F','8','7','a'}; static BYTE data20[] = {'G','I','F','8','9','a'}; static BYTE data21[] = {'G','I','F','8','7'}; static BYTE data22[] = {'G','i','F','8','7','a'}; static BYTE data23[] = {'G','i','F','8','8','a'}; static BYTE data24[] = {'g','i','f','8','7','a'}; static BYTE data25[] = {'G','i','F','8','7','A'}; static BYTE data26[] = {'G','i','F','8','7','a','<','h','t','m','l','>'}; static BYTE data27[] = {0x30,'G','i','F','8','7','A'}; static BYTE data28[] = {0x42,0x4d,0x6e,0x42,0x1c,0x00,0x00,0x00,0x00,0x00,0x36,0x00,0x00,0x00}; static BYTE data29[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x','x'}; static BYTE data30[] = {0x42,0x4d,'x','x','x','x',0x00,0x01,0x00,0x00,'x','x','x','x'}; static BYTE data31[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'<','h','t','m','l','>'}; static BYTE data32[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'}; static BYTE data33[] = {0x00,0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'}; static BYTE data34[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'}; static BYTE data35[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x','x','x','x',0}; static BYTE data36[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,'x','x'}; static BYTE data37[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'<','h','t','m','l','>'}; static BYTE data38[] = {0x00,0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'}; static BYTE data39[] = {0x4d,0x4d,0x00,0x2a,0xff}; static BYTE data40[] = {0x4d,0x4d,0x00,0x2a,'<','h','t','m','l','>',0}; static BYTE data41[] = {0x4d,0x4d,0xff}; static BYTE data42[] = {0x4d,0x4d}; static BYTE data43[] = {0x00,0x4d,0x4d,0x00}; static BYTE data44[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff}; static BYTE data45[] = {'R','I','F','f',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff}; static BYTE data46[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20}; static BYTE data47[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x21,0xff}; static BYTE data48[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,'<','h','t','m','l','>'}; static BYTE data49[] = {'R','I','F','F',0x0f,0x0f,0xf0,0xf0,'A','V','I',0x20,0xf0,0x00}; static BYTE data50[] = {0x00,0x00,0x01,0xb3,0xff}; static BYTE data51[] = {0x00,0x00,0x01,0xba,0xff}; static BYTE data52[] = {0x00,0x00,0x01,0xb8,0xff}; static BYTE data53[] = {0x00,0x00,0x01,0xba}; static BYTE data54[] = {0x00,0x00,0x01,0xba,'<','h','t','m','l','>'}; static BYTE data55[] = {0x1f,0x8b,'x'}; static BYTE data56[] = {0x1f}; static BYTE data57[] = {0x1f,0x8b,'<','h','t','m','l','>','t','e','s','t',0}; static BYTE data58[] = {0x1f,0x8b}; static BYTE data59[] = {0x50,0x4b,'x'}; static BYTE data60[] = {0x50,0x4b}; static BYTE data61[] = {0x50,0x4b,'<','h','t','m','l','>',0}; static BYTE data62[] = {0xca,0xfe,0xba,0xbe,'x'}; static BYTE data63[] = {0xca,0xfe,0xba,0xbe}; static BYTE data64[] = {0xca,0xfe,0xba,0xbe,'<','h','t','m','l','>',0}; static BYTE data65[] = {0x25,0x50,0x44,0x46,'x'}; static BYTE data66[] = {0x25,0x50,0x44,0x46}; static BYTE data67[] = {0x25,0x50,0x44,0x46,'x','<','h','t','m','l','>'}; static BYTE data68[] = {'M','Z','x'}; static BYTE data69[] = {'M','Z'}; static BYTE data70[] = {'M','Z','<','h','t','m','l','>',0xff}; static BYTE data71[] = {'{','\\','r','t','f',0}; static BYTE data72[] = {'{','\\','r','t','f'}; static BYTE data73[] = {' ','{','\\','r','t','f',' '}; static BYTE data74[] = {'{','\\','r','t','f','<','h','t','m','l','>',' '}; static BYTE data75[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V','E',0xff}; static BYTE data76[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V','E'}; static BYTE data77[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V',0xff,0xff}; static BYTE data78[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'<','h','t','m','l','>',0xff}; static BYTE data79[] = {'%','!',0xff}; static BYTE data80[] = {'%','!'}; static BYTE data81[] = {'%','!','P','S','<','h','t','m','l','>'}; static BYTE data82[] = {'.','s','n','d',0}; static BYTE data83[] = {'.','s','n','d'}; static BYTE data84[] = {'.','s','n','d',0,'<','h','t','m','l','>',1,1}; static BYTE data85[] = {'.','S','N','D',0}; static BYTE data86[] = {0x49,0x49,0x2a,0xff}; static BYTE data87[] = {' ','<','h','e','a','d'}; static BYTE data88[] = {' ','<','h','e','a','d','>'}; static BYTE data89[] = {'\t','\r','<','h','e','a','d','>'}; static BYTE data90[] = {'<','H','e','A','d',' '}; static BYTE data91[] = {'<','?','x','m','l',' ',0}; static BYTE data92[] = {'a','b','c','<','?','x','m','l',' ',' '}; static BYTE data93[] = {'<','?','x','m','l',' ',' ','<','h','t','m','l','>'}; static BYTE data94[] = {'<','h','t','m','l','>','<','?','x','m','l',' ',' '}; static BYTE data95[] = {'{','\\','r','t','f','<','?','x','m','l',' ',' '}; static BYTE data96[] = {'<','?','x','m','l',' '}; static BYTE data97[] = " ARRAY_SIZE(wszFile) && !memcmp(sec_url, wszFile, sizeof(wszFile)-sizeof(WCHAR)), "Encoded url = %s\n", wine_dbgstr_w(sec_url)); CoTaskMemFree(sec_url); } CHECK_CALLED(QI_IInternetProtocolInfo); CHECK_CALLED(ParseUrl); } hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest); ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres); hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres); hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf2, &IID_NULL, wszTest, 0, NULL, 0); ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres); hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL, wszTest, 0, NULL, 0); ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres); hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL, wszTest, 0, NULL, 0); ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres); SET_EXPECT(QI_IInternetProtocolInfo); SET_EXPECT(ParseUrl_ENCODE); hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres); CHECK_CALLED(QI_IInternetProtocolInfo); CHECK_CALLED(ParseUrl_ENCODE); hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest); ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres); SET_EXPECT(QI_IInternetProtocolInfo); SET_EXPECT(ParseUrl_ENCODE); hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres); CHECK_CALLED(QI_IInternetProtocolInfo); CHECK_CALLED(ParseUrl_ENCODE); hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest); ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres); expect_cf = &test_protocol_cf2; SET_EXPECT(QI_IInternetProtocolInfo); SET_EXPECT(ParseUrl_ENCODE); hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres); CHECK_CALLED(QI_IInternetProtocolInfo); CHECK_CALLED(ParseUrl_ENCODE); hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest); ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres); hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest); ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres); hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, NULL); ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres); hres = IInternetSession_UnregisterNameSpace(session, NULL, wszTest); ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres); hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf2, wszTest); ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres); hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, ARRAY_SIZE(buf), &size, 0); ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres); IInternetSession_Release(session); } static void test_MimeFilter(void) { IInternetSession *session; HRESULT hres; static const WCHAR mimeW[] = {'t','e','s','t','/','m','i','m','e',0}; hres = pCoInternetGetSession(0, &session, 0); ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres); if(FAILED(hres)) return; hres = IInternetSession_RegisterMimeFilter(session, &test_cf, &IID_NULL, mimeW); ok(hres == S_OK, "RegisterMimeFilter failed: %08x\n", hres); hres = IInternetSession_UnregisterMimeFilter(session, &test_cf, mimeW); ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres); hres = IInternetSession_UnregisterMimeFilter(session, &test_cf, mimeW); ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres); hres = IInternetSession_UnregisterMimeFilter(session, (void*)0xdeadbeef, mimeW); ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres); IInternetSession_Release(session); } static ULONG WINAPI unk_Release(IUnknown *iface) { CHECK_EXPECT(unk_Release); return 0; } static const IUnknownVtbl unk_vtbl = { (void*)0xdeadbeef, (void*)0xdeadbeef, unk_Release }; static void test_ReleaseBindInfo(void) { BINDINFO bi; IUnknown unk = { &unk_vtbl }; pReleaseBindInfo(NULL); /* shouldn't crash */ memset(&bi, 0, sizeof(bi)); bi.cbSize = sizeof(BINDINFO); bi.pUnk = &unk; SET_EXPECT(unk_Release); pReleaseBindInfo(&bi); ok(bi.cbSize == sizeof(BINDINFO), "bi.cbSize=%d\n", bi.cbSize); ok(bi.pUnk == NULL, "bi.pUnk=%p, expected NULL\n", bi.pUnk); CHECK_CALLED(unk_Release); memset(&bi, 0, sizeof(bi)); bi.cbSize = offsetof(BINDINFO, pUnk); bi.pUnk = &unk; pReleaseBindInfo(&bi); ok(bi.cbSize == offsetof(BINDINFO, pUnk), "bi.cbSize=%d\n", bi.cbSize); ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk); memset(&bi, 0, sizeof(bi)); bi.pUnk = &unk; pReleaseBindInfo(&bi); ok(!bi.cbSize, "bi.cbSize=%d, expected 0\n", bi.cbSize); ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk); } static void test_CopyStgMedium(void) { STGMEDIUM src, dst; HGLOBAL empty, hg; char *ptr1, *ptr2; HRESULT hres; int size; static WCHAR fileW[] = {'f','i','l','e',0}; memset(&src, 0xf0, sizeof(src)); memset(&dst, 0xe0, sizeof(dst)); memset(&empty, 0xf0, sizeof(empty)); src.tymed = TYMED_NULL; src.pUnkForRelease = NULL; hres = pCopyStgMedium(&src, &dst); ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres); ok(dst.tymed == TYMED_NULL, "tymed=%d\n", dst.tymed); ok(dst.u.hGlobal == empty, "u=%p\n", dst.u.hGlobal); ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease); memset(&dst, 0xe0, sizeof(dst)); src.tymed = TYMED_ISTREAM; src.u.pstm = NULL; src.pUnkForRelease = NULL; hres = pCopyStgMedium(&src, &dst); ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres); ok(dst.tymed == TYMED_ISTREAM, "tymed=%d\n", dst.tymed); ok(!dst.u.pstm, "pstm=%p\n", dst.u.pstm); ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease); memset(&dst, 0xe0, sizeof(dst)); src.tymed = TYMED_FILE; src.u.lpszFileName = fileW; src.pUnkForRelease = NULL; hres = pCopyStgMedium(&src, &dst); ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres); ok(dst.tymed == TYMED_FILE, "tymed=%d\n", dst.tymed); ok(dst.u.lpszFileName && dst.u.lpszFileName != fileW, "lpszFileName=%p\n", dst.u.lpszFileName); ok(!lstrcmpW(dst.u.lpszFileName, fileW), "wrong file name\n"); ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease); ReleaseStgMedium(&dst); /* TYMED_HGLOBAL */ hg = GlobalAlloc(GMEM_MOVEABLE, 10); ptr1 = GlobalLock(hg); memset(ptr1, 0xfa, 10); memset(&dst, 0xe0, sizeof(dst)); src.tymed = TYMED_HGLOBAL; src.u.hGlobal = hg; hres = pCopyStgMedium(&src, &dst); ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres); ok(dst.tymed == TYMED_HGLOBAL, "tymed=%d\n", dst.tymed); ok(dst.u.hGlobal != hg, "got %p, %p\n", dst.u.hGlobal, hg); size = GlobalSize(dst.u.hGlobal); ok(size == 10, "got size %d\n", size); /* compare contents */ ptr2 = GlobalLock(dst.u.hGlobal); ok(!memcmp(ptr1, ptr2, 10), "got wrong data\n"); GlobalUnlock(ptr2); GlobalUnlock(ptr1); ok(GlobalFlags(dst.u.hGlobal) == 0, "got 0x%08x\n", GlobalFlags(dst.u.hGlobal)); GlobalFree(hg); ReleaseStgMedium(&dst); memset(&dst, 0xe0, sizeof(dst)); src.tymed = TYMED_HGLOBAL; src.u.hGlobal = NULL; hres = pCopyStgMedium(&src, &dst); ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres); ok(dst.u.hGlobal == NULL, "got %p\n", dst.u.hGlobal); hres = pCopyStgMedium(&src, NULL); ok(hres == E_POINTER, "CopyStgMedium failed: %08x, expected E_POINTER\n", hres); hres = pCopyStgMedium(NULL, &dst); ok(hres == E_POINTER, "CopyStgMedium failed: %08x, expected E_POINTER\n", hres); } static void test_CopyBindInfo(void) { BINDINFO src[2], dest[2]; SECURITY_DESCRIPTOR sec_desc; HRESULT hres; int i; hres = pCopyBindInfo(NULL, NULL); ok(hres == E_POINTER, "CopyBindInfo returned %08x, expected E_POINTER\n", hres); memset(src, 0, sizeof(BINDINFO[2])); memset(dest, 0xde, sizeof(BINDINFO[2])); hres = pCopyBindInfo(src, dest); ok(hres == E_INVALIDARG, "CopyBindInfo returned: %08x, expected E_INVALIDARG\n", hres); memset(src, 0, sizeof(BINDINFO[2])); memset(dest, 0xde, sizeof(BINDINFO[2])); src[0].cbSize = sizeof(BINDINFO); dest[0].cbSize = 0; hres = pCopyBindInfo(src, dest); ok(hres == E_INVALIDARG, "CopyBindInfo returned: %08x, expected E_INVALIDARG\n", hres); memset(src, 0, sizeof(BINDINFO[2])); memset(dest, 0xde, sizeof(BINDINFO[2])); src[0].cbSize = 1; dest[0].cbSize = sizeof(BINDINFO)+sizeof(DWORD); hres = pCopyBindInfo(src, dest); ok(hres == S_OK, "CopyBindInfo failed: %08x\n", hres); ok(dest[0].cbSize == sizeof(BINDINFO)+sizeof(DWORD), "incorrect cbSize: %d\n", dest[0].cbSize); for(i=1; i 0, "size=%d, expected non-zero\n", size); size = 2; str[0] = 'a'; hres = pObtainUserAgentString(0, str, &size); ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres); ok(size > 0, "size=%d, expected non-zero\n", size); ok(str[0] == 'a', "str[0]=%c, expected 'a'\n", str[0]); size = 0; hres = pObtainUserAgentString(1, str, &size); ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres); ok(size > 0, "size=%d, expected non-zero\n", size); str2 = HeapAlloc(GetProcessHeap(), 0, (size+20)*sizeof(CHAR)); saved = size; hres = pObtainUserAgentString(0, str2, &size); ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres); ok(size == saved, "size=%d, expected %d\n", size, saved); ok(strlen(expected) <= strlen(str2) && !memcmp(expected, str2, strlen(expected)*sizeof(CHAR)), "user agent was \"%s\", expected to start with \"%s\"\n", str2, expected); GetVersionExW(&os_info); if (sizeof(void*) == 4) IsWow64Process(GetCurrentProcess(), &is_wow); for(i = 1; i < 12; i++) { const char *p = ua; if (i != 7) { size = sizeof(ua); hres = pObtainUserAgentString(i | UAS_EXACTLEGACY, ua, &size); ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres); ok(size == strlen(ua) + 1, "unexpected size %u, expected %u\n", size, strlen(ua) + 1); ok(!strcmp(ua, str2), "unexpected UA for version %u %s, expected %s\n", i, wine_dbgstr_a(ua), wine_dbgstr_a(str2)); } size = sizeof(ua); hres = pObtainUserAgentString(i != 1 ? i : UAS_EXACTLEGACY | 7, ua, &size); ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres); ok(size == strlen(ua) + 1, "unexpected size %u, expected %u\n", size, strlen(ua) + 1); if(i < 8 && i != 1) ok(!strcmp(ua, str2), "unexpected UA for version %u %s, expected %s\n", i, wine_dbgstr_a(ua), wine_dbgstr_a(str2)); if (winetest_debug > 1) trace("version=%u user-agent=%s\n", i, wine_dbgstr_a(ua)); p += check_prefix(p, "Mozilla/"); p += check_prefix(p, i < 9 ? "4.0 (" : "5.0 ("); if(i < 11) { p += check_prefix(p, "compatible; "); sprintf(buf, "MSIE %u.0; ", max(i, 7)); p += check_prefix(p, buf); } sprintf(buf, "Windows NT %u.%u; ", os_info.dwMajorVersion, os_info.dwMinorVersion); p += check_prefix(p, buf); if(is_wow) { p += check_prefix(p, "WOW64; "); }else if(sizeof(void*) == 8) { p += check_prefix(p, "Win64; "); #ifdef __x86_64__ todo_wine p += check_prefix(p, "x64; "); #endif } if(i != 1) { p += check_prefix(p, "Trident/"); ok('5' <= *p && *p <= '8', "unexpected version '%c'\n", *p); if(*p < '7') { win_skip("skipping UA tests, too old IE\n"); break; } p++; /* skip version number */ p += check_prefix(p, ".0"); } if(i == 11) { p += check_prefix(p, "; rv:11.0) like Gecko"); }else if (i >= 9) { p += check_prefix(p, ")"); }else { #ifdef __x86_64__ todo_wine #endif /* This assumes that either p points at some property before * '; .NET', or that there is more than one such occurrence. */ ok(strstr(p, "; .NET") != NULL, "no '; .NET' in %s\n", wine_dbgstr_a(p)); p = strchr(p, ')'); p += check_prefix(p, ")"); } ok(!*p, "unexpected suffix %s for version %u\n", wine_dbgstr_a(p), i); } size = saved+10; hres = pObtainUserAgentString(0, str2, &size); ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres); ok(size == saved, "size=%d, expected %d\n", size, saved); size = 0; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, NULL, 0, &size, 0); ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres); ok(size, "size == 0\n"); size = 0xdeadbeef; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, NULL, 1000, &size, 0); ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres); ok(size, "size == 0\n"); saved = size; size = 0; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved+10, &size, 0); ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres); ok(size == saved, "size = %d, expected %d\n", size, saved); ok(sizeof(expected) <= strlen(str2) && !memcmp(expected, str2, sizeof(expected)-1), "user agent was \"%s\", expected to start with \"%s\"\n", str2, expected); size = 0; str2[0] = 0; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, &size, 0); ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres); ok(size == saved, "size = %d, expected %d\n", size, saved); ok(sizeof(expected) <= strlen(str2) && !memcmp(expected, str2, sizeof(expected)-1), "user agent was \"%s\", expected to start with \"%s\"\n", str2, expected); size = saved; str2[0] = 0; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved-1, &size, 0); ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres); ok(size == saved, "size = %d, expected %d\n", size, saved); ok(!str2[0], "buf changed\n"); size = saved; str2[0] = 0; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, NULL, 0); ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres); ok(!str2[0], "buf changed\n"); hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, test_str, sizeof(test_str), 0); ok(hres == S_OK, "UrlMkSetSessionOption failed: %08x\n", hres); size = 0; str2[0] = 0; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, &size, 0); ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres); ok(size == sizeof(test_str) && !memcmp(str2, test_str, sizeof(test_str)), "wrong user agent\n"); hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, test2_str, sizeof(test2_str), 0); ok(hres == S_OK, "UrlMkSetSessionOption failed: %08x\n", hres); size = 0; str2[0] = 0; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, &size, 0); ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres); ok(size == sizeof(test_str) && !memcmp(str2, test_str, sizeof(test_str)), "wrong user agent\n"); hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, test_str, 2, 0); ok(hres == S_OK, "UrlMkSetSessionOption failed: %08x\n", hres); size = 0; str2[0] = 0; hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, &size, 0); ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres); ok(size == 3 && !strcmp(str2, "te"), "wrong user agent\n"); hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, test_str, 0, 0); ok(hres == E_INVALIDARG, "UrlMkSetSessionOption failed: %08x\n", hres); hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, NULL, sizeof(test_str), 0); ok(hres == E_INVALIDARG, "UrlMkSetSessionOption failed: %08x\n", hres); hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, NULL, 0, 0); ok(hres == E_INVALIDARG, "UrlMkSetSessionOption failed: %08x\n", hres); HeapFree(GetProcessHeap(), 0, str2); } static void test_MkParseDisplayNameEx(void) { IMoniker *mon = NULL; LPWSTR name; DWORD issys; ULONG eaten = 0; IBindCtx *bctx; HRESULT hres; static const WCHAR clsid_nameW[] = {'c','l','s','i','d',':', '2','0','D','0','4','F','E','0','-','3','A','E','A','-','1','0','6','9','-','A','2','D','8', '-','0','8','0','0','2','B','3','0','3','0','9','D',':',0}; const struct { LPBC *ppbc; LPCWSTR szDisplayName; ULONG *pchEaten; LPMONIKER *ppmk; } invalid_parameters[] = { {NULL, NULL, NULL, NULL}, {NULL, NULL, NULL, &mon}, {NULL, NULL, &eaten, NULL}, {NULL, NULL, &eaten, &mon}, {NULL, wszEmpty, NULL, NULL}, {NULL, wszEmpty, NULL, &mon}, {NULL, wszEmpty, &eaten, NULL}, {NULL, wszEmpty, &eaten, &mon}, {&bctx, NULL, NULL, NULL}, {&bctx, NULL, NULL, &mon}, {&bctx, NULL, &eaten, NULL}, {&bctx, NULL, &eaten, &mon}, {&bctx, wszEmpty, NULL, NULL}, {&bctx, wszEmpty, NULL, &mon}, {&bctx, wszEmpty, &eaten, NULL}, {&bctx, wszEmpty, &eaten, &mon}, }; int i; CreateBindCtx(0, &bctx); for (i = 0; i < ARRAY_SIZE(invalid_parameters); i++) { eaten = 0xdeadbeef; mon = (IMoniker *)0xdeadbeef; hres = MkParseDisplayNameEx(invalid_parameters[i].ppbc ? *invalid_parameters[i].ppbc : NULL, invalid_parameters[i].szDisplayName, invalid_parameters[i].pchEaten, invalid_parameters[i].ppmk); ok(hres == E_INVALIDARG, "[%d] Expected MkParseDisplayNameEx to return E_INVALIDARG, got %08x\n", i, hres); ok(eaten == 0xdeadbeef, "[%d] Expected eaten to be 0xdeadbeef, got %u\n", i, eaten); ok(mon == (IMoniker *)0xdeadbeef, "[%d] Expected mon to be 0xdeadbeef, got %p\n", i, mon); } hres = MkParseDisplayNameEx(bctx, url9, &eaten, &mon); ok(hres == S_OK, "MkParseDisplayNameEx failed: %08x\n", hres); ok(eaten == ARRAY_SIZE(url9)-1, "eaten=%d\n", eaten); ok(mon != NULL, "mon == NULL\n"); hres = IMoniker_GetDisplayName(mon, NULL, 0, &name); ok(hres == S_OK, "GetDisplayName failed: %08x\n", hres); ok(!lstrcmpW(name, url9), "wrong display name %s\n", wine_dbgstr_w(name)); CoTaskMemFree(name); hres = IMoniker_IsSystemMoniker(mon, &issys); ok(hres == S_OK, "IsSystemMoniker failed: %08x\n", hres); ok(issys == MKSYS_URLMONIKER, "issys=%x\n", issys); IMoniker_Release(mon); hres = MkParseDisplayNameEx(bctx, clsid_nameW, &eaten, &mon); ok(hres == S_OK, "MkParseDisplayNameEx failed: %08x\n", hres); ok(eaten == ARRAY_SIZE(clsid_nameW)-1, "eaten=%d\n", eaten); ok(mon != NULL, "mon == NULL\n"); hres = IMoniker_IsSystemMoniker(mon, &issys); ok(hres == S_OK, "IsSystemMoniker failed: %08x\n", hres); ok(issys == MKSYS_CLASSMONIKER, "issys=%x\n", issys); IMoniker_Release(mon); hres = MkParseDisplayNameEx(bctx, url8, &eaten, &mon); ok(FAILED(hres), "MkParseDisplayNameEx succeeded: %08x\n", hres); IBindCtx_Release(bctx); } static void test_IsValidURL(void) { HRESULT hr; IBindCtx *bctx = NULL; hr = IsValidURL(NULL, 0, 0); ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr); hr = IsValidURL(NULL, wszHttpWineHQ, 0); ok(hr == S_OK, "Expected S_OK, got %08x\n", hr); CreateBindCtx(0, &bctx); hr = IsValidURL(bctx, wszHttpWineHQ, 0); ok(hr == S_OK, "Expected S_OK, got %08x\n", hr); IBindCtx_Release(bctx); } static const struct { INTERNETFEATURELIST feature; DWORD get_flags; HRESULT expected; BOOL todo; } default_feature_tests[] = { {FEATURE_OBJECT_CACHING,GET_FEATURE_FROM_PROCESS,S_OK}, {FEATURE_ZONE_ELEVATION,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_MIME_HANDLING,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_MIME_SNIFFING,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_WINDOW_RESTRICTIONS,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_WEBOC_POPUPMANAGEMENT,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_BEHAVIORS,GET_FEATURE_FROM_PROCESS,S_OK}, {FEATURE_DISABLE_MK_PROTOCOL,GET_FEATURE_FROM_PROCESS,S_OK}, {FEATURE_LOCALMACHINE_LOCKDOWN,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_SECURITYBAND,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_RESTRICT_ACTIVEXINSTALL,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_VALIDATE_NAVIGATE_URL,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_RESTRICT_FILEDOWNLOAD,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_ADDON_MANAGEMENT,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_PROTOCOL_LOCKDOWN,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_HTTP_USERNAME_PASSWORD_DISABLE,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_SAFE_BINDTOOBJECT,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_UNC_SAVEDFILECHECK,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_GET_URL_DOM_FILEPATH_UNENCODED,GET_FEATURE_FROM_PROCESS,S_OK}, {FEATURE_TABBED_BROWSING,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_SSLUX,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_DISABLE_NAVIGATION_SOUNDS,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_DISABLE_LEGACY_COMPRESSION,GET_FEATURE_FROM_PROCESS,S_OK}, {FEATURE_FORCE_ADDR_AND_STATUS,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_XMLHTTP,GET_FEATURE_FROM_PROCESS,S_OK}, {FEATURE_DISABLE_TELNET_PROTOCOL,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_FEEDS,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_BLOCK_INPUT_PROMPTS,GET_FEATURE_FROM_PROCESS,S_FALSE} }; static void test_internet_feature_defaults(void) { HRESULT hres; DWORD i; for(i = 0; i < ARRAY_SIZE(default_feature_tests); ++i) { hres = pCoInternetIsFeatureEnabled(default_feature_tests[i].feature, default_feature_tests[i].get_flags); todo_wine_if (default_feature_tests[i].todo) ok(hres == default_feature_tests[i].expected, "CoInternetIsFeatureEnabled returned %08x, expected %08x on test %d\n", hres, default_feature_tests[i].expected, i); } } /* With older versions of IE (IE 7 and earlier), urlmon caches * the FeatureControl values from the registry when it's loaded * into memory. Newer versions of IE conditionally cache the * the FeatureControl registry values (i.e. When a call to * CoInternetIsFeatureEnabled and a corresponding CoInternetSetFeatureEnabled * call hasn't already been made for the specified Feature). Because of * this we skip these tests on IE 7 and earlier. */ static const char* szFeatureControlKey = "Software\\Microsoft\\Internet Explorer\\Main\\FeatureControl"; static void test_internet_features_registry(void) { HRESULT hres; DWORD res; char module[MAX_PATH]; char *name; HKEY feature_control; HKEY feature; DWORD value; BOOL skip_zone; BOOL delete_feature_key = TRUE; static const char* szFeatureBehaviorsKey = "FEATURE_BEHAVIORS"; static const char* szFeatureZoneElevationKey = "FEATURE_ZONE_ELEVATION"; if(!pIEInstallScope) { win_skip("Skipping internet feature registry tests, IE is too old...\n"); return; } res = GetModuleFileNameA(NULL, module, sizeof(module)); ok(res, "GetModuleFileName failed: %d\n", GetLastError()); name = strrchr(module, '\\')+1; /* Some Windows machines don't have a FeatureControl key in HKCU. */ res = RegOpenKeyA(HKEY_CURRENT_USER, szFeatureControlKey, &feature_control); ok(res == ERROR_SUCCESS, "RegCreateKey failed: %d\n", res); res = RegOpenKeyA(feature_control, szFeatureBehaviorsKey, &feature); if(res == ERROR_SUCCESS) { /* FEATURE_BEHAVIORS already existed, so don't delete it when we're done. */ delete_feature_key = FALSE; }else { res = RegCreateKeyA(feature_control, szFeatureBehaviorsKey, &feature); ok(res == ERROR_SUCCESS, "RegCreateKey failed: %d\n", res); } value = 0; res = RegSetValueExA(feature, name, 0, REG_DWORD, (BYTE*)&value, sizeof(DWORD)); ok(res == ERROR_SUCCESS, "RegSetValueEx failed: %d\n", res); hres = pCoInternetIsFeatureEnabled(FEATURE_BEHAVIORS, GET_FEATURE_FROM_PROCESS); ok(hres == S_FALSE, "CoInternetIsFeatureEnabled returned %08x, expected S_FALSE\n", hres); if(delete_feature_key) { RegCloseKey(feature); RegDeleteKeyA(feature_control, szFeatureBehaviorsKey); } else { RegDeleteValueA(feature, name); RegCloseKey(feature); } /* IE's feature control cached the value it got from the registry earlier. */ hres = pCoInternetIsFeatureEnabled(FEATURE_BEHAVIORS, GET_FEATURE_FROM_PROCESS); ok(hres == S_FALSE, "CoInternetIsFeatureEnabled returned %08x, expected S_FALSE\n", hres); /* Restore this feature back to its default value. */ hres = pCoInternetSetFeatureEnabled(FEATURE_BEHAVIORS, SET_FEATURE_ON_PROCESS, TRUE); ok(hres == S_OK, "CoInternetSetFeatureEnabled failed: %08x\n", hres); RegCloseKey(feature_control); res = RegOpenKeyA(HKEY_LOCAL_MACHINE, szFeatureControlKey, &feature_control); ok(res == ERROR_SUCCESS, "RegOpenKey failed: %d\n", res); res = RegOpenKeyA(feature_control, szFeatureZoneElevationKey, &feature); ok(res == ERROR_SUCCESS, "RegOpenKey failed: %d\n", res); value = 1; res = RegSetValueExA(feature, "*", 0, REG_DWORD, (BYTE*)&value, sizeof(DWORD)); if (res == ERROR_ACCESS_DENIED) { skip("Not allowed to modify zone elevation\n"); skip_zone = TRUE; } else { skip_zone = FALSE; ok(res == ERROR_SUCCESS, "RegSetValueEx failed: %d\n", res); hres = pCoInternetIsFeatureEnabled(FEATURE_ZONE_ELEVATION, GET_FEATURE_FROM_PROCESS); ok(hres == S_OK, "CoInternetIsFeatureEnabled returned %08x, expected S_OK\n", hres); } RegDeleteValueA(feature, "*"); RegCloseKey(feature); RegCloseKey(feature_control); /* Value is still cached from last time. */ if (!skip_zone) { hres = pCoInternetIsFeatureEnabled(FEATURE_ZONE_ELEVATION, GET_FEATURE_FROM_PROCESS); ok(hres == S_OK, "CoInternetIsFeatureEnabled returned %08x, expected S_OK\n", hres); hres = pCoInternetSetFeatureEnabled(FEATURE_ZONE_ELEVATION, SET_FEATURE_ON_PROCESS, FALSE); ok(hres == S_OK, "CoInternetSetFeatureEnabled failed: %08x\n", hres); } test_internet_feature_defaults(); } static void test_CoInternetIsFeatureEnabled(void) { HRESULT hres; hres = pCoInternetIsFeatureEnabled(FEATURE_ENTRY_COUNT, GET_FEATURE_FROM_PROCESS); ok(hres == E_FAIL, "CoInternetIsFeatureEnabled returned %08x, expected E_FAIL\n", hres); } static const struct { INTERNETFEATURELIST feature; DWORD set_flags; BOOL enable; HRESULT set_expected; BOOL set_todo; DWORD get_flags; HRESULT get_expected; BOOL get_todo; } internet_feature_tests[] = { {FEATURE_OBJECT_CACHING,SET_FEATURE_ON_PROCESS,FALSE,S_OK,FALSE,GET_FEATURE_FROM_PROCESS,S_FALSE}, {FEATURE_WEBOC_POPUPMANAGEMENT,SET_FEATURE_ON_PROCESS,TRUE,S_OK,FALSE,GET_FEATURE_FROM_PROCESS,S_OK}, {FEATURE_LOCALMACHINE_LOCKDOWN,SET_FEATURE_ON_PROCESS,TRUE,S_OK,FALSE,GET_FEATURE_FROM_PROCESS,S_OK} }; static void test_CoInternetSetFeatureEnabled(void) { HRESULT hres; DWORD i; hres = pCoInternetSetFeatureEnabled(FEATURE_ENTRY_COUNT,SET_FEATURE_ON_PROCESS,TRUE); ok(hres == E_FAIL, "CoInternetSetFeatureEnabled returned %08x, expected E_FAIL\n", hres); for(i = 0; i < ARRAY_SIZE(internet_feature_tests); ++i) { hres = pCoInternetSetFeatureEnabled(internet_feature_tests[i].feature, internet_feature_tests[i].set_flags, internet_feature_tests[i].enable); todo_wine_if (internet_feature_tests[i].set_todo) ok(hres == internet_feature_tests[i].set_expected, "CoInternetSetFeatureEnabled returned %08x, expected %08x on test %d\n", hres, internet_feature_tests[i].set_expected, i); hres = pCoInternetIsFeatureEnabled(internet_feature_tests[i].feature, internet_feature_tests[i].set_flags); todo_wine_if (internet_feature_tests[i].get_todo) ok(hres == internet_feature_tests[i].get_expected, "CoInternetIsFeatureEnabled returned %08x, expected %08x on test %d\n", hres, internet_feature_tests[i].get_expected, i); } } static void test_internet_features(void) { HKEY key; DWORD res; if(!pCoInternetIsFeatureEnabled || !pCoInternetSetFeatureEnabled) { win_skip("Skipping internet feature tests, IE is too old\n"); return; } /* IE10 takes FeatureControl key into account only if it's available upon process start. */ res = RegOpenKeyA(HKEY_CURRENT_USER, szFeatureControlKey, &key); if(res != ERROR_SUCCESS) { PROCESS_INFORMATION pi; STARTUPINFOA si = { 0 }; char cmdline[MAX_PATH]; char **argv; BOOL ret; res = RegCreateKeyA(HKEY_CURRENT_USER, szFeatureControlKey, &key); ok(res == ERROR_SUCCESS, "RegCreateKey failed: %d\n", res); trace("Running features tests in a separated process.\n"); winetest_get_mainargs( &argv ); sprintf(cmdline, "\"%s\" %s internet_features", argv[0], argv[1]); ret = CreateProcessA(argv[0], cmdline, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi); ok(ret, "Could not create process: %u\n", GetLastError()); wait_child_process( pi.hProcess ); CloseHandle(pi.hThread); CloseHandle(pi.hProcess); RegDeleteKeyA(HKEY_CURRENT_USER, szFeatureControlKey); return; } test_internet_features_registry(); test_CoInternetIsFeatureEnabled(); test_CoInternetSetFeatureEnabled(); } static BINDINFO rem_bindinfo = { sizeof(rem_bindinfo) }, in_bindinfo; static DWORD rem_bindf; static HRESULT WINAPI BindStatusCallback_QueryInterface(IBindStatusCallbackEx *iface, REFIID riid, void **ppv) { if(IsEqualGUID(&IID_IBindStatusCallbackEx, riid) || IsEqualGUID(&IID_IBindStatusCallback, riid) || IsEqualGUID(&IID_IUnknown, riid)) { *ppv = iface; return S_OK; } *ppv = NULL; return E_NOINTERFACE; } static ULONG WINAPI BindStatusCallback_AddRef(IBindStatusCallbackEx *iface) { return 2; } static ULONG WINAPI BindStatusCallback_Release(IBindStatusCallbackEx *iface) { return 1; } static HRESULT WINAPI BindStatusCallback_OnStartBinding(IBindStatusCallbackEx *iface, DWORD dwReserved, IBinding *pib) { ok(0, "unexpected call\n"); return S_OK; } static HRESULT WINAPI BindStatusCallback_GetPriority(IBindStatusCallbackEx *iface, LONG *pnPriority) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI BindStatusCallback_OnLowResource(IBindStatusCallbackEx *iface, DWORD reserved) { ok(0, "unexpected call\n"); return E_NOTIMPL; } static HRESULT WINAPI BindStatusCallback_OnProgress(IBindStatusCallbackEx *iface, ULONG ulProgress, ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText) { ok(0, "unexpected call\n"); return S_OK; } static HRESULT WINAPI BindStatusCallback_OnStopBinding(IBindStatusCallbackEx *iface, HRESULT hresult, LPCWSTR szError) { ok(0, "unexpected call\n"); return S_OK; } static HRESULT WINAPI BindStatusCallback_GetBindInfo(IBindStatusCallbackEx *iface, DWORD *grfBINDF, BINDINFO *pbindinfo) { in_bindinfo = *pbindinfo; *grfBINDF = rem_bindf; *pbindinfo = rem_bindinfo; return S_OK; } static STGMEDIUM in_stgmed, rem_stgmed; static HRESULT WINAPI BindStatusCallback_OnDataAvailable(IBindStatusCallbackEx *iface, DWORD grfBSCF, DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed) { in_stgmed = *pstgmed; *pstgmed = rem_stgmed; return S_OK; } static HRESULT WINAPI BindStatusCallback_OnObjectAvailable(IBindStatusCallbackEx *iface, REFIID riid, IUnknown *punk) { ok(0, "unexpected call\n"); return S_OK; } static HRESULT WINAPI BindStatusCallbackEx_GetBindInfoEx(IBindStatusCallbackEx *iface, DWORD *grfBINDF, BINDINFO *pbindinfo, DWORD *grfBINDF2, DWORD *pdwReserved) { in_bindinfo = *pbindinfo; *grfBINDF = rem_bindf; *pbindinfo = rem_bindinfo; *grfBINDF2 = 11; *pdwReserved = 12; return S_OK; } static const IBindStatusCallbackExVtbl BindStatusCallbackExVtbl = { BindStatusCallback_QueryInterface, BindStatusCallback_AddRef, BindStatusCallback_Release, BindStatusCallback_OnStartBinding, BindStatusCallback_GetPriority, BindStatusCallback_OnLowResource, BindStatusCallback_OnProgress, BindStatusCallback_OnStopBinding, BindStatusCallback_GetBindInfo, BindStatusCallback_OnDataAvailable, BindStatusCallback_OnObjectAvailable, BindStatusCallbackEx_GetBindInfoEx }; static IBindStatusCallbackEx BindStatusCallback = { &BindStatusCallbackExVtbl }; typedef struct { IUnknown IUnknown_iface; LONG ref; } RefUnk; static inline RefUnk *impl_from_IUnknown(IUnknown *iface) { return CONTAINING_RECORD(iface, RefUnk, IUnknown_iface); } static HRESULT WINAPI RefUnk_QueryInterface(IUnknown *iface, REFIID riid, void **ppv) { if(!IsEqualGUID(&IID_IUnknown, riid)) { *ppv = NULL; return E_NOINTERFACE; } IUnknown_AddRef(iface); *ppv = iface; return S_OK; } static ULONG WINAPI RefUnk_AddRef(IUnknown *iface) { RefUnk *This = impl_from_IUnknown(iface); return InterlockedIncrement(&This->ref); } static ULONG WINAPI RefUnk_Release(IUnknown *iface) { RefUnk *This = impl_from_IUnknown(iface); return InterlockedDecrement(&This->ref); } static const IUnknownVtbl RefUnkVtbl = { RefUnk_QueryInterface, RefUnk_AddRef, RefUnk_Release }; static RefUnk unk_in = {{&RefUnkVtbl}}, unk_out = {{&RefUnkVtbl}}; static HANDLE thread_ready; static DWORD WINAPI bsc_thread(void *arg) { IStream *stream = arg; LARGE_INTEGER zero; MSG msg; HRESULT hres; CoInitialize(NULL); hres = CoMarshalInterface(stream, &IID_IBindStatusCallback, (IUnknown*)&BindStatusCallback, MSHCTX_INPROC, NULL, MSHLFLAGS_NORMAL); ok(hres == S_OK, "CoMarshalInterface failed: %08x\n", hres); zero.QuadPart = 0; hres = IStream_Seek(stream, zero, STREAM_SEEK_SET, NULL); ok(hres == S_OK, "Seek failed: 0x%08x\n", hres); SetEvent(thread_ready); while(GetMessageW(&msg, NULL, 0, 0)) { TranslateMessage(&msg); DispatchMessageW(&msg); } CoUninitialize(); return 0; } static void test_bsc_marshaling(void) { FORMATETC formatetc = {0, NULL, 1, -1, TYMED_ISTREAM}; IBindStatusCallbackEx *callbackex; IBindStatusCallback *bsc; BINDINFO bindinfo; IStream *stream, *binding_stream; HANDLE thread; WCHAR *extra_info_out; WCHAR *verb_out; LARGE_INTEGER zero; STGMEDIUM stgmed; void *buf; DWORD bindf; HRESULT hres; hres = CreateStreamOnHGlobal(NULL, TRUE, &stream); ok(hres == S_OK, "CreateStreamOnHGlobal returned: %08x\n", hres); thread_ready = CreateEventW(NULL, TRUE, FALSE, NULL); thread = CreateThread(NULL, 0, bsc_thread, stream, 0, NULL); WaitForSingleObject(thread_ready, INFINITE); hres = CoUnmarshalInterface(stream, &IID_IBindStatusCallback, (void**)&bsc); ok(hres == S_OK, "CoUnmarshalInterface failed: %08x\n", hres); hres = CreateStreamOnHGlobal(NULL, TRUE, &binding_stream); ok(hres == S_OK, "CreateStreamOnHGlobal returned: %08x\n", hres); hres = IStream_Write(binding_stream, "xxx", 3, NULL); ok(hres == S_OK, "Write failed: %08x\n", hres); rem_bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; bindf = 0xdeadbeef; memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); bindinfo.grfBindInfoF = 12; bindinfo.dwBindVerb = 13; bindinfo.cbstgmedData = 19; bindinfo.dwOptions = 14; bindinfo.dwOptionsFlags = 15; bindinfo.dwCodePage = 16; bindinfo.securityAttributes.nLength = 30; bindinfo.securityAttributes.lpSecurityDescriptor = (void*)0xdead0001; bindinfo.securityAttributes.bInheritHandle = 31; bindinfo.iid.Data1 = 17; bindinfo.pUnk = (IUnknown*)0xdeadbeef; bindinfo.dwReserved = 18; bindinfo.stgmedData.pUnkForRelease = &unk_in.IUnknown_iface; unk_in.ref = 1; memset(&rem_bindinfo, 0, sizeof(bindinfo)); rem_bindinfo.cbSize = sizeof(rem_bindinfo); rem_bindinfo.szExtraInfo = extra_info_out = a2co("extra info out"); rem_bindinfo.grfBindInfoF = 22; rem_bindinfo.dwBindVerb = 23; rem_bindinfo.szCustomVerb = verb_out = a2co("custom verb out"); rem_bindinfo.cbstgmedData = 29; rem_bindinfo.dwOptions = 24; rem_bindinfo.dwOptionsFlags = 25; rem_bindinfo.dwCodePage = 16; rem_bindinfo.securityAttributes.nLength = 40; rem_bindinfo.securityAttributes.lpSecurityDescriptor = (void*)0xdead0002; rem_bindinfo.securityAttributes.bInheritHandle = 41; rem_bindinfo.iid.Data1 = 27; rem_bindinfo.pUnk = (IUnknown*)0xdeadbeef; rem_bindinfo.dwReserved = 18; rem_bindinfo.stgmedData.pUnkForRelease = &unk_out.IUnknown_iface; unk_out.ref = 1; hres = IBindStatusCallback_GetBindInfo(bsc, &bindf, &bindinfo); ok(hres == S_OK, "GetBindInfo failed: %08x\n", hres); ok(bindf == rem_bindf, "bindf = %x, expected %x\n", bindf, rem_bindf); ok(in_bindinfo.cbSize == sizeof(in_bindinfo), "cbSize = %u\n", in_bindinfo.cbSize); ok(!in_bindinfo.szExtraInfo, "szExtraInfo = %s\n", wine_dbgstr_w(in_bindinfo.szExtraInfo)); ok(in_bindinfo.grfBindInfoF == 12, "cbSize = %u\n", in_bindinfo.grfBindInfoF); ok(in_bindinfo.dwBindVerb == 13, "dwBindVerb = %u\n", in_bindinfo.dwBindVerb); ok(!in_bindinfo.szCustomVerb, "szCustomVerb = %s\n", wine_dbgstr_w(in_bindinfo.szCustomVerb)); ok(in_bindinfo.cbstgmedData == 19, "cbstgmedData = %u\n", in_bindinfo.cbstgmedData); ok(!in_bindinfo.dwOptions, "dwOptions = %u\n", in_bindinfo.dwOptions); ok(!in_bindinfo.dwOptionsFlags, "dwOptionsFlags = %u\n", in_bindinfo.dwOptionsFlags); ok(!in_bindinfo.dwCodePage, "dwCodePage = %u\n", in_bindinfo.dwCodePage); ok(!in_bindinfo.iid.Data1, "iid.Data1 = %u\n", in_bindinfo.iid.Data1); ok(!in_bindinfo.pUnk, "pUnk = %p\n", in_bindinfo.pUnk); ok(!in_bindinfo.dwReserved, "dwReserved = %u\n", in_bindinfo.dwReserved); ok(!in_bindinfo.securityAttributes.nLength, "securityAttributes.nLength = %u\n", in_bindinfo.securityAttributes.nLength); ok(!in_bindinfo.securityAttributes.lpSecurityDescriptor, "securityAttributes.lpSecurityDescriptor = %p\n", in_bindinfo.securityAttributes.lpSecurityDescriptor); ok(!in_bindinfo.securityAttributes.bInheritHandle, "securityAttributes.bInheritHandle = %u\n", in_bindinfo.securityAttributes.bInheritHandle); ok(!in_bindinfo.stgmedData.pUnkForRelease, "pUnkForRelease = %p\n", in_bindinfo.stgmedData.pUnkForRelease); ok(bindinfo.cbSize == sizeof(rem_bindinfo), "cbSize = %u\n", rem_bindinfo.cbSize); ok(!lstrcmpW(bindinfo.szExtraInfo, L"extra info out"), "szExtraInfo = %s\n", wine_dbgstr_w(bindinfo.szExtraInfo)); ok(bindinfo.grfBindInfoF == 22, "grfBindInfoF = %u\n", rem_bindinfo.grfBindInfoF); ok(bindinfo.dwBindVerb == 23, "dwBindVerb = %u\n", bindinfo.dwBindVerb); ok(bindinfo.szCustomVerb != verb_out, "szCustomVerb == inbuf\n"); ok(!lstrcmpW(bindinfo.szCustomVerb, L"custom verb out"), "szCustomVerb = %s\n", wine_dbgstr_w(bindinfo.szCustomVerb)); ok(bindinfo.cbstgmedData == 29, "cbstgmedData = %u\n", bindinfo.cbstgmedData); ok(bindinfo.dwOptions == 24, "dwOptions = %u\n", bindinfo.dwOptions); ok(bindinfo.dwOptionsFlags == 25, "dwOptionsFlags = %u\n", bindinfo.dwOptionsFlags); ok(bindinfo.dwCodePage, "dwCodePage = %u\n", bindinfo.dwCodePage); ok(!bindinfo.iid.Data1, "iid.Data1 = %u\n", bindinfo.iid.Data1); ok(!bindinfo.pUnk, "pUnk = %p\n", bindinfo.pUnk); ok(bindinfo.dwReserved == 18, "dwReserved = %u\n", bindinfo.dwReserved); ok(bindinfo.securityAttributes.nLength == 30, "securityAttributes.nLength = %u\n", bindinfo.securityAttributes.nLength); ok(bindinfo.securityAttributes.lpSecurityDescriptor == (void*)0xdead0001, "securityAttributes.lpSecurityDescriptor = %p\n", bindinfo.securityAttributes.lpSecurityDescriptor); ok(bindinfo.securityAttributes.bInheritHandle == 31, "securityAttributes.bInheritHandle = %u\n", bindinfo.securityAttributes.bInheritHandle); ok(bindinfo.stgmedData.pUnkForRelease == &unk_in.IUnknown_iface, "pUnkForRelease = %p\n", bindinfo.stgmedData.pUnkForRelease); ok(unk_out.ref == 1, "unk_out.ref = %u\n", unk_out.ref); bindinfo.stgmedData.pUnkForRelease = NULL; ReleaseBindInfo(&bindinfo); zero.QuadPart = 0; hres = IStream_Seek(binding_stream, zero, STREAM_SEEK_SET, NULL); ok(hres == S_OK, "Seek failed: 0x%08x\n", hres); /* Return IStream stgmed from GetBindInfo, it's not marshaled back */ rem_bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; bindf = 0xdeadbeef; memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); memset(&rem_bindinfo, 0, sizeof(rem_bindinfo)); rem_bindinfo.cbSize = sizeof(rem_bindinfo); rem_bindinfo.stgmedData.tymed = TYMED_ISTREAM; rem_bindinfo.stgmedData.u.pstm = binding_stream; rem_bindinfo.cbstgmedData = 3; IStream_AddRef(binding_stream); hres = IBindStatusCallback_GetBindInfo(bsc, &bindf, &bindinfo); ok(hres == S_OK, "GetBindInfo failed: %08x\n", hres); ok(bindf == rem_bindf, "bindf = %x, expected %x\n", bindf, rem_bindf); ok(in_bindinfo.cbSize == sizeof(in_bindinfo), "cbSize = %u\n", in_bindinfo.cbSize); ok(!in_bindinfo.pUnk, "pUnk = %p\n", in_bindinfo.pUnk); ok(in_bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", in_bindinfo.stgmedData.tymed); ok(bindinfo.cbSize == sizeof(bindinfo), "cbSize = %u\n", bindinfo.cbSize); ok(!bindinfo.pUnk, "pUnk = %p\n", bindinfo.pUnk); ok(bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", bindinfo.stgmedData.tymed); ok(!bindinfo.stgmedData.u.pstm, "stm = %p\n", bindinfo.stgmedData.u.pstm); ok(!bindinfo.stgmedData.pUnkForRelease, "pUnkForRelease = %p\n", bindinfo.stgmedData.pUnkForRelease); ok(bindinfo.cbstgmedData == 3, "cbstgmedData = %u\n", bindinfo.cbstgmedData); ReleaseBindInfo(&bindinfo); /* Same, but with pUnkForRelease, it's not marshaled back */ rem_bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; bindf = 0xdeadbeef; memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); memset(&rem_bindinfo, 0, sizeof(rem_bindinfo)); rem_bindinfo.cbSize = sizeof(rem_bindinfo); rem_bindinfo.stgmedData.tymed = TYMED_ISTREAM; rem_bindinfo.stgmedData.u.pstm = binding_stream; rem_bindinfo.stgmedData.pUnkForRelease = &unk_out.IUnknown_iface; unk_out.ref = 1; rem_bindinfo.cbstgmedData = 3; IStream_AddRef(binding_stream); hres = IBindStatusCallback_GetBindInfo(bsc, &bindf, &bindinfo); ok(hres == S_OK, "GetBindInfo failed: %08x\n", hres); ok(bindf == rem_bindf, "bindf = %x, expected %x\n", bindf, rem_bindf); ok(in_bindinfo.cbSize == sizeof(in_bindinfo), "cbSize = %u\n", in_bindinfo.cbSize); ok(!in_bindinfo.pUnk, "pUnk = %p\n", in_bindinfo.pUnk); ok(in_bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", in_bindinfo.stgmedData.tymed); ok(bindinfo.cbSize == sizeof(bindinfo), "cbSize = %u\n", bindinfo.cbSize); ok(!bindinfo.pUnk, "pUnk = %p\n", bindinfo.pUnk); ok(bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", bindinfo.stgmedData.tymed); ok(!bindinfo.stgmedData.u.pstm, "stm = %p\n", bindinfo.stgmedData.u.pstm); ok(!bindinfo.stgmedData.pUnkForRelease, "pUnkForRelease = %p\n", bindinfo.stgmedData.pUnkForRelease); ok(bindinfo.cbstgmedData == 3, "cbstgmedData = %u\n", bindinfo.cbstgmedData); ReleaseBindInfo(&bindinfo); /* Return HGLOBAL stgmed from GetBindInfo, it's not marshaled back */ rem_bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; bindf = 0xdeadbeef; memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); memset(&rem_bindinfo, 0, sizeof(rem_bindinfo)); rem_bindinfo.cbSize = sizeof(rem_bindinfo); rem_bindinfo.stgmedData.tymed = TYMED_HGLOBAL; buf = GlobalAlloc(0, 5); strcpy(buf, "test"); rem_bindinfo.stgmedData.u.hGlobal = buf; rem_bindinfo.cbstgmedData = 5; hres = IBindStatusCallback_GetBindInfo(bsc, &bindf, &bindinfo); ok(hres == S_OK, "GetBindInfo failed: %08x\n", hres); ok(bindf == rem_bindf, "bindf = %x, expected %x\n", bindf, rem_bindf); ok(in_bindinfo.cbSize == sizeof(in_bindinfo), "cbSize = %u\n", in_bindinfo.cbSize); ok(!in_bindinfo.pUnk, "pUnk = %p\n", in_bindinfo.pUnk); ok(in_bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", in_bindinfo.stgmedData.tymed); ok(bindinfo.cbSize == sizeof(bindinfo), "cbSize = %u\n", bindinfo.cbSize); ok(!bindinfo.pUnk, "pUnk = %p\n", bindinfo.pUnk); ok(bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", bindinfo.stgmedData.tymed); ok(!bindinfo.stgmedData.u.pstm, "stm = %p\n", bindinfo.stgmedData.u.pstm); ok(!bindinfo.stgmedData.pUnkForRelease, "pUnkForRelease = %p\n", bindinfo.stgmedData.pUnkForRelease); ok(bindinfo.cbstgmedData == 5, "cbstgmedData = %u\n", bindinfo.cbstgmedData); ReleaseBindInfo(&bindinfo); /* Same with GetBindInfoEx */ hres = IBindStatusCallback_QueryInterface(bsc, &IID_IBindStatusCallbackEx, (void**)&callbackex); if(SUCCEEDED(hres)) { DWORD bindf2, reserved; rem_bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; bindf = bindf2 = reserved = 0xdeadbeef; memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); bindinfo.grfBindInfoF = 12; bindinfo.dwBindVerb = 13; bindinfo.cbstgmedData = 19; bindinfo.dwOptions = 14; bindinfo.dwOptionsFlags = 15; bindinfo.dwCodePage = 16; bindinfo.securityAttributes.nLength = 30; bindinfo.securityAttributes.lpSecurityDescriptor = (void*)0xdead0001; bindinfo.securityAttributes.bInheritHandle = 31; bindinfo.iid.Data1 = 17; bindinfo.pUnk = (IUnknown*)0xdeadbeef; bindinfo.dwReserved = 18; bindinfo.stgmedData.pUnkForRelease = &unk_in.IUnknown_iface; unk_in.ref = 1; memset(&rem_bindinfo, 0, sizeof(bindinfo)); rem_bindinfo.cbSize = sizeof(rem_bindinfo); rem_bindinfo.szExtraInfo = extra_info_out = a2co("extra info out"); rem_bindinfo.grfBindInfoF = 22; rem_bindinfo.dwBindVerb = 23; rem_bindinfo.szCustomVerb = verb_out = a2co("custom verb out"); rem_bindinfo.cbstgmedData = 29; rem_bindinfo.dwOptions = 24; rem_bindinfo.dwOptionsFlags = 25; rem_bindinfo.dwCodePage = 16; rem_bindinfo.securityAttributes.nLength = 40; rem_bindinfo.securityAttributes.lpSecurityDescriptor = (void*)0xdead0002; rem_bindinfo.securityAttributes.bInheritHandle = 41; rem_bindinfo.iid.Data1 = 27; rem_bindinfo.pUnk = (IUnknown*)0xdeadbeef; rem_bindinfo.dwReserved = 18; rem_bindinfo.stgmedData.pUnkForRelease = &unk_out.IUnknown_iface; unk_out.ref = 1; hres = IBindStatusCallbackEx_GetBindInfoEx(callbackex, &bindf, &bindinfo, &bindf2, &reserved); ok(hres == S_OK, "GetBindInfo failed: %08x\n", hres); ok(bindf == rem_bindf, "bindf = %x, expected %x\n", bindf, rem_bindf); ok(bindf2 == 11, "bindf2 = %x\n", bindf); ok(reserved == 12, "reserved = %x\n", reserved); ok(in_bindinfo.cbSize == sizeof(in_bindinfo), "cbSize = %u\n", in_bindinfo.cbSize); ok(!in_bindinfo.szExtraInfo, "szExtraInfo = %s\n", wine_dbgstr_w(in_bindinfo.szExtraInfo)); ok(in_bindinfo.grfBindInfoF == 12, "cbSize = %u\n", in_bindinfo.grfBindInfoF); ok(in_bindinfo.dwBindVerb == 13, "dwBindVerb = %u\n", in_bindinfo.dwBindVerb); ok(!in_bindinfo.szCustomVerb, "szCustomVerb = %s\n", wine_dbgstr_w(in_bindinfo.szCustomVerb)); ok(in_bindinfo.cbstgmedData == 19, "cbstgmedData = %u\n", in_bindinfo.cbstgmedData); ok(!in_bindinfo.dwOptions, "dwOptions = %u\n", in_bindinfo.dwOptions); ok(!in_bindinfo.dwOptionsFlags, "dwOptionsFlags = %u\n", in_bindinfo.dwOptionsFlags); ok(!in_bindinfo.dwCodePage, "dwCodePage = %u\n", in_bindinfo.dwCodePage); ok(!in_bindinfo.iid.Data1, "iid.Data1 = %u\n", in_bindinfo.iid.Data1); ok(!in_bindinfo.pUnk, "pUnk = %p\n", in_bindinfo.pUnk); ok(!in_bindinfo.dwReserved, "dwReserved = %u\n", in_bindinfo.dwReserved); ok(!in_bindinfo.securityAttributes.nLength, "securityAttributes.nLength = %u\n", in_bindinfo.securityAttributes.nLength); ok(!in_bindinfo.securityAttributes.lpSecurityDescriptor, "securityAttributes.lpSecurityDescriptor = %p\n", in_bindinfo.securityAttributes.lpSecurityDescriptor); ok(!in_bindinfo.securityAttributes.bInheritHandle, "securityAttributes.bInheritHandle = %u\n", in_bindinfo.securityAttributes.bInheritHandle); ok(!in_bindinfo.stgmedData.pUnkForRelease, "pUnkForRelease = %p\n", in_bindinfo.stgmedData.pUnkForRelease); ok(bindinfo.cbSize == sizeof(rem_bindinfo), "cbSize = %u\n", rem_bindinfo.cbSize); ok(!lstrcmpW(bindinfo.szExtraInfo, L"extra info out"), "szExtraInfo = %s\n", wine_dbgstr_w(bindinfo.szExtraInfo)); ok(bindinfo.grfBindInfoF == 22, "grfBindInfoF = %u\n", rem_bindinfo.grfBindInfoF); ok(bindinfo.dwBindVerb == 23, "dwBindVerb = %u\n", bindinfo.dwBindVerb); ok(bindinfo.szCustomVerb != verb_out, "szCustomVerb == inbuf\n"); ok(!lstrcmpW(bindinfo.szCustomVerb, L"custom verb out"), "szCustomVerb = %s\n", wine_dbgstr_w(bindinfo.szCustomVerb)); ok(bindinfo.cbstgmedData == 29, "cbstgmedData = %u\n", bindinfo.cbstgmedData); ok(bindinfo.dwOptions == 24, "dwOptions = %u\n", bindinfo.dwOptions); ok(bindinfo.dwOptionsFlags == 25, "dwOptionsFlags = %u\n", bindinfo.dwOptionsFlags); ok(bindinfo.dwCodePage, "dwCodePage = %u\n", bindinfo.dwCodePage); ok(!bindinfo.iid.Data1, "iid.Data1 = %u\n", bindinfo.iid.Data1); ok(!bindinfo.pUnk, "pUnk = %p\n", bindinfo.pUnk); ok(bindinfo.dwReserved == 18, "dwReserved = %u\n", bindinfo.dwReserved); ok(bindinfo.securityAttributes.nLength == 30, "securityAttributes.nLength = %u\n", bindinfo.securityAttributes.nLength); ok(bindinfo.securityAttributes.lpSecurityDescriptor == (void*)0xdead0001, "securityAttributes.lpSecurityDescriptor = %p\n", bindinfo.securityAttributes.lpSecurityDescriptor); ok(bindinfo.securityAttributes.bInheritHandle == 31, "securityAttributes.bInheritHandle = %u\n", bindinfo.securityAttributes.bInheritHandle); ok(bindinfo.stgmedData.pUnkForRelease == &unk_in.IUnknown_iface, "pUnkForRelease = %p\n", bindinfo.stgmedData.pUnkForRelease); ok(unk_out.ref == 1, "unk_out.ref = %u\n", unk_out.ref); bindinfo.stgmedData.pUnkForRelease = NULL; ReleaseBindInfo(&bindinfo); zero.QuadPart = 0; hres = IStream_Seek(binding_stream, zero, STREAM_SEEK_SET, NULL); ok(hres == S_OK, "Seek failed: 0x%08x\n", hres); /* Return IStream stgmed from GetBindInfoEx, it's not marshaled back */ rem_bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; bindf = bindf2 = reserved = 0xdeadbeef; memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); memset(&rem_bindinfo, 0, sizeof(rem_bindinfo)); rem_bindinfo.cbSize = sizeof(rem_bindinfo); rem_bindinfo.stgmedData.tymed = TYMED_ISTREAM; rem_bindinfo.stgmedData.u.pstm = binding_stream; rem_bindinfo.cbstgmedData = 3; IStream_AddRef(binding_stream); hres = IBindStatusCallbackEx_GetBindInfoEx(callbackex, &bindf, &bindinfo, &bindf2, &reserved); ok(hres == S_OK, "GetBindInfoEx failed: %08x\n", hres); ok(bindf == rem_bindf, "bindf = %x, expected %x\n", bindf, rem_bindf); ok(in_bindinfo.cbSize == sizeof(in_bindinfo), "cbSize = %u\n", in_bindinfo.cbSize); ok(!in_bindinfo.pUnk, "pUnk = %p\n", in_bindinfo.pUnk); ok(in_bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", in_bindinfo.stgmedData.tymed); ok(bindinfo.cbSize == sizeof(bindinfo), "cbSize = %u\n", bindinfo.cbSize); ok(!bindinfo.pUnk, "pUnk = %p\n", bindinfo.pUnk); ok(bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", bindinfo.stgmedData.tymed); ok(!bindinfo.stgmedData.u.pstm, "stm = %p\n", bindinfo.stgmedData.u.pstm); ok(!bindinfo.stgmedData.pUnkForRelease, "pUnkForRelease = %p\n", bindinfo.stgmedData.pUnkForRelease); ok(bindinfo.cbstgmedData == 3, "cbstgmedData = %u\n", bindinfo.cbstgmedData); ReleaseBindInfo(&bindinfo); /* Same, but with pUnkForRelease, it's not marshaled back */ rem_bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; bindf = bindf2 = reserved = 0xdeadbeef; memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); memset(&rem_bindinfo, 0, sizeof(rem_bindinfo)); rem_bindinfo.cbSize = sizeof(rem_bindinfo); rem_bindinfo.stgmedData.tymed = TYMED_ISTREAM; rem_bindinfo.stgmedData.u.pstm = binding_stream; rem_bindinfo.stgmedData.pUnkForRelease = &unk_out.IUnknown_iface; unk_out.ref = 1; rem_bindinfo.cbstgmedData = 3; IStream_AddRef(binding_stream); hres = IBindStatusCallbackEx_GetBindInfoEx(callbackex, &bindf, &bindinfo, &bindf2, &reserved); ok(hres == S_OK, "GetBindInfoEx failed: %08x\n", hres); ok(bindf == rem_bindf, "bindf = %x, expected %x\n", bindf, rem_bindf); ok(in_bindinfo.cbSize == sizeof(in_bindinfo), "cbSize = %u\n", in_bindinfo.cbSize); ok(!in_bindinfo.pUnk, "pUnk = %p\n", in_bindinfo.pUnk); ok(in_bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", in_bindinfo.stgmedData.tymed); ok(bindinfo.cbSize == sizeof(bindinfo), "cbSize = %u\n", bindinfo.cbSize); ok(!bindinfo.pUnk, "pUnk = %p\n", bindinfo.pUnk); ok(bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", bindinfo.stgmedData.tymed); ok(!bindinfo.stgmedData.u.pstm, "stm = %p\n", bindinfo.stgmedData.u.pstm); ok(!bindinfo.stgmedData.pUnkForRelease, "pUnkForRelease = %p\n", bindinfo.stgmedData.pUnkForRelease); ok(bindinfo.cbstgmedData == 3, "cbstgmedData = %u\n", bindinfo.cbstgmedData); ReleaseBindInfo(&bindinfo); /* Return HGLOBAL stgmed from GetBindInfoEx, it's not marshaled back */ rem_bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA; bindf = bindf2 = reserved = 0xdeadbeef; memset(&bindinfo, 0, sizeof(bindinfo)); bindinfo.cbSize = sizeof(bindinfo); memset(&rem_bindinfo, 0, sizeof(rem_bindinfo)); rem_bindinfo.cbSize = sizeof(rem_bindinfo); rem_bindinfo.stgmedData.tymed = TYMED_HGLOBAL; buf = GlobalAlloc(0, 5); strcpy(buf, "test"); rem_bindinfo.stgmedData.u.hGlobal = buf; rem_bindinfo.cbstgmedData = 5; hres = IBindStatusCallbackEx_GetBindInfoEx(callbackex, &bindf, &bindinfo, &bindf2, &reserved); ok(hres == S_OK, "GetBindInfoEx failed: %08x\n", hres); ok(bindf == rem_bindf, "bindf = %x, expected %x\n", bindf, rem_bindf); ok(in_bindinfo.cbSize == sizeof(in_bindinfo), "cbSize = %u\n", in_bindinfo.cbSize); ok(!in_bindinfo.pUnk, "pUnk = %p\n", in_bindinfo.pUnk); ok(in_bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", in_bindinfo.stgmedData.tymed); ok(bindinfo.cbSize == sizeof(bindinfo), "cbSize = %u\n", bindinfo.cbSize); ok(!bindinfo.pUnk, "pUnk = %p\n", bindinfo.pUnk); ok(bindinfo.stgmedData.tymed == TYMED_NULL, "tymed = %u\n", bindinfo.stgmedData.tymed); ok(!bindinfo.stgmedData.u.pstm, "stm = %p\n", bindinfo.stgmedData.u.pstm); ok(!bindinfo.stgmedData.pUnkForRelease, "pUnkForRelease = %p\n", bindinfo.stgmedData.pUnkForRelease); ok(bindinfo.cbstgmedData == 5, "cbstgmedData = %u\n", bindinfo.cbstgmedData); ReleaseBindInfo(&bindinfo); IBindStatusCallbackEx_Release(callbackex); }else { win_skip("IBindStatusCallbackEx not supported\n"); } /* Test marshaling stgmed from OnDataAvailable */ memset(&in_stgmed, 0xcc, sizeof(in_stgmed)); stgmed.tymed = TYMED_ISTREAM; stgmed.u.pstm = binding_stream; stgmed.pUnkForRelease = NULL; hres = IBindStatusCallback_OnDataAvailable(bsc, 1, 10, &formatetc, &stgmed); ok(hres == S_OK, "OnDataAvailable failed: %08x\n", hres); ok(in_stgmed.tymed == TYMED_ISTREAM, "tymed = %u\n", in_stgmed.tymed); ok(in_stgmed.u.pstm != NULL, "pstm = NULL\n"); ok(!in_stgmed.pUnkForRelease, "pUnkForRelease = %p\n", in_stgmed.pUnkForRelease); /* OnDataAvailable with both IStream and pUnkForRelease */ memset(&in_stgmed, 0xcc, sizeof(in_stgmed)); stgmed.tymed = TYMED_ISTREAM; stgmed.u.pstm = binding_stream; stgmed.pUnkForRelease = &unk_in.IUnknown_iface; unk_in.ref = 1; hres = IBindStatusCallback_OnDataAvailable(bsc, 1, 10, &formatetc, &stgmed); ok(hres == S_OK, "OnDataAvailable failed: %08x\n", hres); ok(in_stgmed.tymed == TYMED_ISTREAM, "tymed = %u\n", in_stgmed.tymed); ok(in_stgmed.u.pstm != NULL, "pstm = NULL\n"); ok(in_stgmed.pUnkForRelease != NULL, "pUnkForRelease = %p\n", in_stgmed.pUnkForRelease); ok(unk_in.ref > 1, "ref = %u\n", unk_in.ref); /* OnDataAvailable with TYMED_ISTREAM, but NULL stream */ memset(&in_stgmed, 0xcc, sizeof(in_stgmed)); stgmed.tymed = TYMED_ISTREAM; stgmed.u.pstm = binding_stream; stgmed.pUnkForRelease = NULL; hres = IBindStatusCallback_OnDataAvailable(bsc, 1, 10, &formatetc, &stgmed); ok(hres == S_OK, "OnDataAvailable failed: %08x\n", hres); ok(in_stgmed.tymed == TYMED_ISTREAM, "tymed = %u\n", in_stgmed.tymed); ok(in_stgmed.u.pstm != NULL, "pstm = NULL\n"); ok(!in_stgmed.pUnkForRelease, "pUnkForRelease = %p\n", in_stgmed.pUnkForRelease); /* OnDataAvailable with TYMED_NULL and pUnkForRelease */ memset(&in_stgmed, 0xcc, sizeof(in_stgmed)); stgmed.tymed = TYMED_NULL; stgmed.u.pstm = binding_stream; stgmed.pUnkForRelease = &unk_in.IUnknown_iface; unk_in.ref = 1; hres = IBindStatusCallback_OnDataAvailable(bsc, 1, 10, &formatetc, &stgmed); ok(hres == S_OK, "OnDataAvailable failed: %08x\n", hres); ok(in_stgmed.tymed == TYMED_NULL, "tymed = %u\n", in_stgmed.tymed); ok(!in_stgmed.u.pstm, "pstm != NULL\n"); ok(in_stgmed.pUnkForRelease != NULL, "pUnkForRelease = %p\n", in_stgmed.pUnkForRelease); ok(unk_in.ref == 1, "ref = %u\n", unk_in.ref); IStream_Release(binding_stream); IBindStatusCallback_Release(bsc); TerminateThread(thread, 0); } START_TEST(misc) { HMODULE hurlmon; int argc; char **argv; argc = winetest_get_mainargs(&argv); hurlmon = GetModuleHandleA("urlmon.dll"); pCoInternetCompareUrl = (void *) GetProcAddress(hurlmon, "CoInternetCompareUrl"); pCoInternetGetSecurityUrl = (void*) GetProcAddress(hurlmon, "CoInternetGetSecurityUrl"); pCoInternetGetSession = (void*) GetProcAddress(hurlmon, "CoInternetGetSession"); pCoInternetParseUrl = (void*) GetProcAddress(hurlmon, "CoInternetParseUrl"); pCoInternetQueryInfo = (void*) GetProcAddress(hurlmon, "CoInternetQueryInfo"); pCopyStgMedium = (void*) GetProcAddress(hurlmon, "CopyStgMedium"); pCopyBindInfo = (void*) GetProcAddress(hurlmon, "CopyBindInfo"); pFindMimeFromData = (void*) GetProcAddress(hurlmon, "FindMimeFromData"); pObtainUserAgentString = (void*) GetProcAddress(hurlmon, "ObtainUserAgentString"); pReleaseBindInfo = (void*) GetProcAddress(hurlmon, "ReleaseBindInfo"); pUrlMkGetSessionOption = (void*) GetProcAddress(hurlmon, "UrlMkGetSessionOption"); pCompareSecurityIds = (void*) GetProcAddress(hurlmon, "CompareSecurityIds"); pCoInternetIsFeatureEnabled = (void*) GetProcAddress(hurlmon, "CoInternetIsFeatureEnabled"); pCoInternetSetFeatureEnabled = (void*) GetProcAddress(hurlmon, "CoInternetSetFeatureEnabled"); pIEInstallScope = (void*) GetProcAddress(hurlmon, "IEInstallScope"); if (!pCoInternetCompareUrl || !pCoInternetGetSecurityUrl || !pCoInternetGetSession || !pCoInternetParseUrl || !pCompareSecurityIds) { win_skip("Various needed functions not present, too old IE\n"); return; } OleInitialize(NULL); if(argc <= 2 || strcmp(argv[2], "internet_features")) { register_protocols(); test_CreateFormatEnum(); test_RegisterFormatEnumerator(); test_CoInternetParseUrl(); test_CoInternetCompareUrl(); test_CoInternetQueryInfo(); test_FindMimeFromData(); test_NameSpace(); test_MimeFilter(); test_ReleaseBindInfo(); test_CopyStgMedium(); test_CopyBindInfo(); test_UrlMkGetSessionOption(); test_user_agent(); test_MkParseDisplayNameEx(); test_IsValidURL(); test_bsc_marshaling(); } test_internet_features(); OleUninitialize(); }