chrome/browser/install_verification/win/module_verification_test.cc

   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
  13 std::set<size_t> ModuleVerificationTest::reported_module_ids_;
  14
  15 void ModuleVerificationTest::SetUp() {
  16   reported_module_ids_.clear();
  17 }
  18
  19 bool 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
  30 void ModuleVerificationTest::ReportModule(size_t module_id) {
  31   reported_module_ids_.insert(module_id);
  32 }