106 lines
3.8 KiB
C++
106 lines
3.8 KiB
C++
/*
|
|
* Copyright (C) 2013-2016 Apple Inc. All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
|
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
|
|
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
|
|
* THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#if ENABLE(CONTENT_FILTERING)
|
|
|
|
#include "CachedResourceHandle.h"
|
|
#include "PlatformContentFilter.h"
|
|
#include "ResourceError.h"
|
|
#include <functional>
|
|
#include <wtf/Forward.h>
|
|
#include <wtf/UniqueRef.h>
|
|
|
|
namespace WebCore {
|
|
|
|
class CachedRawResource;
|
|
class ContentFilterClient;
|
|
class DocumentLoader;
|
|
class ResourceRequest;
|
|
class ResourceResponse;
|
|
class SubstituteData;
|
|
|
|
class ContentFilter {
|
|
WTF_MAKE_FAST_ALLOCATED;
|
|
WTF_MAKE_NONCOPYABLE(ContentFilter);
|
|
|
|
public:
|
|
template <typename T> static void addType() { types().append(type<T>()); }
|
|
|
|
static std::unique_ptr<ContentFilter> create(ContentFilterClient&);
|
|
~ContentFilter();
|
|
|
|
static const char* urlScheme() { return "x-apple-content-filter"; }
|
|
|
|
void startFilteringMainResource(CachedRawResource&);
|
|
void stopFilteringMainResource();
|
|
|
|
bool continueAfterWillSendRequest(ResourceRequest&, const ResourceResponse&);
|
|
bool continueAfterResponseReceived(const ResourceResponse&);
|
|
bool continueAfterDataReceived(const uint8_t* data, int length);
|
|
bool continueAfterNotifyFinished(CachedResource&);
|
|
|
|
static bool continueAfterSubstituteDataRequest(const DocumentLoader& activeLoader, const SubstituteData&);
|
|
bool willHandleProvisionalLoadFailure(const ResourceError&) const;
|
|
void handleProvisionalLoadFailure(const ResourceError&);
|
|
|
|
private:
|
|
using State = PlatformContentFilter::State;
|
|
|
|
struct Type {
|
|
Function<UniqueRef<PlatformContentFilter>()> create;
|
|
};
|
|
template <typename T> static Type type();
|
|
WEBCORE_EXPORT static Vector<Type>& types();
|
|
|
|
using Container = Vector<UniqueRef<PlatformContentFilter>>;
|
|
friend std::unique_ptr<ContentFilter> std::make_unique<ContentFilter>(Container&&, ContentFilterClient&);
|
|
ContentFilter(Container&&, ContentFilterClient&);
|
|
|
|
template <typename Function> void forEachContentFilterUntilBlocked(Function&&);
|
|
void didDecide(State);
|
|
void deliverResourceData(CachedResource&);
|
|
|
|
Container m_contentFilters;
|
|
ContentFilterClient& m_client;
|
|
CachedResourceHandle<CachedRawResource> m_mainResource;
|
|
const PlatformContentFilter* m_blockingContentFilter { nullptr };
|
|
State m_state { State::Stopped };
|
|
ResourceError m_blockedError;
|
|
bool m_isLoadingBlockedPage { false };
|
|
};
|
|
|
|
template <typename T>
|
|
ContentFilter::Type ContentFilter::type()
|
|
{
|
|
static_assert(std::is_base_of<PlatformContentFilter, T>::value, "Type must be a PlatformContentFilter.");
|
|
return { T::create };
|
|
}
|
|
|
|
} // namespace WebCore
|
|
|
|
#endif // ENABLE(CONTENT_FILTERING)
|