OLD | NEW |
| (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 | |
OLD | NEW |