base/sampling_heap_profiler/module_cache.cc

// 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) {
  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();
                         });
  if (it != modules_.end())
    return it->get();

  std::unique_ptr<Module> module = CreateModuleForAddress(address);
  if (!module)
    return nullptr;
  modules_.push_back(std::move(module));
  return modules_.back().get();
}

std::vector<const ModuleCache::Module*> ModuleCache::GetModules() const {
  std::vector<const Module*> result;
  result.reserve(modules_.size());
  for (const std::unique_ptr<Module>& module : modules_)
    result.push_back(module.get());
  return result;
}

void ModuleCache::InjectModuleForTesting(std::unique_ptr<Module> module) {
  modules_.push_back(std::move(module));
}

}  // namespace base