11320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Copyright (c) 2014 The Chromium Authors. All rights reserved. 21320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Use of this source code is governed by a BSD-style license that can be 31320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// found in the LICENSE file. 41320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 51320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/test/trace_to_file.h" 61320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 71320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/base_switches.h" 81320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/command_line.h" 91320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/debug/trace_event_impl.h" 101320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_util.h" 111320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/run_loop.h" 121320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 131320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccinamespace base { 141320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccinamespace test { 151320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 161320f92c476a1ad9d19dba2a48c72b75566198e9Primiano TucciTraceToFile::TraceToFile() : started_(false) { 171320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 181320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 191320f92c476a1ad9d19dba2a48c72b75566198e9Primiano TucciTraceToFile::~TraceToFile() { 201320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci EndTracingIfNeeded(); 211320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 221320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 231320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid TraceToFile::BeginTracingFromCommandLineOptions() { 241320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK(CommandLine::InitializedForCurrentProcess()); 251320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK(!started_); 261320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 271320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kTraceToFile)) 281320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci return; 291320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 301320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // Empty filter (i.e. just --trace-to-file) turns into default categories in 311320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // TraceEventImpl 321320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci std::string filter = CommandLine::ForCurrentProcess()->GetSwitchValueASCII( 331320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci switches::kTraceToFile); 341320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 351320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci FilePath path; 361320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kTraceToFileName)) { 371320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci path = FilePath(CommandLine::ForCurrentProcess() 381320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci ->GetSwitchValuePath(switches::kTraceToFileName)); 391320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci } else { 401320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci path = FilePath(FILE_PATH_LITERAL("trace.json")); 411320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci } 421320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 431320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci BeginTracing(path, filter); 441320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 451320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 461320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid TraceToFile::BeginTracing(const FilePath& path, 471320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci const std::string& categories) { 481320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK(!started_); 491320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci started_ = true; 501320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci path_ = path; 511320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci WriteFileHeader(); 521320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 531320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci debug::TraceLog::GetInstance()->SetEnabled( 541320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci debug::CategoryFilter(categories), 551320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci debug::TraceLog::RECORDING_MODE, 561320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci debug::TraceOptions(debug::RECORD_UNTIL_FULL)); 571320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 581320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 591320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid TraceToFile::WriteFileHeader() { 601320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci const char str[] = "{\"traceEvents\": ["; 611320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci WriteFile(path_, str, static_cast<int>(strlen(str))); 621320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 631320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 641320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid TraceToFile::AppendFileFooter() { 651320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci const char str[] = "]}"; 661320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci AppendToFile(path_, str, static_cast<int>(strlen(str))); 671320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 681320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 691320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid TraceToFile::TraceOutputCallback(const std::string& data) { 701320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci int ret = AppendToFile(path_, data.c_str(), static_cast<int>(data.size())); 711320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK_NE(-1, ret); 721320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 731320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 741320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccistatic void OnTraceDataCollected( 751320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci Closure quit_closure, 761320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci debug::TraceResultBuffer* buffer, 771320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci const scoped_refptr<RefCountedString>& json_events_str, 781320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci bool has_more_events) { 791320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci buffer->AddFragment(json_events_str->data()); 801320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!has_more_events) 811320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci quit_closure.Run(); 821320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 831320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 841320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid TraceToFile::EndTracingIfNeeded() { 851320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!started_) 861320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci return; 871320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci started_ = false; 881320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 891320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci debug::TraceLog::GetInstance()->SetDisabled(); 901320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 911320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci debug::TraceResultBuffer buffer; 921320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci buffer.SetOutputCallback( 931320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci Bind(&TraceToFile::TraceOutputCallback, Unretained(this))); 941320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 951320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci RunLoop run_loop; 961320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci debug::TraceLog::GetInstance()->Flush( 971320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci Bind(&OnTraceDataCollected, run_loop.QuitClosure(), Unretained(&buffer))); 981320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci run_loop.Run(); 991320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1001320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci AppendFileFooter(); 1011320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 1021320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1031320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} // namespace test 1041320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} // namespace base 105