#ifndef PROCESS_STAT_H#define PROCESS_STAT_H#ifdef __cplusplusextern “C” {
#endiftypedef 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_H*//*** 需要连接到psapi.lib*/#include#include #include#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 TaskManager)// we may call this function on a process that has just exited butwe 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;
}#include “process_stat.h”#include#includeint 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;
}
|
#ifndef PROCESS_STAT_H#define PROCESS_STAT_H#ifdef __cplusplusextern “C” {
#endiftypedef 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_H*//*** 需要连接到psapi.lib*/#include#include #include#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 TaskManager)// we may call this function on a process that has just exited butwe 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;
}#include “process_stat.h”#include#includeint 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;
}
|