分享一下自己做的一個獲取Linux系統狀態的類（c++）（代碼整理后的，在物理機上測試可以，在vm虛擬機上測試，磁盤計算有問題）

頭文件

#ifndef?LINUXSERVERSTATE_H #define?LINUXSERVERSTATE_H #include?<sys/types.h> #include?<unistd.h> #include?<string> #include?<vector> #include?<pthread.h> #include?<map>using?std::string; using?std::vector; using?std::map;#pragma?pack(push) #pragma?pack(8) struct?SysMemInfo {unsigned?long?total;//單位：Munsigned?long?free;//單位：M }; struct?SysDiskInfo {string???name;//sda1,sad2等等unsigned?long?total;//單位：M，不包括交換分區、CDROM等unsigned?long?free;//單位：M }; struct?SysNetInfo {float?send;//kbfloat?recv;//kbfloat?total;//kb }; struct?SysProcInfo {string???name;//進程名字int??????pid;//進程idint??????cpu;//cpu使用率long?????mem;//內存使用，單位Kb，linux內存指的是vmrss，即程序正在使用的物理內存，-1表示未獲取到或獲取錯誤 }; struct?SysProcTimeInfo {unsigned?long?user;//進程在用戶態執行的時間unsigned?long?kernel;//進程在內核態執行的時間 }; //linux?src目錄中的include/net/tcp_states.h，source目錄一般位于/usr/src/linux＊目錄中 enum?SysNetState{TCP_ESTABLISHED?=?1,TCP_SYN_SENT,TCP_SYN_RECV,TCP_FIN_WAIT1,TCP_FIN_WAIT2,TCP_TIME_WAIT,TCP_CLOSE,TCP_CLOSE_WAIT,TCP_LAST_ACK,TCP_LISTEN,TCP_CLOSING,????/*?Now?a?valid?state?*/TCP_MAX_STATES??/*?Leave?at?the?end!?*/ }; enum?SockType{UNKNOWN=0,TCP?,?UDP,?TCP6,?UDP6}; struct?SysNetConnInfo {int?????????????id;SockType????????protocol;std::string?????localAddr;int?????????????localPort;std::string?????remoteAddr;int?????????????remotePort;SysNetState?????state; }; #pragma?pack(pop)void?get_netstat(unsigned?long&?inOctets,unsigned?long&?outOctets,time_t&?sysTime); void?get_cpuusage(unsigned?long&?cpuTime,unsigned?long&?idleTime);class?LinuxServerState; void?get_procinfo(LinuxServerState*?pMain,map<pid_t,SysProcTimeInfo>&?procTimeInfo); void*?CPUNetProcessCount(void*?arg);class?LinuxServerState { public:LinuxServerState();virtual?????~LinuxServerState();int?????GetMemInfo(SysMemInfo&?memInfo);int?????GetDiskInfo(vector<SysDiskInfo>&?diskInfo,unsigned?long?&total,unsigned?long?&available);//返回cpu的利用率，返回－1表示超時或者其他錯誤int?????GetCpuUsage();//sys?net?infoSysNetInfo??GetNetInfo();//process?infovoid????GetProcInfo(vector<SysProcInfo>&?procinfo);//net?connection?infovoid????GetNetConnectionInfo(vector<SysNetConnInfo>&?netConnInfo); private:void????initialize();friend??void*?CPUNetProcessCount(void*?arg);friend??void?get_procinfo(LinuxServerState*?pMain,map<pid_t,SysProcTimeInfo>&?procTimeInfo);int?????????????????m_cpuUsage;?//多個CPU的總的使用率pthread_mutex_t?????m_mutex;SysNetInfo??????????m_netInfo;pthread_t???????????m_countThread;vector<SysProcInfo>?*m_procinfo_cal;vector<SysProcInfo>?*m_procinfo;void????????????????*m_threadRet;LinuxServerState(const?LinuxServerState&?other);virtual?LinuxServerState&?operator=(const?LinuxServerState&?other);virtual?bool?operator==(const?LinuxServerState&?other)?const;bool????????????????m_running; };#endif

源文件

