| Index: chrome_frame/pin_module.cc
|
| diff --git a/chrome_frame/pin_module.cc b/chrome_frame/pin_module.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..b4cebbe2674a9169d1766e84c82b5dc565600f93
|
| --- /dev/null
|
| +++ b/chrome_frame/pin_module.cc
|
| @@ -0,0 +1,49 @@
|
| +// 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_frame/pin_module.h"
|
| +
|
| +#include <windows.h>
|
| +
|
| +#include "base/logging.h"
|
| +#include "chrome_frame/utils.h"
|
| +
|
| +extern "C" IMAGE_DOS_HEADER __ImageBase;
|
| +
|
| +namespace chrome_frame {
|
| +
|
| +namespace {
|
| +
|
| +PinModuleCallbackFn g_pin_module_callback = NULL;
|
| +
|
| +} // namespace
|
| +
|
| +void SetPinModuleCallback(PinModuleCallbackFn callback) {
|
| + g_pin_module_callback = callback;
|
| +}
|
| +
|
| +void PinModule() {
|
| + static bool s_pinned = false;
|
| + if (!s_pinned && !IsUnpinnedMode()) {
|
| + wchar_t system_buffer[MAX_PATH];
|
| + HMODULE this_module = reinterpret_cast<HMODULE>(&__ImageBase);
|
| + system_buffer[0] = 0;
|
| + if (GetModuleFileName(this_module, system_buffer,
|
| + arraysize(system_buffer)) != 0) {
|
| + HMODULE unused;
|
| + if (!GetModuleHandleEx(GET_MODULE_HANDLE_EX_FLAG_PIN, system_buffer,
|
| + &unused)) {
|
| + DPLOG(FATAL) << "Failed to pin module " << system_buffer;
|
| + } else {
|
| + s_pinned = true;
|
| + if (g_pin_module_callback)
|
| + g_pin_module_callback();
|
| + }
|
| + } else {
|
| + DPLOG(FATAL) << "Could not get module path.";
|
| + }
|
| + }
|
| +}
|
| +
|
| +} // namespace chrome_frame
|
|
|