Teach the mojo_shell --trace-startup flag to gather data from services

shell/tracer.h

 // Copyright 2015 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 SHELL_TRACER_H_
 #define SHELL_TRACER_H_
 
 #include <stdio.h>
 
 #include <string>
 
 #include "base/callback.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted_memory.h"
+#include "mojo/common/data_pipe_drainer.h"
+#include "mojo/common/trace_controller_impl.h"
+#include "services/tracing/tracing.mojom.h"
 
 namespace shell {
 
 // Tracer collects tracing data from base/trace_event and from externally
 // configured sources, aggregates it into a single stream, and writes it out to
 // a file. It should be constructed very early in a process' lifetime before any
 // initialization that may be interesting to trace has occured and be shut down
 // as late as possible to capture as much initialization/shutdown code as
 // possible.
-class Tracer {
+class Tracer : public mojo::common::DataPipeDrainer::Client {
  public:
   Tracer();
-  ~Tracer();
+  ~Tracer() override;
 
   // Starts tracing the current process with the given set of categories.
   void Start(const std::string& categories);
 
+  // Starts collecting data from the tracing service with the given set of
+  // categories.
+  void StartCollectingFromTracingService(
+      tracing::TraceCoordinatorPtr coordinator);
+
   // Stops tracing and flushes all collected trace data to the given filename.
   // Blocks until the file write is complete. May be called after the message
   // loop is shut down.
   void StopAndFlushToFile(const std::string& filename);
 
+  void ConnectToController(
+      mojo::InterfaceRequest<tracing::TraceController> request);
+
  private:
   void StopTracingAndFlushToDisk(const std::string& filename);
 
   // Called from the flush thread. When all data is collected this runs
   // |done_callback| on the flush thread.
   void EndTraceAndFlush(const std::string& filename,
                         const base::Closure& done_callback);
 
   // Called from the flush thread.
   void WriteTraceDataCollected(
       const base::Closure& done_callback,
       const scoped_refptr<base::RefCountedString>& events_str,
       bool has_more_events);
 
-  // Whether we're currently tracing. Main thread only.
+  // mojo::common::DataPipeDrainer::Client implementation.
+  void OnDataAvailable(const void* data, size_t num_bytes) override;
+  void OnDataComplete() override;
+
+  // Emits a comma if needed.
+  void WriteCommaIfNeeded();
+
+  // Writes trace file footer and closes out the file.
+  void WriteFooterAndClose();
+
+  // Set when connected to the tracing service.
+  tracing::TraceCoordinatorPtr coordinator_;
+  scoped_ptr<mojo::common::DataPipeDrainer> drainer_;
+
+  // Whether we're currently tracing.
   bool tracing_;
 
-  // Whether we've written the first chunk. Flush thread only.
+  // Whether we've written the first chunk.
   bool first_chunk_written_;
+  std::string trace_service_data_;
 
-  // Trace file, if open. Flush thread only.
+  // Trace file, if open.
   FILE* trace_file_;
+  std::string trace_filename_;
 
   DISALLOW_COPY_AND_ASSIGN(Tracer);
 };
 
 }  // namespace shell
 
 #endif  // SHELL_TRACER_H_