headless: Add support for net logging

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/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
+  base::DictionaryValue* dict = new base::DictionaryValue();
+
+  dict->SetString("name", "headless");
+  dict->SetString(
+      "command_line",
+      base::CommandLine::ForCurrentProcess()->GetCommandLineString());
+
+  constants_dict->Set("clientInfo", dict);
+
+  return 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