Replace WTF::move with WTFMove
[WebKit-https.git] / Source / WebCore / contentextensions / ContentExtension.cpp
1 /*
2  * Copyright (C) 2015 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. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "ContentExtension.h"
28
29 #include "CompiledContentExtension.h"
30 #include "ContentExtensionsBackend.h"
31 #include "StyleSheetContents.h"
32 #include <wtf/text/StringBuilder.h>
33
34 #if ENABLE(CONTENT_EXTENSIONS)
35
36 namespace WebCore {
37 namespace ContentExtensions {
38
39 RefPtr<ContentExtension> ContentExtension::create(const String& identifier, Ref<CompiledContentExtension>&& compiledExtension)
40 {
41     return adoptRef(*new ContentExtension(identifier, WTFMove(compiledExtension)));
42 }
43
44 ContentExtension::ContentExtension(const String& identifier, Ref<CompiledContentExtension>&& compiledExtension)
45     : m_identifier(identifier)
46     , m_compiledExtension(WTFMove(compiledExtension))
47 {
48     DFABytecodeInterpreter withoutDomains(m_compiledExtension->filtersWithoutDomainsBytecode(), m_compiledExtension->filtersWithoutDomainsBytecodeLength());
49     DFABytecodeInterpreter withDomains(m_compiledExtension->filtersWithDomainsBytecode(), m_compiledExtension->filtersWithDomainsBytecodeLength());
50     for (uint64_t action : withoutDomains.actionsMatchingEverything()) {
51         ASSERT(static_cast<uint32_t>(action) == action);
52         m_universalActionsWithoutDomains.append(static_cast<uint32_t>(action));
53     }
54     for (uint64_t action : withDomains.actionsMatchingEverything()) {
55         ASSERT((action & ~IfDomainFlag) == static_cast<uint32_t>(action));
56         m_universalActionsWithDomains.append(action);
57     }
58     
59     compileGlobalDisplayNoneStyleSheet();
60     m_universalActionsWithoutDomains.shrinkToFit();
61     m_universalActionsWithDomains.shrinkToFit();
62 }
63
64 uint32_t ContentExtension::findFirstIgnorePreviousRules() const
65 {
66     auto* actions = m_compiledExtension->actions();
67     uint32_t actionsLength = m_compiledExtension->actionsLength();
68     uint32_t currentActionIndex = 0;
69     while (currentActionIndex < actionsLength) {
70         if (Action::deserializeType(actions, actionsLength, currentActionIndex) == ActionType::IgnorePreviousRules)
71             return currentActionIndex;
72         currentActionIndex += Action::serializedLength(actions, actionsLength, currentActionIndex);
73     }
74     return std::numeric_limits<uint32_t>::max();
75 }
76     
77 StyleSheetContents* ContentExtension::globalDisplayNoneStyleSheet()
78 {
79     return m_globalDisplayNoneStyleSheet.get();
80 }
81
82 void ContentExtension::compileGlobalDisplayNoneStyleSheet()
83 {
84     uint32_t firstIgnorePreviousRules = findFirstIgnorePreviousRules();
85     
86     auto* actions = m_compiledExtension->actions();
87     uint32_t actionsLength = m_compiledExtension->actionsLength();
88
89     auto inGlobalDisplayNoneStyleSheet = [&](const uint32_t location)
90     {
91         return location < firstIgnorePreviousRules && Action::deserializeType(actions, actionsLength, location) == ActionType::CSSDisplayNoneSelector;
92     };
93     
94     StringBuilder css;
95     for (uint32_t universalActionLocation : m_universalActionsWithoutDomains) {
96         if (inGlobalDisplayNoneStyleSheet(universalActionLocation)) {
97             if (!css.isEmpty())
98                 css.append(',');
99             Action action = Action::deserialize(actions, actionsLength, universalActionLocation);
100             ASSERT(action.type() == ActionType::CSSDisplayNoneSelector);
101             css.append(action.stringArgument());
102         }
103     }
104     if (css.isEmpty())
105         return;
106     css.append("{");
107     css.append(ContentExtensionsBackend::displayNoneCSSRule());
108     css.append("}");
109
110     m_globalDisplayNoneStyleSheet = StyleSheetContents::create();
111     m_globalDisplayNoneStyleSheet->setIsUserStyleSheet(true);
112     if (!m_globalDisplayNoneStyleSheet->parseString(css.toString()))
113         m_globalDisplayNoneStyleSheet = nullptr;
114
115     // These actions don't need to be applied individually any more. They will all be applied to every page as a precompiled style sheet.
116     m_universalActionsWithoutDomains.removeAllMatching(inGlobalDisplayNoneStyleSheet);
117 }
118
119 void ContentExtension::populateDomainCacheIfNeeded(const String& domain)
120 {
121     if (m_cachedDomain != domain) {
122         DFABytecodeInterpreter interpreter(m_compiledExtension->domainFiltersBytecode(), m_compiledExtension->domainFiltersBytecodeLength());
123         const uint16_t allLoadTypesAndResourceTypes = LoadTypeMask | ResourceTypeMask;
124         auto domainActions = interpreter.interpret(domain.utf8(), allLoadTypesAndResourceTypes);
125         
126         m_cachedDomainActions.clear();
127         for (uint64_t action : domainActions)
128             m_cachedDomainActions.add(action);
129         
130         m_cachedUniversalDomainActions.clear();
131         for (uint64_t action : m_universalActionsWithDomains) {
132             ASSERT_WITH_MESSAGE((action & ~IfDomainFlag) == static_cast<uint32_t>(action), "Universal actions with domains should not have flags.");
133             if (!!(action & IfDomainFlag) == m_cachedDomainActions.contains(action))
134                 m_cachedUniversalDomainActions.append(static_cast<uint32_t>(action));
135         }
136         m_cachedUniversalDomainActions.shrinkToFit();
137         m_cachedDomain = domain;
138     }
139 }
140
141 const DFABytecodeInterpreter::Actions& ContentExtension::cachedDomainActions(const String& domain)
142 {
143     populateDomainCacheIfNeeded(domain);
144     return m_cachedDomainActions;
145 }
146
147 const Vector<uint32_t>& ContentExtension::universalActionsWithDomains(const String& domain)
148 {
149     populateDomainCacheIfNeeded(domain);
150     return m_cachedUniversalDomainActions;
151 }
152     
153 } // namespace ContentExtensions
154 } // namespace WebCore
155
156 #endif // ENABLE(CONTENT_EXTENSIONS)