ui/qt/resolved_addresses_dialog.cpp

   1 /* resolved_addresses_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 "resolved_addresses_dialog.h"
  11 #include <ui_resolved_addresses_dialog.h>
  12
  13 #include "config.h"
  14
  15 #include "file.h"
  16
  17 #include "epan/addr_resolv.h"
  18 #include <wiretap/wtap.h>
  19
  20 #include <QMenu>
  21 #include <QPushButton>
  22 #include <QTextCursor>
  23 #include <QSortFilterProxyModel>
  24
  25 #include "capture_file.h"
  26 #include "main_application.h"
  27
  28 #include <ui/qt/models/astringlist_list_model.h>
  29 #include <ui/qt/models/resolved_addresses_models.h>
  30
  31 const QString no_entries_ = QObject::tr("No entries.");
  32 const QString entry_count_ = QObject::tr("%1 entries.");
  33
  34 ResolvedAddressesDialog::ResolvedAddressesDialog(QWidget *parent, QString captureFile, wtap* wth) :
  35     GeometryStateDialog(parent),
  36     ui(new Ui::ResolvedAddressesDialog),
  37     file_name_(tr("[no file]"))
  38 {
  39     ui->setupUi(this);
  40     setAttribute(Qt::WA_DeleteOnClose, true);
  41
  42     QStringList title_parts = QStringList() << tr("Resolved Addresses");
  43
  44     copy_bt_ = ui->buttonBox->addButton(tr("Copy"), QDialogButtonBox::ActionRole);
  45
  46     save_bt_ = ui->buttonBox->addButton(tr("Save as…"), QDialogButtonBox::ActionRole);
  47     connect(save_bt_, &QPushButton::clicked, this, &ResolvedAddressesDialog::saveAs);
  48
  49     if (!captureFile.isEmpty()) {
  50         file_name_ = captureFile;
  51         title_parts << file_name_;
  52     }
  53     setWindowTitle(mainApp->windowTitleString(title_parts));
  54
  55     ui->plainTextEdit->setFont(mainApp->monospaceFont());
  56     ui->plainTextEdit->setReadOnly(true);
  57     ui->plainTextEdit->setWordWrapMode(QTextOption::NoWrap);
  58
  59     if (wth) {
  60         // might return null
  61         wtap_block_t nrb_hdr;
  62
  63         /*
  64             * XXX - support multiple NRBs.
  65             */
  66         nrb_hdr = wtap_file_get_nrb(wth);
  67         if (nrb_hdr != NULL) {
  68             char *str;
  69
  70             /*
  71                 * XXX - support multiple comments.
  72                 */
  73             if (wtap_block_get_nth_string_option_value(nrb_hdr, OPT_COMMENT, 0, &str) == WTAP_OPTTYPE_SUCCESS) {
  74                 comment_ = str;
  75             }
  76         }
  77     }
  78
  79     fillBlocks();
  80
  81     ethSortModel = new AStringListListSortFilterProxyModel(this);
  82     ethTypeModel = new AStringListListSortFilterProxyModel(this);
  83     EthernetAddressModel * ethModel = new EthernetAddressModel(this);
  84     ethSortModel->setSourceModel(ethModel);
  85     ethSortModel->setColumnsToFilter(QList<int>() << 1 << 2);
  86     ethSortModel->setFilterCaseSensitivity(Qt::CaseInsensitive);
  87     ethTypeModel->setSourceModel(ethSortModel);
  88     ethTypeModel->setColumnToFilter(0);
  89     ethTypeModel->setColumnToHide(0);
  90     ui->tblAddresses->setModel(ethTypeModel);
  91     ui->tblAddresses->resizeColumnsToContents();
  92     ui->tblAddresses->sortByColumn(1, Qt::AscendingOrder);
  93     ui->cmbDataType->addItems(ethModel->filterValues());
  94
  95     portSortModel = new AStringListListSortFilterProxyModel(this);
  96     portTypeModel = new AStringListListSortFilterProxyModel(this);
  97     PortsModel * portModel = new PortsModel(this);
  98     portSortModel->setSourceModel(portModel);
  99     portSortModel->setColumnAsNumeric(PORTS_COL_PORT);
 100     portSortModel->setColumnsToFilter(QList<int>() << PORTS_COL_NAME << PORTS_COL_PROTOCOL);
 101     portSortModel->setFilterCaseSensitivity(Qt::CaseInsensitive);
 102     portTypeModel->setSourceModel(portSortModel);
 103     portTypeModel->setColumnToFilter(PORTS_COL_PROTOCOL);
 104     portTypeModel->setColumnAsNumeric(PORTS_COL_PORT);
 105     ui->tblPorts->setModel(portTypeModel);
 106     ui->tblPorts->resizeColumnsToContents();
 107     ui->tblPorts->sortByColumn(PORTS_COL_PORT, Qt::AscendingOrder);
 108     ui->cmbPortFilterType->addItems(portModel->filterValues());
 109
 110     tabChanged(ui->tabWidget->currentIndex());
 111     connect(ui->tabWidget, &QTabWidget::currentChanged, this, &ResolvedAddressesDialog::tabChanged);
 112 }
 113
 114 ResolvedAddressesDialog::~ResolvedAddressesDialog()
 115 {
 116     delete ui;
 117 }
 118
 119 void ResolvedAddressesDialog::tabChanged(int index)
 120 {
 121     QWidget *currentTab = ui->tabWidget->widget(index);
 122     ResolvedAddressesView *addressView = nullptr;
 123     if (currentTab != nullptr) {
 124         addressView = currentTab->findChild<ResolvedAddressesView*>();
 125         if (addressView != nullptr) {
 126             QMenu* oldMenu = copy_bt_->menu();
 127             copy_bt_->setMenu(addressView->createCopyMenu(false, copy_bt_));
 128             if (oldMenu != nullptr) {
 129                 delete oldMenu;
 130             }
 131         }
 132     }
 133     foreach (QAbstractButton *button, ui->buttonBox->buttons()) {
 134         if (ui->buttonBox->buttonRole(button) == QDialogButtonBox::ActionRole) {
 135             button->setEnabled(addressView != nullptr);
 136         }
 137     }
 138 }
 139
 140 void ResolvedAddressesDialog::on_cmbDataType_currentIndexChanged(int index)
 141 {
 142     if (! ethSortModel)
 143         return;
 144
 145     QString filter = ui->cmbDataType->itemText(index);
 146     if (index == 0)
 147     {
 148         filter.clear();
 149         ethTypeModel->setFilterType(AStringListListSortFilterProxyModel::FilterNone, 0);
 150     }
 151     else
 152         ethTypeModel->setFilterType(AStringListListSortFilterProxyModel::FilterByEquivalent, 0);
 153     ethTypeModel->setFilter(filter);
 154 }
 155
 156 void ResolvedAddressesDialog::on_txtSearchFilter_textChanged(QString)
 157 {
 158     QString filter = ui->txtSearchFilter->text();
 159     if (!ethSortModel || (!filter.isEmpty() && filter.length() < 3))
 160         return;
 161
 162     ethSortModel->setFilter(filter);
 163 }
 164
 165 void ResolvedAddressesDialog::on_cmbPortFilterType_currentIndexChanged(int index)
 166 {
 167     if (! portSortModel)
 168         return;
 169
 170     QString filter = ui->cmbPortFilterType->itemText(index);
 171     if (index == 0)
 172     {
 173         filter.clear();
 174         portTypeModel->setFilterType(AStringListListSortFilterProxyModel::FilterNone, PORTS_COL_PROTOCOL);
 175     }
 176     else
 177         portTypeModel->setFilterType(AStringListListSortFilterProxyModel::FilterByEquivalent, PORTS_COL_PROTOCOL);
 178     portTypeModel->setFilter(filter);
 179 }
 180
 181 void ResolvedAddressesDialog::on_txtPortFilter_textChanged(QString val)
 182 {
 183     if (! portSortModel)
 184         return;
 185
 186     portSortModel->setFilter(val);
 187 }
 188
 189 void ResolvedAddressesDialog::changeEvent(QEvent *event)
 190 {
 191     if (0 != event)
 192     {
 193         switch (event->type())
 194         {
 195         case QEvent::LanguageChange:
 196             ui->retranslateUi(this);
 197             fillBlocks();
 198             break;
 199         default:
 200             break;
 201         }
 202     }
 203     QDialog::changeEvent(event);
 204 }
 205
 206 void ResolvedAddressesDialog::fillBlocks()
 207 {
 208     setUpdatesEnabled(false);
 209     ui->plainTextEdit->clear();
 210
 211     QString lines;
 212     ui->plainTextEdit->appendPlainText(tr("# Resolved addresses found in %1").arg(file_name_));
 213
 214     if (ui->actionComment->isChecked()) {
 215         lines = "\n";
 216         lines.append(tr("# Comments\n#\n# "));
 217         if (!comment_.isEmpty()) {
 218             lines.append("\n\n");
 219             lines.append(comment_);
 220             lines.append("\n");
 221         } else {
 222             lines.append(no_entries_);
 223         }
 224         ui->plainTextEdit->appendPlainText(lines);
 225     }
 226
 227     ui->plainTextEdit->moveCursor(QTextCursor::Start);
 228     setUpdatesEnabled(true);
 229 }
 230
 231 void ResolvedAddressesDialog::saveAs()
 232 {
 233     QWidget *currentTab = ui->tabWidget->currentWidget();
 234     if (currentTab == nullptr) {
 235         return;
 236     }
 237
 238     ResolvedAddressesView *addressView = currentTab->findChild<ResolvedAddressesView*>();
 239     if (addressView == nullptr) {
 240         return;
 241     }
 242
 243     addressView->saveAs();
 244 }