#include?"sys_stat.h" #include?<sys/sysinfo.h>//getmemInfo #include?<stdio.h> #include?<stdlib.h> #include?<fstream> #include?<climits> #include?<cstring> #include?<time.h> #include?<dirent.h> #include?<errno.h> #include?<iostream> #include?<fcntl.h> #include?<sys/types.h> #include?<sys/stat.h> #include?<netinet/in.h> #include?<sys/socket.h> #include?<arpa/inet.h> #include?<algorithm>using?namespace?std;void?get_netstat(unsigned?long&?inOctets,unsigned?long&?outOctets,time_t&?sysTime) {unsigned?long???inNoRoutes?=?0,?inTruncPkts?=?0;unsigned?long???inMcastPkts?=?0,?outMcastPkts?=?0;unsigned?long???inBcastPkts?=?0,?outBcastPkts?=?0;unsigned?long???inMcastOctets?=?0,?outMcastOctets?=?0;unsigned?long???inBcastOctets?=?0,?outBcastOctets?=?0;fstream?????fProcNetStat("/proc/net/netstat",ios_base::in);char????????bufNetStat[2048]?=?{'\0'},?ipExt[16]?=?{'\0'};while(!fProcNetStat.eof()){fProcNetStat.getline(bufNetStat,2048);if(strncasecmp(bufNetStat,"tcpext",6)?==?0)continue;if(strncasecmp(bufNetStat,"ipext",5)?==?0){fProcNetStat.getline(bufNetStat,2048);sscanf(bufNetStat,"%s?%ld?%ld?%ld?%ld?%ld?%ld?%ld?%ld?%ld?%ld?%ld",ipExt,&inNoRoutes,&inTruncPkts,&inMcastPkts,&outMcastPkts,&inBcastPkts,&outBcastPkts,&inOctets,&outOctets,&inMcastOctets,&outMcastOctets,&inBcastOctets,&outBcastOctets);sysTime?=?time(NULL);break;}} }void?get_cpuusage(unsigned?long&?cpuTime,unsigned?long&?idleTime) {fstream?????fProcStat("/proc/stat",ios_base::in);char??????????cpubuf[8]?=?{'\0'};unsigned?long?userT?=?0,?niceT?=?0,?systemT?=?0,?idleT?=?0;unsigned?long?ioT?=?0,?irqT?=?0,?softirqT?=?0;fProcStat>>cpubuf>>userT>>niceT>>systemT>>idleT>>ioT>>irqT>>softirqT;cpuTime?=?userT?+?niceT?+?systemT?+?idleT?+?ioT?+?irqT?+?softirqT;idleTime?=?idleT; }void?get_procinfo(LinuxServerState*?pMain,map<pid_t,SysProcTimeInfo>&?procTimeInfo) {struct?dirent*?ent??=?NULL;char?*endptr????????=?NULL;DIR*?pDirProc???????=?opendir("/proc");long?utime?=?0,?stime?=?0;if(pDirProc?==?NULL)return?;char?dirName[256]?=?{'\0'};char?tmpbuf[2048]?=?{'\0'};if?(pMain->m_procinfo_cal?==?NULL?)return;pMain->m_procinfo_cal->clear();while((ent?=?readdir(pDirProc))!=?NULL){long?pidNum?=?strtol(ent->d_name,&endptr,10);if(pidNum?==?LONG_MAX?||?pidNum?==?LONG_MIN?||?endptr?==?ent->d_name)continue;fstream?fPidStat,fPidMem;int?i?=?0;SysProcTimeInfo?tmpProcTimeInfo;SysProcInfo?????tmpProcInfo;snprintf(dirName,256,"%s%d%s","/proc/",pidNum,"/stat");try{fPidStat.open(dirName,ios_base::in);}catch(std::exception&?e){continue;}fPidStat>>tmpbuf;//獲取pidi++;fPidStat>>tmpbuf;//獲取進程名字，名字由"()"包括i++;//?process?nameif(strlen(tmpbuf)?>?2)tmpProcInfo.name??=?string(tmpbuf).substr(1,strlen(tmpbuf)?-2);elsecontinue;for(;?i?<?14;i++)fPidStat>>tmpbuf;//14項表示utime,15項表示stimefPidStat?>>?utime;i++;fPidStat?>>?stime;i++;tmpProcTimeInfo.user????=?utime;tmpProcTimeInfo.kernel??=?stime;//?process?cpu?usagemap<pid_t,SysProcTimeInfo>::iterator?it?=?procTimeInfo.find((pid_t)pidNum);if(it?==?procTimeInfo.end()){procTimeInfo.insert(pair<pid_t,SysProcTimeInfo>(pidNum,tmpProcTimeInfo));tmpProcInfo.cpu?=?0;}else{tmpProcInfo.cpu?=?utime?+?stime?-?it->second.kernel?-?it->second.user;if(tmpProcInfo.cpu?>?100)tmpProcInfo.cpu?=?100;if(tmpProcInfo.cpu?<?0)tmpProcInfo.cpu?=?0;it->second.kernel???=?stime;it->second.user?????=?utime;}utime???????????=?0;stime???????????=?0;tmpProcInfo.pid?=?pidNum;//?process?memchar?*pstrtolong;memset(dirName,'\0',128);snprintf(dirName,128,"%s%d%s","/proc/",pidNum,"/status");fPidMem.open(dirName,ios_base::in);while(!fPidMem.eof()){fPidMem?>>?tmpbuf;if(strncasecmp(tmpbuf,"vmrss",5)?==?0){fPidMem?>>?tmpbuf;unsigned?long?vmrss?=?strtol(tmpbuf,&pstrtolong,10);if(vmrss?==?LONG_MAX?||?vmrss?==?LONG_MIN?||?pstrtolong?==?tmpbuf){tmpProcInfo.mem?=?-1;}elsetmpProcInfo.mem?=?vmrss;if(0?==?pthread_mutex_trylock(&pMain->m_mutex)){pMain->m_procinfo_cal->push_back(tmpProcInfo);pthread_mutex_unlock(&pMain->m_mutex);}break;}}memset(dirName,'\0',256);memset(tmpbuf,'\0',2048);}closedir(pDirProc);pDirProc????=?NULL;if(0?==?pthread_mutex_trylock(&pMain->m_mutex)){std::swap(pMain->m_procinfo_cal?,pMain->m_procinfo);pthread_mutex_unlock(&pMain->m_mutex);} }void*?CPUNetProcessCount(void*?arg) {LinuxServerState*?pMain?=?(LinuxServerState*)arg;unsigned?long???inOctets?=?0,?outOctets?=?0;time_t??????????sysTime;unsigned?long???cpuTime?=?0,?idleTime?=?0;map<pid_t,SysProcTimeInfo>?procTimeInfo;while(pMain->m_running){//cpu?total?usageunsigned?long?cpuTimeNow?=?0,?idleTimeNow?=?0;int?cpu?=?0;get_cpuusage(cpuTimeNow,idleTimeNow);cpu?=?100?-?(idleTimeNow?-?idleTime)*100/(cpuTimeNow?-?cpuTime);idleTime??=?idleTimeNow;cpuTime???=?cpuTimeNow;if(0?==?pthread_mutex_trylock(&pMain->m_mutex)){pMain->m_cpuUsage?=?cpu;pthread_mutex_unlock(&pMain->m_mutex);}//else暫時不處理//net?statunsigned?long?inOctetsTmp?=?0,outOctetsTmp?=?0;time_t????timeNow?=?time(NULL);get_netstat(inOctetsTmp,outOctetsTmp,timeNow);SysNetInfo?tmpNetInfo;tmpNetInfo.recv?????=?(inOctetsTmp?-?inOctets)/1024/(timeNow?-?sysTime);tmpNetInfo.send?????=?(outOctetsTmp?-?outOctets)/1024/(timeNow?-?sysTime);tmpNetInfo.total????=?tmpNetInfo.recv?+?tmpNetInfo.send;sysTime???=?timeNow;inOctets??=?inOctetsTmp;outOctets?=?outOctetsTmp;if(0?==?pthread_mutex_trylock(&pMain->m_mutex)){pMain->m_netInfo?=?tmpNetInfo;pthread_mutex_unlock(&pMain->m_mutex);}//proc?infoget_procinfo(pMain,procTimeInfo);sleep(1);}return?NULL; }LinuxServerState::LinuxServerState() {m_cpuUsage??????????=?0;m_countThread???????=?0;m_threadRet?????????=?NULL;m_netInfo.total?????=?0;m_netInfo.recv??????=?0;m_netInfo.send??????=?0;m_procinfo_cal??????=?new?vector<SysProcInfo>();m_procinfo??????????=?new?vector<SysProcInfo>();m_running???????????=?true;initialize(); }LinuxServerState::LinuxServerState(const?LinuxServerState&?other) { }LinuxServerState::~LinuxServerState() {m_running?=?false;if(m_countThread?!=?0)pthread_join(m_countThread,NULL);pthread_mutex_destroy(&m_mutex);if?(?m_procinfo?!=?NULL?)delete?m_procinfo;if?(?m_procinfo_cal?!=?NULL?)delete?m_procinfo_cal; }LinuxServerState&?LinuxServerState::operator=(const?LinuxServerState&?other) {return?*this; }bool?LinuxServerState::operator==(const?LinuxServerState&?other)?const {///TODO:?return?...; }void?LinuxServerState::initialize() {pthread_mutex_init(&m_mutex,NULL);pthread_create(&m_countThread,NULL,&CPUNetProcessCount,this); }int?LinuxServerState::GetCpuUsage() {timespec??tmp;tmp.tv_nsec???=?0;tmp.tv_sec????=?1;int?cpu???=?0;if(0?==?pthread_mutex_timedlock(&m_mutex,&tmp)){cpu?????=?m_cpuUsage;pthread_mutex_unlock(&m_mutex);}if(cpu?>?100)return?100;if(cpu?<?0)return?0;return?cpu; }int?LinuxServerState::GetMemInfo(SysMemInfo&?memInfo) {struct?sysinfo????tmp;int??ret?=?sysinfo(&tmp);if(ret?==?0){memInfo.free????=?(unsigned?long)tmp.freeram/(1024*1024);memInfo.total???=?(unsigned?long)tmp.totalram/(1024*1024);}return?ret; }int?LinuxServerState::GetDiskInfo(std::vector<SysDiskInfo>&?diskInfo,unsigned?long?&total,unsigned?long?&available) {total?????=?0;available?=?0;char??????buf[128]?=?{'\0'};string????fileName;char??????*p;fstream???fProc("/proc/partitions",ios_base::in);unsigned?long?total2?=?0,?free2?=?0,?sumTotal?=?0,?sumFree?=?0;int???????major?=?0,?minor?=?0;//首先計算總大小及各個分區的大小，包括交換分區，然后用df命令獲取可用空間while(!fProc.eof()){fProc>>fileName;total2??=?strtol(fileName.c_str(),&p,10);if(total2?==?0)continue;else?if(total2?==?LONG_MAX?||?total2?==?LONG_MIN?||?p?==?fileName.c_str())return?total2;else{major?????=?total2;fProc>>minor;fProc>>total2;fProc>>fileName;SysDiskInfo?tmp;tmp.total?=?(unsigned?long)(total2/1024);tmp.free??=?0;tmp.name??=?fileName;diskInfo.push_back(tmp);if((fileName.find("sd")!=?string::npos?||?fileName.find("hd")?!=?string::npos)?&&!(fileName[fileName.size()-1]?>='0'?&&??fileName[fileName.size()-1]?<='9'))sumTotal?+=?tmp.total;fileName.clear();}}//獲取可用空間const?char????????*command?=?"df";const?char????????*type????=?"r";FILE??*pFile??????=?popen(command,type);char??buf2[1024]??=?{'\0'};char??filesys[64]?=?{'\0'},?usage[8]?=?{'\0'},?mountedon[64]?=?{'\0'};unsigned?long?kblocks?=?0,?used?=?0,?available2?=?0;if(pFile?==?NULL)return?-1;vector<SysDiskInfo>::iterator?it;while?(!feof?(pFile)?){if?(fgets?(buf2?,?1024?,?pFile)?==?NULL?)break;sscanf(buf2,"%s?%ld?%ld?%ld?%s?%s",filesys,&kblocks,&used,&available2,usage,mountedon);string?ff(filesys);if(ff.rfind("/")?==?string::npos)continue;for(it?=?diskInfo.begin();?it?!=?diskInfo.end();it++){if(it->name?==?ff.substr(ff.rfind("/")+1)){it->free???=?(unsigned?long)(available2/1024);sumFree???+=?it->free;}}memset(buf2,'\0',1024);memset(filesys,'\0',64);memset(mountedon,'\0',64);memset(usage,'\0',8);kblocks?????=?0;used????????=?0;available2??=?0;}pclose(pFile);total???????=?sumTotal;available???=?sumFree;return?0; }SysNetInfo?LinuxServerState::GetNetInfo() {SysNetInfo????tmpNetInfo;tmpNetInfo.recv???=?0;tmpNetInfo.send???=?0;tmpNetInfo.total??=?0;timespec??tmp;tmp.tv_nsec???=?0;tmp.tv_sec????=?1;if(0?==?pthread_mutex_timedlock(&m_mutex,&tmp)){tmpNetInfo??????=?m_netInfo;pthread_mutex_unlock(&m_mutex);}return?tmpNetInfo; }void?LinuxServerState::GetProcInfo(std::vector<SysProcInfo>&?procinfo) {timespec??tmp;tmp.tv_nsec???=?0;tmp.tv_sec????=?1;if(0?==?pthread_mutex_timedlock(&m_mutex,&tmp)){if?(?m_procinfo?==?NULL?)return;procinfo????????=?*m_procinfo;pthread_mutex_unlock(&m_mutex);} }void?LinuxServerState::GetNetConnectionInfo(std::vector<SysNetConnInfo>&?netConnInfo) {int?i?=?0,id?=?0;char?buf[1024]?=?{'\0'};while(i++?<?4){fstream?????fInfo;SockType????????stateType?=?UNKNOWN;if(i?==?1){fInfo.open("/proc/net/tcp",ios_base::in);stateType??=?TCP;}else?if(i?==?2){fInfo.open("/proc/net/udp",ios_base::in);stateType?=?UDP;}else?if(i?==?3){fInfo.open("/proc/net/tcp6",ios_base::in);stateType?=?TCP6;}else?if(i?==?4){fInfo.open("/proc/net/udp6",ios_base::in);stateType?=?UDP6;}if(stateType?==?UNKNOWN)continue;while(!fInfo.eof()){SysNetConnInfo????tmpConnInfo;char?sl[6]?=?{'\0'},localaddr[32]?=?{'\0'},remoteaddr[32]?=?{'\0'};fInfo.getline(buf,1024);unsigned?long?llocal?=?0,?lremote?=?0;int???????localPort?=?0,?remotePort?=?0,?st?=?0;int?ret;if(i?==?1?||?i?==?2){ret?=?sscanf(buf,"%s%x:%x%x:%x%x",sl,&llocal,&localPort,&lremote,&remotePort,&st);if(ret?!=?6)continue;in_addr?????????tmpAddr;tmpAddr.s_addr??????=?llocal;tmpConnInfo.localAddr???=?inet_ntoa(tmpAddr);tmpAddr.s_addr??????=?lremote;tmpConnInfo.remoteAddr??=?inet_ntoa(tmpAddr);}else?if(i?==?3?||?i?==?4?){ret?=?sscanf(buf,"%s%32s:%x%32s:%x%x",sl,&localaddr,&localPort,&remoteaddr,&remotePort,&st);if(ret?!=?6)continue;char????tmp[64]?=?{'\0'};int?pos?=?0;for(int?j?=?0;j?<?32;?j++){tmp[pos++]?=?localaddr[j];if((j?%?4)?==?3?&&?j?!=?31)tmp[pos++]?=?':';}tmpConnInfo.localAddr???=?string(tmp);memset(tmp,'\0',64);pos??=?0;for(int?j?=?0;j?<?32;?j++){tmp[pos++]?=?remoteaddr[j];if((j?%?4)?==?3?&&?j?!=?31)tmp[pos++]?=?':';}tmpConnInfo.remoteAddr??=?string(tmp);}tmpConnInfo.localPort?=?localPort;tmpConnInfo.remotePort????=?remotePort;tmpConnInfo.id????????=?id++;tmpConnInfo.protocol??=?stateType;tmpConnInfo.state?????=?(SysNetState)st;netConnInfo.push_back(tmpConnInfo);}fInfo.close();} }

