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