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