Enable Pointer Events on watchOS
[WebKit-https.git] / Source / WebCore / testing / MockContentFilter.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 "MockContentFilter.h"
28
29 #if ENABLE(CONTENT_FILTERING)
30
31 #include "ContentFilter.h"
32 #include "ContentFilterUnblockHandler.h"
33 #include "Logging.h"
34 #include "ResourceRequest.h"
35 #include "ResourceResponse.h"
36 #include "SharedBuffer.h"
37 #include <mutex>
38 #include <wtf/text/CString.h>
39
40 namespace WebCore {
41
42 using Decision = MockContentFilterSettings::Decision;
43 using DecisionPoint = MockContentFilterSettings::DecisionPoint;
44
45 void MockContentFilter::ensureInstalled()
46 {
47     static std::once_flag onceFlag;
48     std::call_once(onceFlag, []{
49         ContentFilter::addType<MockContentFilter>();
50     });
51 }
52
53 static inline MockContentFilterSettings& settings()
54 {
55     return MockContentFilterSettings::singleton();
56 }
57
58 bool MockContentFilter::enabled()
59 {
60     bool enabled = settings().enabled();
61     LOG(ContentFiltering, "MockContentFilter is %s.\n", enabled ? "enabled" : "not enabled");
62     return enabled;
63 }
64
65 std::unique_ptr<MockContentFilter> MockContentFilter::create()
66 {
67     return std::make_unique<MockContentFilter>();
68 }
69
70 void MockContentFilter::willSendRequest(ResourceRequest& request, const ResourceResponse& redirectResponse)
71 {
72     if (!enabled()) {
73         m_state = State::Allowed;
74         return;
75     }
76
77     if (redirectResponse.isNull())
78         maybeDetermineStatus(DecisionPoint::AfterWillSendRequest);
79     else
80         maybeDetermineStatus(DecisionPoint::AfterRedirect);
81
82     if (m_state == State::Filtering)
83         return;
84
85     String modifiedRequestURLString { settings().modifiedRequestURL() };
86     if (modifiedRequestURLString.isEmpty())
87         return;
88
89     URL modifiedRequestURL { request.url(), modifiedRequestURLString };
90     if (!modifiedRequestURL.isValid()) {
91         LOG(ContentFiltering, "MockContentFilter failed to convert %s to a  URL.\n", modifiedRequestURL.string().ascii().data());
92         return;
93     }
94
95     request.setURL(modifiedRequestURL);
96 }
97
98 void MockContentFilter::responseReceived(const ResourceResponse&)
99 {
100     maybeDetermineStatus(DecisionPoint::AfterResponse);
101 }
102
103 void MockContentFilter::addData(const char*, int)
104 {
105     maybeDetermineStatus(DecisionPoint::AfterAddData);
106 }
107
108 void MockContentFilter::finishedAddingData()
109 {
110     maybeDetermineStatus(DecisionPoint::AfterFinishedAddingData);
111 }
112
113 Ref<SharedBuffer> MockContentFilter::replacementData() const
114 {
115     ASSERT(didBlockData());
116     return SharedBuffer::create(m_replacementData.data(), m_replacementData.size());
117 }
118
119 ContentFilterUnblockHandler MockContentFilter::unblockHandler() const
120 {
121     ASSERT(didBlockData());
122     using DecisionHandlerFunction = ContentFilterUnblockHandler::DecisionHandlerFunction;
123
124     return ContentFilterUnblockHandler {
125         MockContentFilterSettings::unblockURLHost(), [](DecisionHandlerFunction decisionHandler) {
126             bool shouldAllow { settings().unblockRequestDecision() == Decision::Allow };
127             if (shouldAllow)
128                 settings().setDecision(Decision::Allow);
129             LOG(ContentFiltering, "MockContentFilter %s the unblock request.\n", shouldAllow ? "allowed" : "did not allow");
130             decisionHandler(shouldAllow);
131         }
132     };
133 }
134
135 String MockContentFilter::unblockRequestDeniedScript() const
136 {
137     return "unblockRequestDenied()"_s;
138 }
139
140 void MockContentFilter::maybeDetermineStatus(DecisionPoint decisionPoint)
141 {
142     if (m_state != State::Filtering || decisionPoint != settings().decisionPoint())
143         return;
144
145     LOG(ContentFiltering, "MockContentFilter stopped buffering with state %u at decision point %u.\n", m_state, decisionPoint);
146
147     m_state = settings().decision() == Decision::Allow ? State::Allowed : State::Blocked;
148     if (m_state != State::Blocked)
149         return;
150
151     m_replacementData.clear();
152     const CString utf8BlockedString = settings().blockedString().utf8();
153     m_replacementData.append(utf8BlockedString.data(), utf8BlockedString.length());
154 }
155
156 } // namespace WebCore
157
158 #endif // ENABLE(CONTENT_FILTERING)