159 lines
4 KiB
C++
159 lines
4 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;
|
|
}
|
|
}
|
|
|
|
|
|
}
|