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