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