dlls/urlmon/tests/protocol.c

   1 /*
   2  * Copyright 2005-2007 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 <stdio.h>
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "ole2.h"
  29 #include "urlmon.h"
  30 #include "wininet.h"
  31
  32 #define DEFINE_EXPECT(func) \
  33     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  34
  35 #define SET_EXPECT(func) \
  36     expect_ ## func = TRUE
  37
  38 #define CHECK_EXPECT2(func) \
  39     do { \
  40         ok(expect_ ##func, "unexpected call " #func  "\n"); \
  41         called_ ## func = TRUE; \
  42     }while(0)
  43
  44 #define CHECK_EXPECT(func) \
  45     do { \
  46         CHECK_EXPECT2(func);     \
  47         expect_ ## func = FALSE; \
  48     }while(0)
  49
  50 #define CHECK_CALLED(func) \
  51     do { \
  52         ok(called_ ## func, "expected " #func "\n"); \
  53         expect_ ## func = called_ ## func = FALSE; \
  54     }while(0)
  55
  56 #define CHECK_NOT_CALLED(func) \
  57     do { \
  58         ok(!called_ ## func, "unexpected " #func "\n"); \
  59         expect_ ## func = called_ ## func = FALSE; \
  60     }while(0)
  61
  62 #define CLEAR_CALLED(func) \
  63     expect_ ## func = called_ ## func = FALSE
  64
  65 DEFINE_EXPECT(GetBindInfo);
  66 DEFINE_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
  67 DEFINE_EXPECT(ReportProgress_DIRECTBIND);
  68 DEFINE_EXPECT(ReportProgress_RAWMIMETYPE);
  69 DEFINE_EXPECT(ReportProgress_FINDINGRESOURCE);
  70 DEFINE_EXPECT(ReportProgress_CONNECTING);
  71 DEFINE_EXPECT(ReportProgress_SENDINGREQUEST);
  72 DEFINE_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE);
  73 DEFINE_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE);
  74 DEFINE_EXPECT(ReportProgress_PROTOCOLCLASSID);
  75 DEFINE_EXPECT(ReportProgress_COOKIE_SENT);
  76 DEFINE_EXPECT(ReportProgress_REDIRECTING);
  77 DEFINE_EXPECT(ReportData);
  78 DEFINE_EXPECT(ReportResult);
  79 DEFINE_EXPECT(GetBindString_ACCEPT_MIMES);
  80 DEFINE_EXPECT(GetBindString_USER_AGENT);
  81 DEFINE_EXPECT(GetBindString_POST_COOKIE);
  82 DEFINE_EXPECT(QueryService_HttpNegotiate);
  83 DEFINE_EXPECT(QueryService_InternetProtocol);
  84 DEFINE_EXPECT(BeginningTransaction);
  85 DEFINE_EXPECT(GetRootSecurityId);
  86 DEFINE_EXPECT(OnResponse);
  87 DEFINE_EXPECT(Switch);
  88 DEFINE_EXPECT(Continue);
  89 DEFINE_EXPECT(CreateInstance);
  90 DEFINE_EXPECT(Start);
  91 DEFINE_EXPECT(Terminate);
  92 DEFINE_EXPECT(Read);
  93 DEFINE_EXPECT(SetPriority);
  94 DEFINE_EXPECT(LockRequest);
  95 DEFINE_EXPECT(UnlockRequest);
  96
  97 static const WCHAR wszIndexHtml[] = {'i','n','d','e','x','.','h','t','m','l',0};
  98 static const WCHAR index_url[] =
  99     {'f','i','l','e',':','i','n','d','e','x','.','h','t','m','l',0};
 100
 101 static const WCHAR acc_mimeW[] = {'*','/','*',0};
 102 static const WCHAR user_agentW[] = {'W','i','n','e',0};
 103 static const WCHAR text_htmlW[] = {'t','e','x','t','/','h','t','m','l',0};
 104 static const WCHAR hostW[] = {'w','w','w','.','w','i','n','e','h','q','.','o','r','g',0};
 105 static const WCHAR winehq_ipW[] = {'2','0','9','.','4','6','.','2','5','.','1','3','4',0};
 106 static const WCHAR emptyW[] = {0};
 107
 108 static HRESULT expect_hrResult;
 109 static LPCWSTR file_name, http_url, expect_wsz;
 110 static IInternetProtocol *http_protocol = NULL;
 111 static BOOL first_data_notif = FALSE, http_is_first = FALSE,
 112     http_post_test = FALSE;
 113 static int state = 0, prot_state;
 114 static DWORD bindf = 0, ex_priority = 0;
 115 static IInternetProtocol *binding_protocol;
 116 static IInternetBindInfo *prot_bind_info;
 117 static IInternetProtocolSink *binding_sink;
 118 static void *expect_pv;
 119 static HANDLE event_complete, event_complete2;
 120 static BOOL binding_test;
 121 static PROTOCOLDATA protocoldata, *pdata;
 122 static DWORD prot_read;
 123
 124 static enum {
 125     FILE_TEST,
 126     HTTP_TEST,
 127     MK_TEST,
 128     BIND_TEST
 129 } tested_protocol;
 130
 131 static const WCHAR protocol_names[][10] = {
 132     {'f','i','l','e',0},
 133     {'h','t','t','p',0},
 134     {'m','k',0},
 135     {'t','e','s','t',0}
 136 };
 137
 138 static const WCHAR binding_urls[][30] = {
 139     {'f','i','l','e',':','t','e','s','t','.','h','t','m','l',0},
 140     {'h','t','t','p',':','/','/','t','e','s','t','/','t','e','s','t','.','h','t','m','l',0},
 141     {'m','k',':','t','e','s','t',0},
 142     {'t','e','s','t',':','/','/','f','i','l','e','.','h','t','m','l',0}
 143 };
 144
 145 static const char *debugstr_w(LPCWSTR str)
 146 {
 147     static char buf[512];
 148     if(!str)
 149         return "(null)";
 150     WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL);
 151     return buf;
 152 }
 153
 154 static const char *debugstr_guid(REFIID riid)
 155 {
 156     static char buf[50];
 157
 158     sprintf(buf, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
 159             riid->Data1, riid->Data2, riid->Data3, riid->Data4[0],
 160             riid->Data4[1], riid->Data4[2], riid->Data4[3], riid->Data4[4],
 161             riid->Data4[5], riid->Data4[6], riid->Data4[7]);
 162
 163     return buf;
 164 }
 165
 166 static HRESULT WINAPI HttpNegotiate_QueryInterface(IHttpNegotiate2 *iface, REFIID riid, void **ppv)
 167 {
 168     if(IsEqualGUID(&IID_IUnknown, riid)
 169             || IsEqualGUID(&IID_IHttpNegotiate, riid)
 170             || IsEqualGUID(&IID_IHttpNegotiate2, riid)) {
 171         *ppv = iface;
 172         return S_OK;
 173     }
 174
 175     ok(0, "unexpected call\n");
 176     return E_NOINTERFACE;
 177 }
 178
 179 static ULONG WINAPI HttpNegotiate_AddRef(IHttpNegotiate2 *iface)
 180 {
 181     return 2;
 182 }
 183
 184 static ULONG WINAPI HttpNegotiate_Release(IHttpNegotiate2 *iface)
 185 {
 186     return 1;
 187 }
 188
 189 static HRESULT WINAPI HttpNegotiate_BeginningTransaction(IHttpNegotiate2 *iface, LPCWSTR szURL,
 190         LPCWSTR szHeaders, DWORD dwReserved, LPWSTR *pszAdditionalHeaders)
 191 {
 192     LPWSTR addl_headers;
 193
 194     static const WCHAR wszHeaders[] =
 195         {'C','o','n','t','e','n','t','-','T','y','p','e',':',' ','a','p','p','l','i','c','a','t',
 196          'i','o','n','/','x','-','w','w','w','-','f','o','r','m','-','u','r','l','e','n','c','o',
 197          'd','e','d','\r','\n',0};
 198
 199     CHECK_EXPECT(BeginningTransaction);
 200
 201     if(binding_test)
 202         ok(!lstrcmpW(szURL, binding_urls[tested_protocol]), "szURL != http_url\n");
 203     else
 204         ok(!lstrcmpW(szURL, http_url), "szURL != http_url\n");
 205     ok(!dwReserved, "dwReserved=%d, expected 0\n", dwReserved);
 206     ok(pszAdditionalHeaders != NULL, "pszAdditionalHeaders == NULL\n");
 207     if(pszAdditionalHeaders)
 208     {
 209         ok(*pszAdditionalHeaders == NULL, "*pszAdditionalHeaders != NULL\n");
 210         if (http_post_test)
 211         {
 212             addl_headers = CoTaskMemAlloc(sizeof(wszHeaders));
 213             if (!addl_headers)
 214             {
 215                 http_post_test = FALSE;
 216                 skip("Out of memory\n");
 217                 return E_OUTOFMEMORY;
 218             }
 219             lstrcpyW(addl_headers, wszHeaders);
 220             *pszAdditionalHeaders = addl_headers;
 221         }
 222     }
 223
 224     return S_OK;
 225 }
 226
 227 static HRESULT WINAPI HttpNegotiate_OnResponse(IHttpNegotiate2 *iface, DWORD dwResponseCode,
 228         LPCWSTR szResponseHeaders, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders)
 229 {
 230     CHECK_EXPECT(OnResponse);
 231
 232     ok(dwResponseCode == 200, "dwResponseCode=%d, expected 200\n", dwResponseCode);
 233     ok(szResponseHeaders != NULL, "szResponseHeaders == NULL\n");
 234     ok(szRequestHeaders == NULL, "szRequestHeaders != NULL\n");
 235     ok(pszAdditionalRequestHeaders == NULL, "pszAdditionalHeaders != NULL\n");
 236
 237     return S_OK;
 238 }
 239
 240 static HRESULT WINAPI HttpNegotiate_GetRootSecurityId(IHttpNegotiate2 *iface,
 241         BYTE *pbSecurityId, DWORD *pcbSecurityId, DWORD_PTR dwReserved)
 242 {
 243     static const BYTE sec_id[] = {'h','t','t','p',':','t','e','s','t',1,0,0,0};
 244
 245     CHECK_EXPECT(GetRootSecurityId);
 246
 247     ok(!dwReserved, "dwReserved=%ld, expected 0\n", dwReserved);
 248     ok(pbSecurityId != NULL, "pbSecurityId == NULL\n");
 249     ok(pcbSecurityId != NULL, "pcbSecurityId == NULL\n");
 250
 251     if(pcbSecurityId) {
 252         ok(*pcbSecurityId == 512, "*pcbSecurityId=%d, expected 512\n", *pcbSecurityId);
 253         *pcbSecurityId = sizeof(sec_id);
 254     }
 255
 256     if(pbSecurityId)
 257         memcpy(pbSecurityId, sec_id, sizeof(sec_id));
 258
 259     return E_FAIL;
 260 }
 261
 262 static IHttpNegotiate2Vtbl HttpNegotiateVtbl = {
 263     HttpNegotiate_QueryInterface,
 264     HttpNegotiate_AddRef,
 265     HttpNegotiate_Release,
 266     HttpNegotiate_BeginningTransaction,
 267     HttpNegotiate_OnResponse,
 268     HttpNegotiate_GetRootSecurityId
 269 };
 270
 271 static IHttpNegotiate2 http_negotiate = { &HttpNegotiateVtbl };
 272
 273 static HRESULT QueryInterface(REFIID,void**);
 274
 275 static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv)
 276 {
 277     return QueryInterface(riid, ppv);
 278 }
 279
 280 static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface)
 281 {
 282     return 2;
 283 }
 284
 285 static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface)
 286 {
 287     return 1;
 288 }
 289
 290 static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface, REFGUID guidService,
 291         REFIID riid, void **ppv)
 292 {
 293     if(IsEqualGUID(&IID_IHttpNegotiate, guidService) || IsEqualGUID(&IID_IHttpNegotiate2, riid)) {
 294         CHECK_EXPECT2(QueryService_HttpNegotiate);
 295         return IHttpNegotiate2_QueryInterface(&http_negotiate, riid, ppv);
 296     }
 297
 298     if(IsEqualGUID(&IID_IInternetProtocol, guidService)) {
 299         ok(IsEqualGUID(&IID_IInternetProtocol, riid), "unexpected riid\n");
 300         CHECK_EXPECT(QueryService_InternetProtocol);
 301         return E_NOINTERFACE;
 302     }
 303
 304     ok(0, "unexpected service %s\n", debugstr_guid(guidService));
 305     return E_FAIL;
 306 }
 307
 308 static const IServiceProviderVtbl ServiceProviderVtbl = {
 309     ServiceProvider_QueryInterface,
 310     ServiceProvider_AddRef,
 311     ServiceProvider_Release,
 312     ServiceProvider_QueryService
 313 };
 314
 315 static IServiceProvider service_provider = { &ServiceProviderVtbl };
 316
 317 static HRESULT WINAPI ProtocolSink_QueryInterface(IInternetProtocolSink *iface, REFIID riid, void **ppv)
 318 {
 319     return QueryInterface(riid, ppv);
 320 }
 321
 322 static ULONG WINAPI ProtocolSink_AddRef(IInternetProtocolSink *iface)
 323 {
 324     return 2;
 325 }
 326
 327 static ULONG WINAPI ProtocolSink_Release(IInternetProtocolSink *iface)
 328 {
 329     return 1;
 330 }
 331
 332 static HRESULT WINAPI ProtocolSink_Switch(IInternetProtocolSink *iface, PROTOCOLDATA *pProtocolData)
 333 {
 334     HRESULT hres;
 335
 336     CHECK_EXPECT(Switch);
 337     ok(pProtocolData != NULL, "pProtocolData == NULL\n");
 338
 339     pdata = pProtocolData;
 340
 341     if(binding_test) {
 342         SetEvent(event_complete);
 343         WaitForSingleObject(event_complete2, INFINITE);
 344         return S_OK;
 345     }
 346
 347     if (!state) {
 348         if (http_is_first) {
 349             CHECK_CALLED(ReportProgress_FINDINGRESOURCE);
 350             CHECK_CALLED(ReportProgress_CONNECTING);
 351         } else todo_wine {
 352             CHECK_NOT_CALLED(ReportProgress_FINDINGRESOURCE);
 353             CHECK_NOT_CALLED(ReportProgress_CONNECTING);
 354         }
 355         CHECK_CALLED(ReportProgress_SENDINGREQUEST);
 356         SET_EXPECT(OnResponse);
 357         SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
 358         if(bindf & BINDF_NEEDFILE)
 359             SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE);
 360     }
 361
 362     SET_EXPECT(ReportData);
 363     hres = IInternetProtocol_Continue(http_protocol, pProtocolData);
 364     ok(hres == S_OK, "Continue failed: %08x\n", hres);
 365     CHECK_CALLED(ReportData);
 366
 367     if (!state) {
 368         state = 1;
 369         CHECK_CALLED(OnResponse);
 370         CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE);
 371         if(bindf & BINDF_NEEDFILE)
 372             CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE);
 373     }
 374
 375     SetEvent(event_complete);
 376
 377     return S_OK;
 378 }
 379
 380 static HRESULT WINAPI ProtocolSink_ReportProgress(IInternetProtocolSink *iface, ULONG ulStatusCode,
 381         LPCWSTR szStatusText)
 382 {
 383     static const WCHAR null_guid[] = {'{','0','0','0','0','0','0','0','0','-','0','0','0','0','-',
 384         '0','0','0','0','-','0','0','0','0','-','0','0','0','0','0','0','0','0','0','0','0','0','}',0};
 385     static const WCHAR text_plain[] = {'t','e','x','t','/','p','l','a','i','n',0};
 386     static const WCHAR post_host[] =
 387         {'c','r','o','s','s','o','v','e','r','.','c','o','d','e',
 388          'w','e','a','v','e','r','s','.','c','o','m',0};
 389     static const WCHAR wszCrossoverIP[] =
 390         {'2','0','9','.','4','6','.','2','5','.','1','3','2',0};
 391     /* I'm not sure if it's a good idea to hardcode here the IP address... */
 392
 393     switch(ulStatusCode) {
 394     case BINDSTATUS_MIMETYPEAVAILABLE:
 395         CHECK_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
 396         ok(szStatusText != NULL, "szStatusText == NULL\n");
 397         if(szStatusText) {
 398             if(tested_protocol == BIND_TEST)
 399                 ok(szStatusText == expect_wsz, "unexpected szStatusText\n");
 400             else if (http_post_test)
 401                 ok(lstrlenW(text_plain) <= lstrlenW(szStatusText) &&
 402                    !memcmp(szStatusText, text_plain, lstrlenW(text_plain)*sizeof(WCHAR)),
 403                    "szStatusText != text/plain\n");
 404             else
 405                 ok(lstrlenW(text_htmlW) <= lstrlenW(szStatusText) &&
 406                    !memcmp(szStatusText, text_htmlW, lstrlenW(text_htmlW)*sizeof(WCHAR)),
 407                    "szStatusText != text/html\n");
 408         }
 409         break;
 410     case BINDSTATUS_DIRECTBIND:
 411         CHECK_EXPECT2(ReportProgress_DIRECTBIND);
 412         ok(szStatusText == NULL, "szStatusText != NULL\n");
 413         break;
 414     case BINDSTATUS_RAWMIMETYPE:
 415         CHECK_EXPECT2(ReportProgress_RAWMIMETYPE);
 416         ok(szStatusText != NULL, "szStatusText == NULL\n");
 417         if(szStatusText)
 418             ok(lstrlenW(szStatusText) < lstrlenW(text_htmlW) ||
 419                !memcmp(szStatusText, text_htmlW, lstrlenW(text_htmlW)*sizeof(WCHAR)),
 420                "szStatusText != text/html\n");
 421         break;
 422     case BINDSTATUS_CACHEFILENAMEAVAILABLE:
 423         CHECK_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE);
 424         ok(szStatusText != NULL, "szStatusText == NULL\n");
 425         if(szStatusText) {
 426             if(binding_test)
 427                 ok(!lstrcmpW(szStatusText, expect_wsz), "unexpected szStatusText\n");
 428             else if(tested_protocol == FILE_TEST)
 429                 ok(!lstrcmpW(szStatusText, file_name), "szStatusText = \"%s\"\n", debugstr_w(szStatusText));
 430             else
 431                 ok(szStatusText != NULL, "szStatusText == NULL\n");
 432         }
 433         break;
 434     case BINDSTATUS_FINDINGRESOURCE:
 435         CHECK_EXPECT(ReportProgress_FINDINGRESOURCE);
 436         ok(szStatusText != NULL, "szStatusText == NULL\n");
 437         if(szStatusText)
 438         {
 439             if (!http_post_test)
 440                 ok(!lstrcmpW(szStatusText, hostW),
 441                    "szStatustext != \"www.winehq.org\"\n");
 442             else
 443                 ok(!lstrcmpW(szStatusText, post_host),
 444                    "szStatustext != \"crossover.codeweavers.com\"\n");
 445         }
 446         break;
 447     case BINDSTATUS_CONNECTING:
 448         CHECK_EXPECT(ReportProgress_CONNECTING);
 449         ok(szStatusText != NULL, "szStatusText == NULL\n");
 450         if(szStatusText)
 451             ok(!lstrcmpW(szStatusText, http_post_test ?
 452                          wszCrossoverIP : winehq_ipW),
 453                "Unexpected szStatusText\n");
 454         break;
 455     case BINDSTATUS_SENDINGREQUEST:
 456         CHECK_EXPECT(ReportProgress_SENDINGREQUEST);
 457         if(tested_protocol == FILE_TEST) {
 458             ok(szStatusText != NULL, "szStatusText == NULL\n");
 459             if(szStatusText)
 460                 ok(!*szStatusText, "wrong szStatusText\n");
 461         }
 462         break;
 463     case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE:
 464         CHECK_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE);
 465         ok(szStatusText != NULL, "szStatusText == NULL\n");
 466         if(szStatusText)
 467             ok(!lstrcmpW(szStatusText, text_htmlW), "szStatusText != text/html\n");
 468         break;
 469     case BINDSTATUS_PROTOCOLCLASSID:
 470         CHECK_EXPECT(ReportProgress_PROTOCOLCLASSID);
 471         ok(szStatusText != NULL, "szStatusText == NULL\n");
 472         ok(!lstrcmpW(szStatusText, null_guid), "unexpected szStatusText\n");
 473         break;
 474     case BINDSTATUS_COOKIE_SENT:
 475         CHECK_EXPECT(ReportProgress_COOKIE_SENT);
 476         ok(szStatusText == NULL, "szStatusText != NULL\n");
 477         break;
 478     case BINDSTATUS_REDIRECTING:
 479         CHECK_EXPECT(ReportProgress_REDIRECTING);
 480         ok(szStatusText == NULL, "szStatusText != NULL\n");
 481         break;
 482     default:
 483         ok(0, "Unexpected status %d\n", ulStatusCode);
 484     };
 485
 486     return S_OK;
 487 }
 488
 489 static HRESULT WINAPI ProtocolSink_ReportData(IInternetProtocolSink *iface, DWORD grfBSCF,
 490         ULONG ulProgress, ULONG ulProgressMax)
 491 {
 492     if(tested_protocol == FILE_TEST) {
 493         CHECK_EXPECT2(ReportData);
 494
 495         ok(ulProgress == ulProgressMax, "ulProgress (%d) != ulProgressMax (%d)\n",
 496            ulProgress, ulProgressMax);
 497         ok(ulProgressMax == 13, "ulProgressMax=%d, expected 13\n", ulProgressMax);
 498         ok(grfBSCF == (BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION),
 499                 "grcfBSCF = %08x\n", grfBSCF);
 500     }else if(!binding_test && tested_protocol == HTTP_TEST) {
 501         if(!(grfBSCF & BSCF_LASTDATANOTIFICATION))
 502             CHECK_EXPECT(ReportData);
 503         else if (http_post_test)
 504             ok(ulProgress == 13, "Read %u bytes instead of 13\n", ulProgress);
 505
 506         ok(ulProgress, "ulProgress == 0\n");
 507
 508         if(first_data_notif) {
 509             ok(grfBSCF == BSCF_FIRSTDATANOTIFICATION, "grcfBSCF = %08x\n", grfBSCF);
 510             first_data_notif = FALSE;
 511         } else {
 512             ok(grfBSCF == BSCF_INTERMEDIATEDATANOTIFICATION
 513                || grfBSCF == (BSCF_LASTDATANOTIFICATION|BSCF_INTERMEDIATEDATANOTIFICATION),
 514                "grcfBSCF = %08x\n", grfBSCF);
 515         }
 516
 517         if(!(bindf & BINDF_FROMURLMON) &&
 518            !(grfBSCF & BSCF_LASTDATANOTIFICATION)) {
 519             if(!state) {
 520                 state = 1;
 521                 if(http_is_first) {
 522                     CHECK_CALLED(ReportProgress_FINDINGRESOURCE);
 523                     CHECK_CALLED(ReportProgress_CONNECTING);
 524                 } else todo_wine {
 525                     CHECK_NOT_CALLED(ReportProgress_FINDINGRESOURCE);
 526                     CHECK_NOT_CALLED(ReportProgress_CONNECTING);
 527                 }
 528                 CHECK_CALLED(ReportProgress_SENDINGREQUEST);
 529                 CHECK_CALLED(OnResponse);
 530                 CHECK_CALLED(ReportProgress_RAWMIMETYPE);
 531             }
 532             SetEvent(event_complete);
 533         }
 534     }else {
 535         BYTE buf[1000];
 536         ULONG read;
 537         HRESULT hres;
 538
 539         CHECK_EXPECT(ReportData);
 540
 541         if(tested_protocol == BIND_TEST)
 542             return S_OK;
 543
 544         do {
 545             SET_EXPECT(Read);
 546             hres = IInternetProtocol_Read(binding_protocol, expect_pv=buf, sizeof(buf), &read);
 547             CHECK_CALLED(Read);
 548         }while(hres == S_OK);
 549     }
 550
 551     return S_OK;
 552 }
 553
 554 static HRESULT WINAPI ProtocolSink_ReportResult(IInternetProtocolSink *iface, HRESULT hrResult,
 555         DWORD dwError, LPCWSTR szResult)
 556 {
 557     CHECK_EXPECT(ReportResult);
 558
 559     ok(hrResult == expect_hrResult, "hrResult = %08x, expected: %08x\n",
 560             hrResult, expect_hrResult);
 561     if(SUCCEEDED(hrResult))
 562         ok(dwError == ERROR_SUCCESS, "dwError = %d, expected ERROR_SUCCESS\n", dwError);
 563     else
 564         ok(dwError != ERROR_SUCCESS, "dwError == ERROR_SUCCESS\n");
 565     ok(!szResult, "szResult != NULL\n");
 566
 567     return S_OK;
 568 }
 569
 570 static IInternetProtocolSinkVtbl protocol_sink_vtbl = {
 571     ProtocolSink_QueryInterface,
 572     ProtocolSink_AddRef,
 573     ProtocolSink_Release,
 574     ProtocolSink_Switch,
 575     ProtocolSink_ReportProgress,
 576     ProtocolSink_ReportData,
 577     ProtocolSink_ReportResult
 578 };
 579
 580 static IInternetProtocolSink protocol_sink = { &protocol_sink_vtbl };
 581
 582 static HRESULT QueryInterface(REFIID riid, void **ppv)
 583 {
 584     *ppv = NULL;
 585
 586     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocolSink, riid))
 587         *ppv = &protocol_sink;
 588     if(IsEqualGUID(&IID_IServiceProvider, riid))
 589         *ppv = &service_provider;
 590
 591     if(*ppv)
 592         return S_OK;
 593
 594     return E_NOINTERFACE;
 595 }
 596
 597 static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface, REFIID riid, void **ppv)
 598 {
 599     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetBindInfo, riid)) {
 600         *ppv = iface;
 601         return S_OK;
 602     }
 603     return E_NOINTERFACE;
 604 }
 605
 606 static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface)
 607 {
 608     return 2;
 609 }
 610
 611 static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface)
 612 {
 613     return 1;
 614 }
 615
 616 static HRESULT WINAPI BindInfo_GetBindInfo(IInternetBindInfo *iface, DWORD *grfBINDF, BINDINFO *pbindinfo)
 617 {
 618     DWORD cbSize;
 619
 620     static const CHAR szPostData[] = "mode=Test";
 621
 622     CHECK_EXPECT(GetBindInfo);
 623
 624     ok(grfBINDF != NULL, "grfBINDF == NULL\n");
 625     ok(pbindinfo != NULL, "pbindinfo == NULL\n");
 626     ok(pbindinfo->cbSize == sizeof(BINDINFO), "wrong size of pbindinfo: %d\n", pbindinfo->cbSize);
 627
 628     *grfBINDF = bindf;
 629     if(binding_test)
 630         *grfBINDF |= BINDF_FROMURLMON;
 631     cbSize = pbindinfo->cbSize;
 632     memset(pbindinfo, 0, cbSize);
 633     pbindinfo->cbSize = cbSize;
 634
 635     if (http_post_test)
 636     {
 637         /* Must be GMEM_FIXED, GMEM_MOVABLE does not work properly
 638          * with urlmon on native (Win98 and WinXP) */
 639         U(pbindinfo->stgmedData).hGlobal = GlobalAlloc(GPTR, sizeof(szPostData));
 640         if (!U(pbindinfo->stgmedData).hGlobal)
 641         {
 642             http_post_test = FALSE;
 643             skip("Out of memory\n");
 644             return E_OUTOFMEMORY;
 645         }
 646         lstrcpy((LPSTR)U(pbindinfo->stgmedData).hGlobal, szPostData);
 647         pbindinfo->cbstgmedData = sizeof(szPostData)-1;
 648         pbindinfo->dwBindVerb = BINDVERB_POST;
 649         pbindinfo->stgmedData.tymed = TYMED_HGLOBAL;
 650     }
 651
 652     return S_OK;
 653 }
 654
 655 static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface, ULONG ulStringType,
 656         LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
 657 {
 658     ok(ppwzStr != NULL, "ppwzStr == NULL\n");
 659     ok(pcElFetched != NULL, "pcElFetched == NULL\n");
 660
 661     switch(ulStringType) {
 662     case BINDSTRING_ACCEPT_MIMES:
 663         CHECK_EXPECT(GetBindString_ACCEPT_MIMES);
 664         ok(cEl == 256, "cEl=%d, expected 256\n", cEl);
 665         if(pcElFetched) {
 666             ok(*pcElFetched == 256, "*pcElFetched=%d, expected 256\n", *pcElFetched);
 667             *pcElFetched = 1;
 668         }
 669         if(ppwzStr) {
 670             *ppwzStr = CoTaskMemAlloc(sizeof(acc_mimeW));
 671             memcpy(*ppwzStr, acc_mimeW, sizeof(acc_mimeW));
 672         }
 673         return S_OK;
 674     case BINDSTRING_USER_AGENT:
 675         CHECK_EXPECT(GetBindString_USER_AGENT);
 676         ok(cEl == 1, "cEl=%d, expected 1\n", cEl);
 677         if(pcElFetched) {
 678             ok(*pcElFetched == 0, "*pcElFetch=%d, expectd 0\n", *pcElFetched);
 679             *pcElFetched = 1;
 680         }
 681         if(ppwzStr) {
 682             *ppwzStr = CoTaskMemAlloc(sizeof(user_agentW));
 683             memcpy(*ppwzStr, user_agentW, sizeof(user_agentW));
 684         }
 685         return S_OK;
 686     case BINDSTRING_POST_COOKIE:
 687         CHECK_EXPECT(GetBindString_POST_COOKIE);
 688         ok(cEl == 1, "cEl=%d, expected 1\n", cEl);
 689         if(pcElFetched)
 690             ok(*pcElFetched == 0, "*pcElFetch=%d, expectd 0\n", *pcElFetched);
 691         return S_OK;
 692     default:
 693         ok(0, "unexpected call\n");
 694     }
 695
 696     return E_NOTIMPL;
 697 }
 698
 699 static IInternetBindInfoVtbl bind_info_vtbl = {
 700     BindInfo_QueryInterface,
 701     BindInfo_AddRef,
 702     BindInfo_Release,
 703     BindInfo_GetBindInfo,
 704     BindInfo_GetBindString
 705 };
 706
 707 static IInternetBindInfo bind_info = { &bind_info_vtbl };
 708
 709 static HRESULT WINAPI InternetPriority_QueryInterface(IInternetPriority *iface,
 710                                                   REFIID riid, void **ppv)
 711 {
 712     ok(0, "unexpected call\n");
 713     return E_NOINTERFACE;
 714 }
 715
 716 static ULONG WINAPI InternetPriority_AddRef(IInternetPriority *iface)
 717 {
 718     return 2;
 719 }
 720
 721 static ULONG WINAPI InternetPriority_Release(IInternetPriority *iface)
 722 {
 723     return 1;
 724 }
 725
 726 static HRESULT WINAPI InternetPriority_SetPriority(IInternetPriority *iface, LONG nPriority)
 727 {
 728     CHECK_EXPECT(SetPriority);
 729     ok(nPriority == ex_priority, "nPriority=%d\n", nPriority);
 730     return S_OK;
 731 }
 732
 733 static HRESULT WINAPI InternetPriority_GetPriority(IInternetPriority *iface, LONG *pnPriority)
 734 {
 735     ok(0, "unexpected call\n");
 736     return E_NOTIMPL;
 737 }
 738
 739
 740 static const IInternetPriorityVtbl InternetPriorityVtbl = {
 741     InternetPriority_QueryInterface,
 742     InternetPriority_AddRef,
 743     InternetPriority_Release,
 744     InternetPriority_SetPriority,
 745     InternetPriority_GetPriority
 746 };
 747
 748 static IInternetPriority InternetPriority = { &InternetPriorityVtbl };
 749
 750 static HRESULT WINAPI Protocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
 751 {
 752     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetProtocol, riid)) {
 753         *ppv = iface;
 754         return S_OK;
 755     }
 756
 757     if(IsEqualGUID(&IID_IInternetPriority, riid)) {
 758         *ppv = &InternetPriority;
 759         return S_OK;
 760     }
 761
 762     ok(0, "unexpected riid %s\n", debugstr_guid(riid));
 763     *ppv = NULL;
 764     return E_NOINTERFACE;
 765 }
 766
 767 static ULONG WINAPI Protocol_AddRef(IInternetProtocol *iface)
 768 {
 769     return 2;
 770 }
 771
 772 static ULONG WINAPI Protocol_Release(IInternetProtocol *iface)
 773 {
 774     return 1;
 775 }
 776
 777 static DWORD WINAPI thread_proc(PVOID arg)
 778 {
 779     HRESULT hres;
 780
 781     memset(&protocoldata, -1, sizeof(protocoldata));
 782
 783     prot_state = 0;
 784
 785     SET_EXPECT(ReportProgress_FINDINGRESOURCE);
 786     hres = IInternetProtocolSink_ReportProgress(binding_sink,
 787             BINDSTATUS_FINDINGRESOURCE, hostW);
 788     CHECK_CALLED(ReportProgress_FINDINGRESOURCE);
 789     ok(hres == S_OK, "ReportProgress failed: %08x\n", hres);
 790
 791     SET_EXPECT(ReportProgress_CONNECTING);
 792     hres = IInternetProtocolSink_ReportProgress(binding_sink,
 793             BINDSTATUS_CONNECTING, winehq_ipW);
 794     CHECK_CALLED(ReportProgress_CONNECTING);
 795     ok(hres == S_OK, "ReportProgress failed: %08x\n", hres);
 796
 797     SET_EXPECT(ReportProgress_SENDINGREQUEST);
 798     hres = IInternetProtocolSink_ReportProgress(binding_sink,
 799             BINDSTATUS_SENDINGREQUEST, NULL);
 800     CHECK_CALLED(ReportProgress_SENDINGREQUEST);
 801     ok(hres == S_OK, "ReportProgress failed: %08x\n", hres);
 802
 803     prot_state = 1;
 804     SET_EXPECT(Switch);
 805     hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata);
 806     CHECK_CALLED(Switch);
 807     ok(hres == S_OK, "Switch failed: %08x\n", hres);
 808
 809     prot_state = 2;
 810     SET_EXPECT(Switch);
 811     hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata);
 812     CHECK_CALLED(Switch);
 813     ok(hres == S_OK, "Switch failed: %08x\n", hres);
 814
 815     prot_state = 2;
 816     SET_EXPECT(Switch);
 817     hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata);
 818     CHECK_CALLED(Switch);
 819     ok(hres == S_OK, "Switch failed: %08x\n", hres);
 820
 821     prot_state = 3;
 822     SET_EXPECT(Switch);
 823     hres = IInternetProtocolSink_Switch(binding_sink, &protocoldata);
 824     CHECK_CALLED(Switch);
 825     ok(hres == S_OK, "Switch failed: %08x\n", hres);
 826
 827     SetEvent(event_complete);
 828
 829     return 0;
 830 }
 831
 832 static HRESULT WINAPI Protocol_Start(IInternetProtocol *iface, LPCWSTR szUrl,
 833         IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
 834         DWORD grfPI, DWORD dwReserved)
 835 {
 836     BINDINFO bindinfo, exp_bindinfo;
 837     DWORD cbindf = 0;
 838     HRESULT hres;
 839
 840     CHECK_EXPECT(Start);
 841
 842     ok(pOIProtSink != NULL, "pOIProtSink == NULL\n");
 843     ok(pOIBindInfo != NULL, "pOIBindInfo == NULL\n");
 844     ok(pOIProtSink != &protocol_sink, "unexpected pOIProtSink\n");
 845     ok(pOIBindInfo != &bind_info, "unexpected pOIBindInfo\n");
 846     ok(!grfPI, "grfPI = %x\n", grfPI);
 847     ok(!dwReserved, "dwReserved = %d\n", dwReserved);
 848
 849     memset(&bindinfo, 0, sizeof(bindinfo));
 850     bindinfo.cbSize = sizeof(bindinfo);
 851     memcpy(&exp_bindinfo, &bindinfo, sizeof(bindinfo));
 852     SET_EXPECT(GetBindInfo);
 853     hres = IInternetBindInfo_GetBindInfo(pOIBindInfo, &cbindf, &bindinfo);
 854     ok(hres == S_OK, "GetBindInfo failed: %08x\n", hres);
 855     CHECK_CALLED(GetBindInfo);
 856     ok(cbindf == (bindf|BINDF_FROMURLMON), "bindf = %x, expected %x\n",
 857        cbindf, (bindf|BINDF_FROMURLMON));
 858     ok(!memcmp(&exp_bindinfo, &bindinfo, sizeof(bindinfo)), "unexpected bindinfo\n");
 859     ReleaseBindInfo(&bindinfo);
 860
 861     SET_EXPECT(ReportProgress_SENDINGREQUEST);
 862     hres = IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_SENDINGREQUEST, emptyW);
 863     ok(hres == S_OK, "ReportProgress(BINDSTATUS_SENDINGREQUEST) failed: %08x\n", hres);
 864     CHECK_CALLED(ReportProgress_SENDINGREQUEST);
 865
 866     if(tested_protocol == HTTP_TEST) {
 867         IServiceProvider *service_provider;
 868         IHttpNegotiate *http_negotiate;
 869         IHttpNegotiate2 *http_negotiate2;
 870         LPWSTR ua = (LPWSTR)0xdeadbeef, accept_mimes[256];
 871         LPWSTR additional_headers = NULL;
 872         BYTE sec_id[100];
 873         DWORD fetched = 0, size = 100;
 874         DWORD tid;
 875
 876         SET_EXPECT(GetBindString_USER_AGENT);
 877         hres = IInternetBindInfo_GetBindString(pOIBindInfo, BINDSTRING_USER_AGENT,
 878                                                &ua, 1, &fetched);
 879         CHECK_CALLED(GetBindString_USER_AGENT);
 880         ok(hres == S_OK, "GetBindString(BINDSTRING_USER_AGETNT) failed: %08x\n", hres);
 881         ok(fetched == 1, "fetched = %d, expected 254\n", fetched);
 882         ok(ua != NULL, "ua =  %p\n", ua);
 883         ok(!lstrcmpW(ua, user_agentW), "unexpected user agent %s\n", debugstr_w(ua));
 884         CoTaskMemFree(ua);
 885
 886         fetched = 256;
 887         SET_EXPECT(GetBindString_ACCEPT_MIMES);
 888         hres = IInternetBindInfo_GetBindString(pOIBindInfo, BINDSTRING_ACCEPT_MIMES,
 889                                                accept_mimes, 256, &fetched);
 890         CHECK_CALLED(GetBindString_ACCEPT_MIMES);
 891
 892         ok(hres == S_OK,
 893            "GetBindString(BINDSTRING_ACCEPT_MIMES) failed: %08x\n", hres);
 894         ok(fetched == 1, "fetched = %d, expected 1\n", fetched);
 895         ok(!lstrcmpW(acc_mimeW, accept_mimes[0]), "unexpected mimes %s\n", debugstr_w(accept_mimes[0]));
 896
 897         hres = IInternetBindInfo_QueryInterface(pOIBindInfo, &IID_IServiceProvider,
 898                                                 (void**)&service_provider);
 899         ok(hres == S_OK, "QueryInterface failed: %08x\n", hres);
 900
 901         SET_EXPECT(QueryService_HttpNegotiate);
 902         hres = IServiceProvider_QueryService(service_provider, &IID_IHttpNegotiate,
 903                 &IID_IHttpNegotiate, (void**)&http_negotiate);
 904         CHECK_CALLED(QueryService_HttpNegotiate);
 905         ok(hres == S_OK, "QueryService failed: %08x\n", hres);
 906
 907         SET_EXPECT(BeginningTransaction);
 908         hres = IHttpNegotiate_BeginningTransaction(http_negotiate, binding_urls[tested_protocol],
 909                                                    NULL, 0, &additional_headers);
 910         CHECK_CALLED(BeginningTransaction);
 911         IHttpNegotiate_Release(http_negotiate);
 912         ok(hres == S_OK, "BeginningTransction failed: %08x\n", hres);
 913         ok(additional_headers == NULL, "additional_headers=%p\n", additional_headers);
 914
 915         SET_EXPECT(QueryService_HttpNegotiate);
 916         hres = IServiceProvider_QueryService(service_provider, &IID_IHttpNegotiate2,
 917                 &IID_IHttpNegotiate2, (void**)&http_negotiate2);
 918         CHECK_CALLED(QueryService_HttpNegotiate);
 919         ok(hres == S_OK, "QueryService failed: %08x\n", hres);
 920
 921         size = 512;
 922         SET_EXPECT(GetRootSecurityId);
 923         hres = IHttpNegotiate2_GetRootSecurityId(http_negotiate2, sec_id, &size, 0);
 924         CHECK_CALLED(GetRootSecurityId);
 925         IHttpNegotiate2_Release(http_negotiate2);
 926         ok(hres == E_FAIL, "GetRootSecurityId failed: %08x, expected E_FAIL\n", hres);
 927         ok(size == 13, "size=%d\n", size);
 928
 929         IServiceProvider_Release(service_provider);
 930
 931         CreateThread(NULL, 0, thread_proc, NULL, 0, &tid);
 932
 933         return S_OK;
 934     }
 935
 936     SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE);
 937     hres = IInternetProtocolSink_ReportProgress(pOIProtSink,
 938             BINDSTATUS_CACHEFILENAMEAVAILABLE, expect_wsz = emptyW);
 939     ok(hres == S_OK, "ReportProgress(BINDSTATUS_CACHEFILENAMEAVAILABLE) failed: %08x\n", hres);
 940     CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE);
 941
 942     SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
 943     hres = IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE,
 944                                                 expect_wsz = text_htmlW);
 945     ok(hres == S_OK,
 946        "ReportProgress(BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE) failed: %08x\n", hres);
 947     CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE);
 948
 949     SET_EXPECT(ReportData);
 950     hres = IInternetProtocolSink_ReportData(pOIProtSink,
 951             BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION, 13, 13);
 952     ok(hres == S_OK, "ReportData failed: %08x\n", hres);
 953     CHECK_CALLED(ReportData);
 954
 955     if(tested_protocol == BIND_TEST) {
 956         hres = IInternetProtocol_Terminate(binding_protocol, 0);
 957         ok(hres == E_FAIL, "Termiante failed: %08x\n", hres);
 958     }
 959
 960     SET_EXPECT(ReportResult);
 961     hres = IInternetProtocolSink_ReportResult(pOIProtSink, S_OK, 0, NULL);
 962     ok(hres == S_OK, "ReportResult failed: %08x\n", hres);
 963     CHECK_CALLED(ReportResult);
 964
 965     return S_OK;
 966 }
 967
 968 static HRESULT WINAPI Protocol_Continue(IInternetProtocol *iface,
 969         PROTOCOLDATA *pProtocolData)
 970 {
 971     DWORD bscf = 0;
 972     HRESULT hres;
 973
 974     CHECK_EXPECT(Continue);
 975
 976     ok(pProtocolData != NULL, "pProtocolData == NULL\n");
 977     if(!pProtocolData || tested_protocol == BIND_TEST)
 978         return S_OK;
 979
 980     switch(prot_state) {
 981     case 1: {
 982         IServiceProvider *service_provider;
 983         IHttpNegotiate *http_negotiate;
 984         static WCHAR header[] = {'?',0};
 985
 986         hres = IInternetProtocolSink_QueryInterface(binding_sink, &IID_IServiceProvider,
 987                                                     (void**)&service_provider);
 988         ok(hres == S_OK, "Could not get IServiceProvicder\n");
 989
 990         SET_EXPECT(QueryService_HttpNegotiate);
 991         hres = IServiceProvider_QueryService(service_provider, &IID_IHttpNegotiate,
 992                                              &IID_IHttpNegotiate, (void**)&http_negotiate);
 993         IServiceProvider_Release(service_provider);
 994         CHECK_CALLED(QueryService_HttpNegotiate);
 995         ok(hres == S_OK, "Could not get IHttpNegotiate\n");
 996
 997         SET_EXPECT(OnResponse);
 998         hres = IHttpNegotiate_OnResponse(http_negotiate, 200, header, NULL, NULL);
 999         IHttpNegotiate_Release(http_negotiate);
1000         CHECK_CALLED(OnResponse);
1001         IHttpNegotiate_Release(http_negotiate);
1002         ok(hres == S_OK, "OnResponse failed: %08x\n", hres);
1003
1004         SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
1005         hres = IInternetProtocolSink_ReportProgress(binding_sink,
1006                 BINDSTATUS_MIMETYPEAVAILABLE, text_htmlW);
1007         CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE);
1008         ok(hres == S_OK,
1009            "ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE) failed: %08x\n", hres);
1010
1011         bscf |= BSCF_FIRSTDATANOTIFICATION;
1012         break;
1013     }
1014     case 2:
1015     case 3:
1016         bscf = BSCF_INTERMEDIATEDATANOTIFICATION;
1017         break;
1018     }
1019
1020     SET_EXPECT(ReportData);
1021     hres = IInternetProtocolSink_ReportData(binding_sink, bscf, 100, 400);
1022     CHECK_CALLED(ReportData);
1023     ok(hres == S_OK, "ReportData failed: %08x\n", hres);
1024
1025     if(prot_state == 3)
1026         prot_state = 4;
1027
1028     return S_OK;
1029 }
1030
1031 static HRESULT WINAPI Protocol_Abort(IInternetProtocol *iface, HRESULT hrReason,
1032         DWORD dwOptions)
1033 {
1034     ok(0, "unexpected call\n");
1035     return E_NOTIMPL;
1036 }
1037
1038 static HRESULT WINAPI Protocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
1039 {
1040     CHECK_EXPECT(Terminate);
1041     ok(!dwOptions, "dwOptions=%d\n", dwOptions);
1042     return S_OK;
1043 }
1044
1045 static HRESULT WINAPI Protocol_Suspend(IInternetProtocol *iface)
1046 {
1047     ok(0, "unexpected call\n");
1048     return E_NOTIMPL;
1049 }
1050
1051 static HRESULT WINAPI Protocol_Resume(IInternetProtocol *iface)
1052 {
1053     ok(0, "unexpected call\n");
1054     return E_NOTIMPL;
1055 }
1056
1057 static HRESULT WINAPI Protocol_Read(IInternetProtocol *iface, void *pv,
1058         ULONG cb, ULONG *pcbRead)
1059 {
1060     static BOOL b = TRUE;
1061
1062     CHECK_EXPECT(Read);
1063
1064     ok(pv == expect_pv, "pv != expect_pv\n");
1065     ok(cb == 1000, "cb=%d\n", cb);
1066     ok(pcbRead != NULL, "pcbRead == NULL\n");
1067     ok(!*pcbRead, "*pcbRead = %d\n", *pcbRead);
1068
1069     if(prot_state == 3) {
1070         HRESULT hres;
1071
1072         SET_EXPECT(ReportResult);
1073         hres = IInternetProtocolSink_ReportResult(binding_sink, S_OK, ERROR_SUCCESS, NULL);
1074         CHECK_CALLED(ReportResult);
1075
1076         return S_FALSE;
1077     }
1078
1079     if((b = !b))
1080         return tested_protocol == HTTP_TEST ? E_PENDING : S_FALSE;
1081
1082     memset(pv, 'x', 100);
1083     prot_read += *pcbRead = 100;
1084     return S_OK;
1085 }
1086
1087 static HRESULT WINAPI Protocol_Seek(IInternetProtocol *iface,
1088         LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
1089 {
1090     ok(0, "unexpected call\n");
1091     return E_NOTIMPL;
1092 }
1093
1094 static HRESULT WINAPI Protocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
1095 {
1096     CHECK_EXPECT(LockRequest);
1097     ok(dwOptions == 0, "dwOptions=%x\n", dwOptions);
1098     return S_OK;
1099 }
1100
1101 static HRESULT WINAPI Protocol_UnlockRequest(IInternetProtocol *iface)
1102 {
1103     CHECK_EXPECT(UnlockRequest);
1104     return S_OK;
1105 }
1106
1107 static const IInternetProtocolVtbl ProtocolVtbl = {
1108     Protocol_QueryInterface,
1109     Protocol_AddRef,
1110     Protocol_Release,
1111     Protocol_Start,
1112     Protocol_Continue,
1113     Protocol_Abort,
1114     Protocol_Terminate,
1115     Protocol_Suspend,
1116     Protocol_Resume,
1117     Protocol_Read,
1118     Protocol_Seek,
1119     Protocol_LockRequest,
1120     Protocol_UnlockRequest
1121 };
1122
1123 static IInternetProtocol Protocol = { &ProtocolVtbl };
1124
1125 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
1126 {
1127     ok(0, "unexpected call\n");
1128     return E_NOINTERFACE;
1129 }
1130
1131 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
1132 {
1133     return 2;
1134 }
1135
1136 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
1137 {
1138     return 1;
1139 }
1140
1141 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
1142                                         REFIID riid, void **ppv)
1143 {
1144     CHECK_EXPECT(CreateInstance);
1145
1146     ok(pOuter == (IUnknown*)prot_bind_info, "pOuter != protocol_unk\n");
1147     ok(IsEqualGUID(&IID_IUnknown, riid), "unexpected riid %s\n", debugstr_guid(riid));
1148     ok(ppv != NULL, "ppv == NULL\n");
1149
1150     *ppv = &Protocol;
1151     return S_OK;
1152 }
1153
1154 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
1155 {
1156     ok(0, "unexpected call\n");
1157     return S_OK;
1158 }
1159
1160 static const IClassFactoryVtbl ClassFactoryVtbl = {
1161     ClassFactory_QueryInterface,
1162     ClassFactory_AddRef,
1163     ClassFactory_Release,
1164     ClassFactory_CreateInstance,
1165     ClassFactory_LockServer
1166 };
1167
1168 static IClassFactory ClassFactory = { &ClassFactoryVtbl };
1169
1170 static void test_priority(IInternetProtocol *protocol)
1171 {
1172     IInternetPriority *priority;
1173     LONG pr;
1174     HRESULT hres;
1175
1176     hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority,
1177                                             (void**)&priority);
1178     ok(hres == S_OK, "QueryInterface(IID_IInternetPriority) failed: %08x\n", hres);
1179     if(FAILED(hres))
1180         return;
1181
1182     hres = IInternetPriority_GetPriority(priority, &pr);
1183     ok(hres == S_OK, "GetPriority failed: %08x\n", hres);
1184     ok(pr == 0, "pr=%d, expected 0\n", pr);
1185
1186     hres = IInternetPriority_SetPriority(priority, 1);
1187     ok(hres == S_OK, "SetPriority failed: %08x\n", hres);
1188
1189     hres = IInternetPriority_GetPriority(priority, &pr);
1190     ok(hres == S_OK, "GetPriority failed: %08x\n", hres);
1191     ok(pr == 1, "pr=%d, expected 1\n", pr);
1192
1193     IInternetPriority_Release(priority);
1194 }
1195
1196 static void file_protocol_start(IInternetProtocol *protocol, LPCWSTR url, BOOL is_first)
1197 {
1198     HRESULT hres;
1199
1200     SET_EXPECT(GetBindInfo);
1201     if(!(bindf & BINDF_FROMURLMON))
1202        SET_EXPECT(ReportProgress_DIRECTBIND);
1203     if(is_first) {
1204         SET_EXPECT(ReportProgress_SENDINGREQUEST);
1205         SET_EXPECT(ReportProgress_CACHEFILENAMEAVAILABLE);
1206         if(bindf & BINDF_FROMURLMON)
1207             SET_EXPECT(ReportProgress_VERIFIEDMIMETYPEAVAILABLE);
1208         else
1209             SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
1210     }
1211     SET_EXPECT(ReportData);
1212     if(is_first)
1213         SET_EXPECT(ReportResult);
1214
1215     expect_hrResult = S_OK;
1216
1217     hres = IInternetProtocol_Start(protocol, url, &protocol_sink, &bind_info, 0, 0);
1218     ok(hres == S_OK, "Start failed: %08x\n", hres);
1219
1220     CHECK_CALLED(GetBindInfo);
1221     if(!(bindf & BINDF_FROMURLMON))
1222        CHECK_CALLED(ReportProgress_DIRECTBIND);
1223     if(is_first) {
1224         CHECK_CALLED(ReportProgress_SENDINGREQUEST);
1225         CHECK_CALLED(ReportProgress_CACHEFILENAMEAVAILABLE);
1226         if(bindf & BINDF_FROMURLMON)
1227             CHECK_CALLED(ReportProgress_VERIFIEDMIMETYPEAVAILABLE);
1228         else
1229             CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE);
1230     }
1231     CHECK_CALLED(ReportData);
1232     if(is_first)
1233         CHECK_CALLED(ReportResult);
1234 }
1235
1236 static void test_file_protocol_url(LPCWSTR url)
1237 {
1238     IInternetProtocolInfo *protocol_info;
1239     IUnknown *unk;
1240     IClassFactory *factory;
1241     HRESULT hres;
1242
1243     hres = CoGetClassObject(&CLSID_FileProtocol, CLSCTX_INPROC_SERVER, NULL,
1244             &IID_IUnknown, (void**)&unk);
1245     ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
1246     if(!SUCCEEDED(hres))
1247         return;
1248
1249     hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
1250     ok(hres == E_NOINTERFACE,
1251             "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n", hres);
1252
1253     hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
1254     ok(hres == S_OK, "Could not get IClassFactory interface\n");
1255     if(SUCCEEDED(hres)) {
1256         IInternetProtocol *protocol;
1257         BYTE buf[512];
1258         ULONG cb;
1259         hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol);
1260         ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
1261
1262         if(SUCCEEDED(hres)) {
1263             file_protocol_start(protocol, url, TRUE);
1264             hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
1265             ok(hres == S_OK, "Read failed: %08x\n", hres);
1266             ok(cb == 2, "cb=%u expected 2\n", cb);
1267             hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
1268             ok(hres == S_FALSE, "Read failed: %08x\n", hres);
1269             hres = IInternetProtocol_Read(protocol, buf, sizeof(buf), &cb);
1270             ok(hres == S_FALSE, "Read failed: %08x expected S_FALSE\n", hres);
1271             ok(cb == 0, "cb=%u expected 0\n", cb);
1272             hres = IInternetProtocol_UnlockRequest(protocol);
1273             ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
1274
1275             file_protocol_start(protocol, url, FALSE);
1276             hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
1277             ok(hres == S_FALSE, "Read failed: %08x\n", hres);
1278             hres = IInternetProtocol_LockRequest(protocol, 0);
1279             ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
1280             hres = IInternetProtocol_UnlockRequest(protocol);
1281             ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
1282
1283             IInternetProtocol_Release(protocol);
1284         }
1285
1286         hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol);
1287         ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
1288
1289         if(SUCCEEDED(hres)) {
1290             file_protocol_start(protocol, url, TRUE);
1291             hres = IInternetProtocol_LockRequest(protocol, 0);
1292             ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
1293             hres = IInternetProtocol_Terminate(protocol, 0);
1294             ok(hres == S_OK, "Terminate failed: %08x\n", hres);
1295             hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
1296             ok(hres == S_OK, "Read failed: %08x\n\n", hres);
1297             hres = IInternetProtocol_UnlockRequest(protocol);
1298             ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
1299             hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
1300             ok(hres == S_OK, "Read failed: %08x\n", hres);
1301             hres = IInternetProtocol_Terminate(protocol, 0);
1302             ok(hres == S_OK, "Terminate failed: %08x\n", hres);
1303
1304             IInternetProtocol_Release(protocol);
1305         }
1306
1307         hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol, (void**)&protocol);
1308         ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
1309
1310         if(SUCCEEDED(hres)) {
1311             file_protocol_start(protocol, url, TRUE);
1312             hres = IInternetProtocol_Terminate(protocol, 0);
1313             ok(hres == S_OK, "Terminate failed: %08x\n", hres);
1314             hres = IInternetProtocol_Read(protocol, buf, 2, &cb);
1315             ok(hres == S_OK, "Read failed: %08x\n", hres);
1316             ok(cb == 2, "cb=%u expected 2\n", cb);
1317
1318             IInternetProtocol_Release(protocol);
1319         }
1320
1321         IClassFactory_Release(factory);
1322     }
1323
1324     IUnknown_Release(unk);
1325 }
1326
1327 static void test_file_protocol_fail(void)
1328 {
1329     IInternetProtocol *protocol;
1330     HRESULT hres;
1331
1332     static const WCHAR index_url2[] =
1333         {'f','i','l','e',':','/','/','i','n','d','e','x','.','h','t','m','l',0};
1334
1335     hres = CoCreateInstance(&CLSID_FileProtocol, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
1336             &IID_IInternetProtocol, (void**)&protocol);
1337     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
1338     if(FAILED(hres))
1339         return;
1340
1341     SET_EXPECT(GetBindInfo);
1342     expect_hrResult = MK_E_SYNTAX;
1343     hres = IInternetProtocol_Start(protocol, wszIndexHtml, &protocol_sink, &bind_info, 0, 0);
1344     ok(hres == MK_E_SYNTAX ||
1345        hres == E_INVALIDARG,
1346        "Start failed: %08x, expected MK_E_SYNTAX or E_INVALIDARG\n", hres);
1347     CLEAR_CALLED(GetBindInfo); /* GetBindInfo not called in IE7 */
1348
1349     SET_EXPECT(GetBindInfo);
1350     if(!(bindf & BINDF_FROMURLMON))
1351         SET_EXPECT(ReportProgress_DIRECTBIND);
1352     SET_EXPECT(ReportProgress_SENDINGREQUEST);
1353     SET_EXPECT(ReportResult);
1354     expect_hrResult = INET_E_RESOURCE_NOT_FOUND;
1355     hres = IInternetProtocol_Start(protocol, index_url, &protocol_sink, &bind_info, 0, 0);
1356     ok(hres == INET_E_RESOURCE_NOT_FOUND,
1357             "Start failed: %08x expected INET_E_RESOURCE_NOT_FOUND\n", hres);
1358     CHECK_CALLED(GetBindInfo);
1359     if(!(bindf & BINDF_FROMURLMON))
1360         CHECK_CALLED(ReportProgress_DIRECTBIND);
1361     CHECK_CALLED(ReportProgress_SENDINGREQUEST);
1362     CHECK_CALLED(ReportResult);
1363
1364     IInternetProtocol_Release(protocol);
1365
1366     hres = CoCreateInstance(&CLSID_FileProtocol, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
1367             &IID_IInternetProtocol, (void**)&protocol);
1368     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
1369     if(FAILED(hres))
1370         return;
1371
1372     SET_EXPECT(GetBindInfo);
1373     if(!(bindf & BINDF_FROMURLMON))
1374         SET_EXPECT(ReportProgress_DIRECTBIND);
1375     SET_EXPECT(ReportProgress_SENDINGREQUEST);
1376     SET_EXPECT(ReportResult);
1377     expect_hrResult = INET_E_RESOURCE_NOT_FOUND;
1378
1379     hres = IInternetProtocol_Start(protocol, index_url2, &protocol_sink, &bind_info, 0, 0);
1380     ok(hres == INET_E_RESOURCE_NOT_FOUND,
1381             "Start failed: %08x, expected INET_E_RESOURCE_NOT_FOUND\n", hres);
1382     CHECK_CALLED(GetBindInfo);
1383     if(!(bindf & BINDF_FROMURLMON))
1384         CHECK_CALLED(ReportProgress_DIRECTBIND);
1385     CHECK_CALLED(ReportProgress_SENDINGREQUEST);
1386     CHECK_CALLED(ReportResult);
1387
1388     SET_EXPECT(GetBindInfo);
1389     hres = IInternetProtocol_Start(protocol, NULL, &protocol_sink, &bind_info, 0, 0);
1390     ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres);
1391     CLEAR_CALLED(GetBindInfo); /* GetBindInfo not called in IE7 */
1392
1393     SET_EXPECT(GetBindInfo);
1394     hres = IInternetProtocol_Start(protocol, emptyW, &protocol_sink, &bind_info, 0, 0);
1395     ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres);
1396     CLEAR_CALLED(GetBindInfo); /* GetBindInfo not called in IE7 */
1397
1398     IInternetProtocol_Release(protocol);
1399 }
1400
1401 static void test_file_protocol(void) {
1402     WCHAR buf[INTERNET_MAX_URL_LENGTH], file_name_buf[MAX_PATH];
1403     DWORD size;
1404     ULONG len;
1405     HANDLE file;
1406
1407     static const WCHAR wszFile[] = {'f','i','l','e',':',0};
1408     static const WCHAR wszFile2[] = {'f','i','l','e',':','/','/',0};
1409     static const WCHAR wszFile3[] = {'f','i','l','e',':','/','/','/',0};
1410     static const char html_doc[] = "<HTML></HTML>";
1411
1412     trace("Testing file protocol...\n");
1413     tested_protocol = FILE_TEST;
1414
1415     file = CreateFileW(wszIndexHtml, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
1416             FILE_ATTRIBUTE_NORMAL, NULL);
1417     ok(file != INVALID_HANDLE_VALUE, "CreateFile failed\n");
1418     if(file == INVALID_HANDLE_VALUE)
1419         return;
1420     WriteFile(file, html_doc, sizeof(html_doc)-1, &size, NULL);
1421     CloseHandle(file);
1422
1423     file_name = wszIndexHtml;
1424     bindf = 0;
1425     test_file_protocol_url(index_url);
1426     bindf = BINDF_FROMURLMON;
1427     test_file_protocol_url(index_url);
1428     bindf = BINDF_FROMURLMON | BINDF_NEEDFILE;
1429     test_file_protocol_url(index_url);
1430
1431     memcpy(buf, wszFile, sizeof(wszFile));
1432     len = sizeof(wszFile)/sizeof(WCHAR)-1;
1433     len += GetCurrentDirectoryW(sizeof(buf)/sizeof(WCHAR)-len, buf+len);
1434     buf[len++] = '\\';
1435     memcpy(buf+len, wszIndexHtml, sizeof(wszIndexHtml));
1436
1437     file_name = buf + sizeof(wszFile)/sizeof(WCHAR)-1;
1438     bindf = 0;
1439     test_file_protocol_url(buf);
1440     bindf = BINDF_FROMURLMON;
1441     test_file_protocol_url(buf);
1442
1443     memcpy(buf, wszFile2, sizeof(wszFile2));
1444     len = GetCurrentDirectoryW(sizeof(file_name_buf)/sizeof(WCHAR), file_name_buf);
1445     file_name_buf[len++] = '\\';
1446     memcpy(file_name_buf+len, wszIndexHtml, sizeof(wszIndexHtml));
1447     lstrcpyW(buf+sizeof(wszFile2)/sizeof(WCHAR)-1, file_name_buf);
1448     file_name = file_name_buf;
1449     bindf = 0;
1450     test_file_protocol_url(buf);
1451     bindf = BINDF_FROMURLMON;
1452     test_file_protocol_url(buf);
1453
1454     buf[sizeof(wszFile2)/sizeof(WCHAR)] = '|';
1455     test_file_protocol_url(buf);
1456
1457     memcpy(buf, wszFile3, sizeof(wszFile3));
1458     len = sizeof(wszFile3)/sizeof(WCHAR)-1;
1459     len += GetCurrentDirectoryW(sizeof(buf)/sizeof(WCHAR)-len, buf+len);
1460     buf[len++] = '\\';
1461     memcpy(buf+len, wszIndexHtml, sizeof(wszIndexHtml));
1462
1463     file_name = buf + sizeof(wszFile3)/sizeof(WCHAR)-1;
1464     bindf = 0;
1465     test_file_protocol_url(buf);
1466     bindf = BINDF_FROMURLMON;
1467     test_file_protocol_url(buf);
1468
1469     DeleteFileW(wszIndexHtml);
1470
1471     bindf = 0;
1472     test_file_protocol_fail();
1473     bindf = BINDF_FROMURLMON;
1474     test_file_protocol_fail();
1475 }
1476
1477 static BOOL http_protocol_start(LPCWSTR url, BOOL is_first)
1478 {
1479     static BOOL got_user_agent = FALSE;
1480     HRESULT hres;
1481
1482     first_data_notif = TRUE;
1483     state = 0;
1484
1485     SET_EXPECT(GetBindInfo);
1486     if (!(bindf & BINDF_FROMURLMON))
1487         SET_EXPECT(ReportProgress_DIRECTBIND);
1488     SET_EXPECT(GetBindString_USER_AGENT);
1489     SET_EXPECT(GetBindString_ACCEPT_MIMES);
1490     SET_EXPECT(QueryService_HttpNegotiate);
1491     SET_EXPECT(BeginningTransaction);
1492     SET_EXPECT(GetRootSecurityId);
1493     if (http_post_test)
1494         SET_EXPECT(GetBindString_POST_COOKIE);
1495
1496     hres = IInternetProtocol_Start(http_protocol, url, &protocol_sink, &bind_info, 0, 0);
1497     ok(hres == S_OK, "Start failed: %08x\n", hres);
1498     if(FAILED(hres))
1499         return FALSE;
1500
1501     CHECK_CALLED(GetBindInfo);
1502     if (!(bindf & BINDF_FROMURLMON))
1503         CHECK_CALLED(ReportProgress_DIRECTBIND);
1504     if (!got_user_agent)
1505     {
1506         CHECK_CALLED(GetBindString_USER_AGENT);
1507         got_user_agent = TRUE;
1508     }
1509     else todo_wine
1510     {
1511         /* user agent only retrieved once, even with different URLs */
1512         CHECK_NOT_CALLED(GetBindString_USER_AGENT);
1513     }
1514     CHECK_CALLED(GetBindString_ACCEPT_MIMES);
1515     CHECK_CALLED(QueryService_HttpNegotiate);
1516     CHECK_CALLED(BeginningTransaction);
1517     /* GetRootSecurityId called on WinXP but not on Win98 */
1518     CLEAR_CALLED(GetRootSecurityId);
1519     if (http_post_test)
1520         CHECK_CALLED(GetBindString_POST_COOKIE);
1521
1522     return TRUE;
1523 }
1524
1525 /* is_first refers to whether this is the first call to this function
1526  * _for this url_ */
1527 static void test_http_protocol_url(LPCWSTR url, BOOL is_first)
1528 {
1529     IInternetProtocolInfo *protocol_info;
1530     IClassFactory *factory;
1531     IUnknown *unk;
1532     HRESULT hres;
1533
1534     http_url = url;
1535     http_is_first = is_first;
1536
1537     hres = CoGetClassObject(&CLSID_HttpProtocol, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
1538     ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
1539     if(!SUCCEEDED(hres))
1540         return;
1541
1542     hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
1543     ok(hres == E_NOINTERFACE,
1544         "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n",
1545         hres);
1546
1547     hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
1548     ok(hres == S_OK, "Could not get IClassFactory interface\n");
1549     IUnknown_Release(unk);
1550     if(FAILED(hres))
1551         return;
1552
1553     hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol,
1554                                         (void**)&http_protocol);
1555     ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
1556     if(SUCCEEDED(hres)) {
1557         BYTE buf[3600];
1558         DWORD cb;
1559         int *called = (bindf & BINDF_FROMURLMON) ? &called_Switch : &called_ReportData;
1560
1561         test_priority(http_protocol);
1562
1563         SET_EXPECT(ReportProgress_FINDINGRESOURCE);
1564         SET_EXPECT(ReportProgress_CONNECTING);
1565         SET_EXPECT(ReportProgress_SENDINGREQUEST);
1566         if(!(bindf & BINDF_FROMURLMON)) {
1567             SET_EXPECT(OnResponse);
1568             SET_EXPECT(ReportProgress_RAWMIMETYPE);
1569             SET_EXPECT(ReportData);
1570         } else {
1571             SET_EXPECT(Switch);
1572         }
1573
1574         if(!http_protocol_start(url, is_first))
1575             return;
1576
1577         SET_EXPECT(ReportResult);
1578         expect_hrResult = S_OK;
1579
1580         hres = IInternetProtocol_Read(http_protocol, buf, 1, &cb);
1581         ok((!*called && hres == E_PENDING && cb==0) ||
1582            (*called && hres == S_OK && cb==1), "Read failed: %08x (%d bytes)\n", hres, cb);
1583
1584         WaitForSingleObject(event_complete, INFINITE);
1585         if(bindf & BINDF_FROMURLMON)
1586             CHECK_CALLED(Switch);
1587         else
1588             CHECK_CALLED(ReportData);
1589
1590         while(1) {
1591             if(bindf & BINDF_FROMURLMON)
1592                 SET_EXPECT(Switch);
1593             else
1594                 SET_EXPECT(ReportData);
1595             hres = IInternetProtocol_Read(http_protocol, buf, sizeof(buf), &cb);
1596             if(hres == E_PENDING) {
1597                 hres = IInternetProtocol_Read(http_protocol, buf, 1, &cb);
1598                 ok((!*called && hres == E_PENDING && cb==0) ||
1599                    (*called && hres == S_OK && cb==1), "Read failed: %08x (%d bytes)\n", hres, cb);
1600                 WaitForSingleObject(event_complete, INFINITE);
1601                 if(bindf & BINDF_FROMURLMON)
1602                     CHECK_CALLED(Switch);
1603                 else
1604                     CHECK_CALLED(ReportData);
1605             } else {
1606                 if(bindf & BINDF_FROMURLMON)
1607                     CHECK_NOT_CALLED(Switch);
1608                 else
1609                     CHECK_NOT_CALLED(ReportData);
1610                 if(cb == 0) break;
1611             }
1612         }
1613         ok(hres == S_FALSE, "Read failed: %08x\n", hres);
1614         CHECK_CALLED(ReportResult);
1615
1616         hres = IInternetProtocol_LockRequest(http_protocol, 0);
1617         ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
1618
1619         hres = IInternetProtocol_Read(http_protocol, buf, 1, &cb);
1620         ok(hres == S_FALSE, "Read failed: %08x\n", hres);
1621
1622         hres = IInternetProtocol_Terminate(http_protocol, 0);
1623         ok(hres == S_OK, "Terminate failed: %08x\n", hres);
1624
1625         /* This wait is to give the internet handles being freed in Terminate
1626          * enough time to actually terminate in all cases. Internet handles
1627          * terminate asynchronously and native reuses the main InternetOpen
1628          * handle. The only case in which this seems to be necessary is on
1629          * wine with native wininet and urlmon, resulting in the next time
1630          * test_http_protocol_url being called the first data notification actually
1631          * being an extra last data notification from the previous connection
1632          * about once out of every ten times. */
1633         Sleep(100);
1634
1635         hres = IInternetProtocol_UnlockRequest(http_protocol);
1636         ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
1637
1638         IInternetProtocol_Release(http_protocol);
1639     }
1640
1641     IClassFactory_Release(factory);
1642 }
1643
1644 static void test_http_protocol(void)
1645 {
1646     static const WCHAR winehq_url[] =
1647         {'h','t','t','p',':','/','/','w','w','w','.','w','i','n','e','h','q','.',
1648             'o','r','g','/','s','i','t','e','/','a','b','o','u','t',0};
1649     static const WCHAR posttest_url[] =
1650         {'h','t','t','p',':','/','/','c','r','o','s','s','o','v','e','r','.',
1651          'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m','/',
1652          'p','o','s','t','t','e','s','t','.','p','h','p',0};
1653
1654     trace("Testing http protocol (not from urlmon)...\n");
1655     tested_protocol = HTTP_TEST;
1656     bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA;
1657     test_http_protocol_url(winehq_url, TRUE);
1658
1659     trace("Testing http protocol (from urlmon)...\n");
1660     bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON;
1661     test_http_protocol_url(winehq_url, FALSE);
1662
1663     trace("Testing http protocol (to file)...\n");
1664     bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NEEDFILE;
1665     test_http_protocol_url(winehq_url, FALSE);
1666
1667     trace("Testing http protocol (post data)...\n");
1668     http_post_test = TRUE;
1669     /* Without this flag we get a ReportProgress_CACHEFILENAMEAVAILABLE
1670      * notification with BINDVERB_POST */
1671     bindf = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA | BINDF_FROMURLMON | BINDF_NOWRITECACHE;
1672     test_http_protocol_url(posttest_url, TRUE);
1673     http_post_test = FALSE;
1674 }
1675
1676 static void test_mk_protocol(void)
1677 {
1678     IInternetProtocolInfo *protocol_info;
1679     IInternetProtocol *protocol;
1680     IClassFactory *factory;
1681     IUnknown *unk;
1682     HRESULT hres;
1683
1684     static const WCHAR wrong_url1[] = {'t','e','s','t',':','@','M','S','I','T','S','t','o','r','e',
1685                                        ':',':','/','t','e','s','t','.','h','t','m','l',0};
1686     static const WCHAR wrong_url2[] = {'m','k',':','/','t','e','s','t','.','h','t','m','l',0};
1687
1688     trace("Testing mk protocol...\n");
1689     tested_protocol = MK_TEST;
1690
1691     hres = CoGetClassObject(&CLSID_MkProtocol, CLSCTX_INPROC_SERVER, NULL,
1692             &IID_IUnknown, (void**)&unk);
1693     ok(hres == S_OK, "CoGetClassObject failed: %08x\n", hres);
1694
1695     hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocolInfo, (void**)&protocol_info);
1696     ok(hres == E_NOINTERFACE,
1697         "Could not get IInternetProtocolInfo interface: %08x, expected E_NOINTERFACE\n",
1698         hres);
1699
1700     hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&factory);
1701     ok(hres == S_OK, "Could not get IClassFactory interface\n");
1702     IUnknown_Release(unk);
1703     if(FAILED(hres))
1704         return;
1705
1706     hres = IClassFactory_CreateInstance(factory, NULL, &IID_IInternetProtocol,
1707                                         (void**)&protocol);
1708     IClassFactory_Release(factory);
1709     ok(hres == S_OK, "Could not get IInternetProtocol: %08x\n", hres);
1710
1711     SET_EXPECT(GetBindInfo);
1712     hres = IInternetProtocol_Start(protocol, wrong_url1, &protocol_sink, &bind_info, 0, 0);
1713     ok(hres == MK_E_SYNTAX || hres == INET_E_INVALID_URL,
1714        "Start failed: %08x, expected MK_E_SYNTAX or INET_E_INVALID_URL\n", hres);
1715     CLEAR_CALLED(GetBindInfo);
1716
1717     SET_EXPECT(GetBindInfo);
1718     SET_EXPECT(ReportProgress_DIRECTBIND);
1719     SET_EXPECT(ReportProgress_SENDINGREQUEST);
1720     SET_EXPECT(ReportProgress_MIMETYPEAVAILABLE);
1721     SET_EXPECT(ReportResult);
1722     expect_hrResult = INET_E_RESOURCE_NOT_FOUND;
1723
1724     hres = IInternetProtocol_Start(protocol, wrong_url2, &protocol_sink, &bind_info, 0, 0);
1725     ok(hres == INET_E_RESOURCE_NOT_FOUND ||
1726        hres == INET_E_INVALID_URL, /* win2k3 */
1727        "Start failed: %08x, expected INET_E_RESOURCE_NOT_FOUND or INET_E_INVALID_URL\n", hres);
1728
1729     if (hres == INET_E_RESOURCE_NOT_FOUND) {
1730         CHECK_CALLED(GetBindInfo);
1731         CLEAR_CALLED(ReportProgress_DIRECTBIND);
1732         CHECK_CALLED(ReportProgress_SENDINGREQUEST);
1733         CHECK_CALLED(ReportProgress_MIMETYPEAVAILABLE);
1734         CHECK_CALLED(ReportResult);
1735     }else {
1736         CLEAR_CALLED(GetBindInfo);
1737         CLEAR_CALLED(ReportProgress_DIRECTBIND);
1738         CLEAR_CALLED(ReportProgress_SENDINGREQUEST);
1739         CLEAR_CALLED(ReportProgress_MIMETYPEAVAILABLE);
1740         CLEAR_CALLED(ReportResult);
1741     }
1742
1743     IInternetProtocol_Release(protocol);
1744 }
1745
1746 static void test_CreateBinding(void)
1747 {
1748     IInternetProtocolSink *sink;
1749     IInternetProtocol *protocol;
1750     IInternetPriority *priority;
1751     IInternetSession *session;
1752     LONG p;
1753     BYTE buf[1000];
1754     DWORD read;
1755     HRESULT hres;
1756
1757     static const WCHAR test_url[] =
1758         {'t','e','s','t',':','/','/','f','i','l','e','.','h','t','m','l',0};
1759     static const WCHAR wsz_test[] = {'t','e','s','t',0};
1760
1761     trace("Testing CreateBinding...\n");
1762     tested_protocol = BIND_TEST;
1763     binding_test = TRUE;
1764
1765     hres = CoInternetGetSession(0, &session, 0);
1766     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
1767
1768     hres = IInternetSession_RegisterNameSpace(session, &ClassFactory, &IID_NULL, wsz_test, 0, NULL, 0);
1769     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1770
1771     hres = IInternetSession_CreateBinding(session, NULL, test_url, NULL, NULL, &protocol, 0);
1772     binding_protocol = protocol;
1773     ok(hres == S_OK, "CreateBinding failed: %08x\n", hres);
1774     ok(protocol != NULL, "protocol == NULL\n");
1775
1776     hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetBindInfo, (void**)&prot_bind_info);
1777     ok(hres == S_OK, "QueryInterface(IID_IInternetBindInfo) failed: %08x\n", hres);
1778
1779     hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetProtocolSink, (void**)&sink);
1780     ok(hres == S_OK, "Could not get IInternetProtocolSink: %08x\n", hres);
1781
1782     hres = IInternetProtocol_Start(protocol, test_url, NULL, &bind_info, 0, 0);
1783     ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres);
1784     hres = IInternetProtocol_Start(protocol, test_url, &protocol_sink, NULL, 0, 0);
1785     ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres);
1786     hres = IInternetProtocol_Start(protocol, NULL, &protocol_sink, &bind_info, 0, 0);
1787     ok(hres == E_INVALIDARG, "Start failed: %08x, expected E_INVALIDARG\n", hres);
1788
1789     hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority);
1790     ok(hres == S_OK, "QueryInterface(IID_IInternetPriority) failed: %08x\n", hres);
1791
1792     p = 0xdeadbeef;
1793     hres = IInternetPriority_GetPriority(priority, &p);
1794     ok(hres == S_OK, "GetPriority failed: %08x\n", hres);
1795     ok(!p, "p=%d\n", p);
1796
1797     ex_priority = 100;
1798     hres = IInternetPriority_SetPriority(priority, 100);
1799     ok(hres == S_OK, "SetPriority failed: %08x\n", hres);
1800
1801     p = 0xdeadbeef;
1802     hres = IInternetPriority_GetPriority(priority, &p);
1803     ok(hres == S_OK, "GetPriority failed: %08x\n", hres);
1804     ok(p == 100, "p=%d\n", p);
1805
1806     SET_EXPECT(QueryService_InternetProtocol);
1807     SET_EXPECT(CreateInstance);
1808     SET_EXPECT(ReportProgress_PROTOCOLCLASSID);
1809     SET_EXPECT(SetPriority);
1810     SET_EXPECT(Start);
1811
1812     expect_hrResult = S_OK;
1813     hres = IInternetProtocol_Start(protocol, test_url, &protocol_sink, &bind_info, 0, 0);
1814     ok(hres == S_OK, "Start failed: %08x\n", hres);
1815
1816     CHECK_CALLED(QueryService_InternetProtocol);
1817     CHECK_CALLED(CreateInstance);
1818     CHECK_CALLED(ReportProgress_PROTOCOLCLASSID);
1819     CHECK_CALLED(SetPriority);
1820     CHECK_CALLED(Start);
1821
1822     SET_EXPECT(Read);
1823     read = 0xdeadbeef;
1824     hres = IInternetProtocol_Read(protocol, expect_pv = buf, sizeof(buf), &read);
1825     ok(hres == S_OK, "Read failed: %08x\n", hres);
1826     ok(read == 100, "read = %d\n", read);
1827     CHECK_CALLED(Read);
1828
1829     SET_EXPECT(Read);
1830     read = 0xdeadbeef;
1831     hres = IInternetProtocol_Read(protocol, expect_pv = buf, sizeof(buf), &read);
1832     ok(hres == S_FALSE, "Read failed: %08x\n", hres);
1833     ok(!read, "read = %d\n", read);
1834     CHECK_CALLED(Read);
1835
1836     p = 0xdeadbeef;
1837     hres = IInternetPriority_GetPriority(priority, &p);
1838     ok(hres == S_OK, "GetPriority failed: %08x\n", hres);
1839     ok(p == 100, "p=%d\n", p);
1840
1841     hres = IInternetPriority_SetPriority(priority, 101);
1842     ok(hres == S_OK, "SetPriority failed: %08x\n", hres);
1843
1844     SET_EXPECT(Terminate);
1845     hres = IInternetProtocol_Terminate(protocol, 0xdeadbeef);
1846     ok(hres == S_OK, "Terminate failed: %08x\n", hres);
1847     CHECK_CALLED(Terminate);
1848
1849     SET_EXPECT(Continue);
1850     hres = IInternetProtocolSink_Switch(sink, &protocoldata);
1851     ok(hres == S_OK, "Switch failed: %08x\n", hres);
1852     CHECK_CALLED(Continue);
1853
1854     hres = IInternetProtocolSink_ReportProgress(sink,
1855             BINDSTATUS_CACHEFILENAMEAVAILABLE, expect_wsz = emptyW);
1856     ok(hres == S_OK, "ReportProgress(BINDSTATUS_CACHEFILENAMEAVAILABLE) failed: %08x\n", hres);
1857
1858     hres = IInternetProtocolSink_ReportResult(sink, S_OK, ERROR_SUCCESS, NULL);
1859     ok(hres == E_FAIL, "ReportResult failed: %08x, expected E_FAIL\n", hres);
1860
1861     hres = IInternetProtocolSink_ReportData(sink, 0, 0, 0);
1862     ok(hres == S_OK, "ReportData failed: %08x\n", hres);
1863
1864     IInternetProtocolSink_Release(sink);
1865     IInternetPriority_Release(priority);
1866     IInternetBindInfo_Release(prot_bind_info);
1867     IInternetProtocol_Release(protocol);
1868     IInternetSession_Release(session);
1869 }
1870
1871 static void test_binding(int prot)
1872 {
1873     IInternetProtocol *protocol;
1874     IInternetSession *session;
1875     ULONG ref;
1876     HRESULT hres;
1877
1878     trace("Testing %s binding...\n", debugstr_w(protocol_names[prot]));
1879
1880     tested_protocol = prot;
1881     binding_test = TRUE;
1882     first_data_notif = TRUE;
1883     prot_read = 0;
1884
1885     hres = CoInternetGetSession(0, &session, 0);
1886     ok(hres == S_OK, "CoInternetGetSession failed: %08x\n", hres);
1887
1888     hres = IInternetSession_RegisterNameSpace(session, &ClassFactory, &IID_NULL, protocol_names[prot], 0, NULL, 0);
1889     ok(hres == S_OK, "RegisterNameSpace failed: %08x\n", hres);
1890
1891     hres = IInternetSession_CreateBinding(session, NULL, binding_urls[prot], NULL, NULL, &protocol, 0);
1892     binding_protocol = protocol;
1893     IInternetSession_Release(session);
1894     ok(hres == S_OK, "CreateBinding failed: %08x\n", hres);
1895     ok(protocol != NULL, "protocol == NULL\n");
1896
1897     hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetBindInfo, (void**)&prot_bind_info);
1898     ok(hres == S_OK, "QueryInterface(IID_IInternetBindInfo) failed: %08x\n", hres);
1899
1900     hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetProtocolSink, (void**)&binding_sink);
1901     ok(hres == S_OK, "QueryInterface(IID_IInternetProtocolSink) failed: %08x\n", hres);
1902
1903     ex_priority = 0;
1904     SET_EXPECT(QueryService_InternetProtocol);
1905     SET_EXPECT(CreateInstance);
1906     SET_EXPECT(ReportProgress_PROTOCOLCLASSID);
1907     SET_EXPECT(SetPriority);
1908     SET_EXPECT(Start);
1909
1910     expect_hrResult = S_OK;
1911     hres = IInternetProtocol_Start(protocol, binding_urls[prot], &protocol_sink, &bind_info, 0, 0);
1912     ok(hres == S_OK, "Start failed: %08x\n", hres);
1913
1914     CHECK_CALLED(QueryService_InternetProtocol);
1915     CHECK_CALLED(CreateInstance);
1916     CHECK_CALLED(ReportProgress_PROTOCOLCLASSID);
1917     CHECK_CALLED(SetPriority);
1918     CHECK_CALLED(Start);
1919
1920     if(prot == HTTP_TEST) {
1921         while(prot_state < 4) {
1922             WaitForSingleObject(event_complete, INFINITE);
1923             SET_EXPECT(Continue);
1924             IInternetProtocol_Continue(protocol, pdata);
1925             CHECK_CALLED(Continue);
1926             SetEvent(event_complete2);
1927         }
1928
1929         WaitForSingleObject(event_complete, INFINITE);
1930     }else {
1931         SET_EXPECT(LockRequest);
1932         hres = IInternetProtocol_LockRequest(protocol, 0);
1933         ok(hres == S_OK, "LockRequest failed: %08x\n", hres);
1934         CHECK_CALLED(LockRequest);
1935
1936         SET_EXPECT(UnlockRequest);
1937         hres = IInternetProtocol_UnlockRequest(protocol);
1938         ok(hres == S_OK, "UnlockRequest failed: %08x\n", hres);
1939         CHECK_CALLED(UnlockRequest);
1940     }
1941
1942     SET_EXPECT(Terminate);
1943     hres = IInternetProtocol_Terminate(protocol, 0);
1944     ok(hres == S_OK, "Terminate failed: %08x\n", hres);
1945     CHECK_CALLED(Terminate);
1946
1947     IInternetBindInfo_Release(prot_bind_info);
1948     IInternetProtocolSink_Release(binding_sink);
1949     ref = IInternetProtocol_Release(protocol);
1950     ok(!ref, "ref=%u, expected 0\n", ref);
1951 }
1952
1953 START_TEST(protocol)
1954 {
1955     OleInitialize(NULL);
1956
1957     event_complete = CreateEvent(NULL, FALSE, FALSE, NULL);
1958     event_complete2 = CreateEvent(NULL, FALSE, FALSE, NULL);
1959
1960     test_file_protocol();
1961     test_http_protocol();
1962     test_mk_protocol();
1963     test_CreateBinding();
1964     test_binding(FILE_TEST);
1965     test_binding(HTTP_TEST);
1966
1967     CloseHandle(event_complete);
1968     CloseHandle(event_complete2);
1969
1970     OleUninitialize();
1971 }