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