Skeleton of the Safe Browsing Subresource Filter.

third_party/WebKit/Source/core/loader/DocumentLoader.h

 /*
  * Copyright (C) 2006, 2007, 2008, 2009 Apple Inc. All rights reserved.
  * Copyright (C) 2011 Google 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.
@@ skipping 40 matching lines @@
 #include "wtf/RefPtr.h"
 
 namespace blink {
 
 class ApplicationCacheHost;
 class ResourceFetcher;
 class DocumentInit;
 class LocalFrame;
 class FrameLoader;
 class ResourceLoader;
+class WebDocumentSubresourceFilter;
 
 class CORE_EXPORT DocumentLoader : public GarbageCollectedFinalized<DocumentLoader>, private RawResourceClient {
 public:
     static DocumentLoader* create(LocalFrame* frame, const ResourceRequest& request, const SubstituteData& data)
     {
         return new DocumentLoader(frame, request, data);
     }
     ~DocumentLoader() override;
 
     LocalFrame* frame() const { return m_frame; }
 
     virtual void detachFromFrame();
 
     unsigned long mainResourceIdentifier() const;
 
     void replaceDocumentWhileExecutingJavaScriptURL(const DocumentInit&, const String& source);
 
     const AtomicString& mimeType() const;
 
     const ResourceRequest& originalRequest() const;
 
     const ResourceRequest& request() const;
 
     ResourceFetcher* fetcher() const { return m_fetcher.get(); }
 
+    void setSubresourceFilter(PassOwnPtr<WebDocumentSubresourceFilter>);
+    WebDocumentSubresourceFilter* subresourceFilter() const { return m_subresourceFilter.get(); }
+
     const SubstituteData& substituteData() const { return m_substituteData; }
 
     const KURL& url() const;
     const KURL& unreachableURL() const;
     const KURL& urlForHistory() const;
 
     const AtomicString& responseMIMEType() const;
 
     void didChangePerformanceTiming();
     void didObserveLoadingBehavior(WebLoadingBehaviorFlag);
@@ skipping 77 matching lines @@
     String debugName() const override { return "DocumentLoader"; }
 
     bool maybeLoadEmpty();
 
     bool isRedirectAfterPost(const ResourceRequest&, const ResourceResponse&);
 
     bool shouldContinueForResponse() const;
 
     Member<LocalFrame> m_frame;
     Member<ResourceFetcher> m_fetcher;
+    OwnPtr<WebDocumentSubresourceFilter> m_subresourceFilter;
 
     Member<RawResource> m_mainResource;
 
     Member<DocumentWriter> m_writer;
 
     // A reference to actual request used to create the data source.
     // The only part of this request that should change is the url, and
     // that only in the case of a same-document navigation.
     ResourceRequest m_originalRequest;
 
@@ skipping 36 matching lines @@
     // Used to protect against reentrancy into dataReceived().
     bool m_inDataReceived;
     RefPtr<SharedBuffer> m_dataBuffer;
 };
 
 DECLARE_WEAK_IDENTIFIER_MAP(DocumentLoader);
 
 } // namespace blink
 
 #endif // DocumentLoader_h