src/third_party/WebKit/Source/core/workers/WorkerScriptLoader.h

   1 /*
   2  * Copyright (C) 2009 Apple Inc. All Rights Reserved.
   3  * Copyright (C) 2009, 2011 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 #ifndef WorkerScriptLoader_h
  29 #define WorkerScriptLoader_h
  30
  31 #include "core/loader/ThreadableLoader.h"
  32 #include "core/loader/ThreadableLoaderClient.h"
  33 #include "platform/network/ResourceRequest.h"
  34 #include "platform/weborigin/KURL.h"
  35 #include "public/platform/WebURLRequest.h"
  36 #include "wtf/FastAllocBase.h"
  37 #include "wtf/PassRefPtr.h"
  38 #include "wtf/RefCounted.h"
  39 #include "wtf/text/StringBuilder.h"
  40
  41 namespace blink {
  42
  43 class ResourceRequest;
  44 class ResourceResponse;
  45 class ExecutionContext;
  46 class TextResourceDecoder;
  47 class WorkerScriptLoaderClient;
  48
  49 class WorkerScriptLoader final : public RefCounted<WorkerScriptLoader>, public ThreadableLoaderClient {
  50     WTF_MAKE_FAST_ALLOCATED;
  51 public:
  52     static PassRefPtr<WorkerScriptLoader> create()
  53     {
  54         return adoptRef(new WorkerScriptLoader());
  55     }
  56
  57     void loadSynchronously(ExecutionContext&, const KURL&, CrossOriginRequestPolicy);
  58     void loadAsynchronously(ExecutionContext&, const KURL&, CrossOriginRequestPolicy, WorkerScriptLoaderClient*);
  59
  60     void notifyError();
  61
  62     // This will immediately lead to notifyFinished() if loadAsynchronously
  63     // is in progress.
  64     void cancel();
  65
  66     void setClient(WorkerScriptLoaderClient* client) { m_client = client; }
  67
  68     String script();
  69     const KURL& url() const { return m_url; }
  70     const KURL& responseURL() const;
  71     bool failed() const { return m_failed; }
  72     unsigned long identifier() const { return m_identifier; }
  73
  74     virtual void didReceiveResponse(unsigned long /*identifier*/, const ResourceResponse&, PassOwnPtr<WebDataConsumerHandle>) override;
  75     virtual void didReceiveData(const char* data, unsigned dataLength) override;
  76     virtual void didFinishLoading(unsigned long identifier, double) override;
  77     virtual void didFail(const ResourceError&) override;
  78     virtual void didFailRedirectCheck() override;
  79
  80     void setRequestContext(WebURLRequest::RequestContext requestContext) { m_requestContext = requestContext; }
  81
  82 private:
  83     friend class WTF::RefCounted<WorkerScriptLoader>;
  84
  85     WorkerScriptLoader();
  86     virtual ~WorkerScriptLoader();
  87
  88     PassOwnPtr<ResourceRequest> createResourceRequest();
  89     void notifyFinished();
  90
  91     WorkerScriptLoaderClient* m_client;
  92     RefPtr<ThreadableLoader> m_threadableLoader;
  93     String m_responseEncoding;
  94     OwnPtr<TextResourceDecoder> m_decoder;
  95     StringBuilder m_script;
  96     KURL m_url;
  97     KURL m_responseURL;
  98     bool m_failed;
  99     unsigned long m_identifier;
 100     bool m_finishing;
 101     WebURLRequest::RequestContext m_requestContext;
 102 };
 103
 104 } // namespace blink
 105
 106 #endif // WorkerScriptLoader_h