Add CSP support to service workers
[WebKit-https.git] / Source / WebCore / 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 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 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 #pragma once
28
29 #include "ContentSecurityPolicyResponseHeaders.h"
30 #include "ResourceError.h"
31 #include "ResourceRequest.h"
32 #include "ThreadableLoader.h"
33 #include "ThreadableLoaderClient.h"
34 #include "URL.h"
35 #include <memory>
36 #include <wtf/FastMalloc.h>
37 #include <wtf/RefCounted.h>
38 #include <wtf/RefPtr.h>
39 #include <wtf/text/StringBuilder.h>
40
41 namespace WebCore {
42
43 class ResourceResponse;
44 class ScriptExecutionContext;
45 class TextResourceDecoder;
46 class WorkerScriptLoaderClient;
47
48 class WorkerScriptLoader : public RefCounted<WorkerScriptLoader>, public ThreadableLoaderClient {
49     WTF_MAKE_FAST_ALLOCATED;
50 public:
51     static Ref<WorkerScriptLoader> create()
52     {
53         return adoptRef(*new WorkerScriptLoader);
54     }
55
56     void loadSynchronously(ScriptExecutionContext*, const URL&, FetchOptions::Mode, FetchOptions::Cache, ContentSecurityPolicyEnforcement, const String& initiatorIdentifier);
57     void loadAsynchronously(ScriptExecutionContext&, ResourceRequest&&, FetchOptions::Mode, FetchOptions::Cache, FetchOptions::Redirect, ContentSecurityPolicyEnforcement, WorkerScriptLoaderClient&);
58
59     void notifyError();
60
61     String script();
62     const ContentSecurityPolicyResponseHeaders& contentSecurityPolicy() const { return m_contentSecurityPolicy; }
63     const URL& url() const { return m_url; }
64     const URL& responseURL() const;
65     const String& responseMIMEType() const { return m_responseMIMEType; }
66     bool failed() const { return m_failed; }
67     unsigned long identifier() const { return m_identifier; }
68     const ResourceError& error() const { return m_error; }
69
70     void didReceiveResponse(unsigned long identifier, const ResourceResponse&) override;
71     void didReceiveData(const char* data, int dataLength) override;
72     void didFinishLoading(unsigned long identifier) override;
73     void didFail(const ResourceError&) override;
74
75     void cancel();
76
77 private:
78     friend class WTF::RefCounted<WorkerScriptLoader>;
79
80     WorkerScriptLoader();
81     ~WorkerScriptLoader();
82
83     std::unique_ptr<ResourceRequest> createResourceRequest(const String& initiatorIdentifier);
84     void notifyFinished();
85
86     WorkerScriptLoaderClient* m_client { nullptr };
87     RefPtr<ThreadableLoader> m_threadableLoader;
88     String m_responseEncoding;
89     RefPtr<TextResourceDecoder> m_decoder;
90     StringBuilder m_script;
91     URL m_url;
92     URL m_responseURL;
93     String m_responseMIMEType;
94     ContentSecurityPolicyResponseHeaders m_contentSecurityPolicy;
95     unsigned long m_identifier { 0 };
96     bool m_failed { false };
97     bool m_finishing { false };
98     ResourceError m_error;
99 };
100
101 } // namespace WebCore