Differential D23342 Diff 64293 plugins/process/network/helper/main.cpp

Changeset View

Standalone View

plugins/process/network/helper/main.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 <iomanip>
23		#include <iostream>
24		#include <thread>
25		#include <atomic>
26		#include <csignal>
27
28		#include <getopt.h>
29
30		#include "Accumulator.h"
31		#include "Capture.h"
32		#include "ConnectionMapping.h"
33		#include "Packet.h"
34		#include "TimeStamps.h"
35
36		static std::atomic_bool g_running{false};
37
38		int main(int argc, char **argv)
39		{
40		static struct option long_options[] = {
41		{"help", 0, nullptr, 'h'},
42		{"stats", 0, nullptr, 's'},
43		{nullptr, 0, nullptr, 0}
44		};
45
46		auto statsRequested = false;
47		auto optionIndex = 0;
48		auto option = -1;
49		while ((option = getopt_long(argc, argv, "", long_options, &optionIndex)) != -1) {
50		switch (option) {
51		case 's':
52		statsRequested = true;
53		break;
54		default:
55		std::cerr << "Usage: " << argv[0] << " [options]\n";
56		std::cerr << "This is a helper application for tracking per-process network usage.\n";
57		std::cerr << "\n";
58		std::cerr << "Options:\n";
59		std::cerr << " --stats Print packet capture statistics.\n";
60		std::cerr << " --help Display this help.\n";
61		return 0;
62		}
63		}
64
65		auto mapping = std::make_shared<ConnectionMapping>();
66
67		auto capture = std::make_shared<Capture>();
68		if (!capture->start()) {
69		std::cerr << capture->lastError() << std::endl;
70		return 1;
71		}
72
73		auto accumulator = std::make_shared<Accumulator>(capture, mapping);
74
75		signal(SIGINT, [](int) { g_running = false; });
76		signal(SIGTERM, [](int) { g_running = false; });
77
78		g_running = true;
79		while(g_running) {
80		auto data = accumulator->data();
81		auto timeStamp = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
82
83		if (statsRequested != 0) {
84		capture->reportStatistics();
85		}
86
87		if (data.empty()) {
88		std::cout << std::put_time(std::localtime(&timeStamp), "%T") << std::endl;
89		} else {
90		for (auto itr = data.begin(); itr != data.end(); ++itr) {
91		std::cout << std::put_time(std::localtime(&timeStamp), "%T");
92		std::cout << "\|PID\|" << (itr).first << "\|IN\|" << (itr).second.first << "\|OUT\|" << (*itr).second.second;
93		std::cout << std::endl;
94		}
95		}
96
97		std::this_thread::sleep_for(std::chrono::seconds(1));
98		}
	mevenUnsubmitted Not Done Could the refresh rate be a parameter or at least a constant instead of an hardcoded value. meven: Could the refresh rate be a parameter or at least a constant instead of an hardcoded value.
	ahiemstraAuthorUnsubmitted Not Done The helper reports a download/upload rate per second. This is done by capturing the amount of incoming/outgoing bytes and each second reporting that value, then resetting the counter. Changing this value would thus result in changing the reported unit, so I'd rather keep it a hardcoded constant here. ahiemstra: The helper reports a download/upload rate per second. This is done by capturing the amount of…
99
100		accumulator->stop();
101		capture->stop();
102
103		return 0;
104		}