chrome/browser/devtools/devtools_network_controller.cc


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

// 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;
}