urlmon: Avoid sizeof in traces.
[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 #define NONAMELESSUNION
22
23 #include <wine/test.h>
24 #include <stdarg.h>
25 #include <stddef.h>
26
27 #include "windef.h"
28 #include "winbase.h"
29 #include "ole2.h"
30 #include "urlmon.h"
31
32 #include "initguid.h"
33
34 DEFINE_GUID(CLSID_AboutProtocol, 0x3050F406, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
35
36 #define DEFINE_EXPECT(func) \
37     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
38
39 #define SET_EXPECT(func) \
40     expect_ ## func = TRUE
41
42 #define CHECK_EXPECT(func) \
43     do { \
44         ok(expect_ ##func, "unexpected call " #func "\n"); \
45         expect_ ## func = FALSE; \
46         called_ ## func = TRUE; \
47     }while(0)
48
49 #define CHECK_EXPECT2(func) \
50     do { \
51         ok(expect_ ##func, "unexpected call " #func "\n"); \
52         called_ ## func = TRUE; \
53     }while(0)
54
55 #define CHECK_CALLED(func) \
56     do { \
57         ok(called_ ## func, "expected " #func "\n"); \
58         expect_ ## func = called_ ## func = FALSE; \
59     }while(0)
60
61 DEFINE_EXPECT(ParseUrl);
62 DEFINE_EXPECT(QI_IInternetProtocolInfo);
63 DEFINE_EXPECT(CreateInstance);
64 DEFINE_EXPECT(unk_Release);
65
66 static const char *debugstr_w(LPCWSTR str)
67 {
68     static char buf[1024];
69     WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL);
70     return buf;
71 }
72
73 static void test_CreateFormatEnum(void)
74 {
75     IEnumFORMATETC *fenum = NULL, *fenum2 = NULL;
76     FORMATETC fetc[5];
77     ULONG ul;
78     HRESULT hres;
79
80     static DVTARGETDEVICE dev = {sizeof(dev),0,0,0,0,{0}};
81     static FORMATETC formatetc[] = {
82         {0,&dev,0,0,0},
83         {0,&dev,0,1,0},
84         {0,NULL,0,2,0},
85         {0,NULL,0,3,0},
86         {0,NULL,0,4,0}
87     };
88
89     hres = CreateFormatEnumerator(0, formatetc, &fenum);
90     ok(hres == E_FAIL, "CreateFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
91     hres = CreateFormatEnumerator(0, formatetc, NULL);
92     ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
93     hres = CreateFormatEnumerator(5, formatetc, NULL);
94     ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
95
96
97     hres = CreateFormatEnumerator(5, formatetc, &fenum);
98     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
99     if(FAILED(hres))
100         return;
101
102     hres = IEnumFORMATETC_Next(fenum, 2, NULL, &ul);
103     ok(hres == E_INVALIDARG, "Next failed: %08x, expected E_INVALIDARG\n", hres);
104     ul = 100;
105     hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul);
106     ok(hres == S_OK, "Next failed: %08x\n", hres);
107     ok(ul == 0, "ul=%d, expected 0\n", ul);
108
109     hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul);
110     ok(hres == S_OK, "Next failed: %08x\n", hres);
111     ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex);
112     ok(fetc[1].lindex == 1, "fetc[1].lindex=%d, expected 1\n", fetc[1].lindex);
113     ok(fetc[0].ptd == &dev, "fetc[0].ptd=%p, expected %p\n", fetc[0].ptd, &dev);
114     ok(ul == 2, "ul=%d, expected 2\n", ul);
115
116     hres = IEnumFORMATETC_Skip(fenum, 1);
117     ok(hres == S_OK, "Skip failed: %08x\n", hres);
118
119     hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul);
120     ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres);
121     ok(fetc[0].lindex == 3, "fetc[0].lindex=%d, expected 3\n", fetc[0].lindex);
122     ok(fetc[1].lindex == 4, "fetc[1].lindex=%d, expected 4\n", fetc[1].lindex);
123     ok(fetc[0].ptd == NULL, "fetc[0].ptd=%p, expected NULL\n", fetc[0].ptd);
124     ok(ul == 2, "ul=%d, expected 2\n", ul);
125
126     hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul);
127     ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres);
128     ok(ul == 0, "ul=%d, expected 0\n", ul);
129     ul = 100;
130     hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul);
131     ok(hres == S_OK, "Next failed: %08x\n", hres);
132     ok(ul == 0, "ul=%d, expected 0\n", ul);
133
134     hres = IEnumFORMATETC_Skip(fenum, 3);
135     ok(hres == S_FALSE, "Skip failed: %08x, expected S_FALSE\n", hres);
136
137     hres = IEnumFORMATETC_Reset(fenum);
138     ok(hres == S_OK, "Reset failed: %08x\n", hres);
139
140     hres = IEnumFORMATETC_Next(fenum, 5, fetc, NULL);
141     ok(hres == S_OK, "Next failed: %08x\n", hres);
142     ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex);
143
144     hres = IEnumFORMATETC_Reset(fenum);
145     ok(hres == S_OK, "Reset failed: %08x\n", hres);
146
147     hres = IEnumFORMATETC_Skip(fenum, 2);
148     ok(hres == S_OK, "Skip failed: %08x\n", hres);
149
150     hres = IEnumFORMATETC_Clone(fenum, NULL);
151     ok(hres == E_INVALIDARG, "Clone failed: %08x, expected E_INVALIDARG\n", hres);
152
153     hres = IEnumFORMATETC_Clone(fenum, &fenum2);
154     ok(hres == S_OK, "Clone failed: %08x\n", hres);
155
156     if(SUCCEEDED(hres)) {
157         ok(fenum != fenum2, "fenum == fenum2\n");
158
159         hres = IEnumFORMATETC_Next(fenum2, 2, fetc, &ul);
160         ok(hres == S_OK, "Next failed: %08x\n", hres);
161         ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex);
162
163         IEnumFORMATETC_Release(fenum2);
164     }
165
166     hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul);
167     ok(hres == S_OK, "Next failed: %08x\n", hres);
168     ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex);
169
170     hres = IEnumFORMATETC_Skip(fenum, 1);
171     ok(hres == S_OK, "Skip failed: %08x\n", hres);
172     
173     IEnumFORMATETC_Release(fenum);
174 }
175
176 static void test_RegisterFormatEnumerator(void)
177 {
178     IBindCtx *bctx = NULL;
179     IEnumFORMATETC *format = NULL, *format2 = NULL;
180     IUnknown *unk = NULL;
181     HRESULT hres;
182
183     static FORMATETC formatetc = {0,NULL,0,0,0};
184     static WCHAR wszEnumFORMATETC[] =
185         {'_','E','n','u','m','F','O','R','M','A','T','E','T','C','_',0};
186
187     CreateBindCtx(0, &bctx);
188
189     hres = CreateFormatEnumerator(1, &formatetc, &format);
190     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
191     if(FAILED(hres))
192         return;
193
194     hres = RegisterFormatEnumerator(NULL, format, 0);
195     ok(hres == E_INVALIDARG,
196             "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
197     hres = RegisterFormatEnumerator(bctx, NULL, 0);
198     ok(hres == E_INVALIDARG,
199             "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
200
201     hres = RegisterFormatEnumerator(bctx, format, 0);
202     ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
203
204     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
205     ok(hres == S_OK, "GetObjectParam failed: %08x\n", hres);
206     ok(unk == (IUnknown*)format, "unk != format\n");
207
208     hres = RevokeFormatEnumerator(NULL, format);
209     ok(hres == E_INVALIDARG,
210             "RevokeFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
211
212     hres = RevokeFormatEnumerator(bctx, format);
213     ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
214
215     hres = RevokeFormatEnumerator(bctx, format);
216     ok(hres == E_FAIL, "RevokeFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
217
218     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
219     ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
220
221     hres = RegisterFormatEnumerator(bctx, format, 0);
222     ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
223
224     hres = CreateFormatEnumerator(1, &formatetc, &format2);
225     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
226
227     if(SUCCEEDED(hres)) {
228         hres = RevokeFormatEnumerator(bctx, format);
229         ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
230
231         IEnumFORMATETC_Release(format2);
232     }
233
234     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
235     ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
236
237     IEnumFORMATETC_Release(format);
238
239     hres = RegisterFormatEnumerator(bctx, format, 0);
240     ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
241     hres = RevokeFormatEnumerator(bctx, NULL);
242     ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
243     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
244     ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
245
246     IEnumFORMATETC_Release(format);
247     IBindCtx_Release(bctx);
248 }
249
250 static const WCHAR url1[] = {'r','e','s',':','/','/','m','s','h','t','m','l','.','d','l','l',
251         '/','b','l','a','n','k','.','h','t','m',0};
252 static const WCHAR url2[] = {'i','n','d','e','x','.','h','t','m',0};
253 static const WCHAR url3[] = {'f','i','l','e',':','/','/','c',':','\\','I','n','d','e','x','.','h','t','m',0};
254 static const WCHAR url4[] = {'f','i','l','e',':','s','o','m','e','%','2','0','f','i','l','e',
255         '%','2','e','j','p','g',0};
256 static const WCHAR url5[] = {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q',
257         '.','o','r','g',0};
258 static const WCHAR url6[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
259 static const WCHAR url7[] = {'f','t','p',':','/','/','w','i','n','e','h','q','.','o','r','g','/',
260         'f','i','l','e','.','t','e','s','t',0};
261 static const WCHAR url8[] = {'t','e','s','t',':','1','2','3','a','b','c',0};
262 static const WCHAR url9[] =
263     {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q','.','o','r','g',
264      '/','s','i','t','e','/','a','b','o','u','t',0};
265 static const WCHAR url10[] = {'f','i','l','e',':','/','/','s','o','m','e','%','2','0','f','i','l','e',
266         '.','j','p','g',0};
267
268 static const WCHAR url4e[] = {'f','i','l','e',':','s','o','m','e',' ','f','i','l','e',
269         '.','j','p','g',0};
270
271 static const WCHAR path3[] = {'c',':','\\','I','n','d','e','x','.','h','t','m',0};
272 static const WCHAR path4[] = {'s','o','m','e',' ','f','i','l','e','.','j','p','g',0};
273
274 static const WCHAR wszRes[] = {'r','e','s',0};
275 static const WCHAR wszFile[] = {'f','i','l','e',0};
276 static const WCHAR wszHttp[] = {'h','t','t','p',0};
277 static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
278 static const WCHAR wszEmpty[] = {0};
279
280 struct parse_test {
281     LPCWSTR url;
282     HRESULT secur_hres;
283     LPCWSTR encoded_url;
284     HRESULT path_hres;
285     LPCWSTR path;
286     LPCWSTR schema;
287 };
288
289 static const struct parse_test parse_tests[] = {
290     {url1, S_OK,   url1,  E_INVALIDARG, NULL, wszRes},
291     {url2, E_FAIL, url2,  E_INVALIDARG, NULL, wszEmpty},
292     {url3, E_FAIL, url3,  S_OK, path3,        wszFile},
293     {url4, E_FAIL, url4e, S_OK, path4,        wszFile},
294     {url5, E_FAIL, url5,  E_INVALIDARG, NULL, wszHttp},
295     {url6, S_OK,   url6,  E_INVALIDARG, NULL, wszAbout}
296 };
297
298 static void test_CoInternetParseUrl(void)
299 {
300     HRESULT hres;
301     DWORD size;
302     int i;
303
304     static WCHAR buf[4096];
305
306     memset(buf, 0xf0, sizeof(buf));
307     hres = CoInternetParseUrl(parse_tests[0].url, PARSE_SCHEMA, 0, buf,
308             3, &size, 0);
309     ok(hres == E_POINTER, "schema failed: %08x, expected E_POINTER\n", hres);
310
311     for(i=0; i < sizeof(parse_tests)/sizeof(parse_tests[0]); i++) {
312         memset(buf, 0xf0, sizeof(buf));
313         hres = CoInternetParseUrl(parse_tests[i].url, PARSE_SECURITY_URL, 0, buf,
314                 sizeof(buf)/sizeof(WCHAR), &size, 0);
315         ok(hres == parse_tests[i].secur_hres, "[%d] security url failed: %08x, expected %08x\n",
316                 i, hres, parse_tests[i].secur_hres);
317
318         memset(buf, 0xf0, sizeof(buf));
319         hres = CoInternetParseUrl(parse_tests[i].url, PARSE_ENCODE, 0, buf,
320                 sizeof(buf)/sizeof(WCHAR), &size, 0);
321         ok(hres == S_OK, "[%d] encoding failed: %08x\n", i, hres);
322         ok(size == lstrlenW(parse_tests[i].encoded_url), "[%d] wrong size\n", i);
323         ok(!lstrcmpW(parse_tests[i].encoded_url, buf), "[%d] wrong encoded url\n", i);
324
325         memset(buf, 0xf0, sizeof(buf));
326         hres = CoInternetParseUrl(parse_tests[i].url, PARSE_PATH_FROM_URL, 0, buf,
327                 sizeof(buf)/sizeof(WCHAR), &size, 0);
328         ok(hres == parse_tests[i].path_hres, "[%d] path failed: %08x, expected %08x\n",
329                 i, hres, parse_tests[i].path_hres);
330         if(parse_tests[i].path) {
331             ok(size == lstrlenW(parse_tests[i].path), "[%d] wrong size\n", i);
332             ok(!lstrcmpW(parse_tests[i].path, buf), "[%d] wrong path\n", i);
333         }
334
335         memset(buf, 0xf0, sizeof(buf));
336         hres = CoInternetParseUrl(parse_tests[i].url, PARSE_SCHEMA, 0, buf,
337                 sizeof(buf)/sizeof(WCHAR), &size, 0);
338         ok(hres == S_OK, "[%d] schema failed: %08x\n", i, hres);
339         ok(size == lstrlenW(parse_tests[i].schema), "[%d] wrong size\n", i);
340         ok(!lstrcmpW(parse_tests[i].schema, buf), "[%d] wrong schema\n", i);
341     }
342 }
343
344 static void test_CoInternetCompareUrl(void)
345 {
346     HRESULT hres;
347
348     hres = CoInternetCompareUrl(url1, url1, 0);
349     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
350
351     hres = CoInternetCompareUrl(url1, url3, 0);
352     ok(hres == S_FALSE, "CoInternetParseUrl failed: %08x\n", hres);
353
354     hres = CoInternetCompareUrl(url3, url1, 0);
355     ok(hres == S_FALSE, "CoInternetParseUrl failed: %08x\n", hres);
356 }
357
358 static const struct {
359     LPCWSTR url;
360     DWORD uses_net;
361 } query_info_tests[] = {
362     {url1, 0},
363     {url2, 0},
364     {url3, 0},
365     {url4, 0},
366     {url5, 0},
367     {url6, 0},
368     {url7, 0},
369     {url8, 0}
370 };
371
372 static void test_CoInternetQueryInfo(void)
373 {
374     BYTE buf[100];
375     DWORD cb, i;
376     HRESULT hres;
377
378     for(i=0; i < sizeof(query_info_tests)/sizeof(query_info_tests[0]); i++) {
379         cb = 0xdeadbeef;
380         memset(buf, '?', sizeof(buf));
381         hres = CoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, sizeof(buf), &cb, 0);
382         ok(hres == S_OK, "[%d] CoInternetQueryInfo failed: %08x\n", i, hres);
383         ok(cb == sizeof(DWORD), "[%d] cb = %d\n", i, cb);
384         ok(*(DWORD*)buf == query_info_tests[i].uses_net, "[%d] ret %x, expected %x\n",
385            i, *(DWORD*)buf, query_info_tests[i].uses_net);
386
387         hres = CoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, 3, &cb, 0);
388         ok(hres == E_FAIL, "[%d] CoInternetQueryInfo failed: %08x, expected E_FAIL\n", i, hres);
389         hres = CoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, NULL, sizeof(buf), &cb, 0);
390         ok(hres == E_FAIL, "[%d] CoInternetQueryInfo failed: %08x, expected E_FAIL\n", i, hres);
391
392         memset(buf, '?', sizeof(buf));
393         hres = CoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, sizeof(buf), NULL, 0);
394         ok(hres == S_OK, "[%d] CoInternetQueryInfo failed: %08x\n", i, hres);
395         ok(*(DWORD*)buf == query_info_tests[i].uses_net, "[%d] ret %x, expected %x\n",
396            i, *(DWORD*)buf, query_info_tests[i].uses_net);
397     }
398 }
399
400 static const WCHAR mimeTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
401 static const WCHAR mimeTextPlain[] = {'t','e','x','t','/','p','l','a','i','n',0};
402 static const WCHAR mimeTextRichtext[] = {'t','e','x','t','/','r','i','c','h','t','e','x','t',0};
403 static const WCHAR mimeAppOctetStream[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
404     'o','c','t','e','t','-','s','t','r','e','a','m',0};
405 static const WCHAR mimeImagePjpeg[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
406 static const WCHAR mimeImageGif[] = {'i','m','a','g','e','/','g','i','f',0};
407 static const WCHAR mimeImageBmp[] = {'i','m','a','g','e','/','b','m','p',0};
408 static const WCHAR mimeImageXPng[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
409 static const WCHAR mimeImageTiff[] = {'i','m','a','g','e','/','t','i','f','f',0};
410 static const WCHAR mimeVideoAvi[] = {'v','i','d','e','o','/','a','v','i',0};
411 static const WCHAR mimeVideoMpeg[] = {'v','i','d','e','o','/','m','p','e','g',0};
412 static const WCHAR mimeAppPostscript[] =
413     {'a','p','p','l','i','c','a','t','i','o','n','/','p','o','s','t','s','c','r','i','p','t',0};
414 static const WCHAR mimeAppXCompressed[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
415                                     'x','-','c','o','m','p','r','e','s','s','e','d',0};
416 static const WCHAR mimeAppXZip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
417                                     'x','-','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
418 static const WCHAR mimeAppXGzip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
419                                     'x','-','g','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
420 static const WCHAR mimeAppJava[] = {'a','p','p','l','i','c','a','t','i','o','n','/','j','a','v','a',0};
421 static const WCHAR mimeAppPdf[] = {'a','p','p','l','i','c','a','t','i','o','n','/','p','d','f',0};
422 static const WCHAR mimeAppXMSDownload[] =
423     {'a','p','p','l','i','c','a','t','i','o','n','/','x','-','m','s','d','o','w','n','l','o','a','d',0};
424 static const WCHAR mimeAudioWav[] = {'a','u','d','i','o','/','w','a','v',0};
425 static const WCHAR mimeAudioBasic[] = {'a','u','d','i','o','/','b','a','s','i','c',0};
426
427 static const struct {
428     LPCWSTR url;
429     LPCWSTR mime;
430     HRESULT hres;
431 } mime_tests[] = {
432     {url1, mimeTextHtml, S_OK},
433     {url2, mimeTextHtml, S_OK},
434     {url3, mimeTextHtml, S_OK},
435     {url4, NULL, E_FAIL},
436     {url5, NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)},
437     {url6, NULL, E_FAIL},
438     {url7, NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)}
439 };
440
441 static BYTE data1[] = "test data\n";
442 static BYTE data2[] = {31,'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0};
443 static BYTE data3[] = {0,0,0};
444 static BYTE data4[] = {'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0,0};
445 static BYTE data5[] = {0xa,0xa,0xa,'x',32,'x',0};
446 static BYTE data6[] = {0xfa,0xfa,0xfa,0xfa,'\n','\r','\t','x','x','x',1};
447 static BYTE data7[] = "<html>blahblah";
448 static BYTE data8[] = {'t','e','s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0};
449 static BYTE data9[] = {'t','e',0,'s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0};
450 static BYTE data10[] = "<HtmL>blahblah";
451 static BYTE data11[] = "blah<HTML>blahblah";
452 static BYTE data12[] = "blah<HTMLblahblah";
453 static BYTE data13[] = "blahHTML>blahblah";
454 static BYTE data14[] = "blah<HTMblahblah";
455 static BYTE data15[] = {0xff,0xd8};
456 static BYTE data16[] = {0xff,0xd8,'h'};
457 static BYTE data17[] = {0,0xff,0xd8};
458 static BYTE data18[] = {0xff,0xd8,'<','h','t','m','l','>'};
459 static BYTE data19[] = {'G','I','F','8','7','a'};
460 static BYTE data20[] = {'G','I','F','8','9','a'};
461 static BYTE data21[] = {'G','I','F','8','7'};
462 static BYTE data22[] = {'G','i','F','8','7','a'};
463 static BYTE data23[] = {'G','i','F','8','8','a'};
464 static BYTE data24[] = {'g','i','f','8','7','a'};
465 static BYTE data25[] = {'G','i','F','8','7','A'};
466 static BYTE data26[] = {'G','i','F','8','7','a','<','h','t','m','l','>'};
467 static BYTE data27[] = {0x30,'G','i','F','8','7','A'};
468 static BYTE data28[] = {0x42,0x4d,0x6e,0x42,0x1c,0x00,0x00,0x00,0x00,0x00,0x36,0x00,0x00,0x00};
469 static BYTE data29[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x','x'};
470 static BYTE data30[] = {0x42,0x4d,'x','x','x','x',0x00,0x01,0x00,0x00,'x','x','x','x'};
471 static BYTE data31[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'<','h','t','m','l','>'};
472 static BYTE data32[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'};
473 static BYTE data33[] = {0x00,0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'};
474 static BYTE data34[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'};
475 static BYTE data35[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x','x','x','x',0};
476 static BYTE data36[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,'x','x'};
477 static BYTE data37[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'<','h','t','m','l','>'};
478 static BYTE data38[] = {0x00,0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'};
479 static BYTE data39[] = {0x4d,0x4d,0x00,0x2a};
480 static BYTE data40[] = {0x4d,0x4d,0x00,0x2a,'<','h','t','m','l','>',0};
481 static BYTE data41[] = {0x4d,0x4d,0xff};
482 static BYTE data42[] = {0x4d,0x4d};
483 static BYTE data43[] = {0x00,0x4d,0x4d,0x00};
484 static BYTE data44[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff};
485 static BYTE data45[] = {'R','I','F','f',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff};
486 static BYTE data46[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20};
487 static BYTE data47[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x21,0xff};
488 static BYTE data48[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,'<','h','t','m','l','>'};
489 static BYTE data49[] = {'R','I','F','F',0x0f,0x0f,0xf0,0xf0,'A','V','I',0x20,0xf0,0x00};
490 static BYTE data50[] = {0x00,0x00,0x01,0xb3,0xff};
491 static BYTE data51[] = {0x00,0x00,0x01,0xba,0xff};
492 static BYTE data52[] = {0x00,0x00,0x01,0xb8,0xff};
493 static BYTE data53[] = {0x00,0x00,0x01,0xba};
494 static BYTE data54[] = {0x00,0x00,0x01,0xba,'<','h','t','m','l','>'};
495 static BYTE data55[] = {0x1f,0x8b,'x'};
496 static BYTE data56[] = {0x1f};
497 static BYTE data57[] = {0x1f,0x8b,'<','h','t','m','l','>','t','e','s','t',0};
498 static BYTE data58[] = {0x1f,0x8b};
499 static BYTE data59[] = {0x50,0x4b,'x'};
500 static BYTE data60[] = {0x50,0x4b};
501 static BYTE data61[] = {0x50,0x4b,'<','h','t','m','l','>',0};
502 static BYTE data62[] = {0xca,0xfe,0xba,0xbe,'x'};
503 static BYTE data63[] = {0xca,0xfe,0xba,0xbe};
504 static BYTE data64[] = {0xca,0xfe,0xba,0xbe,'<','h','t','m','l','>',0};
505 static BYTE data65[] = {0x25,0x50,0x44,0x46,'x'};
506 static BYTE data66[] = {0x25,0x50,0x44,0x46};
507 static BYTE data67[] = {0x25,0x50,0x44,0x46,'x','<','h','t','m','l','>'};
508 static BYTE data68[] = {'M','Z','x'};
509 static BYTE data69[] = {'M','Z'};
510 static BYTE data70[] = {'M','Z','<','h','t','m','l','>',0xff};
511 static BYTE data71[] = {'{','\\','r','t','f',0};
512 static BYTE data72[] = {'{','\\','r','t','f'};
513 static BYTE data73[] = {' ','{','\\','r','t','f',' '};
514 static BYTE data74[] = {'{','\\','r','t','f','<','h','t','m','l','>',' '};
515 static BYTE data75[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V','E',0xff};
516 static BYTE data76[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V','E'};
517 static BYTE data77[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V',0xff,0xff};
518 static BYTE data78[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'<','h','t','m','l','>',0xff};
519 static BYTE data79[] = {'%','!',0xff};
520 static BYTE data80[] = {'%','!'};
521 static BYTE data81[] = {'%','!','P','S','<','h','t','m','l','>'};
522 static BYTE data82[] = {'.','s','n','d',0};
523 static BYTE data83[] = {'.','s','n','d'};
524 static BYTE data84[] = {'.','s','n','d',0,'<','h','t','m','l','>',1,1};
525 static BYTE data85[] = {'.','S','N','D',0};
526
527 static const struct {
528     BYTE *data;
529     DWORD size;
530     LPCWSTR mime;
531 } mime_tests2[] = {
532     {data1, sizeof(data1), mimeTextPlain},
533     {data2, sizeof(data2), mimeAppOctetStream},
534     {data3, sizeof(data3), mimeAppOctetStream},
535     {data4, sizeof(data4), mimeAppOctetStream},
536     {data5, sizeof(data5), mimeTextPlain},
537     {data6, sizeof(data6), mimeTextPlain},
538     {data7, sizeof(data7), mimeTextHtml},
539     {data8, sizeof(data8), mimeTextHtml},
540     {data9, sizeof(data9), mimeTextHtml},
541     {data10, sizeof(data10), mimeTextHtml},
542     {data11, sizeof(data11), mimeTextHtml},
543     {data12, sizeof(data12), mimeTextHtml},
544     {data13, sizeof(data13), mimeTextPlain},
545     {data14, sizeof(data14), mimeTextPlain},
546     {data15, sizeof(data15), mimeTextPlain},
547     {data16, sizeof(data16), mimeImagePjpeg},
548     {data17, sizeof(data17), mimeAppOctetStream},
549     {data18, sizeof(data18), mimeTextHtml},
550     {data19, sizeof(data19), mimeImageGif},
551     {data20, sizeof(data20), mimeImageGif},
552     {data21, sizeof(data21), mimeTextPlain},
553     {data22, sizeof(data22), mimeImageGif},
554     {data23, sizeof(data23), mimeTextPlain},
555     {data24, sizeof(data24), mimeImageGif},
556     {data25, sizeof(data25), mimeImageGif},
557     {data26, sizeof(data26), mimeTextHtml},
558     {data27, sizeof(data27), mimeTextPlain},
559     {data28, sizeof(data28), mimeImageBmp},
560     {data29, sizeof(data29), mimeImageBmp},
561     {data30, sizeof(data30), mimeAppOctetStream},
562     {data31, sizeof(data31), mimeTextHtml},
563     {data32, sizeof(data32), mimeAppOctetStream},
564     {data33, sizeof(data33), mimeAppOctetStream},
565     {data34, sizeof(data34), mimeImageXPng},
566     {data35, sizeof(data35), mimeImageXPng},
567     {data36, sizeof(data36), mimeAppOctetStream},
568     {data37, sizeof(data37), mimeTextHtml},
569     {data38, sizeof(data38), mimeAppOctetStream},
570     {data39, sizeof(data39), mimeImageTiff},
571     {data40, sizeof(data40), mimeTextHtml},
572     {data41, sizeof(data41), mimeImageTiff},
573     {data42, sizeof(data42), mimeTextPlain},
574     {data43, sizeof(data43), mimeAppOctetStream},
575     {data44, sizeof(data44), mimeVideoAvi},
576     {data45, sizeof(data45), mimeTextPlain},
577     {data46, sizeof(data46), mimeTextPlain},
578     {data47, sizeof(data47), mimeTextPlain},
579     {data48, sizeof(data48), mimeTextHtml},
580     {data49, sizeof(data49), mimeVideoAvi},
581     {data50, sizeof(data50), mimeVideoMpeg},
582     {data51, sizeof(data51), mimeVideoMpeg},
583     {data52, sizeof(data52), mimeAppOctetStream},
584     {data53, sizeof(data53), mimeAppOctetStream},
585     {data54, sizeof(data54), mimeTextHtml},
586     {data55, sizeof(data55), mimeAppXGzip},
587     {data56, sizeof(data56), mimeTextPlain},
588     {data57, sizeof(data57), mimeTextHtml},
589     {data58, sizeof(data58), mimeAppOctetStream},
590     {data59, sizeof(data59), mimeAppXZip},
591     {data60, sizeof(data60), mimeTextPlain},
592     {data61, sizeof(data61), mimeTextHtml},
593     {data62, sizeof(data62), mimeAppJava},
594     {data63, sizeof(data63), mimeTextPlain},
595     {data64, sizeof(data64), mimeTextHtml},
596     {data65, sizeof(data65), mimeAppPdf},
597     {data66, sizeof(data66), mimeTextPlain},
598     {data67, sizeof(data67), mimeTextHtml},
599     {data68, sizeof(data68), mimeAppXMSDownload},
600     {data69, sizeof(data69), mimeTextPlain},
601     {data70, sizeof(data70), mimeTextHtml},
602     {data71, sizeof(data71), mimeTextRichtext},
603     {data72, sizeof(data72), mimeTextPlain},
604     {data73, sizeof(data73), mimeTextPlain},
605     {data74, sizeof(data74), mimeTextHtml},
606     {data75, sizeof(data75), mimeAudioWav},
607     {data76, sizeof(data76), mimeTextPlain},
608     {data77, sizeof(data77), mimeTextPlain},
609     {data78, sizeof(data78), mimeTextHtml},
610     {data79, sizeof(data79), mimeAppPostscript},
611     {data80, sizeof(data80), mimeTextPlain},
612     {data81, sizeof(data81), mimeTextHtml},
613     {data82, sizeof(data82), mimeAudioBasic},
614     {data83, sizeof(data83), mimeTextPlain},
615     {data84, sizeof(data84), mimeTextHtml},
616     {data85, sizeof(data85), mimeTextPlain}
617 };
618
619 static void test_FindMimeFromData(void)
620 {
621     HRESULT hres;
622     LPWSTR mime;
623     int i;
624
625     for(i=0; i<sizeof(mime_tests)/sizeof(mime_tests[0]); i++) {
626         mime = (LPWSTR)0xf0f0f0f0;
627         hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, NULL, 0, &mime, 0);
628         if(mime_tests[i].mime) {
629             ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
630             ok(!lstrcmpW(mime, mime_tests[i].mime), "[%d] wrong mime\n", i);
631             CoTaskMemFree(mime);
632         }else {
633             ok(hres == E_FAIL || hres == mime_tests[i].hres,
634                "[%d] FindMimeFromData failed: %08x, expected %08x\n",
635                i, hres, mime_tests[i].hres);
636             ok(mime == (LPWSTR)0xf0f0f0f0, "[%d] mime != 0xf0f0f0f0\n", i);
637         }
638
639         mime = (LPWSTR)0xf0f0f0f0;
640         hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, mimeTextPlain, 0, &mime, 0);
641         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
642         ok(!lstrcmpW(mime, mimeTextPlain), "[%d] wrong mime\n", i);
643         CoTaskMemFree(mime);
644
645         mime = (LPWSTR)0xf0f0f0f0;
646         hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, mimeAppOctetStream, 0, &mime, 0);
647         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
648         ok(!lstrcmpW(mime, mimeAppOctetStream), "[%d] wrong mime\n", i);
649         CoTaskMemFree(mime);
650     }
651
652     for(i=0; i < sizeof(mime_tests2)/sizeof(mime_tests2[0]); i++) {
653         hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
654                 NULL, 0, &mime, 0);
655         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
656         ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime: %s\n", i, debugstr_w(mime));
657         CoTaskMemFree(mime);
658
659         hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
660                 mimeTextHtml, 0, &mime, 0);
661         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
662         if(!lstrcmpW(mimeAppOctetStream, mime_tests2[i].mime)
663            || !lstrcmpW(mimeTextPlain, mime_tests2[i].mime))
664             ok(!lstrcmpW(mime, mimeTextHtml), "[%d] wrong mime\n", i);
665         else
666             ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
667         CoTaskMemFree(mime);
668
669         hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
670                 mimeImagePjpeg, 0, &mime, 0);
671         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
672         if(!lstrcmpW(mimeAppOctetStream, mime_tests2[i].mime) || i == 17)
673             ok(!lstrcmpW(mime, mimeImagePjpeg), "[%d] wrong mime\n", i);
674         else
675             ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
676
677         CoTaskMemFree(mime);
678     }
679
680     hres = FindMimeFromData(NULL, url1, data1, sizeof(data1), NULL, 0, &mime, 0);
681     ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
682     ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
683     CoTaskMemFree(mime);
684
685     hres = FindMimeFromData(NULL, url1, data1, sizeof(data1), mimeAppOctetStream, 0, &mime, 0);
686     ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
687     ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
688     CoTaskMemFree(mime);
689
690     hres = FindMimeFromData(NULL, url4, data1, sizeof(data1), mimeAppOctetStream, 0, &mime, 0);
691     ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
692     ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
693     CoTaskMemFree(mime);
694
695     hres = FindMimeFromData(NULL, NULL, NULL, 0, NULL, 0, &mime, 0);
696     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, excepted E_INVALIDARG\n", hres);
697
698     hres = FindMimeFromData(NULL, NULL, NULL, 0, mimeTextPlain, 0, &mime, 0);
699     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
700
701     hres = FindMimeFromData(NULL, NULL, data1, 0, NULL, 0, &mime, 0);
702     ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
703
704     hres = FindMimeFromData(NULL, url1, data1, 0, NULL, 0, &mime, 0);
705     ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
706
707     hres = FindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, &mime, 0);
708     ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
709     ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
710     CoTaskMemFree(mime);
711
712     hres = FindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, NULL, 0);
713     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
714 }
715
716 static const BYTE secid1[] = {'f','i','l','e',':',0,0,0,0};
717 static const BYTE secid5[] = {'h','t','t','p',':','w','w','w','.','w','i','n','e','h','q',
718     '.','o','r','g',3,0,0,0};
719 static const BYTE secid6[] = {'a','b','o','u','t',':','b','l','a','n','k',3,0,0,0};
720 static const BYTE secid7[] = {'f','t','p',':','w','i','n','e','h','q','.','o','r','g',
721                               3,0,0,0};
722 static const BYTE secid10[] =
723     {'f','i','l','e',':','s','o','m','e','%','2','0','f','i','l','e','.','j','p','g',3,0,0,0};
724 static const BYTE secid10_2[] =
725     {'f','i','l','e',':','s','o','m','e',' ','f','i','l','e','.','j','p','g',3,0,0,0};
726
727 static struct secmgr_test {
728     LPCWSTR url;
729     DWORD zone;
730     HRESULT zone_hres;
731     DWORD secid_size;
732     const BYTE *secid;
733     HRESULT secid_hres;
734 } secmgr_tests[] = {
735     {url1, 0,   S_OK, sizeof(secid1), secid1, S_OK},
736     {url2, 100, 0x80041001, 0, NULL, E_INVALIDARG},
737     {url3, 0,   S_OK, sizeof(secid1), secid1, S_OK},
738     {url5, 3,   S_OK, sizeof(secid5), secid5, S_OK},
739     {url6, 3,   S_OK, sizeof(secid6), secid6, S_OK},
740     {url7, 3,   S_OK, sizeof(secid7), secid7, S_OK}
741 };
742
743 static void test_SecurityManager(void)
744 {
745     int i;
746     IInternetSecurityManager *secmgr = NULL;
747     BYTE buf[512];
748     DWORD zone, size, policy;
749     HRESULT hres;
750
751     hres = CoInternetCreateSecurityManager(NULL, &secmgr, 0);
752     ok(hres == S_OK, "CoInternetCreateSecurityManager failed: %08x\n", hres);
753     if(FAILED(hres))
754         return;
755
756     for(i=0; i < sizeof(secmgr_tests)/sizeof(secmgr_tests[0]); i++) {
757         zone = 100;
758         hres = IInternetSecurityManager_MapUrlToZone(secmgr, secmgr_tests[i].url,
759                                                      &zone, 0);
760         ok(hres == secmgr_tests[i].zone_hres /* IE <=6 */
761            || (FAILED(secmgr_tests[i].zone_hres) && hres == E_INVALIDARG), /* IE7 */
762            "[%d] MapUrlToZone failed: %08x, expected %08x\n",
763                 i, hres, secmgr_tests[i].zone_hres);
764         if(SUCCEEDED(hres))
765             ok(zone == secmgr_tests[i].zone, "[%d] zone=%d, expected %d\n", i, zone,
766                secmgr_tests[i].zone);
767         else
768             ok(zone == secmgr_tests[i].zone || zone == -1, "[%d] zone=%d\n", i, zone);
769
770         size = sizeof(buf);
771         memset(buf, 0xf0, sizeof(buf));
772         hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[i].url,
773                 buf, &size, 0);
774         ok(hres == secmgr_tests[i].secid_hres,
775            "[%d] GetSecurityId failed: %08x, expected %08x\n",
776            i, hres, secmgr_tests[i].secid_hres);
777         if(secmgr_tests[i].secid) {
778             ok(size == secmgr_tests[i].secid_size, "[%d] size=%d, expected %d\n",
779                     i, size, secmgr_tests[i].secid_size);
780             ok(!memcmp(buf, secmgr_tests[i].secid, size), "[%d] wrong secid\n", i);
781         }
782     }
783
784     zone = 100;
785     hres = IInternetSecurityManager_MapUrlToZone(secmgr, url10, &zone, 0);
786     ok(hres == S_OK, "MapUrlToZone failed: %08x, expected S_OK\n", hres);
787     ok(zone == 3, "zone=%d, expected 3\n", zone);
788
789     /* win2k3 translates %20 into a space */
790     size = sizeof(buf);
791     memset(buf, 0xf0, sizeof(buf));
792     hres = IInternetSecurityManager_GetSecurityId(secmgr, url10, buf, &size, 0);
793     ok(hres == S_OK, "GetSecurityId failed: %08x, expected S_OK\n", hres);
794     ok(size == sizeof(secid10) ||
795        size == sizeof(secid10_2), /* win2k3 */
796        "size=%d\n", size);
797     ok(!memcmp(buf, secid10, size) ||
798        !memcmp(buf, secid10_2, size), /* win2k3 */
799        "wrong secid\n");
800
801     zone = 100;
802     hres = IInternetSecurityManager_MapUrlToZone(secmgr, NULL, &zone, 0);
803     ok(hres == E_INVALIDARG, "MapUrlToZone failed: %08x, expected E_INVALIDARG\n", hres);
804     ok(zone == 100 || zone == -1, "zone=%d\n", zone);
805
806     size = sizeof(buf);
807     hres = IInternetSecurityManager_GetSecurityId(secmgr, NULL, buf, &size, 0);
808     ok(hres == E_INVALIDARG,
809        "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
810     hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[1].url,
811                                                   NULL, &size, 0);
812     ok(hres == E_INVALIDARG,
813        "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
814     hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[1].url,
815                                                   buf, NULL, 0);
816     ok(hres == E_INVALIDARG,
817        "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
818
819     hres = IInternetSecurityManager_ProcessUrlAction(secmgr, NULL, URLACTION_SCRIPT_RUN, (BYTE*)&policy,
820             sizeof(WCHAR), NULL, 0, 0, 0);
821     ok(hres == E_INVALIDARG, "ProcessUrlAction failed: %08x, expected E_INVALIDARG\n", hres);
822
823     IInternetSecurityManager_Release(secmgr);
824 }
825
826 static void test_url_action(IInternetSecurityManager *secmgr, IInternetZoneManager *zonemgr, DWORD action)
827 {
828     DWORD res, size, policy, reg_policy;
829     char buf[10];
830     HKEY hkey;
831     HRESULT hres;
832
833     res = RegOpenKeyA(HKEY_LOCAL_MACHINE,
834             "Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings\\Zones\\3", &hkey);
835     if(res != ERROR_SUCCESS) {
836         ok(0, "Could not open zone key\n");
837         return;
838     }
839
840     wsprintf(buf, "%X", action);
841     size = sizeof(DWORD);
842     res = RegQueryValueExA(hkey, buf, NULL, NULL, (BYTE*)&reg_policy, &size);
843     RegCloseKey(hkey);
844     if(res != ERROR_SUCCESS || size != sizeof(DWORD)) {
845         policy = 0xdeadbeef;
846         hres = IInternetSecurityManager_ProcessUrlAction(secmgr, url9, action, (BYTE*)&policy,
847                 sizeof(WCHAR), NULL, 0, 0, 0);
848         ok(hres == E_FAIL, "ProcessUrlAction(%x) failed: %08x, expected E_FAIL\n", action, hres);
849         ok(policy == 0xdeadbeef, "(%x) policy=%x\n", action, policy);
850
851         policy = 0xdeadbeef;
852         hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, action, (BYTE*)&policy,
853                 sizeof(DWORD), URLZONEREG_DEFAULT);
854         ok(hres == E_FAIL, "GetZoneActionPolicy failed: %08x, expected E_FAIL\n", hres);
855         ok(policy == 0xdeadbeef, "(%x) policy=%x\n", action, policy);
856         return;
857     }
858
859     policy = 0xdeadbeef;
860     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, action, (BYTE*)&policy,
861             sizeof(DWORD), URLZONEREG_DEFAULT);
862     ok(hres == S_OK, "GetZoneActionPolicy failed: %08x\n", hres);
863     ok(policy == reg_policy, "(%x) policy=%x, expected %x\n", action, policy, reg_policy);
864
865     if(policy != URLPOLICY_QUERY) {
866         policy = 0xdeadbeef;
867         hres = IInternetSecurityManager_ProcessUrlAction(secmgr, url9, action, (BYTE*)&policy,
868                 sizeof(WCHAR), NULL, 0, 0, 0);
869         if(reg_policy == URLPOLICY_DISALLOW)
870             ok(hres == S_FALSE, "ProcessUrlAction(%x) failed: %08x, expected S_FALSE\n", action, hres);
871         else
872             ok(hres == S_OK, "ProcessUrlAction(%x) failed: %08x\n", action, hres);
873         ok(policy == 0xdeadbeef, "(%x) policy=%x\n", action, policy);
874     }
875 }
876
877 static void test_special_url_action(IInternetSecurityManager *secmgr, IInternetZoneManager *zonemgr, DWORD action)
878 {
879     DWORD policy;
880     HRESULT hres;
881
882     policy = 0xdeadbeef;
883     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, action, (BYTE*)&policy,
884             sizeof(DWORD), URLZONEREG_DEFAULT);
885     ok(hres == S_OK, "GetZoneActionPolicy failed: %08x\n", hres);
886     ok(policy == URLPOLICY_DISALLOW, "(%x) policy=%x, expected URLPOLIVY_DISALLOW\n", action, policy);
887
888     policy = 0xdeadbeef;
889     hres = IInternetSecurityManager_ProcessUrlAction(secmgr, url1, action, (BYTE*)&policy,
890             sizeof(WCHAR), NULL, 0, 0, 0);
891     ok(hres == S_FALSE, "ProcessUrlAction(%x) failed: %08x, expected S_FALSE\n", action, hres);
892 }
893
894 static void test_polices(void)
895 {
896     IInternetZoneManager *zonemgr = NULL;
897     IInternetSecurityManager *secmgr = NULL;
898     HRESULT hres;
899
900     hres = CoInternetCreateSecurityManager(NULL, &secmgr, 0);
901     ok(hres == S_OK, "CoInternetCreateSecurityManager failed: %08x\n", hres);
902     hres = CoInternetCreateZoneManager(NULL, &zonemgr, 0);
903     ok(hres == S_OK, "CoInternetCreateZoneManager failed: %08x\n", hres);
904
905     test_url_action(secmgr, zonemgr, URLACTION_SCRIPT_RUN);
906     test_url_action(secmgr, zonemgr, URLACTION_ACTIVEX_OVERRIDE_OBJECT_SAFETY);
907     test_url_action(secmgr, zonemgr, URLACTION_CHANNEL_SOFTDIST_PERMISSIONS);
908     test_url_action(secmgr, zonemgr, 0xdeadbeef);
909
910     test_special_url_action(secmgr, zonemgr, URLACTION_SCRIPT_OVERRIDE_SAFETY);
911
912     IInternetSecurityManager_Release(secmgr);
913     IInternetZoneManager_Release(zonemgr);
914 }
915
916 static void test_ZoneManager(void)
917 {
918     IInternetZoneManager *zonemgr = NULL;
919     BYTE buf[32];
920     HRESULT hres;
921
922     hres = CoInternetCreateZoneManager(NULL, &zonemgr, 0);
923     ok(hres == S_OK, "CoInternetCreateZoneManager failed: %08x\n", hres);
924     if(FAILED(hres))
925         return;
926
927     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, buf,
928             sizeof(DWORD), URLZONEREG_DEFAULT);
929     ok(hres == S_OK, "GetZoneActionPolicy failed: %08x\n", hres);
930     ok(*(DWORD*)buf == 1, "policy=%d, expected 1\n", *(DWORD*)buf);
931
932     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, NULL,
933             sizeof(DWORD), URLZONEREG_DEFAULT);
934     ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
935
936     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, buf,
937             2, URLZONEREG_DEFAULT);
938     ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
939
940     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1fff, buf,
941             sizeof(DWORD), URLZONEREG_DEFAULT);
942     ok(hres == E_FAIL, "GetZoneActionPolicy failed: %08x, expected E_FAIL\n", hres);
943
944     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 13, 0x1a10, buf,
945             sizeof(DWORD), URLZONEREG_DEFAULT);
946     ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
947
948     IInternetZoneManager_Release(zonemgr);
949 }
950
951 static void register_protocols(void)
952 {
953     IInternetSession *session;
954     IClassFactory *factory;
955     HRESULT hres;
956
957     static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
958
959     hres = CoInternetGetSession(0, &session, 0);
960     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
961     if(FAILED(hres))
962         return;
963
964     hres = CoGetClassObject(&CLSID_AboutProtocol, CLSCTX_INPROC_SERVER, NULL,
965             &IID_IClassFactory, (void**)&factory);
966     ok(hres == S_OK, "Coud not get AboutProtocol factory: %08x\n", hres);
967     if(FAILED(hres))
968         return;
969
970     IInternetSession_RegisterNameSpace(session, factory, &CLSID_AboutProtocol,
971                                        wszAbout, 0, NULL, 0);
972     IClassFactory_Release(factory);
973
974     IInternetSession_Release(session);
975 }
976
977 static HRESULT WINAPI InternetProtocolInfo_QueryInterface(IInternetProtocolInfo *iface,
978                                                           REFIID riid, void **ppv)
979 {
980     ok(0, "unexpected call\n");
981     return E_NOINTERFACE;
982 }
983
984 static ULONG WINAPI InternetProtocolInfo_AddRef(IInternetProtocolInfo *iface)
985 {
986     return 2;
987 }
988
989 static ULONG WINAPI InternetProtocolInfo_Release(IInternetProtocolInfo *iface)
990 {
991     return 1;
992 }
993
994 static HRESULT WINAPI InternetProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
995         PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult,
996         DWORD *pcchResult, DWORD dwReserved)
997 {
998     CHECK_EXPECT(ParseUrl);
999     return E_NOTIMPL;
1000 }
1001
1002 static HRESULT WINAPI InternetProtocolInfo_CombineUrl(IInternetProtocolInfo *iface,
1003         LPCWSTR pwzBaseUrl, LPCWSTR pwzRelativeUrl, DWORD dwCombineFlags,
1004         LPWSTR pwzResult, DWORD cchResult, DWORD *pcchResult, DWORD dwReserved)
1005 {
1006     ok(0, "unexpected call\n");
1007     return E_NOTIMPL;
1008 }
1009
1010 static HRESULT WINAPI InternetProtocolInfo_CompareUrl(IInternetProtocolInfo *iface,
1011         LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags)
1012 {
1013     ok(0, "unexpected call\n");
1014     return E_NOTIMPL;
1015 }
1016
1017 static HRESULT WINAPI InternetProtocolInfo_QueryInfo(IInternetProtocolInfo *iface,
1018         LPCWSTR pwzUrl, QUERYOPTION OueryOption, DWORD dwQueryFlags, LPVOID pBuffer,
1019         DWORD cbBuffer, DWORD *pcbBuf, DWORD dwReserved)
1020 {
1021     ok(0, "unexpected call\n");
1022     return E_NOTIMPL;
1023 }
1024
1025 static const IInternetProtocolInfoVtbl InternetProtocolInfoVtbl = {
1026     InternetProtocolInfo_QueryInterface,
1027     InternetProtocolInfo_AddRef,
1028     InternetProtocolInfo_Release,
1029     InternetProtocolInfo_ParseUrl,
1030     InternetProtocolInfo_CombineUrl,
1031     InternetProtocolInfo_CompareUrl,
1032     InternetProtocolInfo_QueryInfo
1033 };
1034
1035 static IInternetProtocolInfo protocol_info = { &InternetProtocolInfoVtbl };
1036
1037 static HRESULT qiret;
1038 static IClassFactory *expect_cf;
1039
1040 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
1041 {
1042     if(IsEqualGUID(&IID_IInternetProtocolInfo, riid)) {
1043         CHECK_EXPECT(QI_IInternetProtocolInfo);
1044         ok(iface == expect_cf, "unexpected iface\n");
1045         *ppv = &protocol_info;
1046         return qiret;
1047     }
1048
1049     ok(0, "unexpected call\n");
1050     return E_NOINTERFACE;
1051 }
1052
1053 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
1054 {
1055     return 2;
1056 }
1057
1058 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
1059 {
1060     return 1;
1061 }
1062
1063 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
1064                                         REFIID riid, void **ppv)
1065 {
1066     ok(0, "unexpected call\n");
1067     return E_NOTIMPL;
1068 }
1069
1070 static HRESULT WINAPI ProtocolCF_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
1071                                         REFIID riid, void **ppv)
1072 {
1073     CHECK_EXPECT(CreateInstance);
1074
1075     ok(iface == expect_cf, "unexpected iface\n");
1076     ok(pOuter == NULL, "pOuter = %p\n", pOuter);
1077     ok(IsEqualGUID(&IID_IInternetProtocolInfo, riid), "unexpected riid\n");
1078     ok(ppv != NULL, "ppv == NULL\n");
1079
1080     *ppv = &protocol_info;
1081     return S_OK;
1082 }
1083
1084 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
1085 {
1086     ok(0, "unexpected call\n");
1087     return S_OK;
1088 }
1089
1090 static const IClassFactoryVtbl ClassFactoryVtbl = {
1091     ClassFactory_QueryInterface,
1092     ClassFactory_AddRef,
1093     ClassFactory_Release,
1094     ClassFactory_CreateInstance,
1095     ClassFactory_LockServer
1096 };
1097
1098 static const IClassFactoryVtbl ProtocolCFVtbl = {
1099     ClassFactory_QueryInterface,
1100     ClassFactory_AddRef,
1101     ClassFactory_Release,
1102     ProtocolCF_CreateInstance,
1103     ClassFactory_LockServer
1104 };
1105
1106 static IClassFactory test_protocol_cf = { &ProtocolCFVtbl };
1107 static IClassFactory test_protocol_cf2 = { &ProtocolCFVtbl };
1108 static IClassFactory test_cf = { &ClassFactoryVtbl };
1109
1110 static void test_NameSpace(void)
1111 {
1112     IInternetSession *session;
1113     WCHAR buf[200];
1114     DWORD size;
1115     HRESULT hres;
1116
1117     static const WCHAR wszTest[] = {'t','e','s','t',0};
1118
1119     hres = CoInternetGetSession(0, &session, 0);
1120     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
1121     if(FAILED(hres))
1122         return;
1123
1124     hres = IInternetSession_RegisterNameSpace(session, NULL, &IID_NULL,
1125                                               wszTest, 0, NULL, 0);
1126     ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
1127
1128     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
1129                                               NULL, 0, NULL, 0);
1130     ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
1131
1132     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
1133                                               wszTest, 0, NULL, 0);
1134     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1135
1136     qiret = E_NOINTERFACE;
1137     expect_cf = &test_protocol_cf;
1138     SET_EXPECT(QI_IInternetProtocolInfo);
1139     SET_EXPECT(CreateInstance);
1140     SET_EXPECT(ParseUrl);
1141
1142     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1143                               &size, 0);
1144     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1145
1146     CHECK_CALLED(QI_IInternetProtocolInfo);
1147     CHECK_CALLED(CreateInstance);
1148     CHECK_CALLED(ParseUrl);
1149
1150     qiret = S_OK;
1151     SET_EXPECT(QI_IInternetProtocolInfo);
1152     SET_EXPECT(ParseUrl);
1153
1154     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1155                               &size, 0);
1156     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1157
1158     CHECK_CALLED(QI_IInternetProtocolInfo);
1159     CHECK_CALLED(ParseUrl);
1160
1161     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1162     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1163
1164     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1165                               &size, 0);
1166     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1167
1168     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf2, &IID_NULL,
1169                                               wszTest, 0, NULL, 0);
1170     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1171
1172     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
1173                                               wszTest, 0, NULL, 0);
1174     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1175
1176     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
1177                                               wszTest, 0, NULL, 0);
1178     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1179
1180     SET_EXPECT(QI_IInternetProtocolInfo);
1181     SET_EXPECT(ParseUrl);
1182
1183     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1184                               &size, 0);
1185     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1186
1187     CHECK_CALLED(QI_IInternetProtocolInfo);
1188     CHECK_CALLED(ParseUrl);
1189
1190     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1191     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1192
1193     SET_EXPECT(QI_IInternetProtocolInfo);
1194     SET_EXPECT(ParseUrl);
1195
1196     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1197                               &size, 0);
1198     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1199
1200     CHECK_CALLED(QI_IInternetProtocolInfo);
1201     CHECK_CALLED(ParseUrl);
1202
1203     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1204     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1205
1206     expect_cf = &test_protocol_cf2;
1207     SET_EXPECT(QI_IInternetProtocolInfo);
1208     SET_EXPECT(ParseUrl);
1209
1210     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1211                               &size, 0);
1212     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1213
1214     CHECK_CALLED(QI_IInternetProtocolInfo);
1215     CHECK_CALLED(ParseUrl);
1216
1217     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1218     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1219     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1220     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1221     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, NULL);
1222     ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres);
1223     hres = IInternetSession_UnregisterNameSpace(session, NULL, wszTest);
1224     ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres);
1225
1226     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf2, wszTest);
1227     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1228
1229     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1230                               &size, 0);
1231     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1232
1233     IInternetSession_Release(session);
1234 }
1235
1236 static void test_MimeFilter(void)
1237 {
1238     IInternetSession *session;
1239     HRESULT hres;
1240
1241     static const WCHAR mimeW[] = {'t','e','s','t','/','m','i','m','e',0};
1242
1243     hres = CoInternetGetSession(0, &session, 0);
1244     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
1245     if(FAILED(hres))
1246         return;
1247
1248     hres = IInternetSession_RegisterMimeFilter(session, &test_cf, &IID_NULL, mimeW);
1249     ok(hres == S_OK, "RegisterMimeFilter failed: %08x\n", hres);
1250
1251     hres = IInternetSession_UnregisterMimeFilter(session, &test_cf, mimeW);
1252     ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres);
1253
1254     hres = IInternetSession_UnregisterMimeFilter(session, &test_cf, mimeW);
1255     ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres);
1256
1257     hres = IInternetSession_UnregisterMimeFilter(session, (void*)0xdeadbeef, mimeW);
1258     ok(hres == S_OK, "UnregisterMimeFilter failed: %08x\n", hres);
1259
1260     IInternetSession_Release(session);
1261 }
1262
1263 static ULONG WINAPI unk_Release(IUnknown *iface)
1264 {
1265     CHECK_EXPECT(unk_Release);
1266     return 0;
1267 }
1268
1269 static const IUnknownVtbl unk_vtbl = {
1270     (void*)0xdeadbeef,
1271     (void*)0xdeadbeef,
1272     unk_Release
1273 };
1274
1275 static void test_ReleaseBindInfo(void)
1276 {
1277     BINDINFO bi;
1278     IUnknown unk = { &unk_vtbl };
1279
1280     ReleaseBindInfo(NULL); /* shouldn't crash */
1281
1282     memset(&bi, 0, sizeof(bi));
1283     bi.cbSize = sizeof(BINDINFO);
1284     bi.pUnk = &unk;
1285     SET_EXPECT(unk_Release);
1286     ReleaseBindInfo(&bi);
1287     ok(bi.cbSize == sizeof(BINDINFO), "bi.cbSize=%d\n", bi.cbSize);
1288     ok(bi.pUnk == NULL, "bi.pUnk=%p, expected NULL\n", bi.pUnk);
1289     CHECK_CALLED(unk_Release);
1290
1291     memset(&bi, 0, sizeof(bi));
1292     bi.cbSize = offsetof(BINDINFO, pUnk);
1293     bi.pUnk = &unk;
1294     ReleaseBindInfo(&bi);
1295     ok(bi.cbSize == offsetof(BINDINFO, pUnk), "bi.cbSize=%d\n", bi.cbSize);
1296     ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk);
1297
1298     memset(&bi, 0, sizeof(bi));
1299     bi.pUnk = &unk;
1300     ReleaseBindInfo(&bi);
1301     ok(!bi.cbSize, "bi.cbSize=%d, expected 0\n", bi.cbSize);
1302     ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk);
1303 }
1304
1305 static void test_CopyStgMedium(void)
1306 {
1307     STGMEDIUM src, dst;
1308     HRESULT hres;
1309
1310     static WCHAR fileW[] = {'f','i','l','e',0};
1311
1312     memset(&src, 0xf0, sizeof(src));
1313     memset(&dst, 0xe0, sizeof(dst));
1314     src.tymed = TYMED_NULL;
1315     src.pUnkForRelease = NULL;
1316     hres = CopyStgMedium(&src, &dst);
1317     ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres);
1318     ok(dst.tymed == TYMED_NULL, "tymed=%d\n", dst.tymed);
1319     ok(dst.u.hGlobal == (void*)0xf0f0f0f0, "u=%p\n", dst.u.hGlobal);
1320     ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease);
1321
1322     memset(&dst, 0xe0, sizeof(dst));
1323     src.tymed = TYMED_ISTREAM;
1324     src.u.pstm = NULL;
1325     src.pUnkForRelease = NULL;
1326     hres = CopyStgMedium(&src, &dst);
1327     ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres);
1328     ok(dst.tymed == TYMED_ISTREAM, "tymed=%d\n", dst.tymed);
1329     ok(!dst.u.pstm, "pstm=%p\n", dst.u.pstm);
1330     ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease);
1331
1332     memset(&dst, 0xe0, sizeof(dst));
1333     src.tymed = TYMED_FILE;
1334     src.u.lpszFileName = fileW;
1335     src.pUnkForRelease = NULL;
1336     hres = CopyStgMedium(&src, &dst);
1337     ok(hres == S_OK, "CopyStgMedium failed: %08x\n", hres);
1338     ok(dst.tymed == TYMED_FILE, "tymed=%d\n", dst.tymed);
1339     ok(dst.u.lpszFileName && dst.u.lpszFileName != fileW, "lpszFileName=%p\n", dst.u.lpszFileName);
1340     ok(!lstrcmpW(dst.u.lpszFileName, fileW), "wrong file name\n");
1341     ok(!dst.pUnkForRelease, "pUnkForRelease=%p, expected NULL\n", dst.pUnkForRelease);
1342
1343     hres = CopyStgMedium(&src, NULL);
1344     ok(hres == E_POINTER, "CopyStgMedium failed: %08x, expected E_POINTER\n", hres);
1345     hres = CopyStgMedium(NULL, &dst);
1346     ok(hres == E_POINTER, "CopyStgMedium failed: %08x, expected E_POINTER\n", hres);
1347 }
1348
1349 static void test_UrlMkGetSessionOption(void)
1350 {
1351     DWORD encoding, size;
1352     HRESULT hres;
1353
1354     size = encoding = 0xdeadbeef;
1355     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1356                                  sizeof(encoding), &size, 0);
1357     ok(hres == S_OK, "UrlMkGetSessionOption failed: %08x\n", hres);
1358     ok(encoding != 0xdeadbeef, "encoding not changed\n");
1359     ok(size == sizeof(encoding), "size=%d\n", size);
1360
1361     size = encoding = 0xdeadbeef;
1362     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1363                                  sizeof(encoding)+1, &size, 0);
1364     ok(hres == S_OK, "UrlMkGetSessionOption failed: %08x\n", hres);
1365     ok(encoding != 0xdeadbeef, "encoding not changed\n");
1366     ok(size == sizeof(encoding), "size=%d\n", size);
1367
1368     size = encoding = 0xdeadbeef;
1369     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1370                                  sizeof(encoding)-1, &size, 0);
1371     ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1372     ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1373     ok(size == 0xdeadbeef, "size=%d\n", size);
1374
1375     size = encoding = 0xdeadbeef;
1376     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, NULL,
1377                                  sizeof(encoding)-1, &size, 0);
1378     ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1379     ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1380     ok(size == 0xdeadbeef, "size=%d\n", size);
1381
1382     encoding = 0xdeadbeef;
1383     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1384                                  sizeof(encoding)-1, NULL, 0);
1385     ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1386     ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1387 }
1388
1389 static void test_ObtainUserAgentString(void)
1390 {
1391     static const CHAR expected[] = "Mozilla/4.0 (compatible; MSIE ";
1392     static CHAR str[3];
1393     LPSTR str2 = NULL;
1394     HRESULT hres;
1395     DWORD size, saved;
1396
1397     hres = ObtainUserAgentString(0, NULL, NULL);
1398     ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1399
1400     size = 100;
1401     hres = ObtainUserAgentString(0, NULL, &size);
1402     ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1403     ok(size == 100, "size=%d, expected %d\n", size, 100);
1404
1405     size = 0;
1406     hres = ObtainUserAgentString(0, str, &size);
1407     ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1408     ok(size > 0, "size=%d, expected non-zero\n", size);
1409
1410     size = 2;
1411     str[0] = 'a';
1412     hres = ObtainUserAgentString(0, str, &size);
1413     ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1414     ok(size > 0, "size=%d, expected non-zero\n", size);
1415     ok(str[0] == 'a', "str[0]=%c, expected 'a'\n", str[0]);
1416
1417     size = 0;
1418     hres = ObtainUserAgentString(1, str, &size);
1419     ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1420     ok(size > 0, "size=%d, expected non-zero\n", size);
1421
1422     str2 = HeapAlloc(GetProcessHeap(), 0, (size+20)*sizeof(CHAR));
1423     if (!str2)
1424     {
1425         skip("skipping rest of ObtainUserAgent tests, out of memory\n");
1426     }
1427     else
1428     {
1429         saved = size;
1430         hres = ObtainUserAgentString(0, str2, &size);
1431         ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres);
1432         ok(size == saved, "size=%d, expected %d\n", size, saved);
1433         ok(strlen(expected) <= strlen(str2) &&
1434            !memcmp(expected, str2, strlen(expected)*sizeof(CHAR)),
1435            "user agent was \"%s\", expected to start with \"%s\"\n",
1436            str2, expected);
1437
1438         size = saved+10;
1439         hres = ObtainUserAgentString(0, str2, &size);
1440         ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres);
1441         ok(size == saved, "size=%d, expected %d\n", size, saved);
1442     }
1443     HeapFree(GetProcessHeap(), 0, str2);
1444 }
1445
1446 static void test_MkParseDisplayNameEx(void)
1447 {
1448     IMoniker *mon = NULL;
1449     LPWSTR name;
1450     DWORD issys;
1451     ULONG eaten = 0;
1452     IBindCtx *bctx;
1453     HRESULT hres;
1454
1455     static const WCHAR clsid_nameW[] = {'c','l','s','i','d',':',
1456             '2','0','D','0','4','F','E','0','-','3','A','E','A','-','1','0','6','9','-','A','2','D','8',
1457             '-','0','8','0','0','2','B','3','0','3','0','9','D',':',0};
1458
1459     CreateBindCtx(0, &bctx);
1460
1461     hres = MkParseDisplayNameEx(bctx, url9, &eaten, &mon);
1462     ok(hres == S_OK, "MkParseDisplayNameEx failed: %08x\n", hres);
1463     ok(eaten == sizeof(url9)/sizeof(WCHAR)-1, "eaten=%d\n", eaten);
1464     ok(mon != NULL, "mon == NULL\n");
1465
1466     hres = IMoniker_GetDisplayName(mon, NULL, 0, &name);
1467     ok(hres == S_OK, "GetDiasplayName failed: %08x\n", hres);
1468     ok(!lstrcmpW(name, url9), "wrong display name %s\n", debugstr_w(name));
1469     CoTaskMemFree(name);
1470
1471     hres = IMoniker_IsSystemMoniker(mon, &issys);
1472     ok(hres == S_OK, "IsSystemMoniker failed: %08x\n", hres);
1473     ok(issys == MKSYS_URLMONIKER, "issys=%x\n", issys);
1474
1475     IMoniker_Release(mon);
1476
1477     hres = MkParseDisplayNameEx(bctx, clsid_nameW, &eaten, &mon);
1478     ok(hres == S_OK, "MkParseDisplayNameEx failed: %08x\n", hres);
1479     ok(eaten == sizeof(clsid_nameW)/sizeof(WCHAR)-1, "eaten=%d\n", eaten);
1480     ok(mon != NULL, "mon == NULL\n");
1481
1482     hres = IMoniker_IsSystemMoniker(mon, &issys);
1483     ok(hres == S_OK, "IsSystemMoniker failed: %08x\n", hres);
1484     ok(issys == MKSYS_CLASSMONIKER, "issys=%x\n", issys);
1485
1486     IMoniker_Release(mon);
1487
1488     hres = MkParseDisplayNameEx(bctx, url8, &eaten, &mon);
1489     ok(FAILED(hres), "MkParseDisplayNameEx succeeded: %08x\n", hres);
1490
1491     IBindCtx_Release(bctx);
1492 }
1493
1494 START_TEST(misc)
1495 {
1496     OleInitialize(NULL);
1497
1498     register_protocols();
1499
1500     test_CreateFormatEnum();
1501     test_RegisterFormatEnumerator();
1502     test_CoInternetParseUrl();
1503     test_CoInternetCompareUrl();
1504     test_CoInternetQueryInfo();
1505     test_FindMimeFromData();
1506     test_SecurityManager();
1507     test_polices();
1508     test_ZoneManager();
1509     test_NameSpace();
1510     test_MimeFilter();
1511     test_ReleaseBindInfo();
1512     test_CopyStgMedium();
1513     test_UrlMkGetSessionOption();
1514     test_ObtainUserAgentString();
1515     test_MkParseDisplayNameEx();
1516
1517     OleUninitialize();
1518 }