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
  22 #include <wine/test.h>
  23 #include <stdarg.h>
  24 #include <stddef.h>
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "ole2.h"
  29 #include "urlmon.h"
  30
  31 #include "initguid.h"
  32
  33 DEFINE_GUID(CLSID_AboutProtocol, 0x3050F406, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
  34
  35 #define DEFINE_EXPECT(func) \
  36     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  37
  38 #define SET_EXPECT(func) \
  39     expect_ ## func = TRUE
  40
  41 #define CHECK_EXPECT(func) \
  42     do { \
  43         ok(expect_ ##func, "unexpected call " #func "\n"); \
  44         expect_ ## func = FALSE; \
  45         called_ ## func = TRUE; \
  46     }while(0)
  47
  48 #define CHECK_EXPECT2(func) \
  49     do { \
  50         ok(expect_ ##func, "unexpected call " #func "\n"); \
  51         called_ ## func = TRUE; \
  52     }while(0)
  53
  54 #define CHECK_CALLED(func) \
  55     do { \
  56         ok(called_ ## func, "expected " #func "\n"); \
  57         expect_ ## func = called_ ## func = FALSE; \
  58     }while(0)
  59
  60 DEFINE_EXPECT(ParseUrl);
  61 DEFINE_EXPECT(QI_IInternetProtocolInfo);
  62 DEFINE_EXPECT(CreateInstance);
  63 DEFINE_EXPECT(unk_Release);
  64
  65 static const char *debugstr_w(LPCWSTR str)
  66 {
  67     static char buf[1024];
  68     WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL);
  69     return buf;
  70 }
  71
  72 static void test_CreateFormatEnum(void)
  73 {
  74     IEnumFORMATETC *fenum = NULL, *fenum2 = NULL;
  75     FORMATETC fetc[5];
  76     ULONG ul;
  77     HRESULT hres;
  78
  79     static DVTARGETDEVICE dev = {sizeof(dev),0,0,0,0,{0}};
  80     static FORMATETC formatetc[] = {
  81         {0,&dev,0,0,0},
  82         {0,&dev,0,1,0},
  83         {0,NULL,0,2,0},
  84         {0,NULL,0,3,0},
  85         {0,NULL,0,4,0}
  86     };
  87
  88     hres = CreateFormatEnumerator(0, formatetc, &fenum);
  89     ok(hres == E_FAIL, "CreateFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
  90     hres = CreateFormatEnumerator(0, formatetc, NULL);
  91     ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
  92     hres = CreateFormatEnumerator(5, formatetc, NULL);
  93     ok(hres == E_INVALIDARG, "CreateFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
  94
  95
  96     hres = CreateFormatEnumerator(5, formatetc, &fenum);
  97     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
  98     if(FAILED(hres))
  99         return;
 100
 101     hres = IEnumFORMATETC_Next(fenum, 2, NULL, &ul);
 102     ok(hres == E_INVALIDARG, "Next failed: %08x, expected E_INVALIDARG\n", hres);
 103     ul = 100;
 104     hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul);
 105     ok(hres == S_OK, "Next failed: %08x\n", hres);
 106     ok(ul == 0, "ul=%d, expected 0\n", ul);
 107
 108     hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul);
 109     ok(hres == S_OK, "Next failed: %08x\n", hres);
 110     ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex);
 111     ok(fetc[1].lindex == 1, "fetc[1].lindex=%d, expected 1\n", fetc[1].lindex);
 112     ok(fetc[0].ptd == &dev, "fetc[0].ptd=%p, expected %p\n", fetc[0].ptd, &dev);
 113     ok(ul == 2, "ul=%d, expected 2\n", ul);
 114
 115     hres = IEnumFORMATETC_Skip(fenum, 1);
 116     ok(hres == S_OK, "Skip failed: %08x\n", hres);
 117
 118     hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul);
 119     ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres);
 120     ok(fetc[0].lindex == 3, "fetc[0].lindex=%d, expected 3\n", fetc[0].lindex);
 121     ok(fetc[1].lindex == 4, "fetc[1].lindex=%d, expected 4\n", fetc[1].lindex);
 122     ok(fetc[0].ptd == NULL, "fetc[0].ptd=%p, expected NULL\n", fetc[0].ptd);
 123     ok(ul == 2, "ul=%d, expected 2\n", ul);
 124
 125     hres = IEnumFORMATETC_Next(fenum, 4, fetc, &ul);
 126     ok(hres == S_FALSE, "Next failed: %08x, expected S_FALSE\n", hres);
 127     ok(ul == 0, "ul=%d, expected 0\n", ul);
 128     ul = 100;
 129     hres = IEnumFORMATETC_Next(fenum, 0, fetc, &ul);
 130     ok(hres == S_OK, "Next failed: %08x\n", hres);
 131     ok(ul == 0, "ul=%d, expected 0\n", ul);
 132
 133     hres = IEnumFORMATETC_Skip(fenum, 3);
 134     ok(hres == S_FALSE, "Skip failed: %08x, expected S_FALSE\n", hres);
 135
 136     hres = IEnumFORMATETC_Reset(fenum);
 137     ok(hres == S_OK, "Reset failed: %08x\n", hres);
 138
 139     hres = IEnumFORMATETC_Next(fenum, 5, fetc, NULL);
 140     ok(hres == S_OK, "Next failed: %08x\n", hres);
 141     ok(fetc[0].lindex == 0, "fetc[0].lindex=%d, expected 0\n", fetc[0].lindex);
 142
 143     hres = IEnumFORMATETC_Reset(fenum);
 144     ok(hres == S_OK, "Reset failed: %08x\n", hres);
 145
 146     hres = IEnumFORMATETC_Skip(fenum, 2);
 147     ok(hres == S_OK, "Skip failed: %08x\n", hres);
 148
 149     hres = IEnumFORMATETC_Clone(fenum, NULL);
 150     ok(hres == E_INVALIDARG, "Clone failed: %08x, expected E_INVALIDARG\n", hres);
 151
 152     hres = IEnumFORMATETC_Clone(fenum, &fenum2);
 153     ok(hres == S_OK, "Clone failed: %08x\n", hres);
 154
 155     if(SUCCEEDED(hres)) {
 156         ok(fenum != fenum2, "fenum == fenum2\n");
 157
 158         hres = IEnumFORMATETC_Next(fenum2, 2, fetc, &ul);
 159         ok(hres == S_OK, "Next failed: %08x\n", hres);
 160         ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex);
 161
 162         IEnumFORMATETC_Release(fenum2);
 163     }
 164
 165     hres = IEnumFORMATETC_Next(fenum, 2, fetc, &ul);
 166     ok(hres == S_OK, "Next failed: %08x\n", hres);
 167     ok(fetc[0].lindex == 2, "fetc[0].lindex=%d, expected 2\n", fetc[0].lindex);
 168
 169     hres = IEnumFORMATETC_Skip(fenum, 1);
 170     ok(hres == S_OK, "Skip failed: %08x\n", hres);
 171
 172     IEnumFORMATETC_Release(fenum);
 173 }
 174
 175 static void test_RegisterFormatEnumerator(void)
 176 {
 177     IBindCtx *bctx = NULL;
 178     IEnumFORMATETC *format = NULL, *format2 = NULL;
 179     IUnknown *unk = NULL;
 180     HRESULT hres;
 181
 182     static FORMATETC formatetc = {0,NULL,0,0,0};
 183     static WCHAR wszEnumFORMATETC[] =
 184         {'_','E','n','u','m','F','O','R','M','A','T','E','T','C','_',0};
 185
 186     CreateBindCtx(0, &bctx);
 187
 188     hres = CreateFormatEnumerator(1, &formatetc, &format);
 189     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
 190     if(FAILED(hres))
 191         return;
 192
 193     hres = RegisterFormatEnumerator(NULL, format, 0);
 194     ok(hres == E_INVALIDARG,
 195             "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
 196     hres = RegisterFormatEnumerator(bctx, NULL, 0);
 197     ok(hres == E_INVALIDARG,
 198             "RegisterFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
 199
 200     hres = RegisterFormatEnumerator(bctx, format, 0);
 201     ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
 202
 203     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
 204     ok(hres == S_OK, "GetObjectParam failed: %08x\n", hres);
 205     ok(unk == (IUnknown*)format, "unk != format\n");
 206
 207     hres = RevokeFormatEnumerator(NULL, format);
 208     ok(hres == E_INVALIDARG,
 209             "RevokeFormatEnumerator failed: %08x, expected E_INVALIDARG\n", hres);
 210
 211     hres = RevokeFormatEnumerator(bctx, format);
 212     ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
 213
 214     hres = RevokeFormatEnumerator(bctx, format);
 215     ok(hres == E_FAIL, "RevokeFormatEnumerator failed: %08x, expected E_FAIL\n", hres);
 216
 217     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
 218     ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
 219
 220     hres = RegisterFormatEnumerator(bctx, format, 0);
 221     ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
 222
 223     hres = CreateFormatEnumerator(1, &formatetc, &format2);
 224     ok(hres == S_OK, "CreateFormatEnumerator failed: %08x\n", hres);
 225
 226     if(SUCCEEDED(hres)) {
 227         hres = RevokeFormatEnumerator(bctx, format);
 228         ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
 229
 230         IEnumFORMATETC_Release(format2);
 231     }
 232
 233     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
 234     ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
 235
 236     IEnumFORMATETC_Release(format);
 237
 238     hres = RegisterFormatEnumerator(bctx, format, 0);
 239     ok(hres == S_OK, "RegisterFormatEnumerator failed: %08x\n", hres);
 240     hres = RevokeFormatEnumerator(bctx, NULL);
 241     ok(hres == S_OK, "RevokeFormatEnumerator failed: %08x\n", hres);
 242     hres = IBindCtx_GetObjectParam(bctx, wszEnumFORMATETC, &unk);
 243     ok(hres == E_FAIL, "GetObjectParam failed: %08x, expected E_FAIL\n", hres);
 244
 245     IEnumFORMATETC_Release(format);
 246     IBindCtx_Release(bctx);
 247 }
 248
 249 static const WCHAR url1[] = {'r','e','s',':','/','/','m','s','h','t','m','l','.','d','l','l',
 250         '/','b','l','a','n','k','.','h','t','m',0};
 251 static const WCHAR url2[] = {'i','n','d','e','x','.','h','t','m',0};
 252 static const WCHAR url3[] = {'f','i','l','e',':','c',':','\\','I','n','d','e','x','.','h','t','m',0};
 253 static const WCHAR url4[] = {'f','i','l','e',':','s','o','m','e','%','2','0','f','i','l','e',
 254         '%','2','e','j','p','g',0};
 255 static const WCHAR url5[] = {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q',
 256         '.','o','r','g',0};
 257 static const WCHAR url6[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
 258 static const WCHAR url7[] = {'f','t','p',':','/','/','w','i','n','e','h','q','.','o','r','g','/',
 259         'f','i','l','e','.','t','e','s','t',0};
 260 static const WCHAR url8[] = {'t','e','s','t',':','1','2','3','a','b','c',0};
 261
 262
 263 static const WCHAR url4e[] = {'f','i','l','e',':','s','o','m','e',' ','f','i','l','e',
 264         '.','j','p','g',0};
 265
 266 static const WCHAR path3[] = {'c',':','\\','I','n','d','e','x','.','h','t','m',0};
 267 static const WCHAR path4[] = {'s','o','m','e',' ','f','i','l','e','.','j','p','g',0};
 268
 269 static const WCHAR wszRes[] = {'r','e','s',0};
 270 static const WCHAR wszFile[] = {'f','i','l','e',0};
 271 static const WCHAR wszHttp[] = {'h','t','t','p',0};
 272 static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
 273 static const WCHAR wszEmpty[] = {0};
 274
 275 struct parse_test {
 276     LPCWSTR url;
 277     HRESULT secur_hres;
 278     LPCWSTR encoded_url;
 279     HRESULT path_hres;
 280     LPCWSTR path;
 281     LPCWSTR schema;
 282 };
 283
 284 static const struct parse_test parse_tests[] = {
 285     {url1, S_OK,   url1,  E_INVALIDARG, NULL, wszRes},
 286     {url2, E_FAIL, url2,  E_INVALIDARG, NULL, wszEmpty},
 287     {url3, E_FAIL, url3,  S_OK, path3,        wszFile},
 288     {url4, E_FAIL, url4e, S_OK, path4,        wszFile},
 289     {url5, E_FAIL, url5,  E_INVALIDARG, NULL, wszHttp},
 290     {url6, S_OK,   url6,  E_INVALIDARG, NULL, wszAbout}
 291 };
 292
 293 static void test_CoInternetParseUrl(void)
 294 {
 295     HRESULT hres;
 296     DWORD size;
 297     int i;
 298
 299     static WCHAR buf[4096];
 300
 301     memset(buf, 0xf0, sizeof(buf));
 302     hres = CoInternetParseUrl(parse_tests[0].url, PARSE_SCHEMA, 0, buf,
 303             3, &size, 0);
 304     ok(hres == E_POINTER, "schema failed: %08x, expected E_POINTER\n", hres);
 305
 306     for(i=0; i < sizeof(parse_tests)/sizeof(parse_tests[0]); i++) {
 307         memset(buf, 0xf0, sizeof(buf));
 308         hres = CoInternetParseUrl(parse_tests[i].url, PARSE_SECURITY_URL, 0, buf,
 309                 sizeof(buf)/sizeof(WCHAR), &size, 0);
 310         ok(hres == parse_tests[i].secur_hres, "[%d] security url failed: %08x, expected %08x\n",
 311                 i, hres, parse_tests[i].secur_hres);
 312
 313         memset(buf, 0xf0, sizeof(buf));
 314         hres = CoInternetParseUrl(parse_tests[i].url, PARSE_ENCODE, 0, buf,
 315                 sizeof(buf)/sizeof(WCHAR), &size, 0);
 316         ok(hres == S_OK, "[%d] encoding failed: %08x\n", i, hres);
 317         ok(size == lstrlenW(parse_tests[i].encoded_url), "[%d] wrong size\n", i);
 318         ok(!lstrcmpW(parse_tests[i].encoded_url, buf), "[%d] wrong encoded url\n", i);
 319
 320         memset(buf, 0xf0, sizeof(buf));
 321         hres = CoInternetParseUrl(parse_tests[i].url, PARSE_PATH_FROM_URL, 0, buf,
 322                 sizeof(buf)/sizeof(WCHAR), &size, 0);
 323         ok(hres == parse_tests[i].path_hres, "[%d] path failed: %08x, expected %08x\n",
 324                 i, hres, parse_tests[i].path_hres);
 325         if(parse_tests[i].path) {
 326             ok(size == lstrlenW(parse_tests[i].path), "[%d] wrong size\n", i);
 327             ok(!lstrcmpW(parse_tests[i].path, buf), "[%d] wrong path\n", i);
 328         }
 329
 330         memset(buf, 0xf0, sizeof(buf));
 331         hres = CoInternetParseUrl(parse_tests[i].url, PARSE_SCHEMA, 0, buf,
 332                 sizeof(buf)/sizeof(WCHAR), &size, 0);
 333         ok(hres == S_OK, "[%d] schema failed: %08x\n", i, hres);
 334         ok(size == lstrlenW(parse_tests[i].schema), "[%d] wrong size\n", i);
 335         ok(!lstrcmpW(parse_tests[i].schema, buf), "[%d] wrong schema\n", i);
 336     }
 337 }
 338
 339 static void test_CoInternetCompareUrl(void)
 340 {
 341     HRESULT hres;
 342
 343     hres = CoInternetCompareUrl(url1, url1, 0);
 344     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
 345
 346     hres = CoInternetCompareUrl(url1, url3, 0);
 347     ok(hres == S_FALSE, "CoInternetParseUrl failed: %08x\n", hres);
 348
 349     hres = CoInternetCompareUrl(url3, url1, 0);
 350     ok(hres == S_FALSE, "CoInternetParseUrl failed: %08x\n", hres);
 351 }
 352
 353 static const struct {
 354     LPCWSTR url;
 355     DWORD uses_net;
 356 } query_info_tests[] = {
 357     {url1, 0},
 358     {url2, 0},
 359     {url3, 0},
 360     {url4, 0},
 361     {url5, 0},
 362     {url6, 0},
 363     {url7, 0},
 364     {url8, 0}
 365 };
 366
 367 static void test_CoInternetQueryInfo(void)
 368 {
 369     BYTE buf[100];
 370     DWORD cb, i;
 371     HRESULT hres;
 372
 373     for(i=0; i < sizeof(query_info_tests)/sizeof(query_info_tests[0]); i++) {
 374         cb = 0xdeadbeef;
 375         memset(buf, '?', sizeof(buf));
 376         hres = CoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, sizeof(buf), &cb, 0);
 377         ok(hres == S_OK, "[%d] CoInternetQueryInfo failed: %08x\n", i, hres);
 378         ok(cb == sizeof(DWORD), "[%d] cb = %d\n", i, cb);
 379         ok(*(DWORD*)buf == query_info_tests[i].uses_net, "[%d] ret %x, expected %x\n",
 380            i, *(DWORD*)buf, query_info_tests[i].uses_net);
 381
 382         hres = CoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, 3, &cb, 0);
 383         ok(hres == E_FAIL, "[%d] CoInternetQueryInfo failed: %08x, expected E_FAIL\n", i, hres);
 384         hres = CoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, NULL, sizeof(buf), &cb, 0);
 385         ok(hres == E_FAIL, "[%d] CoInternetQueryInfo failed: %08x, expected E_FAIL\n", i, hres);
 386
 387         memset(buf, '?', sizeof(buf));
 388         hres = CoInternetQueryInfo(query_info_tests[0].url, QUERY_USES_NETWORK, 0, buf, sizeof(buf), NULL, 0);
 389         ok(hres == S_OK, "[%d] CoInternetQueryInfo failed: %08x\n", i, hres);
 390         ok(*(DWORD*)buf == query_info_tests[i].uses_net, "[%d] ret %x, expected %x\n",
 391            i, *(DWORD*)buf, query_info_tests[i].uses_net);
 392     }
 393 }
 394
 395 static const WCHAR mimeTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
 396 static const WCHAR mimeTextPlain[] = {'t','e','x','t','/','p','l','a','i','n',0};
 397 static const WCHAR mimeTextRichtext[] = {'t','e','x','t','/','r','i','c','h','t','e','x','t',0};
 398 static const WCHAR mimeAppOctetStream[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
 399     'o','c','t','e','t','-','s','t','r','e','a','m',0};
 400 static const WCHAR mimeImagePjpeg[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
 401 static const WCHAR mimeImageGif[] = {'i','m','a','g','e','/','g','i','f',0};
 402 static const WCHAR mimeImageBmp[] = {'i','m','a','g','e','/','b','m','p',0};
 403 static const WCHAR mimeImageXPng[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
 404 static const WCHAR mimeImageTiff[] = {'i','m','a','g','e','/','t','i','f','f',0};
 405 static const WCHAR mimeVideoAvi[] = {'v','i','d','e','o','/','a','v','i',0};
 406 static const WCHAR mimeVideoMpeg[] = {'v','i','d','e','o','/','m','p','e','g',0};
 407 static const WCHAR mimeAppPostscript[] =
 408     {'a','p','p','l','i','c','a','t','i','o','n','/','p','o','s','t','s','c','r','i','p','t',0};
 409 static const WCHAR mimeAppXCompressed[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
 410                                     'x','-','c','o','m','p','r','e','s','s','e','d',0};
 411 static const WCHAR mimeAppXZip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
 412                                     'x','-','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
 413 static const WCHAR mimeAppXGzip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
 414                                     'x','-','g','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
 415 static const WCHAR mimeAppJava[] = {'a','p','p','l','i','c','a','t','i','o','n','/','j','a','v','a',0};
 416 static const WCHAR mimeAppPdf[] = {'a','p','p','l','i','c','a','t','i','o','n','/','p','d','f',0};
 417 static const WCHAR mimeAppXMSDownload[] =
 418     {'a','p','p','l','i','c','a','t','i','o','n','/','x','-','m','s','d','o','w','n','l','o','a','d',0};
 419 static const WCHAR mimeAudioWav[] = {'a','u','d','i','o','/','w','a','v',0};
 420
 421 static const struct {
 422     LPCWSTR url;
 423     LPCWSTR mime;
 424     HRESULT hres;
 425 } mime_tests[] = {
 426     {url1, mimeTextHtml, S_OK},
 427     {url2, mimeTextHtml, S_OK},
 428     {url3, mimeTextHtml, S_OK},
 429     {url4, NULL, E_FAIL},
 430     {url5, NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)},
 431     {url6, NULL, E_FAIL},
 432     {url7, NULL, __HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND)}
 433 };
 434
 435 static BYTE data1[] = "test data\n";
 436 static BYTE data2[] = {31,'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0};
 437 static BYTE data3[] = {0,0,0};
 438 static BYTE data4[] = {'t','e','s',0xfa,'t',' ','d','a','t','a','\n',0,0};
 439 static BYTE data5[] = {0xa,0xa,0xa,'x',32,'x',0};
 440 static BYTE data6[] = {0xfa,0xfa,0xfa,0xfa,'\n','\r','\t','x','x','x',1};
 441 static BYTE data7[] = "<html>blahblah";
 442 static BYTE data8[] = {'t','e','s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0};
 443 static BYTE data9[] = {'t','e',0,'s',0xfa,'t',' ','<','h','t','m','l','>','d','a','t','a','\n',0,0};
 444 static BYTE data10[] = "<HtmL>blahblah";
 445 static BYTE data11[] = "blah<HTML>blahblah";
 446 static BYTE data12[] = "blah<HTMLblahblah";
 447 static BYTE data13[] = "blahHTML>blahblah";
 448 static BYTE data14[] = "blah<HTMblahblah";
 449 static BYTE data15[] = {0xff,0xd8};
 450 static BYTE data16[] = {0xff,0xd8,'h'};
 451 static BYTE data17[] = {0,0xff,0xd8};
 452 static BYTE data18[] = {0xff,0xd8,'<','h','t','m','l','>'};
 453 static BYTE data19[] = {'G','I','F','8','7','a'};
 454 static BYTE data20[] = {'G','I','F','8','9','a'};
 455 static BYTE data21[] = {'G','I','F','8','7'};
 456 static BYTE data22[] = {'G','i','F','8','7','a'};
 457 static BYTE data23[] = {'G','i','F','8','8','a'};
 458 static BYTE data24[] = {'g','i','f','8','7','a'};
 459 static BYTE data25[] = {'G','i','F','8','7','A'};
 460 static BYTE data26[] = {'G','i','F','8','7','a','<','h','t','m','l','>'};
 461 static BYTE data27[] = {0x30,'G','i','F','8','7','A'};
 462 static BYTE data28[] = {0x42,0x4d,0x6e,0x42,0x1c,0x00,0x00,0x00,0x00,0x00,0x36,0x00,0x00,0x00};
 463 static BYTE data29[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x','x'};
 464 static BYTE data30[] = {0x42,0x4d,'x','x','x','x',0x00,0x01,0x00,0x00,'x','x','x','x'};
 465 static BYTE data31[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'<','h','t','m','l','>'};
 466 static BYTE data32[] = {0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'};
 467 static BYTE data33[] = {0x00,0x42,0x4d,'x','x','x','x',0x00,0x00,0x00,0x00,'x','x','x'};
 468 static BYTE data34[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'};
 469 static BYTE data35[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x','x','x','x',0};
 470 static BYTE data36[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,'x','x'};
 471 static BYTE data37[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'<','h','t','m','l','>'};
 472 static BYTE data38[] = {0x00,0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a,'x'};
 473 static BYTE data39[] = {0x4d,0x4d,0x00,0x2a};
 474 static BYTE data40[] = {0x4d,0x4d,0x00,0x2a,'<','h','t','m','l','>',0};
 475 static BYTE data41[] = {0x4d,0x4d,0xff};
 476 static BYTE data42[] = {0x4d,0x4d};
 477 static BYTE data43[] = {0x00,0x4d,0x4d,0x00};
 478 static BYTE data44[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff};
 479 static BYTE data45[] = {'R','I','F','f',0xff,0xff,0xff,0xff,'A','V','I',0x20,0xff};
 480 static BYTE data46[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20};
 481 static BYTE data47[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x21,0xff};
 482 static BYTE data48[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'A','V','I',0x20,'<','h','t','m','l','>'};
 483 static BYTE data49[] = {'R','I','F','F',0x0f,0x0f,0xf0,0xf0,'A','V','I',0x20,0xf0,0x00};
 484 static BYTE data50[] = {0x00,0x00,0x01,0xb3,0xff};
 485 static BYTE data51[] = {0x00,0x00,0x01,0xba,0xff};
 486 static BYTE data52[] = {0x00,0x00,0x01,0xb8,0xff};
 487 static BYTE data53[] = {0x00,0x00,0x01,0xba};
 488 static BYTE data54[] = {0x00,0x00,0x01,0xba,'<','h','t','m','l','>'};
 489 static BYTE data55[] = {0x1f,0x8b,'x'};
 490 static BYTE data56[] = {0x1f};
 491 static BYTE data57[] = {0x1f,0x8b,'<','h','t','m','l','>','t','e','s','t',0};
 492 static BYTE data58[] = {0x1f,0x8b};
 493 static BYTE data59[] = {0x50,0x4b,'x'};
 494 static BYTE data60[] = {0x50,0x4b};
 495 static BYTE data61[] = {0x50,0x4b,'<','h','t','m','l','>',0};
 496 static BYTE data62[] = {0xca,0xfe,0xba,0xbe,'x'};
 497 static BYTE data63[] = {0xca,0xfe,0xba,0xbe};
 498 static BYTE data64[] = {0xca,0xfe,0xba,0xbe,'<','h','t','m','l','>',0};
 499 static BYTE data65[] = {0x25,0x50,0x44,0x46,'x'};
 500 static BYTE data66[] = {0x25,0x50,0x44,0x46};
 501 static BYTE data67[] = {0x25,0x50,0x44,0x46,'x','<','h','t','m','l','>'};
 502 static BYTE data68[] = {'M','Z','x'};
 503 static BYTE data69[] = {'M','Z'};
 504 static BYTE data70[] = {'M','Z','<','h','t','m','l','>',0xff};
 505 static BYTE data71[] = {'{','\\','r','t','f',0};
 506 static BYTE data72[] = {'{','\\','r','t','f'};
 507 static BYTE data73[] = {' ','{','\\','r','t','f',' '};
 508 static BYTE data74[] = {'{','\\','r','t','f','<','h','t','m','l','>',' '};
 509 static BYTE data75[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V','E',0xff};
 510 static BYTE data76[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V','E'};
 511 static BYTE data77[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'W','A','V',0xff,0xff};
 512 static BYTE data78[] = {'R','I','F','F',0xff,0xff,0xff,0xff,'<','h','t','m','l','>',0xff};
 513 static BYTE data79[] = {'%','!',0xff};
 514 static BYTE data80[] = {'%','!'};
 515 static BYTE data81[] = {'%','!','P','S','<','h','t','m','l','>'};
 516
 517 static const struct {
 518     BYTE *data;
 519     DWORD size;
 520     LPCWSTR mime;
 521 } mime_tests2[] = {
 522     {data1, sizeof(data1), mimeTextPlain},
 523     {data2, sizeof(data2), mimeAppOctetStream},
 524     {data3, sizeof(data3), mimeAppOctetStream},
 525     {data4, sizeof(data4), mimeAppOctetStream},
 526     {data5, sizeof(data5), mimeTextPlain},
 527     {data6, sizeof(data6), mimeTextPlain},
 528     {data7, sizeof(data7), mimeTextHtml},
 529     {data8, sizeof(data8), mimeTextHtml},
 530     {data9, sizeof(data9), mimeTextHtml},
 531     {data10, sizeof(data10), mimeTextHtml},
 532     {data11, sizeof(data11), mimeTextHtml},
 533     {data12, sizeof(data12), mimeTextHtml},
 534     {data13, sizeof(data13), mimeTextPlain},
 535     {data14, sizeof(data14), mimeTextPlain},
 536     {data15, sizeof(data15), mimeTextPlain},
 537     {data16, sizeof(data16), mimeImagePjpeg},
 538     {data17, sizeof(data17), mimeAppOctetStream},
 539     {data18, sizeof(data18), mimeTextHtml},
 540     {data19, sizeof(data19), mimeImageGif},
 541     {data20, sizeof(data20), mimeImageGif},
 542     {data21, sizeof(data21), mimeTextPlain},
 543     {data22, sizeof(data22), mimeImageGif},
 544     {data23, sizeof(data23), mimeTextPlain},
 545     {data24, sizeof(data24), mimeImageGif},
 546     {data25, sizeof(data25), mimeImageGif},
 547     {data26, sizeof(data26), mimeTextHtml},
 548     {data27, sizeof(data27), mimeTextPlain},
 549     {data28, sizeof(data28), mimeImageBmp},
 550     {data29, sizeof(data29), mimeImageBmp},
 551     {data30, sizeof(data30), mimeAppOctetStream},
 552     {data31, sizeof(data31), mimeTextHtml},
 553     {data32, sizeof(data32), mimeAppOctetStream},
 554     {data33, sizeof(data33), mimeAppOctetStream},
 555     {data34, sizeof(data34), mimeImageXPng},
 556     {data35, sizeof(data35), mimeImageXPng},
 557     {data36, sizeof(data36), mimeAppOctetStream},
 558     {data37, sizeof(data37), mimeTextHtml},
 559     {data38, sizeof(data38), mimeAppOctetStream},
 560     {data39, sizeof(data39), mimeImageTiff},
 561     {data40, sizeof(data40), mimeTextHtml},
 562     {data41, sizeof(data41), mimeImageTiff},
 563     {data42, sizeof(data42), mimeTextPlain},
 564     {data43, sizeof(data43), mimeAppOctetStream},
 565     {data44, sizeof(data44), mimeVideoAvi},
 566     {data45, sizeof(data45), mimeTextPlain},
 567     {data46, sizeof(data46), mimeTextPlain},
 568     {data47, sizeof(data47), mimeTextPlain},
 569     {data48, sizeof(data48), mimeTextHtml},
 570     {data49, sizeof(data49), mimeVideoAvi},
 571     {data50, sizeof(data50), mimeVideoMpeg},
 572     {data51, sizeof(data51), mimeVideoMpeg},
 573     {data52, sizeof(data52), mimeAppOctetStream},
 574     {data53, sizeof(data53), mimeAppOctetStream},
 575     {data54, sizeof(data54), mimeTextHtml},
 576     {data55, sizeof(data55), mimeAppXGzip},
 577     {data56, sizeof(data56), mimeTextPlain},
 578     {data57, sizeof(data57), mimeTextHtml},
 579     {data58, sizeof(data58), mimeAppOctetStream},
 580     {data59, sizeof(data59), mimeAppXZip},
 581     {data60, sizeof(data60), mimeTextPlain},
 582     {data61, sizeof(data61), mimeTextHtml},
 583     {data62, sizeof(data62), mimeAppJava},
 584     {data63, sizeof(data63), mimeTextPlain},
 585     {data64, sizeof(data64), mimeTextHtml},
 586     {data65, sizeof(data65), mimeAppPdf},
 587     {data66, sizeof(data66), mimeTextPlain},
 588     {data67, sizeof(data67), mimeTextHtml},
 589     {data68, sizeof(data68), mimeAppXMSDownload},
 590     {data69, sizeof(data69), mimeTextPlain},
 591     {data70, sizeof(data70), mimeTextHtml},
 592     {data71, sizeof(data71), mimeTextRichtext},
 593     {data72, sizeof(data72), mimeTextPlain},
 594     {data73, sizeof(data73), mimeTextPlain},
 595     {data74, sizeof(data74), mimeTextHtml},
 596     {data75, sizeof(data75), mimeAudioWav},
 597     {data76, sizeof(data76), mimeTextPlain},
 598     {data77, sizeof(data77), mimeTextPlain},
 599     {data78, sizeof(data78), mimeTextHtml},
 600     {data79, sizeof(data79), mimeAppPostscript},
 601     {data80, sizeof(data80), mimeTextPlain},
 602     {data81, sizeof(data81), mimeTextHtml}
 603 };
 604
 605 static void test_FindMimeFromData(void)
 606 {
 607     HRESULT hres;
 608     LPWSTR mime;
 609     int i;
 610
 611     for(i=0; i<sizeof(mime_tests)/sizeof(mime_tests[0]); i++) {
 612         mime = (LPWSTR)0xf0f0f0f0;
 613         hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, NULL, 0, &mime, 0);
 614         if(mime_tests[i].mime) {
 615             ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
 616             ok(!lstrcmpW(mime, mime_tests[i].mime), "[%d] wrong mime\n", i);
 617             CoTaskMemFree(mime);
 618         }else {
 619             ok(hres == E_FAIL || hres == mime_tests[i].hres,
 620                "[%d] FindMimeFromData failed: %08x, expected %08x\n",
 621                i, hres, mime_tests[i].hres);
 622             ok(mime == (LPWSTR)0xf0f0f0f0, "[%d] mime != 0xf0f0f0f0\n", i);
 623         }
 624
 625         mime = (LPWSTR)0xf0f0f0f0;
 626         hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, mimeTextPlain, 0, &mime, 0);
 627         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
 628         ok(!lstrcmpW(mime, mimeTextPlain), "[%d] wrong mime\n", i);
 629         CoTaskMemFree(mime);
 630
 631         mime = (LPWSTR)0xf0f0f0f0;
 632         hres = FindMimeFromData(NULL, mime_tests[i].url, NULL, 0, mimeAppOctetStream, 0, &mime, 0);
 633         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
 634         ok(!lstrcmpW(mime, mimeAppOctetStream), "[%d] wrong mime\n", i);
 635         CoTaskMemFree(mime);
 636     }
 637
 638     for(i=0; i < sizeof(mime_tests2)/sizeof(mime_tests2[0]); i++) {
 639         hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
 640                 NULL, 0, &mime, 0);
 641         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
 642         ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime: %s\n", i, debugstr_w(mime));
 643         CoTaskMemFree(mime);
 644
 645         hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
 646                 mimeTextHtml, 0, &mime, 0);
 647         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
 648         if(!lstrcmpW(mimeAppOctetStream, mime_tests2[i].mime)
 649            || !lstrcmpW(mimeTextPlain, mime_tests2[i].mime))
 650             ok(!lstrcmpW(mime, mimeTextHtml), "[%d] wrong mime\n", i);
 651         else
 652             ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
 653         CoTaskMemFree(mime);
 654
 655         hres = FindMimeFromData(NULL, NULL, mime_tests2[i].data, mime_tests2[i].size,
 656                 mimeImagePjpeg, 0, &mime, 0);
 657         ok(hres == S_OK, "[%d] FindMimeFromData failed: %08x\n", i, hres);
 658         if(!lstrcmpW(mimeAppOctetStream, mime_tests2[i].mime) || i == 17)
 659             ok(!lstrcmpW(mime, mimeImagePjpeg), "[%d] wrong mime\n", i);
 660         else
 661             ok(!lstrcmpW(mime, mime_tests2[i].mime), "[%d] wrong mime\n", i);
 662
 663         CoTaskMemFree(mime);
 664     }
 665
 666     hres = FindMimeFromData(NULL, url1, data1, sizeof(data1), NULL, 0, &mime, 0);
 667     ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
 668     ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
 669     CoTaskMemFree(mime);
 670
 671     hres = FindMimeFromData(NULL, url1, data1, sizeof(data1), mimeAppOctetStream, 0, &mime, 0);
 672     ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
 673     ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
 674     CoTaskMemFree(mime);
 675
 676     hres = FindMimeFromData(NULL, url4, data1, sizeof(data1), mimeAppOctetStream, 0, &mime, 0);
 677     ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
 678     ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
 679     CoTaskMemFree(mime);
 680
 681     hres = FindMimeFromData(NULL, NULL, NULL, 0, NULL, 0, &mime, 0);
 682     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, excepted E_INVALIDARG\n", hres);
 683
 684     hres = FindMimeFromData(NULL, NULL, NULL, 0, mimeTextPlain, 0, &mime, 0);
 685     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
 686
 687     hres = FindMimeFromData(NULL, NULL, data1, 0, NULL, 0, &mime, 0);
 688     ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
 689
 690     hres = FindMimeFromData(NULL, url1, data1, 0, NULL, 0, &mime, 0);
 691     ok(hres == E_FAIL, "FindMimeFromData failed: %08x, expected E_FAIL\n", hres);
 692
 693     hres = FindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, &mime, 0);
 694     ok(hres == S_OK, "FindMimeFromData failed: %08x\n", hres);
 695     ok(!lstrcmpW(mime, mimeTextPlain), "wrong mime\n");
 696     CoTaskMemFree(mime);
 697
 698     hres = FindMimeFromData(NULL, NULL, data1, 0, mimeTextPlain, 0, NULL, 0);
 699     ok(hres == E_INVALIDARG, "FindMimeFromData failed: %08x, expected E_INVALIDARG\n", hres);
 700 }
 701
 702 static const BYTE secid1[] = {'f','i','l','e',':',0,0,0,0};
 703 static const BYTE secid4[] ={'f','i','l','e',':',3,0,0,0};
 704 static const BYTE secid5[] = {'h','t','t','p',':','w','w','w','.','w','i','n','e','h','q',
 705     '.','o','r','g',3,0,0,0};
 706 static const BYTE secid6[] = {'a','b','o','u','t',':','b','l','a','n','k',3,0,0,0};
 707 static const BYTE secid7[] = {'f','t','p',':','w','i','n','e','h','q','.','o','r','g',
 708                               3,0,0,0};
 709
 710 static struct secmgr_test {
 711     LPCWSTR url;
 712     DWORD zone;
 713     HRESULT zone_hres;
 714     DWORD secid_size;
 715     const BYTE *secid;
 716     HRESULT secid_hres;
 717 } secmgr_tests[] = {
 718     {url1, 0,   S_OK, sizeof(secid1), secid1, S_OK},
 719     {url2, 100, 0x80041001, 0, NULL, E_INVALIDARG},
 720     {url3, 0,   S_OK, sizeof(secid1), secid1, S_OK},
 721     {url4, 3,   S_OK, sizeof(secid4), secid4, S_OK},
 722     {url5, 3,   S_OK, sizeof(secid5), secid5, S_OK},
 723     {url6, 3,   S_OK, sizeof(secid6), secid6, S_OK},
 724     {url7, 3,   S_OK, sizeof(secid7), secid7, S_OK}
 725 };
 726
 727 static void test_SecurityManager(void)
 728 {
 729     int i;
 730     IInternetSecurityManager *secmgr = NULL;
 731     BYTE buf[512];
 732     DWORD zone, size;
 733     HRESULT hres;
 734
 735     hres = CoInternetCreateSecurityManager(NULL, &secmgr, 0);
 736     ok(hres == S_OK, "CoInternetCreateSecurityManager failed: %08x\n", hres);
 737     if(FAILED(hres))
 738         return;
 739
 740     for(i=0; i < sizeof(secmgr_tests)/sizeof(secmgr_tests[0]); i++) {
 741         zone = 100;
 742         hres = IInternetSecurityManager_MapUrlToZone(secmgr, secmgr_tests[i].url,
 743                                                      &zone, 0);
 744         ok(hres == secmgr_tests[i].zone_hres,
 745            "[%d] MapUrlToZone failed: %08x, expected %08x\n",
 746                 i, hres, secmgr_tests[i].zone_hres);
 747         if(SUCCEEDED(hres))
 748             ok(zone == secmgr_tests[i].zone, "[%d] zone=%d, expected %d\n", i, zone,
 749                secmgr_tests[i].zone);
 750         else
 751             ok(zone == secmgr_tests[i].zone || zone == -1, "[%d] zone=%d\n", i, zone);
 752
 753         size = sizeof(buf);
 754         memset(buf, 0xf0, sizeof(buf));
 755         hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[i].url,
 756                 buf, &size, 0);
 757         ok(hres == secmgr_tests[i].secid_hres,
 758            "[%d] GetSecurityId failed: %08x, expected %08x\n",
 759            i, hres, secmgr_tests[i].secid_hres);
 760         if(secmgr_tests[i].secid) {
 761             ok(size == secmgr_tests[i].secid_size, "[%d] size=%d, expected %d\n",
 762                     i, size, secmgr_tests[i].secid_size);
 763             ok(!memcmp(buf, secmgr_tests[i].secid, size), "[%d] wrong secid\n", i);
 764         }
 765     }
 766
 767     zone = 100;
 768     hres = IInternetSecurityManager_MapUrlToZone(secmgr, NULL, &zone, 0);
 769     ok(hres == E_INVALIDARG, "MapUrlToZone failed: %08x, expected E_INVALIDARG\n", hres);
 770     ok(zone == 100 || zone == -1, "zone=%d\n", zone);
 771
 772     size = sizeof(buf);
 773     hres = IInternetSecurityManager_GetSecurityId(secmgr, NULL, buf, &size, 0);
 774     ok(hres == E_INVALIDARG,
 775        "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
 776     hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[1].url,
 777                                                   NULL, &size, 0);
 778     ok(hres == E_INVALIDARG,
 779        "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
 780     hres = IInternetSecurityManager_GetSecurityId(secmgr, secmgr_tests[1].url,
 781                                                   buf, NULL, 0);
 782     ok(hres == E_INVALIDARG,
 783        "GetSecurityId failed: %08x, expected E_INVALIDARG\n", hres);
 784
 785     IInternetSecurityManager_Release(secmgr);
 786 }
 787
 788 static void test_ZoneManager(void)
 789 {
 790     IInternetZoneManager *zonemgr = NULL;
 791     BYTE buf[32];
 792     HRESULT hres;
 793
 794     hres = CoInternetCreateZoneManager(NULL, &zonemgr, 0);
 795     ok(hres == S_OK, "CoInternetCreateZoneManager failed: %08x\n", hres);
 796     if(FAILED(hres))
 797         return;
 798
 799     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, buf,
 800             sizeof(DWORD), URLZONEREG_DEFAULT);
 801     ok(hres == S_OK, "GetZoneActionPolicy failed: %08x\n", hres);
 802     ok(*(DWORD*)buf == 1, "policy=%d, expected 1\n", *(DWORD*)buf);
 803
 804     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, NULL,
 805             sizeof(DWORD), URLZONEREG_DEFAULT);
 806     ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
 807
 808     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1a10, buf,
 809             2, URLZONEREG_DEFAULT);
 810     ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
 811
 812     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 3, 0x1fff, buf,
 813             sizeof(DWORD), URLZONEREG_DEFAULT);
 814     ok(hres == E_FAIL, "GetZoneActionPolicy failed: %08x, expected E_FAIL\n", hres);
 815
 816     hres = IInternetZoneManager_GetZoneActionPolicy(zonemgr, 13, 0x1a10, buf,
 817             sizeof(DWORD), URLZONEREG_DEFAULT);
 818     ok(hres == E_INVALIDARG, "GetZoneActionPolicy failed: %08x, expected E_INVALIDARG\n", hres);
 819
 820     IInternetZoneManager_Release(zonemgr);
 821 }
 822
 823 static void register_protocols(void)
 824 {
 825     IInternetSession *session;
 826     IClassFactory *factory;
 827     HRESULT hres;
 828
 829     static const WCHAR wszAbout[] = {'a','b','o','u','t',0};
 830
 831     hres = CoInternetGetSession(0, &session, 0);
 832     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
 833     if(FAILED(hres))
 834         return;
 835
 836     hres = CoGetClassObject(&CLSID_AboutProtocol, CLSCTX_INPROC_SERVER, NULL,
 837             &IID_IClassFactory, (void**)&factory);
 838     ok(hres == S_OK, "Coud not get AboutProtocol factory: %08x\n", hres);
 839     if(FAILED(hres))
 840         return;
 841
 842     IInternetSession_RegisterNameSpace(session, factory, &CLSID_AboutProtocol,
 843                                        wszAbout, 0, NULL, 0);
 844     IClassFactory_Release(factory);
 845
 846     IInternetSession_Release(session);
 847 }
 848
 849 static HRESULT WINAPI InternetProtocolInfo_QueryInterface(IInternetProtocolInfo *iface,
 850                                                           REFIID riid, void **ppv)
 851 {
 852     ok(0, "unexpected call\n");
 853     return E_NOINTERFACE;
 854 }
 855
 856 static ULONG WINAPI InternetProtocolInfo_AddRef(IInternetProtocolInfo *iface)
 857 {
 858     return 2;
 859 }
 860
 861 static ULONG WINAPI InternetProtocolInfo_Release(IInternetProtocolInfo *iface)
 862 {
 863     return 1;
 864 }
 865
 866 static HRESULT WINAPI InternetProtocolInfo_ParseUrl(IInternetProtocolInfo *iface, LPCWSTR pwzUrl,
 867         PARSEACTION ParseAction, DWORD dwParseFlags, LPWSTR pwzResult, DWORD cchResult,
 868         DWORD *pcchResult, DWORD dwReserved)
 869 {
 870     CHECK_EXPECT(ParseUrl);
 871     return E_NOTIMPL;
 872 }
 873
 874 static HRESULT WINAPI InternetProtocolInfo_CombineUrl(IInternetProtocolInfo *iface,
 875         LPCWSTR pwzBaseUrl, LPCWSTR pwzRelativeUrl, DWORD dwCombineFlags,
 876         LPWSTR pwzResult, DWORD cchResult, DWORD *pcchResult, DWORD dwReserved)
 877 {
 878     ok(0, "unexpected call\n");
 879     return E_NOTIMPL;
 880 }
 881
 882 static HRESULT WINAPI InternetProtocolInfo_CompareUrl(IInternetProtocolInfo *iface,
 883         LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags)
 884 {
 885     ok(0, "unexpected call\n");
 886     return E_NOTIMPL;
 887 }
 888
 889 static HRESULT WINAPI InternetProtocolInfo_QueryInfo(IInternetProtocolInfo *iface,
 890         LPCWSTR pwzUrl, QUERYOPTION OueryOption, DWORD dwQueryFlags, LPVOID pBuffer,
 891         DWORD cbBuffer, DWORD *pcbBuf, DWORD dwReserved)
 892 {
 893     ok(0, "unexpected call\n");
 894     return E_NOTIMPL;
 895 }
 896
 897 static const IInternetProtocolInfoVtbl InternetProtocolInfoVtbl = {
 898     InternetProtocolInfo_QueryInterface,
 899     InternetProtocolInfo_AddRef,
 900     InternetProtocolInfo_Release,
 901     InternetProtocolInfo_ParseUrl,
 902     InternetProtocolInfo_CombineUrl,
 903     InternetProtocolInfo_CompareUrl,
 904     InternetProtocolInfo_QueryInfo
 905 };
 906
 907 static IInternetProtocolInfo protocol_info = { &InternetProtocolInfoVtbl };
 908
 909 static HRESULT qiret;
 910 static IClassFactory *expect_cf;
 911
 912 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
 913 {
 914     if(IsEqualGUID(&IID_IInternetProtocolInfo, riid)) {
 915         CHECK_EXPECT(QI_IInternetProtocolInfo);
 916         ok(iface == expect_cf, "unexpected iface\n");
 917         *ppv = &protocol_info;
 918         return qiret;
 919     }
 920
 921     ok(0, "unexpected call\n");
 922     return E_NOINTERFACE;
 923 }
 924
 925 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
 926 {
 927     return 2;
 928 }
 929
 930 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
 931 {
 932     return 1;
 933 }
 934
 935 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
 936                                         REFIID riid, void **ppv)
 937 {
 938     CHECK_EXPECT(CreateInstance);
 939
 940     ok(iface == expect_cf, "unexpected iface\n");
 941     ok(pOuter == NULL, "pOuter = %p\n", pOuter);
 942     ok(IsEqualGUID(&IID_IInternetProtocolInfo, riid), "unexpected riid\n");
 943     ok(ppv != NULL, "ppv == NULL\n");
 944
 945     *ppv = &protocol_info;
 946     return S_OK;
 947 }
 948
 949 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
 950 {
 951     ok(0, "unexpected call\n");
 952     return S_OK;
 953 }
 954
 955 static const IClassFactoryVtbl ClassFactoryVtbl = {
 956     ClassFactory_QueryInterface,
 957     ClassFactory_AddRef,
 958     ClassFactory_Release,
 959     ClassFactory_CreateInstance,
 960     ClassFactory_LockServer
 961 };
 962
 963 static IClassFactory test_protocol_cf = { &ClassFactoryVtbl };
 964 static IClassFactory test_protocol_cf2 = { &ClassFactoryVtbl };
 965
 966 static void test_NameSpace(void)
 967 {
 968     IInternetSession *session;
 969     WCHAR buf[200];
 970     DWORD size;
 971     HRESULT hres;
 972
 973     static const WCHAR wszTest[] = {'t','e','s','t',0};
 974
 975     hres = CoInternetGetSession(0, &session, 0);
 976     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
 977     if(FAILED(hres))
 978         return;
 979
 980     hres = IInternetSession_RegisterNameSpace(session, NULL, &IID_NULL,
 981                                               wszTest, 0, NULL, 0);
 982     ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
 983
 984     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
 985                                               NULL, 0, NULL, 0);
 986     ok(hres == E_INVALIDARG, "RegisterNameSpace failed: %08x\n", hres);
 987
 988     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
 989                                               wszTest, 0, NULL, 0);
 990     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
 991
 992     qiret = E_NOINTERFACE;
 993     expect_cf = &test_protocol_cf;
 994     SET_EXPECT(QI_IInternetProtocolInfo);
 995     SET_EXPECT(CreateInstance);
 996     SET_EXPECT(ParseUrl);
 997
 998     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
 999                               &size, 0);
