Store NetLog in BrowserClient

This commit is contained in:
Cheng Zhao 2015-08-11 18:29:55 +08:00
parent edda9d0488
commit 8a92ef34a7
8 changed files with 46 additions and 79 deletions

View file

@ -63,7 +63,9 @@ net::URLRequestContextGetter* BrowserClient::CreateRequestContext(
content::ProtocolHandlerMap* protocol_handlers, content::ProtocolHandlerMap* protocol_handlers,
content::URLRequestInterceptorScopedVector protocol_interceptors) { content::URLRequestInterceptorScopedVector protocol_interceptors) {
auto context = static_cast<BrowserContext*>(browser_context); auto context = static_cast<BrowserContext*>(browser_context);
return context->CreateRequestContext(protocol_handlers, protocol_interceptors.Pass()); return context->CreateRequestContext(static_cast<NetLog*>(GetNetLog()),
protocol_handlers,
protocol_interceptors.Pass());
} }
content::MediaObserver* BrowserClient::GetMediaObserver() { content::MediaObserver* BrowserClient::GetMediaObserver() {
@ -81,7 +83,7 @@ void BrowserClient::GetAdditionalAllowedSchemesForFileSystem(
} }
net::NetLog* BrowserClient::GetNetLog() { net::NetLog* BrowserClient::GetNetLog() {
return browser_context()->GetNetLog(); return &net_log_;
} }
base::FilePath BrowserClient::GetDefaultDownloadDirectory() { base::FilePath BrowserClient::GetDefaultDownloadDirectory() {

View file

@ -5,12 +5,14 @@
#ifndef BRIGHTRAY_BROWSER_BROWSER_CLIENT_H_ #ifndef BRIGHTRAY_BROWSER_BROWSER_CLIENT_H_
#define BRIGHTRAY_BROWSER_BROWSER_CLIENT_H_ #define BRIGHTRAY_BROWSER_BROWSER_CLIENT_H_
#include "browser/net_log.h"
#include "content/public/browser/content_browser_client.h" #include "content/public/browser/content_browser_client.h"
namespace brightray { namespace brightray {
class BrowserContext; class BrowserContext;
class BrowserMainParts; class BrowserMainParts;
class NetLog;
class BrowserClient : public content::ContentBrowserClient { class BrowserClient : public content::ContentBrowserClient {
public: public:
@ -35,8 +37,6 @@ class BrowserClient : public content::ContentBrowserClient {
content::BrowserContext* browser_context, content::BrowserContext* browser_context,
content::ProtocolHandlerMap* protocol_handlers, content::ProtocolHandlerMap* protocol_handlers,
content::URLRequestInterceptorScopedVector protocol_interceptors) override; content::URLRequestInterceptorScopedVector protocol_interceptors) override;
private:
content::BrowserMainParts* CreateBrowserMainParts( content::BrowserMainParts* CreateBrowserMainParts(
const content::MainFunctionParams&) override; const content::MainFunctionParams&) override;
content::MediaObserver* GetMediaObserver() override; content::MediaObserver* GetMediaObserver() override;
@ -46,7 +46,6 @@ class BrowserClient : public content::ContentBrowserClient {
net::NetLog* GetNetLog() override; net::NetLog* GetNetLog() override;
base::FilePath GetDefaultDownloadDirectory() override; base::FilePath GetDefaultDownloadDirectory() override;
content::DevToolsManagerDelegate* GetDevToolsManagerDelegate() override; content::DevToolsManagerDelegate* GetDevToolsManagerDelegate() override;
#if defined(OS_POSIX) && !defined(OS_MACOSX) #if defined(OS_POSIX) && !defined(OS_MACOSX)
void GetAdditionalMappedFilesForChildProcess( void GetAdditionalMappedFilesForChildProcess(
const base::CommandLine& command_line, const base::CommandLine& command_line,
@ -61,6 +60,8 @@ class BrowserClient : public content::ContentBrowserClient {
base::ScopedFD v8_snapshot_fd_; base::ScopedFD v8_snapshot_fd_;
#endif // OS_POSIX && !OS_MACOSX #endif // OS_POSIX && !OS_MACOSX
NetLog net_log_;
DISALLOW_COPY_AND_ASSIGN(BrowserClient); DISALLOW_COPY_AND_ASSIGN(BrowserClient);
}; };

View file

@ -100,11 +100,13 @@ void BrowserContext::RegisterInternalPrefs(PrefRegistrySimple* registry) {
} }
net::URLRequestContextGetter* BrowserContext::CreateRequestContext( net::URLRequestContextGetter* BrowserContext::CreateRequestContext(
NetLog* net_log,
content::ProtocolHandlerMap* protocol_handlers, content::ProtocolHandlerMap* protocol_handlers,
content::URLRequestInterceptorScopedVector protocol_interceptors) { content::URLRequestInterceptorScopedVector protocol_interceptors) {
DCHECK(!url_request_getter_.get()); DCHECK(!url_request_getter_.get());
url_request_getter_ = new URLRequestContextGetter( url_request_getter_ = new URLRequestContextGetter(
this, this,
net_log,
GetPath(), GetPath(),
BrowserThread::UnsafeGetMessageLoopForThread(BrowserThread::IO), BrowserThread::UnsafeGetMessageLoopForThread(BrowserThread::IO),
BrowserThread::UnsafeGetMessageLoopForThread(BrowserThread::FILE), BrowserThread::UnsafeGetMessageLoopForThread(BrowserThread::FILE),
@ -114,10 +116,6 @@ net::URLRequestContextGetter* BrowserContext::CreateRequestContext(
return url_request_getter_.get(); return url_request_getter_.get();
} }
net::NetLog* BrowserContext::GetNetLog() {
return url_request_getter_->net_log();
}
net::NetworkDelegate* BrowserContext::CreateNetworkDelegate() { net::NetworkDelegate* BrowserContext::CreateNetworkDelegate() {
return new NetworkDelegate; return new NetworkDelegate;
} }

View file

@ -46,9 +46,9 @@ class BrowserContext : public content::BrowserContext,
content::PermissionManager* GetPermissionManager() override; content::PermissionManager* GetPermissionManager() override;
net::URLRequestContextGetter* CreateRequestContext( net::URLRequestContextGetter* CreateRequestContext(
NetLog* net_log,
content::ProtocolHandlerMap* protocol_handlers, content::ProtocolHandlerMap* protocol_handlers,
content::URLRequestInterceptorScopedVector protocol_interceptors); content::URLRequestInterceptorScopedVector protocol_interceptors);
net::NetLog* GetNetLog();
net::URLRequestContextGetter* url_request_context_getter() const { net::URLRequestContextGetter* url_request_context_getter() const {
return url_request_getter_.get(); return url_request_getter_.get();

View file

@ -4,15 +4,13 @@
#include "browser/net_log.h" #include "browser/net_log.h"
#include "browser/browser_context.h"
#include "base/command_line.h" #include "base/command_line.h"
#include "base/files/file_util.h" #include "base/files/file_path.h"
#include "base/json/json_writer.h" #include "base/values.h"
#include "base/logging.h"
#include "content/public/common/content_switches.h" #include "content/public/common/content_switches.h"
#include "net/log/net_log_util.h" #include "net/log/net_log_util.h"
#include "net/log/net_log_capture_mode.h"
#include "net/url_request/url_request_context.h" namespace brightray {
namespace { namespace {
@ -31,16 +29,12 @@ scoped_ptr<base::DictionaryValue> GetConstants() {
} // namespace } // namespace
namespace brightray { NetLog::NetLog() {
NetLog::NetLog(net::URLRequestContext* context)
: added_events_(false),
context_(context) {
auto command_line = base::CommandLine::ForCurrentProcess(); auto command_line = base::CommandLine::ForCurrentProcess();
if (command_line->HasSwitch(switches::kLogNetLog)) { if (!command_line->HasSwitch(switches::kLogNetLog))
base::FilePath log_path = return;
command_line->GetSwitchValuePath(switches::kLogNetLog);
base::FilePath log_path = command_line->GetSwitchValuePath(switches::kLogNetLog);
#if defined(OS_WIN) #if defined(OS_WIN)
log_file_.reset(_wfopen(log_path.value().c_str(), L"w")); log_file_.reset(_wfopen(log_path.value().c_str(), L"w"));
#elif defined(OS_POSIX) #elif defined(OS_POSIX)
@ -50,43 +44,19 @@ NetLog::NetLog(net::URLRequestContext* context)
if (!log_file_) { if (!log_file_) {
LOG(ERROR) << "Could not open file: " << log_path.value() LOG(ERROR) << "Could not open file: " << log_path.value()
<< "for net logging"; << "for net logging";
} else { return;
std::string json;
scoped_ptr<base::Value> constants(GetConstants().Pass());
base::JSONWriter::Write(constants.get(), &json);
fprintf(log_file_.get(), "{\"constants\": %s, \n", json.c_str());
fprintf(log_file_.get(), "\"events\": [\n");
if (context_) {
DCHECK(context_->CalledOnValidThread());
std::set<net::URLRequestContext*> contexts;
contexts.insert(context_);
net::CreateNetLogEntriesForActiveObjects(contexts, this);
}
DeprecatedAddObserver(this, net::NetLogCaptureMode::Default());
}
} }
} }
NetLog::~NetLog() { NetLog::~NetLog() {
if (log_file_) {
DeprecatedRemoveObserver(this);
// Ending events array.
fprintf(log_file_.get(), "]}");
log_file_.reset();
}
} }
void NetLog::OnAddEntry(const net::NetLog::Entry& entry) { void NetLog::StartLogging(net::URLRequestContext* url_request_context) {
std::string json; if (!log_file_)
base::JSONWriter::Write(entry.ToValue(), &json); return;
fprintf(log_file_.get(), "%s%s", (added_events_ ? ",\n" : ""), json.c_str()); scoped_ptr<base::Value> constants(GetConstants());
added_events_ = true; write_to_file_observer_.StartObserving(this, log_file_.Pass(), constants.get(), url_request_context);
} }
} // namespace brightray } // namespace brightray

View file

@ -5,29 +5,22 @@
#ifndef BROWSER_NET_LOG_H_ #ifndef BROWSER_NET_LOG_H_
#define BROWSER_NET_LOG_H_ #define BROWSER_NET_LOG_H_
#include "base/files/file_path.h"
#include "base/files/scoped_file.h" #include "base/files/scoped_file.h"
#include "net/log/net_log.h" #include "net/log/net_log.h"
#include "net/log/write_to_file_net_log_observer.h"
namespace net {
class URLRequestContext;
}
namespace brightray { namespace brightray {
class NetLog : public net::NetLog, class NetLog : public net::NetLog {
public net::NetLog::ThreadSafeObserver {
public: public:
explicit NetLog(net::URLRequestContext* context); NetLog();
~NetLog() override; ~NetLog() override;
void OnAddEntry(const net::NetLog::Entry& entry) override; void StartLogging(net::URLRequestContext* url_request_context);
private: private:
bool added_events_;
// We use raw pointer to prevent reference cycle.
net::URLRequestContext* const context_;
base::ScopedFILE log_file_; base::ScopedFILE log_file_;
net::WriteToFileNetLogObserver write_to_file_observer_;
DISALLOW_COPY_AND_ASSIGN(NetLog); DISALLOW_COPY_AND_ASSIGN(NetLog);
}; };

View file

@ -119,12 +119,14 @@ URLRequestContextGetter::Delegate::CreateHttpCacheBackendFactory(const base::Fil
URLRequestContextGetter::URLRequestContextGetter( URLRequestContextGetter::URLRequestContextGetter(
Delegate* delegate, Delegate* delegate,
NetLog* net_log,
const base::FilePath& base_path, const base::FilePath& base_path,
base::MessageLoop* io_loop, base::MessageLoop* io_loop,
base::MessageLoop* file_loop, base::MessageLoop* file_loop,
content::ProtocolHandlerMap* protocol_handlers, content::ProtocolHandlerMap* protocol_handlers,
content::URLRequestInterceptorScopedVector protocol_interceptors) content::URLRequestInterceptorScopedVector protocol_interceptors)
: delegate_(delegate), : delegate_(delegate),
net_log_(net_log),
base_path_(base_path), base_path_(base_path),
io_loop_(io_loop), io_loop_(io_loop),
file_loop_(file_loop), file_loop_(file_loop),
@ -157,8 +159,8 @@ net::URLRequestContext* URLRequestContextGetter::GetURLRequestContext() {
url_request_context_.reset(new net::URLRequestContext); url_request_context_.reset(new net::URLRequestContext);
// --log-net-log // --log-net-log
net_log_.reset(new NetLog(url_request_context_.get())); net_log_->StartLogging(url_request_context_.get());
url_request_context_->set_net_log(net_log_.get()); url_request_context_->set_net_log(net_log_);
network_delegate_.reset(delegate_->CreateNetworkDelegate()); network_delegate_.reset(delegate_->CreateNetworkDelegate());
url_request_context_->set_network_delegate(network_delegate_.get()); url_request_context_->set_network_delegate(network_delegate_.get());

View file

@ -17,7 +17,6 @@ class MessageLoop;
} }
namespace net { namespace net {
class NetLog;
class HostMappingRules; class HostMappingRules;
class HostResolver; class HostResolver;
class NetworkDelegate; class NetworkDelegate;
@ -28,6 +27,8 @@ class URLRequestJobFactory;
namespace brightray { namespace brightray {
class NetLog;
class URLRequestContextGetter : public net::URLRequestContextGetter { class URLRequestContextGetter : public net::URLRequestContextGetter {
public: public:
class Delegate { class Delegate {
@ -46,6 +47,7 @@ class URLRequestContextGetter : public net::URLRequestContextGetter {
URLRequestContextGetter( URLRequestContextGetter(
Delegate* delegate, Delegate* delegate,
NetLog* net_log,
const base::FilePath& base_path, const base::FilePath& base_path,
base::MessageLoop* io_loop, base::MessageLoop* io_loop,
base::MessageLoop* file_loop, base::MessageLoop* file_loop,
@ -58,17 +60,16 @@ class URLRequestContextGetter : public net::URLRequestContextGetter {
scoped_refptr<base::SingleThreadTaskRunner> GetNetworkTaskRunner() const override; scoped_refptr<base::SingleThreadTaskRunner> GetNetworkTaskRunner() const override;
net::HostResolver* host_resolver(); net::HostResolver* host_resolver();
net::NetLog* net_log() { return net_log_.get(); }
private: private:
Delegate* delegate_; Delegate* delegate_;
NetLog* net_log_;
base::FilePath base_path_; base::FilePath base_path_;
base::MessageLoop* io_loop_; base::MessageLoop* io_loop_;
base::MessageLoop* file_loop_; base::MessageLoop* file_loop_;
scoped_ptr<net::ProxyConfigService> proxy_config_service_; scoped_ptr<net::ProxyConfigService> proxy_config_service_;
scoped_ptr<net::NetLog> net_log_;
scoped_ptr<net::NetworkDelegate> network_delegate_; scoped_ptr<net::NetworkDelegate> network_delegate_;
scoped_ptr<net::URLRequestContextStorage> storage_; scoped_ptr<net::URLRequestContextStorage> storage_;
scoped_ptr<net::URLRequestContext> url_request_context_; scoped_ptr<net::URLRequestContext> url_request_context_;