1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-26 08:57:34 +00:00

ProcessManager: Add a "Network" tab with live adapter and socket stats

This fetches info from /proc/netadapters and /proc/net_tcp, updating
every second. Very cool. :^)
This commit is contained in:
Andreas Kling 2019-08-08 20:43:30 +02:00
parent 899366da9d
commit d6bce37756
8 changed files with 371 additions and 0 deletions

View file

@ -10,6 +10,9 @@ OBJS = \
ProcessMemoryMapModel.o \ ProcessMemoryMapModel.o \
ProcessFileDescriptorMapWidget.o \ ProcessFileDescriptorMapWidget.o \
ProcessFileDescriptorMapModel.o \ ProcessFileDescriptorMapModel.o \
NetworkStatisticsWidget.o \
NetworkAdapterModel.o \
SocketModel.o \
main.o main.o
APP = ProcessManager APP = ProcessManager

View file

@ -0,0 +1,102 @@
#include "NetworkAdapterModel.h"
#include <AK/JsonObject.h>
#include <AK/StringBuilder.h>
#include <LibCore/CFile.h>
void NetworkAdapterModel::update()
{
CFile file("/proc/netadapters");
if (!file.open(CIODevice::ReadOnly)) {
dbg() << "Unable to open " << file.filename();
return;
}
auto json = JsonValue::from_string(file.read_all());
ASSERT(json.is_array());
m_netadapters = json.as_array();
did_update();
}
int NetworkAdapterModel::row_count(const GModelIndex&) const
{
return m_netadapters.size();
}
String NetworkAdapterModel::column_name(int column) const
{
switch (column) {
case Column::Name:
return "Name";
case Column::ClassName:
return "Class";
case Column::MacAddress:
return "MAC";
case Column::IpAddress:
return "IP";
case Column::PacketsIn:
return "Pkt In";
case Column::PacketsOut:
return "Pkt Out";
case Column::BytesIn:
return "Bytes In";
case Column::BytesOut:
return "Bytes Out";
default:
ASSERT_NOT_REACHED();
}
}
GModel::ColumnMetadata NetworkAdapterModel::column_metadata(int column) const
{
switch (column) {
case Column::Name:
return { 32, TextAlignment::CenterLeft };
case Column::ClassName:
return { 120, TextAlignment::CenterLeft };
case Column::MacAddress:
return { 90, TextAlignment::CenterLeft };
case Column::IpAddress:
return { 80, TextAlignment::CenterLeft };
case Column::PacketsIn:
return { 60, TextAlignment::CenterRight };
case Column::PacketsOut:
return { 60, TextAlignment::CenterRight };
case Column::BytesIn:
return { 60, TextAlignment::CenterRight };
case Column::BytesOut:
return { 60, TextAlignment::CenterRight };
default:
ASSERT_NOT_REACHED();
}
return {};
}
GVariant NetworkAdapterModel::data(const GModelIndex& index, Role role) const
{
auto& adapter_object = m_netadapters.at(index.row()).as_object();
if (role == GModel::Role::Display) {
switch (index.column()) {
case Column::Name:
return adapter_object.get("name").to_string();
case Column::ClassName:
return adapter_object.get("class_name").to_string();
case Column::MacAddress:
return adapter_object.get("mac_address").to_string();
case Column::IpAddress:
return adapter_object.get("ipv4_address").to_string();
case Column::PacketsIn:
return adapter_object.get("packets_in").to_u32();
case Column::PacketsOut:
return adapter_object.get("packets_out").to_u32();
case Column::BytesIn:
return adapter_object.get("bytes_in").to_u32();
case Column::BytesOut:
return adapter_object.get("bytes_out").to_u32();
default:
ASSERT_NOT_REACHED();
}
}
return {};
}

View file

@ -0,0 +1,33 @@
#pragma once
#include <AK/JsonArray.h>
#include <LibGUI/GModel.h>
class NetworkAdapterModel final : public GModel {
public:
enum Column {
Name,
ClassName,
MacAddress,
IpAddress,
PacketsIn,
PacketsOut,
BytesIn,
BytesOut,
__Count
};
static NonnullRefPtr<NetworkAdapterModel> create() { return adopt(*new NetworkAdapterModel); }
virtual ~NetworkAdapterModel() override {}
virtual int row_count(const GModelIndex& = GModelIndex()) const override;
virtual int column_count(const GModelIndex& = GModelIndex()) const override { return Column::__Count; }
virtual String column_name(int) const override;
virtual ColumnMetadata column_metadata(int) const override;
virtual GVariant data(const GModelIndex&, Role = Role::Display) const override;
virtual void update() override;
private:
NetworkAdapterModel() {}
JsonArray m_netadapters;
};

