Differential D23342 Diff 64293 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
30		#include <KLocalizedString>
31		#include <KPluginFactory>
32
33		#include "networklogging.h"
34
35		using namespace KSysGuard;
36
37		NetworkPlugin::NetworkPlugin(QObject *parent, const QVariantList &args)
38		: ProcessDataProvider(parent, args)
39		{
40		const auto executable = QStandardPaths::findExecutable(QStringLiteral("ksgrd_network_helper"));
41		if (executable.isEmpty()) {
42		qCWarning(KSYSGUARD_PLUGIN_NETWORK) << "Could not find ksgrd_network_helper";
43		return;
44		}
45
46		qCDebug(KSYSGUARD_PLUGIN_NETWORK) << "Network plugin loading";
47
48		m_inboundSensor = new ProcessAttribute(QStringLiteral("netInbound"), i18n("Download Speed"), this);
49		m_inboundSensor->setShortName(i18n("Download"));
50		m_inboundSensor->setUnit(KSysGuard::UnitByteRate);
51		m_outboundSensor = new ProcessAttribute(QStringLiteral("netOutbound"), i18n("Upload Speed"), this);
52		m_outboundSensor->setShortName(i18n("Upload"));
53		m_outboundSensor->setUnit(KSysGuard::UnitByteRate);
54
55		addProcessAttribute(m_inboundSensor);
56		addProcessAttribute(m_outboundSensor);
57
58		m_process = new QProcess(this);
59		m_process->setProgram(executable);
60
61		connect(m_process, QOverload<int, QProcess::ExitStatus>::of(&QProcess::finished), [=](int exitCode, QProcess::ExitStatus status) {
62		if (exitCode != 0 \|\| status != QProcess::NormalExit) {
63		qCWarning(KSYSGUARD_PLUGIN_NETWORK) << "Helper process terminated abnormally!";
64		qCWarning(KSYSGUARD_PLUGIN_NETWORK) << m_process->readAllStandardError();
65		}
66		});
67
68		connect(m_process, &QProcess::readyReadStandardOutput, this, [=]() {
69		while (m_process->canReadLine()) {
70		const QString line = QString::fromUtf8(m_process->readLine());
71
72		// Each line consists of: timestamp\|PID\|pid\|IN\|in_bytes\|OUT\|out_bytes
73		const auto parts = line.splitRef(QLatin1Char('\|'), QString::SkipEmptyParts);
74		if (parts.size() < 2) {
75		continue;
	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		}
77
78		long pid = parts.at(2).toLong();
79
80		auto timeStamp = QDateTime::currentDateTimeUtc();
81		timeStamp.setTime(QTime::fromString(parts.at(0).toString(), QStringLiteral("HH:mm:ss")));
82
83		auto bytesIn = parts.at(4).toUInt();
84		auto bytesOut = parts.at(6).toUInt();
85
86		auto process = getProcess(pid);
87		if (!process) {
88		return;
89		}
90
91		m_inboundSensor->setData(process, bytesIn);
92		m_outboundSensor->setData(process, bytesOut);
93		}
94		});
95		}
96
97		void NetworkPlugin::handleEnabledChanged(bool enabled)
98		{
99		if (enabled) {
100		qCDebug(KSYSGUARD_PLUGIN_NETWORK) << "Network plugin enabled, starting helper";
101		m_process->start();
102		} else {
103		qCDebug(KSYSGUARD_PLUGIN_NETWORK) << "Network plugin disabled, stopping helper";
104		m_process->terminate();
105		}
106		}
107
108		K_PLUGIN_FACTORY_WITH_JSON(PluginFactory, "networkplugin.json", registerPlugin<NetworkPlugin>();)
109
110		#include "network.moc"