summaryrefslogtreecommitdiffstats
path: root/chrome/browser/install_verification/win/module_list.h
blob: f5b17893adf91cd699d8dd5e1ee24d47b97f582d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
// Copyright 2013 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.

#ifndef CHROME_BROWSER_INSTALL_VERIFICATION_WIN_MODULE_LIST_H_
#define CHROME_BROWSER_INSTALL_VERIFICATION_WIN_MODULE_LIST_H_

#include <Windows.h>

#include <set>
#include <vector>

#include "base/basictypes.h"
#include "base/memory/scoped_ptr.h"

struct ModuleInfo;

// Manages a list of HMODULEs, releasing them upon destruction.
class ModuleList {
 public:
  ~ModuleList();

  // Attempts to AddRef each HMODULE in |snapshot|. If a module was unloaded
  // since |snapshot| was taken it will be silently dropped. Successfully
  // AddRef'd HMODULEs will be inserted in the returned ModuleList.
  //
  // This method _always_ returns a valid ModuleList instance.
  static scoped_ptr<ModuleList> FromLoadedModuleSnapshot(
      const std::vector<HMODULE>& snapshot);

  // Retrieves name and address information for the module list.
  void GetModuleInfoSet(std::set<ModuleInfo>* module_info_set);

  size_t size() const {
    return modules_.size();
  }

  HMODULE operator[](size_t index) const {
    return modules_[index];
  }

 private:
  ModuleList();

  std::vector<HMODULE> modules_;

  DISALLOW_COPY_AND_ASSIGN(ModuleList);
};

#endif  // CHROME_BROWSER_INSTALL_VERIFICATION_WIN_MODULE_LIST_H_