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