Windows下C++获取进程cpu使用率，内存使用，IO情况

https://whyabc.iteye.com/blog/2002685

process_stat.h：

* @brief 进程统计信息函数的声明 
#ifndef PROCESS_STAT_H  
#define PROCESS_STAT_H  
  
#ifdef __cplusplus  
extern "C" {  
#endif  

    typedef long long           int64_t;  
    typedef unsigned long long  uint64_t;  
  
    /// 获取当前进程的cpu使用率，返回-1失败  
    int get_cpu_usage();
  
    /// 获取当前进程内存和虚拟内存使用量，返回-1失败，0成功  
    int get_memory_usage(uint64_t* mem, uint64_t* vmem);
  
    /// 获取当前进程总共读和写的IO字节数，返回-1失败，0成功  
    int get_io_bytes(uint64_t* read_bytes, uint64_t* write_bytes);  
  
#ifdef  __cplusplus  
}  
#endif  
  
#endif

process_stat.cpp：

* @brief 进程统计信息函数的实现 
* 需要连接到psapi.lib 
  
#include <windows.h>  
#include <psapi.h>  
#include <assert.h>  
#include "process_stat.h"  
  
/// 时间转换  
static uint64_t file_time_2_utc(const FILETIME* ftime)  
{  
    LARGE_INTEGER li;  
  
    assert(ftime);  
    li.LowPart = ftime->dwLowDateTime;  
    li.HighPart = ftime->dwHighDateTime;  
    return li.QuadPart;  
}  
  
/// 获得CPU的核数  
static int get_processor_number()  
{  
    SYSTEM_INFO info;  
    GetSystemInfo(&info);  
    return (int)info.dwNumberOfProcessors;  
}  
  
int get_cpu_usage()  
{  
    //cpu数量  
    static int processor_count_ = -1;  
    //上一次的时间  
    static int64_t last_time_ = 0;  
    static int64_t last_system_time_ = 0;  
  
    FILETIME now;  
    FILETIME creation_time;  
    FILETIME exit_time;  
    FILETIME kernel_time;  
    FILETIME user_time;  
    int64_t system_time;  
    int64_t time;  
    int64_t system_time_delta;  
    int64_t time_delta;  
  
    int cpu = -1;  
  
    if(processor_count_ == -1)  
    {  
        processor_count_ = get_processor_number();  
    }  
  
    GetSystemTimeAsFileTime(&now);  
  
    if (!GetProcessTimes(GetCurrentProcess(), &creation_time, &exit_time,  
        &kernel_time, &user_time))  
    {  
        // We don't assert here because in some cases (such as in the Task Manager)  
        // we may call this function on a process that has just exited but we have  
        // not yet received the notification.  
        return -1;  
    }  
system_time=(file_time_2_utc(&kernel_time)+file_time_2_utc(&user_time))/processor_count_;  
    time = file_time_2_utc(&now);  
  
    if ((last_system_time_ == 0) || (last_time_ == 0))  
    {  
        // First call, just set the last values.  
        last_system_time_ = system_time;  
        last_time_ = time;  
        return -1;  
    }  
  
    system_time_delta = system_time - last_system_time_;  
    time_delta = time - last_time_;  
  
    assert(time_delta != 0);  
  
    if (time_delta == 0)  
        return -1;  
  
    // We add time_delta / 2 so the result is rounded.  
    cpu = (int)((system_time_delta * 100 + time_delta / 2) / time_delta);  
    last_system_time_ = system_time;  
    last_time_ = time;  
    return cpu;  
}  
  
int get_memory_usage(uint64_t* mem, uint64_t* vmem)  
{  
    PROCESS_MEMORY_COUNTERS pmc;  
    if(GetProcessMemoryInfo(GetCurrentProcess(), &pmc, sizeof(pmc)))  
    {  
        if(mem) *mem = pmc.WorkingSetSize;  
        if(vmem) *vmem = pmc.PagefileUsage;  
        return 0;  
    }  
    return -1;  
}  
  
int get_io_bytes(uint64_t* read_bytes, uint64_t* write_bytes)  
{  
    IO_COUNTERS io_counter;  
    if(GetProcessIoCounters(GetCurrentProcess(), &io_counter))  
    {  
        if(read_bytes) *read_bytes = io_counter.ReadTransferCount;  
        if(write_bytes) *write_bytes = io_counter.WriteTransferCount;  
        return 0;  
    }  
    return -1;  
}

main.cpp

* @brief 进程统计信息函数的测试 
  
#include "process_stat.h"  
#include <stdio.h>  
#include <Windows.h>  
  
int main()   
{   
    while(1)   
    {  
        int cpu;  
        uint64_t mem, vmem, r, w;  
  
        cpu = get_cpu_usage();  
        get_memory_usage(&mem, &vmem);  
        get_io_bytes(&r, &w);  
  
        printf("CPU使用率: %u\n",cpu);  
        printf("内存使用: %u 字节\n", mem);  
        printf("虚拟内存使用: %u 字节\n", vmem);  
        printf("总共读: %u 字节\n", r);  
        printf("总共写: %u 字节\n", w);   
  
        Sleep(1000);   
    }   
    return 0;   
}

注：需要用到psapi.h和psapi.lib文件。