1// Copyright 2013 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "chrome/browser/install_verification/win/module_verification_test.h"
6
7#include <Windows.h>
8
9#include <vector>
10#include "chrome/browser/install_verification/win/loaded_modules_snapshot.h"
11#include "chrome/browser/install_verification/win/module_list.h"
12
13std::set<size_t> ModuleVerificationTest::reported_module_ids_;
14
15void ModuleVerificationTest::SetUp() {
16  reported_module_ids_.clear();
17}
18
19bool ModuleVerificationTest::GetLoadedModuleInfoSet(
20    std::set<ModuleInfo>* loaded_module_info_set) {
21  std::vector<HMODULE> snapshot;
22  if (!GetLoadedModulesSnapshot(&snapshot))
23    return false;
24  ModuleList::FromLoadedModuleSnapshot(snapshot)->GetModuleInfoSet(
25      loaded_module_info_set);
26  return true;
27}
28
29// static
30void ModuleVerificationTest::ReportModule(size_t module_id) {
31  reported_module_ids_.insert(module_id);
32}
33