chrome/browser/net/chrome_net_log.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/net/chrome_net_log.h"
   6
   7 #include <stdio.h>
   8
   9 #include "base/command_line.h"
  10 #include "base/files/scoped_file.h"
  11 #include "base/logging.h"
  12 #include "base/strings/string_number_conversions.h"
  13 #include "base/strings/string_util.h"
  14 #include "base/values.h"
  15 #include "chrome/browser/net/net_log_temp_file.h"
  16 #include "chrome/browser/ui/webui/net_internals/net_internals_ui.h"
  17 #include "chrome/common/chrome_switches.h"
  18 #include "content/public/common/content_switches.h"
  19 #include "net/log/net_log_logger.h"
  20 #include "net/log/trace_net_log_observer.h"
  21
  22 ChromeNetLog::ChromeNetLog()
  23     : net_log_temp_file_(new NetLogTempFile(this)) {
  24   const base::CommandLine* command_line =
  25       base::CommandLine::ForCurrentProcess();
  26
  27   if (command_line->HasSwitch(switches::kLogNetLog)) {
  28     base::FilePath log_path =
  29         command_line->GetSwitchValuePath(switches::kLogNetLog);
  30     // Much like logging.h, bypass threading restrictions by using fopen
  31     // directly.  Have to write on a thread that's shutdown to handle events on
  32     // shutdown properly, and posting events to another thread as they occur
  33     // would result in an unbounded buffer size, so not much can be gained by
  34     // doing this on another thread.  It's only used when debugging Chrome, so
  35     // performance is not a big concern.
  36     base::ScopedFILE file;
  37 #if defined(OS_WIN)
  38     file.reset(_wfopen(log_path.value().c_str(), L"w"));
  39 #elif defined(OS_POSIX)
  40     file.reset(fopen(log_path.value().c_str(), "w"));
  41 #endif
  42
  43     if (!file) {
  44       LOG(ERROR) << "Could not open file " << log_path.value()
  45                  << " for net logging";
  46     } else {
  47       scoped_ptr<base::Value> constants(NetInternalsUI::GetConstants());
  48       net_log_logger_.reset(new net::NetLogLogger());
  49       if (command_line->HasSwitch(switches::kNetLogLevel)) {
  50         std::string log_level_string =
  51             command_line->GetSwitchValueASCII(switches::kNetLogLevel);
  52         int command_line_log_level;
  53         if (base::StringToInt(log_level_string, &command_line_log_level) &&
  54             command_line_log_level >= LOG_ALL &&
  55             command_line_log_level <= LOG_NONE) {
  56           net_log_logger_->set_log_level(
  57               static_cast<LogLevel>(command_line_log_level));
  58         }
  59       }
  60       net_log_logger_->StartObserving(this, file.Pass(), constants.get(),
  61                                       nullptr);
  62     }
  63   }
  64
  65   trace_net_log_observer_.reset(new net::TraceNetLogObserver());
  66   trace_net_log_observer_->WatchForTraceStart(this);
  67 }
  68
  69 ChromeNetLog::~ChromeNetLog() {
  70   net_log_temp_file_.reset();
  71   // Remove the observers we own before we're destroyed.
  72   if (net_log_logger_)
  73     net_log_logger_->StopObserving(nullptr);
  74   if (trace_net_log_observer_)
  75     trace_net_log_observer_->StopWatchForTraceStart();
  76 }
  77