main.cpp

//спасибо https://stackoverflow.com/questions/63166/how-to-determine-cpu-and-memory-consumption-from-inside-a-process
#include "pch.h"
#include "TCHAR.h"
#include "pdh.h"
#include <string>
#include "GarrysMod\Lua\Interface.h"

#pragma comment(lib,"pdh.lib")

using namespace GarrysMod::Lua;

PDH_HQUERY* cpuQuery;
PDH_HCOUNTER* cpuTotal;
unsigned short numProcessors;


LUA_FUNCTION(GetProcessorLoad)
{
    auto num = static_cast<unsigned short>(LUA->CheckNumber(1));
    //memory miss protection
    if (num < 1 || num > numProcessors) return 0;

    PDH_FMT_COUNTERVALUE counterVal;
    PdhCollectQueryData(cpuQuery[num]);
    PdhGetFormattedCounterValue(cpuTotal[num], PDH_FMT_DOUBLE, NULL, &counterVal);
    //обрезается то шорта, так как мне нужны только целые числа
    //cast работает быстрее чем floor, а значений меньше нуля у меня быть не должно, поэтому вот так
    LUA->PushNumber(static_cast<unsigned short>(counterVal.doubleValue));

    return 1;
}

LUA_FUNCTION(GetProcessorsCount)
{
    LUA->PushNumber(numProcessors);
    return 1;
}

GMOD_MODULE_OPEN()
{
    SYSTEM_INFO sysInfo;
    GetSystemInfo(&sysInfo);
    numProcessors = sysInfo.dwNumberOfProcessors;

    //обращение к первому индекусу - нулевой процессор, ко второму - первый и тд. Сделано для луа стайла
    cpuQuery = new PDH_HQUERY[numProcessors + 1];
    cpuTotal = new PDH_HCOUNTER[numProcessors + 1];

    for (unsigned short core = 1; core <= numProcessors; core++)
    {
        //вроде если я делаю сразу wstr, то в стринге получается белеберда, поэтому сделал так
        std::string str = "\\Processor(" + std::to_string(core - 1) + ")\\% Processor Time";
        std::wstring wstr = std::wstring(str.begin(), str.end());
        PdhOpenQuery(NULL, NULL, &cpuQuery[core]);
        PdhAddEnglishCounter(cpuQuery[core], wstr.c_str(), NULL, &cpuTotal[core]);
        PdhCollectQueryData(cpuQuery[core]);
    }

    LUA->PushSpecial(GarrysMod::Lua::SPECIAL_GLOB);
    LUA->PushCFunction(GetProcessorLoad);
    LUA->SetField(-2, "GetProcessorLoad");
    LUA->Pop();

    LUA->PushSpecial(GarrysMod::Lua::SPECIAL_GLOB);
    LUA->PushCFunction(GetProcessorsCount);
    LUA->SetField(-2, "GetProcessorsCount");
    LUA->Pop();

    return 0;
}

GMOD_MODULE_CLOSE()
{
    delete[] cpuQuery;
    delete[] cpuTotal;
    return 0;
}