Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(320)

Side by Side Diff: components/tracing/trace_config_file.cc

Issue 1309243004: Reland again [Startup Tracing] Add --trace-config-file flag (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: fix unit test Created 5 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 // Copyright (c) 2015 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 "components/tracing/trace_config_file.h"
6
7 #include <string>
8
9 #include "base/command_line.h"
10 #include "base/files/file_util.h"
11 #include "base/json/json_reader.h"
12 #include "base/json/json_writer.h"
13 #include "base/logging.h"
14 #include "base/memory/singleton.h"
15 #include "base/values.h"
16 #include "components/tracing/tracing_switches.h"
17
18 namespace tracing {
19
20 namespace {
21
22 // Maximum trace config file size that will be loaded, in bytes.
23 const size_t kTraceConfigFileSizeLimit = 64 * 1024;
24
25 // Trace config file path:
26 // - Android: /data/local/chrome-trace-config.json
27 // - Others: specified by --trace-config-file flag.
28 #if defined(OS_ANDROID)
29 const base::FilePath::CharType kAndroidTraceConfigFile[] =
30 FILE_PATH_LITERAL("/data/local/chrome-trace-config.json");
31 #endif
32
33 const base::FilePath::CharType kDefaultResultFile[] =
34 FILE_PATH_LITERAL("chrometrace.log");
35
36 // String parameters that can be used to parse the trace config file content.
37 const char kTraceConfigParam[] = "trace_config";
38 const char kStartupDurationParam[] = "startup_duration";
39 const char kResultFileParam[] = "result_file";
40
41 } // namespace
42
43 TraceConfigFile* TraceConfigFile::GetInstance() {
44 return Singleton<TraceConfigFile,
45 DefaultSingletonTraits<TraceConfigFile>>::get();
46 }
47
48 TraceConfigFile::TraceConfigFile()
49 : is_enabled_(false),
50 trace_config_(base::trace_event::TraceConfig()),
51 startup_duration_(0),
52 result_file_(kDefaultResultFile) {
53 #if defined(OS_ANDROID)
54 base::FilePath trace_config_file(kAndroidTraceConfigFile);
55 #else
56 const base::CommandLine& command_line =
57 *base::CommandLine::ForCurrentProcess();
58 if (!command_line.HasSwitch(switches::kTraceConfigFile) ||
59 command_line.HasSwitch(switches::kTraceStartup) ||
60 command_line.HasSwitch(switches::kTraceShutdown)) {
61 return;
62 }
63 base::FilePath trace_config_file =
64 command_line.GetSwitchValuePath(switches::kTraceConfigFile);
65 #endif
66
67 if (trace_config_file.empty()) {
68 // If the trace config file path is not specified, trace Chrome with the
69 // default configuration for 5 sec.
70 startup_duration_ = 5;
71 is_enabled_ = true;
72 return;
73 }
74
75 if (!base::PathExists(trace_config_file))
76 return;
77
78 std::string trace_config_file_content;
79 if (!base::ReadFileToString(trace_config_file,
80 &trace_config_file_content,
81 kTraceConfigFileSizeLimit)) {
82 return;
83 }
84 is_enabled_ = ParseTraceConfigFileContent(trace_config_file_content);
85 }
86
87 TraceConfigFile::~TraceConfigFile() {
88 }
89
90 bool TraceConfigFile::ParseTraceConfigFileContent(std::string content) {
91 scoped_ptr<base::Value> value(base::JSONReader::Read(content));
92 if (!value || !value->IsType(base::Value::TYPE_DICTIONARY))
93 return false;
94
95 scoped_ptr<base::DictionaryValue> dict(
96 static_cast<base::DictionaryValue*>(value.release()));
97
98 base::DictionaryValue* trace_config_dict = NULL;
99 if (!dict->GetDictionary(kTraceConfigParam, &trace_config_dict))
100 return false;
101
102 std::string trace_config_str;
103 base::JSONWriter::Write(*trace_config_dict, &trace_config_str);
104 trace_config_ = base::trace_event::TraceConfig(trace_config_str);
105
106 if (!dict->GetInteger(kStartupDurationParam, &startup_duration_))
107 startup_duration_ = 0;
108
109 if (startup_duration_ < 0)
110 startup_duration_ = 0;
111
112 std::string result_file_str;
113 if (dict->GetString(kResultFileParam, &result_file_str))
114 result_file_ = base::FilePath().AppendASCII(result_file_str);
115
116 return true;
117 }
118
119 bool TraceConfigFile::IsEnabled() const {
120 return is_enabled_;
121 }
122
123 base::trace_event::TraceConfig TraceConfigFile::GetTraceConfig() const {
124 DCHECK(IsEnabled());
125 return trace_config_;
126 }
127
128 int TraceConfigFile::GetStartupDuration() const {
129 DCHECK(IsEnabled());
130 return startup_duration_;
131 }
132
133 #if !defined(OS_ANDROID)
134 base::FilePath TraceConfigFile::GetResultFile() const {
135 DCHECK(IsEnabled());
136 return result_file_;
137 }
138 #endif
139
140 } // namespace tracing
OLDNEW
« no previous file with comments | « components/tracing/trace_config_file.h ('k') | components/tracing/trace_config_file_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698