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