mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-28 08:16:09 +03:00
198 lines
7.5 KiB
C++
198 lines
7.5 KiB
C++
// Copyright (c) 2012 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 "net/url_request/url_request_context.h"
|
|
|
|
#include <inttypes.h>
|
|
|
|
#include "base/compiler_specific.h"
|
|
#include "base/debug/alias.h"
|
|
#include "base/memory/ptr_util.h"
|
|
#include "base/metrics/histogram_macros.h"
|
|
#include "base/strings/string_util.h"
|
|
#include "base/strings/stringprintf.h"
|
|
#include "base/threading/thread_task_runner_handle.h"
|
|
#include "base/trace_event/memory_allocator_dump.h"
|
|
#include "base/trace_event/memory_dump_manager.h"
|
|
#include "base/trace_event/memory_dump_request_args.h"
|
|
#include "base/trace_event/process_memory_dump.h"
|
|
#include "net/cookies/cookie_store.h"
|
|
#include "net/dns/host_resolver.h"
|
|
#include "net/http/http_cache.h"
|
|
#include "net/http/http_transaction_factory.h"
|
|
#include "net/socket/ssl_client_socket_impl.h"
|
|
#include "net/url_request/http_user_agent_settings.h"
|
|
#include "net/url_request/url_request.h"
|
|
|
|
namespace net {
|
|
|
|
URLRequestContext::URLRequestContext()
|
|
: net_log_(nullptr),
|
|
host_resolver_(nullptr),
|
|
cert_verifier_(nullptr),
|
|
channel_id_service_(nullptr),
|
|
http_auth_handler_factory_(nullptr),
|
|
proxy_service_(nullptr),
|
|
network_delegate_(nullptr),
|
|
http_server_properties_(nullptr),
|
|
http_user_agent_settings_(nullptr),
|
|
cookie_store_(nullptr),
|
|
transport_security_state_(nullptr),
|
|
cert_transparency_verifier_(nullptr),
|
|
ct_policy_enforcer_(nullptr),
|
|
http_transaction_factory_(nullptr),
|
|
job_factory_(nullptr),
|
|
throttler_manager_(nullptr),
|
|
network_quality_estimator_(nullptr),
|
|
reporting_service_(nullptr),
|
|
network_error_logging_delegate_(nullptr),
|
|
enable_brotli_(false),
|
|
check_cleartext_permitted_(false),
|
|
name_("unknown"),
|
|
largest_outstanding_requests_count_seen_(0) {
|
|
base::trace_event::MemoryDumpManager::GetInstance()->RegisterDumpProvider(
|
|
this, "URLRequestContext", base::ThreadTaskRunnerHandle::Get());
|
|
}
|
|
|
|
URLRequestContext::~URLRequestContext() {
|
|
DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
|
|
AssertNoURLRequests();
|
|
base::trace_event::MemoryDumpManager::GetInstance()->UnregisterDumpProvider(
|
|
this);
|
|
}
|
|
|
|
void URLRequestContext::CopyFrom(const URLRequestContext* other) {
|
|
// Copy URLRequestContext parameters.
|
|
set_net_log(other->net_log_);
|
|
set_host_resolver(other->host_resolver_);
|
|
set_cert_verifier(other->cert_verifier_);
|
|
set_channel_id_service(other->channel_id_service_);
|
|
set_http_auth_handler_factory(other->http_auth_handler_factory_);
|
|
set_proxy_service(other->proxy_service_);
|
|
set_ssl_config_service(other->ssl_config_service_.get());
|
|
set_network_delegate(other->network_delegate_);
|
|
set_http_server_properties(other->http_server_properties_);
|
|
set_cookie_store(other->cookie_store_);
|
|
set_transport_security_state(other->transport_security_state_);
|
|
set_cert_transparency_verifier(other->cert_transparency_verifier_);
|
|
set_ct_policy_enforcer(other->ct_policy_enforcer_);
|
|
set_http_transaction_factory(other->http_transaction_factory_);
|
|
set_job_factory(other->job_factory_);
|
|
set_throttler_manager(other->throttler_manager_);
|
|
set_http_user_agent_settings(other->http_user_agent_settings_);
|
|
set_network_quality_estimator(other->network_quality_estimator_);
|
|
set_reporting_service(other->reporting_service_);
|
|
set_network_error_logging_delegate(other->network_error_logging_delegate_);
|
|
set_enable_brotli(other->enable_brotli_);
|
|
set_check_cleartext_permitted(other->check_cleartext_permitted_);
|
|
}
|
|
|
|
const HttpNetworkSession::Params* URLRequestContext::GetNetworkSessionParams(
|
|
) const {
|
|
HttpTransactionFactory* transaction_factory = http_transaction_factory();
|
|
if (!transaction_factory)
|
|
return nullptr;
|
|
HttpNetworkSession* network_session = transaction_factory->GetSession();
|
|
if (!network_session)
|
|
return nullptr;
|
|
return &network_session->params();
|
|
}
|
|
|
|
const HttpNetworkSession::Context* URLRequestContext::GetNetworkSessionContext()
|
|
const {
|
|
HttpTransactionFactory* transaction_factory = http_transaction_factory();
|
|
if (!transaction_factory)
|
|
return nullptr;
|
|
HttpNetworkSession* network_session = transaction_factory->GetSession();
|
|
if (!network_session)
|
|
return nullptr;
|
|
return &network_session->context();
|
|
}
|
|
|
|
std::unique_ptr<URLRequest> URLRequestContext::CreateRequest(
|
|
const GURL& url,
|
|
RequestPriority priority,
|
|
URLRequest::Delegate* delegate) const {
|
|
return CreateRequest(url, priority, delegate, MISSING_TRAFFIC_ANNOTATION);
|
|
}
|
|
|
|
std::unique_ptr<URLRequest> URLRequestContext::CreateRequest(
|
|
const GURL& url,
|
|
RequestPriority priority,
|
|
URLRequest::Delegate* delegate,
|
|
NetworkTrafficAnnotationTag traffic_annotation) const {
|
|
return base::WrapUnique(new URLRequest(
|
|
url, priority, delegate, this, network_delegate_, traffic_annotation));
|
|
}
|
|
|
|
void URLRequestContext::set_cookie_store(CookieStore* cookie_store) {
|
|
cookie_store_ = cookie_store;
|
|
}
|
|
|
|
void URLRequestContext::InsertURLRequest(const URLRequest* request) const {
|
|
url_requests_.insert(request);
|
|
if (url_requests_.size() > largest_outstanding_requests_count_seen_) {
|
|
largest_outstanding_requests_count_seen_ = url_requests_.size();
|
|
UMA_HISTOGRAM_COUNTS_1M("Net.URLRequestContext.OutstandingRequests",
|
|
largest_outstanding_requests_count_seen_);
|
|
UMA_HISTOGRAM_SPARSE_SLOWLY(
|
|
"Net.URLRequestContext.OutstandingRequests.Type",
|
|
request->traffic_annotation().unique_id_hash_code);
|
|
}
|
|
}
|
|
|
|
void URLRequestContext::RemoveURLRequest(const URLRequest* request) const {
|
|
DCHECK_EQ(1u, url_requests_.count(request));
|
|
url_requests_.erase(request);
|
|
}
|
|
|
|
void URLRequestContext::AssertNoURLRequests() const {
|
|
int num_requests = url_requests_.size();
|
|
if (num_requests != 0) {
|
|
// We're leaking URLRequests :( Dump the URL of the first one and record how
|
|
// many we leaked so we have an idea of how bad it is.
|
|
char url_buf[128];
|
|
const URLRequest* request = *url_requests_.begin();
|
|
base::strlcpy(url_buf, request->url().spec().c_str(), arraysize(url_buf));
|
|
int load_flags = request->load_flags();
|
|
base::debug::Alias(url_buf);
|
|
base::debug::Alias(&num_requests);
|
|
base::debug::Alias(&load_flags);
|
|
CHECK(false) << "Leaked " << num_requests << " URLRequest(s). First URL: "
|
|
<< request->url().spec().c_str() << ".";
|
|
}
|
|
}
|
|
|
|
void URLRequestContext::AssertURLRequestPresent(
|
|
const URLRequest* request) const {
|
|
CHECK_GE(url_requests_.count(request), 0u);
|
|
}
|
|
|
|
bool URLRequestContext::OnMemoryDump(
|
|
const base::trace_event::MemoryDumpArgs& args,
|
|
base::trace_event::ProcessMemoryDump* pmd) {
|
|
SSLClientSocketImpl::DumpSSLClientSessionMemoryStats(pmd);
|
|
|
|
std::string dump_name =
|
|
base::StringPrintf("net/url_request_context/%s/0x%" PRIxPTR,
|
|
name_.c_str(), reinterpret_cast<uintptr_t>(this));
|
|
base::trace_event::MemoryAllocatorDump* dump =
|
|
pmd->CreateAllocatorDump(dump_name);
|
|
dump->AddScalar(base::trace_event::MemoryAllocatorDump::kNameObjectCount,
|
|
base::trace_event::MemoryAllocatorDump::kUnitsObjects,
|
|
url_requests_.size());
|
|
HttpTransactionFactory* transaction_factory = http_transaction_factory();
|
|
if (transaction_factory) {
|
|
HttpNetworkSession* network_session = transaction_factory->GetSession();
|
|
if (network_session)
|
|
network_session->DumpMemoryStats(pmd, dump->absolute_name());
|
|
HttpCache* http_cache = transaction_factory->GetCache();
|
|
if (http_cache)
|
|
http_cache->DumpMemoryStats(pmd, dump->absolute_name());
|
|
}
|
|
return true;
|
|
}
|
|
|
|
} // namespace net
|