From 5dbad7a3cb16be6e317bf4d7cef15691d133d198 Mon Sep 17 00:00:00 2001
From: alex <rwrx@foxmail.com>
Date: Fri, 23 Aug 2024 23:19:42 +0800
Subject: [PATCH] feat: cycle flush view

---
 include/mainwindow.h        |  2 +-
 include/packetsource.h      |  1 +
 src/packet/packetsource.cpp |  3 +++
 src/window/mainwindow.cpp   | 48 ++++++++++++++++++++++---------------
 4 files changed, 34 insertions(+), 20 deletions(-)

diff --git a/include/mainwindow.h b/include/mainwindow.h
index ce73647..e71168d 100644
--- a/include/mainwindow.h
+++ b/include/mainwindow.h
@@ -43,12 +43,12 @@ class MainWindow : public QMainWindow {
     void tableItemClicked(const QModelIndex& index);
     void toggleStartBtn();
     void initWidgets();
-    void updateCaptureStatusLabel() const;
     void initInterfaceList();
     void about();
     void activateStatsWindow() const;
     void saveAsPcap();
     void initMenus();
+    void updateMajorView(size_t, size_t) const;
 
 private:
     Ui::MainWindow* ui;
diff --git a/include/packetsource.h b/include/packetsource.h
index dd46849..1fdcf1f 100644
--- a/include/packetsource.h
+++ b/include/packetsource.h
@@ -91,6 +91,7 @@ class PacketSource final : public QObject {
     void listen_started(PACKETSOURCE_STATE) const;
     void listen_stopped(PACKETSOURCE_STATE) const;
     void captured(size_t, Packet*);
+    void capture_cycle_flush(size_t, size_t);
 
 public:
     explicit PacketSource();
diff --git a/src/packet/packetsource.cpp b/src/packet/packetsource.cpp
index 030a272..94b47c2 100644
--- a/src/packet/packetsource.cpp
+++ b/src/packet/packetsource.cpp
@@ -126,6 +126,9 @@ void PacketSource::consume_queue()
             emit this->captured(history.size() - 1, p);
             dump_flush(p->get_header(), p->get_payload());
         }
+
+        // 一个捕获周期结束
+        emit this->capture_cycle_flush(period_average, history.size() - 1);
     }
 }
 
diff --git a/src/window/mainwindow.cpp b/src/window/mainwindow.cpp
index 66191cd..4c30cc1 100644
--- a/src/window/mainwindow.cpp
+++ b/src/window/mainwindow.cpp
@@ -136,7 +136,6 @@ void MainWindow::captureInterfaceStarted(packetsource_state state)
 
     ui->layerTree->clear();
     interfaceStatusLabel->setText(state.interface_name.append(": ").append(state.state).c_str());
-    updateCaptureStatusLabel();
 }
 
 void MainWindow::captureInterfaceStopped(packetsource_state state) const
@@ -145,8 +144,6 @@ void MainWindow::captureInterfaceStopped(packetsource_state state) const
     ui->resetBtn->setDisabled(true);
     ui->startBtn->setText("Start");
     interfaceStatusLabel->setText(state.interface_name.append(": ").append(state.state).c_str());
-
-    updateCaptureStatusLabel();
 }
 
 void MainWindow::resetCapture()
@@ -262,17 +259,6 @@ void MainWindow::initWidgets()
     }
 }
 
-void MainWindow::updateCaptureStatusLabel() const
-{
-    size_t count = packetSource->packet_count();
-    if (count == 0) {
-        captureStatusLabel->setText("");
-        return;
-    }
-
-    captureStatusLabel->setText("packets: " + QString::number(count) + "/" + QString::number(count));
-}
-
 void MainWindow::acceptPacket(const int row, Packet* packet) const
 {
     string src = packet->get_host_src();
@@ -310,11 +296,6 @@ void MainWindow::acceptPacket(const int row, Packet* packet) const
     ui->packetsTable->setItem(row, 4, item4);
     ui->packetsTable->setItem(row, 5, item5);
     ui->packetsTable->setItem(row, 6, item6);
-
-    // 滚动到最底部，目前会导致表格卡顿和程序崩溃
-    // ui->packetsTable->verticalScrollBar()->setValue(ui->packetsTable->verticalScrollBar()->maximum());
-
-    updateCaptureStatusLabel();
 }
 
 void MainWindow::initSlots()
@@ -325,6 +306,7 @@ void MainWindow::initSlots()
     connect(packetSource, &PacketSource::listen_started, this, &MainWindow::captureInterfaceStarted);
     connect(packetSource, &PacketSource::listen_stopped, this, &MainWindow::captureInterfaceStopped);
     connect(packetSource, &PacketSource::captured, this, &MainWindow::acceptPacket);
+    connect(packetSource, &PacketSource::capture_cycle_flush, this, &MainWindow::updateMajorView);
     connect(ui->packetsTable, &QTableWidget::clicked, this, &MainWindow::tableItemClicked);
     connect(ui->loadFileBtn, &QPushButton::clicked, this, &MainWindow::loadOfflineFile);
 }
@@ -350,6 +332,34 @@ void MainWindow::loadOfflineFile() const
     packetSource->start_on_interface(nullptr, interface);
 }
 
+void MainWindow::updateMajorView(size_t period_average, size_t sum_capture) const
+{
+    // 滚动到最底部，大流量会导致表格卡顿和程序崩溃
+    ui->packetsTable->verticalScrollBar()->setValue(ui->packetsTable->verticalScrollBar()->maximum());
+
+    size_t count = packetSource->packet_count();
+    if (count == 0) {
+        captureStatusLabel->setText("");
+        return;
+    }
+
+    pcap_stat stats {};
+    pcap_stats(packetSource->get_interface(), &stats);
+
+    string msg = std::format("captured: {} droped: {}", count, stats.ps_drop);
+    captureStatusLabel->setText(msg.c_str());
+
+    // 悬浮气泡
+    std::ostringstream tip;
+    auto duration = std::chrono::high_resolution_clock::now() - time_start;
+    tip << count << " captured in " << std::chrono::duration_cast<std::chrono::seconds>(duration).count() << " seconds" << std::endl;
+    tip << stats.ps_recv << " received by filter" << std::endl;
+    tip << stats.ps_ifdrop << " dropped by interface" << std::endl;
+    tip << stats.ps_drop << " dropped by kernel" << std::endl;
+
+    captureStatusLabel->setToolTip(tip.str().c_str());
+}
+
 void MainWindow::tableItemClicked(const QModelIndex& index)
 {
     auto packet = packetSource->peek(index.row());