Tracing pre-filtering

base/trace_event/trace_config.cc

 // Copyright (c) 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/trace_event/trace_config.h"
 
 #include <stddef.h>
 
 #include <utility>
 
@@ skipping 33 matching lines @@
 
 const char kSyntheticDelayCategoryFilterPrefix[] = "DELAY(";
 
 // String parameters that is used to parse memory dump config in trace config
 // string.
 const char kMemoryDumpConfigParam[] = "memory_dump_config";
 const char kTriggersParam[] = "triggers";
 const char kPeriodicIntervalParam[] = "periodic_interval_ms";
 const char kModeParam[] = "mode";
 
+// String parameters used to parse category event filters.
+const char kCategoryEventFilterParam[] = "category_event_filters";
+const char kPredicateParam[] = "predicate";
+const char kArgsParam[] = "args";
+
 // Default configuration of memory dumps.
 const TraceConfig::MemoryDumpTriggerConfig kDefaultHeavyMemoryDumpTrigger = {
     2000,  // periodic_interval_ms
     MemoryDumpLevelOfDetail::DETAILED};
 const TraceConfig::MemoryDumpTriggerConfig kDefaultLightMemoryDumpTrigger = {
     250,  // periodic_interval_ms
     MemoryDumpLevelOfDetail::LIGHT};
 
 class ConvertableTraceConfigToTraceFormat
     : public base::trace_event::ConvertableToTraceFormat {
  public:
   explicit ConvertableTraceConfigToTraceFormat(const TraceConfig& trace_config)
       : trace_config_(trace_config) {}
   ~ConvertableTraceConfigToTraceFormat() override {}
   void AppendAsTraceFormat(std::string* out) const override {
     out->append(trace_config_.ToString());
   }
 
  private:
   const TraceConfig trace_config_;
 };
 
 }  // namespace
 
+TraceConfig::CategoryEventFilterConfig::CategoryEventFilterConfig() {}
+TraceConfig::CategoryEventFilterConfig::~CategoryEventFilterConfig() {}
+
+TraceConfig::CategoryEventFilterConfig::CategoryEventFilterConfig(
+    const CategoryEventFilterConfig& tc)
+    : predicate_name(tc.predicate_name),
+      included_categories(tc.included_categories),
+      excluded_categories(tc.excluded_categories) {
+  if (tc.args)
+    args = tc.args->CreateDeepCopy();
+}
+
+TraceConfig::CategoryEventFilterConfig& TraceConfig::CategoryEventFilterConfig::
+operator=(const TraceConfig::CategoryEventFilterConfig& rhs) {
+  if (this == &rhs)
+    return *this;
+
+  predicate_name = rhs.predicate_name;
+  included_categories = rhs.included_categories;
+  excluded_categories = rhs.excluded_categories;
+  if (rhs.args)
+    args = rhs.args->CreateDeepCopy();
+
+  return *this;
+}
+
+bool TraceConfig::CategoryEventFilterConfig::IsCategoryGroupEnabled(
+    const char* category_group_name) const {
+  CStringTokenizer category_group_tokens(
+      category_group_name, category_group_name + strlen(category_group_name),
+      ",");
+  while (category_group_tokens.GetNext()) {
+    std::string category_group_token = category_group_tokens.token();
+
+    for (const auto& excluded_category : excluded_categories) {
+      if (base::MatchPattern(category_group_token.c_str(),
+                             excluded_category.c_str())) {
+        return false;
+      }
+    }
+
+    for (const auto& included_category : included_categories) {
+      if (base::MatchPattern(category_group_token.c_str(),
+                             included_category.c_str())) {
+        return true;
+      }
+    }
+  }
+
+  return false;
+}
+
 TraceConfig::TraceConfig() {
   InitializeDefault();
 }
 
 TraceConfig::TraceConfig(const std::string& category_filter_string,
                          const std::string& trace_options_string) {
   InitializeFromStrings(category_filter_string, trace_options_string);
 }
 
 TraceConfig::TraceConfig(const std::string& category_filter_string,
@@ skipping 31 matching lines @@
 
 TraceConfig::TraceConfig(const TraceConfig& tc)
     : record_mode_(tc.record_mode_),
       enable_sampling_(tc.enable_sampling_),
       enable_systrace_(tc.enable_systrace_),
       enable_argument_filter_(tc.enable_argument_filter_),
       memory_dump_config_(tc.memory_dump_config_),
       included_categories_(tc.included_categories_),
       disabled_categories_(tc.disabled_categories_),
       excluded_categories_(tc.excluded_categories_),
-      synthetic_delays_(tc.synthetic_delays_) {}
+      synthetic_delays_(tc.synthetic_delays_),
+      category_event_filters_(tc.category_event_filters_) {}
 
 TraceConfig::~TraceConfig() {
 }
 
 TraceConfig& TraceConfig::operator=(const TraceConfig& rhs) {
   if (this == &rhs)
     return *this;
 
   record_mode_ = rhs.record_mode_;
   enable_sampling_ = rhs.enable_sampling_;
   enable_systrace_ = rhs.enable_systrace_;
   enable_argument_filter_ = rhs.enable_argument_filter_;
   memory_dump_config_ = rhs.memory_dump_config_;
   included_categories_ = rhs.included_categories_;
   disabled_categories_ = rhs.disabled_categories_;
   excluded_categories_ = rhs.excluded_categories_;
   synthetic_delays_ = rhs.synthetic_delays_;
+  category_event_filters_ = rhs.category_event_filters_;
   return *this;
 }
 
 const TraceConfig::StringList& TraceConfig::GetSyntheticDelayValues() const {
   return synthetic_delays_;
 }
 
+const TraceConfig::CategoryEventFilters& TraceConfig::GetCategoryEventFilters()
+    const {
+  return category_event_filters_;
+}
+
 std::string TraceConfig::ToString() const {
   base::DictionaryValue dict;
   ToDict(dict);
 
   std::string json;
   base::JSONWriter::Write(dict, &json);
 
   return json;
 }
 
@@ skipping 172 matching lines @@
 
   if (IsCategoryEnabled(MemoryDumpManager::kTraceCategory)) {
     // If dump triggers not set, the client is using the legacy with just
     // category enabled. So, use the default periodic dump config.
     const base::DictionaryValue* memory_dump_config = nullptr;
     if (dict.GetDictionary(kMemoryDumpConfigParam, &memory_dump_config))
       SetMemoryDumpConfig(*memory_dump_config);
     else
       SetDefaultMemoryDumpConfig();
   }
+
+  const base::ListValue* category_event_filters = nullptr;
+  if (dict.GetList(kCategoryEventFilterParam, &category_event_filters))
+    SetCategoryEventFilters(*category_event_filters);
 }
 
 void TraceConfig::InitializeFromConfigString(const std::string& config_string) {
   std::unique_ptr<Value> value(JSONReader::Read(config_string));
   if (!value)
     return InitializeDefault();
 
   const DictionaryValue* dict = nullptr;
   bool is_dict = value->GetAsDictionary(&dict);
 
@@ skipping 161 matching lines @@
     memory_dump_config_.push_back(dump_config);
   }
 }
 
 void TraceConfig::SetDefaultMemoryDumpConfig() {
   memory_dump_config_.clear();
   memory_dump_config_.push_back(kDefaultHeavyMemoryDumpTrigger);
   memory_dump_config_.push_back(kDefaultLightMemoryDumpTrigger);
 }
 