1000     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1001
1002     CHECK_CALLED(QI_IInternetProtocolInfo);
1003     CHECK_CALLED(CreateInstance);
1004     CHECK_CALLED(ParseUrl);
1005
1006     qiret = S_OK;
1007     SET_EXPECT(QI_IInternetProtocolInfo);
1008     SET_EXPECT(ParseUrl);
1009
1010     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1011                               &size, 0);
1012     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1013
1014     CHECK_CALLED(QI_IInternetProtocolInfo);
1015     CHECK_CALLED(ParseUrl);
1016
1017     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1018     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1019
1020     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1021                               &size, 0);
1022     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1023
1024     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf2, &IID_NULL,
1025                                               wszTest, 0, NULL, 0);
1026     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1027
1028     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
1029                                               wszTest, 0, NULL, 0);
1030     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1031
1032     hres = IInternetSession_RegisterNameSpace(session, &test_protocol_cf, &IID_NULL,
1033                                               wszTest, 0, NULL, 0);
1034     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1035
1036     SET_EXPECT(QI_IInternetProtocolInfo);
1037     SET_EXPECT(ParseUrl);
1038
1039     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1040                               &size, 0);
1041     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1042
1043     CHECK_CALLED(QI_IInternetProtocolInfo);
1044     CHECK_CALLED(ParseUrl);
1045
1046     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1047     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1048
1049     SET_EXPECT(QI_IInternetProtocolInfo);
1050     SET_EXPECT(ParseUrl);
1051
1052     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1053                               &size, 0);
1054     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1055
1056     CHECK_CALLED(QI_IInternetProtocolInfo);
1057     CHECK_CALLED(ParseUrl);
1058
1059     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1060     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1061
1062     expect_cf = &test_protocol_cf2;
1063     SET_EXPECT(QI_IInternetProtocolInfo);
1064     SET_EXPECT(ParseUrl);
1065
1066     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1067                               &size, 0);
1068     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1069
1070     CHECK_CALLED(QI_IInternetProtocolInfo);
1071     CHECK_CALLED(ParseUrl);
1072
1073     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1074     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1075     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, wszTest);
1076     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1077     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf, NULL);
1078     ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres);
1079     hres = IInternetSession_UnregisterNameSpace(session, NULL, wszTest);
1080     ok(hres == E_INVALIDARG, "UnregisterNameSpace failed: %08x\n", hres);
1081
1082     hres = IInternetSession_UnregisterNameSpace(session, &test_protocol_cf2, wszTest);
1083     ok(hres == S_OK, "UnregisterNameSpace failed: %08x\n", hres);
1084
1085     hres = CoInternetParseUrl(url8, PARSE_ENCODE, 0, buf, sizeof(buf)/sizeof(WCHAR),
1086                               &size, 0);
1087     ok(hres == S_OK, "CoInternetParseUrl failed: %08x\n", hres);
1088
1089     IInternetSession_Release(session);
1090 }
1091
1092 static ULONG WINAPI unk_Release(IUnknown *iface)
1093 {
1094     CHECK_EXPECT(unk_Release);
1095     return 0;
1096 }
1097
1098 static const IUnknownVtbl unk_vtbl = {
1099     (void*)0xdeadbeef,
1100     (void*)0xdeadbeef,
1101     unk_Release
1102 };
1103
1104 static void test_ReleaseBindInfo(void)
1105 {
1106     BINDINFO bi;
1107     IUnknown unk = { &unk_vtbl };
1108
1109     ReleaseBindInfo(NULL); /* shouldn't crash */
1110
1111     memset(&bi, 0, sizeof(bi));
1112     bi.cbSize = sizeof(BINDINFO);
1113     bi.pUnk = &unk;
1114     SET_EXPECT(unk_Release);
1115     ReleaseBindInfo(&bi);
1116     ok(bi.cbSize == sizeof(BINDINFO), "bi.cbSize=%d\n", bi.cbSize);
1117     ok(bi.pUnk == NULL, "bi.pUnk=%p, expected NULL\n", bi.pUnk);
1118     CHECK_CALLED(unk_Release);
1119
1120     memset(&bi, 0, sizeof(bi));
1121     bi.cbSize = offsetof(BINDINFO, pUnk);
1122     bi.pUnk = &unk;
1123     ReleaseBindInfo(&bi);
1124     ok(bi.cbSize == offsetof(BINDINFO, pUnk), "bi.cbSize=%d\n", bi.cbSize);
1125     ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk);
1126
1127     memset(&bi, 0, sizeof(bi));
1128     bi.pUnk = &unk;
1129     ReleaseBindInfo(&bi);
1130     ok(!bi.cbSize, "bi.cbSize=%d, expected 0\n", bi.cbSize);
1131     ok(bi.pUnk == &unk, "bi.pUnk=%p, expected %p\n", bi.pUnk, &unk);
1132 }
1133
1134 static void test_UrlMkGetSessionOption(void)
1135 {
1136     DWORD encoding, size;
1137     HRESULT hres;
1138
1139     size = encoding = 0xdeadbeef;
1140     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1141                                  sizeof(encoding), &size, 0);
1142     ok(hres == S_OK, "UrlMkGetSessionOption failed: %08x\n", hres);
1143     ok(encoding != 0xdeadbeef, "encoding not changed\n");
1144     ok(size == sizeof(encoding), "size=%d\n", size);
1145
1146     size = encoding = 0xdeadbeef;
1147     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1148                                  sizeof(encoding)+1, &size, 0);
1149     ok(hres == S_OK, "UrlMkGetSessionOption failed: %08x\n", hres);
1150     ok(encoding != 0xdeadbeef, "encoding not changed\n");
1151     ok(size == sizeof(encoding), "size=%d\n", size);
1152
1153     size = encoding = 0xdeadbeef;
1154     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1155                                  sizeof(encoding)-1, &size, 0);
1156     ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1157     ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1158     ok(size == 0xdeadbeef, "size=%d\n", size);
1159
1160     size = encoding = 0xdeadbeef;
1161     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, NULL,
1162                                  sizeof(encoding)-1, &size, 0);
1163     ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1164     ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1165     ok(size == 0xdeadbeef, "size=%d\n", size);
1166
1167     encoding = 0xdeadbeef;
1168     hres = UrlMkGetSessionOption(URLMON_OPTION_URL_ENCODING, &encoding,
1169                                  sizeof(encoding)-1, NULL, 0);
1170     ok(hres == E_INVALIDARG, "UrlMkGetSessionOption failed: %08x\n", hres);
1171     ok(encoding == 0xdeadbeef, "encoding = %08x, exepcted 0xdeadbeef\n", encoding);
1172 }
1173
1174 static void test_ObtainUserAgentString(void)
1175 {
1176     static const CHAR expected[] = "Mozilla/4.0 (compatible; MSIE ";
1177     static CHAR str[3];
1178     LPSTR str2 = NULL;
1179     HRESULT hres;
1180     DWORD size, saved;
1181
1182     hres = ObtainUserAgentString(0, NULL, NULL);
1183     ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1184
1185     size = 100;
1186     hres = ObtainUserAgentString(0, NULL, &size);
1187     ok(hres == E_INVALIDARG, "ObtainUserAgentString failed: %08x\n", hres);
1188     ok(size == 100, "size=%d, expected %d\n", size, 100);
1189
1190     size = 0;
1191     hres = ObtainUserAgentString(0, str, &size);
1192     ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1193     ok(size > 0, "size=%d, expected non-zero\n", size);
1194
1195     size = 2;
1196     str[0] = 'a';
1197     hres = ObtainUserAgentString(0, str, &size);
1198     ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1199     ok(size > 0, "size=%d, expected non-zero\n", size);
1200     ok(str[0] == 'a', "str[0]=%c, expected 'a'\n", str[0]);
1201
1202     size = 0;
1203     hres = ObtainUserAgentString(1, str, &size);
1204     ok(hres == E_OUTOFMEMORY, "ObtainUserAgentString failed: %08x\n", hres);
1205     ok(size > 0, "size=%d, expected non-zero\n", size);
1206
1207     str2 = HeapAlloc(GetProcessHeap(), 0, (size+20)*sizeof(CHAR));
1208     if (!str2)
1209     {
1210         skip("skipping rest of ObtainUserAgent tests, out of memory\n");
1211     }
1212     else
1213     {
1214         saved = size;
1215         hres = ObtainUserAgentString(0, str2, &size);
1216         ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres);
1217         ok(size == saved, "size=%d, expected %d\n", size, saved);
1218         ok(strlen(expected) <= strlen(str2) &&
1219            !memcmp(expected, str2, strlen(expected)*sizeof(CHAR)),
1220            "user agent was \"%s\", expected to start with \"%s\"\n",
1221            str2, expected);
1222
1223         size = saved+10;
1224         hres = ObtainUserAgentString(0, str2, &size);
1225         ok(hres == S_OK, "ObtainUserAgentString failed: %08x\n", hres);
1226         ok(size == saved, "size=%d, expected %d\n", size, saved);
1227     }
1228     HeapFree(GetProcessHeap(), 0, str2);
1229 }
1230
1231 START_TEST(misc)
1232 {
1233     OleInitialize(NULL);
1234
1235     register_protocols();
1236
1237     test_CreateFormatEnum();
1238     test_RegisterFormatEnumerator();
1239     test_CoInternetParseUrl();
1240     test_CoInternetCompareUrl();
1241     test_CoInternetQueryInfo();
1242     test_FindMimeFromData();
1243     test_SecurityManager();
1244     test_ZoneManager();
1245     test_NameSpace();
1246     test_ReleaseBindInfo();
1247     test_UrlMkGetSessionOption();
1248     test_ObtainUserAgentString();
1249
1250     OleUninitialize();
1251 }