| // Copyright 2018 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/sampling_heap_profiler/module_cache.h" |
| |
| #include <utility> |
| |
| namespace base { |
| |
| ModuleCache::ModuleCache() = default; |
| ModuleCache::~ModuleCache() = default; |
| |
| const ModuleCache::Module* ModuleCache::GetModuleForAddress(uintptr_t address) { |
| Module* module = FindModuleForAddress(non_native_modules_, address); |
| if (module) |
| return module; |
| |
| module = FindModuleForAddress(native_modules_, address); |
| if (module) |
| return module; |
| |
| std::unique_ptr<Module> new_module = CreateModuleForAddress(address); |
| if (!new_module) |
| return nullptr; |
| native_modules_.push_back(std::move(new_module)); |
| return native_modules_.back().get(); |
| } |
| |
| std::vector<const ModuleCache::Module*> ModuleCache::GetModules() const { |
| std::vector<const Module*> result; |
| result.reserve(native_modules_.size()); |
| for (const std::unique_ptr<Module>& module : native_modules_) |
| result.push_back(module.get()); |
| return result; |
| } |
| |
| void ModuleCache::AddNonNativeModule(std::unique_ptr<Module> module) { |
| DCHECK(!module->IsNative()); |
| non_native_modules_.push_back(std::move(module)); |
| } |
| |
| void ModuleCache::InjectModuleForTesting(std::unique_ptr<Module> module) { |
| native_modules_.push_back(std::move(module)); |
| } |
| |
| // static |
| ModuleCache::Module* ModuleCache::FindModuleForAddress( |
| const std::vector<std::unique_ptr<Module>>& modules, |
| uintptr_t address) { |
| auto it = std::find_if(modules.begin(), modules.end(), |
| [address](const std::unique_ptr<Module>& module) { |
| return address >= module->GetBaseAddress() && |
| address < module->GetBaseAddress() + |
| module->GetSize(); |
| }); |
| return it != modules.end() ? it->get() : nullptr; |
| } |
| |
| } // namespace base |