1 | /* |
2 | * Copyright (C) 2009 Apple 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 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 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 | #pragma once |
27 | |
28 | #include <wtf/URL.h> |
29 | #include "UserContentTypes.h" |
30 | #include "UserScriptTypes.h" |
31 | #include <wtf/Vector.h> |
32 | |
33 | namespace WebCore { |
34 | |
35 | class UserScript { |
36 | WTF_MAKE_FAST_ALLOCATED; |
37 | public: |
38 | UserScript() = default; |
39 | ~UserScript() = default; |
40 | |
41 | UserScript(String&& source, URL&& url, Vector<String>&& whitelist, Vector<String>&& blacklist, UserScriptInjectionTime injectionTime, UserContentInjectedFrames injectedFrames) |
42 | : m_source(WTFMove(source)) |
43 | , m_url(WTFMove(url)) |
44 | , m_whitelist(WTFMove(whitelist)) |
45 | , m_blacklist(WTFMove(blacklist)) |
46 | , m_injectionTime(injectionTime) |
47 | , m_injectedFrames(injectedFrames) |
48 | { |
49 | } |
50 | |
51 | const String& source() const { return m_source; } |
52 | const URL& url() const { return m_url; } |
53 | const Vector<String>& whitelist() const { return m_whitelist; } |
54 | const Vector<String>& blacklist() const { return m_blacklist; } |
55 | UserScriptInjectionTime injectionTime() const { return m_injectionTime; } |
56 | UserContentInjectedFrames injectedFrames() const { return m_injectedFrames; } |
57 | |
58 | template<class Encoder> void encode(Encoder&) const; |
59 | template<class Decoder> static bool decode(Decoder&, UserScript&); |
60 | |
61 | private: |
62 | String m_source; |
63 | URL m_url; |
64 | Vector<String> m_whitelist; |
65 | Vector<String> m_blacklist; |
66 | UserScriptInjectionTime m_injectionTime { InjectAtDocumentStart }; |
67 | UserContentInjectedFrames m_injectedFrames { InjectInAllFrames }; |
68 | }; |
69 | |
70 | template<class Encoder> |
71 | void UserScript::encode(Encoder& encoder) const |
72 | { |
73 | encoder << m_source; |
74 | encoder << m_url; |
75 | encoder << m_whitelist; |
76 | encoder << m_blacklist; |
77 | encoder.encodeEnum(m_injectionTime); |
78 | encoder.encodeEnum(m_injectedFrames); |
79 | } |
80 | |
81 | template<class Decoder> |
82 | bool UserScript::decode(Decoder& decoder, UserScript& userScript) |
83 | { |
84 | String source; |
85 | if (!decoder.decode(source)) |
86 | return false; |
87 | |
88 | URL url; |
89 | if (!decoder.decode(url)) |
90 | return false; |
91 | |
92 | Vector<String> whitelist; |
93 | if (!decoder.decode(whitelist)) |
94 | return false; |
95 | |
96 | Vector<String> blacklist; |
97 | if (!decoder.decode(blacklist)) |
98 | return false; |
99 | |
100 | UserScriptInjectionTime injectionTime; |
101 | if (!decoder.decodeEnum(injectionTime)) |
102 | return false; |
103 | |
104 | UserContentInjectedFrames injectedFrames; |
105 | if (!decoder.decodeEnum(injectedFrames)) |
106 | return false; |
107 | |
108 | userScript = { WTFMove(source), WTFMove(url), WTFMove(whitelist), WTFMove(blacklist), injectionTime, injectedFrames }; |
109 | return true; |
110 | } |
111 | |
112 | } // namespace WebCore |
113 | |