File indexing completed on 2024-04-28 16:49:55

 /*
     SPDX-FileCopyrightText: 2007 John Tapsell <tapsell@kde.org>
 
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "processes_atop_p.h"
 #include "atop_p.h"
 #include "process.h"
 #include "processcore_debug.h"
 
 #include <zlib.h>
 
 #include <QByteArray>
 #include <QFile>
 #include <QHash>
 #include <QTextStream>
 #include <QtEndian>
 
 #include <QDebug>
 
 namespace KSysGuard
 {
 class ProcessesATop::Private
 {
 public:
     Private();
     ~Private();
     QFile atopLog;
     bool ready;
 
     bool loadDataForHistory(int index);
     bool loadHistoryFile(const QString &filename);
 
     RawHeader rh;
     RawRecord rr;
     //      SStat        sstats;
     PStat *pstats;
     QList<long> pids; // This is a list of process pid's, in the exact same order as pstats
     QString lastError;
 
     QList<long> historyOffsets; //< The file offset where each history is stored
     QList<QPair<QDateTime, uint>> historyTimes; //< The end time for each history record and its interval, probably in order from oldest to newest
     int currentlySelectedIndex;
 };
 
 ProcessesATop::Private::Private()
     : ready(false)
     , pstats(nullptr)
     , currentlySelectedIndex(-1)
 {
 }
 
 ProcessesATop::Private::~Private()
 {
 }
 
 QString ProcessesATop::historyFileName() const
 {
     return d->atopLog.fileName();
 }
 
 bool ProcessesATop::loadHistoryFile(const QString &filename)
 {
     return d->loadHistoryFile(filename);
 }
 
 bool ProcessesATop::Private::loadHistoryFile(const QString &filename)
 {
     atopLog.setFileName(filename);
     ready = false;
     currentlySelectedIndex = -1;
     if (!atopLog.exists()) {
         lastError = QLatin1String("File ") + filename + QLatin1String(" does not exist");
         return false;
     }
 
     if (!atopLog.open(QIODevice::ReadOnly)) {
         lastError = QLatin1String("Could not open file ") + filename;
         return false;
     }
 
     int sizeRead = atopLog.read((char *)(&rh), sizeof(RawHeader));
     if (sizeRead != sizeof(RawHeader)) {
         lastError = QLatin1String("Could not read header from file ") + filename;
         return false;
     }
     if (rh.magic != ATOPLOGMAGIC) {
         lastError = QLatin1String("File ") + filename + QLatin1String(" does not contain raw atop/atopsar output (wrong magic number)");
         return false;
     }
     if (/*rh.sstatlen   != sizeof(SStat)    ||*/
         rh.pstatlen != sizeof(PStat) || rh.rawheadlen != sizeof(RawHeader) || rh.rawreclen != sizeof(RawRecord)) {
         lastError = QLatin1String("File ") + filename + QLatin1String(" has incompatible format");
         if (rh.aversion & 0x8000) {
             lastError = QStringLiteral("(created by version %1.%2. This program understands the format written by version 1.23")
                             .arg((rh.aversion >> 8) & 0x7f)
                             .arg(rh.aversion & 0xff);
         }
         return false;
     }
 
     /* Read the first data header */
     int offset = atopLog.pos();
     historyTimes.clear();
     historyOffsets.clear();
     while (!atopLog.atEnd() && atopLog.read((char *)(&rr), sizeof(RawRecord)) == sizeof(RawRecord)) {
         historyOffsets << offset;
         historyTimes << QPair<QDateTime, uint>(QDateTime::fromSecsSinceEpoch(rr.curtime), rr.interval);
         offset += sizeof(RawRecord) + rr.scomplen + rr.pcomplen;
         atopLog.seek(offset);
     }
     if (currentlySelectedIndex >= historyOffsets.size())
         currentlySelectedIndex = historyOffsets.size() - 1;
 
     ready = true;
     return true;
 }
 
 bool ProcessesATop::Private::loadDataForHistory(int index)
 {
     delete[] pstats;
     pstats = nullptr;
     atopLog.seek(historyOffsets.at(index));
     /*Read the first data header */
     if (atopLog.read((char *)(&rr), sizeof(RawRecord)) != sizeof(RawRecord)) {
         lastError = QStringLiteral("Could not read data header");
         return false;
     }
 
     if (historyTimes.at(index).first != QDateTime::fromSecsSinceEpoch(rr.curtime) || historyTimes.at(index).second != rr.interval) {
         lastError = QStringLiteral("INTERNAL ERROR WITH loadDataForHistory");
         ready = false;
         return false;
     }
 
     atopLog.seek(atopLog.pos() + rr.scomplen);
     QByteArray processRecord;
     processRecord.resize(rr.pcomplen);
     //    qToBigEndian( rr.pcomplen, (uchar*)processRecord.data() );
     unsigned int dataRead = 0;
     do {
         int ret = atopLog.read(processRecord.data() + dataRead, rr.pcomplen - dataRead);
         if (ret == -1) {
             lastError = QStringLiteral("Stream interrupted while being read");
             return false;
         }
         dataRead += ret;
     } while (dataRead < rr.pcomplen);
     Q_ASSERT(dataRead == rr.pcomplen);
     // Q_ASSERT( (index + 1 ==historyTimes.count()) || atopLog.pos() == historyTimes.at(index+1));
 
     pstats = new PStat[rr.nlist];
     unsigned long uncompressedLength = sizeof(struct PStat) * rr.nlist;
     int ret = uncompress((Byte *)pstats, &uncompressedLength, (Byte *)processRecord.constData(), rr.pcomplen);
     if (ret != Z_OK && ret != Z_STREAM_END && ret != Z_NEED_DICT) {
         switch (ret) {
         case Z_MEM_ERROR:
             lastError = QStringLiteral("Could not uncompress record data due to lack of memory");
             break;
         case Z_BUF_ERROR:
             lastError = QStringLiteral("Could not uncompress record data due to lack of room in buffer");
             break;
         case Z_DATA_ERROR:
             lastError = QStringLiteral("Could not uncompress record data due to corrupted data");
             break;
         default:
             lastError = QLatin1String("Could not uncompress record data due to unexpected error: ") + QString::number(ret);
             break;
         }
         delete[] pstats;
         pstats = nullptr;
         return false;
     }
 
     pids.clear();
     for (uint i = 0; i < rr.nlist; i++) {
         pids << pstats[i].gen.pid;
     }
     return true;
 }
 
 ProcessesATop::ProcessesATop(bool loadDefaultFile)
     : d(new Private())
 {
     if (loadDefaultFile)
         loadHistoryFile(QStringLiteral("/var/log/atop.log"));
 }
 
 bool ProcessesATop::isHistoryAvailable() const
 {
     return d->ready;
 }
 
 long ProcessesATop::getParentPid(long pid)
 {
     int index = d->pids.indexOf(pid);
     if (index < 0)
         return 0;
     return d->pstats[index].gen.ppid;
 }
 
 bool ProcessesATop::updateProcessInfo(long pid, Process *process)
 {
     int index = d->pids.indexOf(pid);
     if (index < 0)
         return false;
     PStat &p = d->pstats[index];
     process->setParentPid(p.gen.ppid);
     process->setUid(p.gen.ruid);
     process->setEuid(p.gen.ruid);
     process->setSuid(p.gen.ruid);
     process->setFsuid(p.gen.ruid);
     process->setGid(p.gen.rgid);
     process->setEgid(p.gen.rgid);
     process->setSgid(p.gen.rgid);
     process->setFsgid(p.gen.rgid);
     process->setTracerpid(-1);
     process->setNumThreads(p.gen.nthr);
     //    process->setTty
     process->setUserTime(p.cpu.utime * 100 / d->rh.hertz); // check - divide by interval maybe?
     process->setSysTime(p.cpu.stime * 100 / d->rh.hertz); // check
     process->setUserUsage(process->userTime() / d->rr.interval);
     process->setSysUsage(process->sysTime() / d->rr.interval);
     process->setNiceLevel(p.cpu.nice);
     //    process->setscheduler(p.cpu.policy);
     process->setVmSize(p.mem.vmem);
     process->setVmRSS(p.mem.rmem);
     process->vmSizeChange() = p.mem.vgrow;
     process->vmRSSChange() = p.mem.rgrow;
     process->setVmURSS(0);
     process->vmURSSChange() = 0;
 
     /* Fill in name and command */
     QString name = QString::fromUtf8(p.gen.name, qstrnlen(p.gen.name, PNAMLEN));
     QString command = QString::fromUtf8(p.gen.cmdline, qstrnlen(p.gen.cmdline, CMDLEN));
     // cmdline separates parameters with the NULL character
     if (!command.isEmpty()) {
         if (command.startsWith(name)) {
             int index = command.indexOf(QLatin1Char('\0'));
             name = command.left(index);
         }
         command.replace(QLatin1Char('\0'), QLatin1Char(' '));
     }
     process->setName(name);
     process->setCommand(command);
 
     /* Fill in state */
     switch (p.gen.state) {
     case 'E':
         process->setStatus(Process::Ended);
         break;
     case 'R':
         process->setStatus(Process::Running);
         break;
     case 'S':
         process->setStatus(Process::Sleeping);
         break;
     case 'D':
         process->setStatus(Process::DiskSleep);
         break;
     case 'Z':
         process->setStatus(Process::Zombie);
         break;
     case 'T':
         process->setStatus(Process::Stopped);
         break;
     case 'W':
         process->setStatus(Process::Paging);
         break;
     default:
         process->setStatus(Process::OtherStatus);
         break;
     }
 
     return true;
 }
 QDateTime ProcessesATop::viewingTime() const
 {
     if (!d->ready)
         return QDateTime();
     return d->historyTimes.at(d->currentlySelectedIndex).first;
 }
 bool ProcessesATop::setViewingTime(const QDateTime &when)
 {
     QPair<QDateTime, uint> tmpWhen(when, 0);
     QList<QPair<QDateTime, uint>>::iterator i = std::upper_bound(d->historyTimes.begin(), d->historyTimes.end(), tmpWhen);
 
     if (i->first == when || (i->first > when && i->first.addSecs(-i->second) <= when)) {
         // We found the time :)
         d->currentlySelectedIndex = i - d->historyTimes.begin();
         bool success = d->loadDataForHistory(d->currentlySelectedIndex);
         if (!success)
             qCWarning(LIBKSYSGUARD_PROCESSCORE) << d->lastError;
         return success;
     }
     return false;
 }
 QList<QPair<QDateTime, uint>> ProcessesATop::historiesAvailable() const
 {
     return d->historyTimes;
 }
 
 QSet<long> ProcessesATop::getAllPids()
 {
     const QSet<long> pids(d->pids.cbegin(), d->pids.cend());
     return pids;
 }
 
 Processes::Error ProcessesATop::sendSignal(long pid, int sig)
 {
     Q_UNUSED(pid);
     Q_UNUSED(sig);
 
     return Processes::NotSupported;
 }
 
 Processes::Error ProcessesATop::setNiceness(long pid, int priority)
 {
     Q_UNUSED(pid);
     Q_UNUSED(priority);
 
     return Processes::NotSupported;
 }
 
 Processes::Error ProcessesATop::setScheduler(long pid, int priorityClass, int priority)
 {
     Q_UNUSED(pid);
     Q_UNUSED(priorityClass);
     Q_UNUSED(priority);
 
     return Processes::NotSupported;
 }
 
 Processes::Error ProcessesATop::setIoNiceness(long pid, int priorityClass, int priority)
 {
     Q_UNUSED(pid);
     Q_UNUSED(priorityClass);
     Q_UNUSED(priority);
 
     return Processes::NotSupported;
 }
 
 bool ProcessesATop::supportsIoNiceness()
 {
     return false;
 }
 
 long long ProcessesATop::totalPhysicalMemory()
 {
     return 0;
 }
 
 ProcessesATop::~ProcessesATop()
 {
     delete d;
 }
 
 }