khanat-opennel-code/code/nel/src/misc/cpu_time_stat.cpp

159 lines
3.9 KiB
C++

// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2010 Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
#include "stdmisc.h"
#include "nel/misc/cpu_time_stat.h"
#include "nel/misc/common.h"
#ifdef NL_OS_UNIX
#include <unistd.h>
#else
#include <process.h>
#endif
using namespace std;
namespace NLMISC
{
// Get absolute ticks value for the whole cpu set
bool CCPUTimeStat::getCPUTicks(uint64& user, uint64& nice, uint64& system, uint64& idle, uint64& iowait)
{
#ifdef NL_OS_UNIX
const char* statfile = "/proc/stat";
FILE* f = fopen(statfile, "r");
if (f == NULL)
return false;
// /proc/stat
// cpu [user] [nice] [system] [idle] [iowait] [irq] [softirq]
if (fscanf(f, "cpu %"NL_I64"u %"NL_I64"u %"NL_I64"u %"NL_I64"u %"NL_I64"u", &user, &nice, &system, &idle, &iowait) != 5)
nlwarning("Failed to parse /proc/stat");
fclose(f);
return true;
#else
return false;
#endif
}
// Get absolute ticks values for a specified pid
bool CCPUTimeStat::getPIDTicks(uint64& utime, uint64& stime, uint64& cutime, uint64& cstime, uint pid)
{
#ifdef NL_OS_UNIX
std::string statfile = NLMISC::toString("/proc/%u/stat", pid);
FILE* f = fopen(statfile.c_str(), "r");
if (f == NULL)
return false;
// /proc/<pid>/stat
// pid com sta ppi pgi ses tty tpg fla mif maf cmi cma [utime] [stime] [cutime] [cstime] ...
if (fscanf(f, "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %"NL_I64"u %"NL_I64"u %"NL_I64"u %"NL_I64"u", &utime, &stime, &cutime, &cstime) != 4)
nlwarning("Failed to parse /proc/<pid>/stat");
fclose(f);
return true;
#else
utime = 0;
stime = 0;
cutime = 0;
cstime = 0;
nlunreferenced(pid);
return false;
#endif
}
CCPUTimeStat::CCPUTimeStat() : _PID(0), _FirstTime(true), _LastCPUTicks(0), _LastPIDTicks(0)
{
#ifdef NL_OS_UNIX
_PID = getpid();
#else
_PID = _getpid();
#endif
}
// Peek measure
void CCPUTimeStat::peekMeasures()
{
NLMISC::TTime ctime = NLMISC::CTime::getLocalTime();
uint64 u, n, s, i, io;
uint64 ut, st, cut, cst;
if (getCPUTicks(u, n, s, i, io) && getPIDTicks(ut, st, cut, cst, _PID))
{
uint64 cpuTicks = u+n+s+i+io;
uint64 pidTicks = ut+st+cut+cst;
// only compute diff
if (_LastCPUTicks == cpuTicks || _LastPIDTicks == pidTicks)
return;
_LastCPUTicks = cpuTicks;
_CPUUser.computeDiff(u);
_CPUNice.computeDiff(n);
_CPUSystem.computeDiff(s);
_CPUIdle.computeDiff(i);
_CPUIOWait.computeDiff(io);
_LastPIDTicks = pidTicks;
_PIDUTime.computeDiff(ut);
_PIDSTime.computeDiff(st);
_PIDCUTime.computeDiff(cut);
_PIDCSTime.computeDiff(cst);
if (!_FirstTime)
{
uint32 cpuTotal = _CPUUser.Diff+_CPUNice.Diff+_CPUSystem.Diff+_CPUIdle.Diff+_CPUIOWait.Diff;
_CPUUser.computeLoad(cpuTotal, ctime);
_CPUNice.computeLoad(cpuTotal, ctime);
_CPUSystem.computeLoad(cpuTotal, ctime);
_CPUIdle.computeLoad(cpuTotal, ctime);
_CPUIOWait.computeLoad(cpuTotal, ctime);
uint32 pidTotal = _PIDUTime.Diff+_PIDSTime.Diff+_PIDCUTime.Diff+_PIDCSTime.Diff;
if (pidTotal > cpuTotal)
{
// EEK! should not happen!!
cpuTotal = pidTotal;
}
_PIDUTime.computeLoad(cpuTotal, ctime);
_PIDSTime.computeLoad(cpuTotal, ctime);
_PIDCUTime.computeLoad(cpuTotal, ctime);
_PIDCSTime.computeLoad(cpuTotal, ctime);
}
_FirstTime = false;
}
}
}