Refactor WebRtcLoggingHandlerHost in preparation of automatic upload of WebRtcEventLogs.

chrome/browser/media/webrtc_text_log_handler.h

+// Copyright 2016 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.
+
+#ifndef CHROME_BROWSER_MEDIA_WEBRTC_TEXT_LOG_HANDLER_H_
+#define CHROME_BROWSER_MEDIA_WEBRTC_TEXT_LOG_HANDLER_H_
+
+#include <map>
+#include <memory>
+#include <string>
+
+#include "base/callback.h"
+#include "base/threading/thread_checker.h"
+#include "chrome/common/media/webrtc_logging_message_data.h"
+#include "chrome/common/partial_circular_buffer.h"
+#include "net/base/network_interfaces.h"
+
+#if defined(OS_ANDROID)
+const size_t kWebRtcLogSize = 1 * 1024 * 1024;  // 1 MB
+#else
+const size_t kWebRtcLogSize = 6 * 1024 * 1024;  // 6 MB
+#endif
+
+typedef std::map<std::string, std::string> MetaDataMap;
+
+class WebRtcLogUploader;
+
+class WebRtcLogBuffer {
+ public:
+  WebRtcLogBuffer();
+  ~WebRtcLogBuffer();
+
+  void Log(const std::string& message);
+
+  // Returns a circular buffer instance for reading the internal log buffer.
+  // Must only be called after the log has been marked as complete
+  // (see SetComplete) and the caller must ensure that the WebRtcLogBuffer
+  // instance remains in scope for the lifetime of the returned circular buffer.
+  PartialCircularBuffer Read();
+
+  // Switches the buffer to read-only mode, where access to the internal
+  // buffer is allowed from different threads than were used to contribute
+  // to the log.  Calls to Log() won't be allowed after calling
+  // SetComplete() and the call to SetComplete() must be done on the same
+  // thread as constructed the buffer and calls Log().
+  void SetComplete();
+
+ private:
+  base::ThreadChecker thread_checker_;
+  uint8_t buffer_[kWebRtcLogSize];
+  PartialCircularBuffer circular_;
+  bool read_only_;
+};
+
+class WebRtcTextLogHandler
+    : public base::RefCountedThreadSafe<WebRtcTextLogHandler> {
+ public:
+  // States used for protecting from function calls made at non-allowed points
+  // in time. For example, StartLogging() is only allowed in CLOSED state.
+  // Transitions: SetMetaData():    CLOSED -> CLOSED.
+  //              StartLogging():   CLOSED -> STARTING.
+  //              StartDone():      STARTING -> STARTED.
+  //              StopLogging():    STARTED -> STOPPING.
+  //              StopDone():       STOPPING -> STOPPED.
+  //              DiscardLog():     STOPPED -> CLOSED or
+  //                                CHANNEL_CLOSING -> CHANNEL_CLOSING.
+  //              ReleaseLog():     STOPPED -> CLOSED. or
+  //                                CHANNEL_CLOSING -> CHANNEL_CLOSING.
+  //              ChannelClosing(): ANY -> CHANNEL_CLOSING.
+  enum LoggingState {
+    CLOSED,           // Logging not started, no log in memory.
+    STARTING,         // Start logging is in progress.
+    STARTED,          // Logging started.
+    STOPPING,         // Stop logging is in progress.
+    STOPPED,          // Logging has been stopped, log still open in memory.
+    CHANNEL_CLOSING,  // Renderer is closing. The log (if there is one) can
+                      // still be uploaded, but no new logs can be created and
+                      // and no state transitions can be made.
+  };
+
+  typedef base::Callback<void(bool, const std::string&)> GenericDoneCallback;
+
+  explicit WebRtcTextLogHandler(int render_process_id);
+
+  // Returns the current state of the log. Must be called on the IO thread.
+  LoggingState GetState() const { return logging_state_; }
+
+  void SetMetaData(std::unique_ptr<MetaDataMap> meta_data,
+                   const GenericDoneCallback& callback);
+
+  // Opens a log and starts logging if allowed by the LogUploader.
+  // Returns false if logging could not be started.
+  // Must be called on the IO thread.
+  bool StartLogging(WebRtcLogUploader* log_uploader,
+                    const GenericDoneCallback& callback);
+
+  // Stops logging. Log will remain open until UploadLog or DiscardLog is
+  // called. Must be called on the IO thread.
+  bool StopLogging(const GenericDoneCallback& callback);
+
+  // Called by the WebRtcLoggingHandlerHost when logging has stopped in the
+  // renderer. Should only be called in response to a
+  // WebRtcLoggingMsg_LoggingStopped IPC message.
+  // Must be called on the IO thread.
+  void StopDone();
+
+  // Signals that the renderer is closing, which de facto stops logging but
+  // keeps the log in memory.
+  // Can be called in any state except CLOSED. Must be called on the IO thread.
+  void ChannelClosing();
+
+  // Discards a stopped log. Must be called on the IO thread.
+  void DiscardLog();
+
+  // Releases a stopped log to the caller. Must be called on the IO thread.
+  void ReleaseLog(std::unique_ptr<WebRtcLogBuffer>* log_buffer,
+                  std::unique_ptr<MetaDataMap>* meta_data);
+
+  // Adds a message to the log. Must be called on the IO thread.
+  void LogMessage(const std::string& message);
+
+  // Adds a message to the log. Must be called on the IO thread.
+  void LogWebRtcLoggingMessageData(const WebRtcLoggingMessageData& message);
+
+  // Returns true if the logging state is CLOSED and fires an the callback
+  // with an error message otherwise. Must be called on the IO thread.
+  bool ExpectLoggingStateStopped(const GenericDoneCallback& callback);
+
+  void FireGenericDoneCallback(const GenericDoneCallback& callback,
+                               bool success,
+                               const std::string& error_message);
+
+ private:
+  friend class RefCountedThreadSafe<WebRtcTextLogHandler>;
+  ~WebRtcTextLogHandler();
+
+  void StartDone(const GenericDoneCallback& callback);
+
+  void LogToCircularBuffer(const std::string& message);
+
+  void LogInitialInfoOnFileThread(const GenericDoneCallback& callback);
+  void LogInitialInfoOnIOThread(const net::NetworkInterfaceList& network_list,
+                                const GenericDoneCallback& callback);
+  void EnableBrowserProcessLoggingOnUIThread();
+  void DisableBrowserProcessLoggingOnUIThread();
+
+  // The render process ID this object belongs to.
+  const int render_process_id_;
+
+  std::unique_ptr<WebRtcLogBuffer> log_buffer_;
+
+  // These are only accessed on the IO thread, except when in STARTING state. In
+  // this state we are protected since entering any function that alters the
+  // state is not allowed.
+  std::unique_ptr<MetaDataMap> meta_data_;
+
+  // These are only accessed on the IO thread.
+  GenericDoneCallback stop_callback_;
+
+  // Only accessed on the IO thread.
+  LoggingState logging_state_;
+
+  // The system time in ms when logging is started. Reset when logging_state_
+  // changes to STOPPED.
+  base::Time logging_started_time_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebRtcTextLogHandler);
+};
+
+#endif  // CHROME_BROWSER_MEDIA_WEBRTC_TEXT_LOG_HANDLER_H_