全景AR增强监视系统对接SkeyeIVMS视频云管控系统实现软硬件资源的健康状态管理(二)

int MonitorApi::getProcessCount(const QString &processName)
{
    //进程计数
    static bool initialized = false;

    int count = 0;
    QString fileName = QFileInfo(processName).fileName();
    QString fullName = QDir::toNativeSeparators(QFileInfo(processName).absoluteFilePath());

#ifdef Q_OS_WIN
    if (!initialized) {
        //提升进程权限
        LUID luidTmp;
        HANDLE tokenHnadle;
        TOKEN_PRIVILEGES tkp;
        if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &tokenHnadle)) {
            if(!LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &luidTmp)) {
                qDebug() << "LookupPrivilegeValue Faile! Error =" << GetLastError();
                CloseHandle(tokenHnadle);
            }

            tkp.PrivilegeCount = 1;
            tkp.Privileges[0].Luid = luidTmp;
            tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

            if(!AdjustTokenPrivileges(tokenHnadle, FALSE, &tkp, sizeof(tkp), NULL, NULL)) {
                qDebug() << "AdjustTokenPrivileges Faile! Error =" << GetLastError();
                CloseHandle(tokenHnadle);
            }
            CloseHandle(tokenHnadle);
        }
        initialized = true;
    }

    PROCESSENTRY32 pe32;
    pe32.dwSize = sizeof(PROCESSENTRY32);

    //创建进程快照句柄
    HANDLE processSnapshotHandle = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);

    //创建句柄失败
    if (processSnapshotHandle == INVALID_HANDLE_VALUE)
        return -1;

    //找第一个
    bool isExist = Process32First(processSnapshotHandle, &pe32);

    while(isExist) {
        auto name = QString::fromWCharArray(pe32.szExeFile);
        //进程名不区分大小写
        if(fileName.compare(name, Qt::CaseInsensitive) == 0) {
            HMODULE modules[512] = {0};
            DWORD needed = 0;
            TCHAR mouduleName[MAX_PATH];
            BOOL is64Process;
            HANDLE processHandle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pe32.th32ProcessID);
            if (processHandle) {
                IsWow64Process(processHandle, &is64Process); //判断是32位还是64位进程
                EnumProcessModulesEx(processHandle, modules, sizeof(modules), &needed, is64Process ? LIST_MODULES_32BIT : LIST_MODULES_64BIT);
                if (needed > 0) {
                    GetModuleFileNameEx(processHandle, modules[0], mouduleName, MAX_PATH);
                    if (QString::fromWCharArray(mouduleName) == fullName)
                        count++;
                }
                CloseHandle(processHandle);
            }
        }
        //找下一个
        isExist = Process32Next(processSnapshotHandle, &pe32);
    }
#endif

    return count;
}