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
22 #include <wine/test.h>
33 DEFINE_GUID(CLSID_AboutProtocol, 0x3050F406, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
35 #define DEFINE_EXPECT(func) \
36 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
38 #define SET_EXPECT(func) \
39 expect_ ## func = TRUE
41 #define CHECK_EXPECT(func) \
43 ok(expect_ ##func, "unexpected call " #func "\n"); \
44 expect_ ## func = FALSE; \
45 called_ ## func = TRUE; \
48 #define CHECK_EXPECT2(func) \
50 ok(expect_ ##func, "unexpected call " #func "\n"); \
51 called_ ## func = TRUE; \
54 #define CHECK_CALLED(func) \
56 ok(called_ ## func, "expected " #func "\n"); \
57 expect_ ## func = called_ ## func = FALSE; \
60 DEFINE_EXPECT(ParseUrl);
61 DEFINE_EXPECT(QI_IInternetProtocolInfo);
62 DEFINE_EXPECT(CreateInstance);
63 DEFINE_EXPECT(unk_Release);
65 static void test_CreateFormatEnum(void)
67 IEnumFORMATETC *fenum = NULL, *fenum2 = NULL;
72 static DVTARGETDEVICE dev = {sizeof(dev),0,0,0,0,{0}};
73 static FORMATETC formatetc[] = {
81 hres = CreateFormatEnumerator(0, formatetc, &fenum);
82 ok(hres == E_FAIL, "CreateFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
83 hres = CreateFormatEnumerator(0, formatetc, NULL);
84 ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
85 hres = CreateFormatEnumerator(5, formatetc, NULL);
86 ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
89 hres = CreateFormatEnumerator(5, formatetc, &fenum);
90 ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
94 hres = IEnumFORMATETC_Next(fenum, 2, NULL, &ul);
95 ok(hres == E_INVALIDARG, "Next failed: %08x, expected E_INVALIDARG\n", hres);
97 hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul);
98 ok(hres == S_OK, "Next failed: %08x\n", hres);
99 ok(ul == 0, "ul=%d, expected 0\n", ul);
101 hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul);
102 ok(hres == S_OK, "Next failed: %08x\n", hres);
103 ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex);
104 ok(fetc[1].lindex == 1, "fetc[1].lindex=%d, expected 1\n", fetc[1].lindex);
105 ok(fetc[0].ptd == &dev, "fetc[0].ptd=%p, expected %p\n", fetc[0].ptd, &dev);
106 ok(ul == 2, "ul=%d, expected 2\n", ul);
108 hres = IEnumFORMATETC_Skip(fenum, 1);
109 ok(hres == S_OK, "Skip failed: %08x\n", hres);
111 hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul);
112 ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres);
113 ok(fetc[0].lindex == 3, "fetc[0].lindex=%d, expected 3\n", fetc[0].lindex);
114 ok(fetc[1].lindex == 4, "fetc[1].lindex=%d, expected 4\n", fetc[1].lindex);
115 ok(fetc[0].ptd == NULL, "fetc[0].ptd=%p, expected NULL\n", fetc[0].ptd);
116 ok(ul == 2, "ul=%d, expected 2\n", ul);
118 hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul);
119 ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres);
120 ok(ul == 0, "ul=%d, expected 0\n", ul);
122 hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul);
123 ok(hres == S_OK, "Next failed: %08x\n", hres);
124 ok(ul == 0, "ul=%d, expected 0\n", ul);
126 hres = IEnumFORMATETC_Skip(fenum, 3);
127 ok(hres == S_FALSE, "Skip failed: %08x, expected S_FALSE\n", hres);
129 hres = IEnumFORMATETC_Reset(fenum);
130 ok(hres == S_OK, "Reset failed: %08x\n", hres);
132 hres = IEnumFORMATETC_Next(fenum, 5, fetc, NULL);
133 ok(hres == S_OK, "Next failed: %08x\n", hres);
134 ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex);
136 hres = IEnumFORMATETC_Reset(fenum);
137 ok(hres == S_OK, "Reset failed: %08x\n", hres);
139 hres = IEnumFORMATETC_Skip(fenum, 2);
140 ok(hres == S_OK, "Skip failed: %08x\n", hres);
142 hres = IEnumFORMATETC_Clone(fenum, NULL);
143 ok(hres == E_INVALIDARG, "Clone failed: %08x, expected E_INVALIDARG\n", hres);
145 hres = IEnumFORMATETC_Clone(fenum, &fenum2);
146 ok(hres == S_OK, "Clone failed: %08x\n", hres);
148 if(SUCCEEDED(hres)) {
149 ok(fenum != fenum2, "fenum == fenum2\n");
151 hres = IEnumFORMATETC_Next(fenum2, 2, fetc, &ul);
152 ok(hres == S_OK, "Next failed: %08x\n", hres);
153 ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex);
155 IEnumFORMATETC_Release(fenum2);
158 hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul);
159 ok(hres == S_OK, "Next failed: %08x\n", hres);
160 ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex);
162 hres = IEnumFORMATETC_Skip(fenum, 1);
163 ok(hres == S_OK, "Skip failed: %08x\n", hres);
165 IEnumFORMATETC_Release(fenum);
168 static void test_RegisterFormatEnumerator(void)
170 IBindCtx *bctx = NULL;
171 IEnumFORMATETC *format = NULL, *format2 = NULL;
172 IUnknown *unk = NULL;
175 static FORMATETC formatetc = {0,NULL,0,0,0};
176 static WCHAR wszEnumFORMATETC[] =
177 {'_','E','n','u','m','F','O','R','M','A','T','E','T','C','_',0};
179 CreateBindCtx(0, &bctx);
181 hres = CreateFormatEnumerator(1, &formatetc, &format);
182 ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
186 hres = RegisterFormatEnumerator(NULL, format, 0);
187 ok(hres == E_INVALIDARG,
188 "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
189 hres = RegisterFormatEnumerator(bctx, NULL, 0);
190 ok(hres == E_INVALIDARG,
191 "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
193 hres = RegisterFormatEnumerator(bctx, format, 0);
194 ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
196 hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
197 ok(hres == S_OK, "GetObjectParam failed: %08x\n", hres);
198 ok(unk == (IUnknown*)format, "unk != format\n");
200 hres = RevokeFormatEnumerator(NULL, format);
201 ok(hres == E_INVALIDARG,
202 "RevokeFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
204 hres = RevokeFormatEnumerator(bctx, format);
205 ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
207 hres = RevokeFormatEnumerator(bctx, format);
208 ok(hres == E_FAIL, "RevokeFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
210 hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
211 ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
213 hres = RegisterFormatEnumerator(bctx, format, 0);
214 ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
216 hres = CreateFormatEnumerator(1, &formatetc, &format2);
217 ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
219 if(SUCCEEDED(hres)) {
220 hres = RevokeFormatEnumerator(bctx, format);
221 ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
223 IEnumFORMATETC_Release(format2);
226 hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
227 ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
229 IEnumFORMATETC_Release(format);
231 hres = RegisterFormatEnumerator(bctx, format, 0);
232 ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
233 hres = RevokeFormatEnumerator(bctx, NULL);
234 ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
235 hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
236 ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
238 IBindCtx_Release(bctx);
241 static const WCHAR url1[] = {'r','e','s',':','/','/','m','s','h','t','m','l','.','d','l','l',
242 '/','b','l','a','n','k','.','h','t','m',0};
243 static const WCHAR url2[] = {'i','n','d','e','x','.','h','t','m',0};
244 static const WCHAR url3[] = {'f','i','l','e',':','c',':','\\','I','n','d','e','x','.','h','t','m',0};
245 static const WCHAR url4[] = {'f','i','l','e',':','s','o','m','e','%','2','0','f','i','l','e',
246 '%','2','e','j','p','g',0};
247 static const WCHAR url5[] = {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q',
249 static const WCHAR url6[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
250 static const WCHAR url7[] = {'f','t','p',':','/','/','w','i','n','e','h','q','.','o','r','g','/',
251 'f','i','l','e','.','t','e','s','t',0};
252 static const WCHAR url8[] = {'t','e','s','t',':','1','2','3','a','b','c',0};
255 static const WCHAR url4e[] = {'f','i','l','e',':','s','o','m','e',' ','f','i','l','e',
258 static const WCHAR path3[] = {'c',':','\\','I','n','d','e','x','.','h','t','m',0};
259 static const WCHAR path4[] = {'s','o','m','e',' ','f','i','l','e','.','j','p','g',0};
261 static const WCHAR wszRes[] = {'r','e','s',0};
262 static const WCHAR wszFile[] = {'f','i','l','e',0};
263 static const WCHAR wszHttp[] = {'h','t','t','p',0};
264 static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
265 static const WCHAR wszEmpty[] = {0};
276 static const struct parse_test parse_tests[] = {
277 {url1, S_OK, url1, E_INVALIDARG, NULL, wszRes},
278 {url2, E_FAIL, url2, E_INVALIDARG, NULL, wszEmpty},
279 {url3, E_FAIL, url3, S_OK, path3, wszFile},
280 {url4, E_FAIL, url4e, S_OK, path4, wszFile},
281 {url5, E_FAIL, url5, E_INVALIDARG, NULL, wszHttp},
282 {url6, S_OK, url6, E_INVALIDARG, NULL, wszAbout}
285 static void test_CoInternetParseUrl(void)
291 static WCHAR buf[4096];
293 memset(buf, 0xf0, sizeof(buf));
294 hres = CoInternetParseUrl(parse_tests[0].url, PARSE_SCHEMA, 0, buf,
296 ok(hres == E_POINTER, "schema failed: %08x, expected E_POINTER\n", hres);
298 for(i=0; i < sizeof(parse_tests)/sizeof(parse_tests[0]); i++) {
299 memset(buf, 0xf0, sizeof(buf));
300 hres = CoInternetParseUrl(parse_tests[i].url, PARSE_SECURITY_URL, 0, buf,
301 sizeof(buf)/sizeof(WCHAR), &size, 0);
302 ok(hres == parse_tests[i].secur_hres, "[%d] security url failed: %08x, expected %08x\n",
303 i, hres, parse_tests[i].secur_hres);
305 memset(buf, 0xf0, sizeof(buf));
306 hres = CoInternetParseUrl(parse_tests[i].url, PARSE_ENCODE, 0, buf,
307 sizeof(buf)/sizeof(WCHAR), &size, 0);
308 ok(hres == S_OK, "[%d] encoding failed: %08x\n", i, hres);
309 ok(size == lstrlenW(parse_tests[i].encoded_url), "[%d] wrong size\n", i);
310 ok(!lstrcmpW(parse_tests[i].encoded_url, buf), "[%d] wrong encoded url\n", i);
312 memset(buf, 0xf0, sizeof(buf));
313 hres = CoInternetParseUrl(parse_tests[i].url, PARSE_PATH_FROM_URL, 0, buf,
314 sizeof(buf)/sizeof(WCHAR), &size, 0);
315 ok(hres == parse_tests[i].path_hres, "[%d] path failed: %08x, expected %08x\n",
316 i, hres, parse_tests[i].path_hres);
317 if(parse_tests[i].path) {
318 ok(size == lstrlenW(parse_tests[i].path), "[%d] wrong size\n", i);
319 ok(!lstrcmpW(parse_tests[i].path, buf), "[%d] wrong path\n", i);
322 memset(buf, 0xf0, sizeof(buf));
323 hres = CoInternetParseUrl(parse_tests[i].url, PARSE_SCHEMA, 0, buf,
324 sizeof(buf)/sizeof(WCHAR), &size, 0);
325 ok(hres == S_OK, "[%d] schema failed: %08x\n", i, hres);
326 ok(size == lstrlenW(parse_tests[i].schema), "[%d] wrong size\n", i);
327 ok(!lstrcmpW(parse_tests[i].schema, buf), "[%d] wrong schema\n", i);
331 static void test_CoInternetCompareUrl(void)
335 hres = CoInternetCompareUrl(url1, url1, 0);
336 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
338 hres = CoInternetCompareUrl(url1, url3, 0);
339 ok(hres == S_FALSE, "CoInternetParseUrl failed: %08x\n", hres);
341 hres = CoInternetCompareUrl(url3, url1, 0);
342 ok(hres == S_FALSE, "CoInternetParseUrl failed: %08x\n", hres);
345 static const WCHAR mimeTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
346 static const WCHAR mimeTextPlain[] = {'t','e','x','t','/','p','l','a','i','n',0};
347 static const WCHAR mimeAppOctetStream[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
348 'o','c','t','e','t','-','s','t','r','e','a','m',0};
349 static const WCHAR mimeImagePjpeg[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
350 static const WCHAR mimeImageGif[] = {'i','m','a','g','e','/','g','i','f',0};
351 static const WCHAR mimeImageBmp[] = {'i','m','a','g','e','/','b','m','p',0};
352 static const WCHAR mimeImageXPng[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
353 static const WCHAR mimeImageTiff[] = {'i','m','a','g','e','/','t','i','f','f',0};
354 static const WCHAR mimeVideoAvi[] = {'v','i','d','e','o','/','a','v','i',0};
355 static const WCHAR mimeVideoMpeg[] = {'v','i','d','e','o','/','m','p','e','g',0};
356 static const WCHAR mimeAppXCompressed[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
357 'x','-','c','o','m','p','r','e','s','s','e','d',0};
358 static const WCHAR mimeAppXZip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
359 'x','-','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
360 static const WCHAR mimeAppXGzip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
361 'x','-','g','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
362 static const WCHAR mimeAppJava[] = {'a','p','p','l','i','c','a','t','i','o','n','/','j','a','v','a',0};
363 static const WCHAR mimeAppPdf[] = {'a','p','p','l','i','c','a','t','i','o','n','/','p','d','f',0};
364 static const WCHAR mimeAppXMSDownload[] =
365 {'a','p','p','l','i','c','a','t','i','o','n','/','x','-','m','s','d','o','w','n','l','o','a','d',0};
367 static const struct {
372 {url1, mimeTextHtml, S_OK},
373 {url2, mimeTextHtml, S_OK},
374 {url3, mimeTextHtml, S_OK},
375 {url4, NULL, E_FAIL},
376 {url5, NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)},
377 {url6, NULL, E_FAIL},
378 {url7, NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)}
381 static BYTE data1[] = "test data\n";
382 static BYTE data2[] = {31,'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0};
383 static BYTE data3[] = {0,0,0};
384 static BYTE data4[] = {'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0,0};
385 static BYTE data5[] = {0xa,0xa,0xa,'x',32,'x',0};
386 static BYTE data6[] = {0xfa,0xfa,0xfa,0xfa,'\n','\r','\t','x','x','x',1};
387 static BYTE data7[] = "<html>blahblah";
388 static BYTE data8[] = {'t','e','s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0};
389 static BYTE data9[] = {'t','e',0,'s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0};
390 static BYTE data10[] = "<HtmL>blahblah";
391 static BYTE data11[] = "blah<HTML>blahblah";
392 static BYTE data12[] = "blah<HTMLblahblah";
393 static BYTE data13[] = "blahHTML>blahblah";
394 static BYTE data14[] = "blah<HTMblahblah";
395 static BYTE data15[] = {0xff,0xd8};
396 static BYTE data16[] = {0xff,0xd8,'h'};
397 static BYTE data17[] = {0,0xff,0xd8};
398 static BYTE data18[] = {0xff,0xd8,'<','h','t','m','l','>'};
399 static BYTE data19[] = {'G','I','F','8','7','a'};
400 static BYTE data20[] = {'G','I','F','8','9','a'};
401 static BYTE data21[] = {'G','I','F','8','7'};
402 static BYTE data22[] = {'G','i','F','8','7','a'};
403 static BYTE data23[] = {'G','i','F','8','8','a'};
404 static BYTE data24[] = {'g','i','f','8','7','a'};
405 static BYTE data25[] = {'G','i','F','8','7','A'};
406 static BYTE data26[] = {'G','i','F','8','7','a','<','h','t','m','l','>'};
407 static BYTE data27[] = {0x30,'G','i','F','8','7','A'};
408 static BYTE data28[] = {0x42,0x4d,0x6e,0x42,0x1c,0x00,0x00,0x00,0x00,0x00,0x36,0x00,0x00,0x00};
409 static BYTE data29[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x','x'};
410 static BYTE data30[] = {0x42,0x4d,'x','x','x','x',0x00,0x01,0x00,0x00,'x','x','x','x'};
411 static BYTE data31[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'<','h','t','m','l','>'};
412 static BYTE data32[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'};
413 static BYTE data33[] = {0x00,0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'};
414 static BYTE data34[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'};
415 static BYTE data35[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x','x','x','x',0};
416 static BYTE data36[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,'x','x'};
417 static BYTE data37[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'<','h','t','m','l','>'};
418 static BYTE data38[] = {0x00,0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'};
419 static BYTE data39[] = {0x4d,0x4d,0x00,0x2a};
420 static BYTE data40[] = {0x4d,0x4d,0x00,0x2a,'<','h','t','m','l','>',0};
421 static BYTE data41[] = {0x4d,0x4d,0xff};
422 static BYTE data42[] = {0x4d,0x4d};
423 static BYTE data43[] = {0x00,0x4d,0x4d,0x00};
424 static BYTE data44[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff};
425 static BYTE data45[] = {'R','I','F','f',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff};
426 static BYTE data46[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20};
427 static BYTE data47[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x21,0xff};
428 static BYTE data48[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,'<','h','t','m','l','>'};
429 static BYTE data49[] = {'R','I','F','F',0x0f,0x0f,0xf0,0xf0,'A','V','I',0x20,0xf0,0x00};
430 static BYTE data50[] = {0x00,0x00,0x01,0xb3,0xff};
431 static BYTE data51[] = {0x00,0x00,0x01,0xba,0xff};
432 static BYTE data52[] = {0x00,0x00,0x01,0xb8,0xff};
433 static BYTE data53[] = {0x00,0x00,0x01,0xba};
434 static BYTE data54[] = {0x00,0x00,0x01,0xba,'<','h','t','m','l','>'};
435 static BYTE data55[] = {0x1f,0x8b,'x'};
436 static BYTE data56[] = {0x1f};
437 static BYTE data57[] = {0x1f,0x8b,'<','h','t','m','l','>','t','e','s','t',0};
438 static BYTE data58[] = {0x1f,0x8b};
439 static BYTE data59[] = {0x50,0x4b,'x'};
440 static BYTE data60[] = {0x50,0x4b};
441 static BYTE data61[] = {0x50,0x4b,'<','h','t','m','l','>',0};
442 static BYTE data62[] = {0xca,0xfe,0xba,0xbe,'x'};
443 static BYTE data63[] = {0xca,0xfe,0xba,0xbe};
444 static BYTE data64[] = {0xca,0xfe,0xba,0xbe,'<','h','t','m','l','>',0};
445 static BYTE data65[] = {0x25,0x50,0x44,0x46,'x'};
446 static BYTE data66[] = {0x25,0x50,0x44,0x46};
447 static BYTE data67[] = {0x25,0x50,0x44,0x46,'x','<','h','t','m','l','>'};
448 static BYTE data68[] = {'M','Z','x'};
449 static BYTE data69[] = {'M','Z'};
450 static BYTE data70[] = {'M','Z','<','h','t','m','l','>',0xff};
452 static const struct {
457 {data1, sizeof(data1), mimeTextPlain},
458 {data2, sizeof(data2), mimeAppOctetStream},
459 {data3, sizeof(data3), mimeAppOctetStream},
460 {data4, sizeof(data4), mimeAppOctetStream},
461 {data5, sizeof(data5), mimeTextPlain},
462 {data6, sizeof(data6), mimeTextPlain},
463 {data7, sizeof(data7), mimeTextHtml},
464 {data8, sizeof(data8), mimeTextHtml},
465 {data9, sizeof(data9), mimeTextHtml},
466 {data10, sizeof(data10), mimeTextHtml},
467 {data11, sizeof(data11), mimeTextHtml},
468 {data12, sizeof(data12), mimeTextHtml},
469 {data13, sizeof(data13), mimeTextPlain},
470 {data14, sizeof(data14), mimeTextPlain},
471 {data15, sizeof(data15), mimeTextPlain},
472 {data16, sizeof(data16), mimeImagePjpeg},
473 {data17, sizeof(data17), mimeAppOctetStream},
474 {data18, sizeof(data18), mimeTextHtml},
475 {data19, sizeof(data19), mimeImageGif},
476 {data20, sizeof(data20), mimeImageGif},
477 {data21, sizeof(data21), mimeTextPlain},
478 {data22, sizeof(data22), mimeImageGif},
479 {data23, sizeof(data23), mimeTextPlain},
480 {data24, sizeof(data24), mimeImageGif},
481 {data25, sizeof(data25), mimeImageGif},
482 {data26, sizeof(data26), mimeTextHtml},
483 {data27, sizeof(data27), mimeTextPlain},
484 {data28, sizeof(data28), mimeImageBmp},
485 {data29, sizeof(data29), mimeImageBmp},
486 {data30, sizeof(data30), mimeAppOctetStream},
487 {data31, sizeof(data31), mimeTextHtml},
488 {data32, sizeof(data32), mimeAppOctetStream},
489 {data33, sizeof(data33), mimeAppOctetStream},
490 {data34, sizeof(data34), mimeImageXPng},
491 {data35, sizeof(data35), mimeImageXPng},
492 {data36, sizeof(data36), mimeAppOctetStream},
493 {data37, sizeof(data37), mimeTextHtml},
494 {data38, sizeof(data38), mimeAppOctetStream},
495 {data39, sizeof(data39), mimeImageTiff},
496 {data40, sizeof(data40), mimeTextHtml},
497 {data41, sizeof(data41), mimeImageTiff},
498 {data42, sizeof(data42), mimeTextPlain},
499 {data43, sizeof(data43), mimeAppOctetStream},
500 {data44, sizeof(data44), mimeVideoAvi},
501 {data45, sizeof(data45), mimeTextPlain},
502 {data46, sizeof(data46), mimeTextPlain},
503 {data47, sizeof(data47), mimeTextPlain},
504 {data48, sizeof(data48), mimeTextHtml},
505 {data49, sizeof(data49), mimeVideoAvi},
506 {data50, sizeof(data50), mimeVideoMpeg},
507 {data51, sizeof(data51), mimeVideoMpeg},
508 {data52, sizeof(data52), mimeAppOctetStream},
509 {data53, sizeof(data53), mimeAppOctetStream},
510 {data54, sizeof(data54), mimeTextHtml},
511 {data55, sizeof(data55), mimeAppXGzip},
512 {data56, sizeof(data56), mimeTextPlain},
513 {data57, sizeof(data57), mimeTextHtml},
514 {data58, sizeof(data58), mimeAppOctetStream},
515 {data59, sizeof(data59), mimeAppXZip},
516 {data60, sizeof(data60), mimeTextPlain},
517 {data61, sizeof(data61), mimeTextHtml},
518 {data62, sizeof(data62), mimeAppJava},
519 {data63, sizeof(data63), mimeTextPlain},
520 {data64, sizeof(data64), mimeTextHtml},
521 {data65, sizeof(data65), mimeAppPdf},
522 {data66, sizeof(data66), mimeTextPlain},
523 {data67, sizeof(data67), mimeTextHtml},
524 {data68, sizeof(data68), mimeAppXMSDownload},
525 {data69, sizeof(data69), mimeTextPlain},
526 {data70, sizeof(data70), mimeTextHtml}
529 static void test_FindMimeFromData(void)
535 for(i=0; i<sizeof(mime_tests)/sizeof(mime_tests[0]); i++) {
536 mime = (LPWSTR)0xf0f0f0f0;
537 hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, NULL, 0, &mime, 0);
538 if(mime_tests[i].mime) {
539 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
540 ok(!lstrcmpW(mime, mime_tests[i].mime), "[%d] wrong mime\n", i);
543 ok(hres == E_FAIL || hres == mime_tests[i].hres,
544 "[%d] FindMimeFromData failed: %08x, expected %08x\n",
545 i, hres, mime_tests[i].hres);
546 ok(mime == (LPWSTR)0xf0f0f0f0, "[%d] mime != 0xf0f0f0f0\n", i);
549 mime = (LPWSTR)0xf0f0f0f0;
550 hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, mimeTextPlain, 0, &mime, 0);
551 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
552 ok(!lstrcmpW(mime, mimeTextPlain), "[%d] wrong mime\n", i);
555 mime = (LPWSTR)0xf0f0f0f0;
556 hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, mimeAppOctetStream, 0, &mime, 0);
557 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
558 ok(!lstrcmpW(mime, mimeAppOctetStream), "[%d] wrong mime\n", i);
562 for(i=0; i < sizeof(mime_tests2)/sizeof(mime_tests2[0]); i++) {
563 hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
565 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
566 ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
569 hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
570 mimeTextHtml, 0, &mime, 0);
571 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
572 if(!lstrcmpW(mimeAppOctetStream, mime_tests2[i].mime)
573 || !lstrcmpW(mimeTextPlain, mime_tests2[i].mime))
574 ok(!lstrcmpW(mime, mimeTextHtml), "[%d] wrong mime\n", i);
576 ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
578 hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
579 mimeImagePjpeg, 0, &mime, 0);
580 ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
581 if(!lstrcmpW(mimeAppOctetStream, mime_tests2[i].mime) || i == 17)
582 ok(!lstrcmpW(mime, mimeImagePjpeg), "[%d] wrong mime\n", i);
584 ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
589 hres = FindMimeFromData(NULL, url1, data1, sizeof(data1), NULL, 0, &mime, 0);
590 ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
591 ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
594 hres = FindMimeFromData(NULL, url1, data1, sizeof(data1), mimeAppOctetStream, 0, &mime, 0);
595 ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
596 ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
599 hres = FindMimeFromData(NULL, url4, data1, sizeof(data1), mimeAppOctetStream, 0, &mime, 0);
600 ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
601 ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
604 hres = FindMimeFromData(NULL, NULL, NULL, 0, NULL, 0, &mime, 0);
605 ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, excepted E_INVALIDARG\n", hres);
607 hres = FindMimeFromData(NULL, NULL, NULL, 0, mimeTextPlain, 0, &mime, 0);
608 ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
610 hres = FindMimeFromData(NULL, NULL, data1, 0, NULL, 0, &mime, 0);
611 ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
613 hres = FindMimeFromData(NULL, url1, data1, 0, NULL, 0, &mime, 0);
614 ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
616 hres = FindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, &mime, 0);
617 ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
618 ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
621 hres = FindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, NULL, 0);
622 ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
625 static const BYTE secid1[] = {'f','i','l','e',':',0,0,0,0};
626 static const BYTE secid4[] ={'f','i','l','e',':',3,0,0,0};
627 static const BYTE secid5[] = {'h','t','t','p',':','w','w','w','.','w','i','n','e','h','q',
628 '.','o','r','g',3,0,0,0};
629 static const BYTE secid6[] = {'a','b','o','u','t',':','b','l','a','n','k',3,0,0,0};
630 static const BYTE secid7[] = {'f','t','p',':','w','i','n','e','h','q','.','o','r','g',
633 static struct secmgr_test {
641 {url1, 0, S_OK, sizeof(secid1), secid1, S_OK},
642 {url2, 100, 0x80041001, 0, NULL, E_INVALIDARG},
643 {url3, 0, S_OK, sizeof(secid1), secid1, S_OK},
644 {url4, 3, S_OK, sizeof(secid4), secid4, S_OK},
645 {url5, 3, S_OK, sizeof(secid5), secid5, S_OK},
646 {url6, 3, S_OK, sizeof(secid6), secid6, S_OK},
647 {url7, 3, S_OK, sizeof(secid7), secid7, S_OK}
650 static void test_SecurityManager(void)
653 IInternetSecurityManager *secmgr = NULL;
658 hres = CoInternetCreateSecurityManager(NULL, &secmgr, 0);
659 ok(hres == S_OK, "CoInternetCreateSecurityManager failed: %08x\n", hres);
663 for(i=0; i < sizeof(secmgr_tests)/sizeof(secmgr_tests[0]); i++) {
665 hres = IInternetSecurityManager_MapUrlToZone(secmgr, secmgr_tests[i].url,
667 ok(hres == secmgr_tests[i].zone_hres,
668 "[%d] MapUrlToZone failed: %08x, expected %08x\n",
669 i, hres, secmgr_tests[i].zone_hres);
671 ok(zone == secmgr_tests[i].zone, "[%d] zone=%d, expected %d\n", i, zone,
672 secmgr_tests[i].zone);
674 ok(zone == secmgr_tests[i].zone || zone == -1, "[%d] zone=%d\n", i, zone);
677 memset(buf, 0xf0, sizeof(buf));
678 hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[i].url,
680 ok(hres == secmgr_tests[i].secid_hres,
681 "[%d] GetSecurityId failed: %08x, expected %08x\n",
682 i, hres, secmgr_tests[i].secid_hres);
683 if(secmgr_tests[i].secid) {
684 ok(size == secmgr_tests[i].secid_size, "[%d] size=%d, expected %d\n",
685 i, size, secmgr_tests[i].secid_size);
686 ok(!memcmp(buf, secmgr_tests[i].secid, size), "[%d] wrong secid\n", i);
691 hres = IInternetSecurityManager_MapUrlToZone(secmgr, NULL, &zone, 0);
692 ok(hres == E_INVALIDARG, "MapUrlToZone failed: %08x, expected E_INVALIDARG\n", hres);
693 ok(zone == 100, "zone=%d\n", zone);
696 hres = IInternetSecurityManager_GetSecurityId(secmgr, NULL, buf, &size, 0);
697 ok(hres == E_INVALIDARG,
698 "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
699 hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[1].url,
701 ok(hres == E_INVALIDARG,
702 "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
703 hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[1].url,
705 ok(hres == E_INVALIDARG,
706 "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
708 IInternetSecurityManager_Release(secmgr);
711 static void test_ZoneManager(void)
713 IInternetZoneManager *zonemgr = NULL;
717 hres = CoInternetCreateZoneManager(NULL, &zonemgr, 0);
718 ok(hres == S_OK, "CoInternetCreateZoneManager failed: %08x\n", hres);
722 hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, buf,
723 sizeof(DWORD), URLZONEREG_DEFAULT);
724 ok(hres == S_OK, "GetZoneActionPolicy failed: %08x\n", hres);
725 ok(*(DWORD*)buf == 1, "policy=%d, expected 1\n", *(DWORD*)buf);
727 hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, NULL,
728 sizeof(DWORD), URLZONEREG_DEFAULT);
729 ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
731 hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, buf,
732 2, URLZONEREG_DEFAULT);
733 ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
735 hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1fff, buf,
736 sizeof(DWORD), URLZONEREG_DEFAULT);
737 ok(hres == E_FAIL, "GetZoneActionPolicy failed: %08x, expected E_FAIL\n", hres);
739 hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 13, 0x1a10, buf,
740 sizeof(DWORD), URLZONEREG_DEFAULT);
741 ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
743 IInternetZoneManager_Release(zonemgr);
746 static void register_protocols(void)
748 IInternetSession *session;
749 IClassFactory *factory;
752 static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
754 hres = CoInternetGetSession(0, &session, 0);
755 ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
759 hres = CoGetClassObject(&CLSID_AboutProtocol, CLSCTX_INPROC_SERVER, NULL,
760 &IID_IClassFactory, (void**)&factory);
761 ok(hres == S_OK, "Coud not get AboutProtocol factory: %08x\n", hres);
765 IInternetSession_RegisterNameSpace(session, factory, &CLSID_AboutProtocol,
766 wszAbout, 0, NULL, 0);
767 IClassFactory_Release(factory);
771 static HRESULT WINAPI InternetProtocolInfo_QueryInterface(IInternetProtocolInfo *iface,
772 REFIID riid, void **ppv)
774 ok(0, "unexpected call\n");
775 return E_NOINTERFACE;
778 static ULONG WINAPI InternetProtocolInfo_AddRef(IInternetProtocolInfo *iface)
783 static ULONG WINAPI InternetProtocolInfo_Release(IInternetProtocolInfo *iface)
788 static HRESULT WINAPI InternetProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
789 PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult,
790 DWORD *pcchResult, DWORD dwReserved)
792 CHECK_EXPECT(ParseUrl);
796 static HRESULT WINAPI InternetProtocolInfo_CombineUrl(IInternetProtocolInfo *iface,
797 LPCWSTR pwzBaseUrl, LPCWSTR pwzRelativeUrl, DWORD dwCombineFlags,
798 LPWSTR pwzResult, DWORD cchResult, DWORD *pcchResult, DWORD dwReserved)
800 ok(0, "unexpected call\n");
804 static HRESULT WINAPI InternetProtocolInfo_CompareUrl(IInternetProtocolInfo *iface,
805 LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags)
807 ok(0, "unexpected call\n");
811 static HRESULT WINAPI InternetProtocolInfo_QueryInfo(IInternetProtocolInfo *iface,
812 LPCWSTR pwzUrl, QUERYOPTION OueryOption, DWORD dwQueryFlags, LPVOID pBuffer,
813 DWORD cbBuffer, DWORD *pcbBuf, DWORD dwReserved)
815 ok(0, "unexpected call\n");
819 static const IInternetProtocolInfoVtbl InternetProtocolInfoVtbl = {
820 InternetProtocolInfo_QueryInterface,
821 InternetProtocolInfo_AddRef,
822 InternetProtocolInfo_Release,
823 InternetProtocolInfo_ParseUrl,
824 InternetProtocolInfo_CombineUrl,
825 InternetProtocolInfo_CompareUrl,
826 InternetProtocolInfo_QueryInfo
829 static IInternetProtocolInfo protocol_info = { &InternetProtocolInfoVtbl };
831 static HRESULT qiret;
832 static IClassFactory *expect_cf;
834 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
836 if(IsEqualGUID(&IID_IInternetProtocolInfo, riid)) {
837 CHECK_EXPECT(QI_IInternetProtocolInfo);
838 ok(iface == expect_cf, "unexpected iface\n");
839 *ppv = &protocol_info;
843 ok(0, "unexpected call\n");
844 return E_NOINTERFACE;
847 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
852 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
857 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
858 REFIID riid, void **ppv)
860 CHECK_EXPECT(CreateInstance);
862 ok(iface == expect_cf, "unexpected iface\n");
863 ok(pOuter == NULL, "pOuter = %p\n", pOuter);
864 ok(IsEqualGUID(&IID_IInternetProtocolInfo, riid), "unexpected riid\n");
865 ok(ppv != NULL, "ppv == NULL\n");
867 *ppv = &protocol_info;
871 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
873 ok(0, "unexpected call\n");
877 static const IClassFactoryVtbl ClassFactoryVtbl = {
878 ClassFactory_QueryInterface,
880 ClassFactory_Release,
881 ClassFactory_CreateInstance,
882 ClassFactory_LockServer
885 static IClassFactory test_protocol_cf = { &ClassFactoryVtbl };
886 static IClassFactory test_protocol_cf2 = { &ClassFactoryVtbl };
888 static void test_NameSpace(void)
890 IInternetSession *session;
895 static const WCHAR wszTest[] = {'t','e','s','t',0};
897 hres = CoInternetGetSession(0, &session, 0);
898 ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
902 hres = IInternetSession_RegisterNameSpace(session, NULL, &IID_NULL,
903 wszTest, 0, NULL, 0);
904 ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
906 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
908 ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
910 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
911 wszTest, 0, NULL, 0);
912 ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
914 qiret = E_NOINTERFACE;
915 expect_cf = &test_protocol_cf;
916 SET_EXPECT(QI_IInternetProtocolInfo);
917 SET_EXPECT(CreateInstance);
918 SET_EXPECT(ParseUrl);
920 hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
922 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
924 CHECK_CALLED(QI_IInternetProtocolInfo);
925 CHECK_CALLED(CreateInstance);
926 CHECK_CALLED(ParseUrl);
929 SET_EXPECT(QI_IInternetProtocolInfo);
930 SET_EXPECT(ParseUrl);
932 hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
934 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
936 CHECK_CALLED(QI_IInternetProtocolInfo);
937 CHECK_CALLED(ParseUrl);
939 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
940 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
942 hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
944 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
946 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf2, &IID_NULL,
947 wszTest, 0, NULL, 0);
948 ok(hres == S_OK, "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 hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
955 wszTest, 0, NULL, 0);
956 ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
958 SET_EXPECT(QI_IInternetProtocolInfo);
959 SET_EXPECT(ParseUrl);
961 hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
963 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
965 CHECK_CALLED(QI_IInternetProtocolInfo);
966 CHECK_CALLED(ParseUrl);
968 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
969 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
971 SET_EXPECT(QI_IInternetProtocolInfo);
972 SET_EXPECT(ParseUrl);
974 hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
976 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
978 CHECK_CALLED(QI_IInternetProtocolInfo);
979 CHECK_CALLED(ParseUrl);
981 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
982 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
984 expect_cf = &test_protocol_cf2;
985 SET_EXPECT(QI_IInternetProtocolInfo);
986 SET_EXPECT(ParseUrl);
988 hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
990 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
992 CHECK_CALLED(QI_IInternetProtocolInfo);
993 CHECK_CALLED(ParseUrl);
995 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
996 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
997 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
998 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
999 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, NULL);
1000 ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres);
1001 hres = IInternetSession_UnregisterNameSpace(session, NULL, wszTest);
1002 ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres);
1004 hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf2, wszTest);
1005 ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1007 hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1009 ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1011 IInternetSession_Release(session);
1014 static ULONG WINAPI unk_Release(IUnknown *iface)
1016 CHECK_EXPECT(unk_Release);
1020 static const IUnknownVtbl unk_vtbl = {
1026 static void test_ReleaseBindInfo(void)
1029 IUnknown unk = { &unk_vtbl };
1031 ReleaseBindInfo(NULL); /* shouldn't crash */
1033 memset(&bi, 0, sizeof(bi));
1034 bi.cbSize = sizeof(BINDINFO);
1036 SET_EXPECT(unk_Release);
1037 ReleaseBindInfo(&bi);
1038 ok(bi.cbSize == sizeof(BINDINFO), "bi.cbSize=%d\n", bi.cbSize);
1039 ok(bi.pUnk == NULL, "bi.pUnk=%p, expected NULL\n", bi.pUnk);
1040 CHECK_CALLED(unk_Release);
1042 memset(&bi, 0, sizeof(bi));
1043 bi.cbSize = offsetof(BINDINFO, pUnk);
1045 ReleaseBindInfo(&bi);
1046 ok(bi.cbSize == offsetof(BINDINFO, pUnk), "bi.cbSize=%d\n", bi.cbSize);
1047 ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk);
1049 memset(&bi, 0, sizeof(bi));
1051 ReleaseBindInfo(&bi);
1052 ok(!bi.cbSize, "bi.cbSize=%d, expected 0\n", bi.cbSize);
1053 ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk);
1056 static void test_UrlMkGetSessionOption(void)
1058 DWORD encoding, size;
1061 size = encoding = 0xdeadbeef;
1062 hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1063 sizeof(encoding), &size, 0);
1064 ok(hres == S_OK, "UrlMkGetSessionOption failed: %08x\n", hres);
1065 ok(encoding != 0xdeadbeef, "encoding not changed\n");
1066 ok(size == sizeof(encoding), "size=%d\n", size);
1068 size = encoding = 0xdeadbeef;
1069 hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1070 sizeof(encoding)+1, &size, 0);
1071 ok(hres == S_OK, "UrlMkGetSessionOption failed: %08x\n", hres);
1072 ok(encoding != 0xdeadbeef, "encoding not changed\n");
1073 ok(size == sizeof(encoding), "size=%d\n", size);
1075 size = encoding = 0xdeadbeef;
1076 hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1077 sizeof(encoding)-1, &size, 0);
1078 ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1079 ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1080 ok(size == 0xdeadbeef, "size=%d\n", size);
1082 size = encoding = 0xdeadbeef;
1083 hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, NULL,
1084 sizeof(encoding)-1, &size, 0);
1085 ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1086 ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1087 ok(size == 0xdeadbeef, "size=%d\n", size);
1089 encoding = 0xdeadbeef;
1090 hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1091 sizeof(encoding)-1, NULL, 0);
1092 ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1093 ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1096 static void test_ObtainUserAgentString(void)
1098 static const CHAR expected[] = "Mozilla/4.0 (compatible; MSIE ";
1104 hres = ObtainUserAgentString(0, NULL, NULL);
1105 ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1108 hres = ObtainUserAgentString(0, NULL, &size);
1109 ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1110 ok(size == 100, "size=%d, expected %d\n", size, 100);
1113 hres = ObtainUserAgentString(0, str, &size);
1114 ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1115 ok(size > 0, "size=%d, expected non-zero\n", size);
1119 hres = ObtainUserAgentString(0, str, &size);
1120 ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1121 ok(size > 0, "size=%d, expected non-zero\n", size);
1122 ok(str[0] == 'a', "str[0]=%c, expected 'a'\n", str[0]);
1125 hres = ObtainUserAgentString(1, str, &size);
1126 ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1127 ok(size > 0, "size=%d, expected non-zero\n", size);
1129 str2 = HeapAlloc(GetProcessHeap(), 0, (size+20)*sizeof(CHAR));
1132 skip("skipping rest of ObtainUserAgent tests, out of memory\n");
1137 hres = ObtainUserAgentString(0, str2, &size);
1138 ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres);
1139 ok(size == saved, "size=%d, expected %d\n", size, saved);
1140 ok(strlen(expected) <= strlen(str2) &&
1141 !memcmp(expected, str2, strlen(expected)*sizeof(CHAR)),
1142 "user agent was \"%s\", expected to start with \"%s\"\n",
1146 hres = ObtainUserAgentString(0, str2, &size);
1147 ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres);
1148 ok(size == saved, "size=%d, expected %d\n", size, saved);
1150 HeapFree(GetProcessHeap(), 0, str2);
1155 OleInitialize(NULL);
1157 register_protocols();
1159 test_CreateFormatEnum();
1160 test_RegisterFormatEnumerator();
1161 test_CoInternetParseUrl();
1162 test_CoInternetCompareUrl();
1163 test_FindMimeFromData();
1164 test_SecurityManager();
1167 test_ReleaseBindInfo();
1168 test_UrlMkGetSessionOption();
1169 test_ObtainUserAgentString();