headless/lib/browser/headless_net_log.cc

// Copyright 2017 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 "headless/lib/browser/headless_net_log.h"

#include <stdio.h>

#include <utility>

#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/files/scoped_file.h"
#include "base/memory/ptr_util.h"
#include "base/values.h"
#include "build/build_config.h"
#include "content/public/common/content_switches.h"
#include "net/log/net_log_util.h"
#include "net/log/write_to_file_net_log_observer.h"

namespace headless {
namespace {

std::unique_ptr<base::Value> GetHeadlessConstants() {
  std::unique_ptr<base::DictionaryValue> constants_dict =
      net::GetNetConstants();

  // Add a dictionary with client information
  auto dict = base::MakeUnique<base::DictionaryValue>();

  dict->SetString("name", "headless");
  dict->SetString(
      "command_line",
      base::CommandLine::ForCurrentProcess()->GetCommandLineString());

  constants_dict->Set("clientInfo", std::move(dict));

  return std::move(constants_dict);
}

}  // namespace

HeadlessNetLog::HeadlessNetLog(const base::FilePath& log_path) {
  // TODO(mmenke):  Other than a different set of constants, this code is
  //     identical to code in ChromeNetLog.  Consider merging the code.

  // Much like logging.h, bypass threading restrictions by using fopen
  // directly.  Have to write on a thread that's shutdown to handle events on
  // shutdown properly, and posting events to another thread as they occur
  // would result in an unbounded buffer size, so not much can be gained by
  // doing this on another thread.  It's only used when debugging, so
  // performance is not a big concern.
  base::ScopedFILE file;
#if defined(OS_WIN)
  file.reset(_wfopen(log_path.value().c_str(), L"w"));
#elif defined(OS_POSIX)
  file.reset(fopen(log_path.value().c_str(), "w"));
#endif

  if (!file) {
    LOG(ERROR) << "Could not open file " << log_path.value()
               << " for net logging";
  } else {
    std::unique_ptr<base::Value> constants(GetHeadlessConstants());
    write_to_file_observer_.reset(new net::WriteToFileNetLogObserver());
    write_to_file_observer_->StartObserving(this, std::move(file),
                                            constants.get(), nullptr);
  }
}

HeadlessNetLog::~HeadlessNetLog() {
  // Remove the observer we own before we're destroyed.
  if (write_to_file_observer_)
    write_to_file_observer_->StopObserving(nullptr);
}

}  // namespace headless