зеркало из https://github.com/mozilla/gecko-dev.git
62 строки
2.1 KiB
C++
62 строки
2.1 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef mozilla_WindowsEnumProcessModules_h
|
|
#define mozilla_WindowsEnumProcessModules_h
|
|
|
|
#include <windows.h>
|
|
#include <psapi.h>
|
|
|
|
#include "mozilla/FunctionRef.h"
|
|
#include "mozilla/NativeNt.h"
|
|
#include "mozilla/UniquePtr.h"
|
|
#include "mozilla/WinHeaderOnlyUtils.h"
|
|
|
|
namespace mozilla {
|
|
|
|
// Why don't we use CreateToolhelp32Snapshot instead of EnumProcessModules?
|
|
// CreateToolhelp32Snapshot gets the ANSI versions of module path strings
|
|
// via ntdll!RtlQueryProcessDebugInformation and stores them into a snapshot.
|
|
// Module32FirstW/Module32NextW re-converts ANSI into Unicode, but it cannot
|
|
// restore lost information. This means we still need GetModuleFileNameEx
|
|
// even when we use CreateToolhelp32Snapshot, but EnumProcessModules is faster.
|
|
inline bool EnumerateProcessModules(
|
|
const FunctionRef<void(const wchar_t*, HMODULE)>& aCallback) {
|
|
DWORD modulesSize;
|
|
if (!::EnumProcessModules(nt::kCurrentProcess, nullptr, 0, &modulesSize)) {
|
|
return false;
|
|
}
|
|
|
|
DWORD modulesNum = modulesSize / sizeof(HMODULE);
|
|
UniquePtr<HMODULE[]> modules = MakeUnique<HMODULE[]>(modulesNum);
|
|
if (!::EnumProcessModules(nt::kCurrentProcess, modules.get(),
|
|
modulesNum * sizeof(HMODULE), &modulesSize)) {
|
|
return false;
|
|
}
|
|
|
|
// The list may have shrunk between calls
|
|
if (modulesSize / sizeof(HMODULE) < modulesNum) {
|
|
modulesNum = modulesSize / sizeof(HMODULE);
|
|
}
|
|
|
|
for (DWORD i = 0; i < modulesNum; ++i) {
|
|
UniquePtr<wchar_t[]> modulePath = GetFullModulePath(modules[i]);
|
|
if (!modulePath) {
|
|
continue;
|
|
}
|
|
|
|
// Please note that modules[i] could be invalid if the module
|
|
// was unloaded after GetFullModulePath succeeded.
|
|
aCallback(modulePath.get(), modules[i]);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif // mozilla_WindowsEnumProcessModules_h
|