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