| // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef DEBUGD_SRC_PERF_TOOL_H_ |
| #define DEBUGD_SRC_PERF_TOOL_H_ |
| |
| #include <stdint.h> |
| |
| #include <memory> |
| #include <string> |
| #include <vector> |
| |
| #include <base/macros.h> |
| #include <dbus-c++/dbus.h> |
| |
| namespace debugd { |
| |
| class RandomSelector; |
| |
| class PerfTool { |
| public: |
| // Class that returns the CPU arch and model. This is in a separate class so |
| // it can be mocked out for testing. |
| class CPUInfoReader { |
| public: |
| CPUInfoReader(); |
| virtual ~CPUInfoReader() {} |
| |
| // Accessors |
| const std::string& arch() const { |
| return arch_; |
| } |
| const std::string& model() const { |
| return model_; |
| } |
| |
| protected: |
| // The CPU arch and model info. |
| std::string arch_; |
| std::string model_; |
| }; |
| |
| PerfTool(); |
| // This is a special constructor for testing that takes in CPUInfoReader and |
| // RandomSelector args. In particular, it takes ownership of |
| // |random_selector|. |
| PerfTool(const CPUInfoReader& cpu_info, RandomSelector* random_selector); |
| ~PerfTool() = default; |
| |
| // Randomly runs the perf tool in various modes and collects various events |
| // for |duration_secs| seconds and returns a protobuf containing the collected |
| // data. |
| std::vector<uint8_t> GetRichPerfData(const uint32_t& duration_secs, |
| DBus::Error* error); |
| |
| private: |
| // Helper function that runs perf for a given |duration_secs| returning the |
| // collected data in |data_string|. |
| void GetPerfDataHelper(const uint32_t& duration_secs, |
| const std::vector<std::string>& perf_args, |
| DBus::Error* error, |
| std::string* data_string); |
| |
| std::unique_ptr<RandomSelector> random_selector_; |
| |
| DISALLOW_COPY_AND_ASSIGN(PerfTool); |
| }; |
| |
| } // namespace debugd |
| |
| #endif // DEBUGD_SRC_PERF_TOOL_H_ |