1 /* 2 * Copyright (C) 2015 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef SIMPLE_PERF_THREAD_TREE_H_ 18 #define SIMPLE_PERF_THREAD_TREE_H_ 19 20 #include <stdint.h> 21 22 #include <limits> 23 #include <memory> 24 #include <set> 25 26 #include "dso.h" 27 28 namespace simpleperf { 29 30 struct MapEntry { 31 uint64_t start_addr; 32 uint64_t len; 33 uint64_t pgoff; 34 uint64_t time; // Map creation time. 35 Dso* dso; 36 MapEntryMapEntry37 MapEntry(uint64_t start_addr, uint64_t len, uint64_t pgoff, uint64_t time, Dso* dso) 38 : start_addr(start_addr), len(len), pgoff(pgoff), time(time), dso(dso) { 39 } MapEntryMapEntry40 MapEntry() { 41 } 42 get_end_addrMapEntry43 uint64_t get_end_addr() const { 44 return start_addr + len; 45 } 46 }; 47 48 struct MapComparator { 49 bool operator()(const MapEntry* map1, const MapEntry* map2) const; 50 }; 51 52 struct ThreadEntry { 53 int pid; 54 int tid; 55 const char* comm; // It always refers to the latest comm. 56 std::set<MapEntry*, MapComparator> maps; 57 }; 58 59 class ThreadTree { 60 public: ThreadTree()61 ThreadTree() : unknown_symbol_("unknown", 0, std::numeric_limits<unsigned long long>::max()) { 62 unknown_dso_ = Dso::CreateDso(DSO_ELF_FILE, "unknown"); 63 unknown_map_ = 64 MapEntry(0, std::numeric_limits<unsigned long long>::max(), 0, 0, unknown_dso_.get()); 65 } 66 67 void AddThread(int pid, int tid, const std::string& comm); 68 void ForkThread(int pid, int tid, int ppid, int ptid); 69 ThreadEntry* FindThreadOrNew(int pid, int tid); 70 void AddKernelMap(uint64_t start_addr, uint64_t len, uint64_t pgoff, uint64_t time, 71 const std::string& filename); 72 void AddThreadMap(int pid, int tid, uint64_t start_addr, uint64_t len, uint64_t pgoff, 73 uint64_t time, const std::string& filename); 74 const MapEntry* FindMap(const ThreadEntry* thread, uint64_t ip, bool in_kernel); 75 const Symbol* FindSymbol(const MapEntry* map, uint64_t ip); UnknownMap()76 const MapEntry* UnknownMap() const { 77 return &unknown_map_; 78 } 79 80 void Clear(); 81 82 private: 83 Dso* FindKernelDsoOrNew(const std::string& filename); 84 Dso* FindUserDsoOrNew(const std::string& filename); 85 MapEntry* AllocateMap(const MapEntry& value); 86 void FixOverlappedMap(std::set<MapEntry*, MapComparator>* map_set, const MapEntry* map); 87 88 std::unordered_map<int, std::unique_ptr<ThreadEntry>> thread_tree_; 89 std::vector<std::unique_ptr<std::string>> thread_comm_storage_; 90 91 std::set<MapEntry*, MapComparator> kernel_map_tree_; 92 std::vector<std::unique_ptr<MapEntry>> map_storage_; 93 MapEntry unknown_map_; 94 95 std::unique_ptr<Dso> kernel_dso_; 96 std::unordered_map<std::string, std::unique_ptr<Dso>> module_dso_tree_; 97 std::unordered_map<std::string, std::unique_ptr<Dso>> user_dso_tree_; 98 std::unique_ptr<Dso> unknown_dso_; 99 Symbol unknown_symbol_; 100 }; 101 102 } // namespace simpleperf 103 104 using MapEntry = simpleperf::MapEntry; 105 using ThreadEntry = simpleperf::ThreadEntry; 106 using ThreadTree = simpleperf::ThreadTree; 107 108 struct Record; 109 110 void BuildThreadTree(const Record& record, ThreadTree* thread_tree); 111 112 #endif // SIMPLE_PERF_THREAD_TREE_H_ 113