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

   1 /*
   2  * Copyright 2006-2010 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "config.h"
  20
  21 #include <stdarg.h>
  22 #include <assert.h>
  23
  24 #define COBJMACROS
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "winreg.h"
  30 #include "ole2.h"
  31 #include "shlguid.h"
  32 #include "wininet.h"
  33 #include "shlwapi.h"
  34
  35 #include "wine/debug.h"
  36
  37 #include "mshtml_private.h"
  38 #include "binding.h"
  39
  40 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  41
  42 #define NS_IOSERVICE_CLASSNAME "nsIOService"
  43 #define NS_IOSERVICE_CONTRACTID "@mozilla.org/network/io-service;1"
  44
  45 static const IID NS_IOSERVICE_CID =
  46     {0x9ac9e770, 0x18bc, 0x11d3, {0x93, 0x37, 0x00, 0x10, 0x4b, 0xa0, 0xfd, 0x40}};
  47 static const IID IID_nsWineURI =
  48     {0x5088272e, 0x900b, 0x11da, {0xc6,0x87, 0x00,0x0f,0xea,0x57,0xf2,0x1a}};
  49
  50 static nsIIOService *nsio = NULL;
  51 static nsINetUtil *net_util;
  52
  53 static const 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
 893 static void start_binding_task_destr(task_t *_task)
 894 {
 895     start_binding_task_t *task = (start_binding_task_t*)_task;
 896
 897     IBindStatusCallback_Release(&task->bscallback->bsc.IBindStatusCallback_iface);
 898     heap_free(task);
 899 }
 900
 901 static nsresult async_open(nsChannel *This, HTMLWindow *window, BOOL is_doc_channel, nsIStreamListener *listener,
 902         nsISupports *context)
 903 {
 904     nsChannelBSC *bscallback;
 905     IMoniker *mon = NULL;
 906     HRESULT hres;
 907
 908     hres = CreateURLMonikerEx2(NULL, This->uri->uri, &mon, 0);
 909     if(FAILED(hres)) {
 910         WARN("CreateURLMoniker failed: %08x\n", hres);
 911         return NS_ERROR_UNEXPECTED;
 912     }
 913
 914     if(is_doc_channel)
 915         set_current_mon(window, mon);
 916
 917     hres = create_channelbsc(mon, NULL, NULL, 0, &bscallback);
 918     IMoniker_Release(mon);
 919     if(FAILED(hres))
 920         return NS_ERROR_UNEXPECTED;
 921
 922     channelbsc_set_channel(bscallback, This, listener, context);
 923
 924     if(is_doc_channel) {
 925         set_window_bscallback(window, bscallback);
 926         async_start_doc_binding(window, bscallback);
 927         IUnknown_Release((IUnknown*)bscallback);
 928     }else {
 929         start_binding_task_t *task = heap_alloc(sizeof(start_binding_task_t));
 930
 931         task->doc = window->doc;
 932         task->bscallback = bscallback;
 933         push_task(&task->header, start_binding_proc, start_binding_task_destr, window->doc->basedoc.task_magic);
 934     }
 935
 936     return NS_OK;
 937 }
 938
 939 static nsresult NSAPI nsChannel_AsyncOpen(nsIHttpChannel *iface, nsIStreamListener *aListener,
 940                                           nsISupports *aContext)
 941 {
 942     nsChannel *This = impl_from_nsIHttpChannel(iface);
 943     HTMLWindow *window = NULL;
 944     BOOL cancel = FALSE;
 945     nsresult nsres = NS_OK;
 946
 947     TRACE("(%p)->(%p %p)\n", This, aListener, aContext);
 948
 949     if(!ensure_uri(This->uri))
 950         return NS_ERROR_FAILURE;
 951
 952     if(This->uri->is_doc_uri) {
 953         window = get_channel_window(This);
 954         if(window) {
 955             set_uri_window(This->uri, window);
 956         }else if(This->uri->container) {
 957             BOOL b;
 958
 959             /* nscontainer->doc should be NULL which means navigation to a new window */
 960             if(This->uri->container->doc)
 961                 FIXME("nscontainer->doc = %p\n", This->uri->container->doc);
 962
 963             nsres = before_async_open(This, This->uri->container, &b);
 964             if(NS_FAILED(nsres))
 965                 return nsres;
 966             if(b)
 967                 FIXME("Navigation not cancelled\n");
 968             return NS_ERROR_UNEXPECTED;
 969         }
 970     }
 971
 972     if(!window) {
 973         if(This->uri->window_ref && This->uri->window_ref->window) {
 974             window = This->uri->window_ref->window;
 975             IHTMLWindow2_AddRef(&window->IHTMLWindow2_iface);
 976         }else {
 977             /* FIXME: Analyze removing get_window_from_load_group call */
 978             if(This->load_group)
 979                 window = get_window_from_load_group(This);
 980             if(!window)
 981                 window = get_channel_window(This);
 982             if(window)
 983                 set_uri_window(This->uri, window);
 984         }
 985     }
 986
 987     if(!window) {
 988         ERR("window = NULL\n");
 989         return NS_ERROR_UNEXPECTED;
 990     }
 991
 992     if(This->uri->is_doc_uri && window == window->doc_obj->basedoc.window) {
 993         if(This->uri->channel_bsc) {
 994             channelbsc_set_channel(This->uri->channel_bsc, This, aListener, aContext);
 995
 996             if(window->doc_obj->mime) {
 997                 heap_free(This->content_type);
 998                 This->content_type = heap_strdupWtoA(window->doc_obj->mime);
 999             }
1000
1001             cancel = TRUE;
1002         }else {
1003             nsres = before_async_open(This, window->doc_obj->nscontainer, &cancel);
1004             if(NS_SUCCEEDED(nsres)  && cancel) {
1005                 TRACE("canceled\n");
1006                 nsres = NS_BINDING_ABORTED;
1007             }
1008         }
1009     }
1010
1011     if(!cancel)
1012         nsres = async_open(This, window, This->uri->is_doc_uri, aListener, aContext);
1013
1014     if(NS_SUCCEEDED(nsres) && This->load_group) {
1015         nsres = nsILoadGroup_AddRequest(This->load_group, (nsIRequest*)&This->nsIHttpChannel_iface,
1016                 aContext);
1017         if(NS_FAILED(nsres))
1018             ERR("AddRequest failed: %08x\n", nsres);
1019     }
1020
1021     IHTMLWindow2_Release(&window->IHTMLWindow2_iface);
1022     return nsres;
1023 }
1024
1025 static nsresult NSAPI nsChannel_GetRequestMethod(nsIHttpChannel *iface, nsACString *aRequestMethod)
1026 {
1027     nsChannel *This = impl_from_nsIHttpChannel(iface);
1028
1029     TRACE("(%p)->(%p)\n", This, aRequestMethod);
1030
1031     nsACString_SetData(aRequestMethod, request_method_strings[This->request_method]);
1032     return NS_OK;
1033 }
1034
1035 static nsresult NSAPI nsChannel_SetRequestMethod(nsIHttpChannel *iface,
1036                                                  const nsACString *aRequestMethod)
1037 {
1038     nsChannel *This = impl_from_nsIHttpChannel(iface);
1039     const char *method;
1040     unsigned i;
1041
1042     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aRequestMethod));
1043
1044     nsACString_GetData(aRequestMethod, &method);
1045     for(i=0; i < sizeof(request_method_strings)/sizeof(*request_method_strings); i++) {
1046         if(!strcasecmp(method, request_method_strings[i])) {
1047             This->request_method = i;
1048             return NS_OK;
1049         }
1050     }
1051
1052     ERR("Invalid method %s\n", debugstr_a(method));
1053     return NS_ERROR_UNEXPECTED;
1054 }
1055
1056 static nsresult NSAPI nsChannel_GetReferrer(nsIHttpChannel *iface, nsIURI **aReferrer)
1057 {
1058     nsChannel *This = impl_from_nsIHttpChannel(iface);
1059
1060     TRACE("(%p)->(%p)\n", This, aReferrer);
1061
1062     if(This->referrer)
1063         nsIURI_AddRef(This->referrer);
1064     *aReferrer = This->referrer;
1065     return NS_OK;
1066 }
1067
1068 static nsresult NSAPI nsChannel_SetReferrer(nsIHttpChannel *iface, nsIURI *aReferrer)
1069 {
1070     nsChannel *This = impl_from_nsIHttpChannel(iface);
1071
1072     TRACE("(%p)->(%p)\n", This, aReferrer);
1073
1074     if(aReferrer)
1075         nsIURI_AddRef(aReferrer);
1076     if(This->referrer)
1077         nsIURI_Release(This->referrer);
1078     This->referrer = aReferrer;
1079     return NS_OK;
1080 }
1081
1082 static nsresult NSAPI nsChannel_GetRequestHeader(nsIHttpChannel *iface,
1083          const nsACString *aHeader, nsACString *_retval)
1084 {
1085     nsChannel *This = impl_from_nsIHttpChannel(iface);
1086
1087     TRACE("(%p)->(%s %p)\n", This, debugstr_nsacstr(aHeader), _retval);
1088
1089     return get_channel_http_header(&This->request_headers, aHeader, _retval);
1090 }
1091
1092 static nsresult NSAPI nsChannel_SetRequestHeader(nsIHttpChannel *iface,
1093          const nsACString *aHeader, const nsACString *aValue, PRBool aMerge)
1094 {
1095     nsChannel *This = impl_from_nsIHttpChannel(iface);
1096
1097     TRACE("(%p)->(%s %s %x)\n", This, debugstr_nsacstr(aHeader), debugstr_nsacstr(aValue), aMerge);
1098
1099     if(aMerge)
1100         FIXME("aMerge not supported\n");
1101
1102     return set_channel_http_header(&This->request_headers, aHeader, aValue);
1103 }
1104
1105 static nsresult NSAPI nsChannel_VisitRequestHeaders(nsIHttpChannel *iface,
1106                                                     nsIHttpHeaderVisitor *aVisitor)
1107 {
1108     nsChannel *This = impl_from_nsIHttpChannel(iface);
1109
1110     FIXME("(%p)->(%p)\n", This, aVisitor);
1111
1112     return NS_ERROR_NOT_IMPLEMENTED;
1113 }
1114
1115 static nsresult NSAPI nsChannel_GetAllowPipelining(nsIHttpChannel *iface, PRBool *aAllowPipelining)
1116 {
1117     nsChannel *This = impl_from_nsIHttpChannel(iface);
1118
1119     FIXME("(%p)->(%p)\n", This, aAllowPipelining);
1120
1121     return NS_ERROR_NOT_IMPLEMENTED;
1122 }
1123
1124 static nsresult NSAPI nsChannel_SetAllowPipelining(nsIHttpChannel *iface, PRBool aAllowPipelining)
1125 {
1126     nsChannel *This = impl_from_nsIHttpChannel(iface);
1127
1128     FIXME("(%p)->(%x)\n", This, aAllowPipelining);
1129
1130     return NS_ERROR_NOT_IMPLEMENTED;
1131 }
1132
1133 static nsresult NSAPI nsChannel_GetRedirectionLimit(nsIHttpChannel *iface, PRUint32 *aRedirectionLimit)
1134 {
1135     nsChannel *This = impl_from_nsIHttpChannel(iface);
1136
1137     FIXME("(%p)->(%p)\n", This, aRedirectionLimit);
1138
1139     return NS_ERROR_NOT_IMPLEMENTED;
1140 }
1141
1142 static nsresult NSAPI nsChannel_SetRedirectionLimit(nsIHttpChannel *iface, PRUint32 aRedirectionLimit)
1143 {
1144     nsChannel *This = impl_from_nsIHttpChannel(iface);
1145
1146     FIXME("(%p)->(%u)\n", This, aRedirectionLimit);
1147
1148     return NS_ERROR_NOT_IMPLEMENTED;
1149 }
1150
1151 static nsresult NSAPI nsChannel_GetResponseStatus(nsIHttpChannel *iface, PRUint32 *aResponseStatus)
1152 {
1153     nsChannel *This = impl_from_nsIHttpChannel(iface);
1154
1155     TRACE("(%p)->(%p)\n", This, aResponseStatus);
1156
1157     if(This->response_status) {
1158         *aResponseStatus = This->response_status;
1159         return NS_OK;
1160     }
1161
1162     WARN("No response status\n");
1163     return NS_ERROR_UNEXPECTED;
1164 }
1165
1166 static nsresult NSAPI nsChannel_GetResponseStatusText(nsIHttpChannel *iface,
1167                                                       nsACString *aResponseStatusText)
1168 {
1169     nsChannel *This = impl_from_nsIHttpChannel(iface);
1170
1171     FIXME("(%p)->(%p)\n", This, aResponseStatusText);
1172
1173     return NS_ERROR_NOT_IMPLEMENTED;
1174 }
1175
1176 static nsresult NSAPI nsChannel_GetRequestSucceeded(nsIHttpChannel *iface,
1177                                                     PRBool *aRequestSucceeded)
1178 {
1179     nsChannel *This = impl_from_nsIHttpChannel(iface);
1180
1181     TRACE("(%p)->(%p)\n", This, aRequestSucceeded);
1182
1183     if(!This->response_status)
1184         return NS_ERROR_NOT_AVAILABLE;
1185
1186     *aRequestSucceeded = This->response_status/100 == 2;
1187
1188     return NS_OK;
1189 }
1190
1191 static nsresult NSAPI nsChannel_GetResponseHeader(nsIHttpChannel *iface,
1192          const nsACString *header, nsACString *_retval)
1193 {
1194     nsChannel *This = impl_from_nsIHttpChannel(iface);
1195
1196     TRACE("(%p)->(%s %p)\n", This, debugstr_nsacstr(header), _retval);
1197
1198     return get_channel_http_header(&This->response_headers, header, _retval);
1199 }
1200
1201 static nsresult NSAPI nsChannel_SetResponseHeader(nsIHttpChannel *iface,
1202         const nsACString *header, const nsACString *value, PRBool merge)
1203 {
1204     nsChannel *This = impl_from_nsIHttpChannel(iface);
1205
1206     FIXME("(%p)->(%s %s %x)\n", This, debugstr_nsacstr(header), debugstr_nsacstr(value), merge);
1207
1208     return NS_ERROR_NOT_IMPLEMENTED;
1209 }
1210
1211 static nsresult NSAPI nsChannel_VisitResponseHeaders(nsIHttpChannel *iface,
1212         nsIHttpHeaderVisitor *aVisitor)
1213 {
1214     nsChannel *This = impl_from_nsIHttpChannel(iface);
1215
1216     TRACE("(%p)->(%p)\n", This, aVisitor);
1217
1218     return visit_http_headers(&This->response_headers, aVisitor);
1219 }
1220
1221 static nsresult NSAPI nsChannel_IsNoStoreResponse(nsIHttpChannel *iface, PRBool *_retval)
1222 {
1223     nsChannel *This = impl_from_nsIHttpChannel(iface);
1224     http_header_t *header;
1225
1226     static const WCHAR cache_controlW[] = {'C','a','c','h','e','-','C','o','n','t','r','o','l'};
1227     static const WCHAR no_storeW[] = {'n','o','-','s','t','o','r','e',0};
1228
1229     TRACE("(%p)->(%p)\n", This, _retval);
1230
1231     header = find_http_header(&This->response_headers, cache_controlW, sizeof(cache_controlW)/sizeof(WCHAR));
1232     *_retval = header && !strcmpiW(header->data, no_storeW);
1233     return NS_OK;
1234 }
1235
1236 static nsresult NSAPI nsChannel_IsNoCacheResponse(nsIHttpChannel *iface, PRBool *_retval)
1237 {
1238     nsChannel *This = impl_from_nsIHttpChannel(iface);
1239
1240     FIXME("(%p)->(%p)\n", This, _retval);
1241
1242     return NS_ERROR_NOT_IMPLEMENTED;
1243 }
1244
1245 static const nsIHttpChannelVtbl nsChannelVtbl = {
1246     nsChannel_QueryInterface,
1247     nsChannel_AddRef,
1248     nsChannel_Release,
1249     nsChannel_GetName,
1250     nsChannel_IsPending,
1251     nsChannel_GetStatus,
1252     nsChannel_Cancel,
1253     nsChannel_Suspend,
1254     nsChannel_Resume,
1255     nsChannel_GetLoadGroup,
1256     nsChannel_SetLoadGroup,
1257     nsChannel_GetLoadFlags,
1258     nsChannel_SetLoadFlags,
1259     nsChannel_GetOriginalURI,
1260     nsChannel_SetOriginalURI,
1261     nsChannel_GetURI,
1262     nsChannel_GetOwner,
1263     nsChannel_SetOwner,
1264     nsChannel_GetNotificationCallbacks,
1265     nsChannel_SetNotificationCallbacks,
1266     nsChannel_GetSecurityInfo,
1267     nsChannel_GetContentType,
1268     nsChannel_SetContentType,
1269     nsChannel_GetContentCharset,
1270     nsChannel_SetContentCharset,
1271     nsChannel_GetContentLength,
1272     nsChannel_SetContentLength,
1273     nsChannel_Open,
1274     nsChannel_AsyncOpen,
1275     nsChannel_GetRequestMethod,
1276     nsChannel_SetRequestMethod,
1277     nsChannel_GetReferrer,
1278     nsChannel_SetReferrer,
1279     nsChannel_GetRequestHeader,
1280     nsChannel_SetRequestHeader,
1281     nsChannel_VisitRequestHeaders,
1282     nsChannel_GetAllowPipelining,
1283     nsChannel_SetAllowPipelining,
1284     nsChannel_GetRedirectionLimit,
1285     nsChannel_SetRedirectionLimit,
1286     nsChannel_GetResponseStatus,
1287     nsChannel_GetResponseStatusText,
1288     nsChannel_GetRequestSucceeded,
1289     nsChannel_GetResponseHeader,
1290     nsChannel_SetResponseHeader,
1291     nsChannel_VisitResponseHeaders,
1292     nsChannel_IsNoStoreResponse,
1293     nsChannel_IsNoCacheResponse
1294 };
1295
1296 static inline nsChannel *impl_from_nsIUploadChannel(nsIUploadChannel *iface)
1297 {
1298     return CONTAINING_RECORD(iface, nsChannel, nsIUploadChannel_iface);
1299 }
1300
1301 static nsresult NSAPI nsUploadChannel_QueryInterface(nsIUploadChannel *iface, nsIIDRef riid,
1302         void **result)
1303 {
1304     nsChannel *This = impl_from_nsIUploadChannel(iface);
1305     return nsIChannel_QueryInterface(&This->nsIHttpChannel_iface, riid, result);
1306 }
1307
1308 static nsrefcnt NSAPI nsUploadChannel_AddRef(nsIUploadChannel *iface)
1309 {
1310     nsChannel *This = impl_from_nsIUploadChannel(iface);
1311     return nsIChannel_AddRef(&This->nsIHttpChannel_iface);
1312 }
1313
1314 static nsrefcnt NSAPI nsUploadChannel_Release(nsIUploadChannel *iface)
1315 {
1316     nsChannel *This = impl_from_nsIUploadChannel(iface);
1317     return nsIChannel_Release(&This->nsIHttpChannel_iface);
1318 }
1319
1320 static nsresult NSAPI nsUploadChannel_SetUploadStream(nsIUploadChannel *iface,
1321         nsIInputStream *aStream, const nsACString *aContentType, PRInt32 aContentLength)
1322 {
1323     nsChannel *This = impl_from_nsIUploadChannel(iface);
1324     const char *content_type;
1325
1326     static const WCHAR content_typeW[] =
1327         {'C','o','n','t','e','n','t','-','T','y','p','e',0};
1328
1329     TRACE("(%p)->(%p %s %d)\n", This, aStream, debugstr_nsacstr(aContentType), aContentLength);
1330
1331     This->post_data_contains_headers = TRUE;
1332
1333     if(aContentType) {
1334         nsACString_GetData(aContentType, &content_type);
1335         if(*content_type) {
1336             WCHAR *ct;
1337
1338             ct = heap_strdupAtoW(content_type);
1339             if(!ct)
1340                 return NS_ERROR_UNEXPECTED;
1341
1342             set_http_header(&This->request_headers, content_typeW,
1343                     sizeof(content_typeW)/sizeof(WCHAR), ct, strlenW(ct));
1344             heap_free(ct);
1345             This->post_data_contains_headers = FALSE;
1346         }
1347     }
1348
1349     if(This->post_data_stream)
1350         nsIInputStream_Release(This->post_data_stream);
1351
1352     if(aContentLength != -1)
1353         FIXME("Unsupported acontentLength = %d\n", aContentLength);
1354
1355     if(This->post_data_stream)
1356         nsIInputStream_Release(This->post_data_stream);
1357     This->post_data_stream = aStream;
1358     if(aStream)
1359         nsIInputStream_AddRef(aStream);
1360
1361     This->request_method = METHOD_POST;
1362     return NS_OK;
1363 }
1364
1365 static nsresult NSAPI nsUploadChannel_GetUploadStream(nsIUploadChannel *iface,
1366         nsIInputStream **aUploadStream)
1367 {
1368     nsChannel *This = impl_from_nsIUploadChannel(iface);
1369
1370     TRACE("(%p)->(%p)\n", This, aUploadStream);
1371
1372     if(This->post_data_stream)
1373         nsIInputStream_AddRef(This->post_data_stream);
1374
1375     *aUploadStream = This->post_data_stream;
1376     return NS_OK;
1377 }
1378
1379 static const nsIUploadChannelVtbl nsUploadChannelVtbl = {
1380     nsUploadChannel_QueryInterface,
1381     nsUploadChannel_AddRef,
1382     nsUploadChannel_Release,
1383     nsUploadChannel_SetUploadStream,
1384     nsUploadChannel_GetUploadStream
1385 };
1386
1387 static inline nsChannel *impl_from_nsIHttpChannelInternal(nsIHttpChannelInternal *iface)
1388 {
1389     return CONTAINING_RECORD(iface, nsChannel, nsIHttpChannelInternal_iface);
1390 }
1391
1392 static nsresult NSAPI nsHttpChannelInternal_QueryInterface(nsIHttpChannelInternal *iface, nsIIDRef riid,
1393         void **result)
1394 {
1395     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1396     return nsIChannel_QueryInterface(&This->nsIHttpChannel_iface, riid, result);
1397 }
1398
1399 static nsrefcnt NSAPI nsHttpChannelInternal_AddRef(nsIHttpChannelInternal *iface)
1400 {
1401     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1402     return nsIChannel_AddRef(&This->nsIHttpChannel_iface);
1403 }
1404
1405 static nsrefcnt NSAPI nsHttpChannelInternal_Release(nsIHttpChannelInternal *iface)
1406 {
1407     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1408     return nsIChannel_Release(&This->nsIHttpChannel_iface);
1409 }
1410
1411 static nsresult NSAPI nsHttpChannelInternal_GetDocumentURI(nsIHttpChannelInternal *iface, nsIURI **aDocumentURI)
1412 {
1413     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1414
1415     FIXME("(%p)->()\n", This);
1416
1417     return NS_ERROR_NOT_IMPLEMENTED;
1418 }
1419
1420 static nsresult NSAPI nsHttpChannelInternal_SetDocumentURI(nsIHttpChannelInternal *iface, nsIURI *aDocumentURI)
1421 {
1422     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1423
1424     FIXME("(%p)->()\n", This);
1425
1426     return NS_ERROR_NOT_IMPLEMENTED;
1427 }
1428
1429 static nsresult NSAPI nsHttpChannelInternal_GetRequestVersion(nsIHttpChannelInternal *iface, PRUint32 *major, PRUint32 *minor)
1430 {
1431     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1432
1433     FIXME("(%p)->()\n", This);
1434
1435     return NS_ERROR_NOT_IMPLEMENTED;
1436 }
1437
1438 static nsresult NSAPI nsHttpChannelInternal_GetResponseVersion(nsIHttpChannelInternal *iface, PRUint32 *major, PRUint32 *minor)
1439 {
1440     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1441
1442     FIXME("(%p)->()\n", This);
1443
1444     return NS_ERROR_NOT_IMPLEMENTED;
1445 }
1446
1447 static nsresult NSAPI nsHttpChannelInternal_SetCookie(nsIHttpChannelInternal *iface, const char *aCookieHeader)
1448 {
1449     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1450
1451     FIXME("(%p)->()\n", This);
1452
1453     return NS_ERROR_NOT_IMPLEMENTED;
1454 }
1455
1456 static nsresult NSAPI nsHttpChannelInternal_SetupFallbackChannel(nsIHttpChannelInternal *iface, const char *aFallbackKey)
1457 {
1458     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1459
1460     FIXME("(%p)->()\n", This);
1461
1462     return NS_ERROR_NOT_IMPLEMENTED;
1463 }
1464
1465 static nsresult NSAPI nsHttpChannelInternal_GetForceAllowThirdPartyCookie(nsIHttpChannelInternal *iface, PRBool *aForceThirdPartyCookie)
1466 {
1467     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1468
1469     FIXME("(%p)->()\n", This);
1470
1471     return NS_ERROR_NOT_IMPLEMENTED;
1472 }
1473
1474 static nsresult NSAPI nsHttpChannelInternal_SetForceAllowThirdPartyCookie(nsIHttpChannelInternal *iface, PRBool aForceThirdPartyCookie)
1475 {
1476     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1477
1478     FIXME("(%p)->()\n", This);
1479
1480     return NS_ERROR_NOT_IMPLEMENTED;
1481 }
1482
1483 static nsresult NSAPI nsHttpChannelInternal_GetCanceled(nsIHttpChannelInternal *iface, PRBool *aCanceled)
1484 {
1485     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1486
1487     FIXME("(%p)->(%p)\n", This, aCanceled);
1488
1489     return NS_ERROR_NOT_IMPLEMENTED;
1490 }
1491
1492 static nsresult NSAPI nsHttpChannelInternal_GetChannelIsForDownload(nsIHttpChannelInternal *iface, PRBool *aCanceled)
1493 {
1494     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1495
1496     FIXME("(%p)->(%p)\n", This, aCanceled);
1497
1498     return NS_ERROR_NOT_IMPLEMENTED;
1499 }
1500
1501 static nsresult NSAPI nsHttpChannelInternal_SetChannelIsForDownload(nsIHttpChannelInternal *iface, PRBool aCanceled)
1502 {
1503     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1504
1505     FIXME("(%p)->(%x)\n", This, aCanceled);
1506
1507     return NS_ERROR_NOT_IMPLEMENTED;
1508 }
1509
1510 static nsresult NSAPI nsHttpChannelInternal_GetLocalAddress(nsIHttpChannelInternal *iface, nsACString *aLocalAddress)
1511 {
1512     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1513
1514     FIXME("(%p)->(%p)\n", This, aLocalAddress);
1515
1516     return NS_ERROR_NOT_IMPLEMENTED;
1517 }
1518
1519 static nsresult NSAPI nsHttpChannelInternal_GetLocalPort(nsIHttpChannelInternal *iface, PRInt32 *aLocalPort)
1520 {
1521     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1522
1523     FIXME("(%p)->(%p)\n", This, aLocalPort);
1524
1525     return NS_ERROR_NOT_IMPLEMENTED;
1526 }
1527
1528 static nsresult NSAPI nsHttpChannelInternal_GetRemoteAddress(nsIHttpChannelInternal *iface, nsACString *aRemoteAddress)
1529 {
1530     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1531
1532     FIXME("(%p)->(%p)\n", This, aRemoteAddress);
1533
1534     return NS_ERROR_NOT_IMPLEMENTED;
1535 }
1536
1537 static nsresult NSAPI nsHttpChannelInternal_GetRemotePort(nsIHttpChannelInternal *iface, PRInt32 *aRemotePort)
1538 {
1539     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1540
1541     FIXME("(%p)->(%p)\n", This, aRemotePort);
1542
1543     return NS_ERROR_NOT_IMPLEMENTED;
1544 }
1545
1546 static nsresult NSAPI nsHttpChannelInternal_SetCacheKeysRedirectChain(nsIHttpChannelInternal *iface, void *cacheKeys)
1547 {
1548     nsChannel *This = impl_from_nsIHttpChannelInternal(iface);
1549
1550     FIXME("(%p)->(%p)\n", This, cacheKeys);
1551
1552     return NS_ERROR_NOT_IMPLEMENTED;
1553 }
1554
1555 static const nsIHttpChannelInternalVtbl nsHttpChannelInternalVtbl = {
1556     nsHttpChannelInternal_QueryInterface,
1557     nsHttpChannelInternal_AddRef,
1558     nsHttpChannelInternal_Release,
1559     nsHttpChannelInternal_GetDocumentURI,
1560     nsHttpChannelInternal_SetDocumentURI,
1561     nsHttpChannelInternal_GetRequestVersion,
1562     nsHttpChannelInternal_GetResponseVersion,
1563     nsHttpChannelInternal_SetCookie,
1564     nsHttpChannelInternal_SetupFallbackChannel,
1565     nsHttpChannelInternal_GetForceAllowThirdPartyCookie,
1566     nsHttpChannelInternal_SetForceAllowThirdPartyCookie,
1567     nsHttpChannelInternal_GetCanceled,
1568     nsHttpChannelInternal_GetChannelIsForDownload,
1569     nsHttpChannelInternal_SetChannelIsForDownload,
1570     nsHttpChannelInternal_GetLocalAddress,
1571     nsHttpChannelInternal_GetLocalPort,
1572     nsHttpChannelInternal_GetRemoteAddress,
1573     nsHttpChannelInternal_GetRemotePort,
1574     nsHttpChannelInternal_SetCacheKeysRedirectChain
1575 };
1576
1577
1578 static void invalidate_uri(nsWineURI *This)
1579 {
1580     if(This->uri) {
1581         IUri_Release(This->uri);
1582         This->uri = NULL;
1583     }
1584 }
1585
1586 static BOOL ensure_uri_builder(nsWineURI *This)
1587 {
1588     if(!This->uri_builder) {
1589         HRESULT hres;
1590
1591         if(!ensure_uri(This))
1592             return FALSE;
1593
1594         hres = CreateIUriBuilder(This->uri, 0, 0, &This->uri_builder);
1595         if(FAILED(hres)) {
1596             WARN("CreateIUriBuilder failed: %08x\n", hres);
1597             return FALSE;
1598         }
1599     }
1600
1601     invalidate_uri(This);
1602     return TRUE;
1603 }
1604
1605 static nsresult get_uri_string(nsWineURI *This, Uri_PROPERTY prop, nsACString *ret)
1606 {
1607     char *vala;
1608     BSTR val;
1609     HRESULT hres;
1610
1611     if(!ensure_uri(This))
1612         return NS_ERROR_UNEXPECTED;
1613
1614     hres = IUri_GetPropertyBSTR(This->uri, prop, &val, 0);
1615     if(FAILED(hres)) {
1616         WARN("GetPropertyBSTR failed: %08x\n", hres);
1617         return NS_ERROR_UNEXPECTED;
1618     }
1619
1620     vala = heap_strdupWtoA(val);
1621     SysFreeString(val);
1622     if(!vala)
1623         return NS_ERROR_OUT_OF_MEMORY;
1624
1625     TRACE("ret %s\n", debugstr_a(vala));
1626     nsACString_SetData(ret, vala);
1627     heap_free(vala);
1628     return NS_OK;
1629 }
1630
1631 static inline nsWineURI *impl_from_nsIURL(nsIURL *iface)
1632 {
1633     return CONTAINING_RECORD(iface, nsWineURI, nsIURL_iface);
1634 }
1635
1636 static nsresult NSAPI nsURI_QueryInterface(nsIURL *iface, nsIIDRef riid, void **result)
1637 {
1638     nsWineURI *This = impl_from_nsIURL(iface);
1639
1640     *result = NULL;
1641
1642     if(IsEqualGUID(&IID_nsISupports, riid)) {
1643         TRACE("(%p)->(IID_nsISupports %p)\n", This, result);
1644         *result = &This->nsIURL_iface;
1645     }else if(IsEqualGUID(&IID_nsIURI, riid)) {
1646         TRACE("(%p)->(IID_nsIURI %p)\n", This, result);
1647         *result = &This->nsIURL_iface;
1648     }else if(IsEqualGUID(&IID_nsIURL, riid)) {
1649         TRACE("(%p)->(IID_nsIURL %p)\n", This, result);
1650         *result = &This->nsIURL_iface;
1651     }else if(IsEqualGUID(&IID_nsWineURI, riid)) {
1652         TRACE("(%p)->(IID_nsWineURI %p)\n", This, result);
1653         *result = This;
1654     }
1655
1656     if(*result) {
1657         nsIURI_AddRef(&This->nsIURL_iface);
1658         return NS_OK;
1659     }
1660
1661     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), result);
1662     return This->nsuri ? nsIURI_QueryInterface(This->nsuri, riid, result) : NS_NOINTERFACE;
1663 }
1664
1665 static nsrefcnt NSAPI nsURI_AddRef(nsIURL *iface)
1666 {
1667     nsWineURI *This = impl_from_nsIURL(iface);
1668     LONG ref = InterlockedIncrement(&This->ref);
1669
1670     TRACE("(%p) ref=%d\n", This, ref);
1671
1672     return ref;
1673 }
1674
1675 static nsrefcnt NSAPI nsURI_Release(nsIURL *iface)
1676 {
1677     nsWineURI *This = impl_from_nsIURL(iface);
1678     LONG ref = InterlockedDecrement(&This->ref);
1679
1680     TRACE("(%p) ref=%d\n", This, ref);
1681
1682     if(!ref) {
1683         if(This->window_ref)
1684             windowref_release(This->window_ref);
1685         if(This->container)
1686             nsIWebBrowserChrome_Release(&This->container->nsIWebBrowserChrome_iface);
1687         if(This->nsurl)
1688             nsIURL_Release(This->nsurl);
1689         if(This->nsuri)
1690             nsIURI_Release(This->nsuri);
1691         if(This->uri)
1692             IUri_Release(This->uri);
1693         heap_free(This);
1694     }
1695
1696     return ref;
1697 }
1698
1699 static nsresult NSAPI nsURI_GetSpec(nsIURL *iface, nsACString *aSpec)
1700 {
1701     nsWineURI *This = impl_from_nsIURL(iface);
1702
1703     TRACE("(%p)->(%p)\n", This, aSpec);
1704
1705     return get_uri_string(This, Uri_PROPERTY_DISPLAY_URI, aSpec);
1706 }
1707
1708 static nsresult NSAPI nsURI_SetSpec(nsIURL *iface, const nsACString *aSpec)
1709 {
1710     nsWineURI *This = impl_from_nsIURL(iface);
1711     const char *speca;
1712     WCHAR *spec;
1713     IUri *uri;
1714     HRESULT hres;
1715
1716     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aSpec));
1717
1718     nsACString_GetData(aSpec, &speca);
1719     spec = heap_strdupAtoW(speca);
1720     if(!spec)
1721         return NS_ERROR_OUT_OF_MEMORY;
1722
1723     hres = CreateUri(spec, 0, 0, &uri);
1724     heap_free(spec);
1725     if(FAILED(hres)) {
1726         WARN("CreateUri failed: %08x\n", hres);
1727         return NS_ERROR_FAILURE;
1728     }
1729
1730     invalidate_uri(This);
1731     if(This->uri_builder) {
1732         IUriBuilder_Release(This->uri_builder);
1733         This->uri_builder = NULL;
1734     }
1735
1736     This->uri = uri;
1737     return NS_OK;
1738 }
1739
1740 static nsresult NSAPI nsURI_GetPrePath(nsIURL *iface, nsACString *aPrePath)
1741 {
1742     nsWineURI *This = impl_from_nsIURL(iface);
1743
1744     TRACE("(%p)->(%p)\n", This, aPrePath);
1745
1746     if(This->nsuri)
1747         return nsIURI_GetPrePath(This->nsuri, aPrePath);
1748
1749     FIXME("default action not implemented\n");
1750     return NS_ERROR_NOT_IMPLEMENTED;
1751 }
1752
1753 static nsresult NSAPI nsURI_GetScheme(nsIURL *iface, nsACString *aScheme)
1754 {
1755     nsWineURI *This = impl_from_nsIURL(iface);
1756     DWORD scheme;
1757     HRESULT hres;
1758
1759     TRACE("(%p)->(%p)\n", This, aScheme);
1760
1761     if(!ensure_uri(This))
1762         return NS_ERROR_UNEXPECTED;
1763
1764     hres = IUri_GetScheme(This->uri, &scheme);
1765     if(FAILED(hres)) {
1766         WARN("GetScheme failed: %08x\n", hres);
1767         return NS_ERROR_UNEXPECTED;
1768     }
1769
1770     if(scheme == URL_SCHEME_ABOUT) {
1771         nsACString_SetData(aScheme, "wine");
1772         return NS_OK;
1773     }
1774
1775     return get_uri_string(This, Uri_PROPERTY_SCHEME_NAME, aScheme);
1776 }
1777
1778 static nsresult NSAPI nsURI_SetScheme(nsIURL *iface, const nsACString *aScheme)
1779 {
1780     nsWineURI *This = impl_from_nsIURL(iface);
1781     const char *schemea;
1782     WCHAR *scheme;
1783     HRESULT hres;
1784
1785     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aScheme));
1786
1787     if(!ensure_uri_builder(This))
1788         return NS_ERROR_UNEXPECTED;
1789
1790     nsACString_GetData(aScheme, &schemea);
1791     scheme = heap_strdupAtoW(schemea);
1792     if(!scheme)
1793         return NS_ERROR_OUT_OF_MEMORY;
1794
1795     hres = IUriBuilder_SetSchemeName(This->uri_builder, scheme);
1796     heap_free(scheme);
1797     if(FAILED(hres))
1798         return NS_ERROR_UNEXPECTED;
1799
1800     return NS_OK;
1801 }
1802
1803 static nsresult NSAPI nsURI_GetUserPass(nsIURL *iface, nsACString *aUserPass)
1804 {
1805     nsWineURI *This = impl_from_nsIURL(iface);
1806     BSTR user, pass;
1807     HRESULT hres;
1808
1809     TRACE("(%p)->(%p)\n", This, aUserPass);
1810
1811     if(!ensure_uri(This))
1812         return NS_ERROR_UNEXPECTED;
1813
1814     hres = IUri_GetUserName(This->uri, &user);
1815     if(FAILED(hres))
1816         return NS_ERROR_FAILURE;
1817
1818     hres = IUri_GetPassword(This->uri, &pass);
1819     if(FAILED(hres)) {
1820         SysFreeString(user);
1821         return NS_ERROR_FAILURE;
1822     }
1823
1824     if(*user || *pass) {
1825         FIXME("Construct user:pass string\n");
1826     }else {
1827         nsACString_SetData(aUserPass, "");
1828     }
1829
1830     SysFreeString(user);
1831     SysFreeString(pass);
1832     return NS_OK;
1833 }
1834
1835 static nsresult NSAPI nsURI_SetUserPass(nsIURL *iface, const nsACString *aUserPass)
1836 {
1837     nsWineURI *This = impl_from_nsIURL(iface);
1838     WCHAR *user = NULL, *pass = NULL, *buf = NULL;
1839     const char *user_pass;
1840     HRESULT hres;
1841
1842     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aUserPass));
1843
1844     if(!ensure_uri_builder(This))
1845         return NS_ERROR_UNEXPECTED;
1846
1847     nsACString_GetData(aUserPass, &user_pass);
1848     if(*user_pass) {
1849         WCHAR *ptr;
1850
1851         buf = heap_strdupAtoW(user_pass);
1852         if(!buf)
1853             return NS_ERROR_OUT_OF_MEMORY;
1854
1855         ptr = strchrW(buf, ':');
1856         if(!ptr) {
1857             user = buf;
1858         }else if(ptr != buf) {
1859             *ptr++ = 0;
1860             user = buf;
1861             if(*ptr)
1862                 pass = ptr;
1863         }else {
1864             pass = buf+1;
1865         }
1866     }
1867
1868     hres = IUriBuilder_SetUserName(This->uri_builder, user);
1869     if(SUCCEEDED(hres))
1870         hres = IUriBuilder_SetPassword(This->uri_builder, pass);
1871
1872     heap_free(buf);
1873     return SUCCEEDED(hres) ? NS_OK : NS_ERROR_FAILURE;
1874 }
1875
1876 static nsresult NSAPI nsURI_GetUsername(nsIURL *iface, nsACString *aUsername)
1877 {
1878     nsWineURI *This = impl_from_nsIURL(iface);
1879
1880     TRACE("(%p)->(%p)\n", This, aUsername);
1881
1882     return get_uri_string(This, Uri_PROPERTY_USER_NAME, aUsername);
1883 }
1884
1885 static nsresult NSAPI nsURI_SetUsername(nsIURL *iface, const nsACString *aUsername)
1886 {
1887     nsWineURI *This = impl_from_nsIURL(iface);
1888     const char *usera;
1889     WCHAR *user;
1890     HRESULT hres;
1891
1892     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aUsername));
1893
1894     if(!ensure_uri_builder(This))
1895         return NS_ERROR_UNEXPECTED;
1896
1897     nsACString_GetData(aUsername, &usera);
1898     user = heap_strdupAtoW(usera);
1899     if(!user)
1900         return NS_ERROR_OUT_OF_MEMORY;
1901
1902     hres = IUriBuilder_SetUserName(This->uri_builder, user);
1903     heap_free(user);
1904     if(FAILED(hres))
1905         return NS_ERROR_UNEXPECTED;
1906
1907     return NS_OK;
1908 }
1909
1910 static nsresult NSAPI nsURI_GetPassword(nsIURL *iface, nsACString *aPassword)
1911 {
1912     nsWineURI *This = impl_from_nsIURL(iface);
1913
1914     TRACE("(%p)->(%p)\n", This, aPassword);
1915
1916     return get_uri_string(This, Uri_PROPERTY_PASSWORD, aPassword);
1917 }
1918
1919 static nsresult NSAPI nsURI_SetPassword(nsIURL *iface, const nsACString *aPassword)
1920 {
1921     nsWineURI *This = impl_from_nsIURL(iface);
1922     const char *passa;
1923     WCHAR *pass;
1924     HRESULT hres;
1925
1926     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aPassword));
1927
1928     if(!ensure_uri_builder(This))
1929         return NS_ERROR_UNEXPECTED;
1930
1931     nsACString_GetData(aPassword, &passa);
1932     pass = heap_strdupAtoW(passa);
1933     if(!pass)
1934         return NS_ERROR_OUT_OF_MEMORY;
1935
1936     hres = IUriBuilder_SetPassword(This->uri_builder, pass);
1937     heap_free(pass);
1938     if(FAILED(hres))
1939         return NS_ERROR_UNEXPECTED;
1940
1941     return NS_OK;
1942 }
1943
1944 static nsresult NSAPI nsURI_GetHostPort(nsIURL *iface, nsACString *aHostPort)
1945 {
1946     nsWineURI *This = impl_from_nsIURL(iface);
1947     const WCHAR *ptr;
1948     char *vala;
1949     BSTR val;
1950     HRESULT hres;
1951
1952     TRACE("(%p)->(%p)\n", This, aHostPort);
1953
1954     if(!ensure_uri(This))
1955         return NS_ERROR_UNEXPECTED;
1956
1957     hres = IUri_GetAuthority(This->uri, &val);
1958     if(FAILED(hres)) {
1959         WARN("GetAuthority failed: %08x\n", hres);
1960         return NS_ERROR_UNEXPECTED;
1961     }
1962
1963     ptr = strchrW(val, '@');
1964     if(!ptr)
1965         ptr = val;
1966
1967     vala = heap_strdupWtoA(ptr);
1968     SysFreeString(val);
1969     if(!vala)
1970         return NS_ERROR_OUT_OF_MEMORY;
1971
1972     TRACE("ret %s\n", debugstr_a(vala));
1973     nsACString_SetData(aHostPort, vala);
1974     heap_free(vala);
1975     return NS_OK;
1976 }
1977
1978 static nsresult NSAPI nsURI_SetHostPort(nsIURL *iface, const nsACString *aHostPort)
1979 {
1980     nsWineURI *This = impl_from_nsIURL(iface);
1981
1982     WARN("(%p)->(%s)\n", This, debugstr_nsacstr(aHostPort));
1983
1984     /* Not implemented by Gecko */
1985     return NS_ERROR_NOT_IMPLEMENTED;
1986 }
1987
1988 static nsresult NSAPI nsURI_GetHost(nsIURL *iface, nsACString *aHost)
1989 {
1990     nsWineURI *This = impl_from_nsIURL(iface);
1991
1992     TRACE("(%p)->(%p)\n", This, aHost);
1993
1994     return get_uri_string(This, Uri_PROPERTY_HOST, aHost);
1995 }
1996
1997 static nsresult NSAPI nsURI_SetHost(nsIURL *iface, const nsACString *aHost)
1998 {
1999     nsWineURI *This = impl_from_nsIURL(iface);
2000     const char *hosta;
2001     WCHAR *host;
2002     HRESULT hres;
2003
2004     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aHost));
2005
2006     if(!ensure_uri_builder(This))
2007         return NS_ERROR_UNEXPECTED;
2008
2009     nsACString_GetData(aHost, &hosta);
2010     host = heap_strdupAtoW(hosta);
2011     if(!host)
2012         return NS_ERROR_OUT_OF_MEMORY;
2013
2014     hres = IUriBuilder_SetHost(This->uri_builder, host);
2015     heap_free(host);
2016     if(FAILED(hres))
2017         return NS_ERROR_UNEXPECTED;
2018
2019     return NS_OK;
2020 }
2021
2022 static nsresult NSAPI nsURI_GetPort(nsIURL *iface, PRInt32 *aPort)
2023 {
2024     nsWineURI *This = impl_from_nsIURL(iface);
2025     DWORD port;
2026     HRESULT hres;
2027
2028     TRACE("(%p)->(%p)\n", This, aPort);
2029
2030     if(!ensure_uri(This))
2031         return NS_ERROR_UNEXPECTED;
2032
2033     hres = IUri_GetPort(This->uri, &port);
2034     if(FAILED(hres)) {
2035         WARN("GetPort failed: %08x\n", hres);
2036         return NS_ERROR_UNEXPECTED;
2037     }
2038
2039     *aPort = port ? port : -1;
2040     return NS_OK;
2041 }
2042
2043 static nsresult NSAPI nsURI_SetPort(nsIURL *iface, PRInt32 aPort)
2044 {
2045     nsWineURI *This = impl_from_nsIURL(iface);
2046     HRESULT hres;
2047
2048     TRACE("(%p)->(%d)\n", This, aPort);
2049
2050     if(!ensure_uri_builder(This))
2051         return NS_ERROR_UNEXPECTED;
2052
2053     hres = IUriBuilder_SetPort(This->uri_builder, aPort != -1, aPort);
2054     return SUCCEEDED(hres) ? NS_OK : NS_ERROR_FAILURE;
2055 }
2056
2057 static nsresult NSAPI nsURI_GetPath(nsIURL *iface, nsACString *aPath)
2058 {
2059     nsWineURI *This = impl_from_nsIURL(iface);
2060
2061     TRACE("(%p)->(%p)\n", This, aPath);
2062
2063     return get_uri_string(This, Uri_PROPERTY_PATH, aPath);
2064 }
2065
2066 static nsresult NSAPI nsURI_SetPath(nsIURL *iface, const nsACString *aPath)
2067 {
2068     nsWineURI *This = impl_from_nsIURL(iface);
2069     const char *patha;
2070     WCHAR *path;
2071     HRESULT hres;
2072
2073     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aPath));
2074
2075     if(!ensure_uri_builder(This))
2076         return NS_ERROR_UNEXPECTED;
2077
2078     nsACString_GetData(aPath, &patha);
2079     path = heap_strdupAtoW(patha);
2080     if(!path)
2081         return NS_ERROR_OUT_OF_MEMORY;
2082
2083     hres = IUriBuilder_SetPath(This->uri_builder, path);
2084     heap_free(path);
2085     if(FAILED(hres))
2086         return NS_ERROR_UNEXPECTED;
2087
2088     return NS_OK;
2089 }
2090
2091 static nsresult NSAPI nsURL_GetRef(nsIURL *iface, nsACString *aRef)
2092 {
2093     nsWineURI *This = impl_from_nsIURL(iface);
2094     char *refa = NULL;
2095     BSTR ref;
2096     HRESULT hres;
2097
2098     TRACE("(%p)->(%p)\n", This, aRef);
2099
2100     if(!ensure_uri(This))
2101         return NS_ERROR_UNEXPECTED;
2102
2103     hres = IUri_GetFragment(This->uri, &ref);
2104     if(FAILED(hres))
2105         return NS_ERROR_UNEXPECTED;
2106
2107     refa = heap_strdupWtoA(ref);
2108     SysFreeString(ref);
2109     if(ref && !refa)
2110         return NS_ERROR_OUT_OF_MEMORY;
2111
2112     nsACString_SetData(aRef, refa && *refa == '#' ? refa+1 : refa);
2113     heap_free(refa);
2114     return NS_OK;
2115 }
2116
2117 static nsresult NSAPI nsURL_SetRef(nsIURL *iface, const nsACString *aRef)
2118 {
2119     nsWineURI *This = impl_from_nsIURL(iface);
2120     const char *refa;
2121     WCHAR *ref;
2122     HRESULT hres;
2123
2124     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aRef));
2125
2126     if(!ensure_uri_builder(This))
2127         return NS_ERROR_UNEXPECTED;
2128
2129     nsACString_GetData(aRef, &refa);
2130     ref = heap_strdupAtoW(refa);
2131     if(!ref)
2132         return NS_ERROR_OUT_OF_MEMORY;
2133
2134     hres = IUriBuilder_SetFragment(This->uri_builder, ref);
2135     heap_free(ref);
2136     if(FAILED(hres))
2137         return NS_ERROR_UNEXPECTED;
2138
2139     return NS_OK;
2140 }
2141
2142 static nsresult NSAPI nsURI_Equals(nsIURL *iface, nsIURI *other, PRBool *_retval)
2143 {
2144     nsWineURI *This = impl_from_nsIURL(iface);
2145     nsWineURI *other_obj;
2146     nsresult nsres;
2147     HRESULT hres;
2148
2149     TRACE("(%p)->(%p %p)\n", This, other, _retval);
2150
2151     nsres = nsIURI_QueryInterface(other, &IID_nsWineURI, (void**)&other_obj);
2152     if(NS_FAILED(nsres)) {
2153         TRACE("Could not get nsWineURI interface\n");
2154         *_retval = FALSE;
2155         return NS_OK;
2156     }
2157
2158     if(ensure_uri(This) && ensure_uri(other_obj)) {
2159         hres = IUri_IsEqual(This->uri, other_obj->uri, _retval);
2160         nsres = SUCCEEDED(hres) ? NS_OK : NS_ERROR_FAILURE;
2161     }else {
2162         nsres = NS_ERROR_UNEXPECTED;
2163     }
2164
2165     nsIURI_Release(&other_obj->nsIURL_iface);
2166     return nsres;
2167 }
2168
2169 static nsresult NSAPI nsURI_EqualsExceptRef(nsIURL *iface, nsIURI *other, PRBool *_retval)
2170 {
2171     nsWineURI *This = impl_from_nsIURL(iface);
2172     FIXME("(%p)->(%p %p)\n", This, other, _retval);
2173     return nsIURL_Equals(&This->nsIURL_iface, other, _retval);
2174 }
2175
2176 static nsresult NSAPI nsURI_SchemeIs(nsIURL *iface, const char *scheme, PRBool *_retval)
2177 {
2178     nsWineURI *This = impl_from_nsIURL(iface);
2179     WCHAR buf[INTERNET_MAX_SCHEME_LENGTH];
2180     BSTR scheme_name;
2181     HRESULT hres;
2182
2183     TRACE("(%p)->(%s %p)\n", This, debugstr_a(scheme), _retval);
2184
2185     if(!ensure_uri(This))
2186         return NS_ERROR_UNEXPECTED;
2187
2188     hres = IUri_GetSchemeName(This->uri, &scheme_name);
2189     if(FAILED(hres))
2190         return NS_ERROR_UNEXPECTED;
2191
2192     MultiByteToWideChar(CP_ACP, 0, scheme, -1, buf, sizeof(buf)/sizeof(WCHAR));
2193     *_retval = !strcmpW(scheme_name, buf);
2194     SysFreeString(scheme_name);
2195     return NS_OK;
2196 }
2197
2198 static nsresult NSAPI nsURI_Clone(nsIURL *iface, nsIURI **_retval)
2199 {
2200     nsWineURI *This = impl_from_nsIURL(iface);
2201     nsIURI *nsuri = NULL;
2202     nsWineURI *wine_uri;
2203     nsresult nsres;
2204
2205     TRACE("(%p)->(%p)\n", This, _retval);
2206
2207     if(!ensure_uri(This))
2208         return NS_ERROR_UNEXPECTED;
2209
2210     if(This->nsuri) {
2211         nsres = nsIURI_Clone(This->nsuri, &nsuri);
2212         if(NS_FAILED(nsres)) {
2213             WARN("Clone failed: %08x\n", nsres);
2214             return nsres;
2215         }
2216     }
2217
2218     nsres = create_nsuri(This->uri, nsuri, This->window_ref ? This->window_ref->window : NULL, This->container, &wine_uri);
2219     if(NS_FAILED(nsres)) {
2220         WARN("create_nsuri failed: %08x\n", nsres);
2221         return nsres;
2222     }
2223
2224     *_retval = (nsIURI*)&wine_uri->nsIURL_iface;
2225     return NS_OK;
2226 }
2227
2228 static nsresult NSAPI nsURI_CloneIgnoreRef(nsIURL *iface, nsIURI **_retval)
2229 {
2230     nsWineURI *This = impl_from_nsIURL(iface);
2231     FIXME("(%p)->(%p)\n", This, _retval);
2232     return nsIURL_Clone(&This->nsIURL_iface, _retval);
2233 }
2234
2235 static nsresult NSAPI nsURI_Resolve(nsIURL *iface, const nsACString *aRelativePath,
2236         nsACString *_retval)
2237 {
2238     nsWineURI *This = impl_from_nsIURL(iface);
2239     const char *patha;
2240     IUri *new_uri;
2241     WCHAR *path;
2242     char *reta;
2243     BSTR ret;
2244     HRESULT hres;
2245
2246     TRACE("(%p)->(%s %p)\n", This, debugstr_nsacstr(aRelativePath), _retval);
2247
2248     if(!ensure_uri(This))
2249         return NS_ERROR_UNEXPECTED;
2250
2251     nsACString_GetData(aRelativePath, &patha);
2252     path = heap_strdupAtoW(patha);
2253     if(!path)
2254         return NS_ERROR_OUT_OF_MEMORY;
2255
2256     hres = CoInternetCombineUrlEx(This->uri, path, URL_ESCAPE_SPACES_ONLY|URL_DONT_ESCAPE_EXTRA_INFO, &new_uri, 0);
2257     heap_free(path);
2258     if(FAILED(hres)) {
2259         ERR("CoIntenetCombineUrlEx failed: %08x\n", hres);
2260         return NS_ERROR_FAILURE;
2261     }
2262
2263     hres = IUri_GetDisplayUri(new_uri, &ret);
2264     IUri_Release(new_uri);
2265     if(FAILED(hres))
2266         return NS_ERROR_FAILURE;
2267
2268     reta = heap_strdupWtoA(ret);
2269     SysFreeString(ret);
2270     if(!reta)
2271         return NS_ERROR_OUT_OF_MEMORY;
2272
2273     TRACE("returning %s\n", debugstr_a(reta));
2274     nsACString_SetData(_retval, reta);
2275     heap_free(reta);
2276     return NS_OK;
2277 }
2278
2279 static nsresult NSAPI nsURI_GetAsciiSpec(nsIURL *iface, nsACString *aAsciiSpec)
2280 {
2281     nsWineURI *This = impl_from_nsIURL(iface);
2282
2283     TRACE("(%p)->(%p)\n", This, aAsciiSpec);
2284
2285     return nsIURI_GetSpec(&This->nsIURL_iface, aAsciiSpec);
2286 }
2287
2288 static nsresult NSAPI nsURI_GetAsciiHost(nsIURL *iface, nsACString *aAsciiHost)
2289 {
2290     nsWineURI *This = impl_from_nsIURL(iface);
2291
2292     TRACE("(%p)->(%p)\n", This, aAsciiHost);
2293
2294     if(This->nsuri)
2295         return nsIURI_GetAsciiHost(This->nsuri, aAsciiHost);
2296
2297     WARN("Use Uri_PUNYCODE_IDN_HOST flag\n");
2298     return get_uri_string(This, Uri_PROPERTY_HOST, aAsciiHost);
2299 }
2300
2301 static nsresult NSAPI nsURI_GetOriginCharset(nsIURL *iface, nsACString *aOriginCharset)
2302 {
2303     nsWineURI *This = impl_from_nsIURL(iface);
2304
2305     TRACE("(%p)->(%p)\n", This, aOriginCharset);
2306
2307     if(This->nsuri)
2308         return nsIURI_GetOriginCharset(This->nsuri, aOriginCharset);
2309
2310     FIXME("default action not implemented\n");
2311     return NS_ERROR_NOT_IMPLEMENTED;
2312 }
2313
2314 static nsresult NSAPI nsURL_GetFilePath(nsIURL *iface, nsACString *aFilePath)
2315 {
2316     nsWineURI *This = impl_from_nsIURL(iface);
2317
2318     TRACE("(%p)->(%p)\n", This, aFilePath);
2319
2320     return nsIURL_GetPath(&This->nsIURL_iface, aFilePath);
2321 }
2322
2323 static nsresult NSAPI nsURL_SetFilePath(nsIURL *iface, const nsACString *aFilePath)
2324 {
2325     nsWineURI *This = impl_from_nsIURL(iface);
2326
2327     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aFilePath));
2328
2329     return nsIURL_SetPath(&This->nsIURL_iface, aFilePath);
2330 }
2331
2332 static nsresult NSAPI nsURL_GetParam(nsIURL *iface, nsACString *aParam)
2333 {
2334     nsWineURI *This = impl_from_nsIURL(iface);
2335
2336     TRACE("(%p)->(%p)\n", This, aParam);
2337
2338     if(This->nsurl)
2339         return nsIURL_GetParam(This->nsurl, aParam);
2340
2341     FIXME("default action not implemented\n");
2342     return NS_ERROR_NOT_IMPLEMENTED;
2343 }
2344
2345 static nsresult NSAPI nsURL_SetParam(nsIURL *iface, const nsACString *aParam)
2346 {
2347     nsWineURI *This = impl_from_nsIURL(iface);
2348
2349     WARN("(%p)->(%s)\n", This, debugstr_nsacstr(aParam));
2350
2351     /* Not implemented by Gecko */
2352     return NS_ERROR_NOT_IMPLEMENTED;
2353 }
2354
2355 static nsresult NSAPI nsURL_GetQuery(nsIURL *iface, nsACString *aQuery)
2356 {
2357     nsWineURI *This = impl_from_nsIURL(iface);
2358
2359     TRACE("(%p)->(%p)\n", This, aQuery);
2360
2361     return get_uri_string(This, Uri_PROPERTY_QUERY, aQuery);
2362 }
2363
2364 static nsresult NSAPI nsURL_SetQuery(nsIURL *iface, const nsACString *aQuery)
2365 {
2366     nsWineURI *This = impl_from_nsIURL(iface);
2367     const char *querya;
2368     WCHAR *query;
2369     HRESULT hres;
2370
2371     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aQuery));
2372
2373     if(!ensure_uri_builder(This))
2374         return NS_ERROR_UNEXPECTED;
2375
2376     nsACString_GetData(aQuery, &querya);
2377     query = heap_strdupAtoW(querya);
2378     if(!query)
2379         return NS_ERROR_OUT_OF_MEMORY;
2380
2381     hres = IUriBuilder_SetQuery(This->uri_builder, query);
2382     heap_free(query);
2383     if(FAILED(hres))
2384         return NS_ERROR_UNEXPECTED;
2385
2386     return NS_OK;
2387 }
2388
2389 static nsresult NSAPI nsURL_GetDirectory(nsIURL *iface, nsACString *aDirectory)
2390 {
2391     nsWineURI *This = impl_from_nsIURL(iface);
2392     char *dir = NULL;
2393     WCHAR *ptr;
2394     BSTR path;
2395     HRESULT hres;
2396
2397     TRACE("(%p)->(%p)\n", This, aDirectory);
2398
2399     if(!ensure_uri(This))
2400         return NS_ERROR_UNEXPECTED;
2401
2402     hres = IUri_GetPath(This->uri, &path);
2403     if(FAILED(hres))
2404         return NS_ERROR_FAILURE;
2405
2406     ptr = strrchrW(path, '/');
2407     if(ptr) {
2408         if(ptr[1])
2409             *ptr = 0;
2410         dir = heap_strdupWtoA(path);
2411         if(!dir) {
2412             SysFreeString(path);
2413             return NS_ERROR_OUT_OF_MEMORY;
2414         }
2415     }
2416
2417     SysFreeString(path);
2418     TRACE("ret %s\n", debugstr_a(dir));
2419     nsACString_SetData(aDirectory, dir ? dir : "");
2420     return NS_OK;
2421 }
2422
2423 static nsresult NSAPI nsURL_SetDirectory(nsIURL *iface, const nsACString *aDirectory)
2424 {
2425     nsWineURI *This = impl_from_nsIURL(iface);
2426
2427     WARN("(%p)->(%s)\n", This, debugstr_nsacstr(aDirectory));
2428
2429     /* Not implemented by Gecko */
2430     return NS_ERROR_NOT_IMPLEMENTED;
2431 }
2432
2433 static nsresult NSAPI nsURL_GetFileName(nsIURL *iface, nsACString *aFileName)
2434 {
2435     nsWineURI *This = impl_from_nsIURL(iface);
2436
2437     TRACE("(%p)->(%p)\n", This, aFileName);
2438
2439     if(This->nsurl)
2440         return nsIURL_GetFileName(This->nsurl, aFileName);
2441
2442     FIXME("default action not implemented\n");
2443     return NS_ERROR_NOT_IMPLEMENTED;
2444 }
2445
2446 static nsresult NSAPI nsURL_SetFileName(nsIURL *iface, const nsACString *aFileName)
2447 {
2448     nsWineURI *This = impl_from_nsIURL(iface);
2449
2450     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aFileName));
2451
2452     if(This->nsurl) {
2453         invalidate_uri(This);
2454         return nsIURL_SetFileName(This->nsurl, aFileName);
2455     }
2456
2457     FIXME("default action not implemented\n");
2458     return NS_ERROR_NOT_IMPLEMENTED;
2459 }
2460
2461 static nsresult NSAPI nsURL_GetFileBaseName(nsIURL *iface, nsACString *aFileBaseName)
2462 {
2463     nsWineURI *This = impl_from_nsIURL(iface);
2464
2465     TRACE("(%p)->(%p)\n", This, aFileBaseName);
2466
2467     if(This->nsurl)
2468         return nsIURL_GetFileBaseName(This->nsurl, aFileBaseName);
2469
2470     FIXME("default action not implemented\n");
2471     return NS_ERROR_NOT_IMPLEMENTED;
2472 }
2473
2474 static nsresult NSAPI nsURL_SetFileBaseName(nsIURL *iface, const nsACString *aFileBaseName)
2475 {
2476     nsWineURI *This = impl_from_nsIURL(iface);
2477
2478     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aFileBaseName));
2479
2480     if(This->nsurl) {
2481         invalidate_uri(This);
2482         return nsIURL_SetFileBaseName(This->nsurl, aFileBaseName);
2483     }
2484
2485     FIXME("default action not implemented\n");
2486     return NS_ERROR_NOT_IMPLEMENTED;
2487 }
2488
2489 static nsresult NSAPI nsURL_GetFileExtension(nsIURL *iface, nsACString *aFileExtension)
2490 {
2491     nsWineURI *This = impl_from_nsIURL(iface);
2492
2493     TRACE("(%p)->(%p)\n", This, aFileExtension);
2494
2495     if(This->nsurl)
2496         return nsIURL_GetFileExtension(This->nsurl, aFileExtension);
2497
2498     FIXME("default action not implemented\n");
2499     return NS_ERROR_NOT_IMPLEMENTED;
2500 }
2501
2502 static nsresult NSAPI nsURL_SetFileExtension(nsIURL *iface, const nsACString *aFileExtension)
2503 {
2504     nsWineURI *This = impl_from_nsIURL(iface);
2505
2506     TRACE("(%p)->(%s)\n", This, debugstr_nsacstr(aFileExtension));
2507
2508     if(This->nsurl) {
2509         invalidate_uri(This);
2510         return nsIURL_SetFileExtension(This->nsurl, aFileExtension);
2511     }
2512
2513     FIXME("default action not implemented\n");
2514     return NS_ERROR_NOT_IMPLEMENTED;
2515 }
2516
2517 static nsresult NSAPI nsURL_GetCommonBaseSpec(nsIURL *iface, nsIURI *aURIToCompare, nsACString *_retval)
2518 {
2519     nsWineURI *This = impl_from_nsIURL(iface);
2520
2521     TRACE("(%p)->(%p %p)\n", This, aURIToCompare, _retval);
2522
2523     if(This->nsurl)
2524         return nsIURL_GetCommonBaseSpec(This->nsurl, aURIToCompare, _retval);
2525
2526     FIXME("default action not implemented\n");
2527     return NS_ERROR_NOT_IMPLEMENTED;
2528 }
2529
2530 static nsresult NSAPI nsURL_GetRelativeSpec(nsIURL *iface, nsIURI *aURIToCompare, nsACString *_retval)
2531 {
2532     nsWineURI *This = impl_from_nsIURL(iface);
2533
2534     TRACE("(%p)->(%p %p)\n", This, aURIToCompare, _retval);
2535
2536     if(This->nsurl)
2537         return nsIURL_GetRelativeSpec(This->nsurl, aURIToCompare, _retval);
2538
2539     FIXME("default action not implemented\n");
2540     return NS_ERROR_NOT_IMPLEMENTED;
2541 }
2542
2543 static const nsIURLVtbl nsURLVtbl = {
2544     nsURI_QueryInterface,
2545     nsURI_AddRef,
2546     nsURI_Release,
2547     nsURI_GetSpec,
2548     nsURI_SetSpec,
2549     nsURI_GetPrePath,
2550     nsURI_GetScheme,
2551     nsURI_SetScheme,
2552     nsURI_GetUserPass,
2553     nsURI_SetUserPass,
2554     nsURI_GetUsername,
2555     nsURI_SetUsername,
2556     nsURI_GetPassword,
2557     nsURI_SetPassword,
2558     nsURI_GetHostPort,
2559     nsURI_SetHostPort,
2560     nsURI_GetHost,
2561     nsURI_SetHost,
2562     nsURI_GetPort,
2563     nsURI_SetPort,
2564     nsURI_GetPath,
2565     nsURI_SetPath,
2566     nsURL_GetRef,
2567     nsURL_SetRef,
2568     nsURI_Equals,
2569     nsURI_EqualsExceptRef,
2570     nsURI_SchemeIs,
2571     nsURI_Clone,
2572     nsURI_CloneIgnoreRef,
2573     nsURI_Resolve,
2574     nsURI_GetAsciiSpec,
2575     nsURI_GetAsciiHost,
2576     nsURI_GetOriginCharset,
2577     nsURL_GetFilePath,
2578     nsURL_SetFilePath,
2579     nsURL_GetParam,
2580     nsURL_SetParam,
2581     nsURL_GetQuery,
2582     nsURL_SetQuery,
2583     nsURL_GetDirectory,
2584     nsURL_SetDirectory,
2585     nsURL_GetFileName,
2586     nsURL_SetFileName,
2587     nsURL_GetFileBaseName,
2588     nsURL_SetFileBaseName,
2589     nsURL_GetFileExtension,
2590     nsURL_SetFileExtension,
2591     nsURL_GetCommonBaseSpec,
2592     nsURL_GetRelativeSpec
2593 };
2594
2595 static nsresult create_nsuri(IUri *iuri, nsIURI *nsuri, HTMLWindow *window, NSContainer *container, nsWineURI **_retval)
2596 {
2597     nsWineURI *ret = heap_alloc_zero(sizeof(nsWineURI));
2598
2599     ret->nsIURL_iface.lpVtbl = &nsURLVtbl;
2600     ret->ref = 1;
2601     ret->nsuri = nsuri;
2602
2603     set_uri_nscontainer(ret, container);
2604     set_uri_window(ret, window);
2605
2606     IUri_AddRef(iuri);
2607     ret->uri = iuri;
2608
2609     if(nsuri)
2610         nsIURI_QueryInterface(nsuri, &IID_nsIURL, (void**)&ret->nsurl);
2611
2612     TRACE("retval=%p\n", ret);
2613     *_retval = ret;
2614     return NS_OK;
2615 }
2616
2617 HRESULT create_doc_uri(HTMLWindow *window, WCHAR *url, nsWineURI **ret)
2618 {
2619     nsWineURI *uri;
2620     IUri *iuri;
2621     nsresult nsres;
2622     HRESULT hres;
2623
2624     hres = CreateUri(url, 0, 0, &iuri);
2625     if(FAILED(hres))
2626         return hres;
2627
2628     nsres = create_nsuri(iuri, NULL, window, window->doc_obj->nscontainer, &uri);
2629     IUri_Release(iuri);
2630     if(NS_FAILED(nsres))
2631         return E_FAIL;
2632
2633     uri->is_doc_uri = TRUE;
2634
2635     *ret = uri;
2636     return S_OK;
2637 }
2638
2639 static nsresult create_nschannel(nsWineURI *uri, nsChannel **ret)
2640 {
2641     nsChannel *channel;
2642     HRESULT hres;
2643
2644     if(!ensure_uri(uri))
2645         return NS_ERROR_UNEXPECTED;
2646
2647     channel = heap_alloc_zero(sizeof(nsChannel));
2648     if(!channel)
2649         return NS_ERROR_OUT_OF_MEMORY;
2650
2651     channel->nsIHttpChannel_iface.lpVtbl = &nsChannelVtbl;
2652     channel->nsIUploadChannel_iface.lpVtbl = &nsUploadChannelVtbl;
2653     channel->nsIHttpChannelInternal_iface.lpVtbl = &nsHttpChannelInternalVtbl;
2654     channel->ref = 1;
2655     channel->request_method = METHOD_GET;
2656     list_init(&channel->response_headers);
2657     list_init(&channel->request_headers);
2658
2659     nsIURL_AddRef(&uri->nsIURL_iface);
2660     channel->uri = uri;
2661
2662     hres = IUri_GetScheme(uri->uri, &channel->url_scheme);
2663     if(FAILED(hres))
2664         channel->url_scheme = URL_SCHEME_UNKNOWN;
2665
2666     *ret = channel;
2667     return NS_OK;
2668 }
2669
2670 HRESULT create_redirect_nschannel(const WCHAR *url, nsChannel *orig_channel, nsChannel **ret)
2671 {
2672     HTMLWindow *window = NULL;
2673     nsChannel *channel;
2674     nsWineURI *uri;
2675     IUri *iuri;
2676     nsresult nsres;
2677     HRESULT hres;
2678
2679     hres = CreateUri(url, 0, 0, &iuri);
2680     if(FAILED(hres))
2681         return hres;
2682
2683     if(orig_channel->uri->window_ref)
2684         window = orig_channel->uri->window_ref->window;
2685     nsres = create_nsuri(iuri, NULL, window, NULL, &uri);
2686     IUri_Release(iuri);
2687     if(NS_FAILED(nsres))
2688         return E_FAIL;
2689
2690     nsres = create_nschannel(uri, &channel);
2691     nsIURL_Release(&uri->nsIURL_iface);
2692     if(NS_FAILED(nsres))
2693         return E_FAIL;
2694
2695     if(orig_channel->load_group) {
2696         nsILoadGroup_AddRef(orig_channel->load_group);
2697         channel->load_group = orig_channel->load_group;
2698     }
2699
2700     if(orig_channel->notif_callback) {
2701         nsIInterfaceRequestor_AddRef(orig_channel->notif_callback);
2702         channel->notif_callback = orig_channel->notif_callback;
2703     }
2704
2705     channel->load_flags = orig_channel->load_flags | LOAD_REPLACE;
2706
2707     if(orig_channel->request_method == METHOD_POST)
2708         FIXME("unsupported POST method\n");
2709
2710     if(orig_channel->original_uri) {
2711         nsIURI_AddRef(orig_channel->original_uri);
2712         channel->original_uri = orig_channel->original_uri;
2713     }
2714
2715     if(orig_channel->referrer) {
2716         nsIURI_AddRef(orig_channel->referrer);
2717         channel->referrer = orig_channel->referrer;
2718     }
2719
2720     *ret = channel;
2721     return S_OK;
2722 }
2723
2724 typedef struct {
2725     nsIProtocolHandler nsIProtocolHandler_iface;
2726
2727     LONG ref;
2728
2729     nsIProtocolHandler *nshandler;
2730 } nsProtocolHandler;
2731
2732 static inline nsProtocolHandler *impl_from_nsIProtocolHandler(nsIProtocolHandler *iface)
2733 {
2734     return CONTAINING_RECORD(iface, nsProtocolHandler, nsIProtocolHandler_iface);
2735 }
2736
2737 static nsresult NSAPI nsProtocolHandler_QueryInterface(nsIProtocolHandler *iface, nsIIDRef riid,
2738         void **result)
2739 {
2740     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2741
2742     *result = NULL;
2743
2744     if(IsEqualGUID(&IID_nsISupports, riid)) {
2745         TRACE("(%p)->(IID_nsISupports %p)\n", This, result);
2746         *result = &This->nsIProtocolHandler_iface;
2747     }else if(IsEqualGUID(&IID_nsIProtocolHandler, riid)) {
2748         TRACE("(%p)->(IID_nsIProtocolHandler %p)\n", This, result);
2749         *result = &This->nsIProtocolHandler_iface;
2750     }else if(IsEqualGUID(&IID_nsIExternalProtocolHandler, riid)) {
2751         TRACE("(%p)->(IID_nsIExternalProtocolHandler %p), returning NULL\n", This, result);
2752         return NS_NOINTERFACE;
2753     }
2754
2755     if(*result) {
2756         nsISupports_AddRef((nsISupports*)*result);
2757         return NS_OK;
2758     }
2759
2760     WARN("(%s %p)\n", debugstr_guid(riid), result);
2761     return NS_NOINTERFACE;
2762 }
2763
2764 static nsrefcnt NSAPI nsProtocolHandler_AddRef(nsIProtocolHandler *iface)
2765 {
2766     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2767     LONG ref = InterlockedIncrement(&This->ref);
2768
2769     TRACE("(%p) ref=%d\n", This, ref);
2770
2771     return ref;
2772 }
2773
2774 static nsrefcnt NSAPI nsProtocolHandler_Release(nsIProtocolHandler *iface)
2775 {
2776     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2777     LONG ref = InterlockedDecrement(&This->ref);
2778
2779     TRACE("(%p) ref=%d\n", This, ref);
2780
2781     if(!ref) {
2782         if(This->nshandler)
2783             nsIProtocolHandler_Release(This->nshandler);
2784         heap_free(This);
2785     }
2786
2787     return ref;
2788 }
2789
2790 static nsresult NSAPI nsProtocolHandler_GetScheme(nsIProtocolHandler *iface, nsACString *aScheme)
2791 {
2792     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2793
2794     TRACE("(%p)->(%p)\n", This, aScheme);
2795
2796     if(This->nshandler)
2797         return nsIProtocolHandler_GetScheme(This->nshandler, aScheme);
2798     return NS_ERROR_NOT_IMPLEMENTED;
2799 }
2800
2801 static nsresult NSAPI nsProtocolHandler_GetDefaultPort(nsIProtocolHandler *iface,
2802         PRInt32 *aDefaultPort)
2803 {
2804     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2805
2806     TRACE("(%p)->(%p)\n", This, aDefaultPort);
2807
2808     if(This->nshandler)
2809         return nsIProtocolHandler_GetDefaultPort(This->nshandler, aDefaultPort);
2810     return NS_ERROR_NOT_IMPLEMENTED;
2811 }
2812
2813 static nsresult NSAPI nsProtocolHandler_GetProtocolFlags(nsIProtocolHandler *iface,
2814                                                          PRUint32 *aProtocolFlags)
2815 {
2816     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2817
2818     TRACE("(%p)->(%p)\n", This, aProtocolFlags);
2819
2820     if(This->nshandler)
2821         return nsIProtocolHandler_GetProtocolFlags(This->nshandler, aProtocolFlags);
2822     return NS_ERROR_NOT_IMPLEMENTED;
2823 }
2824
2825 static nsresult NSAPI nsProtocolHandler_NewURI(nsIProtocolHandler *iface,
2826         const nsACString *aSpec, const char *aOriginCharset, nsIURI *aBaseURI, nsIURI **_retval)
2827 {
2828     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2829
2830     TRACE("((%p)->%s %s %p %p)\n", This, debugstr_nsacstr(aSpec), debugstr_a(aOriginCharset),
2831           aBaseURI, _retval);
2832
2833     if(This->nshandler)
2834         return nsIProtocolHandler_NewURI(This->nshandler, aSpec, aOriginCharset, aBaseURI, _retval);
2835     return NS_ERROR_NOT_IMPLEMENTED;
2836 }
2837
2838 static nsresult NSAPI nsProtocolHandler_NewChannel(nsIProtocolHandler *iface,
2839         nsIURI *aURI, nsIChannel **_retval)
2840 {
2841     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2842
2843     TRACE("(%p)->(%p %p)\n", This, aURI, _retval);
2844
2845     if(This->nshandler)
2846         return nsIProtocolHandler_NewChannel(This->nshandler, aURI, _retval);
2847     return NS_ERROR_NOT_IMPLEMENTED;
2848 }
2849
2850 static nsresult NSAPI nsProtocolHandler_AllowPort(nsIProtocolHandler *iface,
2851         PRInt32 port, const char *scheme, PRBool *_retval)
2852 {
2853     nsProtocolHandler *This = impl_from_nsIProtocolHandler(iface);
2854
2855     TRACE("(%p)->(%d %s %p)\n", This, port, debugstr_a(scheme), _retval);
2856
2857     if(This->nshandler)
2858         return nsIProtocolHandler_AllowPort(This->nshandler, port, scheme, _retval);
2859     return NS_ERROR_NOT_IMPLEMENTED;
2860 }
2861
2862 static const nsIProtocolHandlerVtbl nsProtocolHandlerVtbl = {
2863     nsProtocolHandler_QueryInterface,
2864     nsProtocolHandler_AddRef,
2865     nsProtocolHandler_Release,
2866     nsProtocolHandler_GetScheme,
2867     nsProtocolHandler_GetDefaultPort,
2868     nsProtocolHandler_GetProtocolFlags,
2869     nsProtocolHandler_NewURI,
2870     nsProtocolHandler_NewChannel,
2871     nsProtocolHandler_AllowPort
2872 };
2873
2874 static nsIProtocolHandler *create_protocol_handler(nsIProtocolHandler *nshandler)
2875 {
2876     nsProtocolHandler *ret = heap_alloc(sizeof(nsProtocolHandler));
2877
2878     ret->nsIProtocolHandler_iface.lpVtbl = &nsProtocolHandlerVtbl;
2879     ret->ref = 1;
2880     ret->nshandler = nshandler;
2881
2882     return &ret->nsIProtocolHandler_iface;
2883 }
2884
2885 static nsresult NSAPI nsIOService_QueryInterface(nsIIOService*,nsIIDRef,void**);
2886
2887 static nsrefcnt NSAPI nsIOService_AddRef(nsIIOService *iface)
2888 {
2889     return 2;
2890 }
2891
2892 static nsrefcnt NSAPI nsIOService_Release(nsIIOService *iface)
2893 {
2894     return 1;
2895 }
2896
2897 static nsresult NSAPI nsIOService_GetProtocolHandler(nsIIOService *iface, const char *aScheme,
2898                                                      nsIProtocolHandler **_retval)
2899 {
2900     nsIExternalProtocolHandler *nsexthandler;
2901     nsIProtocolHandler *nshandler;
2902     nsresult nsres;
2903
2904     TRACE("(%s %p)\n", debugstr_a(aScheme), _retval);
2905
2906     nsres = nsIIOService_GetProtocolHandler(nsio, aScheme, &nshandler);
2907     if(NS_FAILED(nsres)) {
2908         WARN("GetProtocolHandler failed: %08x\n", nsres);
2909         return nsres;
2910     }
2911
2912     nsres = nsIProtocolHandler_QueryInterface(nshandler, &IID_nsIExternalProtocolHandler,
2913                                               (void**)&nsexthandler);
2914     if(NS_FAILED(nsres)) {
2915         *_retval = nshandler;
2916         return NS_OK;
2917     }
2918
2919     nsIExternalProtocolHandler_Release(nsexthandler);
2920     *_retval = create_protocol_handler(nshandler);
2921     TRACE("return %p\n", *_retval);
2922     return NS_OK;
2923 }
2924
2925 static nsresult NSAPI nsIOService_GetProtocolFlags(nsIIOService *iface, const char *aScheme,
2926                                                     PRUint32 *_retval)
2927 {
2928     TRACE("(%s %p)\n", debugstr_a(aScheme), _retval);
2929     return nsIIOService_GetProtocolFlags(nsio, aScheme, _retval);
2930 }
2931
2932 static BOOL is_gecko_special_uri(const char *spec)
2933 {
2934     static const char *special_schemes[] = {"chrome:", "jar:", "moz-safe-about", "resource:", "javascript:", "wyciwyg:"};
2935     int i;
2936
2937     for(i=0; i < sizeof(special_schemes)/sizeof(*special_schemes); i++) {
2938         if(!strncasecmp(spec, special_schemes[i], strlen(special_schemes[i])))
2939             return TRUE;
2940     }
2941
2942     if(!strncasecmp(spec, "file:", 5)) {
2943         const char *ptr = spec+5;
2944         while(*ptr == '/')
2945             ptr++;
2946         return is_gecko_path(ptr);
2947     }
2948
2949     return FALSE;
2950 }
2951
2952 static nsresult NSAPI nsIOService_NewURI(nsIIOService *iface, const nsACString *aSpec,
2953         const char *aOriginCharset, nsIURI *aBaseURI, nsIURI **_retval)
2954 {
2955     nsWineURI *wine_uri, *base_wine_uri = NULL;
2956     WCHAR new_spec[INTERNET_MAX_URL_LENGTH];
2957     HTMLWindow *window = NULL;
2958     const char *spec = NULL;
2959     nsIURI *uri = NULL;
2960     IUri *urlmon_uri;
2961     nsresult nsres;
2962     HRESULT hres;
2963
2964     TRACE("(%s %s %p %p)\n", debugstr_nsacstr(aSpec), debugstr_a(aOriginCharset),
2965           aBaseURI, _retval);
2966
2967     nsACString_GetData(aSpec, &spec);
2968     if(is_gecko_special_uri(spec))
2969         return nsIIOService_NewURI(nsio, aSpec, aOriginCharset, aBaseURI, _retval);
2970
2971     if(!strncmp(spec, "wine:", 5))
2972         spec += 5;
2973
2974     if(aBaseURI) {
2975         nsres = nsIURI_QueryInterface(aBaseURI, &IID_nsWineURI, (void**)&base_wine_uri);
2976         if(NS_SUCCEEDED(nsres)) {
2977             if(!ensure_uri(base_wine_uri))
2978                 return NS_ERROR_UNEXPECTED;
2979             if(base_wine_uri->window_ref)
2980                 window = base_wine_uri->window_ref->window;
2981         }else {
2982             WARN("Could not get base nsWineURI: %08x\n", nsres);
2983         }
2984     }
2985
2986     MultiByteToWideChar(CP_ACP, 0, spec, -1, new_spec, sizeof(new_spec)/sizeof(WCHAR));
2987
2988     if(base_wine_uri) {
2989         hres = CoInternetCombineUrlEx(base_wine_uri->uri, new_spec, URL_ESCAPE_SPACES_ONLY|URL_DONT_ESCAPE_EXTRA_INFO,
2990                 &urlmon_uri, 0);
2991         if(FAILED(hres))
2992             WARN("CoInternetCombineUrlEx failed: %08x\n", hres);
2993     }else {
2994         hres = CreateUri(new_spec, 0, 0, &urlmon_uri);
2995         if(FAILED(hres))
2996             WARN("CreateUri failed: %08x\n", hres);
2997     }
2998
2999     nsres = nsIIOService_NewURI(nsio, aSpec, aOriginCharset, aBaseURI, &uri);
3000     if(NS_FAILED(nsres))
3001         TRACE("NewURI failed: %08x\n", nsres);
3002
3003     if(FAILED(hres)) {
3004         *_retval = uri;
3005         return nsres;
3006     }
3007
3008     nsres = create_nsuri(urlmon_uri, uri, window, NULL, &wine_uri);
3009     IUri_Release(urlmon_uri);
3010     if(base_wine_uri)
3011         nsIURI_Release(&base_wine_uri->nsIURL_iface);
3012     if(NS_FAILED(nsres))
3013         return nsres;
3014
3015     *_retval = (nsIURI*)&wine_uri->nsIURL_iface;
3016     return nsres;
3017 }
3018
3019 static nsresult NSAPI nsIOService_NewFileURI(nsIIOService *iface, nsIFile *aFile,
3020                                              nsIURI **_retval)
3021 {
3022     TRACE("(%p %p)\n", aFile, _retval);
3023     return nsIIOService_NewFileURI(nsio, aFile, _retval);
3024 }
3025
3026 static nsresult NSAPI nsIOService_NewChannelFromURI(nsIIOService *iface, nsIURI *aURI,
3027                                                      nsIChannel **_retval)
3028 {
3029     nsWineURI *wine_uri;
3030     nsChannel *ret;
3031     nsresult nsres;
3032
3033     TRACE("(%p %p)\n", aURI, _retval);
3034
3035     nsres = nsIURI_QueryInterface(aURI, &IID_nsWineURI, (void**)&wine_uri);
3036     if(NS_FAILED(nsres)) {
3037         TRACE("Could not get nsWineURI: %08x\n", nsres);
3038         return nsIIOService_NewChannelFromURI(nsio, aURI, _retval);
3039     }
3040
3041     nsres = create_nschannel(wine_uri, &ret);
3042     nsIURL_Release(&wine_uri->nsIURL_iface);
3043     if(NS_FAILED(nsres))
3044         return nsres;
3045
3046     nsIURI_AddRef(aURI);
3047     ret->original_uri = aURI;
3048
3049     *_retval = (nsIChannel*)&ret->nsIHttpChannel_iface;
3050     return NS_OK;
3051 }
3052
3053 static nsresult NSAPI nsIOService_NewChannel(nsIIOService *iface, const nsACString *aSpec,
3054         const char *aOriginCharset, nsIURI *aBaseURI, nsIChannel **_retval)
3055 {
3056     TRACE("(%s %s %p %p)\n", debugstr_nsacstr(aSpec), debugstr_a(aOriginCharset), aBaseURI, _retval);
3057     return nsIIOService_NewChannel(nsio, aSpec, aOriginCharset, aBaseURI, _retval);
3058 }
3059
3060 static nsresult NSAPI nsIOService_GetOffline(nsIIOService *iface, PRBool *aOffline)
3061 {
3062     TRACE("(%p)\n", aOffline);
3063     return nsIIOService_GetOffline(nsio, aOffline);
3064 }
3065
3066 static nsresult NSAPI nsIOService_SetOffline(nsIIOService *iface, PRBool aOffline)
3067 {
3068     TRACE("(%x)\n", aOffline);
3069     return nsIIOService_SetOffline(nsio, aOffline);
3070 }
3071
3072 static nsresult NSAPI nsIOService_AllowPort(nsIIOService *iface, PRInt32 aPort,
3073                                              const char *aScheme, PRBool *_retval)
3074 {
3075     TRACE("(%d %s %p)\n", aPort, debugstr_a(aScheme), _retval);
3076     return nsIIOService_AllowPort(nsio, aPort, debugstr_a(aScheme), _retval);
3077 }
3078
3079 static nsresult NSAPI nsIOService_ExtractScheme(nsIIOService *iface, const nsACString *urlString,
3080                                                  nsACString * _retval)
3081 {
3082     TRACE("(%s %p)\n", debugstr_nsacstr(urlString), _retval);
3083     return nsIIOService_ExtractScheme(nsio, urlString, _retval);
3084 }
3085
3086 static const nsIIOServiceVtbl nsIOServiceVtbl = {
3087     nsIOService_QueryInterface,
3088     nsIOService_AddRef,
3089     nsIOService_Release,
3090     nsIOService_GetProtocolHandler,
3091     nsIOService_GetProtocolFlags,
3092     nsIOService_NewURI,
3093     nsIOService_NewFileURI,
3094     nsIOService_NewChannelFromURI,
3095     nsIOService_NewChannel,
3096     nsIOService_GetOffline,
3097     nsIOService_SetOffline,
3098     nsIOService_AllowPort,
3099     nsIOService_ExtractScheme
3100 };
3101
3102 static nsIIOService nsIOService = { &nsIOServiceVtbl };
3103
3104 static nsresult NSAPI nsNetUtil_QueryInterface(nsINetUtil *iface, nsIIDRef riid,
3105         void **result)
3106 {
3107     return nsIIOService_QueryInterface(&nsIOService, riid, result);
3108 }
3109
3110 static nsrefcnt NSAPI nsNetUtil_AddRef(nsINetUtil *iface)
3111 {
3112     return 2;
3113 }
3114
3115 static nsrefcnt NSAPI nsNetUtil_Release(nsINetUtil *iface)
3116 {
3117     return 1;
3118 }
3119
3120 static nsresult NSAPI nsNetUtil_ParseContentType(nsINetUtil *iface, const nsACString *aTypeHeader,
3121         nsACString *aCharset, PRBool *aHadCharset, nsACString *aContentType)
3122 {
3123     TRACE("(%s %p %p %p)\n", debugstr_nsacstr(aTypeHeader), aCharset, aHadCharset, aContentType);
3124
3125     return nsINetUtil_ParseContentType(net_util, aTypeHeader, aCharset, aHadCharset, aContentType);
3126 }
3127
3128 static nsresult NSAPI nsNetUtil_ProtocolHasFlags(nsINetUtil *iface, nsIURI *aURI, PRUint32 aFlags, PRBool *_retval)
3129 {
3130     TRACE("()\n");
3131
3132     return nsINetUtil_ProtocolHasFlags(net_util, aURI, aFlags, _retval);
3133 }
3134
3135 static nsresult NSAPI nsNetUtil_URIChainHasFlags(nsINetUtil *iface, nsIURI *aURI, PRUint32 aFlags, PRBool *_retval)
3136 {
3137     TRACE("(%p %08x %p)\n", aURI, aFlags, _retval);
3138
3139     if(aFlags == (1<<11)) {
3140         *_retval = FALSE;
3141         return NS_OK;
3142     }
3143
3144     return nsINetUtil_URIChainHasFlags(net_util, aURI, aFlags, _retval);
3145 }
3146
3147 static nsresult NSAPI nsNetUtil_ToImmutableURI(nsINetUtil *iface, nsIURI *aURI, nsIURI **_retval)
3148 {
3149     TRACE("(%p %p)\n", aURI, _retval);
3150
3151     return nsINetUtil_ToImmutableURI(net_util, aURI, _retval);
3152 }
3153
3154 static nsresult NSAPI nsNetUtil_NewSimpleNestedURI(nsINetUtil *iface, nsIURI *aURI, nsIURI **_retval)
3155 {
3156     TRACE("(%p %p)\n", aURI, _retval);
3157
3158     return nsINetUtil_NewSimpleNestedURI(net_util, aURI, _retval);
3159 }
3160
3161 static nsresult NSAPI nsNetUtil_EscapeString(nsINetUtil *iface, const nsACString *aString,
3162                                              PRUint32 aEscapeType, nsACString *_retval)
3163 {
3164     TRACE("(%s %x %p)\n", debugstr_nsacstr(aString), aEscapeType, _retval);
3165
3166     return nsINetUtil_EscapeString(net_util, aString, aEscapeType, _retval);
3167 }
3168
3169 static nsresult NSAPI nsNetUtil_EscapeURL(nsINetUtil *iface, const nsACString *aStr, PRUint32 aFlags,
3170                                           nsACString *_retval)
3171 {
3172     TRACE("(%s %08x %p)\n", debugstr_nsacstr(aStr), aFlags, _retval);
3173
3174     return nsINetUtil_EscapeURL(net_util, aStr, aFlags, _retval);
3175 }
3176
3177 static nsresult NSAPI nsNetUtil_UnescapeString(nsINetUtil *iface, const nsACString *aStr,
3178                                                PRUint32 aFlags, nsACString *_retval)
3179 {
3180     TRACE("(%s %08x %p)\n", debugstr_nsacstr(aStr), aFlags, _retval);
3181
3182     return nsINetUtil_UnescapeString(net_util, aStr, aFlags, _retval);
3183 }
3184
3185 static nsresult NSAPI nsNetUtil_ExtractCharsetFromContentType(nsINetUtil *iface, const nsACString *aTypeHeader,
3186         nsACString *aCharset, PRInt32 *aCharsetStart, PRInt32 *aCharsetEnd, PRBool *_retval)
3187 {
3188     TRACE("(%s %p %p %p %p)\n", debugstr_nsacstr(aTypeHeader), aCharset, aCharsetStart,
3189           aCharsetEnd, _retval);
3190
3191     return nsINetUtil_ExtractCharsetFromContentType(net_util, aTypeHeader, aCharset, aCharsetStart, aCharsetEnd, _retval);
3192 }
3193
3194 static const nsINetUtilVtbl nsNetUtilVtbl = {
3195     nsNetUtil_QueryInterface,
3196     nsNetUtil_AddRef,
3197     nsNetUtil_Release,
3198     nsNetUtil_ParseContentType,
3199     nsNetUtil_ProtocolHasFlags,
3200     nsNetUtil_URIChainHasFlags,
3201     nsNetUtil_ToImmutableURI,
3202     nsNetUtil_NewSimpleNestedURI,
3203     nsNetUtil_EscapeString,
3204     nsNetUtil_EscapeURL,
3205     nsNetUtil_UnescapeString,
3206     nsNetUtil_ExtractCharsetFromContentType
3207 };
3208
3209 static nsINetUtil nsNetUtil = { &nsNetUtilVtbl };
3210
3211 static nsresult NSAPI nsIOService_QueryInterface(nsIIOService *iface, nsIIDRef riid,
3212         void **result)
3213 {
3214     *result = NULL;
3215
3216     if(IsEqualGUID(&IID_nsISupports, riid))
3217         *result = &nsIOService;
3218     else if(IsEqualGUID(&IID_nsIIOService, riid))
3219         *result = &nsIOService;
3220     else if(IsEqualGUID(&IID_nsINetUtil, riid))
3221         *result = &nsNetUtil;
3222
3223     if(*result) {
3224         nsISupports_AddRef((nsISupports*)*result);
3225         return NS_OK;
3226     }
3227
3228     FIXME("(%s %p)\n", debugstr_guid(riid), result);
3229     return NS_NOINTERFACE;
3230 }
3231
3232 static nsresult NSAPI nsIOServiceFactory_QueryInterface(nsIFactory *iface, nsIIDRef riid,
3233         void **result)
3234 {
3235     *result = NULL;
3236
3237     if(IsEqualGUID(&IID_nsISupports, riid)) {
3238         TRACE("(IID_nsISupports %p)\n", result);
3239         *result = iface;
3240     }else if(IsEqualGUID(&IID_nsIFactory, riid)) {
3241         TRACE("(IID_nsIFactory %p)\n", result);
3242         *result = iface;
3243     }
3244
3245     if(*result) {
3246         nsIFactory_AddRef(iface);
3247         return NS_OK;
3248     }
3249
3250     WARN("(%s %p)\n", debugstr_guid(riid), result);
3251     return NS_NOINTERFACE;
3252 }
3253
3254 static nsrefcnt NSAPI nsIOServiceFactory_AddRef(nsIFactory *iface)
3255 {
3256     return 2;
3257 }
3258
3259 static nsrefcnt NSAPI nsIOServiceFactory_Release(nsIFactory *iface)
3260 {
3261     return 1;
3262 }
3263
3264 static nsresult NSAPI nsIOServiceFactory_CreateInstance(nsIFactory *iface,
3265         nsISupports *aOuter, const nsIID *iid, void **result)
3266 {
3267     return nsIIOService_QueryInterface(&nsIOService, iid, result);
3268 }
3269
3270 static nsresult NSAPI nsIOServiceFactory_LockFactory(nsIFactory *iface, PRBool lock)
3271 {
3272     WARN("(%x)\n", lock);
3273     return NS_OK;
3274 }
3275
3276 static const nsIFactoryVtbl nsIOServiceFactoryVtbl = {
3277     nsIOServiceFactory_QueryInterface,
3278     nsIOServiceFactory_AddRef,
3279     nsIOServiceFactory_Release,
3280     nsIOServiceFactory_CreateInstance,
3281     nsIOServiceFactory_LockFactory
3282 };
3283
3284 static nsIFactory nsIOServiceFactory = { &nsIOServiceFactoryVtbl };
3285
3286 static BOOL translate_url(HTMLDocumentObj *doc, nsWineURI *uri)
3287 {
3288     OLECHAR *new_url = NULL;
3289     WCHAR *url;
3290     BOOL ret = FALSE;
3291     HRESULT hres;
3292
3293     if(!doc->hostui || !ensure_uri(uri))
3294         return FALSE;
3295
3296     hres = IUri_GetDisplayUri(uri->uri, &url);
3297     if(FAILED(hres))
3298         return FALSE;
3299
3300     hres = IDocHostUIHandler_TranslateUrl(doc->hostui, 0, url, &new_url);
3301     if(hres == S_OK && new_url) {
3302         if(strcmpW(url, new_url)) {
3303             FIXME("TranslateUrl returned new URL %s -> %s\n", debugstr_w(url), debugstr_w(new_url));
3304             ret = TRUE;
3305         }
3306         CoTaskMemFree(new_url);
3307     }
3308
3309     SysFreeString(url);
3310     return ret;
3311 }
3312
3313 nsresult on_start_uri_open(NSContainer *nscontainer, nsIURI *uri, PRBool *_retval)
3314 {
3315     nsWineURI *wine_uri;
3316     nsresult nsres;
3317
3318     *_retval = FALSE;
3319
3320     nsres = nsIURI_QueryInterface(uri, &IID_nsWineURI, (void**)&wine_uri);
3321     if(NS_FAILED(nsres)) {
3322         WARN("Could not get nsWineURI: %08x\n", nsres);
3323         return NS_ERROR_NOT_IMPLEMENTED;
3324     }
3325
3326     if(!wine_uri->is_doc_uri) {
3327         wine_uri->is_doc_uri = TRUE;
3328
3329         if(!wine_uri->container) {
3330             nsIWebBrowserChrome_AddRef(&nscontainer->nsIWebBrowserChrome_iface);
3331             wine_uri->container = nscontainer;
3332         }
3333
3334         if(nscontainer->doc)
3335             *_retval = translate_url(nscontainer->doc, wine_uri);
3336     }
3337
3338     nsIURI_Release(&wine_uri->nsIURL_iface);
3339     return NS_OK;
3340 }
3341
3342 void init_nsio(nsIComponentManager *component_manager, nsIComponentRegistrar *registrar)
3343 {
3344     nsIFactory *old_factory = NULL;
3345     nsresult nsres;
3346
3347     nsres = nsIComponentManager_GetClassObject(component_manager, &NS_IOSERVICE_CID,
3348                                                &IID_nsIFactory, (void**)&old_factory);
3349     if(NS_FAILED(nsres)) {
3350         ERR("Could not get factory: %08x\n", nsres);
3351         return;
3352     }
3353
3354     nsres = nsIFactory_CreateInstance(old_factory, NULL, &IID_nsIIOService, (void**)&nsio);
3355     if(NS_FAILED(nsres)) {
3356         ERR("Couldn not create nsIOService instance %08x\n", nsres);
3357         nsIFactory_Release(old_factory);
3358         return;
3359     }
3360
3361     nsres = nsIIOService_QueryInterface(nsio, &IID_nsINetUtil, (void**)&net_util);
3362     if(NS_FAILED(nsres)) {
3363         WARN("Could not get nsINetUtil interface: %08x\n", nsres);
3364         nsIIOService_Release(nsio);
3365         return;
3366     }
3367
3368     nsres = nsIComponentRegistrar_UnregisterFactory(registrar, &NS_IOSERVICE_CID, old_factory);
3369     nsIFactory_Release(old_factory);
3370     if(NS_FAILED(nsres))
3371         ERR("UnregisterFactory failed: %08x\n", nsres);
3372
3373     nsres = nsIComponentRegistrar_RegisterFactory(registrar, &NS_IOSERVICE_CID,
3374             NS_IOSERVICE_CLASSNAME, NS_IOSERVICE_CONTRACTID, &nsIOServiceFactory);
3375     if(NS_FAILED(nsres))
3376         ERR("RegisterFactory failed: %08x\n", nsres);
3377 }
3378
3379 void release_nsio(void)
3380 {
3381     if(net_util) {
3382         nsINetUtil_Release(net_util);
3383         net_util = NULL;
3384     }
3385
3386     if(nsio) {
3387         nsIIOService_Release(nsio);
3388         nsio = NULL;
3389     }
3390 }