C++注入后获取模块大小

#include <iostream>
#include <windows.h>
#include <psapi.h>

DWORD GetModuleSize(HANDLE hProcess, const char* moduleName) {
    HMODULE hModules[1024];
    DWORD cbNeeded;

    if (EnumProcessModules(hProcess, hModules, sizeof(hModules), &cbNeeded)) {
        int moduleCount = cbNeeded / sizeof(HMODULE);

        for (int i = 0; i < moduleCount; i++) {
            char szModuleName[MAX_PATH];
            if (GetModuleFileNameExA(hProcess, hModules[i], szModuleName, sizeof(szModuleName))) {
                if (strstr(szModuleName, moduleName) != NULL) {
                    MODULEINFO moduleInfo;
                    if (GetModuleInformation(hProcess, hModules[i], &moduleInfo, sizeof(moduleInfo))) {
                        return moduleInfo.SizeOfImage;
                    }
                }
            }
        }
    }

    return 0;
}

int main() {
    const char* targetProcessName = "target.exe";
    const char* injectedModuleName = "injected.dll";

    // 获取目标进程的句柄
    HANDLE hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, GetProcessId(targetProcessName));
    if (hProcess == NULL) {
        std::cout << "Failed to open target process." << std::endl;
        return 1;
    }

    // 获取被注入模块的大小
    DWORD moduleSize = GetModuleSize(hProcess, injectedModuleName);
    if (moduleSize > 0) {
        std::cout << "Size of injected module: " << moduleSize << " bytes." << std::endl;
    } else {
        std::cout << "Failed to get size of injected module." << std::endl;
    }

    // 关闭目标进程的句柄
    CloseHandle(hProcess);

    return 0;
}