View file

@ -0,0 +1,51 @@
#include "NetworkStatisticsWidget.h"
#include "NetworkAdapterModel.h"
#include "SocketModel.h"
#include <LibGUI/GBoxLayout.h>
#include <LibGUI/GGroupBox.h>
#include <LibGUI/GTableView.h>
NetworkStatisticsWidget::NetworkStatisticsWidget(GWidget* parent)
: GWidget(parent)
{
set_layout(make<GBoxLayout>(Orientation::Vertical));
layout()->set_margins({ 4, 4, 4, 4 });
set_fill_with_background_color(true);
set_background_color(Color::WarmGray);
auto* adapters_group_box = new GGroupBox("Adapters", this);
adapters_group_box->set_layout(make<GBoxLayout>(Orientation::Vertical));
adapters_group_box->layout()->set_margins({ 6, 16, 6, 6 });
adapters_group_box->set_size_policy(SizePolicy::Fill, SizePolicy::Fixed);
adapters_group_box->set_preferred_size(0, 120);
m_adapter_table_view = new GTableView(adapters_group_box);
m_adapter_table_view->set_model(NetworkAdapterModel::create());
auto* sockets_group_box = new GGroupBox("Sockets", this);
sockets_group_box->set_layout(make<GBoxLayout>(Orientation::Vertical));
sockets_group_box->layout()->set_margins({ 6, 16, 6, 6 });
sockets_group_box->set_size_policy(SizePolicy::Fill, SizePolicy::Fill);
sockets_group_box->set_preferred_size(0, 0);
m_socket_table_view = new GTableView(sockets_group_box);
m_socket_table_view->set_model(SocketModel::create());
m_update_timer = new CTimer(
1000, [this] {
update_models();
},
this);
update_models();
}
NetworkStatisticsWidget::~NetworkStatisticsWidget()
{
}
void NetworkStatisticsWidget::update_models()
{
m_adapter_table_view->model()->update();
m_socket_table_view->model()->update();
}

View file

@ -0,0 +1,20 @@
#pragma once
#include <LibCore/CTimer.h>
#include <LibGUI/GWidget.h>
class GTableView;
class NetworkStatisticsWidget final : public GWidget {
C_OBJECT(NetworkStatisticsWidget)
public:
explicit NetworkStatisticsWidget(GWidget* parent = nullptr);
virtual ~NetworkStatisticsWidget() override;
private:
void update_models();
GTableView* m_adapter_table_view { nullptr };
GTableView* m_socket_table_view { nullptr };
CTimer* m_update_timer { nullptr };
};

View file

