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