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