42010abd2543849b6c8df942c7f5755a076abf67
[WebKit-https.git] / WebCore / platform / mac / ResourceLoaderMac.mm
1 /*
2  * Copyright (C) 2004 Apple Computer, Inc.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #import "config.h"
27 #import "ResourceLoader.h"
28 #import "ResourceLoaderInternal.h"
29
30 #import "BlockExceptions.h"
31 #import "DocLoader.h"
32 #import "FoundationExtras.h"
33 #import "FrameMac.h"
34 #import "KURL.h"
35 #import "FormDataMac.h"
36 #import "LoaderFunctions.h"
37 #import "WebCoreResourceLoaderImp.h"
38 #import "Logging.h"
39 #import "WebCoreFrameBridge.h"
40
41 namespace WebCore {
42     
43 ResourceLoaderInternal::~ResourceLoaderInternal()
44 {
45     HardRelease(response);
46     HardRelease(loader);
47 }
48
49 ResourceLoader::~ResourceLoader()
50 {
51     // This will cancel the handle, and who knows what that could do
52     BEGIN_BLOCK_OBJC_EXCEPTIONS;
53     [d->loader jobWillBeDeallocated];
54     END_BLOCK_OBJC_EXCEPTIONS;
55     delete d;
56 }
57
58 bool ResourceLoader::start(DocLoader* docLoader)
59 {
60     ref();
61     d->m_loading = true;
62
63     ASSERT(docLoader);
64     
65     FrameMac* frame = Mac(docLoader->frame());
66     if (!frame) {
67         kill();
68         return false;
69     }
70
71     WebCoreFrameBridge* bridge = frame->bridge();
72
73     frame->didTellBridgeAboutLoad(url().url());
74
75     BEGIN_BLOCK_OBJC_EXCEPTIONS;
76
77     WebCoreResourceLoaderImp* resourceLoader = [[WebCoreResourceLoaderImp alloc] initWithJob:this];
78
79     id <WebCoreResourceHandle> handle;
80
81     NSDictionary* headerDict = nil;
82     String headerString = queryMetaData("customHTTPHeader");
83
84     if (!headerString.isEmpty())
85         headerDict = [NSDictionary _webcore_dictionaryWithHeaderString:headerString];
86
87     if (!postData().elements().isEmpty())
88         handle = [bridge startLoadingResource:resourceLoader withMethod:method() URL:url().getNSURL() customHeaders:headerDict postData:arrayFromFormData(postData())];
89     else
90         handle = [bridge startLoadingResource:resourceLoader withMethod:method() URL:url().getNSURL() customHeaders:headerDict];
91     [resourceLoader setHandle:handle];
92     [resourceLoader release];
93
94     if (handle)
95         return true;
96
97     END_BLOCK_OBJC_EXCEPTIONS;
98
99     kill();
100     return false;
101 }
102
103 void ResourceLoader::assembleResponseHeaders() const
104 {
105     if (!d->assembledResponseHeaders) {
106         if ([d->response isKindOfClass:[NSHTTPURLResponse class]]) {
107             NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)d->response;
108             NSDictionary *headers = [httpResponse allHeaderFields];
109             d->responseHeaders = DeprecatedString::fromNSString(HeaderStringFromDictionary(headers, [httpResponse statusCode]));
110         }
111         d->assembledResponseHeaders = true;
112     }
113 }
114
115 void ResourceLoader::retrieveCharset() const
116 {
117     if (!d->retrievedCharset) {
118         NSString *charset = [d->response textEncodingName];
119         if (charset)
120             d->metaData.set("charset", charset);
121         d->retrievedCharset = true;
122     }
123 }
124
125 void ResourceLoader::setLoader(WebCoreResourceLoaderImp *loader)
126 {
127     HardRetain(loader);
128     HardRelease(d->loader);
129     d->loader = loader;
130 }
131
132 void ResourceLoader::receivedResponse(NSURLResponse* response)
133 {
134     d->assembledResponseHeaders = false;
135     d->retrievedCharset = false;
136     d->response = response;
137     HardRetain(d->response);
138     if (d->client)
139         d->client->receivedResponse(this, response);
140 }
141
142 void ResourceLoader::cancel()
143 {
144     [d->loader jobCanceledLoad];
145 }
146
147 } // namespace WebCore