Differential D23342 Diff 66195 plugins/process/network/network.cpp

Changeset View

Standalone View

plugins/process/network/network.cpp

This file was added.

1		/*
2		* This file is part of KSysGuard.
3		* Copyright 2019 Arjen Hiemstra <ahiemstra@heimr.nl>
4		*
5		* This program is free software; you can redistribute it and/or
6		* modify it under the terms of the GNU General Public License as
7		* published by the Free Software Foundation; either version 2 of
8		* the License or (at your option) version 3 or any later version
9		* accepted by the membership of KDE e.V. (or its successor approved
10		* by the membership of KDE e.V.), which shall act as a proxy
11		* defined in Section 14 of version 3 of the license.
12		*
13		* This program is distributed in the hope that it will be useful,
14		* but WITHOUT ANY WARRANTY; without even the implied warranty of
15		* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16		* GNU General Public License for more details.
17		*
18		* You should have received a copy of the GNU General Public License
19		* along with this program. If not, see <http://www.gnu.org/licenses/>.
20		*/
21
22		#include "network.h"
23
24		#include <QDateTime>
25		#include <QDebug>
26		#include <QHash>
27		#include <QProcess>
28		#include <QStandardPaths>
29		#include <QFile>
30
31		#include <KLocalizedString>
32		#include <KPluginFactory>
33
34		#include "networklogging.h"
35		#include "networkconstants.h"
36
37		using namespace KSysGuard;
38
39		NetworkPlugin::NetworkPlugin(QObject *parent, const QVariantList &args)
40		: ProcessDataProvider(parent, args)
41		{
42		const auto executable = NetworkConstants::HelperLocation;
43		if (!QFile::exists(executable)) {
44		qCWarning(KSYSGUARD_PLUGIN_NETWORK) << "Could not find ksgrd_network_helper";
45		return;
46		}
47
48		qCDebug(KSYSGUARD_PLUGIN_NETWORK) << "Network plugin loading";
49		qCDebug(KSYSGUARD_PLUGIN_NETWORK) << "Found helper at" << qPrintable(executable);
50
51		m_inboundSensor = new ProcessAttribute(QStringLiteral("netInbound"), i18n("Download Speed"), this);
52		m_inboundSensor->setShortName(i18n("Download"));
53		m_inboundSensor->setUnit(KSysGuard::UnitByteRate);
54		m_outboundSensor = new ProcessAttribute(QStringLiteral("netOutbound"), i18n("Upload Speed"), this);
55		m_outboundSensor->setShortName(i18n("Upload"));
56		m_outboundSensor->setUnit(KSysGuard::UnitByteRate);
57
58		addProcessAttribute(m_inboundSensor);
59		addProcessAttribute(m_outboundSensor);
60
61		m_process = new QProcess(this);
62		m_process->setProgram(executable);
63
64		connect(m_process, QOverload<int, QProcess::ExitStatus>::of(&QProcess::finished), [=](int exitCode, QProcess::ExitStatus status) {
65		if (exitCode != 0 \|\| status != QProcess::NormalExit) {
66		qCWarning(KSYSGUARD_PLUGIN_NETWORK) << "Helper process terminated abnormally!";
67		qCWarning(KSYSGUARD_PLUGIN_NETWORK) << m_process->readAllStandardError();
68		}
69		});
70
71		connect(m_process, &QProcess::readyReadStandardOutput, this, [=]() {
72		while (m_process->canReadLine()) {
73		const QString line = QString::fromUtf8(m_process->readLine());
74
75		// Each line consists of: timestamp\|PID\|pid\|IN\|in_bytes\|OUT\|out_bytes
	davidedmundsonUnsubmitted Done why 2? We call parts.at(6) Also code style {} davidedmundson: why 2? We call parts.at(6) Also code style {}
	ahiemstraAuthorUnsubmitted Not Done Good point. I cannot recall why I used 2 so changed it to 7 now. ahiemstra: Good point. I cannot recall why I used 2 so changed it to 7 now.
76		const auto parts = line.splitRef(QLatin1Char('\|'), QString::SkipEmptyParts);
77		if (parts.size() < 7) {
78		continue;
79		}
80
81		long pid = parts.at(2).toLong();
82
83		auto timeStamp = QDateTime::currentDateTimeUtc();
84		timeStamp.setTime(QTime::fromString(parts.at(0).toString(), QStringLiteral("HH:mm:ss")));
85
86		auto bytesIn = parts.at(4).toUInt();
87		auto bytesOut = parts.at(6).toUInt();
88
89		auto process = getProcess(pid);
90		if (!process) {
91		return;
92		}
93
94		m_inboundSensor->setData(process, bytesIn);
95		m_outboundSensor->setData(process, bytesOut);
96		}
97		});
98		}
99
100		void NetworkPlugin::handleEnabledChanged(bool enabled)
101		{
102		if (enabled) {
103		qCDebug(KSYSGUARD_PLUGIN_NETWORK) << "Network plugin enabled, starting helper";
104		m_process->start();
105		} else {
106		qCDebug(KSYSGUARD_PLUGIN_NETWORK) << "Network plugin disabled, stopping helper";
107		m_process->terminate();
108		}
109		}
110
111		K_PLUGIN_FACTORY_WITH_JSON(PluginFactory, "networkplugin.json", registerPlugin<NetworkPlugin>();)
112
113		#include "network.moc"