urlmon/tests: Add tests for ObtainUserAgentString.
[wine] / dlls / urlmon / tests / misc.c
1 /*
2  * Copyright 2005-2006 Jacek Caban for CodeWeavers
3  *
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.
8  *
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.
13  *
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
17  */
18
19 #define COBJMACROS
20 #define CONST_VTABLE
21
22 #include <wine/test.h>
23 #include <stdarg.h>
24 #include <stddef.h>
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "ole2.h"
29 #include "urlmon.h"
30
31 #include "initguid.h"
32
33 DEFINE_GUID(CLSID_AboutProtocol, 0x3050F406, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
34
35 #define DEFINE_EXPECT(func) \
36     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
37
38 #define SET_EXPECT(func) \
39     expect_ ## func = TRUE
40
41 #define CHECK_EXPECT(func) \
42     do { \
43         ok(expect_ ##func, "unexpected call " #func "\n"); \
44         expect_ ## func = FALSE; \
45         called_ ## func = TRUE; \
46     }while(0)
47
48 #define CHECK_EXPECT2(func) \
49     do { \
50         ok(expect_ ##func, "unexpected call " #func "\n"); \
51         called_ ## func = TRUE; \
52     }while(0)
53
54 #define CHECK_CALLED(func) \
55     do { \
56         ok(called_ ## func, "expected " #func "\n"); \
57         expect_ ## func = called_ ## func = FALSE; \
58     }while(0)
59
60 DEFINE_EXPECT(ParseUrl);
61 DEFINE_EXPECT(QI_IInternetProtocolInfo);
62 DEFINE_EXPECT(CreateInstance);
63 DEFINE_EXPECT(unk_Release);
64
65 static void test_CreateFormatEnum(void)
66 {
67     IEnumFORMATETC *fenum = NULL, *fenum2 = NULL;
68     FORMATETC fetc[5];
69     ULONG ul;
70     HRESULT hres;
71
72     static DVTARGETDEVICE dev = {sizeof(dev),0,0,0,0,{0}};
73     static FORMATETC formatetc[] = {
74         {0,&dev,0,0,0},
75         {0,&dev,0,1,0},
76         {0,NULL,0,2,0},
77         {0,NULL,0,3,0},
78         {0,NULL,0,4,0}
79     };
80
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);
87
88
89     hres = CreateFormatEnumerator(5, formatetc, &fenum);
90     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
91     if(FAILED(hres))
92         return;
93
94     hres = IEnumFORMATETC_Next(fenum, 2, NULL, &ul);
95     ok(hres == E_INVALIDARG, "Next failed: %08x, expected E_INVALIDARG\n", hres);
96     ul = 100;
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);
100
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);
107
108     hres = IEnumFORMATETC_Skip(fenum, 1);
109     ok(hres == S_OK, "Skip failed: %08x\n", hres);
110
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);
117
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);
121     ul = 100;
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);
125
126     hres = IEnumFORMATETC_Skip(fenum, 3);
127     ok(hres == S_FALSE, "Skip failed: %08x, expected S_FALSE\n", hres);
128
129     hres = IEnumFORMATETC_Reset(fenum);
130     ok(hres == S_OK, "Reset failed: %08x\n", hres);
131
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);
135
136     hres = IEnumFORMATETC_Reset(fenum);
137     ok(hres == S_OK, "Reset failed: %08x\n", hres);
138
139     hres = IEnumFORMATETC_Skip(fenum, 2);
140     ok(hres == S_OK, "Skip failed: %08x\n", hres);
141
142     hres = IEnumFORMATETC_Clone(fenum, NULL);
143     ok(hres == E_INVALIDARG, "Clone failed: %08x, expected E_INVALIDARG\n", hres);
144
145     hres = IEnumFORMATETC_Clone(fenum, &fenum2);
146     ok(hres == S_OK, "Clone failed: %08x\n", hres);
147
148     if(SUCCEEDED(hres)) {
149         ok(fenum != fenum2, "fenum == fenum2\n");
150
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);
154
155         IEnumFORMATETC_Release(fenum2);
156     }
157
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);
161
162     hres = IEnumFORMATETC_Skip(fenum, 1);
163     ok(hres == S_OK, "Skip failed: %08x\n", hres);
164     
165     IEnumFORMATETC_Release(fenum);
166 }
167
168 static void test_RegisterFormatEnumerator(void)
169 {
170     IBindCtx *bctx = NULL;
171     IEnumFORMATETC *format = NULL, *format2 = NULL;
172     IUnknown *unk = NULL;
173     HRESULT hres;
174
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};
178
179     CreateBindCtx(0, &bctx);
180
181     hres = CreateFormatEnumerator(1, &formatetc, &format);
182     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
183     if(FAILED(hres))
184         return;
185
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);
192
193     hres = RegisterFormatEnumerator(bctx, format, 0);
194     ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
195
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");
199
200     hres = RevokeFormatEnumerator(NULL, format);
201     ok(hres == E_INVALIDARG,
202             "RevokeFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
203
204     hres = RevokeFormatEnumerator(bctx, format);
205     ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
206
207     hres = RevokeFormatEnumerator(bctx, format);
208     ok(hres == E_FAIL, "RevokeFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
209
210     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
211     ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
212
213     hres = RegisterFormatEnumerator(bctx, format, 0);
214     ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
215
216     hres = CreateFormatEnumerator(1, &formatetc, &format2);
217     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
218
219     if(SUCCEEDED(hres)) {
220         hres = RevokeFormatEnumerator(bctx, format);
221         ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
222
223         IEnumFORMATETC_Release(format2);
224     }
225
226     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
227     ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
228
229     IEnumFORMATETC_Release(format);
230
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);
237
238     IBindCtx_Release(bctx);
239 }
240
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',
248         '.','o','r','g',0};
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};
253
254
255 static const WCHAR url4e[] = {'f','i','l','e',':','s','o','m','e',' ','f','i','l','e',
256         '.','j','p','g',0};
257
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};
260
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};
266
267 struct parse_test {
268     LPCWSTR url;
269     HRESULT secur_hres;
270     LPCWSTR encoded_url;
271     HRESULT path_hres;
272     LPCWSTR path;
273     LPCWSTR schema;
274 };
275
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}
283 };
284
285 static void test_CoInternetParseUrl(void)
286 {
287     HRESULT hres;
288     DWORD size;
289     int i;
290
291     static WCHAR buf[4096];
292
293     memset(buf, 0xf0, sizeof(buf));
294     hres = CoInternetParseUrl(parse_tests[0].url, PARSE_SCHEMA, 0, buf,
295             3, &size, 0);
296     ok(hres == E_POINTER, "schema failed: %08x, expected E_POINTER\n", hres);
297
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);
304
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);
311
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);
320         }
321
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);
328     }
329 }
330
331 static void test_CoInternetCompareUrl(void)
332 {
333     HRESULT hres;
334
335     hres = CoInternetCompareUrl(url1, url1, 0);
336     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
337
338     hres = CoInternetCompareUrl(url1, url3, 0);
339     ok(hres == S_FALSE, "CoInternetParseUrl failed: %08x\n", hres);
340
341     hres = CoInternetCompareUrl(url3, url1, 0);
342     ok(hres == S_FALSE, "CoInternetParseUrl failed: %08x\n", hres);
343 }
344
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};
366
367 static const struct {
368     LPCWSTR url;
369     LPCWSTR mime;
370     HRESULT hres;
371 } mime_tests[] = {
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)}
379 };
380
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};
451
452 static const struct {
453     BYTE *data;
454     DWORD size;
455     LPCWSTR mime;
456 } mime_tests2[] = {
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}
527 };
528
529 static void test_FindMimeFromData(void)
530 {
531     HRESULT hres;
532     LPWSTR mime;
533     int i;
534
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);
541             CoTaskMemFree(mime);
542         }else {
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);
547         }
548
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);
553         CoTaskMemFree(mime);
554
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);
559         CoTaskMemFree(mime);
560     }
561
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,
564                 NULL, 0, &mime, 0);
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);
567         CoTaskMemFree(mime);
568
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);
575         else
576             ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
577
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);
583         else
584             ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
585
586         CoTaskMemFree(mime);
587     }
588
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");
592     CoTaskMemFree(mime);
593
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");
597     CoTaskMemFree(mime);
598
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");
602     CoTaskMemFree(mime);
603
604     hres = FindMimeFromData(NULL, NULL, NULL, 0, NULL, 0, &mime, 0);
605     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, excepted E_INVALIDARG\n", hres);
606
607     hres = FindMimeFromData(NULL, NULL, NULL, 0, mimeTextPlain, 0, &mime, 0);
608     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
609
610     hres = FindMimeFromData(NULL, NULL, data1, 0, NULL, 0, &mime, 0);
611     ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
612
613     hres = FindMimeFromData(NULL, url1, data1, 0, NULL, 0, &mime, 0);
614     ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
615
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");
619     CoTaskMemFree(mime);
620
621     hres = FindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, NULL, 0);
622     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
623 }
624
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',
631                               3,0,0,0};
632
633 static struct secmgr_test {
634     LPCWSTR url;
635     DWORD zone;
636     HRESULT zone_hres;
637     DWORD secid_size;
638     const BYTE *secid;
639     HRESULT secid_hres;
640 } secmgr_tests[] = {
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}
648 };
649
650 static void test_SecurityManager(void)
651 {
652     int i;
653     IInternetSecurityManager *secmgr = NULL;
654     BYTE buf[512];
655     DWORD zone, size;
656     HRESULT hres;
657
658     hres = CoInternetCreateSecurityManager(NULL, &secmgr, 0);
659     ok(hres == S_OK, "CoInternetCreateSecurityManager failed: %08x\n", hres);
660     if(FAILED(hres))
661         return;
662
663     for(i=0; i < sizeof(secmgr_tests)/sizeof(secmgr_tests[0]); i++) {
664         zone = 100;
665         hres = IInternetSecurityManager_MapUrlToZone(secmgr, secmgr_tests[i].url,
666                                                      &zone, 0);
667         ok(hres == secmgr_tests[i].zone_hres,
668            "[%d] MapUrlToZone failed: %08x, expected %08x\n",
669                 i, hres, secmgr_tests[i].zone_hres);
670         if(SUCCEEDED(hres))
671             ok(zone == secmgr_tests[i].zone, "[%d] zone=%d, expected %d\n", i, zone,
672                secmgr_tests[i].zone);
673         else
674             ok(zone == secmgr_tests[i].zone || zone == -1, "[%d] zone=%d\n", i, zone);
675
676         size = sizeof(buf);
677         memset(buf, 0xf0, sizeof(buf));
678         hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[i].url,
679                 buf, &size, 0);
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);
687         }
688     }
689
690     zone = 100;
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);
694
695     size = sizeof(buf);
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,
700                                                   NULL, &size, 0);
701     ok(hres == E_INVALIDARG,
702        "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
703     hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[1].url,
704                                                   buf, NULL, 0);
705     ok(hres == E_INVALIDARG,
706        "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
707
708     IInternetSecurityManager_Release(secmgr);
709 }
710
711 static void test_ZoneManager(void)
712 {
713     IInternetZoneManager *zonemgr = NULL;
714     BYTE buf[32];
715     HRESULT hres;
716
717     hres = CoInternetCreateZoneManager(NULL, &zonemgr, 0);
718     ok(hres == S_OK, "CoInternetCreateZoneManager failed: %08x\n", hres);
719     if(FAILED(hres))
720         return;
721
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);
726
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);
730
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);
734
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);
738
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);
742
743     IInternetZoneManager_Release(zonemgr);
744 }
745
746 static void register_protocols(void)
747 {
748     IInternetSession *session;
749     IClassFactory *factory;
750     HRESULT hres;
751
752     static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
753
754     hres = CoInternetGetSession(0, &session, 0);
755     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
756     if(FAILED(hres))
757         return;
758
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);
762     if(FAILED(hres))
763         return;
764
765     IInternetSession_RegisterNameSpace(session, factory, &CLSID_AboutProtocol,
766                                        wszAbout, 0, NULL, 0);
767     IClassFactory_Release(factory);
768
769 }
770
771 static HRESULT WINAPI InternetProtocolInfo_QueryInterface(IInternetProtocolInfo *iface,
772                                                           REFIID riid, void **ppv)
773 {
774     ok(0, "unexpected call\n");
775     return E_NOINTERFACE;
776 }
777
778 static ULONG WINAPI InternetProtocolInfo_AddRef(IInternetProtocolInfo *iface)
779 {
780     return 2;
781 }
782
783 static ULONG WINAPI InternetProtocolInfo_Release(IInternetProtocolInfo *iface)
784 {
785     return 1;
786 }
787
788 static HRESULT WINAPI InternetProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
789         PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult,
790         DWORD *pcchResult, DWORD dwReserved)
791 {
792     CHECK_EXPECT(ParseUrl);
793     return E_NOTIMPL;
794 }
795
796 static HRESULT WINAPI InternetProtocolInfo_CombineUrl(IInternetProtocolInfo *iface,
797         LPCWSTR pwzBaseUrl, LPCWSTR pwzRelativeUrl, DWORD dwCombineFlags,
798         LPWSTR pwzResult, DWORD cchResult, DWORD *pcchResult, DWORD dwReserved)
799 {
800     ok(0, "unexpected call\n");
801     return E_NOTIMPL;
802 }
803
804 static HRESULT WINAPI InternetProtocolInfo_CompareUrl(IInternetProtocolInfo *iface,
805         LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags)
806 {
807     ok(0, "unexpected call\n");
808     return E_NOTIMPL;
809 }
810
811 static HRESULT WINAPI InternetProtocolInfo_QueryInfo(IInternetProtocolInfo *iface,
812         LPCWSTR pwzUrl, QUERYOPTION OueryOption, DWORD dwQueryFlags, LPVOID pBuffer,
813         DWORD cbBuffer, DWORD *pcbBuf, DWORD dwReserved)
814 {
815     ok(0, "unexpected call\n");
816     return E_NOTIMPL;
817 }
818
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
827 };
828
829 static IInternetProtocolInfo protocol_info = { &InternetProtocolInfoVtbl };
830
831 static HRESULT qiret;
832 static IClassFactory *expect_cf;
833
834 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
835 {
836     if(IsEqualGUID(&IID_IInternetProtocolInfo, riid)) {
837         CHECK_EXPECT(QI_IInternetProtocolInfo);
838         ok(iface == expect_cf, "unexpected iface\n");
839         *ppv = &protocol_info;
840         return qiret;
841     }
842
843     ok(0, "unexpected call\n");
844     return E_NOINTERFACE;
845 }
846
847 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
848 {
849     return 2;
850 }
851
852 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
853 {
854     return 1;
855 }
856
857 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
858                                         REFIID riid, void **ppv)
859 {
860     CHECK_EXPECT(CreateInstance);
861
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");
866
867     *ppv = &protocol_info;
868     return S_OK;
869 }
870
871 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
872 {
873     ok(0, "unexpected call\n");
874     return S_OK;
875 }
876
877 static const IClassFactoryVtbl ClassFactoryVtbl = {
878     ClassFactory_QueryInterface,
879     ClassFactory_AddRef,
880     ClassFactory_Release,
881     ClassFactory_CreateInstance,
882     ClassFactory_LockServer
883 };
884
885 static IClassFactory test_protocol_cf = { &ClassFactoryVtbl };
886 static IClassFactory test_protocol_cf2 = { &ClassFactoryVtbl };
887
888 static void test_NameSpace(void)
889 {
890     IInternetSession *session;
891     WCHAR buf[200];
892     DWORD size;
893     HRESULT hres;
894
895     static const WCHAR wszTest[] = {'t','e','s','t',0};
896
897     hres = CoInternetGetSession(0, &session, 0);
898     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
899     if(FAILED(hres))
900         return;
901
902     hres = IInternetSession_RegisterNameSpace(session, NULL, &IID_NULL,
903                                               wszTest, 0, NULL, 0);
904     ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
905
906     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
907                                               NULL, 0, NULL, 0);
908     ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
909
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);
913
914     qiret = E_NOINTERFACE;
915     expect_cf = &test_protocol_cf;
916     SET_EXPECT(QI_IInternetProtocolInfo);
917     SET_EXPECT(CreateInstance);
918     SET_EXPECT(ParseUrl);
919
920     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
921                               &size, 0);
922     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
923
924     CHECK_CALLED(QI_IInternetProtocolInfo);
925     CHECK_CALLED(CreateInstance);
926     CHECK_CALLED(ParseUrl);
927
928     qiret = S_OK;
929     SET_EXPECT(QI_IInternetProtocolInfo);
930     SET_EXPECT(ParseUrl);
931
932     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
933                               &size, 0);
934     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
935
936     CHECK_CALLED(QI_IInternetProtocolInfo);
937     CHECK_CALLED(ParseUrl);
938
939     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
940     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
941
942     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
943                               &size, 0);
944     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
945
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);
949
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);
953
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);
957
958     SET_EXPECT(QI_IInternetProtocolInfo);
959     SET_EXPECT(ParseUrl);
960
961     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
962                               &size, 0);
963     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
964
965     CHECK_CALLED(QI_IInternetProtocolInfo);
966     CHECK_CALLED(ParseUrl);
967
968     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
969     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
970
971     SET_EXPECT(QI_IInternetProtocolInfo);
972     SET_EXPECT(ParseUrl);
973
974     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
975                               &size, 0);
976     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
977
978     CHECK_CALLED(QI_IInternetProtocolInfo);
979     CHECK_CALLED(ParseUrl);
980
981     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
982     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
983
984     expect_cf = &test_protocol_cf2;
985     SET_EXPECT(QI_IInternetProtocolInfo);
986     SET_EXPECT(ParseUrl);
987
988     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
989                               &size, 0);
990     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
991
992     CHECK_CALLED(QI_IInternetProtocolInfo);
993     CHECK_CALLED(ParseUrl);
994
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);
1003
1004     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf2, wszTest);
1005     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1006
1007     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1008                               &size, 0);
1009     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1010
1011     IInternetSession_Release(session);
1012 }
1013
1014 static ULONG WINAPI unk_Release(IUnknown *iface)
1015 {
1016     CHECK_EXPECT(unk_Release);
1017     return 0;
1018 }
1019
1020 static const IUnknownVtbl unk_vtbl = {
1021     (void*)0xdeadbeef,
1022     (void*)0xdeadbeef,
1023     unk_Release
1024 };
1025
1026 static void test_ReleaseBindInfo(void)
1027 {
1028     BINDINFO bi;
1029     IUnknown unk = { &unk_vtbl };
1030
1031     ReleaseBindInfo(NULL); /* shouldn't crash */
1032
1033     memset(&bi, 0, sizeof(bi));
1034     bi.cbSize = sizeof(BINDINFO);
1035     bi.pUnk = &unk;
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);
1041
1042     memset(&bi, 0, sizeof(bi));
1043     bi.cbSize = offsetof(BINDINFO, pUnk);
1044     bi.pUnk = &unk;
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);
1048
1049     memset(&bi, 0, sizeof(bi));
1050     bi.pUnk = &unk;
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);
1054 }
1055
1056 static void test_UrlMkGetSessionOption(void)
1057 {
1058     DWORD encoding, size;
1059     HRESULT hres;
1060
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);
1067
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);
1074
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);
1081
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);
1088
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);
1094 }
1095
1096 static void test_ObtainUserAgentString(void)
1097 {
1098     static const CHAR expected[] = "Mozilla/4.0 (compatible; MSIE ";
1099     static CHAR str[3];
1100     LPSTR str2 = NULL;
1101     HRESULT hres;
1102     DWORD size, saved;
1103
1104     hres = ObtainUserAgentString(0, NULL, NULL);
1105     ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1106
1107     size = 100;
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);
1111
1112     size = 0;
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);
1116
1117     size = 2;
1118     str[0] = 'a';
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]);
1123
1124     size = 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);
1128
1129     str2 = HeapAlloc(GetProcessHeap(), 0, (size+20)*sizeof(CHAR));
1130     if (!str2)
1131     {
1132         skip("skipping rest of ObtainUserAgent tests, out of memory\n");
1133     }
1134     else
1135     {
1136         saved = size;
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",
1143            str2, expected);
1144
1145         size = saved+10;
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);
1149     }
1150     HeapFree(GetProcessHeap(), 0, str2);
1151 }
1152
1153 START_TEST(misc)
1154 {
1155     OleInitialize(NULL);
1156
1157     register_protocols();
1158
1159     test_CreateFormatEnum();
1160     test_RegisterFormatEnumerator();
1161     test_CoInternetParseUrl();
1162     test_CoInternetCompareUrl();
1163     test_FindMimeFromData();
1164     test_SecurityManager();
1165     test_ZoneManager();
1166     test_NameSpace();
1167     test_ReleaseBindInfo();
1168     test_UrlMkGetSessionOption();
1169     test_ObtainUserAgentString();
1170
1171     OleUninitialize();
1172 }