Functions.cpp

#include "Functions.h"
bool AttachProcess(const char* processName)
{
    PROCESSENTRY32 procEntry32;
    procEntry32.dwSize = sizeof(PROCESSENTRY32);

    HANDLE hProcSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
    if (hProcSnap == INVALID_HANDLE_VALUE)
        return false;

    while (Process32Next(hProcSnap, &procEntry32))
    {
        if (!strcmp(processName, procEntry32.szExeFile))
        {
            HANDLE temp = OpenProcess(PROCESS_ALL_ACCESS, FALSE, procEntry32.th32ProcessID);
            if (temp == INVALID_HANDLE_VALUE) {
                CloseHandle(hProcSnap);
                return false;
            }

            pID = procEntry32.th32ProcessID;
            pHandle = temp;

            CloseHandle(hProcSnap);
            return true;
        }
    }

    CloseHandle(hProcSnap);
    return false;
}
DWORD GetModule(const char* moduleName)
{
    HANDLE hSnapShot;
    hSnapShot = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, pID);

    MODULEENTRY32 mEntry;
    mEntry.dwSize = sizeof(MODULEENTRY32);

    if (hSnapShot == INVALID_HANDLE_VALUE) {
        CloseHandle(hSnapShot);
        return 0;
    }

    if (Module32First(hSnapShot, &mEntry))
    {
        if (!strcmp(moduleName, mEntry.szModule))
        {
            CloseHandle(hSnapShot);
            return (DWORD)mEntry.modBaseAddr;
        }
    }

    while (Module32Next(hSnapShot, &mEntry))
    {
        if (!strcmp(moduleName, mEntry.szModule))
        {
            CloseHandle(hSnapShot);
            return (DWORD)mEntry.modBaseAddr;
        }
    }
    CloseHandle(hSnapShot);
    return false;
}