@ -0,0 +1,120 @@
#include "SocketModel.h"
#include <AK/JsonObject.h>
#include <AK/StringBuilder.h>
#include <LibCore/CFile.h>
void SocketModel::update()
{
CFile file("/proc/net_tcp");
if (!file.open(CIODevice::ReadOnly)) {
dbg() << "Unable to open " << file.filename();
return;
}
auto json = JsonValue::from_string(file.read_all());
ASSERT(json.is_array());
m_sockets = json.as_array();
did_update();
}
int SocketModel::row_count(const GModelIndex&) const
{
return m_sockets.size();
}
String SocketModel::column_name(int column) const
{
switch (column) {
case Column::PeerAddress:
return "Peer";
case Column::PeerPort:
return "Port";
case Column::LocalAddress:
return "Local";
case Column::LocalPort:
return "Port";
case Column::State:
return "State";
case Column::SeqNumber:
return "Seq#";
case Column::AckNumber:
return "Ack#";
case Column::PacketsIn:
return "Pkt In";
case Column::PacketsOut:
return "Pkt Out";
case Column::BytesIn:
return "Bytes In";
case Column::BytesOut:
return "Bytes Out";
default:
ASSERT_NOT_REACHED();
}
}
GModel::ColumnMetadata SocketModel::column_metadata(int column) const
{
switch (column) {
case Column::PeerAddress:
return { 80, TextAlignment::CenterLeft };
case Column::PeerPort:
return { 30, TextAlignment::CenterRight };
case Column::LocalAddress:
return { 80, TextAlignment::CenterLeft };
case Column::LocalPort:
return { 30, TextAlignment::CenterRight };
case Column::State:
return { 80, TextAlignment::CenterLeft };
case Column::AckNumber:
return { 60, TextAlignment::CenterRight };
case Column::SeqNumber:
return { 60, TextAlignment::CenterRight };
case Column::PacketsIn:
return { 60, TextAlignment::CenterRight };
case Column::PacketsOut:
return { 60, TextAlignment::CenterRight };
case Column::BytesIn:
return { 60, TextAlignment::CenterRight };
case Column::BytesOut:
return { 60, TextAlignment::CenterRight };
default:
ASSERT_NOT_REACHED();
}
return {};
}
GVariant SocketModel::data(const GModelIndex& index, Role role) const
{
auto& socket_object = m_sockets.at(index.row()).as_object();
if (role == GModel::Role::Display) {
switch (index.column()) {
case Column::PeerAddress:
return socket_object.get("peer_address").to_string();
case Column::PeerPort:
return socket_object.get("peer_port").to_u32();
case Column::LocalAddress:
return socket_object.get("local_address").to_string();
case Column::LocalPort:
return socket_object.get("local_port").to_u32();
case Column::State:
return socket_object.get("state").to_string();
case Column::AckNumber:
return socket_object.get("ack_number").to_u32();
case Column::SeqNumber:
return socket_object.get("sequence_number").to_u32();
case Column::PacketsIn:
return socket_object.get("packets_in").to_u32();
case Column::PacketsOut:
return socket_object.get("packets_out").to_u32();
case Column::BytesIn:
return socket_object.get("bytes_in").to_u32();
case Column::BytesOut:
return socket_object.get("bytes_out").to_u32();
default:
ASSERT_NOT_REACHED();
}
}
return {};
}

View file

@ -0,0 +1,38 @@
#pragma once
#include <AK/JsonArray.h>
#include <LibGUI/GModel.h>
class SocketModel final : public GModel {
public:
enum Column {
PeerAddress,
PeerPort,
LocalAddress,
LocalPort,
State,
AckNumber,
SeqNumber,
PacketsIn,
PacketsOut,
BytesIn,
BytesOut,
__Count
};
static NonnullRefPtr<SocketModel> create() { return adopt(*new SocketModel); }
virtual ~SocketModel() override {}
virtual int row_count(const GModelIndex& = GModelIndex()) const override;
virtual int column_count(const GModelIndex& = GModelIndex()) const override { return Column::__Count; }
virtual String column_name(int) const override;
virtual ColumnMetadata column_metadata(int) const override;
virtual GVariant data(const GModelIndex&, Role = Role::Display) const override;
virtual void update() override;
private:
SocketModel() {}
JsonArray m_sockets;
};

View file

@ -4,6 +4,7 @@
#include "ProcessMemoryMapWidget.h" #include "ProcessMemoryMapWidget.h"
#include "ProcessStacksWidget.h" #include "ProcessStacksWidget.h"
#include "ProcessTableView.h" #include "ProcessTableView.h"
#include "NetworkStatisticsWidget.h"
#include <LibCore/CTimer.h> #include <LibCore/CTimer.h>
#include <LibDraw/PNGLoader.h> #include <LibDraw/PNGLoader.h>
#include <LibGUI/GAction.h> #include <LibGUI/GAction.h>
@ -71,6 +72,9 @@ int main(int argc, char** argv)
tabwidget->add_widget("Graphs", graphs_container); tabwidget->add_widget("Graphs", graphs_container);
auto* network_stats_widget = new NetworkStatisticsWidget(nullptr);
tabwidget->add_widget("Network", network_stats_widget);
process_table_container->set_layout(make<GBoxLayout>(Orientation::Vertical)); process_table_container->set_layout(make<GBoxLayout>(Orientation::Vertical));
process_table_container->layout()->set_margins({ 4, 0, 4, 4 }); process_table_container->layout()->set_margins({ 4, 0, 4, 4 });
process_table_container->layout()->set_spacing(0); process_table_container->layout()->set_spacing(0);