git.oblomov.eu Git - wine/blob - dlls/mshtml/nsio.c

   1 /*
   2  * Copyright 2006-2010 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 #include "config.h"
  20
  21 #include <stdarg.h>
  22 #include <assert.h>
  23
  24 #define COBJMACROS
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "winreg.h"
  30 #include "ole2.h"
  31 #include "shlguid.h"
  32 #include "wininet.h"
  33 #include "shlwapi.h"
  34
  35 #include "wine/debug.h"
  36
  37 #include "mshtml_private.h"
  38 #include "binding.h"
  39
  40 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  41
  42 #define NS_IOSERVICE_CLASSNAME "nsIOService"
  43 #define NS_IOSERVICE_CONTRACTID "@mozilla.org/network/io-service;1"
  44
  45 static const IID NS_IOSERVICE_CID =
  46     {0x9ac9e770, 0x18bc, 0x11d3, {0x93, 0x37, 0x00, 0x10, 0x4b, 0xa0, 0xfd, 0x40}};
  47 static const IID IID_nsWineURI =
  48     {0x5088272e, 0x900b, 0x11da, {0xc6,0x87, 0x00,0x0f,0xea,0x57,0xf2,0x1a}};
  49
  50 static nsIIOService *nsio = NULL;
  51 static nsINetUtil *net_util;
  52
  53 static const char *request_method_strings[] = {"GET", "PUT", "POST"};
  54
  55 struct  nsWineURI {
  56     nsIURL nsIURL_iface;
  57     nsIStandardURL nsIStandardURL_iface;
  58
  59     LONG ref;
  60
  61     nsIURI *nsuri;
  62     nsIURL *nsurl;
  63     NSContainer *container;
  64     windowref_t *window_ref;
  65     nsChannelBSC *channel_bsc;
  66     IUri *uri;
  67     IUriBuilder *uri_builder;
  68     BOOL is_doc_uri;
  69 };
  70
  71 static BOOL ensure_uri(nsWineURI *This)
  72 {
  73     HRESULT hres;
  74
  75     assert(This->uri || This->uri_builder);
  76
  77     if(!This->uri) {
  78         hres = IUriBuilder_CreateUriSimple(This->uri_builder, 0, 0, &This->uri);
  79         if(FAILED(hres)) {
  80             WARN("CreateUriSimple failed: %08x\n", hres);
  81             return FALSE;
  82         }
  83     }
  84
  85     return TRUE;
  86 }
  87
  88 IUri *nsuri_get_uri(nsWineURI *nsuri)
  89 {
  90     if(!ensure_uri(nsuri))
  91         return NULL;
  92
  93     IUri_AddRef(nsuri->uri);
  94     return nsuri->uri;
  95 }
  96
  97 static IUri *get_uri_nofrag(IUri *uri)
  98 {
  99     IUriBuilder *uri_builder;
 100     IUri *ret;
 101     BOOL b;
 102     HRESULT hres;
 103
 104     hres = IUri_HasProperty(uri, Uri_PROPERTY_FRAGMENT, &b);
 105     if(SUCCEEDED(hres) && !b) {
 106         IUri_AddRef(uri);
 107         return uri;
 108     }
 109
 110     hres = CreateIUriBuilder(uri, 0, 0, &uri_builder);
 111     if(FAILED(hres))
 112         return NULL;
 113
 114     hres = IUriBuilder_RemoveProperties(uri_builder, Uri_HAS_FRAGMENT);
 115     if(SUCCEEDED(hres))
 116         hres = IUriBuilder_CreateUriSimple(uri_builder, 0, 0, &ret);
 117     IUriBuilder_Release(uri_builder);
 118     if(FAILED(hres))
 119         return NULL;
 120
 121     return ret;
 122 }
 123
 124 BOOL compare_ignoring_frag(IUri *uri1, IUri *uri2)
 125 {
 126     IUri *uri_nofrag1, *uri_nofrag2;
 127     BOOL ret = FALSE;
 128
 129     uri_nofrag1 = get_uri_nofrag(uri1);
 130     if(!uri_nofrag1)
 131         return FALSE;
 132
 133     uri_nofrag2 = get_uri_nofrag(uri2);
 134     if(uri_nofrag2) {
 135         IUri_IsEqual(uri_nofrag1, uri_nofrag2, &ret);
 136         IUri_Release(uri_nofrag2);
 137     }
 138
 139     IUri_Release(uri_nofrag1);
 140     return ret;
 141 }
 142
 143 static nsresult create_nsuri(IUri*,nsIURI*,HTMLWindow*,NSContainer*,nsWineURI**);
 144
 145 static const char *debugstr_nsacstr(const nsACString *nsstr)
 146 {
 147     const char *data;
 148
 149     nsACString_GetData(nsstr, &data);
 150     return debugstr_a(data);
 151 }
 152
 153 HRESULT nsuri_to_url(LPCWSTR nsuri, BOOL ret_empty, BSTR *ret)
 154 {
 155     const WCHAR *ptr = nsuri;
 156
 157     static const WCHAR wine_prefixW[] = {'w','i','n','e',':'};
 158
 159     if(!strncmpW(nsuri, wine_prefixW, sizeof(wine_prefixW)/sizeof(WCHAR)))
 160         ptr += sizeof(wine_prefixW)/sizeof(WCHAR);
 161
 162     if(*ptr || ret_empty) {
 163         *ret = SysAllocString(ptr);
 164         if(!*ret)
 165             return E_OUTOFMEMORY;
 166     }else {
 167         *ret = NULL;
 168     }
 169
 170     TRACE("%s -> %s\n", debugstr_w(nsuri), debugstr_w(*ret));
 171     return S_OK;
 172 }
 173
 174 static BOOL exec_shldocvw_67(HTMLDocumentObj *doc, BSTR url)
 175 {
 176     IOleCommandTarget *cmdtrg = NULL;
 177     HRESULT hres;
 178
 179     hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&cmdtrg);
 180     if(SUCCEEDED(hres)) {
 181         VARIANT varUrl, varRes;
 182
 183         V_VT(&varUrl) = VT_BSTR;
 184         V_BSTR(&varUrl) = url;
 185         V_VT(&varRes) = VT_BOOL;
 186
 187         hres = IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 67, 0, &varUrl, &varRes);
 188
 189         IOleCommandTarget_Release(cmdtrg);
 190
 191         if(SUCCEEDED(hres) && !V_BOOL(&varRes)) {
 192             TRACE("got VARIANT_FALSE, do not load\n");
 193             return FALSE;
 194         }
 195     }
 196
 197     return TRUE;
 198 }
 199
 200 static nsresult before_async_open(nsChannel *channel, NSContainer *container, BOOL *cancel)
 201 {
 202     HTMLDocumentObj *doc = container->doc;
 203     BSTR display_uri;
 204     DWORD hlnf = 0;
 205     HRESULT hres;
 206
 207     if(!doc) {
 208         NSContainer *container_iter = container;
 209
 210         hlnf = HLNF_OPENINNEWWINDOW;
 211         while(!container_iter->doc)
 212             container_iter = container_iter->parent;
 213         doc = container_iter->doc;
 214     }
 215
 216     if(!doc->client) {
 217         *cancel = TRUE;
 218         return NS_OK;
 219     }
 220
 221     hres = IUri_GetDisplayUri(channel->uri->uri, &display_uri);
 222     if(FAILED(hres))
 223         return NS_ERROR_FAILURE;
 224
 225     if(!hlnf) {
 226         BOOL b;
 227
 228         b = !exec_shldocvw_67(doc, display_uri);
 229         if(b) {
 230             SysFreeString(display_uri);
 231             *cancel = FALSE;
 232             return NS_OK;
 233         }
 234     }
 235
 236     hres = hlink_frame_navigate(&doc->basedoc, display_uri, channel, hlnf, cancel);
 237     SysFreeString(display_uri);
 238     if(FAILED(hres))
 239         *cancel = TRUE;
 240     return NS_OK;
 241 }
 242
 243 HRESULT load_nsuri(HTMLWindow *window, nsWineURI *uri, nsChannelBSC *channelbsc, DWORD flags)
 244 {
 245     nsIWebNavigation *web_navigation;
 246     nsIDocShell *doc_shell;
 247     HTMLDocumentNode *doc;
 248     nsresult nsres;
 249
 250     nsres = get_nsinterface((nsISupports*)window->nswindow, &IID_nsIWebNavigation, (void**)&web_navigation);
 251     if(NS_FAILED(nsres)) {
 252         ERR("Could not get nsIWebNavigation interface: %08x\n", nsres);
 253         return E_FAIL;
 254     }
 255
 256     nsres = nsIWebNavigation_QueryInterface(web_navigation, &IID_nsIDocShell, (void**)&doc_shell);
 257     nsIWebNavigation_Release(web_navigation);
 258     if(NS_FAILED(nsres)) {
 259         ERR("Could not get nsIDocShell: %08x\n", nsres);
 260         return E_FAIL;
 261     }
 262
 263     uri->channel_bsc = channelbsc;
 264     doc = window->doc;
 265     doc->skip_mutation_notif = TRUE;
 266     nsres = nsIDocShell_LoadURI(doc_shell, (nsIURI*)&uri->nsIURL_iface, NULL, flags, FALSE);
 267     if(doc == window->doc)
 268         doc->skip_mutation_notif = FALSE;
 269     uri->channel_bsc = NULL;
 270     nsIDocShell_Release(doc_shell);
 271     if(NS_FAILED(nsres)) {
 272         WARN("LoadURI failed: %08x\n", nsres);
 273         return E_FAIL;
 274     }
 275
 276     return S_OK;
 277 }
 278
 279 static void set_uri_nscontainer(nsWineURI *This, NSContainer *nscontainer)
 280 {
 281     if(This->container) {
 282         if(This->container == nscontainer)
 283             return;
 284         TRACE("Changing %p -> %p\n", This->container, nscontainer);
 285         nsIWebBrowserChrome_Release(&This->container->nsIWebBrowserChrome_iface);
 286     }
 287
 288     if(nscontainer)
 289         nsIWebBrowserChrome_AddRef(&nscontainer->nsIWebBrowserChrome_iface);
 290     This->container = nscontainer;
 291 }
 292
 293 static void set_uri_window(nsWineURI *This, HTMLWindow *window)
 294 {
 295     if(This->window_ref) {
 296         if(This->window_ref->window == window)
 297             return;
 298         TRACE("Changing %p -> %p\n", This->window_ref->window, window);
 299         windowref_release(This->window_ref);
 300     }
 301
 302     if(window) {
 303         windowref_addref(window->window_ref);
 304         This->window_ref = window->window_ref;
 305
 306         if(window->doc_obj)
 307             set_uri_nscontainer(This, window->doc_obj->nscontainer);
 308     }else {
 309         This->window_ref = NULL;
 310     }
 311 }
 312
 313 static inline BOOL is_http_channel(nsChannel *This)
 314 {
 315     return This->url_scheme == URL_SCHEME_HTTP || This->url_scheme == URL_SCHEME_HTTPS;
 316 }
 317
 318 static http_header_t *find_http_header(struct list *headers, const WCHAR *name, int len)
 319 {
 320     http_header_t *iter;
 321
 322     LIST_FOR_EACH_ENTRY(iter, headers, http_header_t, entry) {
 323         if(!strcmpiW(iter->header, name))
 324             return iter;
 325     }
 326
 327     return NULL;
 328 }
 329
 330 static nsresult get_channel_http_header(struct list *headers, const nsACString *header_name_str,
 331         nsACString *_retval)
 332 {
 333     const char *header_namea;
 334     http_header_t *header;
 335     WCHAR *header_name;
 336     char *data;
 337
 338     nsACString_GetData(header_name_str, &header_namea);
 339     header_name = heap_strdupAtoW(header_namea);
 340     if(!header_name)
 341         return NS_ERROR_UNEXPECTED;
 342
 343     header = find_http_header(headers, header_name, strlenW(header_name));
 344     heap_free(header_name);
 345     if(!header)
 346         return NS_ERROR_NOT_AVAILABLE;
 347
 348     data = heap_strdupWtoA(header->data);
 349     if(!data)
 350         return NS_ERROR_UNEXPECTED;
 351
 352     TRACE("%s -> %s\n", debugstr_a(header_namea), debugstr_a(data));
 353     nsACString_SetData(_retval, data);
 354     heap_free(data);
 355     return NS_OK;
 356 }
 357
 358 HRESULT set_http_header(struct list *headers, const WCHAR *name, int name_len,
 359         const WCHAR *value, int value_len)
 360 {
 361     http_header_t *header;
 362
 363     TRACE("%s: %s\n", debugstr_wn(name, name_len), debugstr_wn(value, value_len));
 364
 365     header = find_http_header(headers, name, name_len);
 366     if(header) {
 367         WCHAR *new_data;
 368
 369         new_data = heap_strndupW(value, value_len);
 370         if(!new_data)
 371             return E_OUTOFMEMORY;
 372
 373         heap_free(header->data);
 374         header->data = new_data;
 375     }else {
 376         header = heap_alloc(sizeof(http_header_t));
 377         if(!header)
 378             return E_OUTOFMEMORY;
 379
 380         header->header = heap_strndupW(name, name_len);
 381         header->data = heap_strndupW(value, value_len);
 382         if(!header->header || !header->data) {
 383             heap_free(header->header);
 384             heap_free(header->data);
 385             heap_free(header);
 386             return E_OUTOFMEMORY;
 387         }
 388
 389         list_add_tail(headers, &header->entry);
 390     }
 391
 392     return S_OK;
 393 }
 394
 395 static nsresult set_channel_http_header(struct list *headers, const nsACString *name_str,
 396         const nsACString *value_str)
 397 {
 398     const char *namea, *valuea;
 399     WCHAR *name, *value;
 400     HRESULT hres;
 401
 402     nsACString_GetData(name_str, &namea);
 403     name = heap_strdupAtoW(namea);
 404     if(!name)
 405         return NS_ERROR_UNEXPECTED;
 406
 407     nsACString_GetData(value_str, &valuea);
 408     value = heap_strdupAtoW(valuea);
 409     if(!value) {
 410         heap_free(name);
 411         return NS_ERROR_UNEXPECTED;
 412     }
 413
 414     hres = set_http_header(headers, name, strlenW(name), value, strlenW(value));
 415
 416     heap_free(name);
 417     heap_free(value);
 418     return SUCCEEDED(hres) ? NS_OK : NS_ERROR_UNEXPECTED;
 419 }
 420
 421 static nsresult visit_http_headers(struct list *headers, nsIHttpHeaderVisitor *visitor)
 422 {
 423     nsACString header_str, value_str;
 424     char *header, *value;
 425     http_header_t *iter;
 426     nsresult nsres;
 427
 428     LIST_FOR_EACH_ENTRY(iter, headers, http_header_t, entry) {
 429         header = heap_strdupWtoA(iter->header);
 430         if(!header)
 431             return NS_ERROR_OUT_OF_MEMORY;
 432
 433         value = heap_strdupWtoA(iter->data);
 434         if(!value) {
 435             heap_free(header);
 436             return NS_ERROR_OUT_OF_MEMORY;
 437         }
 438
 439         nsACString_InitDepend(&header_str, header);
 440         nsACString_InitDepend(&value_str, value);
 441         nsres = nsIHttpHeaderVisitor_VisitHeader(visitor, &header_str, &value_str);
 442         nsACString_Finish(&header_str);
 443         nsACString_Finish(&value_str);
 444         heap_free(header);
 445         heap_free(value);
 446         if(NS_FAILED(nsres))
 447             break;
 448     }
 449
 450     return NS_OK;
 451 }
 452
 453 static void free_http_headers(struct list *list)
 454 {
 455     http_header_t *iter, *iter_next;
 456
 457     LIST_FOR_EACH_ENTRY_SAFE(iter, iter_next, list, http_header_t, entry) {
 458         list_remove(&iter->entry);
 459         heap_free(iter->header);
 460         heap_free(iter->data);
 461         heap_free(iter);
 462     }
 463 }
 464
 465 static inline nsChannel *impl_from_nsIHttpChannel(nsIHttpChannel *iface)
 466 {
 467     return CONTAINING_RECORD(iface, nsChannel, nsIHttpChannel_iface);
 468 }
 469
 470 static nsresult NSAPI nsChannel_QueryInterface(nsIHttpChannel *iface, nsIIDRef riid, void **result)
 471 {
 472     nsChannel *This = impl_from_nsIHttpChannel(iface);
 473
 474     if(IsEqualGUID(&IID_nsISupports, riid)) {
 475         TRACE("(%p)->(IID_nsISupports %p)\n", This, result);
 476         *result = &This->nsIHttpChannel_iface;
 477     }else if(IsEqualGUID(&IID_nsIRequest, riid)) {
 478         TRACE("(%p)->(IID_nsIRequest %p)\n", This, result);
 479         *result = &This->nsIHttpChannel_iface;
 480     }else if(IsEqualGUID(&IID_nsIChannel, riid)) {
 481         TRACE("(%p)->(IID_nsIChannel %p)\n", This, result);
 482         *result = &This->nsIHttpChannel_iface;
 483     }else if(IsEqualGUID(&IID_nsIHttpChannel, riid)) {
 484         TRACE("(%p)->(IID_nsIHttpChannel %p)\n", This, result);
 485         *result = is_http_channel(This) ? &This->nsIHttpChannel_iface : NULL;
 486     }else if(IsEqualGUID(&IID_nsIUploadChannel, riid)) {
 487         TRACE("(%p)->(IID_nsIUploadChannel %p)\n", This, result);
 488         *result = &This->nsIUploadChannel_iface;
 489     }else if(IsEqualGUID(&IID_nsIHttpChannelInternal, riid)) {
 490         TRACE("(%p)->(IID_nsIHttpChannelInternal %p)\n", This, result);
 491         *result = is_http_channel(This) ? &This->nsIHttpChannelInternal_iface : NULL;
 492     }else {
 493         TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), result);
 494         *result = NULL;
 495     }
 496
 497     if(*result) {
 498         nsIChannel_AddRef(&This->nsIHttpChannel_iface);
 499         return NS_OK;
 500     }
 501
 502     return NS_NOINTERFACE;
 503 }
 504
 505 static nsrefcnt NSAPI nsChannel_AddRef(nsIHttpChannel *iface)
 506 {
 507     nsChannel *This = impl_from_nsIHttpChannel(iface);
 508     nsrefcnt ref = InterlockedIncrement(&This->ref);
 509
 510     TRACE("(%p) ref=%d\n", This, ref);
 511
 512     return ref;
 513 }
 514
 515 static nsrefcnt NSAPI nsChannel_Release(nsIHttpChannel *iface)
 516 {
 517     nsChannel *This = impl_from_nsIHttpChannel(iface);
 518     LONG ref = InterlockedDecrement(&This->ref);
 519
 520     if(!ref) {
 521         nsIURI_Release(&This->uri->nsIURL_iface);
 522         if(This->owner)
 523             nsISupports_Release(This->owner);
 524         if(This->post_data_stream)
 525             nsIInputStream_Release(This->post_data_stream);
 526         if(This->load_group)
 527             nsILoadGroup_Release(This->load_group);
 528         if(This->notif_callback)
 529             nsIInterfaceRequestor_Release(This->notif_callback);
 530         if(This->original_uri)
 531             nsIURI_Release(This->original_uri);
 532         if(This->referrer)
 533             nsIURI_Release(This->referrer);
 534
 535         free_http_headers(&This->response_headers);
 536         free_http_headers(&This->request_headers);
 537
 538         heap_free(This->content_type);
 539         heap_free(This->charset);
 540         heap_free(This);
 541     }
 542
 543     return ref;
 544 }
 545
 546 static nsresult NSAPI nsChannel_GetName(nsIHttpChannel *iface, nsACString *aName)
 547 {
 548     nsChannel *This = impl_from_nsIHttpChannel(iface);
 549
 550     TRACE("(%p)->(%p)\n", This, aName);
 551
 552     return nsIURI_GetSpec(&This->uri->nsIURL_iface, aName);
 553 }
 554
 555 static nsresult NSAPI nsChannel_IsPending(nsIHttpChannel *iface, PRBool *_retval)
 556 {
 557     nsChannel *This = impl_from_nsIHttpChannel(iface);
 558
 559     FIXME("(%p)->(%p)\n", This, _retval);
 560
 561     return NS_ERROR_NOT_IMPLEMENTED;
 562 }
 563
 564 static nsresult NSAPI nsChannel_GetStatus(nsIHttpChannel *iface, nsresult *aStatus)
 565 {
 566     nsChannel *This = impl_from_nsIHttpChannel(iface);
 567
 568     WARN("(%p)->(%p) returning NS_OK\n", This, aStatus);
 569
 570     return *aStatus = NS_OK;
 571 }
 572
 573 static nsresult NSAPI nsChannel_Cancel(nsIHttpChannel *iface, nsresult aStatus)
 574 {
 575     nsChannel *This = impl_from_nsIHttpChannel(iface);
 576
 577     FIXME("(%p)->(%08x)\n", This, aStatus);
 578
 579     return NS_ERROR_NOT_IMPLEMENTED;
 580 }
 581
 582 static nsresult NSAPI nsChannel_Suspend(nsIHttpChannel *iface)
 583 {
 584     nsChannel *This = impl_from_nsIHttpChannel(iface);
 585
 586     FIXME("(%p)\n", This);
 587
 588     return NS_ERROR_NOT_IMPLEMENTED;
 589 }
 590
 591 static nsresult NSAPI nsChannel_Resume(nsIHttpChannel *iface)
 592 {
 593     nsChannel *This = impl_from_nsIHttpChannel(iface);
 594
 595     FIXME("(%p)\n", This);
 596
 597     return NS_ERROR_NOT_IMPLEMENTED;
 598 }
 599
 600 static nsresult NSAPI nsChannel_GetLoadGroup(nsIHttpChannel *iface, nsILoadGroup **aLoadGroup)
 601 {
 602     nsChannel *This = impl_from_nsIHttpChannel(iface);
 603
 604     TRACE("(%p)->(%p)\n", This, aLoadGroup);
 605
 606     if(This->load_group)
 607         nsILoadGroup_AddRef(This->load_group);
 608
 609     *aLoadGroup = This->load_group;
 610     return NS_OK;
 611 }
 612
 613 static nsresult NSAPI nsChannel_SetLoadGroup(nsIHttpChannel *iface, nsILoadGroup *aLoadGroup)
 614 {
 615     nsChannel *This = impl_from_nsIHttpChannel(iface);
 616
 617     TRACE("(%p)->(%p)\n", This, aLoadGroup);
 618
 619     if(This->load_group)
 620         nsILoadGroup_Release(This->load_group);
 621     if(aLoadGroup)
 622         nsILoadGroup_AddRef(aLoadGroup);
 623     This->load_group = aLoadGroup;
 624
 625     return NS_OK;
 626 }
 627
 628 static nsresult NSAPI nsChannel_GetLoadFlags(nsIHttpChannel *iface, nsLoadFlags *aLoadFlags)
 629 {
 630     nsChannel *This = impl_from_nsIHttpChannel(iface);
 631
 632     TRACE("(%p)->(%p)\n", This, aLoadFlags);
 633
 634     *aLoadFlags = This->load_flags;
 635     return NS_OK;
 636 }
 637
 638 static nsresult NSAPI nsChannel_SetLoadFlags(nsIHttpChannel *iface, nsLoadFlags aLoadFlags)
 639 {
 640     nsChannel *This = impl_from_nsIHttpChannel(iface);
 641
 642     TRACE("(%p)->(%08x)\n", This, aLoadFlags);
 643
 644     This->load_flags = aLoadFlags;
 645     return NS_OK;
 646 }
 647
 648 static nsresult NSAPI nsChannel_GetOriginalURI(nsIHttpChannel *iface, nsIURI **aOriginalURI)
 649 {
 650     nsChannel *This = impl_from_nsIHttpChannel(iface);
 651
 652     TRACE("(%p)->(%p)\n", This, aOriginalURI);
 653
 654     if(This->original_uri)
 655         nsIURI_AddRef(This->original_uri);
 656
 657     *aOriginalURI = This->original_uri;
 658     return NS_OK;
 659 }
 660
 661 static nsresult NSAPI nsChannel_SetOriginalURI(nsIHttpChannel *iface, nsIURI *aOriginalURI)
 662 {
 663     nsChannel *This = impl_from_nsIHttpChannel(iface);
 664
 665     TRACE("(%p)->(%p)\n", This, aOriginalURI);
 666
 667     if(This->original_uri)
 668         nsIURI_Release(This->original_uri);
 669
 670     nsIURI_AddRef(aOriginalURI);
 671     This->original_uri = aOriginalURI;
 672     return NS_OK;
 673 }
 674
 675 static nsresult NSAPI nsChannel_GetURI(nsIHttpChannel *iface, nsIURI **aURI)
 676 {
 677     nsChannel *This = impl_from_nsIHttpChannel(iface);
 678
 679     TRACE("(%p)->(%p)\n", This, aURI);
 680
 681     nsIURI_AddRef(&This->uri->nsIURL_iface);
 682     *aURI = (nsIURI*)This->uri;
 683
 684     return NS_OK;
 685 }
 686
 687 static nsresult NSAPI nsChannel_GetOwner(nsIHttpChannel *iface, nsISupports **aOwner)
 688 {
 689     nsChannel *This = impl_from_nsIHttpChannel(iface);
 690
 691     TRACE("(%p)->(%p)\n", This, aOwner);
 692
 693     if(This->owner)
 694         nsISupports_AddRef(This->owner);
 695     *aOwner = This->owner;
 696
 697     return NS_OK;
 698 }
 699
 700 static nsresult NSAPI nsChannel_SetOwner(nsIHttpChannel *iface, nsISupports *aOwner)
 701 {
 702     nsChannel *This = impl_from_nsIHttpChannel(iface);
 703
 704     TRACE("(%p)->(%p)\n", This, aOwner);
 705
 706     if(aOwner)
 707         nsISupports_AddRef(aOwner);
 708     if(This->owner)
 709         nsISupports_Release(This->owner);
 710     This->owner = aOwner;
 711
 712     return NS_OK;
 713 }
 714
 715 static nsresult NSAPI nsChannel_GetNotificationCallbacks(nsIHttpChannel *iface,
 716         nsIInterfaceRequestor **aNotificationCallbacks)
 717 {
 718     nsChannel *This = impl_from_nsIHttpChannel(iface);
 719
 720     TRACE("(%p)->(%p)\n", This, aNotificationCallbacks);
 721
 722     if(This->notif_callback)
 723         nsIInterfaceRequestor_AddRef(This->notif_callback);
 724     *aNotificationCallbacks = This->notif_callback;
 725
 726     return NS_OK;
 727 }
 728
 729 static nsresult NSAPI nsChannel_SetNotificationCallbacks(nsIHttpChannel *iface,
 730         nsIInterfaceRequestor *aNotificationCallbacks)
 731 {
 732     nsChannel *This = impl_from_nsIHttpChannel(iface);
 733
 734     TRACE("(%p)->(%p)\n", This, aNotificationCallbacks);
 735
 736     if(This->notif_callback)
 737         nsIInterfaceRequestor_Release(This->notif_callback);
 738     if(aNotificationCallbacks)
 739         nsIInterfaceRequestor_AddRef(aNotificationCallbacks);
 740
 741     This->notif_callback = aNotificationCallbacks;
 742
 743     return NS_OK;
 744 }
 745
 746 static nsresult NSAPI nsChannel_GetSecurityInfo(nsIHttpChannel *iface, nsISupports **aSecurityInfo)
 747 {
 748     nsChannel *This = impl_from_nsIHttpChannel(iface);
 749
 750     TRACE("(%p)->(%p)\n", This, aSecurityInfo);
 751
 752     return NS_ERROR_NOT_IMPLEMENTED;
 753 }
 754
 755 static nsresult NSAPI nsChannel_GetContentType(nsIHttpChannel *iface, nsACString *aContentType)
 756 {
 757     nsChannel *This = impl_from_nsIHttpChannel(iface);
 758
 759     TRACE("(%p)->(%p)\n", This, aContentType);
 760
 761     if(This->content_type) {
 762         nsACString_SetData(aContentType, This->content_type);
 763         return S_OK;
 764     }
 765
 766     WARN("unknown type\n");
 767     return NS_ERROR_FAILURE;
 768 }
 769
 770 static nsresult NSAPI nsChannel_SetContentType(nsIHttpChannel *iface,
 771                                                const nsACString *aContentType)
 772 {
 773     nsChannel *This = impl_from_nsIHttpChannel(iface);
 774     const char *content_type;
 775
 776     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aContentType));
 777
 778     nsACString_GetData(aContentType, &content_type);
 779     heap_free(This->content_type);
 780     This->content_type = heap_strdupA(content_type);
 781
 782     return NS_OK;
 783 }
 784
 785 static nsresult NSAPI nsChannel_GetContentCharset(nsIHttpChannel *iface,
 786                                                   nsACString *aContentCharset)
 787 {
 788     nsChannel *This = impl_from_nsIHttpChannel(iface);
 789
 790     TRACE("(%p)->(%p)\n", This, aContentCharset);
 791
 792     if(This->charset) {
 793         nsACString_SetData(aContentCharset, This->charset);
 794         return NS_OK;
 795     }
 796
 797     nsACString_SetData(aContentCharset, "");
 798     return NS_OK;
 799 }
 800
 801 static nsresult NSAPI nsChannel_SetContentCharset(nsIHttpChannel *iface,
 802                                                   const nsACString *aContentCharset)
 803 {
 804     nsChannel *This = impl_from_nsIHttpChannel(iface);
 805
 806     FIXME("(%p)->(%s)\n", This, debugstr_nsacstr(aContentCharset));
 807
 808     return NS_ERROR_NOT_IMPLEMENTED;
 809 }
 810
 811 static nsresult NSAPI nsChannel_GetContentLength(nsIHttpChannel *iface, PRInt32 *aContentLength)
 812 {
 813     nsChannel *This = impl_from_nsIHttpChannel(iface);
 814
 815     FIXME("(%p)->(%p)\n", This, aContentLength);
 816
 817     return NS_ERROR_NOT_IMPLEMENTED;
 818 }
 819
 820 static nsresult NSAPI nsChannel_SetContentLength(nsIHttpChannel *iface, PRInt32 aContentLength)
 821 {
 822     nsChannel *This = impl_from_nsIHttpChannel(iface);
 823
 824     FIXME("(%p)->(%d)\n", This, aContentLength);
 825
 826     return NS_ERROR_NOT_IMPLEMENTED;
 827 }
 828
 829 static nsresult NSAPI nsChannel_Open(nsIHttpChannel *iface, nsIInputStream **_retval)
 830 {
 831     nsChannel *This = impl_from_nsIHttpChannel(iface);
 832
 833     FIXME("(%p)->(%p)\n", This, _retval);
 834
 835     return NS_ERROR_NOT_IMPLEMENTED;
 836 }
 837
 838 static HTMLWindow *get_window_from_load_group(nsChannel *This)
 839 {
 840     HTMLWindow *window;
 841     nsIChannel *channel;
 842     nsIRequest *req;
 843     nsWineURI *wine_uri;
 844     nsIURI *uri;
 845     nsresult nsres;
 846
 847     nsres = nsILoadGroup_GetDefaultLoadRequest(This->load_group, &req);
 848     if(NS_FAILED(nsres)) {
 849         ERR("GetDefaultLoadRequest failed: %08x\n", nsres);
 850         return NULL;
 851     }
 852
 853     if(!req)
 854         return NULL;
 855
 856     nsres = nsIRequest_QueryInterface(req, &IID_nsIChannel, (void**)&channel);
 857     nsIRequest_Release(req);
 858     if(NS_FAILED(nsres)) {
 859         WARN("Could not get nsIChannel interface: %08x\n", nsres);
 860         return NULL;
 861     }
 862
 863     nsres = nsIChannel_GetURI(channel, &uri);
 864     nsIChannel_Release(channel);
 865     if(NS_FAILED(nsres)) {
 866         ERR("GetURI failed: %08x\n", nsres);
 867         return NULL;
 868     }
 869
 870     nsres = nsIURI_QueryInterface(uri, &IID_nsWineURI, (void**)&wine_uri);
 871     nsIURI_Release(uri);
 872     if(NS_FAILED(nsres)) {
 873         TRACE("Could not get nsWineURI: %08x\n", nsres);
 874         return NULL;
 875     }
 876
 877     window = wine_uri->window_ref ? wine_uri->window_ref->window : NULL;
 878     if(window)
 879         IHTMLWindow2_AddRef(&window->IHTMLWindow2_iface);
 880     nsIURI_Release(&wine_uri->nsIURL_iface);
 881
 882     return window;
 883 }
 884
 885 static HTMLWindow *get_channel_window(nsChannel *This)
 886 {
 887     nsIWebProgress *web_progress;
 888     nsIDOMWindow *nswindow;
 889     HTMLWindow *window;
 890     nsresult nsres;
 891
 892     if(This->load_group) {
 893         nsIRequestObserver *req_observer;
 894
 895         nsres = nsILoadGroup_GetGroupObserver(This->load_group, &req_observer);
 896         if(NS_FAILED(nsres) || !req_observer) {
 897             ERR("GetGroupObserver failed: %08x\n", nsres);
 898             return NULL;
 899         }
 900
 901         nsres = nsIRequestObserver_QueryInterface(req_observer, &IID_nsIWebProgress, (void**)&web_progress);
 902         nsIRequestObserver_Release(req_observer);
 903         if(NS_FAILED(nsres)) {
 904             ERR("Could not get nsIWebProgress iface: %08x\n", nsres);
 905             return NULL;
 906         }
 907     }else if(This->notif_callback) {
 908         nsres = nsIInterfaceRequestor_GetInterface(This->notif_callback, &IID_nsIWebProgress, (void**)&web_progress);
 909         if(NS_FAILED(nsres)) {
 910             ERR("GetInterface(IID_nsIWebProgress failed: %08x\n", nsres);
 911             return NULL;
 912         }
 913     }else {
 914         ERR("no load group nor notif callback\n");
 915         return NULL;
 916     }
 917
 918     nsres = nsIWebProgress_GetDOMWindow(web_progress, &nswindow);
 919     nsIWebProgress_Release(web_progress);
 920     if(NS_FAILED(nsres) || !nswindow) {
 921         ERR("GetDOMWindow failed: %08x\n", nsres);
 922         return NULL;
 923     }
 924
 925     window = nswindow_to_window(nswindow);
 926     nsIDOMWindow_Release(nswindow);
 927
 928     if(window)
 929         IHTMLWindow2_AddRef(&window->IHTMLWindow2_iface);
 930     else
 931         FIXME("NULL window for %p\n", nswindow);
 932     return window;
 933 }
 934
 935 typedef struct {
 936     task_t header;
 937     HTMLDocumentNode *doc;
 938     nsChannelBSC *bscallback;
 939 } start_binding_task_t;
 940
 941 static void start_binding_proc(task_t *_task)
 942 {
 943     start_binding_task_t *task = (start_binding_task_t*)_task;
 944
 945     start_binding(NULL, task->doc, (BSCallback*)task->bscallback, NULL);
 946 }
 947
 948 static void start_binding_task_destr(task_t *_task)
 949 {
 950     start_binding_task_t *task = (start_binding_task_t*)_task;
 951
 952     IBindStatusCallback_Release(&task->bscallback->bsc.IBindStatusCallback_iface);
 953     heap_free(task);
 954 }
 955
 956 static nsresult async_open(nsChannel *This, HTMLWindow *window, BOOL is_doc_channel, nsIStreamListener *listener,
 957         nsISupports *context)
 958 {
 959     nsChannelBSC *bscallback;
 960     IMoniker *mon = NULL;
 961     HRESULT hres;
 962
 963     hres = CreateURLMonikerEx2(NULL, This->uri->uri, &mon, 0);
 964     if(FAILED(hres)) {
 965         WARN("CreateURLMoniker failed: %08x\n", hres);
 966         return NS_ERROR_UNEXPECTED;
 967     }
 968
 969     if(is_doc_channel)
 970         set_current_mon(window, mon);
 971
 972     hres = create_channelbsc(mon, NULL, NULL, 0, &bscallback);
 973     IMoniker_Release(mon);
 974     if(FAILED(hres))
 975         return NS_ERROR_UNEXPECTED;
 976
 977     channelbsc_set_channel(bscallback, This, listener, context);
 978
 979     if(is_doc_channel) {
 980         set_window_bscallback(window, bscallback);
 981         async_start_doc_binding(window, bscallback);
 982         IUnknown_Release((IUnknown*)bscallback);
 983     }else {
 984         start_binding_task_t *task = heap_alloc(sizeof(start_binding_task_t));
 985
 986         task->doc = window->doc;
 987         task->bscallback = bscallback;
 988         push_task(&task->header, start_binding_proc, start_binding_task_destr, window->doc->basedoc.task_magic);
 989     }
 990
 991     return NS_OK;
 992 }
 993
 994 static nsresult NSAPI nsChannel_AsyncOpen(nsIHttpChannel *iface, nsIStreamListener *aListener,
 995                                           nsISupports *aContext)
 996 {
 997     nsChannel *This = impl_from_nsIHttpChannel(iface);
 998     HTMLWindow *window = NULL;
 999     BOOL cancel = FALSE;
1000     nsresult nsres = NS_OK;
1001
1002     TRACE("(%p)->(%p %p)\n", This, aListener, aContext);
1003
1004     if(!ensure_uri(This->uri))
1005         return NS_ERROR_FAILURE;
1006
1007     if(TRACE_ON(mshtml)) {
1008         BSTR uri_str;
1009         IUri_GetDisplayUri(This->uri->uri, &uri_str);
1010         TRACE("opening %s\n", debugstr_w(uri_str));
1011         SysFreeString(uri_str);
1012     }
1013
1014     if(This->uri->is_doc_uri) {
1015         window = get_channel_window(This);
1016         if(window) {
1017             set_uri_window(This->uri, window);
1018         }else if(This->uri->container) {
1019             BOOL b;
1020
1021             /* nscontainer->doc should be NULL which means navigation to a new window */
1022             if(This->uri->container->doc)
1023                 FIXME("nscontainer->doc = %p\n", This->uri->container->doc);
1024
1025             nsres = before_async_open(This, This->uri->container, &b);
1026             if(NS_FAILED(nsres))
1027                 return nsres;
1028             if(b)
1029                 FIXME("Navigation not cancelled\n");
1030             return NS_ERROR_UNEXPECTED;
1031         }
1032     }
1033
1034     if(!window) {
1035         if(This->uri->window_ref && This->uri->window_ref->window) {
1036             window = This->uri->window_ref->window;
1037             IHTMLWindow2_AddRef(&window->IHTMLWindow2_iface);
1038         }else {
1039             /* FIXME: Analyze removing get_window_from_load_group call */
1040             if(This->load_group)
1041                 window = get_window_from_load_group(This);
1042             if(!window)
1043                 window = get_channel_window(This);
1044             if(window)
1045                 set_uri_window(This->uri, window);
1046         }
1047     }
1048
1049     if(!window) {
1050         ERR("window = NULL\n");
1051         return NS_ERROR_UNEXPECTED;
1052     }
1053
1054     if(This->uri->is_doc_uri && window == window->doc_obj->basedoc.window) {
1055         if(This->uri->channel_bsc) {
1056             channelbsc_set_channel(This->uri->channel_bsc, This, aListener, aContext);
1057
1058             if(window->doc_obj->mime) {
1059                 heap_free(This->content_type);
1060                 This->content_type = heap_strdupWtoA(window->doc_obj->mime);
1061             }
1062
1063             cancel = TRUE;
1064         }else {
1065             nsres = before_async_open(This, window->doc_obj->nscontainer, &cancel);
1066             if(NS_SUCCEEDED(nsres)  && cancel) {
1067                 TRACE("canceled\n");
1068                 nsres = NS_BINDING_ABORTED;
1069             }
1070         }
1071     }
1072
1073     if(!cancel)
1074         nsres = async_open(This, window, This->uri->is_doc_uri, aListener, aContext);
1075
1076     if(NS_SUCCEEDED(nsres) && This->load_group) {
1077         nsres = nsILoadGroup_AddRequest(This->load_group, (nsIRequest*)&This->nsIHttpChannel_iface,
1078                 aContext);
1079         if(NS_FAILED(nsres))
1080             ERR("AddRequest failed: %08x\n", nsres);
1081     }
1082
1083     IHTMLWindow2_Release(&window->IHTMLWindow2_iface);
1084     return nsres;
1085 }
1086
1087 static nsresult NSAPI nsChannel_GetRequestMethod(nsIHttpChannel *iface, nsACString *aRequestMethod)
1088 {
1089     nsChannel *This = impl_from_nsIHttpChannel(iface);
1090
1091     TRACE("(%p)->(%p)\n", This, aRequestMethod);
1092
1093     nsACString_SetData(aRequestMethod, request_method_strings[This->request_method]);
1094     return NS_OK;
1095 }
1096
1097 static nsresult NSAPI nsChannel_SetRequestMethod(nsIHttpChannel *iface,
1098                                                  const nsACString *aRequestMethod)
1099 {
1100     nsChannel *This = impl_from_nsIHttpChannel(iface);
1101     const char *method;
1102     unsigned i;
1103
1104     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aRequestMethod));
1105
1106     nsACString_GetData(aRequestMethod, &method);
1107     for(i=0; i < sizeof(request_method_strings)/sizeof(*request_method_strings); i++) {
1108         if(!strcasecmp(method, request_method_strings[i])) {
1109             This->request_method = i;
1110             return NS_OK;
1111         }
1112     }
1113
1114     ERR("Invalid method %s\n", debugstr_a(method));
1115     return NS_ERROR_UNEXPECTED;
1116 }
1117
1118 static nsresult NSAPI nsChannel_GetReferrer(nsIHttpChannel *iface, nsIURI **aReferrer)
1119 {
1120     nsChannel *This = impl_from_nsIHttpChannel(iface);
1121
1122     TRACE("(%p)->(%p)\n", This, aReferrer);
1123
1124     if(This->referrer)
1125         nsIURI_AddRef(This->referrer);
1126     *aReferrer = This->referrer;
1127     return NS_OK;
1128 }
1129
1130 static nsresult NSAPI nsChannel_SetReferrer(nsIHttpChannel *iface, nsIURI *aReferrer)
1131 {
1132     nsChannel *This = impl_from_nsIHttpChannel(iface);
1133
1134     TRACE("(%p)->(%p)\n", This, aReferrer);
1135
1136     if(aReferrer)
1137         nsIURI_AddRef(aReferrer);
1138     if(This->referrer)
1139         nsIURI_Release(This->referrer);
1140     This->referrer = aReferrer;
1141     return NS_OK;
1142 }
1143
1144 static nsresult NSAPI nsChannel_GetRequestHeader(nsIHttpChannel *iface,
1145          const nsACString *aHeader, nsACString *_retval)
1146 {
1147     nsChannel *This = impl_from_nsIHttpChannel(iface);
1148
1149     TRACE("(%p)->(%s %p)\n", This, debugstr_nsacstr(aHeader), _retval);
1150
1151     return get_channel_http_header(&This->request_headers, aHeader, _retval);
1152 }
1153
1154 static nsresult NSAPI nsChannel_SetRequestHeader(nsIHttpChannel *iface,
1155          const nsACString *aHeader, const nsACString *aValue, PRBool aMerge)
1156 {
1157     nsChannel *This = impl_from_nsIHttpChannel(iface);
1158
1159     TRACE("(%p)->(%s %s %x)\n", This, debugstr_nsacstr(aHeader), debugstr_nsacstr(aValue), aMerge);
1160
1161     if(aMerge)
1162         FIXME("aMerge not supported\n");
1163
1164     return set_channel_http_header(&This->request_headers, aHeader, aValue);
1165 }
1166
1167 static nsresult NSAPI nsChannel_VisitRequestHeaders(nsIHttpChannel *iface,
1168                                                     nsIHttpHeaderVisitor *aVisitor)
1169 {
1170     nsChannel *This = impl_from_nsIHttpChannel(iface);
1171
1172     FIXME("(%p)->(%p)\n", This, aVisitor);
1173
1174     return NS_ERROR_NOT_IMPLEMENTED;
1175 }
1176
1177 static nsresult NSAPI nsChannel_GetAllowPipelining(nsIHttpChannel *iface, PRBool *aAllowPipelining)
1178 {
1179     nsChannel *This = impl_from_nsIHttpChannel(iface);
1180
1181     FIXME("(%p)->(%p)\n", This, aAllowPipelining);
1182
1183     return NS_ERROR_NOT_IMPLEMENTED;
1184 }
1185
1186 static nsresult NSAPI nsChannel_SetAllowPipelining(nsIHttpChannel *iface, PRBool aAllowPipelining)
1187 {
1188     nsChannel *This = impl_from_nsIHttpChannel(iface);
1189
1190     FIXME("(%p)->(%x)\n", This, aAllowPipelining);
1191
1192     return NS_ERROR_NOT_IMPLEMENTED;
1193 }
1194
1195 static nsresult NSAPI nsChannel_GetRedirectionLimit(nsIHttpChannel *iface, PRUint32 *aRedirectionLimit)
1196 {
1197     nsChannel *This = impl_from_nsIHttpChannel(iface);
1198
1199     FIXME("(%p)->(%p)\n", This, aRedirectionLimit);
1200
1201     return NS_ERROR_NOT_IMPLEMENTED;
1202 }
1203
1204 static nsresult NSAPI nsChannel_SetRedirectionLimit(nsIHttpChannel *iface, PRUint32 aRedirectionLimit)
1205 {
1206     nsChannel *This = impl_from_nsIHttpChannel(iface);
1207
1208     FIXME("(%p)->(%u)\n", This, aRedirectionLimit);
1209
1210     return NS_ERROR_NOT_IMPLEMENTED;
1211 }
1212
1213 static nsresult NSAPI nsChannel_GetResponseStatus(nsIHttpChannel *iface, PRUint32 *aResponseStatus)
1214 {
1215     nsChannel *This = impl_from_nsIHttpChannel(iface);
1216
1217     TRACE("(%p)->(%p)\n", This, aResponseStatus);
1218
1219     if(This->response_status) {
1220         *aResponseStatus = This->response_status;
1221         return NS_OK;
1222     }
1223
1224     WARN("No response status\n");
1225     return NS_ERROR_UNEXPECTED;
1226 }
1227
1228 static nsresult NSAPI nsChannel_GetResponseStatusText(nsIHttpChannel *iface,
1229                                                       nsACString *aResponseStatusText)
1230 {
1231     nsChannel *This = impl_from_nsIHttpChannel(iface);
1232
1233     FIXME("(%p)->(%p)\n", This, aResponseStatusText);
1234
1235     return NS_ERROR_NOT_IMPLEMENTED;
1236 }
1237
1238 static nsresult NSAPI nsChannel_GetRequestSucceeded(nsIHttpChannel *iface,
1239                                                     PRBool *aRequestSucceeded)
1240 {
1241     nsChannel *This = impl_from_nsIHttpChannel(iface);
1242
1243     TRACE("(%p)->(%p)\n", This, aRequestSucceeded);
1244
1245     if(!This->response_status)
1246         return NS_ERROR_NOT_AVAILABLE;
1247
1248     *aRequestSucceeded = This->response_status/100 == 2;
1249
1250     return NS_OK;
1251 }
1252
1253 static nsresult NSAPI nsChannel_GetResponseHeader(nsIHttpChannel *iface,
1254          const nsACString *header, nsACString *_retval)
1255 {
1256     nsChannel *This = impl_from_nsIHttpChannel(iface);
1257
1258     TRACE("(%p)->(%s %p)\n", This, debugstr_nsacstr(header), _retval);
1259
1260     return get_channel_http_header(&This->response_headers, header, _retval);
1261 }
1262
1263 static nsresult NSAPI nsChannel_SetResponseHeader(nsIHttpChannel *iface,
1264         const nsACString *header, const nsACString *value, PRBool merge)
1265 {
1266     nsChannel *This = impl_from_nsIHttpChannel(iface);
1267
1268     FIXME("(%p)->(%s %s %x)\n", This, debugstr_nsacstr(header), debugstr_nsacstr(value), merge);
1269
1270     return NS_ERROR_NOT_IMPLEMENTED;
1271 }
1272
1273 static nsresult NSAPI nsChannel_VisitResponseHeaders(nsIHttpChannel *iface,
1274         nsIHttpHeaderVisitor *aVisitor)
1275 {
1276     nsChannel *This = impl_from_nsIHttpChannel(iface);
1277
1278     TRACE("(%p)->(%p)\n", This, aVisitor);
1279
1280     return visit_http_headers(&This->response_headers, aVisitor);
1281 }
1282
1283 static nsresult NSAPI nsChannel_IsNoStoreResponse(nsIHttpChannel *iface, PRBool *_retval)
1284 {
1285     nsChannel *This = impl_from_nsIHttpChannel(iface);
1286     http_header_t *header;
1287
1288     static const WCHAR cache_controlW[] = {'C','a','c','h','e','-','C','o','n','t','r','o','l'};
1289     static const WCHAR no_storeW[] = {'n','o','-','s','t','o','r','e',0};
1290
1291     TRACE("(%p)->(%p)\n", This, _retval);
1292
1293     header = find_http_header(&This->response_headers, cache_controlW, sizeof(cache_controlW)/sizeof(WCHAR));
1294     *_retval = header && !strcmpiW(header->data, no_storeW);
1295     return NS_OK;
1296 }
1297
1298 static nsresult NSAPI nsChannel_IsNoCacheResponse(nsIHttpChannel *iface, PRBool *_retval)
1299 {
1300     nsChannel *This = impl_from_nsIHttpChannel(iface);
1301
1302     FIXME("(%p)->(%p)\n", This, _retval);
1303
1304     return NS_ERROR_NOT_IMPLEMENTED;
1305 }
1306
1307 static const nsIHttpChannelVtbl nsChannelVtbl = {
1308     nsChannel_QueryInterface,
1309     nsChannel_AddRef,
1310     nsChannel_Release,
1311     nsChannel_GetName,
1312     nsChannel_IsPending,
1313     nsChannel_GetStatus,
1314     nsChannel_Cancel,
1315     nsChannel_Suspend,
1316     nsChannel_Resume,
1317     nsChannel_GetLoadGroup,
1318     nsChannel_SetLoadGroup,
1319     nsChannel_GetLoadFlags,
1320     nsChannel_SetLoadFlags,
1321     nsChannel_GetOriginalURI,
1322     nsChannel_SetOriginalURI,
1323     nsChannel_GetURI,
1324     nsChannel_GetOwner,
1325     nsChannel_SetOwner,
1326     nsChannel_GetNotificationCallbacks,
1327     nsChannel_SetNotificationCallbacks,
1328     nsChannel_GetSecurityInfo,
1329     nsChannel_GetContentType,
1330     nsChannel_SetContentType,
1331     nsChannel_GetContentCharset,
1332     nsChannel_SetContentCharset,
1333     nsChannel_GetContentLength,
1334     nsChannel_SetContentLength,
1335     nsChannel_Open,
1336     nsChannel_AsyncOpen,
1337     nsChannel_GetRequestMethod,
1338     nsChannel_SetRequestMethod,
1339     nsChannel_GetReferrer,
1340     nsChannel_SetReferrer,
1341     nsChannel_GetRequestHeader,
1342     nsChannel_SetRequestHeader,
1343     nsChannel_VisitRequestHeaders,
1344     nsChannel_GetAllowPipelining,
1345     nsChannel_SetAllowPipelining,
1346     nsChannel_GetRedirectionLimit,
1347     nsChannel_SetRedirectionLimit,
1348     nsChannel_GetResponseStatus,
1349     nsChannel_GetResponseStatusText,
1350     nsChannel_GetRequestSucceeded,
1351     nsChannel_GetResponseHeader,
1352     nsChannel_SetResponseHeader,
1353     nsChannel_VisitResponseHeaders,
1354     nsChannel_IsNoStoreResponse,
1355     nsChannel_IsNoCacheResponse
1356 };
1357
1358 static inline nsChannel *impl_from_nsIUploadChannel(nsIUploadChannel *iface)
1359 {
1360     return CONTAINING_RECORD(iface, nsChannel, nsIUploadChannel_iface);
1361 }
1362
1363 static nsresult NSAPI nsUploadChannel_QueryInterface(nsIUploadChannel *iface, nsIIDRef riid,
1364         void **result)
1365 {
1366     nsChannel *This = impl_from_nsIUploadChannel(iface);
1367     return nsIChannel_QueryInterface(&This->nsIHttpChannel_iface, riid, result);
1368 }
1369
1370 static nsrefcnt NSAPI nsUploadChannel_AddRef(nsIUploadChannel *iface)
1371 {
1372     nsChannel *This = impl_from_nsIUploadChannel(iface);
1373     return nsIChannel_AddRef(&This->nsIHttpChannel_iface);
1374 }
1375
1376 static nsrefcnt NSAPI nsUploadChannel_Release(nsIUploadChannel *iface)
1377 {
1378     nsChannel *This = impl_from_nsIUploadChannel(iface);
1379     return nsIChannel_Release(&This->nsIHttpChannel_iface);
1380 }
1381
1382 static nsresult NSAPI nsUploadChannel_SetUploadStream(nsIUploadChannel *iface,
1383         nsIInputStream *aStream, const nsACString *aContentType, PRInt32 aContentLength)
1384 {
1385     nsChannel *This = impl_from_nsIUploadChannel(iface);
1386     const char *content_type;
1387
1388     static const WCHAR content_typeW[] =
1389         {'C','o','n','t','e','n','t','-','T','y','p','e',0};
1390
1391     TRACE("(%p)->(%p %s %d)\n", This, aStream, debugstr_nsacstr(aContentType), aContentLength);
1392
1393     This->post_data_contains_headers = TRUE;
1394
1395     if(aContentType) {
1396         nsACString_GetData(aContentType, &content_type);
1397         if(*content_type) {
1398             WCHAR *ct;
1399
1400             ct = heap_strdupAtoW(content_type);
1401             if(!ct)
1402                 return NS_ERROR_UNEXPECTED;
1403
1404             set_http_header(&This->request_headers, content_typeW,
1405                     sizeof(content_typeW)/sizeof(WCHAR), ct, strlenW(ct));
1406             heap_free(ct);
1407             This->post_data_contains_headers = FALSE;
1408         }
1409     }
1410
1411     if(This->post_data_stream)
1412         nsIInputStream_Release(This->post_data_stream);
1413
1414     if(aContentLength != -1)
1415         FIXME("Unsupported acontentLength = %d\n", aContentLength);
1416
1417     if(This->post_data_stream)
1418         nsIInputStream_Release(This->post_data_stream);
1419     This->post_data_stream = aStream;
1420     if(aStream)
1421         nsIInputStream_AddRef(aStream);
1422
1423     This->request_method = METHOD_POST;
1424     return NS_OK;
1425 }
1426
1427 static nsresult NSAPI nsUploadChannel_GetUploadStream(nsIUploadChannel *iface,
1428         nsIInputStream **aUploadStream)
1429 {
1430     nsChannel *This = impl_from_nsIUploadChannel(iface);
1431
1432     TRACE("(%p)->(%p)\n", This, aUploadStream);
1433
1434     if(This->post_data_stream)
1435         nsIInputStream_AddRef(This->post_data_stream);
1436
1437     *aUploadStream = This->post_data_stream;
1438     return NS_OK;
1439 }
1440
1441 static const nsIUploadChannelVtbl nsUploadChannelVtbl = {
1442     nsUploadChannel_QueryInterface,
1443     nsUploadChannel_AddRef,
1444     nsUploadChannel_Release,
1445     nsUploadChannel_SetUploadStream,
1446     nsUploadChannel_GetUploadStream
1447 };
1448
1449 static inline nsChannel *impl_from_nsIHttpChannelInternal(nsIHttpChannelInternal *iface)
1450 {
1451     return CONTAINING_RECORD(iface, nsChannel, nsIHttpChannelInternal_iface);
1452 }
1453
1454 static nsresult NSAPI nsHttpChannelInternal_QueryInterface(nsIHttpChannelInternal *iface, nsIIDRef riid,
1455         void **result)
1456 {
1457     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1458     return nsIChannel_QueryInterface(&This->nsIHttpChannel_iface, riid, result);
1459 }
1460
1461 static nsrefcnt NSAPI nsHttpChannelInternal_AddRef(nsIHttpChannelInternal *iface)
1462 {
1463     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1464     return nsIChannel_AddRef(&This->nsIHttpChannel_iface);
1465 }
1466
1467 static nsrefcnt NSAPI nsHttpChannelInternal_Release(nsIHttpChannelInternal *iface)
1468 {
1469     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1470     return nsIChannel_Release(&This->nsIHttpChannel_iface);
1471 }
1472
1473 static nsresult NSAPI nsHttpChannelInternal_GetDocumentURI(nsIHttpChannelInternal *iface, nsIURI **aDocumentURI)
1474 {
1475     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1476
1477     FIXME("(%p)->()\n", This);
1478
1479     return NS_ERROR_NOT_IMPLEMENTED;
1480 }
1481
1482 static nsresult NSAPI nsHttpChannelInternal_SetDocumentURI(nsIHttpChannelInternal *iface, nsIURI *aDocumentURI)
1483 {
1484     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1485
1486     FIXME("(%p)->()\n", This);
1487
1488     return NS_ERROR_NOT_IMPLEMENTED;
1489 }
1490
1491 static nsresult NSAPI nsHttpChannelInternal_GetRequestVersion(nsIHttpChannelInternal *iface, PRUint32 *major, PRUint32 *minor)
1492 {
1493     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1494
1495     FIXME("(%p)->()\n", This);
1496
1497     return NS_ERROR_NOT_IMPLEMENTED;
1498 }
1499
1500 static nsresult NSAPI nsHttpChannelInternal_GetResponseVersion(nsIHttpChannelInternal *iface, PRUint32 *major, PRUint32 *minor)
1501 {
1502     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1503
1504     FIXME("(%p)->()\n", This);
1505
1506     return NS_ERROR_NOT_IMPLEMENTED;
1507 }
1508
1509 static nsresult NSAPI nsHttpChannelInternal_SetCookie(nsIHttpChannelInternal *iface, const char *aCookieHeader)
1510 {
1511     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1512
1513     FIXME("(%p)->()\n", This);
1514
1515     return NS_ERROR_NOT_IMPLEMENTED;
1516 }
1517
1518 static nsresult NSAPI nsHttpChannelInternal_SetupFallbackChannel(nsIHttpChannelInternal *iface, const char *aFallbackKey)
1519 {
1520     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1521
1522     FIXME("(%p)->()\n", This);
1523
1524     return NS_ERROR_NOT_IMPLEMENTED;
1525 }
1526
1527 static nsresult NSAPI nsHttpChannelInternal_GetForceAllowThirdPartyCookie(nsIHttpChannelInternal *iface, PRBool *aForceThirdPartyCookie)
1528 {
1529     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1530
1531     FIXME("(%p)->()\n", This);
1532
1533     return NS_ERROR_NOT_IMPLEMENTED;
1534 }
1535
1536 static nsresult NSAPI nsHttpChannelInternal_SetForceAllowThirdPartyCookie(nsIHttpChannelInternal *iface, PRBool aForceThirdPartyCookie)
1537 {
1538     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1539
1540     FIXME("(%p)->()\n", This);
1541
1542     return NS_ERROR_NOT_IMPLEMENTED;
1543 }
1544
1545 static nsresult NSAPI nsHttpChannelInternal_GetCanceled(nsIHttpChannelInternal *iface, PRBool *aCanceled)
1546 {
1547     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1548
1549     FIXME("(%p)->(%p)\n", This, aCanceled);
1550
1551     return NS_ERROR_NOT_IMPLEMENTED;
1552 }
1553
1554 static nsresult NSAPI nsHttpChannelInternal_GetChannelIsForDownload(nsIHttpChannelInternal *iface, PRBool *aCanceled)
1555 {
1556     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1557
1558     FIXME("(%p)->(%p)\n", This, aCanceled);
1559
1560     return NS_ERROR_NOT_IMPLEMENTED;
1561 }
1562
1563 static nsresult NSAPI nsHttpChannelInternal_SetChannelIsForDownload(nsIHttpChannelInternal *iface, PRBool aCanceled)
1564 {
1565     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1566
1567     FIXME("(%p)->(%x)\n", This, aCanceled);
1568
1569     return NS_ERROR_NOT_IMPLEMENTED;
1570 }
1571
1572 static nsresult NSAPI nsHttpChannelInternal_GetLocalAddress(nsIHttpChannelInternal *iface, nsACString *aLocalAddress)
1573 {
1574     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1575
1576     FIXME("(%p)->(%p)\n", This, aLocalAddress);
1577
1578     return NS_ERROR_NOT_IMPLEMENTED;
1579 }
1580
1581 static nsresult NSAPI nsHttpChannelInternal_GetLocalPort(nsIHttpChannelInternal *iface, PRInt32 *aLocalPort)
1582 {
1583     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1584
1585     FIXME("(%p)->(%p)\n", This, aLocalPort);
1586
1587     return NS_ERROR_NOT_IMPLEMENTED;
1588 }
1589
1590 static nsresult NSAPI nsHttpChannelInternal_GetRemoteAddress(nsIHttpChannelInternal *iface, nsACString *aRemoteAddress)
1591 {
1592     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1593
1594     FIXME("(%p)->(%p)\n", This, aRemoteAddress);
1595
1596     return NS_ERROR_NOT_IMPLEMENTED;
1597 }
1598
1599 static nsresult NSAPI nsHttpChannelInternal_GetRemotePort(nsIHttpChannelInternal *iface, PRInt32 *aRemotePort)
1600 {
1601     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1602
1603     FIXME("(%p)->(%p)\n", This, aRemotePort);
1604
1605     return NS_ERROR_NOT_IMPLEMENTED;
1606 }
1607
1608 static nsresult NSAPI nsHttpChannelInternal_SetCacheKeysRedirectChain(nsIHttpChannelInternal *iface, void *cacheKeys)
1609 {
1610     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1611
1612     FIXME("(%p)->(%p)\n", This, cacheKeys);
1613
1614     return NS_ERROR_NOT_IMPLEMENTED;
1615 }
1616
1617 static const nsIHttpChannelInternalVtbl nsHttpChannelInternalVtbl = {
1618     nsHttpChannelInternal_QueryInterface,
1619     nsHttpChannelInternal_AddRef,
1620     nsHttpChannelInternal_Release,
1621     nsHttpChannelInternal_GetDocumentURI,
1622     nsHttpChannelInternal_SetDocumentURI,
1623     nsHttpChannelInternal_GetRequestVersion,
1624     nsHttpChannelInternal_GetResponseVersion,
1625     nsHttpChannelInternal_SetCookie,
1626     nsHttpChannelInternal_SetupFallbackChannel,
1627     nsHttpChannelInternal_GetForceAllowThirdPartyCookie,
1628     nsHttpChannelInternal_SetForceAllowThirdPartyCookie,
1629     nsHttpChannelInternal_GetCanceled,
1630     nsHttpChannelInternal_GetChannelIsForDownload,
1631     nsHttpChannelInternal_SetChannelIsForDownload,
1632     nsHttpChannelInternal_GetLocalAddress,
1633     nsHttpChannelInternal_GetLocalPort,
1634     nsHttpChannelInternal_GetRemoteAddress,
1635     nsHttpChannelInternal_GetRemotePort,
1636     nsHttpChannelInternal_SetCacheKeysRedirectChain
1637 };
1638
1639
1640 static void invalidate_uri(nsWineURI *This)
1641 {
1642     if(This->uri) {
1643         IUri_Release(This->uri);
1644         This->uri = NULL;
1645     }
1646 }
1647
1648 static BOOL ensure_uri_builder(nsWineURI *This)
1649 {
1650     if(!This->uri_builder) {
1651         HRESULT hres;
1652
1653         if(!ensure_uri(This))
1654             return FALSE;
1655
1656         hres = CreateIUriBuilder(This->uri, 0, 0, &This->uri_builder);
1657         if(FAILED(hres)) {
1658             WARN("CreateIUriBuilder failed: %08x\n", hres);
1659             return FALSE;
1660         }
1661     }
1662
1663     invalidate_uri(This);
1664     return TRUE;
1665 }
1666
1667 static nsresult get_uri_string(nsWineURI *This, Uri_PROPERTY prop, nsACString *ret)
1668 {
1669     char *vala;
1670     BSTR val;
1671     HRESULT hres;
1672
1673     if(!ensure_uri(This))
1674         return NS_ERROR_UNEXPECTED;
1675
1676     hres = IUri_GetPropertyBSTR(This->uri, prop, &val, 0);
1677     if(FAILED(hres)) {
1678         WARN("GetPropertyBSTR failed: %08x\n", hres);
1679         return NS_ERROR_UNEXPECTED;
1680     }
1681
1682     vala = heap_strdupWtoA(val);
1683     SysFreeString(val);
1684     if(!vala)
1685         return NS_ERROR_OUT_OF_MEMORY;
1686
1687     TRACE("ret %s\n", debugstr_a(vala));
1688     nsACString_SetData(ret, vala);
1689     heap_free(vala);
1690     return NS_OK;
1691 }
1692
1693 static inline nsWineURI *impl_from_nsIURL(nsIURL *iface)
1694 {
1695     return CONTAINING_RECORD(iface, nsWineURI, nsIURL_iface);
1696 }
1697
1698 static nsresult NSAPI nsURI_QueryInterface(nsIURL *iface, nsIIDRef riid, void **result)
1699 {
1700     nsWineURI *This = impl_from_nsIURL(iface);
1701
1702     *result = NULL;
1703
1704     if(IsEqualGUID(&IID_nsISupports, riid)) {
1705         TRACE("(%p)->(IID_nsISupports %p)\n", This, result);
1706         *result = &This->nsIURL_iface;
1707     }else if(IsEqualGUID(&IID_nsIURI, riid)) {
1708         TRACE("(%p)->(IID_nsIURI %p)\n", This, result);
1709         *result = &This->nsIURL_iface;
1710     }else if(IsEqualGUID(&IID_nsIURL, riid)) {
1711         TRACE("(%p)->(IID_nsIURL %p)\n", This, result);
1712         *result = &This->nsIURL_iface;
1713     }else if(IsEqualGUID(&IID_nsIMutable, riid)) {
1714         TRACE("(%p)->(IID_nsIMutable %p)\n", This, result);
1715         *result = &This->nsIStandardURL_iface;
1716     }else if(IsEqualGUID(&IID_nsIStandardURL, riid)) {
1717         TRACE("(%p)->(IID_nsIStandardURL %p)\n", This, result);
1718         *result = &This->nsIStandardURL_iface;
1719     }else if(IsEqualGUID(&IID_nsWineURI, riid)) {
1720         TRACE("(%p)->(IID_nsWineURI %p)\n", This, result);
1721         *result = This;
1722     }
1723
1724     if(*result) {
1725         nsIURI_AddRef(&This->nsIURL_iface);
1726         return NS_OK;
1727     }
1728
1729     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), result);
1730     return This->nsuri ? nsIURI_QueryInterface(This->nsuri, riid, result) : NS_NOINTERFACE;
1731 }
1732
1733 static nsrefcnt NSAPI nsURI_AddRef(nsIURL *iface)
1734 {
1735     nsWineURI *This = impl_from_nsIURL(iface);
1736     LONG ref = InterlockedIncrement(&This->ref);
1737
1738     TRACE("(%p) ref=%d\n", This, ref);
1739
1740     return ref;
1741 }
1742
1743 static nsrefcnt NSAPI nsURI_Release(nsIURL *iface)
1744 {
1745     nsWineURI *This = impl_from_nsIURL(iface);
1746     LONG ref = InterlockedDecrement(&This->ref);
1747
1748     TRACE("(%p) ref=%d\n", This, ref);
1749
1750     if(!ref) {
1751         if(This->window_ref)
1752             windowref_release(This->window_ref);
1753         if(This->container)
1754             nsIWebBrowserChrome_Release(&This->container->nsIWebBrowserChrome_iface);
1755         if(This->nsurl)
1756             nsIURL_Release(This->nsurl);
1757         if(This->nsuri)
1758             nsIURI_Release(This->nsuri);
1759         if(This->uri)
1760             IUri_Release(This->uri);
1761         heap_free(This);
1762     }
1763
1764     return ref;
1765 }
1766
1767 static nsresult NSAPI nsURI_GetSpec(nsIURL *iface, nsACString *aSpec)
1768 {
1769     nsWineURI *This = impl_from_nsIURL(iface);
1770
1771     TRACE("(%p)->(%p)\n", This, aSpec);
1772
1773     return get_uri_string(This, Uri_PROPERTY_DISPLAY_URI, aSpec);
1774 }
1775
1776 static nsresult NSAPI nsURI_SetSpec(nsIURL *iface, const nsACString *aSpec)
1777 {
1778     nsWineURI *This = impl_from_nsIURL(iface);
1779     const char *speca;
1780     WCHAR *spec;
1781     IUri *uri;
1782     HRESULT hres;
1783
1784     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aSpec));
1785
1786     nsACString_GetData(aSpec, &speca);
1787     spec = heap_strdupAtoW(speca);
1788     if(!spec)
1789         return NS_ERROR_OUT_OF_MEMORY;
1790
1791     hres = CreateUri(spec, 0, 0, &uri);
1792     heap_free(spec);
1793     if(FAILED(hres)) {
1794         WARN("CreateUri failed: %08x\n", hres);
1795         return NS_ERROR_FAILURE;
1796     }
1797
1798     invalidate_uri(This);
1799     if(This->uri_builder) {
1800         IUriBuilder_Release(This->uri_builder);
1801         This->uri_builder = NULL;
1802     }
1803
1804     This->uri = uri;
1805     return NS_OK;
1806 }
1807
1808 static nsresult NSAPI nsURI_GetPrePath(nsIURL *iface, nsACString *aPrePath)
1809 {
1810     nsWineURI *This = impl_from_nsIURL(iface);
1811
1812     TRACE("(%p)->(%p)\n", This, aPrePath);
1813
1814     if(This->nsuri)
1815         return nsIURI_GetPrePath(This->nsuri, aPrePath);
1816
1817     FIXME("default action not implemented\n");
1818     return NS_ERROR_NOT_IMPLEMENTED;
1819 }
1820
1821 static nsresult NSAPI nsURI_GetScheme(nsIURL *iface, nsACString *aScheme)
1822 {
1823     nsWineURI *This = impl_from_nsIURL(iface);
1824     DWORD scheme;
1825     HRESULT hres;
1826
1827     TRACE("(%p)->(%p)\n", This, aScheme);
1828
1829     if(!ensure_uri(This))
1830         return NS_ERROR_UNEXPECTED;
1831
1832     hres = IUri_GetScheme(This->uri, &scheme);
1833     if(FAILED(hres)) {
1834         WARN("GetScheme failed: %08x\n", hres);
1835         return NS_ERROR_UNEXPECTED;
1836     }
1837
1838     if(scheme == URL_SCHEME_ABOUT) {
1839         nsACString_SetData(aScheme, "wine");
1840         return NS_OK;
1841     }
1842
1843     return get_uri_string(This, Uri_PROPERTY_SCHEME_NAME, aScheme);
1844 }
1845
1846 static nsresult NSAPI nsURI_SetScheme(nsIURL *iface, const nsACString *aScheme)
1847 {
1848     nsWineURI *This = impl_from_nsIURL(iface);
1849     const char *schemea;
1850     WCHAR *scheme;
1851     HRESULT hres;
1852
1853     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aScheme));
1854
1855     if(!ensure_uri_builder(This))
1856         return NS_ERROR_UNEXPECTED;
1857
1858     nsACString_GetData(aScheme, &schemea);
1859     scheme = heap_strdupAtoW(schemea);
1860     if(!scheme)
1861         return NS_ERROR_OUT_OF_MEMORY;
1862
1863     hres = IUriBuilder_SetSchemeName(This->uri_builder, scheme);
1864     heap_free(scheme);
1865     if(FAILED(hres))
1866         return NS_ERROR_UNEXPECTED;
1867
1868     return NS_OK;
1869 }
1870
1871 static nsresult NSAPI nsURI_GetUserPass(nsIURL *iface, nsACString *aUserPass)
1872 {
1873     nsWineURI *This = impl_from_nsIURL(iface);
1874     BSTR user, pass;
1875     HRESULT hres;
1876
1877     TRACE("(%p)->(%p)\n", This, aUserPass);
1878
1879     if(!ensure_uri(This))
1880         return NS_ERROR_UNEXPECTED;
1881
1882     hres = IUri_GetUserName(This->uri, &user);
1883     if(FAILED(hres))
1884         return NS_ERROR_FAILURE;
1885
1886     hres = IUri_GetPassword(This->uri, &pass);
1887     if(FAILED(hres)) {
1888         SysFreeString(user);
1889         return NS_ERROR_FAILURE;
1890     }
1891
1892     if(*user || *pass) {
1893         FIXME("Construct user:pass string\n");
1894     }else {
1895         nsACString_SetData(aUserPass, "");
1896     }
1897
1898     SysFreeString(user);
1899     SysFreeString(pass);
1900     return NS_OK;
1901 }
1902
1903 static nsresult NSAPI nsURI_SetUserPass(nsIURL *iface, const nsACString *aUserPass)
1904 {
1905     nsWineURI *This = impl_from_nsIURL(iface);
1906     WCHAR *user = NULL, *pass = NULL, *buf = NULL;
1907     const char *user_pass;
1908     HRESULT hres;
1909
1910     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aUserPass));
1911
1912     if(!ensure_uri_builder(This))
1913         return NS_ERROR_UNEXPECTED;
1914
1915     nsACString_GetData(aUserPass, &user_pass);
1916     if(*user_pass) {
1917         WCHAR *ptr;
1918
1919         buf = heap_strdupAtoW(user_pass);
1920         if(!buf)
1921             return NS_ERROR_OUT_OF_MEMORY;
1922
1923         ptr = strchrW(buf, ':');
1924         if(!ptr) {
1925             user = buf;
1926         }else if(ptr != buf) {
1927             *ptr++ = 0;
1928             user = buf;
1929             if(*ptr)
1930                 pass = ptr;
1931         }else {
1932             pass = buf+1;
1933         }
1934     }
1935
1936     hres = IUriBuilder_SetUserName(This->uri_builder, user);
1937     if(SUCCEEDED(hres))
1938         hres = IUriBuilder_SetPassword(This->uri_builder, pass);
1939
1940     heap_free(buf);
1941     return SUCCEEDED(hres) ? NS_OK : NS_ERROR_FAILURE;
1942 }
1943
1944 static nsresult NSAPI nsURI_GetUsername(nsIURL *iface, nsACString *aUsername)
1945 {
1946     nsWineURI *This = impl_from_nsIURL(iface);
1947
1948     TRACE("(%p)->(%p)\n", This, aUsername);
1949
1950     return get_uri_string(This, Uri_PROPERTY_USER_NAME, aUsername);
1951 }
1952
1953 static nsresult NSAPI nsURI_SetUsername(nsIURL *iface, const nsACString *aUsername)
1954 {
1955     nsWineURI *This = impl_from_nsIURL(iface);
1956     const char *usera;
1957     WCHAR *user;
1958     HRESULT hres;
1959
1960     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aUsername));
1961
1962     if(!ensure_uri_builder(This))
1963         return NS_ERROR_UNEXPECTED;
1964
1965     nsACString_GetData(aUsername, &usera);
1966     user = heap_strdupAtoW(usera);
1967     if(!user)
1968         return NS_ERROR_OUT_OF_MEMORY;
1969
1970     hres = IUriBuilder_SetUserName(This->uri_builder, user);
1971     heap_free(user);
1972     if(FAILED(hres))
1973         return NS_ERROR_UNEXPECTED;
1974
1975     return NS_OK;
1976 }
1977
1978 static nsresult NSAPI nsURI_GetPassword(nsIURL *iface, nsACString *aPassword)
1979 {
1980     nsWineURI *This = impl_from_nsIURL(iface);
1981
1982     TRACE("(%p)->(%p)\n", This, aPassword);
1983
1984     return get_uri_string(This, Uri_PROPERTY_PASSWORD, aPassword);
1985 }
1986
1987 static nsresult NSAPI nsURI_SetPassword(nsIURL *iface, const nsACString *aPassword)
1988 {
1989     nsWineURI *This = impl_from_nsIURL(iface);
1990     const char *passa;
1991     WCHAR *pass;
1992     HRESULT hres;
1993
1994     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aPassword));
1995
1996     if(!ensure_uri_builder(This))
1997         return NS_ERROR_UNEXPECTED;
1998
1999     nsACString_GetData(aPassword, &passa);
2000     pass = heap_strdupAtoW(passa);
2001     if(!pass)
2002         return NS_ERROR_OUT_OF_MEMORY;
2003
2004     hres = IUriBuilder_SetPassword(This->uri_builder, pass);
2005     heap_free(pass);
2006     if(FAILED(hres))
2007         return NS_ERROR_UNEXPECTED;
2008
2009     return NS_OK;
2010 }
2011
2012 static nsresult NSAPI nsURI_GetHostPort(nsIURL *iface, nsACString *aHostPort)
2013 {
2014     nsWineURI *This = impl_from_nsIURL(iface);
2015     const WCHAR *ptr;
2016     char *vala;
2017     BSTR val;
2018     HRESULT hres;
2019
2020     TRACE("(%p)->(%p)\n", This, aHostPort);
2021
2022     if(!ensure_uri(This))
2023         return NS_ERROR_UNEXPECTED;
2024
2025     hres = IUri_GetAuthority(This->uri, &val);
2026     if(FAILED(hres)) {
2027         WARN("GetAuthority failed: %08x\n", hres);
2028         return NS_ERROR_UNEXPECTED;
2029     }
2030
2031     ptr = strchrW(val, '@');
2032     if(!ptr)
2033         ptr = val;
2034
2035     vala = heap_strdupWtoA(ptr);
2036     SysFreeString(val);
2037     if(!vala)
2038         return NS_ERROR_OUT_OF_MEMORY;
2039
2040     TRACE("ret %s\n", debugstr_a(vala));
2041     nsACString_SetData(aHostPort, vala);
2042     heap_free(vala);
2043     return NS_OK;
2044 }
2045
2046 static nsresult NSAPI nsURI_SetHostPort(nsIURL *iface, const nsACString *aHostPort)
2047 {
2048     nsWineURI *This = impl_from_nsIURL(iface);
2049
2050     WARN("(%p)->(%s)\n", This, debugstr_nsacstr(aHostPort));
2051
2052     /* Not implemented by Gecko */
2053     return NS_ERROR_NOT_IMPLEMENTED;
2054 }
2055
2056 static nsresult NSAPI nsURI_GetHost(nsIURL *iface, nsACString *aHost)
2057 {
2058     nsWineURI *This = impl_from_nsIURL(iface);
2059
2060     TRACE("(%p)->(%p)\n", This, aHost);
2061
2062     return get_uri_string(This, Uri_PROPERTY_HOST, aHost);
2063 }
2064
2065 static nsresult NSAPI nsURI_SetHost(nsIURL *iface, const nsACString *aHost)
2066 {
2067     nsWineURI *This = impl_from_nsIURL(iface);
2068     const char *hosta;
2069     WCHAR *host;
2070     HRESULT hres;
2071
2072     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aHost));
2073
2074     if(!ensure_uri_builder(This))
2075         return NS_ERROR_UNEXPECTED;
2076
2077     nsACString_GetData(aHost, &hosta);
2078     host = heap_strdupAtoW(hosta);
2079     if(!host)
2080         return NS_ERROR_OUT_OF_MEMORY;
2081
2082     hres = IUriBuilder_SetHost(This->uri_builder, host);
2083     heap_free(host);
2084     if(FAILED(hres))
2085         return NS_ERROR_UNEXPECTED;
2086
2087     return NS_OK;
2088 }
2089
2090 static nsresult NSAPI nsURI_GetPort(nsIURL *iface, PRInt32 *aPort)
2091 {
2092     nsWineURI *This = impl_from_nsIURL(iface);
2093     DWORD port;
2094     HRESULT hres;
2095
2096     TRACE("(%p)->(%p)\n", This, aPort);
2097
2098     if(!ensure_uri(This))
2099         return NS_ERROR_UNEXPECTED;
2100
2101     hres = IUri_GetPort(This->uri, &port);
2102     if(FAILED(hres)) {
2103         WARN("GetPort failed: %08x\n", hres);
2104         return NS_ERROR_UNEXPECTED;
2105     }
2106
2107     *aPort = port ? port : -1;
2108     return NS_OK;
2109 }
2110
2111 static nsresult NSAPI nsURI_SetPort(nsIURL *iface, PRInt32 aPort)
2112 {
2113     nsWineURI *This = impl_from_nsIURL(iface);
2114     HRESULT hres;
2115
2116     TRACE("(%p)->(%d)\n", This, aPort);
2117
2118     if(!ensure_uri_builder(This))
2119         return NS_ERROR_UNEXPECTED;
2120
2121     hres = IUriBuilder_SetPort(This->uri_builder, aPort != -1, aPort);
2122     return SUCCEEDED(hres) ? NS_OK : NS_ERROR_FAILURE;
2123 }
2124
2125 static nsresult NSAPI nsURI_GetPath(nsIURL *iface, nsACString *aPath)
2126 {
2127     nsWineURI *This = impl_from_nsIURL(iface);
2128
2129     TRACE("(%p)->(%p)\n", This, aPath);
2130
2131     return get_uri_string(This, Uri_PROPERTY_PATH, aPath);
2132 }
2133
2134 static nsresult NSAPI nsURI_SetPath(nsIURL *iface, const nsACString *aPath)
2135 {
2136     nsWineURI *This = impl_from_nsIURL(iface);
2137     const char *patha;
2138     WCHAR *path;
2139     HRESULT hres;
2140
2141     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aPath));
2142
2143     if(!ensure_uri_builder(This))
2144         return NS_ERROR_UNEXPECTED;
2145
2146     nsACString_GetData(aPath, &patha);
2147     path = heap_strdupAtoW(patha);
2148     if(!path)
2149         return NS_ERROR_OUT_OF_MEMORY;
2150
2151     hres = IUriBuilder_SetPath(This->uri_builder, path);
2152     heap_free(path);
2153     if(FAILED(hres))
2154         return NS_ERROR_UNEXPECTED;
2155
2156     return NS_OK;
2157 }
2158
2159 static nsresult NSAPI nsURL_GetRef(nsIURL *iface, nsACString *aRef)
2160 {
2161     nsWineURI *This = impl_from_nsIURL(iface);
2162     char *refa = NULL;
2163     BSTR ref;
2164     HRESULT hres;
2165
2166     TRACE("(%p)->(%p)\n", This, aRef);
2167
2168     if(!ensure_uri(This))
2169         return NS_ERROR_UNEXPECTED;
2170
2171     hres = IUri_GetFragment(This->uri, &ref);
2172     if(FAILED(hres))
2173         return NS_ERROR_UNEXPECTED;
2174
2175     refa = heap_strdupWtoA(ref);
2176     SysFreeString(ref);
2177     if(ref && !refa)
2178         return NS_ERROR_OUT_OF_MEMORY;
2179
2180     nsACString_SetData(aRef, refa && *refa == '#' ? refa+1 : refa);
2181     heap_free(refa);
2182     return NS_OK;
2183 }
2184
2185 static nsresult NSAPI nsURL_SetRef(nsIURL *iface, const nsACString *aRef)
2186 {
2187     nsWineURI *This = impl_from_nsIURL(iface);
2188     const char *refa;
2189     WCHAR *ref;
2190     HRESULT hres;
2191
2192     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aRef));
2193
2194     if(!ensure_uri_builder(This))
2195         return NS_ERROR_UNEXPECTED;
2196
2197     nsACString_GetData(aRef, &refa);
2198     ref = heap_strdupAtoW(refa);
2199     if(!ref)
2200         return NS_ERROR_OUT_OF_MEMORY;
2201
2202     hres = IUriBuilder_SetFragment(This->uri_builder, ref);
2203     heap_free(ref);
2204     if(FAILED(hres))
2205         return NS_ERROR_UNEXPECTED;
2206
2207     return NS_OK;
2208 }
2209
2210 static nsresult NSAPI nsURI_Equals(nsIURL *iface, nsIURI *other, PRBool *_retval)
2211 {
2212     nsWineURI *This = impl_from_nsIURL(iface);
2213     nsWineURI *other_obj;
2214     nsresult nsres;
2215     HRESULT hres;
2216
2217     TRACE("(%p)->(%p %p)\n", This, other, _retval);
2218
2219     nsres = nsIURI_QueryInterface(other, &IID_nsWineURI, (void**)&other_obj);
2220     if(NS_FAILED(nsres)) {
2221         TRACE("Could not get nsWineURI interface\n");
2222         *_retval = FALSE;
2223         return NS_OK;
2224     }
2225
2226     if(ensure_uri(This) && ensure_uri(other_obj)) {
2227         BOOL b;
2228
2229         hres = IUri_IsEqual(This->uri, other_obj->uri, &b);
2230         if(SUCCEEDED(hres)) {
2231             *_retval = b;
2232             nsres = NS_OK;
2233         }else {
2234             nsres = NS_ERROR_FAILURE;
2235         }
2236     }else {
2237         nsres = NS_ERROR_UNEXPECTED;
2238     }
2239
2240     nsIURI_Release(&other_obj->nsIURL_iface);
2241     return nsres;
2242 }
2243
2244 static nsresult NSAPI nsURI_SchemeIs(nsIURL *iface, const char *scheme, PRBool *_retval)
2245 {
2246     nsWineURI *This = impl_from_nsIURL(iface);
2247     WCHAR buf[INTERNET_MAX_SCHEME_LENGTH];
2248     BSTR scheme_name;
2249     HRESULT hres;
2250
2251     TRACE("(%p)->(%s %p)\n", This, debugstr_a(scheme), _retval);
2252
2253     if(!ensure_uri(This))
2254         return NS_ERROR_UNEXPECTED;
2255
2256     hres = IUri_GetSchemeName(This->uri, &scheme_name);
2257     if(FAILED(hres))
2258         return NS_ERROR_UNEXPECTED;
2259
2260     MultiByteToWideChar(CP_ACP, 0, scheme, -1, buf, sizeof(buf)/sizeof(WCHAR));
2261     *_retval = !strcmpW(scheme_name, buf);
2262     SysFreeString(scheme_name);
2263     return NS_OK;
2264 }
2265
2266 static nsresult NSAPI nsURI_Clone(nsIURL *iface, nsIURI **_retval)
2267 {
2268     nsWineURI *This = impl_from_nsIURL(iface);
2269     nsIURI *nsuri = NULL;
2270     nsWineURI *wine_uri;
2271     nsresult nsres;
2272
2273     TRACE("(%p)->(%p)\n", This, _retval);
2274
2275     if(!ensure_uri(This))
2276         return NS_ERROR_UNEXPECTED;
2277
2278     if(This->nsuri) {
2279         nsres = nsIURI_Clone(This->nsuri, &nsuri);
2280         if(NS_FAILED(nsres)) {
2281             WARN("Clone failed: %08x\n", nsres);
2282             return nsres;
2283         }
2284     }
2285
2286     nsres = create_nsuri(This->uri, nsuri, This->window_ref ? This->window_ref->window : NULL, This->container, &wine_uri);
2287     if(NS_FAILED(nsres)) {
2288         WARN("create_nsuri failed: %08x\n", nsres);
2289         return nsres;
2290     }
2291
2292     *_retval = (nsIURI*)&wine_uri->nsIURL_iface;
2293     return NS_OK;
2294 }
2295
2296 static nsresult NSAPI nsURI_Resolve(nsIURL *iface, const nsACString *aRelativePath,
2297         nsACString *_retval)
2298 {
2299     nsWineURI *This = impl_from_nsIURL(iface);
2300     const char *patha;
2301     IUri *new_uri;
2302     WCHAR *path;
2303     char *reta;
2304     BSTR ret;
2305     HRESULT hres;
2306
2307     TRACE("(%p)->(%s %p)\n", This, debugstr_nsacstr(aRelativePath), _retval);
2308
2309     if(!ensure_uri(This))
2310         return NS_ERROR_UNEXPECTED;
2311
2312     nsACString_GetData(aRelativePath, &patha);
2313     path = heap_strdupAtoW(patha);
2314     if(!path)
2315         return NS_ERROR_OUT_OF_MEMORY;
2316
2317     hres = CoInternetCombineUrlEx(This->uri, path, URL_ESCAPE_SPACES_ONLY|URL_DONT_ESCAPE_EXTRA_INFO, &new_uri, 0);
2318     heap_free(path);
2319     if(FAILED(hres)) {
2320         ERR("CoIntenetCombineUrlEx failed: %08x\n", hres);
2321         return NS_ERROR_FAILURE;
2322     }
2323
2324     hres = IUri_GetDisplayUri(new_uri, &ret);
2325     IUri_Release(new_uri);
2326     if(FAILED(hres))
2327         return NS_ERROR_FAILURE;
2328
2329     reta = heap_strdupWtoA(ret);
2330     SysFreeString(ret);
2331     if(!reta)
2332         return NS_ERROR_OUT_OF_MEMORY;
2333
2334     TRACE("returning %s\n", debugstr_a(reta));
2335     nsACString_SetData(_retval, reta);
2336     heap_free(reta);
2337     return NS_OK;
2338 }
2339
2340 static nsresult NSAPI nsURI_GetAsciiSpec(nsIURL *iface, nsACString *aAsciiSpec)
2341 {
2342     nsWineURI *This = impl_from_nsIURL(iface);
2343
2344     TRACE("(%p)->(%p)\n", This, aAsciiSpec);
2345
2346     return nsIURI_GetSpec(&This->nsIURL_iface, aAsciiSpec);
2347 }
2348
2349 static nsresult NSAPI nsURI_GetAsciiHost(nsIURL *iface, nsACString *aAsciiHost)
2350 {
2351     nsWineURI *This = impl_from_nsIURL(iface);
2352
2353     WARN("(%p)->(%p) FIXME: Use Uri_PUNYCODE_IDN_HOST flag\n", This, aAsciiHost);
2354
2355     return get_uri_string(This, Uri_PROPERTY_HOST, aAsciiHost);
2356 }
2357
2358 static nsresult NSAPI nsURI_GetOriginCharset(nsIURL *iface, nsACString *aOriginCharset)
2359 {
2360     nsWineURI *This = impl_from_nsIURL(iface);
2361
2362     TRACE("(%p)->(%p)\n", This, aOriginCharset);
2363
2364     if(This->nsuri)
2365         return nsIURI_GetOriginCharset(This->nsuri, aOriginCharset);
2366
2367     FIXME("default action not implemented\n");
2368     return NS_ERROR_NOT_IMPLEMENTED;
2369 }
2370
2371 static nsresult NSAPI nsURI_EqualsExceptRef(nsIURL *iface, nsIURI *other, PRBool *_retval)
2372 {
2373     nsWineURI *This = impl_from_nsIURL(iface);
2374     nsWineURI *other_obj;
2375     nsresult nsres;
2376
2377     TRACE("(%p)->(%p %p)\n", This, other, _retval);
2378
2379     nsres = nsIURI_QueryInterface(other, &IID_nsWineURI, (void**)&other_obj);
2380     if(NS_FAILED(nsres)) {
2381         TRACE("Could not get nsWineURI interface\n");
2382         *_retval = FALSE;
2383         return NS_OK;
2384     }
2385
2386     if(ensure_uri(This) && ensure_uri(other_obj)) {
2387         *_retval = compare_ignoring_frag(This->uri, other_obj->uri);
2388         nsres = NS_OK;
2389     }else {
2390         nsres = NS_ERROR_UNEXPECTED;
2391     }
2392
2393     nsIURI_Release(&other_obj->nsIURL_iface);
2394     return nsres;
2395 }
2396
2397 static nsresult NSAPI nsURI_CloneIgnoreRef(nsIURL *iface, nsIURI **_retval)
2398 {
2399     nsWineURI *This = impl_from_nsIURL(iface);
2400     nsWineURI *wine_uri;
2401     IUri *uri;
2402     nsresult nsres;
2403
2404     TRACE("(%p)->(%p)\n", This, _retval);
2405
2406     if(!ensure_uri(This))
2407         return NS_ERROR_UNEXPECTED;
2408
2409     uri = get_uri_nofrag(This->uri);
2410     if(!uri)
2411         return NS_ERROR_FAILURE;
2412
2413     nsres = create_nsuri(uri, NULL, This->window_ref ? This->window_ref->window : NULL, This->container, &wine_uri);
2414     IUri_Release(uri);
2415     if(NS_FAILED(nsres)) {
2416         WARN("create_nsuri failed: %08x\n", nsres);
2417         return nsres;
2418     }
2419
2420     *_retval = (nsIURI*)&wine_uri->nsIURL_iface;
2421     return NS_OK;
2422 }
2423
2424 static nsresult NSAPI nsURI_GetSpecIgnoringRef(nsIURL *iface, nsACString *aSpecIgnoringRef)
2425 {
2426     nsWineURI *This = impl_from_nsIURL(iface);
2427
2428     FIXME("(%p)->(%p)\n", This, aSpecIgnoringRef);
2429
2430     return nsIURL_GetSpec(&This->nsIURL_iface, aSpecIgnoringRef);
2431 }
2432
2433 static nsresult NSAPI nsURI_GetHasRef(nsIURL *iface, PRBool *aHasRef)
2434 {
2435     nsWineURI *This = impl_from_nsIURL(iface);
2436     BOOL b;
2437     HRESULT hres;
2438
2439     TRACE("(%p)->(%p)\n", This, aHasRef);
2440
2441     if(!ensure_uri(This))
2442         return NS_ERROR_UNEXPECTED;
2443
2444     hres = IUri_HasProperty(This->uri, Uri_PROPERTY_FRAGMENT, &b);
2445     if(FAILED(hres))
2446         return NS_ERROR_FAILURE;
2447
2448     *aHasRef = b;
2449     return NS_OK;
2450 }
2451
2452 static nsresult NSAPI nsURL_GetFilePath(nsIURL *iface, nsACString *aFilePath)
2453 {
2454     nsWineURI *This = impl_from_nsIURL(iface);
2455
2456     TRACE("(%p)->(%p)\n", This, aFilePath);
2457
2458     return nsIURL_GetPath(&This->nsIURL_iface, aFilePath);
2459 }
2460
2461 static nsresult NSAPI nsURL_SetFilePath(nsIURL *iface, const nsACString *aFilePath)
2462 {
2463     nsWineURI *This = impl_from_nsIURL(iface);
2464
2465     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aFilePath));
2466
2467     return nsIURL_SetPath(&This->nsIURL_iface, aFilePath);
2468 }
2469
2470 static nsresult NSAPI nsURL_GetParam(nsIURL *iface, nsACString *aParam)
2471 {
2472     nsWineURI *This = impl_from_nsIURL(iface);
2473
2474     WARN("(%p)->(%p)\n", This, aParam);
2475
2476     /* This is a leftover of ';' special handling in URLs. It will be removed from Gecko soon */
2477     nsACString_SetData(aParam, "");
2478     return NS_OK;
2479 }
2480
2481 static nsresult NSAPI nsURL_SetParam(nsIURL *iface, const nsACString *aParam)
2482 {
2483     nsWineURI *This = impl_from_nsIURL(iface);
2484
2485     WARN("(%p)->(%s)\n", This, debugstr_nsacstr(aParam));
2486
2487     /* Not implemented by Gecko */
2488     return NS_ERROR_NOT_IMPLEMENTED;
2489 }
2490
2491 static nsresult NSAPI nsURL_GetQuery(nsIURL *iface, nsACString *aQuery)
2492 {
2493     nsWineURI *This = impl_from_nsIURL(iface);
2494
2495     TRACE("(%p)->(%p)\n", This, aQuery);
2496
2497     return get_uri_string(This, Uri_PROPERTY_QUERY, aQuery);
2498 }
2499
2500 static nsresult NSAPI nsURL_SetQuery(nsIURL *iface, const nsACString *aQuery)
2501 {
2502     nsWineURI *This = impl_from_nsIURL(iface);
2503     const char *querya;
2504     WCHAR *query;
2505     HRESULT hres;
2506
2507     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aQuery));
2508
2509     if(!ensure_uri_builder(This))
2510         return NS_ERROR_UNEXPECTED;
2511
2512     nsACString_GetData(aQuery, &querya);
2513     query = heap_strdupAtoW(querya);
2514     if(!query)
2515         return NS_ERROR_OUT_OF_MEMORY;
2516
2517     hres = IUriBuilder_SetQuery(This->uri_builder, query);
2518     heap_free(query);
2519     if(FAILED(hres))
2520         return NS_ERROR_UNEXPECTED;
2521
2522     return NS_OK;
2523 }
2524
2525 static nsresult NSAPI nsURL_GetDirectory(nsIURL *iface, nsACString *aDirectory)
2526 {
2527     nsWineURI *This = impl_from_nsIURL(iface);
2528     char *dir = NULL;
2529     WCHAR *ptr;
2530     BSTR path;
2531     HRESULT hres;
2532
2533     TRACE("(%p)->(%p)\n", This, aDirectory);
2534
2535     if(!ensure_uri(This))
2536         return NS_ERROR_UNEXPECTED;
2537
2538     hres = IUri_GetPath(This->uri, &path);
2539     if(FAILED(hres))
2540         return NS_ERROR_FAILURE;
2541
2542     ptr = strrchrW(path, '/');
2543     if(ptr) {
2544         if(ptr[1])
2545             *ptr = 0;
2546         dir = heap_strdupWtoA(path);
2547         if(!dir) {
2548             SysFreeString(path);
2549             return NS_ERROR_OUT_OF_MEMORY;
2550         }
2551     }
2552
2553     SysFreeString(path);
2554     TRACE("ret %s\n", debugstr_a(dir));
2555     nsACString_SetData(aDirectory, dir ? dir : "");
2556     heap_free(dir);
2557     return NS_OK;
2558 }
2559
2560 static nsresult NSAPI nsURL_SetDirectory(nsIURL *iface, const nsACString *aDirectory)
2561 {
2562     nsWineURI *This = impl_from_nsIURL(iface);
2563
2564     WARN("(%p)->(%s)\n", This, debugstr_nsacstr(aDirectory));
2565
2566     /* Not implemented by Gecko */
2567     return NS_ERROR_NOT_IMPLEMENTED;
2568 }
2569
2570 static nsresult NSAPI nsURL_GetFileName(nsIURL *iface, nsACString *aFileName)
2571 {
2572     nsWineURI *This = impl_from_nsIURL(iface);
2573
2574     TRACE("(%p)->(%p)\n", This, aFileName);
2575
2576     if(This->nsurl)
2577         return nsIURL_GetFileName(This->nsurl, aFileName);
2578
2579     FIXME("default action not implemented\n");
2580     return NS_ERROR_NOT_IMPLEMENTED;
2581 }
2582
2583 static nsresult NSAPI nsURL_SetFileName(nsIURL *iface, const nsACString *aFileName)
2584 {
2585     nsWineURI *This = impl_from_nsIURL(iface);
2586
2587     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aFileName));
2588
2589     if(This->nsurl) {
2590         invalidate_uri(This);
2591         return nsIURL_SetFileName(This->nsurl, aFileName);
2592     }
2593
2594     FIXME("default action not implemented\n");
2595     return NS_ERROR_NOT_IMPLEMENTED;
2596 }
2597
2598 static nsresult NSAPI nsURL_GetFileBaseName(nsIURL *iface, nsACString *aFileBaseName)
2599 {
2600     nsWineURI *This = impl_from_nsIURL(iface);
2601
2602     TRACE("(%p)->(%p)\n", This, aFileBaseName);
2603
2604     if(This->nsurl)
2605         return nsIURL_GetFileBaseName(This->nsurl, aFileBaseName);
2606
2607     FIXME("default action not implemented\n");
2608     return NS_ERROR_NOT_IMPLEMENTED;
2609 }
2610
2611 static nsresult NSAPI nsURL_SetFileBaseName(nsIURL *iface, const nsACString *aFileBaseName)
2612 {
2613     nsWineURI *This = impl_from_nsIURL(iface);
2614
2615     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aFileBaseName));
2616
2617     if(This->nsurl) {
2618         invalidate_uri(This);
2619         return nsIURL_SetFileBaseName(This->nsurl, aFileBaseName);
2620     }
2621
2622     FIXME("default action not implemented\n");
2623     return NS_ERROR_NOT_IMPLEMENTED;
2624 }
2625
2626 static nsresult NSAPI nsURL_GetFileExtension(nsIURL *iface, nsACString *aFileExtension)
2627 {
2628     nsWineURI *This = impl_from_nsIURL(iface);
2629
2630     TRACE("(%p)->(%p)\n", This, aFileExtension);
2631
2632     return get_uri_string(This, Uri_PROPERTY_EXTENSION, aFileExtension);
2633 }
2634
2635 static nsresult NSAPI nsURL_SetFileExtension(nsIURL *iface, const nsACString *aFileExtension)
2636 {
2637     nsWineURI *This = impl_from_nsIURL(iface);
2638
2639     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aFileExtension));
2640
2641     if(This->nsurl) {
2642         invalidate_uri(This);
2643         return nsIURL_SetFileExtension(This->nsurl, aFileExtension);
2644     }
2645
2646     FIXME("default action not implemented\n");
2647     return NS_ERROR_NOT_IMPLEMENTED;
2648 }
2649
2650 static nsresult NSAPI nsURL_GetCommonBaseSpec(nsIURL *iface, nsIURI *aURIToCompare, nsACString *_retval)
2651 {
2652     nsWineURI *This = impl_from_nsIURL(iface);
2653
2654     TRACE("(%p)->(%p %p)\n", This, aURIToCompare, _retval);
2655
2656     if(This->nsurl)
2657         return nsIURL_GetCommonBaseSpec(This->nsurl, aURIToCompare, _retval);
2658
2659     FIXME("default action not implemented\n");
2660     return NS_ERROR_NOT_IMPLEMENTED;
2661 }
2662
2663 static nsresult NSAPI nsURL_GetRelativeSpec(nsIURL *iface, nsIURI *aURIToCompare, nsACString *_retval)
2664 {
2665     nsWineURI *This = impl_from_nsIURL(iface);
2666
2667     TRACE("(%p)->(%p %p)\n", This, aURIToCompare, _retval);
2668
2669     if(This->nsurl)
2670         return nsIURL_GetRelativeSpec(This->nsurl, aURIToCompare, _retval);
2671
2672     FIXME("default action not implemented\n");
2673     return NS_ERROR_NOT_IMPLEMENTED;
2674 }
2675
2676 static const nsIURLVtbl nsURLVtbl = {
2677     nsURI_QueryInterface,
2678     nsURI_AddRef,
2679     nsURI_Release,
2680     nsURI_GetSpec,
2681     nsURI_SetSpec,
2682     nsURI_GetPrePath,
2683     nsURI_GetScheme,
2684     nsURI_SetScheme,
2685     nsURI_GetUserPass,
2686     nsURI_SetUserPass,
2687     nsURI_GetUsername,
2688     nsURI_SetUsername,
2689     nsURI_GetPassword,
2690     nsURI_SetPassword,
2691     nsURI_GetHostPort,
2692     nsURI_SetHostPort,
2693     nsURI_GetHost,
2694     nsURI_SetHost,
2695     nsURI_GetPort,
2696     nsURI_SetPort,
2697     nsURI_GetPath,
2698     nsURI_SetPath,
2699     nsURI_Equals,
2700     nsURI_SchemeIs,
2701     nsURI_Clone,
2702     nsURI_Resolve,
2703     nsURI_GetAsciiSpec,
2704     nsURI_GetAsciiHost,
2705     nsURI_GetOriginCharset,
2706     nsURL_GetRef,
2707     nsURL_SetRef,
2708     nsURI_EqualsExceptRef,
2709     nsURI_CloneIgnoreRef,
2710     nsURI_GetSpecIgnoringRef,
2711     nsURI_GetHasRef,
2712     nsURL_GetFilePath,
2713     nsURL_SetFilePath,
2714     nsURL_GetParam,
2715     nsURL_SetParam,
2716     nsURL_GetQuery,
2717     nsURL_SetQuery,
2718     nsURL_GetDirectory,
2719     nsURL_SetDirectory,
2720     nsURL_GetFileName,
2721     nsURL_SetFileName,
2722     nsURL_GetFileBaseName,
2723     nsURL_SetFileBaseName,
2724     nsURL_GetFileExtension,
2725     nsURL_SetFileExtension,
2726     nsURL_GetCommonBaseSpec,
2727     nsURL_GetRelativeSpec
2728 };
2729
2730 static inline nsWineURI *impl_from_nsIStandardURL(nsIStandardURL *iface)
2731 {
2732     return CONTAINING_RECORD(iface, nsWineURI, nsIStandardURL_iface);
2733 }
2734
2735 static nsresult NSAPI nsStandardURL_QueryInterface(nsIStandardURL *iface, nsIIDRef riid,
2736         void **result)
2737 {
2738     nsWineURI *This = impl_from_nsIStandardURL(iface);
2739     return nsIURL_QueryInterface(&This->nsIURL_iface, riid, result);
2740 }
2741
2742 static nsrefcnt NSAPI nsStandardURL_AddRef(nsIStandardURL *iface)
2743 {
2744     nsWineURI *This = impl_from_nsIStandardURL(iface);
2745     return nsIURL_AddRef(&This->nsIURL_iface);
2746 }
2747
2748 static nsrefcnt NSAPI nsStandardURL_Release(nsIStandardURL *iface)
2749 {
2750     nsWineURI *This = impl_from_nsIStandardURL(iface);
2751     return nsIURL_Release(&This->nsIURL_iface);
2752 }
2753
2754 static nsresult NSAPI nsStandardURL_GetMutable(nsIStandardURL *iface, PRBool *aMutable)
2755 {
2756     nsWineURI *This = impl_from_nsIStandardURL(iface);
2757
2758     TRACE("(%p)->(%p)\n", This, aMutable);
2759
2760     *aMutable = TRUE;
2761     return NS_OK;
2762 }
2763
2764 static nsresult NSAPI nsStandardURL_SetMutable(nsIStandardURL *iface, PRBool aMutable)
2765 {
2766     nsWineURI *This = impl_from_nsIStandardURL(iface);
2767     FIXME("(%p)->(%x)\n", This, aMutable);
2768     return NS_ERROR_NOT_IMPLEMENTED;
2769 }
2770
2771 static nsresult NSAPI nsStandardURL_Init(nsIStandardURL *iface, PRUint32 aUrlType, PRInt32 aDefaultPort,
2772         const nsACString *aSpec, const char *aOriginCharset, nsIURI *aBaseURI)
2773 {
2774     nsWineURI *This = impl_from_nsIStandardURL(iface);
2775     FIXME("(%p)->(%d %d %s %s %p)\n", This, aUrlType, aDefaultPort, debugstr_nsacstr(aSpec), debugstr_a(aOriginCharset), aBaseURI);
2776     return NS_ERROR_NOT_IMPLEMENTED;
2777 }
2778
2779 static const nsIStandardURLVtbl nsStandardURLVtbl = {
2780     nsStandardURL_QueryInterface,
2781     nsStandardURL_AddRef,
2782     nsStandardURL_Release,
2783     nsStandardURL_GetMutable,
2784     nsStandardURL_SetMutable,
2785     nsStandardURL_Init
2786 };
2787
2788 static nsresult create_nsuri(IUri *iuri, nsIURI *nsuri, HTMLWindow *window, NSContainer *container, nsWineURI **_retval)
2789 {
2790     nsWineURI *ret = heap_alloc_zero(sizeof(nsWineURI));
2791
2792     ret->nsIURL_iface.lpVtbl = &nsURLVtbl;
2793     ret->nsIStandardURL_iface.lpVtbl = &nsStandardURLVtbl;
2794     ret->ref = 1;
2795     ret->nsuri = nsuri;
2796
2797     set_uri_nscontainer(ret, container);
2798     set_uri_window(ret, window);
2799
2800     IUri_AddRef(iuri);
2801     ret->uri = iuri;
2802
2803     if(nsuri)
2804         nsIURI_QueryInterface(nsuri, &IID_nsIURL, (void**)&ret->nsurl);
2805
2806     TRACE("retval=%p\n", ret);
2807     *_retval = ret;
2808     return NS_OK;
2809 }
2810
2811 HRESULT create_doc_uri(HTMLWindow *window, WCHAR *url, nsWineURI **ret)
2812 {
2813     nsWineURI *uri;
2814     IUri *iuri;
2815     nsresult nsres;
2816     HRESULT hres;
2817
2818     hres = CreateUri(url, 0, 0, &iuri);
2819     if(FAILED(hres))
2820         return hres;
2821
2822     nsres = create_nsuri(iuri, NULL, window, window->doc_obj->nscontainer, &uri);
2823     IUri_Release(iuri);
2824     if(NS_FAILED(nsres))
2825         return E_FAIL;
2826
2827     uri->is_doc_uri = TRUE;
2828
2829     *ret = uri;
2830     return S_OK;
2831 }
2832
2833 static nsresult create_nschannel(nsWineURI *uri, nsChannel **ret)
2834 {
2835     nsChannel *channel;
2836     HRESULT hres;
2837
2838     if(!ensure_uri(uri))
2839         return NS_ERROR_UNEXPECTED;
2840
2841     channel = heap_alloc_zero(sizeof(nsChannel));
2842     if(!channel)
2843         return NS_ERROR_OUT_OF_MEMORY;
2844
2845     channel->nsIHttpChannel_iface.lpVtbl = &nsChannelVtbl;
2846     channel->nsIUploadChannel_iface.lpVtbl = &nsUploadChannelVtbl;
2847     channel->nsIHttpChannelInternal_iface.lpVtbl = &nsHttpChannelInternalVtbl;
2848     channel->ref = 1;
2849     channel->request_method = METHOD_GET;
2850     list_init(&channel->response_headers);
2851     list_init(&channel->request_headers);
2852
2853     nsIURL_AddRef(&uri->nsIURL_iface);
2854     channel->uri = uri;
2855
2856     hres = IUri_GetScheme(uri->uri, &channel->url_scheme);
2857     if(FAILED(hres))
2858         channel->url_scheme = URL_SCHEME_UNKNOWN;
2859
2860     *ret = channel;
2861     return NS_OK;
2862 }
2863
2864 HRESULT create_redirect_nschannel(const WCHAR *url, nsChannel *orig_channel, nsChannel **ret)
2865 {
2866     HTMLWindow *window = NULL;
2867     nsChannel *channel;
2868     nsWineURI *uri;
2869     IUri *iuri;
2870     nsresult nsres;
2871     HRESULT hres;
2872
2873     hres = CreateUri(url, 0, 0, &iuri);
2874     if(FAILED(hres))
2875         return hres;
2876
2877     if(orig_channel->uri->window_ref)
2878         window = orig_channel->uri->window_ref->window;
2879     nsres = create_nsuri(iuri, NULL, window, NULL, &uri);
2880     IUri_Release(iuri);
2881     if(NS_FAILED(nsres))
2882         return E_FAIL;
2883
2884     nsres = create_nschannel(uri, &channel);
2885     nsIURL_Release(&uri->nsIURL_iface);
2886     if(NS_FAILED(nsres))
2887         return E_FAIL;
2888
2889     if(orig_channel->load_group) {
2890         nsILoadGroup_AddRef(orig_channel->load_group);
2891         channel->load_group = orig_channel->load_group;
2892     }
2893
2894     if(orig_channel->notif_callback) {
2895         nsIInterfaceRequestor_AddRef(orig_channel->notif_callback);
2896         channel->notif_callback = orig_channel->notif_callback;
2897     }
2898
2899     channel->load_flags = orig_channel->load_flags | LOAD_REPLACE;
2900
2901     if(orig_channel->request_method == METHOD_POST)
2902         FIXME("unsupported POST method\n");
2903
2904     if(orig_channel->original_uri) {
2905         nsIURI_AddRef(orig_channel->original_uri);
2906         channel->original_uri = orig_channel->original_uri;
2907     }
2908
2909     if(orig_channel->referrer) {
2910         nsIURI_AddRef(orig_channel->referrer);
2911         channel->referrer = orig_channel->referrer;
2912     }
2913
2914     *ret = channel;
2915     return S_OK;
2916 }
2917
2918 typedef struct {
2919     nsIProtocolHandler nsIProtocolHandler_iface;
2920
2921     LONG ref;
2922
2923     nsIProtocolHandler *nshandler;
2924 } nsProtocolHandler;
2925
2926 static inline nsProtocolHandler *impl_from_nsIProtocolHandler(nsIProtocolHandler *iface)
2927 {
2928     return CONTAINING_RECORD(iface, nsProtocolHandler, nsIProtocolHandler_iface);
2929 }
2930
2931 static nsresult NSAPI nsProtocolHandler_QueryInterface(nsIProtocolHandler *iface, nsIIDRef riid,
2932         void **result)
2933 {
2934     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2935
2936     *result = NULL;
2937
2938     if(IsEqualGUID(&IID_nsISupports, riid)) {
2939         TRACE("(%p)->(IID_nsISupports %p)\n", This, result);
2940         *result = &This->nsIProtocolHandler_iface;
2941     }else if(IsEqualGUID(&IID_nsIProtocolHandler, riid)) {
2942         TRACE("(%p)->(IID_nsIProtocolHandler %p)\n", This, result);
2943         *result = &This->nsIProtocolHandler_iface;
2944     }else if(IsEqualGUID(&IID_nsIExternalProtocolHandler, riid)) {
2945         TRACE("(%p)->(IID_nsIExternalProtocolHandler %p), returning NULL\n", This, result);
2946         return NS_NOINTERFACE;
2947     }
2948
2949     if(*result) {
2950         nsISupports_AddRef((nsISupports*)*result);
2951         return NS_OK;
2952     }
2953
2954     WARN("(%s %p)\n", debugstr_guid(riid), result);
2955     return NS_NOINTERFACE;
2956 }
2957
2958 static nsrefcnt NSAPI nsProtocolHandler_AddRef(nsIProtocolHandler *iface)
2959 {
2960     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2961     LONG ref = InterlockedIncrement(&This->ref);
2962
2963     TRACE("(%p) ref=%d\n", This, ref);
2964
2965     return ref;
2966 }
2967
2968 static nsrefcnt NSAPI nsProtocolHandler_Release(nsIProtocolHandler *iface)
2969 {
2970     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2971     LONG ref = InterlockedDecrement(&This->ref);
2972
2973     TRACE("(%p) ref=%d\n", This, ref);
2974
2975     if(!ref) {
2976         if(This->nshandler)
2977             nsIProtocolHandler_Release(This->nshandler);
2978         heap_free(This);
2979     }
2980
2981     return ref;
2982 }
2983
2984 static nsresult NSAPI nsProtocolHandler_GetScheme(nsIProtocolHandler *iface, nsACString *aScheme)
2985 {
2986     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2987
2988     TRACE("(%p)->(%p)\n", This, aScheme);
2989
2990     if(This->nshandler)
2991         return nsIProtocolHandler_GetScheme(This->nshandler, aScheme);
2992     return NS_ERROR_NOT_IMPLEMENTED;
2993 }
2994
2995 static nsresult NSAPI nsProtocolHandler_GetDefaultPort(nsIProtocolHandler *iface,
2996         PRInt32 *aDefaultPort)
2997 {
2998     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2999
3000     TRACE("(%p)->(%p)\n", This, aDefaultPort);
3001
3002     if(This->nshandler)
3003         return nsIProtocolHandler_GetDefaultPort(This->nshandler, aDefaultPort);
3004     return NS_ERROR_NOT_IMPLEMENTED;
3005 }
3006
3007 static nsresult NSAPI nsProtocolHandler_GetProtocolFlags(nsIProtocolHandler *iface,
3008                                                          PRUint32 *aProtocolFlags)
3009 {
3010     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
3011
3012     TRACE("(%p)->(%p)\n", This, aProtocolFlags);
3013
3014     if(This->nshandler)
3015         return nsIProtocolHandler_GetProtocolFlags(This->nshandler, aProtocolFlags);
3016     return NS_ERROR_NOT_IMPLEMENTED;
3017 }
3018
3019 static nsresult NSAPI nsProtocolHandler_NewURI(nsIProtocolHandler *iface,
3020         const nsACString *aSpec, const char *aOriginCharset, nsIURI *aBaseURI, nsIURI **_retval)
3021 {
3022     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
3023
3024     TRACE("((%p)->%s %s %p %p)\n", This, debugstr_nsacstr(aSpec), debugstr_a(aOriginCharset),
3025           aBaseURI, _retval);
3026
3027     if(This->nshandler)
3028         return nsIProtocolHandler_NewURI(This->nshandler, aSpec, aOriginCharset, aBaseURI, _retval);
3029     return NS_ERROR_NOT_IMPLEMENTED;
3030 }
3031
3032 static nsresult NSAPI nsProtocolHandler_NewChannel(nsIProtocolHandler *iface,
3033         nsIURI *aURI, nsIChannel **_retval)
3034 {
3035     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
3036
3037     TRACE("(%p)->(%p %p)\n", This, aURI, _retval);
3038
3039     if(This->nshandler)
3040         return nsIProtocolHandler_NewChannel(This->nshandler, aURI, _retval);
3041     return NS_ERROR_NOT_IMPLEMENTED;
3042 }
3043
3044 static nsresult NSAPI nsProtocolHandler_AllowPort(nsIProtocolHandler *iface,
3045         PRInt32 port, const char *scheme, PRBool *_retval)
3046 {
3047     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
3048
3049     TRACE("(%p)->(%d %s %p)\n", This, port, debugstr_a(scheme), _retval);
3050
3051     if(This->nshandler)
3052         return nsIProtocolHandler_AllowPort(This->nshandler, port, scheme, _retval);
3053     return NS_ERROR_NOT_IMPLEMENTED;
3054 }
3055
3056 static const nsIProtocolHandlerVtbl nsProtocolHandlerVtbl = {
3057     nsProtocolHandler_QueryInterface,
3058     nsProtocolHandler_AddRef,
3059     nsProtocolHandler_Release,
3060     nsProtocolHandler_GetScheme,
3061     nsProtocolHandler_GetDefaultPort,
3062     nsProtocolHandler_GetProtocolFlags,
3063     nsProtocolHandler_NewURI,
3064     nsProtocolHandler_NewChannel,
3065     nsProtocolHandler_AllowPort
3066 };
3067
3068 static nsIProtocolHandler *create_protocol_handler(nsIProtocolHandler *nshandler)
3069 {
3070     nsProtocolHandler *ret = heap_alloc(sizeof(nsProtocolHandler));
3071
3072     ret->nsIProtocolHandler_iface.lpVtbl = &nsProtocolHandlerVtbl;
3073     ret->ref = 1;
3074     ret->nshandler = nshandler;
3075
3076     return &ret->nsIProtocolHandler_iface;
3077 }
3078
3079 static nsresult NSAPI nsIOService_QueryInterface(nsIIOService*,nsIIDRef,void**);
3080
3081 static nsrefcnt NSAPI nsIOService_AddRef(nsIIOService *iface)
3082 {
3083     return 2;
3084 }
3085
3086 static nsrefcnt NSAPI nsIOService_Release(nsIIOService *iface)
3087 {
3088     return 1;
3089 }
3090
3091 static nsresult NSAPI nsIOService_GetProtocolHandler(nsIIOService *iface, const char *aScheme,
3092                                                      nsIProtocolHandler **_retval)
3093 {
3094     nsIExternalProtocolHandler *nsexthandler;
3095     nsIProtocolHandler *nshandler;
3096     nsresult nsres;
3097
3098     TRACE("(%s %p)\n", debugstr_a(aScheme), _retval);
3099
3100     nsres = nsIIOService_GetProtocolHandler(nsio, aScheme, &nshandler);
3101     if(NS_FAILED(nsres)) {
3102         WARN("GetProtocolHandler failed: %08x\n", nsres);
3103         return nsres;
3104     }
3105
3106     nsres = nsIProtocolHandler_QueryInterface(nshandler, &IID_nsIExternalProtocolHandler,
3107                                               (void**)&nsexthandler);
3108     if(NS_FAILED(nsres)) {
3109         *_retval = nshandler;
3110         return NS_OK;
3111     }
3112
3113     nsIExternalProtocolHandler_Release(nsexthandler);
3114     *_retval = create_protocol_handler(nshandler);
3115     TRACE("return %p\n", *_retval);
3116     return NS_OK;
3117 }
3118
3119 static nsresult NSAPI nsIOService_GetProtocolFlags(nsIIOService *iface, const char *aScheme,
3120                                                     PRUint32 *_retval)
3121 {
3122     TRACE("(%s %p)\n", debugstr_a(aScheme), _retval);
3123     return nsIIOService_GetProtocolFlags(nsio, aScheme, _retval);
3124 }
3125
3126 static BOOL is_gecko_special_uri(const char *spec)
3127 {
3128     static const char *special_schemes[] = {"chrome:", "jar:", "moz-safe-about", "resource:", "javascript:", "wyciwyg:"};
3129     int i;
3130
3131     for(i=0; i < sizeof(special_schemes)/sizeof(*special_schemes); i++) {
3132         if(!strncasecmp(spec, special_schemes[i], strlen(special_schemes[i])))
3133             return TRUE;
3134     }
3135
3136     if(!strncasecmp(spec, "file:", 5)) {
3137         const char *ptr = spec+5;
3138         while(*ptr == '/')
3139             ptr++;
3140         return is_gecko_path(ptr);
3141     }
3142
3143     return FALSE;
3144 }
3145
3146 static nsresult NSAPI nsIOService_NewURI(nsIIOService *iface, const nsACString *aSpec,
3147         const char *aOriginCharset, nsIURI *aBaseURI, nsIURI **_retval)
3148 {
3149     nsWineURI *wine_uri, *base_wine_uri = NULL;
3150     WCHAR new_spec[INTERNET_MAX_URL_LENGTH];
3151     HTMLWindow *window = NULL;
3152     const char *spec = NULL;
3153     nsIURI *uri = NULL;
3154     IUri *urlmon_uri;
3155     nsresult nsres;
3156     HRESULT hres;
3157
3158     TRACE("(%s %s %p %p)\n", debugstr_nsacstr(aSpec), debugstr_a(aOriginCharset),
3159           aBaseURI, _retval);
3160
3161     nsACString_GetData(aSpec, &spec);
3162     if(is_gecko_special_uri(spec))
3163         return nsIIOService_NewURI(nsio, aSpec, aOriginCharset, aBaseURI, _retval);
3164
3165     if(!strncmp(spec, "wine:", 5))
3166         spec += 5;
3167
3168     if(aBaseURI) {
3169         nsres = nsIURI_QueryInterface(aBaseURI, &IID_nsWineURI, (void**)&base_wine_uri);
3170         if(NS_SUCCEEDED(nsres)) {
3171             if(!ensure_uri(base_wine_uri))
3172                 return NS_ERROR_UNEXPECTED;
3173             if(base_wine_uri->window_ref)
3174                 window = base_wine_uri->window_ref->window;
3175         }else {
3176             WARN("Could not get base nsWineURI: %08x\n", nsres);
3177         }
3178     }
3179
3180     MultiByteToWideChar(CP_ACP, 0, spec, -1, new_spec, sizeof(new_spec)/sizeof(WCHAR));
3181
3182     if(base_wine_uri) {
3183         hres = CoInternetCombineUrlEx(base_wine_uri->uri, new_spec, URL_ESCAPE_SPACES_ONLY|URL_DONT_ESCAPE_EXTRA_INFO,
3184                 &urlmon_uri, 0);
3185         if(FAILED(hres))
3186             WARN("CoInternetCombineUrlEx failed: %08x\n", hres);
3187     }else {
3188         hres = CreateUri(new_spec, 0, 0, &urlmon_uri);
3189         if(FAILED(hres))
3190             WARN("CreateUri failed: %08x\n", hres);
3191     }
3192
3193     nsres = nsIIOService_NewURI(nsio, aSpec, aOriginCharset, aBaseURI, &uri);
3194     if(NS_FAILED(nsres))
3195         TRACE("NewURI failed: %08x\n", nsres);
3196
3197     if(FAILED(hres)) {
3198         *_retval = uri;
3199         return nsres;
3200     }
3201
3202     nsres = create_nsuri(urlmon_uri, uri, window, NULL, &wine_uri);
3203     IUri_Release(urlmon_uri);
3204     if(base_wine_uri)
3205         nsIURI_Release(&base_wine_uri->nsIURL_iface);
3206     if(NS_FAILED(nsres))
3207         return nsres;
3208
3209     *_retval = (nsIURI*)&wine_uri->nsIURL_iface;
3210     return nsres;
3211 }
3212
3213 static nsresult NSAPI nsIOService_NewFileURI(nsIIOService *iface, nsIFile *aFile,
3214                                              nsIURI **_retval)
3215 {
3216     TRACE("(%p %p)\n", aFile, _retval);
3217     return nsIIOService_NewFileURI(nsio, aFile, _retval);
3218 }
3219
3220 static nsresult NSAPI nsIOService_NewChannelFromURI(nsIIOService *iface, nsIURI *aURI,
3221                                                      nsIChannel **_retval)
3222 {
3223     nsWineURI *wine_uri;
3224     nsChannel *ret;
3225     nsresult nsres;
3226
3227     TRACE("(%p %p)\n", aURI, _retval);
3228
3229     nsres = nsIURI_QueryInterface(aURI, &IID_nsWineURI, (void**)&wine_uri);
3230     if(NS_FAILED(nsres)) {
3231         TRACE("Could not get nsWineURI: %08x\n", nsres);
3232         return nsIIOService_NewChannelFromURI(nsio, aURI, _retval);
3233     }
3234
3235     nsres = create_nschannel(wine_uri, &ret);
3236     nsIURL_Release(&wine_uri->nsIURL_iface);
3237     if(NS_FAILED(nsres))
3238         return nsres;
3239
3240     nsIURI_AddRef(aURI);
3241     ret->original_uri = aURI;
3242
3243     *_retval = (nsIChannel*)&ret->nsIHttpChannel_iface;
3244     return NS_OK;
3245 }
3246
3247 static nsresult NSAPI nsIOService_NewChannel(nsIIOService *iface, const nsACString *aSpec,
3248         const char *aOriginCharset, nsIURI *aBaseURI, nsIChannel **_retval)
3249 {
3250     TRACE("(%s %s %p %p)\n", debugstr_nsacstr(aSpec), debugstr_a(aOriginCharset), aBaseURI, _retval);
3251     return nsIIOService_NewChannel(nsio, aSpec, aOriginCharset, aBaseURI, _retval);
3252 }
3253
3254 static nsresult NSAPI nsIOService_GetOffline(nsIIOService *iface, PRBool *aOffline)
3255 {
3256     TRACE("(%p)\n", aOffline);
3257     return nsIIOService_GetOffline(nsio, aOffline);
3258 }
3259
3260 static nsresult NSAPI nsIOService_SetOffline(nsIIOService *iface, PRBool aOffline)
3261 {
3262     TRACE("(%x)\n", aOffline);
3263     return nsIIOService_SetOffline(nsio, aOffline);
3264 }
3265
3266 static nsresult NSAPI nsIOService_AllowPort(nsIIOService *iface, PRInt32 aPort,
3267                                              const char *aScheme, PRBool *_retval)
3268 {
3269     TRACE("(%d %s %p)\n", aPort, debugstr_a(aScheme), _retval);
3270     return nsIIOService_AllowPort(nsio, aPort, debugstr_a(aScheme), _retval);
3271 }
3272
3273 static nsresult NSAPI nsIOService_ExtractScheme(nsIIOService *iface, const nsACString *urlString,
3274                                                  nsACString * _retval)
3275 {
3276     TRACE("(%s %p)\n", debugstr_nsacstr(urlString), _retval);
3277     return nsIIOService_ExtractScheme(nsio, urlString, _retval);
3278 }
3279
3280 static const nsIIOServiceVtbl nsIOServiceVtbl = {
3281     nsIOService_QueryInterface,
3282     nsIOService_AddRef,
3283     nsIOService_Release,
3284     nsIOService_GetProtocolHandler,
3285     nsIOService_GetProtocolFlags,
3286     nsIOService_NewURI,
3287     nsIOService_NewFileURI,
3288     nsIOService_NewChannelFromURI,
3289     nsIOService_NewChannel,
3290     nsIOService_GetOffline,
3291     nsIOService_SetOffline,
3292     nsIOService_AllowPort,
3293     nsIOService_ExtractScheme
3294 };
3295
3296 static nsIIOService nsIOService = { &nsIOServiceVtbl };
3297
3298 static nsresult NSAPI nsNetUtil_QueryInterface(nsINetUtil *iface, nsIIDRef riid,
3299         void **result)
3300 {
3301     return nsIIOService_QueryInterface(&nsIOService, riid, result);
3302 }
3303
3304 static nsrefcnt NSAPI nsNetUtil_AddRef(nsINetUtil *iface)
3305 {
3306     return 2;
3307 }
3308
3309 static nsrefcnt NSAPI nsNetUtil_Release(nsINetUtil *iface)
3310 {
3311     return 1;
3312 }
3313
3314 static nsresult NSAPI nsNetUtil_ParseContentType(nsINetUtil *iface, const nsACString *aTypeHeader,
3315         nsACString *aCharset, PRBool *aHadCharset, nsACString *aContentType)
3316 {
3317     TRACE("(%s %p %p %p)\n", debugstr_nsacstr(aTypeHeader), aCharset, aHadCharset, aContentType);
3318
3319     return nsINetUtil_ParseContentType(net_util, aTypeHeader, aCharset, aHadCharset, aContentType);
3320 }
3321
3322 static nsresult NSAPI nsNetUtil_ProtocolHasFlags(nsINetUtil *iface, nsIURI *aURI, PRUint32 aFlags, PRBool *_retval)
3323 {
3324     TRACE("()\n");
3325
3326     return nsINetUtil_ProtocolHasFlags(net_util, aURI, aFlags, _retval);
3327 }
3328
3329 static nsresult NSAPI nsNetUtil_URIChainHasFlags(nsINetUtil *iface, nsIURI *aURI, PRUint32 aFlags, PRBool *_retval)
3330 {
3331     TRACE("(%p %08x %p)\n", aURI, aFlags, _retval);
3332
3333     if(aFlags == (1<<11)) {
3334         *_retval = FALSE;
3335         return NS_OK;
3336     }
3337
3338     return nsINetUtil_URIChainHasFlags(net_util, aURI, aFlags, _retval);
3339 }
3340
3341 static nsresult NSAPI nsNetUtil_ToImmutableURI(nsINetUtil *iface, nsIURI *aURI, nsIURI **_retval)
3342 {
3343     TRACE("(%p %p)\n", aURI, _retval);
3344
3345     return nsINetUtil_ToImmutableURI(net_util, aURI, _retval);
3346 }
3347
3348 static nsresult NSAPI nsNetUtil_NewSimpleNestedURI(nsINetUtil *iface, nsIURI *aURI, nsIURI **_retval)
3349 {
3350     TRACE("(%p %p)\n", aURI, _retval);
3351
3352     return nsINetUtil_NewSimpleNestedURI(net_util, aURI, _retval);
3353 }
3354
3355 static nsresult NSAPI nsNetUtil_EscapeString(nsINetUtil *iface, const nsACString *aString,
3356                                              PRUint32 aEscapeType, nsACString *_retval)
3357 {
3358     TRACE("(%s %x %p)\n", debugstr_nsacstr(aString), aEscapeType, _retval);
3359
3360     return nsINetUtil_EscapeString(net_util, aString, aEscapeType, _retval);
3361 }
3362
3363 static nsresult NSAPI nsNetUtil_EscapeURL(nsINetUtil *iface, const nsACString *aStr, PRUint32 aFlags,
3364                                           nsACString *_retval)
3365 {
3366     TRACE("(%s %08x %p)\n", debugstr_nsacstr(aStr), aFlags, _retval);
3367
3368     return nsINetUtil_EscapeURL(net_util, aStr, aFlags, _retval);
3369 }
3370
3371 static nsresult NSAPI nsNetUtil_UnescapeString(nsINetUtil *iface, const nsACString *aStr,
3372                                                PRUint32 aFlags, nsACString *_retval)
3373 {
3374     TRACE("(%s %08x %p)\n", debugstr_nsacstr(aStr), aFlags, _retval);
3375
3376     return nsINetUtil_UnescapeString(net_util, aStr, aFlags, _retval);
3377 }
3378
3379 static nsresult NSAPI nsNetUtil_ExtractCharsetFromContentType(nsINetUtil *iface, const nsACString *aTypeHeader,
3380         nsACString *aCharset, PRInt32 *aCharsetStart, PRInt32 *aCharsetEnd, PRBool *_retval)
3381 {
3382     TRACE("(%s %p %p %p %p)\n", debugstr_nsacstr(aTypeHeader), aCharset, aCharsetStart,
3383           aCharsetEnd, _retval);
3384
3385     return nsINetUtil_ExtractCharsetFromContentType(net_util, aTypeHeader, aCharset, aCharsetStart, aCharsetEnd, _retval);
3386 }
3387
3388 static const nsINetUtilVtbl nsNetUtilVtbl = {
3389     nsNetUtil_QueryInterface,
3390     nsNetUtil_AddRef,
3391     nsNetUtil_Release,
3392     nsNetUtil_ParseContentType,
3393     nsNetUtil_ProtocolHasFlags,
3394     nsNetUtil_URIChainHasFlags,
3395     nsNetUtil_ToImmutableURI,
3396     nsNetUtil_NewSimpleNestedURI,
3397     nsNetUtil_EscapeString,
3398     nsNetUtil_EscapeURL,
3399     nsNetUtil_UnescapeString,
3400     nsNetUtil_ExtractCharsetFromContentType
3401 };
3402
3403 static nsINetUtil nsNetUtil = { &nsNetUtilVtbl };
3404
3405 static nsresult NSAPI nsIOService_QueryInterface(nsIIOService *iface, nsIIDRef riid,
3406         void **result)
3407 {
3408     *result = NULL;
3409
3410     if(IsEqualGUID(&IID_nsISupports, riid))
3411         *result = &nsIOService;
3412     else if(IsEqualGUID(&IID_nsIIOService, riid))
3413         *result = &nsIOService;
3414     else if(IsEqualGUID(&IID_nsINetUtil, riid))
3415         *result = &nsNetUtil;
3416
3417     if(*result) {
3418         nsISupports_AddRef((nsISupports*)*result);
3419         return NS_OK;
3420     }
3421
3422     FIXME("(%s %p)\n", debugstr_guid(riid), result);
3423     return NS_NOINTERFACE;
3424 }
3425
3426 static nsresult NSAPI nsIOServiceFactory_QueryInterface(nsIFactory *iface, nsIIDRef riid,
3427         void **result)
3428 {
3429     *result = NULL;
3430
3431     if(IsEqualGUID(&IID_nsISupports, riid)) {
3432         TRACE("(IID_nsISupports %p)\n", result);
3433         *result = iface;
3434     }else if(IsEqualGUID(&IID_nsIFactory, riid)) {
3435         TRACE("(IID_nsIFactory %p)\n", result);
3436         *result = iface;
3437     }
3438
3439     if(*result) {
3440         nsIFactory_AddRef(iface);
3441         return NS_OK;
3442     }
3443
3444     WARN("(%s %p)\n", debugstr_guid(riid), result);
3445     return NS_NOINTERFACE;
3446 }
3447
3448 static nsrefcnt NSAPI nsIOServiceFactory_AddRef(nsIFactory *iface)
3449 {
3450     return 2;
3451 }
3452
3453 static nsrefcnt NSAPI nsIOServiceFactory_Release(nsIFactory *iface)
3454 {
3455     return 1;
3456 }
3457
3458 static nsresult NSAPI nsIOServiceFactory_CreateInstance(nsIFactory *iface,
3459         nsISupports *aOuter, const nsIID *iid, void **result)
3460 {
3461     return nsIIOService_QueryInterface(&nsIOService, iid, result);
3462 }
3463
3464 static nsresult NSAPI nsIOServiceFactory_LockFactory(nsIFactory *iface, PRBool lock)
3465 {
3466     WARN("(%x)\n", lock);
3467     return NS_OK;
3468 }
3469
3470 static const nsIFactoryVtbl nsIOServiceFactoryVtbl = {
3471     nsIOServiceFactory_QueryInterface,
3472     nsIOServiceFactory_AddRef,
3473     nsIOServiceFactory_Release,
3474     nsIOServiceFactory_CreateInstance,
3475     nsIOServiceFactory_LockFactory
3476 };
3477
3478 static nsIFactory nsIOServiceFactory = { &nsIOServiceFactoryVtbl };
3479
3480 static BOOL translate_url(HTMLDocumentObj *doc, nsWineURI *uri)
3481 {
3482     OLECHAR *new_url = NULL;
3483     WCHAR *url;
3484     BOOL ret = FALSE;
3485     HRESULT hres;
3486
3487     if(!doc->hostui || !ensure_uri(uri))
3488         return FALSE;
3489
3490     hres = IUri_GetDisplayUri(uri->uri, &url);
3491     if(FAILED(hres))
3492         return FALSE;
3493
3494     hres = IDocHostUIHandler_TranslateUrl(doc->hostui, 0, url, &new_url);
3495     if(hres == S_OK && new_url) {
3496         if(strcmpW(url, new_url)) {
3497             FIXME("TranslateUrl returned new URL %s -> %s\n", debugstr_w(url), debugstr_w(new_url));
3498             ret = TRUE;
3499         }
3500         CoTaskMemFree(new_url);
3501     }
3502
3503     SysFreeString(url);
3504     return ret;
3505 }
3506
3507 nsresult on_start_uri_open(NSContainer *nscontainer, nsIURI *uri, PRBool *_retval)
3508 {
3509     nsWineURI *wine_uri;
3510     nsresult nsres;
3511
3512     *_retval = FALSE;
3513
3514     nsres = nsIURI_QueryInterface(uri, &IID_nsWineURI, (void**)&wine_uri);
3515     if(NS_FAILED(nsres)) {
3516         WARN("Could not get nsWineURI: %08x\n", nsres);
3517         return NS_ERROR_NOT_IMPLEMENTED;
3518     }
3519
3520     if(!wine_uri->is_doc_uri) {
3521         wine_uri->is_doc_uri = TRUE;
3522
3523         if(!wine_uri->container) {
3524             nsIWebBrowserChrome_AddRef(&nscontainer->nsIWebBrowserChrome_iface);
3525             wine_uri->container = nscontainer;
3526         }
3527
3528         if(nscontainer->doc)
3529             *_retval = translate_url(nscontainer->doc, wine_uri);
3530     }
3531
3532     nsIURI_Release(&wine_uri->nsIURL_iface);
3533     return NS_OK;
3534 }
3535
3536 void init_nsio(nsIComponentManager *component_manager, nsIComponentRegistrar *registrar)
3537 {
3538     nsIFactory *old_factory = NULL;
3539     nsresult nsres;
3540
3541     nsres = nsIComponentManager_GetClassObject(component_manager, &NS_IOSERVICE_CID,
3542                                                &IID_nsIFactory, (void**)&old_factory);
3543     if(NS_FAILED(nsres)) {
3544         ERR("Could not get factory: %08x\n", nsres);
3545         return;
3546     }
3547
3548     nsres = nsIFactory_CreateInstance(old_factory, NULL, &IID_nsIIOService, (void**)&nsio);
3549     if(NS_FAILED(nsres)) {
3550         ERR("Couldn not create nsIOService instance %08x\n", nsres);
3551         nsIFactory_Release(old_factory);
3552         return;
3553     }
3554
3555     nsres = nsIIOService_QueryInterface(nsio, &IID_nsINetUtil, (void**)&net_util);
3556     if(NS_FAILED(nsres)) {
3557         WARN("Could not get nsINetUtil interface: %08x\n", nsres);
3558         nsIIOService_Release(nsio);
3559         return;
3560     }
3561
3562     nsres = nsIComponentRegistrar_UnregisterFactory(registrar, &NS_IOSERVICE_CID, old_factory);
3563     nsIFactory_Release(old_factory);
3564     if(NS_FAILED(nsres))
3565         ERR("UnregisterFactory failed: %08x\n", nsres);
3566
3567     nsres = nsIComponentRegistrar_RegisterFactory(registrar, &NS_IOSERVICE_CID,
3568             NS_IOSERVICE_CLASSNAME, NS_IOSERVICE_CONTRACTID, &nsIOServiceFactory);
3569     if(NS_FAILED(nsres))
3570         ERR("RegisterFactory failed: %08x\n", nsres);
3571 }
3572
3573 void release_nsio(void)
3574 {
3575     if(net_util) {
3576         nsINetUtil_Release(net_util);
3577         net_util = NULL;
3578     }
3579
3580     if(nsio) {
3581         nsIIOService_Release(nsio);
3582         nsio = NULL;
3583     }
3584 }