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