git.oblomov.eu Git - wine/blob - dlls/urlmon/tests/misc.c

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