Module Module::GetProviding(std::string name)
{
#if defined(_WIN32)
try
{
Module psapi(L"Psapi.dll");
auto EnumProcessModules = (BOOL WINAPI (*)(HANDLE, HMODULE*, DWORD, LPDWORD))psapi["EnumProcessModules"];
auto GetModuleFileNameExW = (DWORD WINAPI (*)(HANDLE, HMODULE, LPWSTR, DWORD))psapi["GetModuleFileNameExW"];
HANDLE process = GetCurrentProcess();
DWORD bytes_needed = 0;
EnumProcessModules(process, nullptr, 0, &bytes_needed);
size_t n_modules = bytes_needed / sizeof(HMODULE);
std::vector<HMODULE> modules(n_modules);
EnumProcessModules(process, modules.data(), modules.size()*sizeof(HMODULE), &bytes_needed);
for (auto m: modules)
{
Module module(m);
if (module.Probe(name))
{
wchar_t filename[MAX_PATH];
GetModuleFileNameExW(process, m, filename, MAX_PATH);
//LOG(Info, "Symbol \"" << name.c_str() << "\" was found in \"" << filename << "\""); // FIXME: global object dependency failure
return module;
}
}
}
catch (std::exception& e)
{
//LOG(Error, "Module enumeration has failed: " << e.what()); // FIXME: global object dependency failure
}
#else
void* m = dlopen(0, RTLD_NOW|RTLD_GLOBAL);
if (dlsym(m, name.c_str()) != nullptr)
{
return Module(m);
}
#endif
return Module();
}
QList<ProcessInfo> runningProcesses()
{
EnumWindowsProcParam param;
HANDLE snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if (!snapshot)
return param.processes;
QStringList deviceList;
const DWORD bufferSize = 1024;
char buffer[bufferSize + 1] = { 0 };
if (QSysInfo::windowsVersion() <= QSysInfo::WV_5_2) {
const DWORD size = GetLogicalDriveStringsA(bufferSize, buffer);
deviceList = QString::fromLatin1(buffer, size).split(QLatin1Char(char(0)), QString::SkipEmptyParts);
}
QLibrary kernel32(QLatin1String("Kernel32.dll"));
kernel32.load();
QueryFullProcessImageNamePtr pQueryFullProcessImageNamePtr = (QueryFullProcessImageNamePtr) kernel32
.resolve("QueryFullProcessImageNameA");
QLibrary psapi(QLatin1String("Psapi.dll"));
psapi.load();
GetProcessImageFileNamePtr pGetProcessImageFileNamePtr = (GetProcessImageFileNamePtr) psapi
.resolve("GetProcessImageFileNameA");
PROCESSENTRY32 processStruct;
processStruct.dwSize = sizeof(PROCESSENTRY32);
bool foundProcess = Process32First(snapshot, &processStruct);
while (foundProcess) {
HANDLE procHandle = OpenProcess(QSysInfo::windowsVersion() > QSysInfo::WV_5_2
? KDSYSINFO_PROCESS_QUERY_LIMITED_INFORMATION : PROCESS_QUERY_INFORMATION, false, processStruct
.th32ProcessID);
bool succ = false;
QString executablePath;
DWORD bufferSize = 1024;
if (QSysInfo::windowsVersion() > QSysInfo::WV_5_2) {
succ = pQueryFullProcessImageNamePtr(procHandle, 0, buffer, &bufferSize);
executablePath = QString::fromLatin1(buffer);
} else if (pGetProcessImageFileNamePtr) {
succ = pGetProcessImageFileNamePtr(procHandle, buffer, bufferSize);
executablePath = QString::fromLatin1(buffer);
for (int i = 0; i < deviceList.count(); ++i) {
executablePath.replace(QString::fromLatin1( "\\Device\\HarddiskVolume%1\\" ).arg(i + 1),
deviceList.at(i));
}
}
if (succ) {
const quint32 pid = processStruct.th32ProcessID;
param.seenIDs.append(pid);
ProcessInfo info;
info.id = pid;
info.name = executablePath;
param.processes.append(info);
}
CloseHandle(procHandle);
foundProcess = Process32Next(snapshot, &processStruct);
}
if (snapshot)
CloseHandle(snapshot);
kernel32.unload();
return param.processes;
}
