diff options
Diffstat (limited to 'src/modules/SystemBackendProc')
-rw-r--r-- | src/modules/SystemBackendProc/SystemBackendProc.cpp | 134 | ||||
-rw-r--r-- | src/modules/SystemBackendProc/SystemBackendProc.h | 67 |
2 files changed, 201 insertions, 0 deletions
diff --git a/src/modules/SystemBackendProc/SystemBackendProc.cpp b/src/modules/SystemBackendProc/SystemBackendProc.cpp new file mode 100644 index 0000000..68f9cec --- /dev/null +++ b/src/modules/SystemBackendProc/SystemBackendProc.cpp @@ -0,0 +1,134 @@ +/* + * SystemBackendProc.cpp + * + * Copyright (C) 2008 Matthias Schiffer <matthias@gamezock.de> + * + * This program is free software: you can redistribute it and/or modify it + * under the terms of the GNU 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +#include "SystemBackendProc.h" + +#include <Common/ActionManager.h> + +#include <cstdio> +#include <cstring> + +#include <boost/bind.hpp> + +#define init SystemBackendProc_LTX_init +#define deinit SystemBackendProc_LTX_deinit + +namespace Mad { +namespace Modules { + +SystemBackendProc *SystemBackendProc::backend = 0; + +bool SystemBackendProc::getUptimeInfo(const boost::function2<void, unsigned long, unsigned long> &callback) { + unsigned long uptime = 0, idleTime = 0; + + uptimeFile.seekg(0, std::ios::beg); + + if(!uptimeFile.good()) + return false; + + float f; + uptimeFile >> f; + if(!uptimeFile.good()) + return false; + + uptime = (unsigned long)f; + + uptimeFile >> f; + if(uptimeFile.good()) + idleTime = (unsigned long)f; + + Common::ActionManager::get()->add(boost::bind(callback, uptime, idleTime)); + + return true; +} + +bool SystemBackendProc::getMemoryInfo(const boost::function4<void, unsigned long, unsigned long, unsigned long, unsigned long> &callback) { + unsigned long totalMem = 0, freeMem = 0, totalSwap = 0, freeSwap = 0; + + meminfoFile.seekg(0, std::ios::beg); + + if(!meminfoFile.good()) + return false; + + while(!meminfoFile.eof() && meminfoFile.good()) { + std::string line; + std::getline(meminfoFile, line); + + char *name = new char[line.length()+1]; // is always long enough + unsigned long value; + + if(std::sscanf(line.c_str(), "%s %lu", name, &value) == 2) { + if(std::strcmp(name, "MemTotal:") == 0) + totalMem = value; + else if(std::strcmp(name, "MemFree:") == 0) + freeMem = value; + else if(std::strcmp(name, "SwapTotal:") == 0) + totalSwap = value; + else if(std::strcmp(name, "SwapFree:") == 0) + freeSwap = value; + } + + delete [] name; + + if(totalMem && freeMem && totalSwap && freeSwap) + break; + } + + Common::ActionManager::get()->add(boost::bind(callback, totalMem, freeMem, totalSwap, freeSwap)); + + return true; +} + +bool SystemBackendProc::getLoadInfo(const boost::function5<void, unsigned long, unsigned long, float, float, float> &callback) { + unsigned long currentLoad = 0, nProcesses = 0; + float loadAvg1 = 0, loadAvg5 = 0, loadAvg15 = 0; + + loadFile.seekg(0, std::ios::beg); + + if(!loadFile.good()) + return false; + + std::string line; + std::getline(loadFile, line); + + if(line.empty()) + return false; + + std::sscanf(line.c_str(), "%f %f %f %lu/%lu", &loadAvg1, &loadAvg5, &loadAvg15, ¤tLoad, &nProcesses); + + Common::ActionManager::get()->add(boost::bind(callback, currentLoad, nProcesses, loadAvg1, loadAvg5, loadAvg15)); + + return true; +} + +} +} + + +extern "C" { + +void init() { + Mad::Modules::SystemBackendProc::registerBackend(); +} + +void deinit() { + Mad::Modules::SystemBackendProc::unregisterBackend(); +} + +} diff --git a/src/modules/SystemBackendProc/SystemBackendProc.h b/src/modules/SystemBackendProc/SystemBackendProc.h new file mode 100644 index 0000000..bf19e22 --- /dev/null +++ b/src/modules/SystemBackendProc/SystemBackendProc.h @@ -0,0 +1,67 @@ +/* + * SystemBackendProc.h + * + * Copyright (C) 2008 Matthias Schiffer <matthias@gamezock.de> + * + * This program is free software: you can redistribute it and/or modify it + * under the terms of the GNU 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +#ifndef MAD_MODULES_SYSTEMBACKENDPROC_H_ +#define MAD_MODULES_SYSTEMBACKENDPROC_H_ + +#include <Common/SystemBackend.h> + +#include <fstream> + +namespace Mad { +namespace Modules { + +class SystemBackendProc : private Common::SystemBackend { + private: + static SystemBackendProc *backend; + + std::ifstream uptimeFile; + std::ifstream meminfoFile; + std::ifstream loadFile; + + SystemBackendProc() : uptimeFile("/proc/uptime"), meminfoFile("/proc/meminfo"), loadFile("/proc/loadavg") {} + + protected: + virtual bool getUptimeInfo(const boost::function2<void, unsigned long, unsigned long> &callback); + virtual bool getMemoryInfo(const boost::function4<void, unsigned long, unsigned long, unsigned long, unsigned long> &callback); + virtual bool getLoadInfo(const boost::function5<void, unsigned long, unsigned long, float, float, float> &callback); + + public: + static void registerBackend() { + if(backend) + return; + + backend = new SystemBackendProc(); + Common::SystemManager::get()->registerBackend(backend); + } + + static void unregisterBackend() { + if(!backend) + return; + + Common::SystemManager::get()->unregisterBackend(backend); + delete backend; + backend = 0; + } +}; + +} +} + +#endif /* MAD_MODULES_SYSTEMBACKENDPROC_H_ */ |