urlmon/tests: Added tests for CoInternetSetFeatureEnabled.
[wine/testsucceed.git] / dlls / urlmon / tests / misc.c
blob4811ab08c80e2892b066a53092e58e7a2f6fb3d2
1 /*
2 * Copyright 2005-2006 Jacek Caban for CodeWeavers
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 #define COBJMACROS
20 #define CONST_VTABLE
21 #define NONAMELESSUNION
23 #include <wine/test.h>
24 #include <stdarg.h>
25 #include <stddef.h>
27 #include "windef.h"
28 #include "winbase.h"
29 #include "ole2.h"
30 #include "urlmon.h"
32 #include "initguid.h"
34 DEFINE_GUID(CLSID_AboutProtocol, 0x3050F406, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
36 #define DEFINE_EXPECT(func) \
37 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
39 #define SET_EXPECT(func) \
40 expect_ ## func = TRUE
42 #define CHECK_EXPECT(func) \
43 do { \
44 ok(expect_ ##func, "unexpected call " #func "\n"); \
45 expect_ ## func = FALSE; \
46 called_ ## func = TRUE; \
47 }while(0)
49 #define CHECK_EXPECT2(func) \
50 do { \
51 ok(expect_ ##func, "unexpected call " #func "\n"); \
52 called_ ## func = TRUE; \
53 }while(0)
55 #define CHECK_CALLED(func) \
56 do { \
57 ok(called_ ## func, "expected " #func "\n"); \
58 expect_ ## func = called_ ## func = FALSE; \
59 }while(0)
61 DEFINE_EXPECT(ParseUrl);
62 DEFINE_EXPECT(QI_IInternetProtocolInfo);
63 DEFINE_EXPECT(CreateInstance);
64 DEFINE_EXPECT(unk_Release);
66 static HRESULT (WINAPI *pCoInternetCompareUrl)(LPCWSTR, LPCWSTR, DWORD);
67 static HRESULT (WINAPI *pCoInternetGetSecurityUrl)(LPCWSTR, LPWSTR*, PSUACTION, DWORD);
68 static HRESULT (WINAPI *pCoInternetGetSession)(DWORD, IInternetSession **, DWORD);
69 static HRESULT (WINAPI *pCoInternetParseUrl)(LPCWSTR, PARSEACTION, DWORD, LPWSTR, DWORD, DWORD *, DWORD);
70 static HRESULT (WINAPI *pCoInternetQueryInfo)(LPCWSTR, QUERYOPTION, DWORD, LPVOID, DWORD, DWORD *, DWORD);
71 static HRESULT (WINAPI *pCopyStgMedium)(const STGMEDIUM *, STGMEDIUM *);
72 static HRESULT (WINAPI *pFindMimeFromData)(LPBC, LPCWSTR, LPVOID, DWORD, LPCWSTR,
73 DWORD, LPWSTR*, DWORD);
74 static HRESULT (WINAPI *pObtainUserAgentString)(DWORD, LPSTR, DWORD*);
75 static HRESULT (WINAPI *pReleaseBindInfo)(BINDINFO*);
76 static HRESULT (WINAPI *pUrlMkGetSessionOption)(DWORD, LPVOID, DWORD, DWORD *, DWORD);
77 static HRESULT (WINAPI *pCompareSecurityIds)(BYTE*,DWORD,BYTE*,DWORD,DWORD);
78 static HRESULT (WINAPI *pCoInternetIsFeatureEnabled)(INTERNETFEATURELIST,DWORD);
79 static HRESULT (WINAPI *pCoInternetSetFeatureEnabled)(INTERNETFEATURELIST,DWORD,BOOL);
81 static void test_CreateFormatEnum(void)
83 IEnumFORMATETC *fenum = NULL, *fenum2 = NULL;
84 FORMATETC fetc[5];
85 ULONG ul;
86 HRESULT hres;
88 static DVTARGETDEVICE dev = {sizeof(dev),0,0,0,0,{0}};
89 static FORMATETC formatetc[] = {
90 {0,&dev,0,0,0},
91 {0,&dev,0,1,0},
92 {0,NULL,0,2,0},
93 {0,NULL,0,3,0},
94 {0,NULL,0,4,0}
97 hres = CreateFormatEnumerator(0, formatetc, &fenum);
98 ok(hres == E_FAIL, "CreateFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
99 hres = CreateFormatEnumerator(0, formatetc, NULL);
100 ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
101 hres = CreateFormatEnumerator(5, formatetc, NULL);
102 ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
105 hres = CreateFormatEnumerator(5, formatetc, &fenum);
106 ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
107 if(FAILED(hres))
108 return;
110 hres = IEnumFORMATETC_Next(fenum, 2, NULL, &ul);
111 ok(hres == E_INVALIDARG, "Next failed: %08x, expected E_INVALIDARG\n", hres);
112 ul = 100;
113 hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul);
114 ok(hres == S_OK, "Next failed: %08x\n", hres);
115 ok(ul == 0, "ul=%d, expected 0\n", ul);
117 hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul);
118 ok(hres == S_OK, "Next failed: %08x\n", hres);
119 ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex);
120 ok(fetc[1].lindex == 1, "fetc[1].lindex=%d, expected 1\n", fetc[1].lindex);
121 ok(fetc[0].ptd == &dev, "fetc[0].ptd=%p, expected %p\n", fetc[0].ptd, &dev);
122 ok(ul == 2, "ul=%d, expected 2\n", ul);
124 hres = IEnumFORMATETC_Skip(fenum, 1);
125 ok(hres == S_OK, "Skip failed: %08x\n", hres);
127 hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul);
128 ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres);
129 ok(fetc[0].lindex == 3, "fetc[0].lindex=%d, expected 3\n", fetc[0].lindex);
130 ok(fetc[1].lindex == 4, "fetc[1].lindex=%d, expected 4\n", fetc[1].lindex);
131 ok(fetc[0].ptd == NULL, "fetc[0].ptd=%p, expected NULL\n", fetc[0].ptd);
132 ok(ul == 2, "ul=%d, expected 2\n", ul);
134 hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul);
135 ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres);
136 ok(ul == 0, "ul=%d, expected 0\n", ul);
137 ul = 100;
138 hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul);
139 ok(hres == S_OK, "Next failed: %08x\n", hres);
140 ok(ul == 0, "ul=%d, expected 0\n", ul);
142 hres = IEnumFORMATETC_Skip(fenum, 3);
143 ok(hres == S_FALSE, "Skip failed: %08x, expected S_FALSE\n", hres);
145 hres = IEnumFORMATETC_Reset(fenum);
146 ok(hres == S_OK, "Reset failed: %08x\n", hres);
148 hres = IEnumFORMATETC_Next(fenum, 5, fetc, NULL);
149 ok(hres == S_OK, "Next failed: %08x\n", hres);
150 ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex);
152 hres = IEnumFORMATETC_Reset(fenum);
153 ok(hres == S_OK, "Reset failed: %08x\n", hres);
155 hres = IEnumFORMATETC_Skip(fenum, 2);
156 ok(hres == S_OK, "Skip failed: %08x\n", hres);
158 hres = IEnumFORMATETC_Clone(fenum, NULL);
159 ok(hres == E_INVALIDARG, "Clone failed: %08x, expected E_INVALIDARG\n", hres);
161 hres = IEnumFORMATETC_Clone(fenum, &fenum2);
162 ok(hres == S_OK, "Clone failed: %08x\n", hres);
164 if(SUCCEEDED(hres)) {
165 ok(fenum != fenum2, "fenum == fenum2\n");
167 hres = IEnumFORMATETC_Next(fenum2, 2, fetc, &ul);
168 ok(hres == S_OK, "Next failed: %08x\n", hres);
169 ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex);
171 IEnumFORMATETC_Release(fenum2);
174 hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul);
175 ok(hres == S_OK, "Next failed: %08x\n", hres);
176 ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex);
178 hres = IEnumFORMATETC_Skip(fenum, 1);
179 ok(hres == S_OK, "Skip failed: %08x\n", hres);
181 IEnumFORMATETC_Release(fenum);
184 static void test_RegisterFormatEnumerator(void)
186 IBindCtx *bctx = NULL;
187 IEnumFORMATETC *format = NULL, *format2 = NULL;
188 IUnknown *unk = NULL;
189 HRESULT hres;
191 static FORMATETC formatetc = {0,NULL,0,0,0};
192 static WCHAR wszEnumFORMATETC[] =
193 {'_','E','n','u','m','F','O','R','M','A','T','E','T','C','_',0};
195 CreateBindCtx(0, &bctx);
197 hres = CreateFormatEnumerator(1, &formatetc, &format);
198 ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
199 if(FAILED(hres))
200 return;
202 hres = RegisterFormatEnumerator(NULL, format, 0);
203 ok(hres == E_INVALIDARG,
204 "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
205 hres = RegisterFormatEnumerator(bctx, NULL, 0);
206 ok(hres == E_INVALIDARG,
207 "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
209 hres = RegisterFormatEnumerator(bctx, format, 0);
210 ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
212 hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
213 ok(hres == S_OK, "GetObjectParam failed: %08x\n", hres);
214 ok(unk == (IUnknown*)format, "unk != format\n");
216 hres = RevokeFormatEnumerator(NULL, format);
217 ok(hres == E_INVALIDARG,
218 "RevokeFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
220 hres = RevokeFormatEnumerator(bctx, format);
221 ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
223 hres = RevokeFormatEnumerator(bctx, format);
224 ok(hres == E_FAIL, "RevokeFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
226 hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
227 ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
229 hres = RegisterFormatEnumerator(bctx, format, 0);
230 ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
232 hres = CreateFormatEnumerator(1, &formatetc, &format2);
233 ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
235 if(SUCCEEDED(hres)) {
236 hres = RevokeFormatEnumerator(bctx, format);
237 ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
239 IEnumFORMATETC_Release(format2);
242 hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
243 ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
245 IEnumFORMATETC_Release(format);
247 hres = RegisterFormatEnumerator(bctx, format, 0);
248 ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
249 hres = RevokeFormatEnumerator(bctx, NULL);
250 ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
251 hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
252 ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
254 IEnumFORMATETC_Release(format);
255 IBindCtx_Release(bctx);
258 static const WCHAR url1[] = {'r','e','s',':','/','/','m','s','h','t','m','l','.','d','l','l',
259 '/','b','l','a','n','k','.','h','t','m',0};
260 static const WCHAR url2[] = {'i','n','d','e','x','.','h','t','m',0};
261 static const WCHAR url3[] = {'f','i','l','e',':','/','/','c',':','\\','I','n','d','e','x','.','h','t','m',0};
262 static const WCHAR url4[] = {'f','i','l','e',':','s','o','m','e','%','2','0','f','i','l','e',
263 '%','2','e','j','p','g',0};
264 static const WCHAR url5[] = {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q',
265 '.','o','r','g',0};
266 static const WCHAR url6[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
267 static const WCHAR url7[] = {'f','t','p',':','/','/','w','i','n','e','h','q','.','o','r','g','/',
268 'f','i','l','e','.','t','e','s','t',0};
269 static const WCHAR url8[] = {'t','e','s','t',':','1','2','3','a','b','c',0};
270 static const WCHAR url9[] =
271 {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q','.','o','r','g',
272 '/','s','i','t','e','/','a','b','o','u','t',0};
273 static const WCHAR url10[] = {'f','i','l','e',':','/','/','s','o','m','e','%','2','0','f','i','l','e',
274 '.','j','p','g',0};
275 static const WCHAR url11[] = {'h','t','t','p',':','/','/','g','o','o','g','l','e','.','*','.',
276 'c','o','m',0};
277 static const WCHAR url4e[] = {'f','i','l','e',':','s','o','m','e',' ','f','i','l','e',
278 '.','j','p','g',0};
280 static const WCHAR path3[] = {'c',':','\\','I','n','d','e','x','.','h','t','m',0};
281 static const WCHAR path4[] = {'s','o','m','e',' ','f','i','l','e','.','j','p','g',0};
283 static const WCHAR wszRes[] = {'r','e','s',0};
284 static const WCHAR wszFile[] = {'f','i','l','e',0};
285 static const WCHAR wszHttp[] = {'h','t','t','p',0};
286 static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
287 static const WCHAR wszEmpty[] = {0};
288 static const WCHAR wszGoogle[] = {'g','o','o','g','l','e','.','*','.','c','o','m',0};
290 static const WCHAR wszWineHQ[] = {'w','w','w','.','w','i','n','e','h','q','.','o','r','g',0};
291 static const WCHAR wszHttpWineHQ[] = {'h','t','t','p',':','/','/','w','w','w','.',
292 'w','i','n','e','h','q','.','o','r','g',0};
293 static const WCHAR wszHttpGoogle[] = {'h','t','t','p',':','/','/','g','o','o','g','l','e',
294 '.','*','.','c','o','m',0};
296 struct parse_test {
297 LPCWSTR url;
298 HRESULT secur_hres;
299 LPCWSTR encoded_url;
300 HRESULT path_hres;
301 LPCWSTR path;
302 LPCWSTR schema;
303 LPCWSTR domain;
304 HRESULT domain_hres;
305 LPCWSTR rootdocument;
306 HRESULT rootdocument_hres;
309 static const struct parse_test parse_tests[] = {
310 {url1, S_OK, url1, E_INVALIDARG, NULL, wszRes, NULL, E_FAIL, NULL, E_FAIL},
311 {url2, E_FAIL, url2, E_INVALIDARG, NULL, wszEmpty, NULL, E_FAIL, NULL, E_FAIL},
312 {url3, E_FAIL, url3, S_OK, path3, wszFile, wszEmpty, S_OK, NULL, E_FAIL},
313 {url4, E_FAIL, url4e, S_OK, path4, wszFile, wszEmpty, S_OK, NULL, E_FAIL},
314 {url5, E_FAIL, url5, E_INVALIDARG, NULL, wszHttp, wszWineHQ, S_OK, wszHttpWineHQ, S_OK},
315 {url6, S_OK, url6, E_INVALIDARG, NULL, wszAbout, NULL, E_FAIL, NULL, E_FAIL},
316 {url11, E_FAIL, url11, E_INVALIDARG, NULL, wszHttp, wszGoogle, S_OK, wszHttpGoogle, S_OK}
319 static void test_CoInternetParseUrl(void)
321 HRESULT hres;
322 DWORD size;
323 int i;
325 static WCHAR buf[4096];
327 memset(buf, 0xf0, sizeof(buf));
328 hres = pCoInternetParseUrl(parse_tests[0].url, PARSE_SCHEMA, 0, buf,
329 3, &size, 0);
330 ok(hres == E_POINTER, "schema failed: %08x, expected E_POINTER\n", hres);
332 for(i=0; i < sizeof(parse_tests)/sizeof(parse_tests[0]); i++) {
333 memset(buf, 0xf0, sizeof(buf));
334 hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_SECURITY_URL, 0, buf,
335 sizeof(buf)/sizeof(WCHAR), &size, 0);
336 ok(hres == parse_tests[i].secur_hres, "[%d] security url failed: %08x, expected %08x\n",
337 i, hres, parse_tests[i].secur_hres);
339 memset(buf, 0xf0, sizeof(buf));
340 hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_ENCODE, 0, buf,
341 sizeof(buf)/sizeof(WCHAR), &size, 0);
342 ok(hres == S_OK, "[%d] encoding failed: %08x\n", i, hres);
343 ok(size == lstrlenW(parse_tests[i].encoded_url), "[%d] wrong size\n", i);
344 ok(!lstrcmpW(parse_tests[i].encoded_url, buf), "[%d] wrong encoded url\n", i);
346 memset(buf, 0xf0, sizeof(buf));
347 hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_PATH_FROM_URL, 0, buf,
348 sizeof(buf)/sizeof(WCHAR), &size, 0);
349 ok(hres == parse_tests[i].path_hres, "[%d] path failed: %08x, expected %08x\n",
350 i, hres, parse_tests[i].path_hres);
351 if(parse_tests[i].path) {
352 ok(size == lstrlenW(parse_tests[i].path), "[%d] wrong size\n", i);
353 ok(!lstrcmpW(parse_tests[i].path, buf), "[%d] wrong path\n", i);
356 memset(buf, 0xf0, sizeof(buf));
357 hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_SCHEMA, 0, buf,
358 sizeof(buf)/sizeof(WCHAR), &size, 0);
359 ok(hres == S_OK, "[%d] schema failed: %08x\n", i, hres);
360 ok(size == lstrlenW(parse_tests[i].schema), "[%d] wrong size\n", i);
361 ok(!lstrcmpW(parse_tests[i].schema, buf), "[%d] wrong schema\n", i);
363 if(memcmp(parse_tests[i].url, wszRes, 3*sizeof(WCHAR))
364 && memcmp(parse_tests[i].url, wszAbout, 5*sizeof(WCHAR))) {
365 memset(buf, 0xf0, sizeof(buf));
366 hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_DOMAIN, 0, buf,
367 sizeof(buf)/sizeof(WCHAR), &size, 0);
368 ok(hres == parse_tests[i].domain_hres, "[%d] domain failed: %08x\n", i, hres);
369 if(parse_tests[i].domain)
370 ok(!lstrcmpW(parse_tests[i].domain, buf), "[%d] wrong domain, received %s\n", i, wine_dbgstr_w(buf));
373 memset(buf, 0xf0, sizeof(buf));
374 hres = pCoInternetParseUrl(parse_tests[i].url, PARSE_ROOTDOCUMENT, 0, buf,
375 sizeof(buf)/sizeof(WCHAR), &size, 0);
376 ok(hres == parse_tests[i].rootdocument_hres, "[%d] rootdocument failed: %08x\n", i, hres);
377 if(parse_tests[i].rootdocument)
378 ok(!lstrcmpW(parse_tests[i].rootdocument, buf), "[%d] wrong rootdocument, received %s\n", i, wine_dbgstr_w(buf));
382 static void test_CoInternetCompareUrl(void)
384 HRESULT hres;
386 hres = pCoInternetCompareUrl(url1, url1, 0);
387 ok(hres == S_OK, "CoInternetCompareUrl failed: %08x\n", hres);
389 hres = pCoInternetCompareUrl(url1, url3, 0);
390 ok(hres == S_FALSE, "CoInternetCompareUrl failed: %08x\n", hres);
392 hres = pCoInternetCompareUrl(url3, url1, 0);
393 ok(hres == S_FALSE, "CoInternetCompareUrl failed: %08x\n", hres);
396 static const struct {
397 LPCWSTR url;
398 DWORD uses_net;
399 } query_info_tests[] = {
400 {url1, 0},
401 {url2, 0},
402 {url3, 0},
403 {url4, 0},
404 {url5, 0},
405 {url6, 0},
406 {url7, 0},
407 {url8, 0}
410 static void test_CoInternetQueryInfo(void)
412 BYTE buf[100];
413 DWORD cb, i;
414 HRESULT hres;
416 for(i=0; i < sizeof(query_info_tests)/sizeof(query_info_tests[0]); i++) {
417 cb = 0xdeadbeef;
418 memset(buf, '?', sizeof(buf));
419 hres = pCoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, sizeof(buf), &cb, 0);
420 ok(hres == S_OK, "[%d] CoInternetQueryInfo failed: %08x\n", i, hres);
421 ok(cb == sizeof(DWORD), "[%d] cb = %d\n", i, cb);
422 ok(*(DWORD*)buf == query_info_tests[i].uses_net, "[%d] ret %x, expected %x\n",
423 i, *(DWORD*)buf, query_info_tests[i].uses_net);
425 hres = pCoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, 3, &cb, 0);
426 ok(hres == E_FAIL, "[%d] CoInternetQueryInfo failed: %08x, expected E_FAIL\n", i, hres);
427 hres = pCoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, NULL, sizeof(buf), &cb, 0);
428 ok(hres == E_FAIL, "[%d] CoInternetQueryInfo failed: %08x, expected E_FAIL\n", i, hres);
430 memset(buf, '?', sizeof(buf));
431 hres = pCoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, sizeof(buf), NULL, 0);
432 ok(hres == S_OK, "[%d] CoInternetQueryInfo failed: %08x\n", i, hres);
433 ok(*(DWORD*)buf == query_info_tests[i].uses_net, "[%d] ret %x, expected %x\n",
434 i, *(DWORD*)buf, query_info_tests[i].uses_net);
438 static const WCHAR mimeTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
439 static const WCHAR mimeTextPlain[] = {'t','e','x','t','/','p','l','a','i','n',0};
440 static const WCHAR mimeTextRichtext[] = {'t','e','x','t','/','r','i','c','h','t','e','x','t',0};
441 static const WCHAR mimeAppOctetStream[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
442 'o','c','t','e','t','-','s','t','r','e','a','m',0};
443 static const WCHAR mimeImagePjpeg[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
444 static const WCHAR mimeImageGif[] = {'i','m','a','g','e','/','g','i','f',0};
445 static const WCHAR mimeImageBmp[] = {'i','m','a','g','e','/','b','m','p',0};
446 static const WCHAR mimeImageXPng[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
447 static const WCHAR mimeImageTiff[] = {'i','m','a','g','e','/','t','i','f','f',0};
448 static const WCHAR mimeVideoAvi[] = {'v','i','d','e','o','/','a','v','i',0};
449 static const WCHAR mimeVideoMpeg[] = {'v','i','d','e','o','/','m','p','e','g',0};
450 static const WCHAR mimeAppPostscript[] =
451 {'a','p','p','l','i','c','a','t','i','o','n','/','p','o','s','t','s','c','r','i','p','t',0};
452 static const WCHAR mimeAppXCompressed[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
453 'x','-','c','o','m','p','r','e','s','s','e','d',0};
454 static const WCHAR mimeAppXZip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
455 'x','-','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
456 static const WCHAR mimeAppXGzip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
457 'x','-','g','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
458 static const WCHAR mimeAppJava[] = {'a','p','p','l','i','c','a','t','i','o','n','/','j','a','v','a',0};
459 static const WCHAR mimeAppPdf[] = {'a','p','p','l','i','c','a','t','i','o','n','/','p','d','f',0};
460 static const WCHAR mimeAppXMSDownload[] =
461 {'a','p','p','l','i','c','a','t','i','o','n','/','x','-','m','s','d','o','w','n','l','o','a','d',0};
462 static const WCHAR mimeAudioWav[] = {'a','u','d','i','o','/','w','a','v',0};
463 static const WCHAR mimeAudioBasic[] = {'a','u','d','i','o','/','b','a','s','i','c',0};
465 static const struct {
466 LPCWSTR url;
467 LPCWSTR mime;
468 HRESULT hres;
469 } mime_tests[] = {
470 {url1, mimeTextHtml, S_OK},
471 {url2, mimeTextHtml, S_OK},
472 {url3, mimeTextHtml, S_OK},
473 {url4, NULL, E_FAIL},
474 {url5, NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)},
475 {url6, NULL, E_FAIL},
476 {url7, NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)}
479 static BYTE data1[] = "test data\n";
480 static BYTE data2[] = {31,'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0};
481 static BYTE data3[] = {0,0,0};
482 static BYTE data4[] = {'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0,0};
483 static BYTE data5[] = {0xa,0xa,0xa,'x',32,'x',0};
484 static BYTE data6[] = {0xfa,0xfa,0xfa,0xfa,'\n','\r','\t','x','x','x',1};
485 static BYTE data7[] = "<html>blahblah";
486 static BYTE data8[] = {'t','e','s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0};
487 static BYTE data9[] = {'t','e',0,'s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0};
488 static BYTE data10[] = "<HtmL>blahblah";
489 static BYTE data11[] = "blah<HTML>blahblah";
490 static BYTE data12[] = "blah<HTMLblahblah";
491 static BYTE data13[] = "blahHTML>blahblah";
492 static BYTE data14[] = "blah<HTMblahblah";
493 static BYTE data15[] = {0xff,0xd8};
494 static BYTE data16[] = {0xff,0xd8,'h'};
495 static BYTE data17[] = {0,0xff,0xd8};
496 static BYTE data18[] = {0xff,0xd8,'<','h','t','m','l','>'};
497 static BYTE data19[] = {'G','I','F','8','7','a'};
498 static BYTE data20[] = {'G','I','F','8','9','a'};
499 static BYTE data21[] = {'G','I','F','8','7'};
500 static BYTE data22[] = {'G','i','F','8','7','a'};
501 static BYTE data23[] = {'G','i','F','8','8','a'};
502 static BYTE data24[] = {'g','i','f','8','7','a'};
503 static BYTE data25[] = {'G','i','F','8','7','A'};
504 static BYTE data26[] = {'G','i','F','8','7','a','<','h','t','m','l','>'};
505 static BYTE data27[] = {0x30,'G','i','F','8','7','A'};
506 static BYTE data28[] = {0x42,0x4d,0x6e,0x42,0x1c,0x00,0x00,0x00,0x00,0x00,0x36,0x00,0x00,0x00};
507 static BYTE data29[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x','x'};
508 static BYTE data30[] = {0x42,0x4d,'x','x','x','x',0x00,0x01,0x00,0x00,'x','x','x','x'};
509 static BYTE data31[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'<','h','t','m','l','>'};
510 static BYTE data32[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'};
511 static BYTE data33[] = {0x00,0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'};
512 static BYTE data34[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'};
513 static BYTE data35[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x','x','x','x',0};
514 static BYTE data36[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,'x','x'};
515 static BYTE data37[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'<','h','t','m','l','>'};
516 static BYTE data38[] = {0x00,0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'};
517 static BYTE data39[] = {0x4d,0x4d,0x00,0x2a};
518 static BYTE data40[] = {0x4d,0x4d,0x00,0x2a,'<','h','t','m','l','>',0};
519 static BYTE data41[] = {0x4d,0x4d,0xff};
520 static BYTE data42[] = {0x4d,0x4d};
521 static BYTE data43[] = {0x00,0x4d,0x4d,0x00};
522 static BYTE data44[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff};
523 static BYTE data45[] = {'R','I','F','f',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff};
524 static BYTE data46[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20};
525 static BYTE data47[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x21,0xff};
526 static BYTE data48[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,'<','h','t','m','l','>'};
527 static BYTE data49[] = {'R','I','F','F',0x0f,0x0f,0xf0,0xf0,'A','V','I',0x20,0xf0,0x00};
528 static BYTE data50[] = {0x00,0x00,0x01,0xb3,0xff};
529 static BYTE data51[] = {0x00,0x00,0x01,0xba,0xff};
530 static BYTE data52[] = {0x00,0x00,0x01,0xb8,0xff};
531 static BYTE data53[] = {0x00,0x00,0x01,0xba};
532 static BYTE data54[] = {0x00,0x00,0x01,0xba,'<','h','t','m','l','>'};
533 static BYTE data55[] = {0x1f,0x8b,'x'};
534 static BYTE data56[] = {0x1f};
535 static BYTE data57[] = {0x1f,0x8b,'<','h','t','m','l','>','t','e','s','t',0};
536 static BYTE data58[] = {0x1f,0x8b};
537 static BYTE data59[] = {0x50,0x4b,'x'};
538 static BYTE data60[] = {0x50,0x4b};
539 static BYTE data61[] = {0x50,0x4b,'<','h','t','m','l','>',0};
540 static BYTE data62[] = {0xca,0xfe,0xba,0xbe,'x'};
541 static BYTE data63[] = {0xca,0xfe,0xba,0xbe};
542 static BYTE data64[] = {0xca,0xfe,0xba,0xbe,'<','h','t','m','l','>',0};
543 static BYTE data65[] = {0x25,0x50,0x44,0x46,'x'};
544 static BYTE data66[] = {0x25,0x50,0x44,0x46};
545 static BYTE data67[] = {0x25,0x50,0x44,0x46,'x','<','h','t','m','l','>'};
546 static BYTE data68[] = {'M','Z','x'};
547 static BYTE data69[] = {'M','Z'};
548 static BYTE data70[] = {'M','Z','<','h','t','m','l','>',0xff};
549 static BYTE data71[] = {'{','\\','r','t','f',0};
550 static BYTE data72[] = {'{','\\','r','t','f'};
551 static BYTE data73[] = {' ','{','\\','r','t','f',' '};
552 static BYTE data74[] = {'{','\\','r','t','f','<','h','t','m','l','>',' '};
553 static BYTE data75[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V','E',0xff};
554 static BYTE data76[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V','E'};
555 static BYTE data77[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V',0xff,0xff};
556 static BYTE data78[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'<','h','t','m','l','>',0xff};
557 static BYTE data79[] = {'%','!',0xff};
558 static BYTE data80[] = {'%','!'};
559 static BYTE data81[] = {'%','!','P','S','<','h','t','m','l','>'};
560 static BYTE data82[] = {'.','s','n','d',0};
561 static BYTE data83[] = {'.','s','n','d'};
562 static BYTE data84[] = {'.','s','n','d',0,'<','h','t','m','l','>',1,1};
563 static BYTE data85[] = {'.','S','N','D',0};
565 static const struct {
566 BYTE *data;
567 DWORD size;
568 LPCWSTR mime, mime_alt;
569 } mime_tests2[] = {
570 {data1, sizeof(data1), mimeTextPlain},
571 {data2, sizeof(data2), mimeAppOctetStream},
572 {data3, sizeof(data3), mimeAppOctetStream},
573 {data4, sizeof(data4), mimeAppOctetStream},
574 {data5, sizeof(data5), mimeTextPlain},
575 {data6, sizeof(data6), mimeTextPlain},
576 {data7, sizeof(data7), mimeTextHtml, mimeTextPlain /* IE8 */},
577 {data8, sizeof(data8), mimeTextHtml, mimeTextPlain /* IE8 */},
578 {data9, sizeof(data9), mimeTextHtml, mimeImagePjpeg /* IE8 */},
579 {data10, sizeof(data10), mimeTextHtml, mimeTextPlain /* IE8 */},
580 {data11, sizeof(data11), mimeTextHtml, mimeTextPlain /* IE8 */},
581 {data12, sizeof(data12), mimeTextHtml, mimeTextPlain /* IE8 */},
582 {data13, sizeof(data13), mimeTextPlain},
583 {data14, sizeof(data14), mimeTextPlain},
584 {data15, sizeof(data15), mimeTextPlain},
585 {data16, sizeof(data16), mimeImagePjpeg},
586 {data17, sizeof(data17), mimeAppOctetStream},
587 {data18, sizeof(data18), mimeTextHtml},
588 {data19, sizeof(data19), mimeImageGif},
589 {data20, sizeof(data20), mimeImageGif},
590 {data21, sizeof(data21), mimeTextPlain},
591 {data22, sizeof(data22), mimeImageGif},
592 {data23, sizeof(data23), mimeTextPlain},
593 {data24, sizeof(data24), mimeImageGif},
594 {data25, sizeof(data25), mimeImageGif},
595 {data26, sizeof(data26), mimeTextHtml, mimeImageGif /* IE8 */},
596 {data27, sizeof(data27), mimeTextPlain},
597 {data28, sizeof(data28), mimeImageBmp},
598 {data29, sizeof(data29), mimeImageBmp},
599 {data30, sizeof(data30), mimeAppOctetStream},
600 {data31, sizeof(data31), mimeTextHtml, mimeImageBmp /* IE8 */},
601 {data32, sizeof(data32), mimeAppOctetStream},
602 {data33, sizeof(data33), mimeAppOctetStream},
603 {data34, sizeof(data34), mimeImageXPng},
604 {data35, sizeof(data35), mimeImageXPng},
605 {data36, sizeof(data36), mimeAppOctetStream},
606 {data37, sizeof(data37), mimeTextHtml, mimeImageXPng /* IE8 */},
607 {data38, sizeof(data38), mimeAppOctetStream},
608 {data39, sizeof(data39), mimeImageTiff},
609 {data40, sizeof(data40), mimeTextHtml, mimeImageTiff /* IE8 */},
610 {data41, sizeof(data41), mimeImageTiff},
611 {data42, sizeof(data42), mimeTextPlain},
612 {data43, sizeof(data43), mimeAppOctetStream},
613 {data44, sizeof(data44), mimeVideoAvi},
614 {data45, sizeof(data45), mimeTextPlain},
615 {data46, sizeof(data46), mimeTextPlain},
616 {data47, sizeof(data47), mimeTextPlain},
617 {data48, sizeof(data48), mimeTextHtml, mimeVideoAvi /* IE8 */},
618 {data49, sizeof(data49), mimeVideoAvi},
619 {data50, sizeof(data50), mimeVideoMpeg},
620 {data51, sizeof(data51), mimeVideoMpeg},
621 {data52, sizeof(data52), mimeAppOctetStream},
622 {data53, sizeof(data53), mimeAppOctetStream},
623 {data54, sizeof(data54), mimeTextHtml, mimeVideoMpeg /* IE8 */},
624 {data55, sizeof(data55), mimeAppXGzip},
625 {data56, sizeof(data56), mimeTextPlain},
626 {data57, sizeof(data57), mimeTextHtml, mimeAppXGzip /* IE8 */},
627 {data58, sizeof(data58), mimeAppOctetStream},
628 {data59, sizeof(data59), mimeAppXZip},
629 {data60, sizeof(data60), mimeTextPlain},
630 {data61, sizeof(data61), mimeTextHtml, mimeAppXZip /* IE8 */},
631 {data62, sizeof(data62), mimeAppJava},
632 {data63, sizeof(data63), mimeTextPlain},
633 {data64, sizeof(data64), mimeTextHtml, mimeAppJava /* IE8 */},
634 {data65, sizeof(data65), mimeAppPdf},
635 {data66, sizeof(data66), mimeTextPlain},
636 {data67, sizeof(data67), mimeTextHtml, mimeAppPdf /* IE8 */},
637 {data68, sizeof(data68), mimeAppXMSDownload},
638 {data69, sizeof(data69), mimeTextPlain},
639 {data70, sizeof(data70), mimeTextHtml, mimeAppXMSDownload /* IE8 */},
640 {data71, sizeof(data71), mimeTextRichtext},
641 {data72, sizeof(data72), mimeTextPlain},
642 {data73, sizeof(data73), mimeTextPlain},
643 {data74, sizeof(data74), mimeTextHtml, mimeTextRichtext /* IE8 */},
644 {data75, sizeof(data75), mimeAudioWav},
645 {data76, sizeof(data76), mimeTextPlain},
646 {data77, sizeof(data77), mimeTextPlain},
647 {data78, sizeof(data78), mimeTextHtml, mimeTextPlain /* IE8 */},
648 {data79, sizeof(data79), mimeAppPostscript},
649 {data80, sizeof(data80), mimeTextPlain},
650 {data81, sizeof(data81), mimeTextHtml, mimeAppPostscript /* IE8 */},
651 {data82, sizeof(data82), mimeAudioBasic},
652 {data83, sizeof(data83), mimeTextPlain},
653 {data84, sizeof(data84), mimeTextHtml, mimeAudioBasic /* IE8 */},
654 {data85, sizeof(data85), mimeTextPlain}
657 static void test_FindMimeFromData(void)
659 HRESULT hres;
660 LPWSTR mime;
661 int i;
663 for(i=0; i<sizeof(mime_tests)/sizeof(mime_tests[0]); i++) {
664 mime = (LPWSTR)0xf0f0f0f0;
665 hres = pFindMimeFromData(NULL, mime_tests[i].url, NULL, 0, NULL, 0, &mime, 0);
666 if(mime_tests[i].mime) {
667 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
668 ok(!lstrcmpW(mime, mime_tests[i].mime), "[%d] wrong mime\n", i);
669 CoTaskMemFree(mime);
670 }else {
671 ok(hres == E_FAIL || hres == mime_tests[i].hres,
672 "[%d] FindMimeFromData failed: %08x, expected %08x\n",
673 i, hres, mime_tests[i].hres);
674 ok(mime == (LPWSTR)0xf0f0f0f0, "[%d] mime != 0xf0f0f0f0\n", i);
677 mime = (LPWSTR)0xf0f0f0f0;
678 hres = pFindMimeFromData(NULL, mime_tests[i].url, NULL, 0, mimeTextPlain, 0, &mime, 0);
679 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
680 ok(!lstrcmpW(mime, mimeTextPlain), "[%d] wrong mime\n", i);
681 CoTaskMemFree(mime);
683 mime = (LPWSTR)0xf0f0f0f0;
684 hres = pFindMimeFromData(NULL, mime_tests[i].url, NULL, 0, mimeAppOctetStream, 0, &mime, 0);
685 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
686 ok(!lstrcmpW(mime, mimeAppOctetStream), "[%d] wrong mime\n", i);
687 CoTaskMemFree(mime);
690 for(i=0; i < sizeof(mime_tests2)/sizeof(mime_tests2[0]); i++) {
691 hres = pFindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
692 NULL, 0, &mime, 0);
693 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
694 ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime: %s\n", i, wine_dbgstr_w(mime));
695 CoTaskMemFree(mime);
697 hres = pFindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
698 mimeTextHtml, 0, &mime, 0);
699 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
700 if(!lstrcmpW(mimeAppOctetStream, mime_tests2[i].mime)
701 || !lstrcmpW(mimeTextPlain, mime_tests2[i].mime))
702 ok(!lstrcmpW(mime, mimeTextHtml), "[%d] wrong mime\n", i);
703 else
704 ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
705 CoTaskMemFree(mime);
707 hres = pFindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
708 mimeImagePjpeg, 0, &mime, 0);
709 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
710 if(!lstrcmpW(mimeAppOctetStream, mime_tests2[i].mime) || i == 17)
711 ok(!lstrcmpW(mime, mimeImagePjpeg), "[%d] wrong mime\n", i);
712 else
713 ok(!lstrcmpW(mime, mime_tests2[i].mime) ||
714 (mime_tests2[i].mime_alt && !lstrcmpW(mime, mime_tests2[i].mime_alt)),
715 "[%d] wrong mime, got %s\n", i, wine_dbgstr_w(mime));
717 CoTaskMemFree(mime);
720 hres = pFindMimeFromData(NULL, url1, data1, sizeof(data1), NULL, 0, &mime, 0);
721 ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
722 ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
723 CoTaskMemFree(mime);
725 hres = pFindMimeFromData(NULL, url1, data1, sizeof(data1), mimeAppOctetStream, 0, &mime, 0);
726 ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
727 ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
728 CoTaskMemFree(mime);
730 hres = pFindMimeFromData(NULL, url4, data1, sizeof(data1), mimeAppOctetStream, 0, &mime, 0);
731 ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
732 ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
733 CoTaskMemFree(mime);
735 hres = pFindMimeFromData(NULL, NULL, NULL, 0, NULL, 0, &mime, 0);
736 ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, excepted E_INVALIDARG\n", hres);
738 hres = pFindMimeFromData(NULL, NULL, NULL, 0, mimeTextPlain, 0, &mime, 0);
739 ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
741 hres = pFindMimeFromData(NULL, NULL, data1, 0, NULL, 0, &mime, 0);
742 ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
744 hres = pFindMimeFromData(NULL, url1, data1, 0, NULL, 0, &mime, 0);
745 ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
747 hres = pFindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, &mime, 0);
748 ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
749 ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
750 CoTaskMemFree(mime);
752 hres = pFindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, NULL, 0);
753 ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
756 static void register_protocols(void)
758 IInternetSession *session;
759 IClassFactory *factory;
760 HRESULT hres;
762 static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
764 hres = pCoInternetGetSession(0, &session, 0);
765 ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
766 if(FAILED(hres))
767 return;
769 hres = CoGetClassObject(&CLSID_AboutProtocol, CLSCTX_INPROC_SERVER, NULL,
770 &IID_IClassFactory, (void**)&factory);
771 ok(hres == S_OK, "Coud not get AboutProtocol factory: %08x\n", hres);
772 if(FAILED(hres))
773 return;
775 IInternetSession_RegisterNameSpace(session, factory, &CLSID_AboutProtocol,
776 wszAbout, 0, NULL, 0);
777 IClassFactory_Release(factory);
779 IInternetSession_Release(session);
782 static HRESULT WINAPI InternetProtocolInfo_QueryInterface(IInternetProtocolInfo *iface,
783 REFIID riid, void **ppv)
785 ok(0, "unexpected call\n");
786 return E_NOINTERFACE;
789 static ULONG WINAPI InternetProtocolInfo_AddRef(IInternetProtocolInfo *iface)
791 return 2;
794 static ULONG WINAPI InternetProtocolInfo_Release(IInternetProtocolInfo *iface)
796 return 1;
799 static HRESULT WINAPI InternetProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
800 PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult,
801 DWORD *pcchResult, DWORD dwReserved)
803 CHECK_EXPECT2(ParseUrl);
805 if(ParseAction == PARSE_SECURITY_URL) {
806 if(pcchResult)
807 *pcchResult = sizeof(url1)/sizeof(WCHAR);
809 if(cchResult<sizeof(url1)/sizeof(WCHAR))
810 return S_FALSE;
812 memcpy(pwzResult, url1, sizeof(url1));
813 return S_OK;
816 return E_NOTIMPL;
819 static HRESULT WINAPI InternetProtocolInfo_CombineUrl(IInternetProtocolInfo *iface,
820 LPCWSTR pwzBaseUrl, LPCWSTR pwzRelativeUrl, DWORD dwCombineFlags,
821 LPWSTR pwzResult, DWORD cchResult, DWORD *pcchResult, DWORD dwReserved)
823 ok(0, "unexpected call\n");
824 return E_NOTIMPL;
827 static HRESULT WINAPI InternetProtocolInfo_CompareUrl(IInternetProtocolInfo *iface,
828 LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags)
830 ok(0, "unexpected call\n");
831 return E_NOTIMPL;
834 static HRESULT WINAPI InternetProtocolInfo_QueryInfo(IInternetProtocolInfo *iface,
835 LPCWSTR pwzUrl, QUERYOPTION OueryOption, DWORD dwQueryFlags, LPVOID pBuffer,
836 DWORD cbBuffer, DWORD *pcbBuf, DWORD dwReserved)
838 ok(0, "unexpected call\n");
839 return E_NOTIMPL;
842 static const IInternetProtocolInfoVtbl InternetProtocolInfoVtbl = {
843 InternetProtocolInfo_QueryInterface,
844 InternetProtocolInfo_AddRef,
845 InternetProtocolInfo_Release,
846 InternetProtocolInfo_ParseUrl,
847 InternetProtocolInfo_CombineUrl,
848 InternetProtocolInfo_CompareUrl,
849 InternetProtocolInfo_QueryInfo
852 static IInternetProtocolInfo protocol_info = { &InternetProtocolInfoVtbl };
854 static HRESULT qiret;
855 static IClassFactory *expect_cf;
857 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
859 if(IsEqualGUID(&IID_IInternetProtocolInfo, riid)) {
860 CHECK_EXPECT2(QI_IInternetProtocolInfo);
861 ok(iface == expect_cf, "unexpected iface\n");
862 *ppv = &protocol_info;
863 return qiret;
866 ok(0, "unexpected call\n");
867 return E_NOINTERFACE;
870 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
872 return 2;
875 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
877 return 1;
880 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
881 REFIID riid, void **ppv)
883 ok(0, "unexpected call\n");
884 return E_NOTIMPL;
887 static HRESULT WINAPI ProtocolCF_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
888 REFIID riid, void **ppv)
890 CHECK_EXPECT(CreateInstance);
892 ok(iface == expect_cf, "unexpected iface\n");
893 ok(pOuter == NULL, "pOuter = %p\n", pOuter);
894 ok(IsEqualGUID(&IID_IInternetProtocolInfo, riid), "unexpected riid\n");
895 ok(ppv != NULL, "ppv == NULL\n");
897 *ppv = &protocol_info;
898 return S_OK;
901 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
903 ok(0, "unexpected call\n");
904 return S_OK;
907 static const IClassFactoryVtbl ClassFactoryVtbl = {
908 ClassFactory_QueryInterface,
909 ClassFactory_AddRef,
910 ClassFactory_Release,
911 ClassFactory_CreateInstance,
912 ClassFactory_LockServer
915 static const IClassFactoryVtbl ProtocolCFVtbl = {
916 ClassFactory_QueryInterface,
917 ClassFactory_AddRef,
918 ClassFactory_Release,
919 ProtocolCF_CreateInstance,
920 ClassFactory_LockServer
923 static IClassFactory test_protocol_cf = { &ProtocolCFVtbl };
924 static IClassFactory test_protocol_cf2 = { &ProtocolCFVtbl };
925 static IClassFactory test_cf = { &ClassFactoryVtbl };
927 static void test_NameSpace(void)
929 IInternetSession *session;
930 WCHAR buf[200];
931 LPWSTR sec_url;
932 DWORD size;
933 HRESULT hres;
935 static const WCHAR wszTest[] = {'t','e','s','t',0};
937 hres = pCoInternetGetSession(0, &session, 0);
938 ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
939 if(FAILED(hres))
940 return;
942 hres = IInternetSession_RegisterNameSpace(session, NULL, &IID_NULL,
943 wszTest, 0, NULL, 0);
944 ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
946 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
947 NULL, 0, NULL, 0);
948 ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
950 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
951 wszTest, 0, NULL, 0);
952 ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
954 qiret = E_NOINTERFACE;
955 expect_cf = &test_protocol_cf;
956 SET_EXPECT(QI_IInternetProtocolInfo);
957 SET_EXPECT(CreateInstance);
958 SET_EXPECT(ParseUrl);
960 hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
961 &size, 0);
962 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
964 CHECK_CALLED(QI_IInternetProtocolInfo);
965 CHECK_CALLED(CreateInstance);
966 CHECK_CALLED(ParseUrl);
968 qiret = S_OK;
969 SET_EXPECT(QI_IInternetProtocolInfo);
970 SET_EXPECT(ParseUrl);
972 hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
973 &size, 0);
974 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
976 CHECK_CALLED(QI_IInternetProtocolInfo);
977 CHECK_CALLED(ParseUrl);
979 SET_EXPECT(QI_IInternetProtocolInfo);
980 SET_EXPECT(ParseUrl);
982 hres = pCoInternetParseUrl(url8, PARSE_SECURITY_URL, 0, buf,
983 sizeof(buf)/sizeof(WCHAR), &size, 0);
984 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
985 ok(size == sizeof(url1)/sizeof(WCHAR), "Size = %d\n", size);
986 if(size == sizeof(url1)/sizeof(WCHAR))
987 ok(!memcmp(buf, url1, sizeof(url1)), "Encoded url = %s\n", wine_dbgstr_w(buf));
989 CHECK_CALLED(QI_IInternetProtocolInfo);
990 CHECK_CALLED(ParseUrl);
992 SET_EXPECT(QI_IInternetProtocolInfo);
993 SET_EXPECT(ParseUrl);
995 if (pCoInternetGetSecurityUrl) {
996 hres = pCoInternetGetSecurityUrl(url8, &sec_url, PSU_SECURITY_URL_ONLY, 0);
997 ok(hres == S_OK, "CoInternetGetSecurityUrl failed: %08x\n", hres);
998 if(hres == S_OK) {
999 ok(lstrlenW(sec_url)>sizeof(wszFile)/sizeof(WCHAR) &&
1000 !memcmp(sec_url, wszFile, sizeof(wszFile)-sizeof(WCHAR)),
1001 "Encoded url = %s\n", wine_dbgstr_w(sec_url));
1002 CoTaskMemFree(sec_url);
1005 CHECK_CALLED(QI_IInternetProtocolInfo);
1006 CHECK_CALLED(ParseUrl);
1009 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1010 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1012 hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1013 &size, 0);
1014 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1016 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf2, &IID_NULL,
1017 wszTest, 0, NULL, 0);
1018 ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1020 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
1021 wszTest, 0, NULL, 0);
1022 ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1024 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
1025 wszTest, 0, NULL, 0);
1026 ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1028 SET_EXPECT(QI_IInternetProtocolInfo);
1029 SET_EXPECT(ParseUrl);
1031 hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1032 &size, 0);
1033 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1035 CHECK_CALLED(QI_IInternetProtocolInfo);
1036 CHECK_CALLED(ParseUrl);
1038 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1039 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1041 SET_EXPECT(QI_IInternetProtocolInfo);
1042 SET_EXPECT(ParseUrl);
1044 hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1045 &size, 0);
1046 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1048 CHECK_CALLED(QI_IInternetProtocolInfo);
1049 CHECK_CALLED(ParseUrl);
1051 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1052 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1054 expect_cf = &test_protocol_cf2;
1055 SET_EXPECT(QI_IInternetProtocolInfo);
1056 SET_EXPECT(ParseUrl);
1058 hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1059 &size, 0);
1060 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1062 CHECK_CALLED(QI_IInternetProtocolInfo);
1063 CHECK_CALLED(ParseUrl);
1065 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1066 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1067 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1068 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1069 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, NULL);
1070 ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres);
1071 hres = IInternetSession_UnregisterNameSpace(session, NULL, wszTest);
1072 ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres);
1074 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf2, wszTest);
1075 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1077 hres = pCoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1078 &size, 0);
1079 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1081 IInternetSession_Release(session);
1084 static void test_MimeFilter(void)
1086 IInternetSession *session;
1087 HRESULT hres;
1089 static const WCHAR mimeW[] = {'t','e','s','t','/','m','i','m','e',0};
1091 hres = pCoInternetGetSession(0, &session, 0);
1092 ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
1093 if(FAILED(hres))
1094 return;
1096 hres = IInternetSession_RegisterMimeFilter(session, &test_cf, &IID_NULL, mimeW);
1097 ok(hres == S_OK, "RegisterMimeFilter failed: %08x\n", hres);
1099 hres = IInternetSession_UnregisterMimeFilter(session, &test_cf, mimeW);
1100 ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres);
1102 hres = IInternetSession_UnregisterMimeFilter(session, &test_cf, mimeW);
1103 ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres);
1105 hres = IInternetSession_UnregisterMimeFilter(session, (void*)0xdeadbeef, mimeW);
1106 ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres);
1108 IInternetSession_Release(session);
1111 static ULONG WINAPI unk_Release(IUnknown *iface)
1113 CHECK_EXPECT(unk_Release);
1114 return 0;
1117 static const IUnknownVtbl unk_vtbl = {
1118 (void*)0xdeadbeef,
1119 (void*)0xdeadbeef,
1120 unk_Release
1123 static void test_ReleaseBindInfo(void)
1125 BINDINFO bi;
1126 IUnknown unk = { &unk_vtbl };
1128 pReleaseBindInfo(NULL); /* shouldn't crash */
1130 memset(&bi, 0, sizeof(bi));
1131 bi.cbSize = sizeof(BINDINFO);
1132 bi.pUnk = &unk;
1133 SET_EXPECT(unk_Release);
1134 pReleaseBindInfo(&bi);
1135 ok(bi.cbSize == sizeof(BINDINFO), "bi.cbSize=%d\n", bi.cbSize);
1136 ok(bi.pUnk == NULL, "bi.pUnk=%p, expected NULL\n", bi.pUnk);
1137 CHECK_CALLED(unk_Release);
1139 memset(&bi, 0, sizeof(bi));
1140 bi.cbSize = offsetof(BINDINFO, pUnk);
1141 bi.pUnk = &unk;
1142 pReleaseBindInfo(&bi);
1143 ok(bi.cbSize == offsetof(BINDINFO, pUnk), "bi.cbSize=%d\n", bi.cbSize);
1144 ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk);
1146 memset(&bi, 0, sizeof(bi));
1147 bi.pUnk = &unk;
1148 pReleaseBindInfo(&bi);
1149 ok(!bi.cbSize, "bi.cbSize=%d, expected 0\n", bi.cbSize);
1150 ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk);
1153 static void test_CopyStgMedium(void)
1155 STGMEDIUM src, dst;
1156 HGLOBAL empty;
1157 HRESULT hres;
1159 static WCHAR fileW[] = {'f','i','l','e',0};
1161 memset(&src, 0xf0, sizeof(src));
1162 memset(&dst, 0xe0, sizeof(dst));
1163 memset(&empty, 0xf0, sizeof(empty));
1164 src.tymed = TYMED_NULL;
1165 src.pUnkForRelease = NULL;
1166 hres = pCopyStgMedium(&src, &dst);
1167 ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres);
1168 ok(dst.tymed == TYMED_NULL, "tymed=%d\n", dst.tymed);
1169 ok(dst.u.hGlobal == empty, "u=%p\n", dst.u.hGlobal);
1170 ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease);
1172 memset(&dst, 0xe0, sizeof(dst));
1173 src.tymed = TYMED_ISTREAM;
1174 src.u.pstm = NULL;
1175 src.pUnkForRelease = NULL;
1176 hres = pCopyStgMedium(&src, &dst);
1177 ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres);
1178 ok(dst.tymed == TYMED_ISTREAM, "tymed=%d\n", dst.tymed);
1179 ok(!dst.u.pstm, "pstm=%p\n", dst.u.pstm);
1180 ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease);
1182 memset(&dst, 0xe0, sizeof(dst));
1183 src.tymed = TYMED_FILE;
1184 src.u.lpszFileName = fileW;
1185 src.pUnkForRelease = NULL;
1186 hres = pCopyStgMedium(&src, &dst);
1187 ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres);
1188 ok(dst.tymed == TYMED_FILE, "tymed=%d\n", dst.tymed);
1189 ok(dst.u.lpszFileName && dst.u.lpszFileName != fileW, "lpszFileName=%p\n", dst.u.lpszFileName);
1190 ok(!lstrcmpW(dst.u.lpszFileName, fileW), "wrong file name\n");
1191 ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease);
1192 ReleaseStgMedium(&dst);
1194 hres = pCopyStgMedium(&src, NULL);
1195 ok(hres == E_POINTER, "CopyStgMedium failed: %08x, expected E_POINTER\n", hres);
1196 hres = pCopyStgMedium(NULL, &dst);
1197 ok(hres == E_POINTER, "CopyStgMedium failed: %08x, expected E_POINTER\n", hres);
1200 static void test_UrlMkGetSessionOption(void)
1202 DWORD encoding, size;
1203 HRESULT hres;
1205 size = encoding = 0xdeadbeef;
1206 hres = pUrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1207 sizeof(encoding), &size, 0);
1208 ok(hres == S_OK, "UrlMkGetSessionOption failed: %08x\n", hres);
1209 ok(encoding != 0xdeadbeef, "encoding not changed\n");
1210 ok(size == sizeof(encoding), "size=%d\n", size);
1212 size = encoding = 0xdeadbeef;
1213 hres = pUrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1214 sizeof(encoding)+1, &size, 0);
1215 ok(hres == S_OK, "UrlMkGetSessionOption failed: %08x\n", hres);
1216 ok(encoding != 0xdeadbeef, "encoding not changed\n");
1217 ok(size == sizeof(encoding), "size=%d\n", size);
1219 size = encoding = 0xdeadbeef;
1220 hres = pUrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1221 sizeof(encoding)-1, &size, 0);
1222 ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1223 ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1224 ok(size == 0xdeadbeef, "size=%d\n", size);
1226 size = encoding = 0xdeadbeef;
1227 hres = pUrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, NULL,
1228 sizeof(encoding)-1, &size, 0);
1229 ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1230 ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1231 ok(size == 0xdeadbeef, "size=%d\n", size);
1233 encoding = 0xdeadbeef;
1234 hres = pUrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1235 sizeof(encoding)-1, NULL, 0);
1236 ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1237 ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1240 static void test_user_agent(void)
1242 static const CHAR expected[] = "Mozilla/4.0 (compatible; MSIE ";
1243 static char test_str[] = "test";
1244 static char test2_str[] = "test\0test";
1245 static CHAR str[3];
1246 LPSTR str2 = NULL;
1247 HRESULT hres;
1248 DWORD size, saved;
1250 hres = pObtainUserAgentString(0, NULL, NULL);
1251 ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1253 size = 100;
1254 hres = pObtainUserAgentString(0, NULL, &size);
1255 ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1256 ok(size == 100, "size=%d, expected %d\n", size, 100);
1258 size = 0;
1259 hres = pObtainUserAgentString(0, str, &size);
1260 ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1261 ok(size > 0, "size=%d, expected non-zero\n", size);
1263 size = 2;
1264 str[0] = 'a';
1265 hres = pObtainUserAgentString(0, str, &size);
1266 ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1267 ok(size > 0, "size=%d, expected non-zero\n", size);
1268 ok(str[0] == 'a', "str[0]=%c, expected 'a'\n", str[0]);
1270 size = 0;
1271 hres = pObtainUserAgentString(1, str, &size);
1272 ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1273 ok(size > 0, "size=%d, expected non-zero\n", size);
1275 str2 = HeapAlloc(GetProcessHeap(), 0, (size+20)*sizeof(CHAR));
1276 saved = size;
1277 hres = pObtainUserAgentString(0, str2, &size);
1278 ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres);
1279 ok(size == saved, "size=%d, expected %d\n", size, saved);
1280 ok(strlen(expected) <= strlen(str2) &&
1281 !memcmp(expected, str2, strlen(expected)*sizeof(CHAR)),
1282 "user agent was \"%s\", expected to start with \"%s\"\n",
1283 str2, expected);
1285 size = saved+10;
1286 hres = pObtainUserAgentString(0, str2, &size);
1287 ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres);
1288 ok(size == saved, "size=%d, expected %d\n", size, saved);
1290 size = 0;
1291 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, NULL, 0, &size, 0);
1292 ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres);
1293 ok(size, "size == 0\n");
1295 size = 0xdeadbeef;
1296 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, NULL, 1000, &size, 0);
1297 ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1298 ok(size, "size == 0\n");
1300 saved = size;
1301 size = 0;
1302 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved+10, &size, 0);
1303 ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres);
1304 ok(size == saved, "size = %d, expected %d\n", size, saved);
1305 ok(sizeof(expected) <= strlen(str2) && !memcmp(expected, str2, sizeof(expected)-1),
1306 "user agent was \"%s\", expected to start with \"%s\"\n",
1307 str2, expected);
1309 size = 0;
1310 str2[0] = 0;
1311 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, &size, 0);
1312 ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres);
1313 ok(size == saved, "size = %d, expected %d\n", size, saved);
1314 ok(sizeof(expected) <= strlen(str2) && !memcmp(expected, str2, sizeof(expected)-1),
1315 "user agent was \"%s\", expected to start with \"%s\"\n",
1316 str2, expected);
1318 size = saved;
1319 str2[0] = 0;
1320 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved-1, &size, 0);
1321 ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres);
1322 ok(size == saved, "size = %d, expected %d\n", size, saved);
1323 ok(!str2[0], "buf changed\n");
1325 size = saved;
1326 str2[0] = 0;
1327 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, NULL, 0);
1328 ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1329 ok(!str2[0], "buf changed\n");
1331 hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, test_str, sizeof(test_str), 0);
1332 ok(hres == S_OK, "UrlMkSetSessionOption failed: %08x\n", hres);
1334 size = 0;
1335 str2[0] = 0;
1336 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, &size, 0);
1337 ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres);
1338 ok(size == sizeof(test_str) && !memcmp(str2, test_str, sizeof(test_str)), "wrong user agent\n");
1340 hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, test2_str, sizeof(test2_str), 0);
1341 ok(hres == S_OK, "UrlMkSetSessionOption failed: %08x\n", hres);
1343 size = 0;
1344 str2[0] = 0;
1345 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, &size, 0);
1346 ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres);
1347 ok(size == sizeof(test_str) && !memcmp(str2, test_str, sizeof(test_str)), "wrong user agent\n");
1349 hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, test_str, 2, 0);
1350 ok(hres == S_OK, "UrlMkSetSessionOption failed: %08x\n", hres);
1352 size = 0;
1353 str2[0] = 0;
1354 hres = pUrlMkGetSessionOption(URLMON_OPTION_USERAGENT, str2, saved, &size, 0);
1355 ok(hres == E_OUTOFMEMORY, "UrlMkGetSessionOption failed: %08x\n", hres);
1356 ok(size == 3 && !strcmp(str2, "te"), "wrong user agent\n");
1358 hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, test_str, 0, 0);
1359 ok(hres == E_INVALIDARG, "UrlMkSetSessionOption failed: %08x\n", hres);
1361 hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, NULL, sizeof(test_str), 0);
1362 ok(hres == E_INVALIDARG, "UrlMkSetSessionOption failed: %08x\n", hres);
1364 hres = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, NULL, 0, 0);
1365 ok(hres == E_INVALIDARG, "UrlMkSetSessionOption failed: %08x\n", hres);
1367 HeapFree(GetProcessHeap(), 0, str2);
1370 static void test_MkParseDisplayNameEx(void)
1372 IMoniker *mon = NULL;
1373 LPWSTR name;
1374 DWORD issys;
1375 ULONG eaten = 0;
1376 IBindCtx *bctx;
1377 HRESULT hres;
1379 static const WCHAR clsid_nameW[] = {'c','l','s','i','d',':',
1380 '2','0','D','0','4','F','E','0','-','3','A','E','A','-','1','0','6','9','-','A','2','D','8',
1381 '-','0','8','0','0','2','B','3','0','3','0','9','D',':',0};
1383 const struct
1385 LPBC *ppbc;
1386 LPCWSTR szDisplayName;
1387 ULONG *pchEaten;
1388 LPMONIKER *ppmk;
1389 } invalid_parameters[] =
1391 {NULL, NULL, NULL, NULL},
1392 {NULL, NULL, NULL, &mon},
1393 {NULL, NULL, &eaten, NULL},
1394 {NULL, NULL, &eaten, &mon},
1395 {NULL, wszEmpty, NULL, NULL},
1396 {NULL, wszEmpty, NULL, &mon},
1397 {NULL, wszEmpty, &eaten, NULL},
1398 {NULL, wszEmpty, &eaten, &mon},
1399 {&bctx, NULL, NULL, NULL},
1400 {&bctx, NULL, NULL, &mon},
1401 {&bctx, NULL, &eaten, NULL},
1402 {&bctx, NULL, &eaten, &mon},
1403 {&bctx, wszEmpty, NULL, NULL},
1404 {&bctx, wszEmpty, NULL, &mon},
1405 {&bctx, wszEmpty, &eaten, NULL},
1406 {&bctx, wszEmpty, &eaten, &mon},
1409 int i;
1411 CreateBindCtx(0, &bctx);
1413 for (i = 0; i < sizeof(invalid_parameters)/sizeof(invalid_parameters[0]); i++)
1415 eaten = 0xdeadbeef;
1416 mon = (IMoniker *)0xdeadbeef;
1417 hres = MkParseDisplayNameEx(invalid_parameters[i].ppbc ? *invalid_parameters[i].ppbc : NULL,
1418 invalid_parameters[i].szDisplayName,
1419 invalid_parameters[i].pchEaten,
1420 invalid_parameters[i].ppmk);
1421 ok(hres == E_INVALIDARG,
1422 "[%d] Expected MkParseDisplayNameEx to return E_INVALIDARG, got %08x\n", i, hres);
1423 ok(eaten == 0xdeadbeef, "[%d] Expected eaten to be 0xdeadbeef, got %u\n", i, eaten);
1424 ok(mon == (IMoniker *)0xdeadbeef, "[%d] Expected mon to be 0xdeadbeef, got %p\n", i, mon);
1427 hres = MkParseDisplayNameEx(bctx, url9, &eaten, &mon);
1428 ok(hres == S_OK, "MkParseDisplayNameEx failed: %08x\n", hres);
1429 ok(eaten == sizeof(url9)/sizeof(WCHAR)-1, "eaten=%d\n", eaten);
1430 ok(mon != NULL, "mon == NULL\n");
1432 hres = IMoniker_GetDisplayName(mon, NULL, 0, &name);
1433 ok(hres == S_OK, "GetDiasplayName failed: %08x\n", hres);
1434 ok(!lstrcmpW(name, url9), "wrong display name %s\n", wine_dbgstr_w(name));
1435 CoTaskMemFree(name);
1437 hres = IMoniker_IsSystemMoniker(mon, &issys);
1438 ok(hres == S_OK, "IsSystemMoniker failed: %08x\n", hres);
1439 ok(issys == MKSYS_URLMONIKER, "issys=%x\n", issys);
1441 IMoniker_Release(mon);
1443 hres = MkParseDisplayNameEx(bctx, clsid_nameW, &eaten, &mon);
1444 ok(hres == S_OK, "MkParseDisplayNameEx failed: %08x\n", hres);
1445 ok(eaten == sizeof(clsid_nameW)/sizeof(WCHAR)-1, "eaten=%d\n", eaten);
1446 ok(mon != NULL, "mon == NULL\n");
1448 hres = IMoniker_IsSystemMoniker(mon, &issys);
1449 ok(hres == S_OK, "IsSystemMoniker failed: %08x\n", hres);
1450 ok(issys == MKSYS_CLASSMONIKER, "issys=%x\n", issys);
1452 IMoniker_Release(mon);
1454 hres = MkParseDisplayNameEx(bctx, url8, &eaten, &mon);
1455 ok(FAILED(hres), "MkParseDisplayNameEx succeeded: %08x\n", hres);
1457 IBindCtx_Release(bctx);
1460 static void test_IsValidURL(void)
1462 HRESULT hr;
1463 IBindCtx *bctx = NULL;
1465 hr = IsValidURL(NULL, 0, 0);
1466 ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
1468 hr = IsValidURL(NULL, wszHttpWineHQ, 0);
1469 ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
1471 CreateBindCtx(0, &bctx);
1473 hr = IsValidURL(bctx, wszHttpWineHQ, 0);
1474 ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
1476 IBindCtx_Release(bctx);
1479 static const struct {
1480 INTERNETFEATURELIST feature;
1481 DWORD get_flags;
1482 HRESULT expected;
1483 BOOL todo;
1484 } default_feature_tests[] = {
1485 {FEATURE_OBJECT_CACHING,GET_FEATURE_FROM_PROCESS,S_OK,TRUE},
1486 {FEATURE_ZONE_ELEVATION,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1487 {FEATURE_MIME_HANDLING,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1488 {FEATURE_MIME_SNIFFING,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1489 {FEATURE_WINDOW_RESTRICTIONS,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1490 {FEATURE_WEBOC_POPUPMANAGEMENT,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1491 {FEATURE_BEHAVIORS,GET_FEATURE_FROM_PROCESS,S_OK,TRUE},
1492 {FEATURE_DISABLE_MK_PROTOCOL,GET_FEATURE_FROM_PROCESS,S_OK,TRUE},
1493 {FEATURE_LOCALMACHINE_LOCKDOWN,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1494 {FEATURE_SECURITYBAND,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1495 {FEATURE_RESTRICT_ACTIVEXINSTALL,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1496 {FEATURE_VALIDATE_NAVIGATE_URL,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1497 {FEATURE_RESTRICT_FILEDOWNLOAD,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1498 {FEATURE_ADDON_MANAGEMENT,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1499 {FEATURE_PROTOCOL_LOCKDOWN,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1500 {FEATURE_HTTP_USERNAME_PASSWORD_DISABLE,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1501 {FEATURE_SAFE_BINDTOOBJECT,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1502 {FEATURE_UNC_SAVEDFILECHECK,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1503 {FEATURE_GET_URL_DOM_FILEPATH_UNENCODED,GET_FEATURE_FROM_PROCESS,S_OK,TRUE},
1504 {FEATURE_TABBED_BROWSING,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1505 {FEATURE_SSLUX,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1506 {FEATURE_DISABLE_NAVIGATION_SOUNDS,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1507 {FEATURE_DISABLE_LEGACY_COMPRESSION,GET_FEATURE_FROM_PROCESS,S_OK,TRUE},
1508 {FEATURE_FORCE_ADDR_AND_STATUS,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1509 {FEATURE_XMLHTTP,GET_FEATURE_FROM_PROCESS,S_OK,TRUE},
1510 {FEATURE_DISABLE_TELNET_PROTOCOL,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1511 {FEATURE_FEEDS,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1512 {FEATURE_BLOCK_INPUT_PROMPTS,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE}
1515 static void test_internet_feature_defaults(void) {
1516 HRESULT hres;
1517 DWORD i;
1519 for(i = 0; i < sizeof(default_feature_tests)/sizeof(default_feature_tests[0]); ++i) {
1520 hres = pCoInternetIsFeatureEnabled(default_feature_tests[i].feature, default_feature_tests[i].get_flags);
1521 if(default_feature_tests[i].todo) {
1522 todo_wine
1523 ok(hres == default_feature_tests[i].expected, "CoInternetIsFeatureEnabled returned %08x, expected %08x on test %d\n",
1524 hres, default_feature_tests[i].expected, i);
1525 } else {
1526 ok(hres == default_feature_tests[i].expected, "CoInternetIsFeatureEnabled returned %08x, expected %08x on test %d\n",
1527 hres, default_feature_tests[i].expected, i);
1532 static void test_CoInternetIsFeatureEnabled(void) {
1533 HRESULT hres;
1535 test_internet_feature_defaults();
1537 hres = pCoInternetIsFeatureEnabled(FEATURE_ENTRY_COUNT, GET_FEATURE_FROM_PROCESS);
1538 todo_wine
1539 ok(hres == E_FAIL, "CoInternetIsFeatureEnabled returned %08x, expected E_FAIL\n", hres);
1542 static const struct {
1543 INTERNETFEATURELIST feature;
1544 DWORD set_flags;
1545 BOOL enable;
1546 HRESULT set_expected;
1547 BOOL set_todo;
1548 DWORD get_flags;
1549 HRESULT get_expected;
1550 BOOL get_todo;
1551 } internet_feature_tests[] = {
1552 {FEATURE_OBJECT_CACHING,SET_FEATURE_ON_PROCESS,FALSE,S_OK,TRUE,GET_FEATURE_FROM_PROCESS,S_FALSE,TRUE},
1553 {FEATURE_WEBOC_POPUPMANAGEMENT,SET_FEATURE_ON_PROCESS,TRUE,S_OK,TRUE,GET_FEATURE_FROM_PROCESS,S_OK,TRUE},
1554 {FEATURE_LOCALMACHINE_LOCKDOWN,SET_FEATURE_ON_PROCESS,TRUE,S_OK,TRUE,GET_FEATURE_FROM_PROCESS,S_OK,TRUE}
1557 static void test_CoInternetSetFeatureEnabled(void) {
1558 HRESULT hres;
1559 DWORD i;
1561 hres = pCoInternetSetFeatureEnabled(FEATURE_ENTRY_COUNT,SET_FEATURE_ON_PROCESS,TRUE);
1562 todo_wine
1563 ok(hres == E_FAIL, "CoInternetSetFeatureEnabled returned %08x, expected E_FAIL\n", hres);
1565 for(i = 0; i < sizeof(internet_feature_tests)/sizeof(internet_feature_tests[0]); ++i) {
1566 hres = pCoInternetSetFeatureEnabled(internet_feature_tests[i].feature, internet_feature_tests[i].set_flags,
1567 internet_feature_tests[i].enable);
1568 if(internet_feature_tests[i].set_todo) {
1569 todo_wine
1570 ok(hres == internet_feature_tests[i].set_expected, "CoInternetSetFeatureEnabled returned %08x, expected %08x on test %d\n",
1571 hres, internet_feature_tests[i].set_expected, i);
1572 } else {
1573 ok(hres == internet_feature_tests[i].set_expected, "CoInternetSetFeatureEnabled returned %08x, expected %08x on test %d\n",
1574 hres, internet_feature_tests[i].set_expected, i);
1577 hres = pCoInternetIsFeatureEnabled(internet_feature_tests[i].feature, internet_feature_tests[i].set_flags);
1578 if(internet_feature_tests[i].get_todo) {
1579 todo_wine
1580 ok(hres == internet_feature_tests[i].get_expected, "CoInternetIsFeatureEnabled returned %08x, expected %08x on test %d\n",
1581 hres, internet_feature_tests[i].get_expected, i);
1582 } else {
1583 ok(hres == internet_feature_tests[i].get_expected, "CoInternetIsFeatureEnabled returned %08x, expected %08x on test %d\n",
1584 hres, internet_feature_tests[i].get_expected, i);
1589 static void test_internet_features(void) {
1590 if(!pCoInternetIsFeatureEnabled || !pCoInternetSetFeatureEnabled) {
1591 win_skip("Skipping internet feature tests, IE is too old\n");
1592 return;
1595 test_CoInternetIsFeatureEnabled();
1596 test_CoInternetSetFeatureEnabled();
1599 START_TEST(misc)
1601 HMODULE hurlmon;
1603 hurlmon = GetModuleHandle("urlmon.dll");
1604 pCoInternetCompareUrl = (void *) GetProcAddress(hurlmon, "CoInternetCompareUrl");
1605 pCoInternetGetSecurityUrl = (void*) GetProcAddress(hurlmon, "CoInternetGetSecurityUrl");
1606 pCoInternetGetSession = (void*) GetProcAddress(hurlmon, "CoInternetGetSession");
1607 pCoInternetParseUrl = (void*) GetProcAddress(hurlmon, "CoInternetParseUrl");
1608 pCoInternetQueryInfo = (void*) GetProcAddress(hurlmon, "CoInternetQueryInfo");
1609 pCopyStgMedium = (void*) GetProcAddress(hurlmon, "CopyStgMedium");
1610 pFindMimeFromData = (void*) GetProcAddress(hurlmon, "FindMimeFromData");
1611 pObtainUserAgentString = (void*) GetProcAddress(hurlmon, "ObtainUserAgentString");
1612 pReleaseBindInfo = (void*) GetProcAddress(hurlmon, "ReleaseBindInfo");
1613 pUrlMkGetSessionOption = (void*) GetProcAddress(hurlmon, "UrlMkGetSessionOption");
1614 pCompareSecurityIds = (void*) GetProcAddress(hurlmon, "CompareSecurityIds");
1615 pCoInternetIsFeatureEnabled = (void*) GetProcAddress(hurlmon, "CoInternetIsFeatureEnabled");
1616 pCoInternetSetFeatureEnabled = (void*) GetProcAddress(hurlmon, "CoInternetSetFeatureEnabled");
1618 if (!pCoInternetCompareUrl || !pCoInternetGetSecurityUrl ||
1619 !pCoInternetGetSession || !pCoInternetParseUrl || !pCompareSecurityIds) {
1620 win_skip("Various needed functions not present, too old IE\n");
1621 return;
1624 OleInitialize(NULL);
1626 register_protocols();
1628 test_CreateFormatEnum();
1629 test_RegisterFormatEnumerator();
1630 test_CoInternetParseUrl();
1631 test_CoInternetCompareUrl();
1632 test_CoInternetQueryInfo();
1633 test_FindMimeFromData();
1634 test_NameSpace();
1635 test_MimeFilter();
1636 test_ReleaseBindInfo();
1637 test_CopyStgMedium();
1638 test_UrlMkGetSessionOption();
1639 test_user_agent();
1640 test_MkParseDisplayNameEx();
1641 test_IsValidURL();
1642 test_internet_features();
1644 OleUninitialize();