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
|
// 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.
#include "chrome/browser/install_verification/win/loaded_modules_snapshot.h"
#include <Windows.h>
#include <algorithm>
#include <vector>
#include "base/bind.h"
#include "base/callback_helpers.h"
#include "chrome/browser/install_verification/win/module_verification_test.h"
#include "testing/gtest/include/gtest/gtest.h"
TEST(LoadedModulesSnapshotTest, TestCase) {
std::vector<HMODULE> snapshot;
ASSERT_TRUE(GetLoadedModulesSnapshot(&snapshot));
size_t original_snapshot_size = snapshot.size();
ASSERT_GT(original_snapshot_size, 0u);
snapshot.clear();
// Load in a new module. Pick msvidc32.dll as it is present from WinXP to
// Win8 and yet rarely used.
ASSERT_EQ(NULL, ::GetModuleHandle(L"msvidc32.dll"));
HMODULE new_dll = ::LoadLibrary(L"msvidc32.dll");
ASSERT_NE(static_cast<HMODULE>(NULL), new_dll);
base::ScopedClosureRunner release_new_dll(
base::Bind(base::IgnoreResult(&::FreeLibrary), new_dll));
ASSERT_TRUE(GetLoadedModulesSnapshot(&snapshot));
ASSERT_GT(snapshot.size(), original_snapshot_size);
ASSERT_NE(snapshot.end(),
std::find(snapshot.begin(), snapshot.end(), new_dll));
}
|