測試文件

#include?<iostream> #include?"sys_stat.h"using?namespace?std;int?main(int?argc,?char?*argv[]) {if?(argc?<?2){cout<<"usage:"<<endl;cout<<"\t./test?mem/disk/cpu/proc/net/conn"<<endl;return?0;}string?opt(argv[1]);int?i?=?0;LinuxServerState?stat;while?(?i++?<?10?){if?(opt?==?"cpu"){cout?<<?"cpu_use:"<<stat.GetCpuUsage()<<endl;}else?if?(opt?==?"mem"){SysMemInfo?meminfo;stat.GetMemInfo(meminfo);cout<<"meminfo:"<<endl;cout<<"\ttotal:"<<meminfo.total<<"M"<<endl;cout<<"\tfree:"<<meminfo.free<<"M"<<endl;cout<<"\t"<<((double)(meminfo.total?-?meminfo.free))*100/meminfo.total<<endl;}else?if?(opt?==?"disk"){vector<SysDiskInfo>?vdiskinfo;unsigned?long?disktotal,diskavail;stat.GetDiskInfo(vdiskinfo,disktotal,diskavail);cout<<"diskinfo:"<<endl;cout<<"\ttotal:"<<disktotal<<"M"<<endl;cout<<"\tfree:"<<diskavail<<"M"<<endl;cout<<"\t"<<((double)(disktotal?-?diskavail))*100/disktotal<<endl;vector<SysDiskInfo>::iterator?itdisk?=?vdiskinfo.begin();cout<<endl;for(;?itdisk?!=?vdiskinfo.end();?++itdisk){cout<<"\tname:"<<itdisk->name<<endl;cout<<"\ttotal:"<<itdisk->total<<"M"<<endl;cout<<"\tfree:"<<itdisk->free<<"M"<<endl;cout<<endl;}}else?if?(opt?==?"net"){SysNetInfo??netinfo?=?stat.GetNetInfo();cout<<"netinfo:"<<endl;cout<<"\trecv:"<<netinfo.recv<<"kb"<<endl;cout<<"\tsend:"<<netinfo.send<<"kb"<<endl;cout<<"\ttotal:"<<netinfo.total<<"kb"<<endl;}else?if?(opt?==?"proc"){vector<SysProcInfo>?procinfo;stat.GetProcInfo(procinfo);cout<<"procinfo:"<<endl;vector<SysProcInfo>::iterator?itproc?=?procinfo.begin();for(;?itproc?!=?procinfo.end();?++itproc){cout<<"\tname:"<<itproc->name<<endl;cout<<"\tpid:"<<itproc->pid<<endl;cout<<"\tmem:"<<itproc->mem<<"Kb"<<endl;cout<<"\tcpu:"<<itproc->cpu<<endl;cout<<endl;}}else?if?(opt?==?"conn"){vector<SysNetConnInfo>?netConnInfo;stat.GetNetConnectionInfo(netConnInfo);vector<SysNetConnInfo>::iterator?itconn?=?netConnInfo.begin();for(;?itconn!=?netConnInfo.end();?++itconn){cout<<"\tid:"<<itconn->id<<endl;cout<<"\tprotocol:"<<itconn->protocol<<endl;cout<<"\tlocalAddr:"<<itconn->localAddr<<":"<<itconn->localPort<<endl;cout<<"\tremoteAddr:"<<itconn->remoteAddr<<":"<<itconn->remotePort<<endl;cout<<"\tstate:"<<itconn->state<<endl;cout<<endl;}}sleep(1);}sleep(1);return?0; }

makefile

target=testsrc=$(wildcard?./*.cpp) objs=$(patsubst?%.cpp,%.o,$(src))%.o:%.cppg++?-g?-c?$<${target}:${objs}g++?-o?$@?${objs}?-lpthreadclean:rm?./*.orm?./test

轉載于:https://blog.51cto.com/hakuyo/1218145

總結

以上是生活随笔為你收集整理的linux系统状态（磁盘大小、内存、进程、cpu使用率、网络连接）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。