First cut at UI for saving net_logs data into a temporary file on

chrome/browser/net/net_log_temp_file.cc

+// 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 "chrome/browser/net/net_log_temp_file.h"
+
+#include <stdio.h>
+
+#include "base/bind.h"
+#include "base/file_util.h"
+#include "base/json/json_writer.h"
+#include "base/logging.h"
+#include "base/message_loop.h"
+#include "base/string_number_conversions.h"
+#include "base/string_util.h"
+#include "base/threading/platform_thread.h"
+#include "base/threading/thread_restrictions.h"
+#include "base/utf_string_conversions.h"
+#include "base/values.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/net/chrome_net_log.h"
+#include "chrome/browser/net/net_log_logger.h"
+#include "chrome/browser/ui/webui/net_internals/net_internals_ui.h"
+#include "content/public/browser/browser_thread.h"
+
+#if defined(OS_ANDROID)
+#include "chrome/browser/android/intent_helper.h"
+#endif
+
+using content::BrowserThread;
+
+NetLogTempFile* NetLogTempFile::GetInstance() {
+  return Singleton<NetLogTempFile>::get();
+}
+
+NetLogTempFile::NetLogTempFile() : state_(UNINITIALIZED) {
+  DetermineInitialState();
+}
+
+NetLogTempFile::~NetLogTempFile() {
+}
+
+void NetLogTempFile::ProcessCommand(Command command) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+  switch (command) {
+    case DO_START:
+      StartNetLog();
+      break;
+    case DO_STOP:
+      StopNetLog();
+      break;
+    case DO_SEND:
+      SendNetLog();
+      break;
+    default:
+      break;
+  }
+}
+
+Value* NetLogTempFile::NetLogTempFileToValue() const {
+  base::DictionaryValue* dict = new base::DictionaryValue;
+  switch (state_) {
+    case STOP:
+      dict->SetString("file", log_path_.MaybeAsASCII());
+      dict->SetString("state", "STOP");
+      break;
+    case START_SEND:
+      dict->SetString("file", log_path_.MaybeAsASCII());
+      dict->SetString("state", "START_SEND");
+      break;
+    default:
+      dict->SetString("file", "");
+      dict->SetString("state", "START");
+      break;
+  }
+  return dict;
+}
+
+bool GetTempNetLogDir(FilePath* path) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+
+  base::ThreadRestrictions::ScopedAllowIO allow_io;
+  FilePath temp_dir;
+  if (!file_util::GetTempDir(&temp_dir))
+    return false;
+
+  FilePath net_log_dir = temp_dir.Append(FILE_PATH_LITERAL("NetLog"));
+  if (file_util::PathExists(net_log_dir)) {
+    *path = net_log_dir;
+    return true;
+  }
+
+  if (!file_util::CreateDirectory(net_log_dir))
+    return false;
+
+  *path = net_log_dir;
+  return true;
+}
+
+void NetLogTempFile::DetermineInitialState() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+
+  base::ThreadRestrictions::ScopedAllowIO allow_io;
+  FilePath net_log_dir;
+  if (!GetTempNetLogDir(&net_log_dir)) {
+    state_ = START;
+    return;
+  }
+
+  if (file_util::IsDirectoryEmpty(net_log_dir))
+    state_ = START;
+  else
+    state_ = START_SEND;
+}
+
+void NetLogTempFile::StartNetLog() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+
+  if (!g_browser_process->net_log())
+    return;
+
+  if (state_ == STOP)
+    return;
+
+  base::ThreadRestrictions::ScopedAllowIO allow_io;
+  FilePath net_log_dir;
+  if (!GetTempNetLogDir(&net_log_dir))
+    return;
+
+  file_util::FileEnumerator file_iter(
+      net_log_dir, false, file_util::FileEnumerator::FILES);
+  for (FilePath current = file_iter.Next(); !current.empty();
+       current = file_iter.Next()) {
+    file_util::Delete(current, false);
+  }
+
+  if (!file_util::CreateTemporaryFileInDir(net_log_dir, &log_path_))
+    return;
+
+  net_log_logger_.reset(new NetLogLogger(log_path_));
+  net_log_logger_->StartObserving(g_browser_process->net_log());
+  state_ = STOP;
+}
+
+void NetLogTempFile::StopNetLog() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+
+  if (!g_browser_process->net_log())
+    return;
+
+  if (state_ != STOP)
+    return;
+
+  net_log_logger_->StopObserving(g_browser_process->net_log());
+  net_log_logger_.reset();
+  state_ = START_SEND;
+}
+
+void NetLogTempFile::SendNetLog() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+
+  if (state_ != START_SEND)
+    return;
+
+  base::ThreadRestrictions::ScopedAllowIO allow_io;
+  FilePath net_log_dir;
+  if (!GetTempNetLogDir(&net_log_dir))
+    return;
+
+  FilePath file_to_send;
+  file_util::FileEnumerator file_iter(
+      net_log_dir, false, file_util::FileEnumerator::FILES);
+  for (FilePath current = file_iter.Next(); !current.empty();
+       current = file_iter.Next()) {
+    file_to_send = current;
+    break;
+  }
+  if (file_to_send.empty())
+    return;
+
+#if defined(OS_POSIX)
+  // Users, group and others can read, write and traverse.
+  int mode = file_util::FILE_PERMISSION_MASK;
+  file_util::SetPosixFilePermissions(file_to_send, mode);
+#endif  // defined(OS_POSIX)
+
+  BrowserThread::PostTask(
+      BrowserThread::UI,
+      FROM_HERE,
+      base::Bind(&NetLogTempFile::SendEmail, file_to_send));
+}
+
+// static
+void NetLogTempFile::SendEmail(const FilePath& file_to_send) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+
+#if defined(OS_ANDROID)
+  std::string email;
+  std::string subject = "net_internals_log";
+  std::string title = "test_title";
+  std::string body = "Net Internals log data";
+  FilePath::StringType file_to_attach(file_to_send.value());
+  chrome::android::SendEmail(
+      UTF8ToUTF16(email), UTF8ToUTF16(subject),
+      UTF8ToUTF16(body), UTF8ToUTF16(title), UTF8ToUTF16(file_to_attach));
+#endif
+}