1 // Copyright (c) 2010 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 "base/vlog.h"
6
7 #include "base/basictypes.h"
8 #include "base/logging.h"
9 #include "base/string_number_conversions.h"
10 #include "base/string_split.h"
11
12 namespace logging {
13
14 const int VlogInfo::kDefaultVlogLevel = 0;
15
16 struct VlogInfo::VmodulePattern {
17 enum MatchTarget { MATCH_MODULE, MATCH_FILE };
18
19 explicit VmodulePattern(const std::string& pattern);
20
21 VmodulePattern();
22
23 std::string pattern;
24 int vlog_level;
25 MatchTarget match_target;
26 };
27
VmodulePattern(const std::string & pattern)28 VlogInfo::VmodulePattern::VmodulePattern(const std::string& pattern)
29 : pattern(pattern),
30 vlog_level(VlogInfo::kDefaultVlogLevel),
31 match_target(MATCH_MODULE) {
32 // If the pattern contains a {forward,back} slash, we assume that
33 // it's meant to be tested against the entire __FILE__ string.
34 std::string::size_type first_slash = pattern.find_first_of("\\/");
35 if (first_slash != std::string::npos)
36 match_target = MATCH_FILE;
37 }
38
VmodulePattern()39 VlogInfo::VmodulePattern::VmodulePattern()
40 : vlog_level(VlogInfo::kDefaultVlogLevel),
41 match_target(MATCH_MODULE) {}
42
VlogInfo(const std::string & v_switch,const std::string & vmodule_switch,int * min_log_level)43 VlogInfo::VlogInfo(const std::string& v_switch,
44 const std::string& vmodule_switch,
45 int* min_log_level)
46 : min_log_level_(min_log_level) {
47 DCHECK(min_log_level != NULL);
48
49 typedef std::pair<std::string, std::string> KVPair;
50 int vlog_level = 0;
51 if (!v_switch.empty()) {
52 if (base::StringToInt(v_switch, &vlog_level)) {
53 SetMaxVlogLevel(vlog_level);
54 } else {
55 LOG(WARNING) << "Could not parse v switch \"" << v_switch << "\"";
56 }
57 }
58
59 std::vector<KVPair> kv_pairs;
60 if (!base::SplitStringIntoKeyValuePairs(
61 vmodule_switch, '=', ',', &kv_pairs)) {
62 LOG(WARNING) << "Could not fully parse vmodule switch \""
63 << vmodule_switch << "\"";
64 }
65 for (std::vector<KVPair>::const_iterator it = kv_pairs.begin();
66 it != kv_pairs.end(); ++it) {
67 VmodulePattern pattern(it->first);
68 if (!base::StringToInt(it->second, &pattern.vlog_level)) {
69 LOG(WARNING) << "Parsed vlog level for \""
70 << it->first << "=" << it->second
71 << "\" as " << pattern.vlog_level;
72 }
73 vmodule_levels_.push_back(pattern);
74 }
75 }
76
~VlogInfo()77 VlogInfo::~VlogInfo() {}
78
79 namespace {
80
81 // Given a path, returns the basename with the extension chopped off
82 // (and any -inl suffix). We avoid using FilePath to minimize the
83 // number of dependencies the logging system has.
GetModule(const base::StringPiece & file)84 base::StringPiece GetModule(const base::StringPiece& file) {
85 base::StringPiece module(file);
86 base::StringPiece::size_type last_slash_pos =
87 module.find_last_of("\\/");
88 if (last_slash_pos != base::StringPiece::npos)
89 module.remove_prefix(last_slash_pos + 1);
90 base::StringPiece::size_type extension_start = module.rfind('.');
91 module = module.substr(0, extension_start);
92 static const char kInlSuffix[] = "-inl";
93 static const int kInlSuffixLen = arraysize(kInlSuffix) - 1;
94 if (module.ends_with(kInlSuffix))
95 module.remove_suffix(kInlSuffixLen);
96 return module;
97 }
98
99 } // namespace
100
GetVlogLevel(const base::StringPiece & file) const101 int VlogInfo::GetVlogLevel(const base::StringPiece& file) const {
102 if (!vmodule_levels_.empty()) {
103 base::StringPiece module(GetModule(file));
104 for (std::vector<VmodulePattern>::const_iterator it =
105 vmodule_levels_.begin(); it != vmodule_levels_.end(); ++it) {
106 base::StringPiece target(
107 (it->match_target == VmodulePattern::MATCH_FILE) ? file : module);
108 if (MatchVlogPattern(target, it->pattern))
109 return it->vlog_level;
110 }
111 }
112 return GetMaxVlogLevel();
113 }
114
SetMaxVlogLevel(int level)115 void VlogInfo::SetMaxVlogLevel(int level) {
116 // Log severity is the negative verbosity.
117 *min_log_level_ = -level;
118 }
119
GetMaxVlogLevel() const120 int VlogInfo::GetMaxVlogLevel() const {
121 return -*min_log_level_;
122 }
123
MatchVlogPattern(const base::StringPiece & string,const base::StringPiece & vlog_pattern)124 bool MatchVlogPattern(const base::StringPiece& string,
125 const base::StringPiece& vlog_pattern) {
126 base::StringPiece p(vlog_pattern);
127 base::StringPiece s(string);
128 // Consume characters until the next star.
129 while (!p.empty() && !s.empty() && (p[0] != '*')) {
130 switch (p[0]) {
131 // A slash (forward or back) must match a slash (forward or back).
132 case '/':
133 case '\\':
134 if ((s[0] != '/') && (s[0] != '\\'))
135 return false;
136 break;
137
138 // A '?' matches anything.
139 case '?':
140 break;
141
142 // Anything else must match literally.
143 default:
144 if (p[0] != s[0])
145 return false;
146 break;
147 }
148 p.remove_prefix(1), s.remove_prefix(1);
149 }
150
151 // An empty pattern here matches only an empty string.
152 if (p.empty())
153 return s.empty();
154
155 // Coalesce runs of consecutive stars. There should be at least
156 // one.
157 while (!p.empty() && (p[0] == '*'))
158 p.remove_prefix(1);
159
160 // Since we moved past the stars, an empty pattern here matches
161 // anything.
162 if (p.empty())
163 return true;
164
165 // Since we moved past the stars and p is non-empty, if some
166 // non-empty substring of s matches p, then we ourselves match.
167 while (!s.empty()) {
168 if (MatchVlogPattern(s, p))
169 return true;
170 s.remove_prefix(1);
171 }
172
173 // Otherwise, we couldn't find a match.
174 return false;
175 }
176
177 } // namespace
178