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     connect(mainApp->mainWindow(), SIGNAL(displayFilterSuccess(bool)), this, SLOT(displayFilterSuccess(bool)));
  74
  75     QPushButton *close_bt = ui->buttonBox->button(QDialogButtonBox::Close);
  76     if (close_bt)
  77         close_bt->setDefault(true);
  78
  79     addProgressFrame(&parent);
  80 }
  81
  82 TrafficTableDialog::~TrafficTableDialog()
  83 {
  84     delete ui;
  85 }
  86
  87 void TrafficTableDialog::addProgressFrame(QObject *parent)
  88 {
  89     ProgressFrame::addToButtonBox(ui->buttonBox, parent);
  90 }
  91
  92 QDialogButtonBox *TrafficTableDialog::buttonBox() const
  93 {
  94     return ui->btnBoxSettings;
  95 }
  96
  97 QVBoxLayout *TrafficTableDialog::getVerticalLayout() const
  98 {
  99     return ui->verticalLayout;
 100 }
 101
 102 QCheckBox *TrafficTableDialog::displayFilterCheckBox() const
 103 {
 104     return ui->displayFilterCheckBox;
 105 }
 106
 107 QCheckBox *TrafficTableDialog::absoluteTimeCheckBox() const
 108 {
 109     return ui->absoluteTimeCheckBox;
 110 }
 111
 112 TrafficTab *TrafficTableDialog::trafficTab() const
 113 {
 114     return ui->trafficTab;
 115 }
 116
 117 TrafficTypesList *TrafficTableDialog::trafficList() const
 118 {
 119     return ui->trafficList;
 120 }
 121
 122 void TrafficTableDialog::currentTabChanged()
 123 {
 124     bool has_resolution = ui->trafficTab->hasNameResolution();
 125     copy_bt_->setMenu(ui->trafficTab->createCopyMenu(copy_bt_));
 126
 127     ui->nameResolutionCheckBox->setEnabled(has_resolution);
 128     if (! has_resolution) {
 129         ui->nameResolutionCheckBox->setChecked(false);
 130         ui->trafficTab->setNameResolution(false);
 131     }
 132 }
 133
 134 void TrafficTableDialog::aggregationSummaryOnlyCheckBoxToggled(bool checked)
 135 {
 136     if (!cap_file_.isValid()) {
 137         return;
 138     }
 139
 140     ATapDataModel * atdm = trafficTab()->dataModelForTabIndex(1);
 141     if(atdm) {
 142         atdm->updateFlags(checked);
 143     }
 144
 145     cap_file_.retapPackets();
 146 }
 147
 148 void TrafficTableDialog::on_nameResolutionCheckBox_toggled(bool checked)
 149 {
 150     ui->trafficTab->setNameResolution(checked);
 151 }
 152
 153 void TrafficTableDialog::displayFilterCheckBoxToggled(bool checked)
 154 {
 155     displayFilterUpdate(checked);
 156 }
 157
 158 void TrafficTableDialog::displayFilterUpdate(bool set_filter)
 159 {
 160     if (!cap_file_.isValid()) {
 161         return;
 162     }
 163
 164     if (set_filter)
 165         trafficTab()->setFilter(cap_file_.displayFilter());
 166     else
 167         trafficTab()->setFilter(QString());
 168
 169     cap_file_.retapPackets();
 170 }
 171
 172 void TrafficTableDialog::displayFilterSuccess(bool success)
 173 {
 174     if (success && ui->displayFilterCheckBox->isEnabled() && ui->displayFilterCheckBox->isChecked()) {
 175        displayFilterUpdate(true);
 176     }
 177 }
 178
 179 void TrafficTableDialog::captureEvent(CaptureEvent e)
 180 {
 181     if (e.captureContext() == CaptureEvent::Retap)
 182     {
 183         switch (e.eventType())
 184         {
 185         case CaptureEvent::Started:
 186             ui->displayFilterCheckBox->setEnabled(false);
 187             break;
 188         case CaptureEvent::Finished:
 189             ui->displayFilterCheckBox->setEnabled(true);
 190             break;
 191         default:
 192             break;
 193         }
 194     }
 195
 196 }