#ifndef ContentFilter_h
#define ContentFilter_h
#if ENABLE(CONTENT_FILTERING)
#include "CachedResourceHandle.h"
#include <functional>
#include <wtf/Vector.h>
namespace WebCore {
class CachedRawResource;
class ContentFilterUnblockHandler;
class DocumentLoader;
class PlatformContentFilter;
class ResourceRequest;
class ResourceResponse;
class SharedBuffer;
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(DocumentLoader&);
~ContentFilter();
static const char* urlScheme() { return "x-apple-content-filter"; }
void startFilteringMainResource(CachedRawResource&);
void stopFilteringMainResource();
bool continueAfterWillSendRequest(ResourceRequest&, const ResourceResponse&);
bool continueAfterResponseReceived(CachedResource*, const ResourceResponse&);
bool continueAfterDataReceived(CachedResource*, const char* data, int length);
bool continueAfterNotifyFinished(CachedResource*);
ContentFilterUnblockHandler unblockHandler() const;
Ref<SharedBuffer> replacementData() const;
String unblockRequestDeniedScript() const;
private:
enum class State {
Stopped,
Filtering,
Allowed,
Blocked,
};
struct Type {
const std::function<std::unique_ptr<PlatformContentFilter>()> create;
};
template <typename T> static Type type();
WEBCORE_EXPORT static Vector<Type>& types();
using Container = Vector<std::unique_ptr<PlatformContentFilter>>;
friend std::unique_ptr<ContentFilter> std::make_unique<ContentFilter>(Container&&, DocumentLoader&);
ContentFilter(Container, DocumentLoader&);
void forEachContentFilterUntilBlocked(std::function<void(PlatformContentFilter&)>);
void didDecide(State);
void deliverResourceData(CachedResource&);
const Container m_contentFilters;
DocumentLoader& m_documentLoader;
CachedResourceHandle<CachedRawResource> m_mainResource;
PlatformContentFilter* m_blockingContentFilter { nullptr };
State m_state { State::Stopped };
};
template <typename T>
ContentFilter::Type ContentFilter::type()
{
static_assert(std::is_base_of<PlatformContentFilter, T>::value, "Type must be a PlatformContentFilter.");
return { T::create };
}
}
#endif // ENABLE(CONTENT_FILTERING)
#endif // ContentFilter_h