enabling network emulation in devtools

brightray/browser/net/devtools_network_transaction.h

@@ -0,0 +1,131 @@
+// 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 BROWSER_DEVTOOLS_NETWORK_TRANSACTION_H_
+#define BROWSER_DEVTOOLS_NETWORK_TRANSACTION_H_
+
+#include <stdint.h>
+
+#include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
+#include "net/base/completion_callback.h"
+#include "net/base/load_states.h"
+#include "net/base/request_priority.h"
+#include "net/http/http_transaction.h"
+#include "net/websockets/websocket_handshake_stream_base.h"
+
+namespace brightray {
+
+class DevToolsNetworkController;
+class DevToolsNetworkInterceptor;
+
+class DevToolsNetworkTransaction : public net::HttpTransaction {
+ public:
+  DevToolsNetworkTransaction(
+      DevToolsNetworkController* controller,
+      scoped_ptr<net::HttpTransaction> network_transaction);
+  ~DevToolsNetworkTransaction() override;
+
+  // Checks if request contains DevTools specific headers. Found values are
+  // remembered and corresponding keys are removed from headers.
+  void ProcessRequest();
+
+  // Runs callback with net::ERR_INTERNET_DISCONNECTED result.
+  void Fail();
+
+  void DecreaseThrottledByteCount(int64_t delta);
+  void FireThrottledCallback();
+
+  // HttpTransaction methods:
+  int Start(const net::HttpRequestInfo* request,
+            const net::CompletionCallback& callback,
+            const net::BoundNetLog& net_log) override;
+  int RestartIgnoringLastError(
+      const net::CompletionCallback& callback) override;
+  int RestartWithCertificate(net::X509Certificate* client_cert,
+                             const net::CompletionCallback& callback) override;
+  int RestartWithAuth(const net::AuthCredentials& credentials,
+                      const net::CompletionCallback& callback) override;
+  bool IsReadyToRestartForAuth() override;
+
+  int Read(net::IOBuffer* buf,
+           int buf_len,
+           const net::CompletionCallback& callback) override;
+  void StopCaching() override;
+  bool GetFullRequestHeaders(net::HttpRequestHeaders* headers) const override;
+  int64_t GetTotalReceivedBytes() const override;
+  void DoneReading() override;
+  const net::HttpResponseInfo* GetResponseInfo() const override;
+  net::LoadState GetLoadState() const override;
+  net::UploadProgress GetUploadProgress() const override;
+  void SetQuicServerInfo(net::QuicServerInfo* quic_server_info) override;
+  bool GetLoadTimingInfo(net::LoadTimingInfo* load_timing_info) const override;
+  void SetPriority(net::RequestPriority priority) override;
+  void SetWebSocketHandshakeStreamCreateHelper(
+      net::WebSocketHandshakeStreamBase::CreateHelper* create_helper) override;
+  void SetBeforeNetworkStartCallback(
+      const BeforeNetworkStartCallback& callback) override;
+  void SetBeforeProxyHeadersSentCallback(
+      const BeforeProxyHeadersSentCallback& callback) override;
+  int ResumeNetworkStart() override;
+  void GetConnectionAttempts(net::ConnectionAttempts* out) const override;
+
+  bool failed() const { return failed_; }
+
+  const net::HttpRequestInfo* request() const { return request_; }
+
+  int64_t throttled_byte_count() const { return throttled_byte_count_; }
+
+  const std::string& client_id() const {
+    return client_id_;
+  }
+
+ private:
+  enum CallbackType {
+      NONE,
+      READ,
+      RESTART_IGNORING_LAST_ERROR,
+      RESTART_WITH_AUTH,
+      RESTART_WITH_CERTIFICATE,
+      START
+  };
+
+  // Proxy callback handler. Runs saved callback.
+  void OnCallback(int result);
+
+  int SetupCallback(
+      net::CompletionCallback callback,
+      int result,
+      CallbackType callback_type);
+  void Throttle(int result);
+
+  DevToolsNetworkController* controller_;
+  base::WeakPtr<DevToolsNetworkInterceptor> interceptor_;
+
+  // Modified request. Should be destructed after |transaction_|
+  scoped_ptr<net::HttpRequestInfo> custom_request_;
+
+  // Original network transaction.
+  scoped_ptr<net::HttpTransaction> transaction_;
+
+  const net::HttpRequestInfo* request_;
+
+  // True if Fail was already invoked.
+  bool failed_;
+
+  // Value of "X-DevTools-Emulate-Network-Conditions-Client-Id" request header.
+  std::string client_id_;
+
+  int throttled_result_;
+  int64_t throttled_byte_count_;
+  CallbackType callback_type_;
+  net::CompletionCallback proxy_callback_;
+  net::CompletionCallback callback_;
+
+  DISALLOW_COPY_AND_ASSIGN(DevToolsNetworkTransaction);
+};
+
+}  // namespace brightray
+
+#endif  // BROWSER_DEVTOOLS_NETWORK_TRANSACTION_H_