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, RedirectOriginCheck redirectOriginCheck)
  51 {
  52     ResourceRequest request(url);
  53     request.setHTTPMethod("GET");
  54
  55     ASSERT(scriptExecutionContext->isWorkerContext());
  56     WorkerThreadableLoader::loadResourceSynchronously(static_cast<WorkerContext*>(scriptExecutionContext), request, *this, AllowStoredCredentials, redirectOriginCheck);
  57 }
  58
  59 void WorkerScriptLoader::loadAsynchronously(ScriptExecutionContext* scriptExecutionContext, const String& url, RedirectOriginCheck redirectOriginCheck, 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, redirectOriginCheck);
  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)