ui/qt/traffic_table_dialog.cpp

   1 /* traffic_table_dialog.cpp
   2  *
   3  * Wireshark - Network traffic analyzer
   4  * By Gerald Combs <gerald@wireshark.org>
   5  * Copyright 1998 Gerald Combs
   6  *
   7  * SPDX-License-Identifier: GPL-2.0-or-later
   8  */
   9
  10 #include "traffic_table_dialog.h"
  11 #include <ui_traffic_table_dialog.h>
  12
  13 #include <epan/addr_resolv.h>
  14 #include <epan/prefs.h>
  15
  16 #include "ui/recent.h"
  17
  18 #include "progress_frame.h"
  19 #include "main_application.h"
  20
  21 #include <ui/qt/widgets/traffic_tab.h>
  22 #include <ui/qt/widgets/traffic_types_list.h>
  23
  24 #include <QCheckBox>
  25 #include <QClipboard>
  26 #include <QContextMenuEvent>
  27 #include <QDialogButtonBox>
  28 #include <QList>
  29 #include <QMap>
  30 #include <QMessageBox>
  31 #include <QPushButton>
  32 #include <QTabWidget>
  33 #include <QTreeWidget>
  34 #include <QTextStream>
  35 #include <QToolButton>
  36 #include <QTreeView>
  37
  38 TrafficTableDialog::TrafficTableDialog(QWidget &parent, CaptureFile &cf, const QString &table_name) :
  39     WiresharkDialog(parent, cf),
  40     ui(new Ui::TrafficTableDialog)
  41 {
  42     ui->setupUi(this);
  43     loadGeometry(parent.width(), parent.height() * 3 / 4);
  44
  45     ui->absoluteTimeCheckBox->hide();
  46     setWindowSubtitle(QStringLiteral("%1s").arg(table_name));
  47     ui->grpSettings->setTitle(QStringLiteral("%1 Settings").arg(table_name));
  48
  49     copy_bt_ = buttonBox()->addButton(tr("Copy"), QDialogButtonBox::ActionRole);
  50     copy_bt_->setMenu(ui->trafficTab->createCopyMenu(copy_bt_));
  51
  52     if (cf.displayFilter().length() > 0) {
  53         ui->displayFilterCheckBox->setChecked(true);
  54         ui->trafficTab->setFilter(cf.displayFilter());
  55     }
  56
  57     ui->trafficTab->setFocus();
  58     ui->trafficTab->useNanosecondTimestamps(cf.timestampPrecision() == WTAP_TSPREC_NSEC);
  59     connect(ui->displayFilterCheckBox, &QCheckBox::toggled, this, &TrafficTableDialog::displayFilterCheckBoxToggled);
  60     connect(ui->trafficList, &TrafficTypesList::protocolsChanged, ui->trafficTab, &TrafficTab::setOpenTabs);
  61     connect(ui->trafficTab, &TrafficTab::tabsChanged, ui->trafficList, &TrafficTypesList::selectProtocols);
  62
  63     connect(mainApp, SIGNAL(addressResolutionChanged()), this, SLOT(currentTabChanged()));
  64     connect(ui->trafficTab, SIGNAL(currentChanged(int)), this, SLOT(currentTabChanged()));
  65     connect(&cap_file_, SIGNAL(captureEvent(CaptureEvent)), this, SLOT(captureEvent(CaptureEvent)));
  66
  67     connect(ui->absoluteTimeCheckBox, &QCheckBox::toggled, ui->trafficTab, &TrafficTab::useAbsoluteTime);
  68     connect(ui->trafficTab, &TrafficTab::retapRequired, &cap_file_, &CaptureFile::delayedRetapPackets);
  69
  70     connect(ui->trafficListSearch, &QLineEdit::textChanged, ui->trafficList, &TrafficTypesList::filterList);
  71     connect(ui->trafficList, &TrafficTypesList::clearFilterList, ui->trafficListSearch, &QLineEdit::clear);
  72
  73     QPushButton *close_bt = ui->buttonBox->button(QDialogButtonBox::Close);
  74     if (close_bt)
  75         close_bt->setDefault(true);
  76
  77     addProgressFrame(&parent);
  78 }
  79
  80 TrafficTableDialog::~TrafficTableDialog()
  81 {
  82     delete ui;
  83 }
  84
  85 void TrafficTableDialog::addProgressFrame(QObject *parent)
  86 {
  87     ProgressFrame::addToButtonBox(ui->buttonBox, parent);
  88 }
  89
  90 QDialogButtonBox *TrafficTableDialog::buttonBox() const
  91 {
  92     return ui->btnBoxSettings;
  93 }
  94
  95 QVBoxLayout *TrafficTableDialog::getVerticalLayout() const
  96 {
  97     return ui->verticalLayout;
  98 }
  99
 100 QCheckBox *TrafficTableDialog::displayFilterCheckBox() const
 101 {
 102     return ui->displayFilterCheckBox;
 103 }
 104
 105 QCheckBox *TrafficTableDialog::absoluteTimeCheckBox() const
 106 {
 107     return ui->absoluteTimeCheckBox;
 108 }
 109
 110 TrafficTab *TrafficTableDialog::trafficTab() const
 111 {
 112     return ui->trafficTab;
 113 }
 114
 115 TrafficTypesList *TrafficTableDialog::trafficList() const
 116 {
 117     return ui->trafficList;
 118 }
 119
 120 void TrafficTableDialog::currentTabChanged()
 121 {
 122     bool has_resolution = ui->trafficTab->hasNameResolution();
 123     copy_bt_->setMenu(ui->trafficTab->createCopyMenu(copy_bt_));
 124
 125     ui->nameResolutionCheckBox->setEnabled(has_resolution);
 126     if (! has_resolution) {
 127         ui->nameResolutionCheckBox->setChecked(false);
 128         ui->trafficTab->setNameResolution(false);
 129     }
 130 }
 131
 132 void TrafficTableDialog::aggregationSummaryOnlyCheckBoxToggled(bool checked)
 133 {
 134     if (!cap_file_.isValid()) {
 135         return;
 136     }
 137
 138     ATapDataModel * atdm = trafficTab()->dataModelForTabIndex(1);
 139     if(atdm) {
 140         atdm->updateFlags(checked);
 141     }
 142
 143     cap_file_.retapPackets();
 144 }
 145
 146 void TrafficTableDialog::on_nameResolutionCheckBox_toggled(bool checked)
 147 {
 148     ui->trafficTab->setNameResolution(checked);
 149 }
 150
 151 void TrafficTableDialog::displayFilterCheckBoxToggled(bool checked)
 152 {
 153     if (!cap_file_.isValid()) {
 154         return;
 155     }
 156
 157     if (checked)
 158         trafficTab()->setFilter(cap_file_.displayFilter());
 159     else
 160         trafficTab()->setFilter(QString());
 161
 162     cap_file_.retapPackets();
 163 }
 164
 165 void TrafficTableDialog::captureEvent(CaptureEvent e)
 166 {
 167     if (e.captureContext() == CaptureEvent::Retap)
 168     {
 169         switch (e.eventType())
 170         {
 171         case CaptureEvent::Started:
 172             ui->displayFilterCheckBox->setEnabled(false);
 173             break;
 174         case CaptureEvent::Finished:
 175             ui->displayFilterCheckBox->setEnabled(true);
 176             break;
 177         default:
 178             break;
 179         }
 180     }
 181
 182 }