diff options
Diffstat (limited to 'simpleperf/RecordFilter.h')
| -rw-r--r-- | simpleperf/RecordFilter.h | 86 |
1 files changed, 26 insertions, 60 deletions
diff --git a/simpleperf/RecordFilter.h b/simpleperf/RecordFilter.h index 4c5ee584..f0c72fdb 100644 --- a/simpleperf/RecordFilter.h +++ b/simpleperf/RecordFilter.h @@ -31,7 +31,7 @@ namespace simpleperf { -#define RECORD_FILTER_OPTION_HELP_MSG_FOR_RECORDING \ +#define RECORD_FILTER_OPTION_HELP_MSG \ "--exclude-pid pid1,pid2,... Exclude samples for selected processes.\n" \ "--exclude-tid tid1,tid2,... Exclude samples for selected threads.\n" \ "--exclude-process-name process_name_regex Exclude samples for processes with name\n" \ @@ -47,84 +47,31 @@ namespace simpleperf { " the regular expression.\n" \ "--include-uid uid1,uid2,... Include samples for processes belonging to selected uids.\n" -#define RECORD_FILTER_OPTION_HELP_MSG_FOR_REPORTING \ - "--exclude-pid pid1,pid2,... Exclude samples for selected processes.\n" \ - "--exclude-tid tid1,tid2,... Exclude samples for selected threads.\n" \ - "--exclude-process-name process_name_regex Exclude samples for processes with name\n" \ - " containing the regular expression.\n" \ - "--exclude-thread-name thread_name_regex Exclude samples for threads with name containing\n" \ - " the regular expression.\n" \ - "--include-pid pid1,pid2,... Include samples for selected processes.\n" \ - "--include-tid tid1,tid2,... Include samples for selected threads.\n" \ - "--include-process-name process_name_regex Include samples for processes with name\n" \ - " containing the regular expression.\n" \ - "--include-thread-name thread_name_regex Include samples for threads with name containing\n" \ - " the regular expression.\n" \ - "--filter-file <file> Use filter file to filter samples based on timestamps. The\n" \ - " file format is in doc/sampler_filter.md.\n" - -inline OptionFormatMap GetRecordFilterOptionFormats(bool for_recording) { - OptionFormatMap option_formats = { - {"--exclude-pid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, - {"--exclude-tid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, - {"--exclude-process-name", - {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, - {"--exclude-thread-name", - {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, - {"--include-pid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, - {"--include-tid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, - {"--include-process-name", - {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, - {"--include-thread-name", - {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, - }; - if (for_recording) { - option_formats.emplace( - "--exclude-uid", - OptionFormat({OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED})); - option_formats.emplace( - "--include-uid", - OptionFormat({OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED})); - } else { - option_formats.emplace( - "--filter-file", - OptionFormat({OptionValueType::STRING, OptionType::SINGLE, AppRunnerType::ALLOWED})); - } - return option_formats; -} - struct RecordFilterCondition { bool used = false; std::set<pid_t> pids; std::set<pid_t> tids; std::vector<std::regex> process_name_regs; std::vector<std::regex> thread_name_regs; - std::set<uint32_t> uids; + std::set<uid_t> uids; }; -class TimeFilter; - // Filter SampleRecords based on the rule below: // out_sample_records = (in_sample_records & ~exclude_conditions) & include_conditions // By default, exclude_conditions = 0, include_conditions = 1. class RecordFilter { public: - RecordFilter(const ThreadTree& thread_tree); - ~RecordFilter(); + RecordFilter(const ThreadTree& thread_tree) : thread_tree_(thread_tree) {} bool ParseOptions(OptionValueMap& options); void AddPids(const std::set<pid_t>& pids, bool exclude); void AddTids(const std::set<pid_t>& tids, bool exclude); void AddProcessNameRegex(const std::string& process_name, bool exclude); void AddThreadNameRegex(const std::string& thread_name, bool exclude); - void AddUids(const std::set<uint32_t>& uids, bool exclude); - bool SetFilterFile(const std::string& filename); + void AddUids(const std::set<uid_t>& uids, bool exclude); // Return true if the record passes filter. bool Check(const SampleRecord* r); - // Check if the clock matches the clock for timestamps in the filter file. - bool CheckClock(const std::string& clock); - RecordFilterCondition& GetCondition(bool exclude) { return exclude ? exclude_condition_ : include_condition_; } @@ -133,13 +80,32 @@ class RecordFilter { private: bool CheckCondition(const SampleRecord* r, const RecordFilterCondition& condition); bool SearchInRegs(const std::string& s, const std::vector<std::regex>& regs); - std::optional<uint32_t> GetUidForProcess(pid_t pid); + std::optional<uid_t> GetUidForProcess(pid_t pid); const ThreadTree& thread_tree_; RecordFilterCondition exclude_condition_; RecordFilterCondition include_condition_; - std::unordered_map<pid_t, std::optional<uint32_t>> pid_to_uid_map_; - std::unique_ptr<TimeFilter> time_filter_; + std::unordered_map<pid_t, std::optional<uid_t>> pid_to_uid_map_; }; +inline const OptionFormatMap& GetRecordFilterOptionFormats() { + static const OptionFormatMap option_formats = { + {"--exclude-pid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--exclude-tid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--exclude-process-name", + {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--exclude-thread-name", + {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--exclude-uid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--include-pid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--include-tid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--include-process-name", + {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--include-thread-name", + {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + {"--include-uid", {OptionValueType::STRING, OptionType::MULTIPLE, AppRunnerType::ALLOWED}}, + }; + return option_formats; +} + } // namespace simpleperf |