DevTools protocol interception, blocking & modification of requests

content/browser/devtools/protocol/network_handler.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CONTENT_BROWSER_DEVTOOLS_PROTOCOL_NETWORK_HANDLER_H_
 #define CONTENT_BROWSER_DEVTOOLS_PROTOCOL_NETWORK_HANDLER_H_
 
 #include <memory>
 
 #include "base/macros.h"
+#include "base/memory/weak_ptr.h"
+#include "base/synchronization/lock.h"
 #include "content/browser/devtools/protocol/devtools_domain_handler.h"
 #include "content/browser/devtools/protocol/network.h"
 #include "net/base/net_errors.h"
 #include "net/cookies/canonical_cookie.h"
 
+namespace net {
+class URLRequest;
+}  // namespace net
+
 namespace content {
 
 class DevToolsAgentHostImpl;
 class RenderFrameHostImpl;
 struct BeginNavigationParams;
 struct CommonNavigationParams;
 struct ResourceRequest;
 struct ResourceRequestCompletionStatus;
 struct ResourceResponseHead;
 
@@ skipping 32 matching lines @@
       Maybe<std::string> path,
       Maybe<bool> secure,
       Maybe<bool> http_only,
       Maybe<std::string> same_site,
       Maybe<double> expires,
       std::unique_ptr<SetCookieCallback> callback) override;
 
   Response SetUserAgentOverride(const std::string& user_agent) override;
   Response CanEmulateNetworkConditions(bool* result) override;
 
+  DispatchResponse EnableFetchInterception(bool enabled) override;
+  void AllowRequest(const String& request_id,
+                    std::unique_ptr<AllowRequestCallback> callback) override;
+  void BlockRequest(const String& request_id,
+                    const std::string& error_reason,
+                    std::unique_ptr<BlockRequestCallback> callback) override;
+  void ModifyRequest(const std::string& request_id,
+                     Maybe<std::string> url,
+                     Maybe<std::string> method,
+                     Maybe<std::string> post_data,
+                     Maybe<protocol::Network::Headers> headers,
+                     std::unique_ptr<ModifyRequestCallback> callback) override;
+  void MockResponse(const std::string& request_id,
+                    const std::string& raw_response,
+                    std::unique_ptr<MockResponseCallback> callback) override;
+
   void NavigationPreloadRequestSent(int worker_version_id,
                                     const std::string& request_id,
                                     const ResourceRequest& request);
   void NavigationPreloadResponseReceived(int worker_version_id,
                                          const std::string& request_id,
                                          const GURL& url,
                                          const ResourceResponseHead& head);
   void NavigationPreloadCompleted(
       const std::string& request_id,
       const ResourceRequestCompletionStatus& completion_status);
   void NavigationFailed(const CommonNavigationParams& common_params,
                         const BeginNavigationParams& begin_params,
                         net::Error error_code);
 
   bool enabled() const { return enabled_; }
   std::string UserAgentOverride() const;
 
+  Network::Frontend* frontend() const { return frontend_.get(); }
+
+  static std::unique_ptr<Network::Request> CreateRequestFromURLRequest(
+      const net::URLRequest* request);
+
  private:
   std::unique_ptr<Network::Frontend> frontend_;
   RenderFrameHostImpl* host_;
   bool enabled_;
+  bool interception_enabled_;
   std::string user_agent_;
+  base::WeakPtrFactory<NetworkHandler> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(NetworkHandler);
 };
 
 }  // namespace protocol
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_NETWORK_HANDLER_H_