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
|
// Copyright 2016 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 "components/data_use_measurement/core/data_use_network_delegate.h"
#include <memory>
#include <utility>
#include "components/data_use_measurement/core/data_use_ascriber.h"
#include "components/data_use_measurement/core/url_request_classifier.h"
#include "net/url_request/url_request.h"
namespace data_use_measurement {
DataUseNetworkDelegate::DataUseNetworkDelegate(
std::unique_ptr<NetworkDelegate> nested_network_delegate,
DataUseAscriber* ascriber,
std::unique_ptr<URLRequestClassifier> url_request_classifier,
const metrics::UpdateUsagePrefCallbackType& metrics_data_use_forwarder)
: net::LayeredNetworkDelegate(std::move(nested_network_delegate)),
ascriber_(ascriber),
data_use_measurement_(std::move(url_request_classifier),
metrics_data_use_forwarder,
ascriber_) {
DCHECK(ascriber);
}
DataUseNetworkDelegate::~DataUseNetworkDelegate() {}
void DataUseNetworkDelegate::OnBeforeURLRequestInternal(
net::URLRequest* request,
const net::CompletionCallback& callback,
GURL* new_url) {
ascriber_->OnBeforeUrlRequest(request);
data_use_measurement_.OnBeforeURLRequest(request);
}
void DataUseNetworkDelegate::OnBeforeRedirectInternal(
net::URLRequest* request,
const GURL& new_location) {
data_use_measurement_.OnBeforeRedirect(*request, new_location);
}
void DataUseNetworkDelegate::OnHeadersReceivedInternal(
net::URLRequest* request,
const net::CompletionCallback& callback,
const net::HttpResponseHeaders* original_response_headers,
scoped_refptr<net::HttpResponseHeaders>* override_response_headers,
GURL* allowed_unsafe_redirect_url) {
data_use_measurement_.OnHeadersReceived(request, original_response_headers);
}
void DataUseNetworkDelegate::OnNetworkBytesReceivedInternal(
net::URLRequest* request,
int64_t bytes_received) {
ascriber_->OnNetworkBytesReceived(request, bytes_received);
data_use_measurement_.OnNetworkBytesReceived(*request, bytes_received);
}
void DataUseNetworkDelegate::OnNetworkBytesSentInternal(
net::URLRequest* request,
int64_t bytes_sent) {
ascriber_->OnNetworkBytesSent(request, bytes_sent);
data_use_measurement_.OnNetworkBytesSent(*request, bytes_sent);
}
void DataUseNetworkDelegate::OnCompletedInternal(net::URLRequest* request,
bool started) {
ascriber_->OnUrlRequestCompleted(request, started);
data_use_measurement_.OnCompleted(*request, started);
}
void DataUseNetworkDelegate::OnURLRequestDestroyedInternal(
net::URLRequest* request) {
ascriber_->OnUrlRequestDestroyed(request);
}
} // namespace data_use_measurement
|