+void TraceConfig::SetCategoryEventFilters(
+    const base::ListValue& category_event_filters) {
+  category_event_filters_.clear();
+
+  for (size_t i = 0; i < category_event_filters.GetSize(); ++i) {
+    const base::DictionaryValue* event_filter = nullptr;
+    if (!category_event_filters.GetDictionary(i, &event_filter))
+      continue;
+
+    CategoryEventFilterConfig new_config;
+    if (!event_filter->GetString(kPredicateParam, &new_config.predicate_name)) {
+      LOG(ERROR) << "Invalid predicate name in category event filter.";
+      continue;
+    }
+
+    const base::ListValue* included_list = nullptr;
+    if (!event_filter->GetList(kIncludedCategoriesParam, &included_list)) {
+      LOG(ERROR) << "Missing included_categories in category event filter.";
+      continue;
+    }
+
+    for (size_t i = 0; i < included_list->GetSize(); ++i) {
+      std::string category;
+      if (included_list->GetString(i, &category))
+        new_config.included_categories.push_back(category);
+    }
+
+    const base::ListValue* excluded_list = nullptr;
+    if (event_filter->GetList(kExcludedCategoriesParam, &excluded_list)) {
+      for (size_t i = 0; i < excluded_list->GetSize(); ++i) {
+        std::string category;
+        if (excluded_list->GetString(i, &category))
+          new_config.excluded_categories.push_back(category);
+      }
+    }
+
+    const base::DictionaryValue* args_dict = nullptr;
+    if (event_filter->GetDictionary(kArgsParam, &args_dict)) {
+      new_config.args = args_dict->CreateDeepCopy();
+    }
+
+    category_event_filters_.push_back(new_config);
+  }
+}
+
 void TraceConfig::ToDict(base::DictionaryValue& dict) const {
   switch (record_mode_) {
     case RECORD_UNTIL_FULL:
       dict.SetString(kRecordModeParam, kRecordUntilFull);
       break;
     case RECORD_CONTINUOUSLY:
       dict.SetString(kRecordModeParam, kRecordContinuously);
       break;
     case RECORD_AS_MUCH_AS_POSSIBLE:
       dict.SetString(kRecordModeParam, kRecordAsMuchAsPossible);
@@ skipping 135 matching lines @@
           str.at(0) == ' ' ||
           str.at(str.length() - 1) == ' ';
 }
 
 bool TraceConfig::HasIncludedPatterns() const {
   return !included_categories_.empty();
 }
 
 }  // namespace trace_event
 }  // namespace base