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/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 DevToolsSession;
+class DevToolsURLInterceptorRequestJob;
 class RenderFrameHostImpl;
 struct BeginNavigationParams;
 struct CommonNavigationParams;
 struct ResourceRequest;
 struct ResourceRequestCompletionStatus;
 struct ResourceResponseHead;
 
 namespace protocol {
 
 class NetworkHandler : public DevToolsDomainHandler,
@@ skipping 28 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;
+  DispatchResponse AllowRequest(const String& intercept_id) override;
+  DispatchResponse BlockRequest(const String& intercept_id,
+                                const std::string& error_reason) override;
+  DispatchResponse ModifyRequest(
+      const String& intercept_id,
+      Maybe<String> url,
+      Maybe<String> method,
+      Maybe<String> post_data,
+      Maybe<protocol::Network::Headers> headers) override;
+  DispatchResponse MockResponse(const String& intercept_id,
+                                const String& raw_response) 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_; }
+  bool interception_enabled() const;
   std::string UserAgentOverride() const;
 
+  Network::Frontend* frontend() const { return frontend_.get(); }
+
+  void RegisterURLInterceptor(const std::string& intercept_id,
+                              DevToolsURLInterceptorRequestJob* interceptor);
+
+  void UnregisterURLInterceptor(const std::string& intercept_id,
+                                DevToolsURLInterceptorRequestJob* interceptor);
+
+  static std::unique_ptr<Network::Request> CreateRequestFromURLRequest(
+      const net::URLRequest* request);
+
  private:
   std::unique_ptr<Network::Frontend> frontend_;
   RenderFrameHostImpl* host_;
   bool enabled_;
   std::string user_agent_;
 
+  mutable base::Lock interception_lock_;  // Protects members below
+  std::unordered_map<std::string, DevToolsURLInterceptorRequestJob*>
+      interceptor_map_;
+  bool interception_enabled_;
+
   DISALLOW_COPY_AND_ASSIGN(NetworkHandler);
 };
 
 }  // namespace protocol
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_NETWORK_HANDLER_H_