8737b881b9a444d8c8804533456964378f323669
[WebKit-https.git] / WebCore / workers / WorkerScriptLoader.cpp
1 /*
2  * Copyright (C) 2009 Apple Inc. All Rights Reserved.
3  * Copyright (C) 2009 Google Inc. All Rights Reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
25  *
26  */
27
28 #include "config.h"
29
30 #if ENABLE(WORKERS)
31
32 #include "WorkerScriptLoader.h"
33
34 #include "ResourceRequest.h"
35 #include "ScriptExecutionContext.h"
36 #include "SecurityOrigin.h"
37 #include "WorkerContext.h"
38 #include "WorkerScriptLoaderClient.h"
39 #include "WorkerThreadableLoader.h"
40
41 namespace WebCore {
42
43 WorkerScriptLoader::WorkerScriptLoader()
44     : m_client(0)
45     , m_failed(false)
46     , m_identifier(0)
47 {
48 }
49
50 void WorkerScriptLoader::loadSynchronously(ScriptExecutionContext* scriptExecutionContext, const String& url, CrossOriginRedirectPolicy crossOriginRedirectPolicy)
51 {
52     ResourceRequest request(url);
53     request.setHTTPMethod("GET");
54
55     ASSERT(scriptExecutionContext->isWorkerContext());
56     WorkerThreadableLoader::loadResourceSynchronously(static_cast<WorkerContext*>(scriptExecutionContext), request, *this, AllowStoredCredentials, crossOriginRedirectPolicy);
57 }
58     
59 void WorkerScriptLoader::loadAsynchronously(ScriptExecutionContext* scriptExecutionContext, const String& url, CrossOriginRedirectPolicy crossOriginRedirectPolicy, WorkerScriptLoaderClient* client)
60 {
61     ASSERT(client);
62     m_client = client;
63
64     ResourceRequest request(url);
65     request.setHTTPMethod("GET");
66
67     m_threadableLoader = ThreadableLoader::create(scriptExecutionContext, this, request, DoNotSendLoadCallbacks, DoNotSniffContent, AllowStoredCredentials, crossOriginRedirectPolicy);
68 }
69     
70 void WorkerScriptLoader::didReceiveResponse(const ResourceResponse& response)
71 {
72     if (response.httpStatusCode() / 100 != 2 && response.httpStatusCode()) {
73         m_failed = true;
74         return;
75     }
76     m_responseEncoding = response.textEncodingName();
77 }
78
79 void WorkerScriptLoader::didReceiveData(const char* data, int len)
80 {
81     if (m_failed)
82         return;
83
84     if (!m_decoder) {
85         if (!m_responseEncoding.isEmpty())
86             m_decoder = TextResourceDecoder::create("text/javascript", m_responseEncoding);
87         else
88             m_decoder = TextResourceDecoder::create("text/javascript", "UTF-8");
89     }
90
91     if (!len)
92         return;
93     
94     if (len == -1)
95         len = strlen(data);
96     
97     m_script += m_decoder->decode(data, len);
98 }
99
100 void WorkerScriptLoader::didFinishLoading(unsigned long identifier)
101 {
102     if (m_failed)
103         return;
104
105     if (m_decoder)
106         m_script += m_decoder->flush();
107
108     m_identifier = identifier;
109     notifyFinished();
110 }
111
112 void WorkerScriptLoader::didFail(const ResourceError&)
113 {
114     m_failed = true;
115     notifyFinished();
116 }
117
118 void WorkerScriptLoader::didFailRedirectCheck()
119 {
120     m_failed = true;
121     notifyFinished();
122 }
123
124 void WorkerScriptLoader::didReceiveAuthenticationCancellation(const ResourceResponse&)
125 {
126     m_failed = true;
127     notifyFinished();
128 }
129     
130 void WorkerScriptLoader::notifyFinished()
131 {
132     if (m_client)
133         m_client->notifyFinished();
134 }
135
136 } // namespace WebCore
137
138 #endif // ENABLE(WORKERS)