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"
+
+using content::BrowserThread;
+
+namespace {
+
+const char kDevToolsRequestInitiator[] = "X-DevTools-Request-Initiator";
+
+}  // namespace
+
+DevToolsNetworkController::DevToolsNetworkController()
+    : weak_ptr_factory_(this) {
+}
+
+DevToolsNetworkController::~DevToolsNetworkController() {
+}
+
+void DevToolsNetworkController::AddTransaction(
+    DevToolsNetworkTransaction* transaction) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  transactions_.insert(transaction);
+}
+
+void DevToolsNetworkController::RemoveTransaction(
+    DevToolsNetworkTransaction* transaction) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(transactions_.find(transaction) != transactions_.end());
+  transactions_.erase(transaction);
+}
+
+void DevToolsNetworkController::SetNetworkState(
+    const std::string& client_id,
+    bool offline) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  BrowserThread::PostTask(
+      content::BrowserThread::IO,
+      FROM_HERE,
+      base::Bind(
+          &DevToolsNetworkController::SetNetworkStateOnIO,
+          weak_ptr_factory_.GetWeakPtr(),
+          client_id,
+          offline));
+}
+
+void DevToolsNetworkController::SetNetworkStateOnIO(
+    const std::string& client_id,
+    bool offline) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (!offline) {
+    clients_.erase(client_id);
+    return;
+  }
+  clients_.insert(client_id);
+
+  // Iterate over a copy of set, because failing of transaction could result in
+  // creating a new one, or (theoretically) destroying one.
+  Transactions old_transactions(transactions_);
+  for (Transactions::iterator it = old_transactions.begin();
+       it != old_transactions.end(); ++it) {
+    if (transactions_.find(*it) == transactions_.end())
+      continue;
+    if (!(*it)->request() || (*it)->failed())
+      continue;
+    if (ShouldFail((*it)->request()))
+      (*it)->Fail();
+  }
+}
+
+bool DevToolsNetworkController::ShouldFail(
+    const net::HttpRequestInfo* request) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(request);
+  if (clients_.empty())
+    return false;
+
+  if (request->extra_headers.HasHeader(kDevToolsRequestInitiator))
+    return false;
+
+  // TODO: Add domain blacklist.
+
+  return true;
+}