Home | History | Annotate | Download | only in feedback 
      1 // Copyright 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/feedback/tracing_manager.h"
      6 
      7 #include "base/bind.h"
      8 #include "base/file_util.h"
      9 #include "base/location.h"
     10 #include "base/memory/ref_counted_memory.h"
     11 #include "base/message_loop/message_loop_proxy.h"
     12 #include "base/prefs/pref_service.h"
     13 #include "chrome/browser/browser_process.h"
     14 #include "chrome/browser/feedback/feedback_util.h"
     15 #include "chrome/common/pref_names.h"
     16 #include "content/public/browser/tracing_controller.h"
     17 
     18 namespace {
     19 // Only once trace manager can exist at a time.
     20 TracingManager* g_tracing_manager = NULL;
     21 // Trace IDs start at 1 and increase.
     22 int g_next_trace_id = 1;
     23 // Name of the file to store the tracing data as.
     24 const base::FilePath::CharType kTracingFilename[] =
     25     FILE_PATH_LITERAL("tracing.json");
     26 }
     27 
     28 TracingManager::TracingManager()
     29     : current_trace_id_(0),
     30       weak_ptr_factory_(this) {
     31   DCHECK(!g_tracing_manager);
     32   g_tracing_manager = this;
     33   StartTracing();
     34 }
     35 
     36 TracingManager::~TracingManager() {
     37   DCHECK(g_tracing_manager == this);
     38   g_tracing_manager = NULL;
     39 }
     40 
     41 int TracingManager::RequestTrace() {
     42   // Return the current trace if one is being collected.
     43   if (current_trace_id_)
     44     return current_trace_id_;
     45 
     46   current_trace_id_ = g_next_trace_id;
     47   ++g_next_trace_id;
     48   content::TracingController::GetInstance()->DisableRecording(
     49       base::FilePath(),
     50       base::Bind(&TracingManager::OnTraceDataCollected,
     51                  weak_ptr_factory_.GetWeakPtr()));
     52   return current_trace_id_;
     53 }
     54 
     55 bool TracingManager::GetTraceData(int id, const TraceDataCallback& callback) {
     56   // If a trace is being collected currently, send it via callback when
     57   // complete.
     58   if (current_trace_id_) {
     59     // Only allow one trace data request at a time.
     60     if (trace_callback_.is_null()) {
     61       trace_callback_ = callback;
     62       return true;
     63     } else {
     64       return false;
     65     }
     66   } else {
     67     std::map<int, scoped_refptr<base::RefCountedString> >::iterator data =
     68         trace_data_.find(id);
     69     if (data == trace_data_.end())
     70       return false;
     71 
     72     // Always return the data asychronously, so the behavior is consistant.
     73     base::MessageLoopProxy::current()->PostTask(
     74         FROM_HERE,
     75         base::Bind(callback, data->second));
     76     return true;
     77   }
     78 }
     79 
     80 void TracingManager::DiscardTraceData(int id) {
     81   trace_data_.erase(id);
     82 
     83   // If the trace is discarded before it is complete, clean up the accumulators.
     84   if (id == current_trace_id_) {
     85     current_trace_id_ = 0;
     86 
     87     // If the trace has already been requested, provide an empty string.
     88     if (!trace_callback_.is_null()) {
     89       trace_callback_.Run(scoped_refptr<base::RefCountedString>());
     90       trace_callback_.Reset();
     91     }
     92   }
     93 }
     94 
     95 void TracingManager::StartTracing() {
     96   content::TracingController::GetInstance()->EnableRecording(
     97       "", content::TracingController::DEFAULT_OPTIONS,
     98       content::TracingController::EnableRecordingDoneCallback());
     99 }
    100 
    101 void TracingManager::OnTraceDataCollected(const base::FilePath& path) {
    102   if (!current_trace_id_)
    103     return;
    104 
    105   std::string data;
    106   if (!base::ReadFileToString(path, &data)) {
    107     LOG(ERROR) << "Failed to read trace data from: " << path.value();
    108     return;
    109   }
    110   base::DeleteFile(path, false);
    111 
    112   std::string output_val;
    113   feedback_util::ZipString(
    114       base::FilePath(kTracingFilename), data, &output_val);
    115 
    116   scoped_refptr<base::RefCountedString> output(
    117       base::RefCountedString::TakeString(&output_val));
    118 
    119   trace_data_[current_trace_id_] = output;
    120 
    121   if (!trace_callback_.is_null()) {
    122     trace_callback_.Run(output);
    123     trace_callback_.Reset();
    124   }
    125 
    126   current_trace_id_ = 0;
    127 
    128   // Tracing has to be restarted asynchronous, so the TracingController can
    129   // clean up.
    130   base::MessageLoopProxy::current()->PostTask(
    131       FROM_HERE,
    132       base::Bind(&TracingManager::StartTracing,
    133                  weak_ptr_factory_.GetWeakPtr()));
    134 }
    135 
    136 // static
    137 scoped_ptr<TracingManager> TracingManager::Create() {
    138   if (g_tracing_manager)
    139     return scoped_ptr<TracingManager>();
    140   return scoped_ptr<TracingManager>(new TracingManager());
    141 }
    142 
    143 TracingManager* TracingManager::Get() {
    144   return g_tracing_manager;
    145 }
    146