Add the ability for DevTools to wrap network transactions.

chrome/browser/devtools/devtools_network_controller.cc

+// 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.
+
+#include "chrome/browser/devtools/devtools_network_controller.h"
+
+#include "chrome/browser/devtools/devtools_network_transaction.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/profiles/profile_io_data.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/resource_context.h"
+
+DevToolsNetworkController::DevToolsNetworkController() {
+}
+
+DevToolsNetworkController::~DevToolsNetworkController() {
+}
+
+void DevToolsNetworkController::AddTransaction(
+    uint64 transaction_id,
+    DevToolsNetworkTransaction* transaction) {
+  transactions_[transaction_id] = transaction;
+}
+
+void DevToolsNetworkController::RemoveTransaction(uint64 transaction_id) {
+  transactions_.erase(transaction_id);
+}
+
+// static
+void DevToolsNetworkController::DisableNetwork(
+    Profile* profile,
+    const std::string& client_id,
+    bool disable_network) {
+  CHECK(profile);
+  content::ResourceContext* resourceContext = profile->GetResourceContext();
+  CHECK(resourceContext);
+  content::BrowserThread::PostTask(
+      content::BrowserThread::IO,
+      FROM_HERE,
+      base::Bind(
+          &DevToolsNetworkController::DisableNetworkOnIO,
+          resourceContext,
+          client_id,
+          disable_network));
+}
+
+// static
+void DevToolsNetworkController::DisableNetworkOnIO(
+    content::ResourceContext* resourceContext,
+    const std::string& client_id,
+    bool disable_network) {
+  CHECK(resourceContext);
+  DevToolsNetworkController* controller = ProfileIOData::FromResourceContext(
+    resourceContext)->network_controller();
+  CHECK(controller);
+  controller->DisableNetwork(client_id, disable_network);
+}
+
+void DevToolsNetworkController::DisableNetwork(
+    const std::string& client_id,
+    bool disable_network) {
+  ClientsSet::iterator client = clients_set_.find(client_id);
+  if (client != clients_set_.end()) {
+    if (!disable_network)
+      clients_set_.erase(client);
+    return;
+  }
+
+  if (!disable_network)
+    return;
+
+  bool first_client = clients_set_.size() == 0;
+  clients_set_.insert(client_id);
+  if (!first_client)
+    return;
+
+  Transactions::iterator it = transactions_.begin();
+  while (it != transactions_.end()) {
+    Transactions::iterator current = it++;
+    DevToolsNetworkTransaction* transaction = current->second;
+    if (IsBlockedURL(transaction->GetURL())) {
+      transactions_.erase(current);
+      transaction->Stop();
+    }
+  }
+}
+
+bool DevToolsNetworkController::IsBlockedURL(const GURL& url) {
+  if (clients_set_.size() == 0)
+    return false;
+
+  // TODO: Add whitelist.
+
+  return true;
+}