6986eb4866
follow_stream_dialog.cpp. Remove the ones that we don't use. Use QMessageBox in place of simple_dialog. Move IP6OPT_HOME_ADDRESS and related definitions to packet-ipv6.c. It looks like we only use it there and it conflicts with WinPcap. Mark some items unused. svn path=/trunk/; revision=51951
1467 lines
56 KiB
C++
1467 lines
56 KiB
C++
/* main_window.cpp
|
|
*
|
|
* $Id$
|
|
*
|
|
* Wireshark - Network traffic analyzer
|
|
* By Gerald Combs <gerald@wireshark.org>
|
|
* Copyright 1998 Gerald Combs
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include "main_window.h"
|
|
#include "ui_main_window.h"
|
|
|
|
#include "globals.h"
|
|
|
|
#include <epan/epan_dissect.h>
|
|
#include <epan/filesystem.h>
|
|
#include <epan/ipproto.h>
|
|
#include <epan/prefs.h>
|
|
|
|
//#include <wiretap/wtap.h>
|
|
|
|
#ifdef HAVE_LIBPCAP
|
|
#include "capture.h"
|
|
#include "capture-pcap-util.h"
|
|
#include "capture_ui_utils.h"
|
|
#include "capture_session.h"
|
|
#endif
|
|
|
|
#include "ui/alert_box.h"
|
|
#include "ui/main_statusbar.h"
|
|
#include "ui/capture_globals.h"
|
|
|
|
#include "wireshark_application.h"
|
|
#include "proto_tree.h"
|
|
#include "byte_view_tab.h"
|
|
#include "display_filter_edit.h"
|
|
#include "import_text_dialog.h"
|
|
#include "export_dissection_dialog.h"
|
|
|
|
#include "qt_ui_utils.h"
|
|
|
|
#include <QTreeWidget>
|
|
#include <QTabWidget>
|
|
#include <QAction>
|
|
#include <QToolButton>
|
|
#include <QKeyEvent>
|
|
#include <QMetaObject>
|
|
#include <QMessageBox>
|
|
|
|
#ifdef QT_MACEXTRAS_LIB
|
|
#include <QtMacExtras/QMacNativeToolBar>
|
|
#endif
|
|
|
|
#include <QDebug>
|
|
|
|
//menu_recent_file_write_all
|
|
|
|
// If we ever add support for multiple windows this will need to be replaced.
|
|
static MainWindow *gbl_cur_main_window = NULL;
|
|
|
|
void pipe_input_set_handler(gint source, gpointer user_data, int *child_process, pipe_input_cb_t input_cb)
|
|
{
|
|
gbl_cur_main_window->setPipeInputHandler(source, user_data, child_process, input_cb);
|
|
}
|
|
|
|
MainWindow::MainWindow(QWidget *parent) :
|
|
QMainWindow(parent),
|
|
main_ui_(new Ui::MainWindow),
|
|
df_combo_box_(new DisplayFilterCombo()),
|
|
cap_file_(NULL),
|
|
previous_focus_(NULL),
|
|
capture_stopping_(false),
|
|
capture_filter_valid_(false),
|
|
#ifdef _WIN32
|
|
pipe_timer_(NULL)
|
|
#else
|
|
pipe_notifier_(NULL)
|
|
#endif
|
|
{
|
|
gbl_cur_main_window = this;
|
|
main_ui_->setupUi(this);
|
|
setTitlebarForCaptureFile();
|
|
setMenusForCaptureFile();
|
|
setForCapturedPackets(false);
|
|
setMenusForSelectedPacket();
|
|
setMenusForSelectedTreeRow();
|
|
setForCaptureInProgress(false);
|
|
setMenusForFileSet(false);
|
|
interfaceSelectionChanged();
|
|
|
|
//To prevent users use features before initialization complete
|
|
//Otherwise unexpected problems may occur
|
|
setFeaturesEnabled(false);
|
|
connect(wsApp, SIGNAL(appInitialized()), this, SLOT(setFeaturesEnabled()));
|
|
|
|
connect(wsApp, SIGNAL(preferencesChanged()), this, SLOT(layoutPanes()));
|
|
|
|
connect(wsApp, SIGNAL(updateRecentItemStatus(const QString &, qint64, bool)), this, SLOT(updateRecentFiles()));
|
|
updateRecentFiles();
|
|
|
|
connect(&summary_dialog_, SIGNAL(captureCommentChanged()), this, SLOT(updateForUnsavedChanges()));
|
|
connect(&follow_stream_dialog_, SIGNAL(updateFilter(QString&, bool)), this, SLOT(filterPackets(QString&, bool)));
|
|
|
|
const DisplayFilterEdit *df_edit = dynamic_cast<DisplayFilterEdit *>(df_combo_box_->lineEdit());
|
|
connect(df_edit, SIGNAL(pushFilterSyntaxStatus(QString&)), main_ui_->statusBar, SLOT(pushFilterStatus(QString&)));
|
|
connect(df_edit, SIGNAL(popFilterSyntaxStatus()), main_ui_->statusBar, SLOT(popFilterStatus()));
|
|
connect(df_edit, SIGNAL(pushFilterSyntaxWarning(QString&)),
|
|
main_ui_->statusBar, SLOT(pushTemporaryStatus(QString&)));
|
|
connect(df_edit, SIGNAL(filterPackets(QString&,bool)), this, SLOT(filterPackets(QString&,bool)));
|
|
connect(df_edit, SIGNAL(addBookmark(QString)), this, SLOT(addDisplayFilterButton(QString)));
|
|
connect(this, SIGNAL(displayFilterSuccess(bool)), df_edit, SLOT(displayFilterSuccess(bool)));
|
|
|
|
// http://standards.freedesktop.org/icon-naming-spec/icon-naming-spec-latest.html
|
|
// http://qt-project.org/doc/qt-4.8/qstyle.html#StandardPixmap-enum
|
|
main_ui_->actionFileOpen->setIcon(
|
|
QIcon().fromTheme("document-open", style()->standardIcon(QStyle::SP_DirIcon)));
|
|
// main_ui_->actionFileSave set in main_window.ui
|
|
main_ui_->actionFileClose->setIcon(
|
|
QIcon().fromTheme("process-stop", style()->standardIcon(QStyle::SP_DialogCloseButton)));
|
|
|
|
// In Qt4 multiple toolbars and "pretty" are mutually exculsive on OS X. If
|
|
// unifiedTitleAndToolBarOnMac is enabled everything ends up in the same row.
|
|
// https://bugreports.qt-project.org/browse/QTBUG-22433
|
|
// This property is obsolete in Qt5 so this issue may be fixed in that version.
|
|
main_ui_->displayFilterToolBar->addWidget(df_combo_box_);
|
|
|
|
main_ui_->goToFrame->hide();
|
|
// XXX For some reason the cursor is drawn funny with an input mask set
|
|
// https://bugreports.qt-project.org/browse/QTBUG-7174
|
|
|
|
main_ui_->searchFrame->hide();
|
|
connect(main_ui_->searchFrame, SIGNAL(pushFilterSyntaxStatus(QString&)),
|
|
main_ui_->statusBar, SLOT(pushTemporaryStatus(QString&)));
|
|
|
|
#if defined(Q_OS_MAC)
|
|
#ifdef QT_MACEXTRAS_LIB
|
|
QMacNativeToolBar *ntb = QtMacExtras::setNativeToolBar(main_ui_->mainToolBar);
|
|
ntb->setIconSize(QSize(24, 24));
|
|
#endif // QT_MACEXTRAS_LIB
|
|
|
|
foreach (QMenu *menu, main_ui_->menuBar->findChildren<QMenu*>()) {
|
|
foreach (QAction *act, menu->actions()) {
|
|
act->setIconVisibleInMenu(false);
|
|
}
|
|
}
|
|
main_ui_->goToLineEdit->setAttribute(Qt::WA_MacSmallSize, true);
|
|
main_ui_->goToGo->setAttribute(Qt::WA_MacSmallSize, true);
|
|
main_ui_->goToCancel->setAttribute(Qt::WA_MacSmallSize, true);
|
|
|
|
main_ui_->actionEditPreferences->setMenuRole(QAction::PreferencesRole);
|
|
|
|
#endif // Q_OS_MAC
|
|
|
|
#ifdef HAVE_SOFTWARE_UPDATE
|
|
QAction *update_sep = main_ui_->menuHelp->insertSeparator(main_ui_->actionHelpAbout);
|
|
QAction *update_action = new QAction(tr("Check for Updates..."), main_ui_->menuHelp);
|
|
main_ui_->menuHelp->insertAction(update_sep, update_action);
|
|
connect(update_action, SIGNAL(triggered()), this, SLOT(on_actionHelpCheckForUpdates_triggered()));
|
|
#endif
|
|
empty_pane_ = new QWidget(main_ui_->mainStack);
|
|
|
|
packet_list_ = new PacketList(main_ui_->mainStack);
|
|
|
|
proto_tree_ = new ProtoTree(main_ui_->mainStack);
|
|
proto_tree_->setHeaderHidden(true);
|
|
proto_tree_->installEventFilter(this);
|
|
|
|
byte_view_tab_ = new ByteViewTab(main_ui_->mainStack);
|
|
byte_view_tab_->setTabPosition(QTabWidget::South);
|
|
byte_view_tab_->setDocumentMode(true);
|
|
|
|
packet_list_->setProtoTree(proto_tree_);
|
|
packet_list_->setByteViewTab(byte_view_tab_);
|
|
packet_list_->installEventFilter(this);
|
|
|
|
master_split_ = NULL;
|
|
extra_split_ = NULL;
|
|
|
|
main_welcome_ = main_ui_->welcomePage;
|
|
|
|
#ifdef HAVE_LIBPCAP
|
|
connect(wsApp, SIGNAL(captureCapturePrepared(capture_session *)),
|
|
this, SLOT(captureCapturePrepared(capture_session *)));
|
|
connect(wsApp, SIGNAL(captureCaptureUpdateStarted(capture_session *)),
|
|
this, SLOT(captureCaptureUpdateStarted(capture_session *)));
|
|
connect(wsApp, SIGNAL(captureCaptureUpdateFinished(capture_session *)),
|
|
this, SLOT(captureCaptureUpdateFinished(capture_session *)));
|
|
connect(wsApp, SIGNAL(captureCaptureFixedStarted(capture_session *)),
|
|
this, SLOT(captureCaptureFixedStarted(capture_session *)));
|
|
connect(wsApp, SIGNAL(captureCaptureFixedFinished(capture_session *)),
|
|
this, SLOT(captureCaptureFixedFinished(capture_session *)));
|
|
connect(wsApp, SIGNAL(captureCaptureStopping(capture_session *)),
|
|
this, SLOT(captureCaptureStopping(capture_session *)));
|
|
connect(wsApp, SIGNAL(captureCaptureFailed(capture_session *)),
|
|
this, SLOT(captureCaptureFailed(capture_session *)));
|
|
#endif
|
|
|
|
connect(wsApp, SIGNAL(captureFileOpened(const capture_file*)),
|
|
this, SLOT(captureFileOpened(const capture_file*)));
|
|
connect(wsApp, SIGNAL(captureFileReadStarted(const capture_file*)),
|
|
this, SLOT(captureFileReadStarted(const capture_file*)));
|
|
connect(wsApp, SIGNAL(captureFileReadFinished(const capture_file*)),
|
|
this, SLOT(captureFileReadFinished(const capture_file*)));
|
|
connect(wsApp, SIGNAL(captureFileClosing(const capture_file*)),
|
|
this, SLOT(captureFileClosing(const capture_file*)));
|
|
connect(wsApp, SIGNAL(captureFileClosed(const capture_file*)),
|
|
this, SLOT(captureFileClosed(const capture_file*)));
|
|
connect(wsApp, SIGNAL(columnsChanged()),
|
|
this, SLOT(recreatePacketList()));
|
|
connect(wsApp, SIGNAL(packetDissectionChanged()),
|
|
this, SLOT(redissectPackets()));
|
|
connect(wsApp, SIGNAL(appInitialized()),
|
|
this, SLOT(filterExpressionsChanged()));
|
|
connect(wsApp, SIGNAL(filterExpressionsChanged()),
|
|
this, SLOT(filterExpressionsChanged()));
|
|
|
|
connect(main_welcome_, SIGNAL(startCapture()),
|
|
this, SLOT(startCapture()));
|
|
connect(main_welcome_, SIGNAL(recentFileActivated(QString&)),
|
|
this, SLOT(openCaptureFile(QString&)));
|
|
connect(main_welcome_, SIGNAL(pushFilterSyntaxStatus(QString&)),
|
|
main_ui_->statusBar, SLOT(pushFilterStatus(QString&)));
|
|
connect(main_welcome_, SIGNAL(popFilterSyntaxStatus()),
|
|
main_ui_->statusBar, SLOT(popFilterStatus()));
|
|
|
|
connect(this, SIGNAL(setCaptureFile(capture_file*)),
|
|
main_ui_->searchFrame, SLOT(setCaptureFile(capture_file*)));
|
|
connect(this, SIGNAL(setCaptureFile(capture_file*)),
|
|
main_ui_->statusBar, SLOT(setCaptureFile(capture_file*)));
|
|
connect(this, SIGNAL(setCaptureFile(capture_file*)),
|
|
packet_list_, SLOT(setCaptureFile(capture_file*)));
|
|
connect(this, SIGNAL(setCaptureFile(capture_file*)),
|
|
byte_view_tab_, SLOT(setCaptureFile(capture_file*)));
|
|
|
|
connect(main_ui_->actionGoNextPacket, SIGNAL(triggered()),
|
|
packet_list_, SLOT(goNextPacket()));
|
|
connect(main_ui_->actionGoPreviousPacket, SIGNAL(triggered()),
|
|
packet_list_, SLOT(goPreviousPacket()));
|
|
connect(main_ui_->actionGoFirstPacket, SIGNAL(triggered()),
|
|
packet_list_, SLOT(goFirstPacket()));
|
|
connect(main_ui_->actionGoLastPacket, SIGNAL(triggered()),
|
|
packet_list_, SLOT(goLastPacket()));
|
|
|
|
connect(main_ui_->actionViewExpandSubtrees, SIGNAL(triggered()),
|
|
proto_tree_, SLOT(expandSubtrees()));
|
|
connect(main_ui_->actionViewExpandAll, SIGNAL(triggered()),
|
|
proto_tree_, SLOT(expandAll()));
|
|
connect(main_ui_->actionViewCollapseAll, SIGNAL(triggered()),
|
|
proto_tree_, SLOT(collapseAll()));
|
|
|
|
connect(packet_list_->selectionModel(), SIGNAL(selectionChanged(QItemSelection,QItemSelection)),
|
|
this, SLOT(setMenusForSelectedPacket()));
|
|
connect(packet_list_, SIGNAL(packetDissectionChanged()),
|
|
this, SLOT(redissectPackets()));
|
|
connect(packet_list_, SIGNAL(setMenusFollowStream()),
|
|
this, SLOT(setMenusForFollowStream()));
|
|
|
|
connect(proto_tree_, SIGNAL(protoItemSelected(QString&)),
|
|
main_ui_->statusBar, SLOT(pushFieldStatus(QString&)));
|
|
|
|
connect(proto_tree_, SIGNAL(protoItemSelected(field_info *)),
|
|
this, SLOT(setMenusForSelectedTreeRow(field_info *)));
|
|
|
|
connect(&file_set_dialog_, SIGNAL(fileSetOpenCaptureFile(QString&)),
|
|
this, SLOT(openCaptureFile(QString&)));
|
|
|
|
QTreeWidget *iface_tree = findChild<QTreeWidget *>("interfaceTree");
|
|
if (iface_tree) {
|
|
connect(iface_tree, SIGNAL(itemSelectionChanged()),
|
|
this, SLOT(interfaceSelectionChanged()));
|
|
}
|
|
connect(main_ui_->welcomePage, SIGNAL(captureFilterSyntaxChanged(bool)),
|
|
this, SLOT(captureFilterSyntaxChanged(bool)));
|
|
|
|
main_ui_->mainStack->setCurrentWidget(main_welcome_);
|
|
}
|
|
|
|
MainWindow::~MainWindow()
|
|
{
|
|
delete main_ui_;
|
|
}
|
|
|
|
QString MainWindow::getFilter()
|
|
{
|
|
return df_combo_box_->itemText(df_combo_box_->count());
|
|
}
|
|
|
|
void MainWindow::setPipeInputHandler(gint source, gpointer user_data, int *child_process, pipe_input_cb_t input_cb)
|
|
{
|
|
pipe_source_ = source;
|
|
pipe_child_process_ = child_process;
|
|
pipe_user_data_ = user_data;
|
|
pipe_input_cb_ = input_cb;
|
|
|
|
#ifdef _WIN32
|
|
/* Tricky to use pipes in win9x, as no concept of wait. NT can
|
|
do this but that doesn't cover all win32 platforms. GTK can do
|
|
this but doesn't seem to work over processes. Attempt to do
|
|
something similar here, start a timer and check for data on every
|
|
timeout. */
|
|
/*g_log(NULL, G_LOG_LEVEL_DEBUG, "pipe_input_set_handler: new");*/
|
|
|
|
if (pipe_timer_) {
|
|
disconnect(pipe_timer_, SIGNAL(timeout()), this, SLOT(pipeTimeout()));
|
|
delete pipe_timer_;
|
|
}
|
|
|
|
pipe_timer_ = new QTimer(this);
|
|
connect(pipe_timer_, SIGNAL(timeout()), this, SLOT(pipeTimeout()));
|
|
connect(pipe_timer_, SIGNAL(destroyed()), this, SLOT(pipeNotifierDestroyed()));
|
|
pipe_timer_->start(200);
|
|
#else
|
|
if (pipe_notifier_) {
|
|
disconnect(pipe_notifier_, SIGNAL(activated(int)), this, SLOT(pipeActivated(int)));
|
|
delete pipe_notifier_;
|
|
}
|
|
|
|
pipe_notifier_ = new QSocketNotifier(pipe_source_, QSocketNotifier::Read);
|
|
// XXX ui/gtk/gui_utils.c sets the encoding. Do we need to do the same?
|
|
connect(pipe_notifier_, SIGNAL(activated(int)), this, SLOT(pipeActivated(int)));
|
|
connect(pipe_notifier_, SIGNAL(destroyed()), this, SLOT(pipeNotifierDestroyed()));
|
|
#endif
|
|
}
|
|
|
|
|
|
bool MainWindow::eventFilter(QObject *obj, QEvent *event) {
|
|
|
|
// The user typed some text. Start filling in a filter.
|
|
// We may need to be more choosy here. We just need to catch events for the packet list,
|
|
// proto tree, and main welcome widgets.
|
|
if (event->type() == QEvent::KeyPress) {
|
|
QKeyEvent *kevt = static_cast<QKeyEvent *>(event);
|
|
if (kevt->text().length() > 0 && kevt->text()[0].isPrint()) {
|
|
df_combo_box_->lineEdit()->insert(kevt->text());
|
|
df_combo_box_->lineEdit()->setFocus();
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return QObject::eventFilter(obj, event);
|
|
}
|
|
|
|
void MainWindow::keyPressEvent(QKeyEvent *event) {
|
|
|
|
// Explicitly focus on the display filter combo.
|
|
if (event->modifiers() & Qt::ControlModifier && event->key() == Qt::Key_Slash) {
|
|
df_combo_box_->setFocus(Qt::ShortcutFocusReason);
|
|
return;
|
|
}
|
|
|
|
if (wsApp->focusWidget() == main_ui_->goToLineEdit) {
|
|
if (event->modifiers() == Qt::NoModifier) {
|
|
if (event->key() == Qt::Key_Escape) {
|
|
on_goToCancel_clicked();
|
|
} else if (event->key() == Qt::Key_Enter || event->key() == Qt::Key_Return) {
|
|
on_goToGo_clicked();
|
|
}
|
|
}
|
|
return; // goToLineEdit didn't want it and we don't either.
|
|
}
|
|
|
|
// Move up & down the packet list.
|
|
if (event->key() == Qt::Key_F7) {
|
|
packet_list_->goPreviousPacket();
|
|
} else if (event->key() == Qt::Key_F8) {
|
|
packet_list_->goNextPacket();
|
|
}
|
|
|
|
// Move along, citizen.
|
|
QMainWindow::keyPressEvent(event);
|
|
}
|
|
|
|
void MainWindow::closeEvent(QCloseEvent *event) {
|
|
/* If we're in the middle of stopping a capture, don't do anything;
|
|
the user can try deleting the window after the capture stops. */
|
|
if (capture_stopping_) {
|
|
event->ignore();
|
|
return;
|
|
}
|
|
|
|
// Make sure we kill any open dumpcap processes.
|
|
delete main_welcome_;
|
|
|
|
if(testCaptureFileClose(true)) {
|
|
/* QCoreApplication::quit() won't exit properly
|
|
* because when during initialization phase we are not in the event loop */
|
|
exit(0);
|
|
}
|
|
else
|
|
{
|
|
event->ignore();
|
|
return;
|
|
}
|
|
|
|
}
|
|
|
|
QWidget* MainWindow::getLayoutWidget(layout_pane_content_e type) {
|
|
switch (type) {
|
|
case layout_pane_content_none:
|
|
return empty_pane_;
|
|
case layout_pane_content_plist:
|
|
return packet_list_;
|
|
case layout_pane_content_pdetails:
|
|
return proto_tree_;
|
|
case layout_pane_content_pbytes:
|
|
return byte_view_tab_;
|
|
default:
|
|
g_assert_not_reached();
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
void MainWindow::mergeCaptureFile()
|
|
{
|
|
QString file_name = "";
|
|
QString display_filter = "";
|
|
dfilter_t *rfcode = NULL;
|
|
int err;
|
|
|
|
if (!cap_file_)
|
|
return;
|
|
|
|
if (prefs.gui_ask_unsaved) {
|
|
if (cf_has_unsaved_data(cap_file_)) {
|
|
QMessageBox msg_dialog;
|
|
gchar *display_basename;
|
|
int response;
|
|
|
|
msg_dialog.setIcon(QMessageBox::Question);
|
|
/* This file has unsaved data; ask the user whether to save
|
|
the capture. */
|
|
if (cap_file_->is_tempfile) {
|
|
msg_dialog.setText(tr("Save packets before merging?"));
|
|
msg_dialog.setInformativeText(tr("A temporary capture file can't be merged."));
|
|
} else {
|
|
/*
|
|
* Format the message.
|
|
*/
|
|
display_basename = g_filename_display_basename(cap_file_->filename);
|
|
msg_dialog.setText(QString(tr("Save changes in \"%1\" before merging?")).arg(display_basename));
|
|
g_free(display_basename);
|
|
msg_dialog.setInformativeText(tr("Changes must be saved before the files can be merged."));
|
|
}
|
|
|
|
msg_dialog.setStandardButtons(QMessageBox::Save | QMessageBox::Cancel);
|
|
msg_dialog.setDefaultButton(QMessageBox::Save);
|
|
|
|
response = msg_dialog.exec();
|
|
|
|
switch (response) {
|
|
|
|
case QMessageBox::Save:
|
|
/* Save the file but don't close it */
|
|
saveCaptureFile(cap_file_, FALSE);
|
|
break;
|
|
|
|
case QMessageBox::Cancel:
|
|
default:
|
|
/* Don't do the merge. */
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
for (;;) {
|
|
CaptureFileDialog merge_dlg(this, cap_file_, display_filter);
|
|
int file_type;
|
|
cf_status_t merge_status;
|
|
char *in_filenames[2];
|
|
char *tmpname;
|
|
|
|
switch (prefs.gui_fileopen_style) {
|
|
|
|
case FO_STYLE_LAST_OPENED:
|
|
/* The user has specified that we should start out in the last directory
|
|
we looked in. If we've already opened a file, use its containing
|
|
directory, if we could determine it, as the directory, otherwise
|
|
use the "last opened" directory saved in the preferences file if
|
|
there was one. */
|
|
/* This is now the default behaviour in file_selection_new() */
|
|
break;
|
|
|
|
case FO_STYLE_SPECIFIED:
|
|
/* The user has specified that we should always start out in a
|
|
specified directory; if they've specified that directory,
|
|
start out by showing the files in that dir. */
|
|
if (prefs.gui_fileopen_dir[0] != '\0')
|
|
merge_dlg.setDirectory(prefs.gui_fileopen_dir);
|
|
break;
|
|
}
|
|
|
|
if (merge_dlg.merge(file_name)) {
|
|
if (dfilter_compile(display_filter.toUtf8().constData(), &rfcode)) {
|
|
cf_set_rfcode(cap_file_, rfcode);
|
|
} else {
|
|
/* Not valid. Tell the user, and go back and run the file
|
|
selection box again once they dismiss the alert. */
|
|
//bad_dfilter_alert_box(top_level, display_filter->str);
|
|
QMessageBox::warning(this, tr("Invalid Display Filter"),
|
|
QString(tr("The filter expression %1 isn't a valid display filter. (%2).").arg(display_filter, dfilter_error_msg)),
|
|
QMessageBox::Ok);
|
|
continue;
|
|
}
|
|
} else {
|
|
return;
|
|
}
|
|
|
|
file_type = cap_file_->cd_t;
|
|
|
|
/* Try to merge or append the two files */
|
|
tmpname = NULL;
|
|
if (merge_dlg.mergeType() == 0) {
|
|
/* chronological order */
|
|
in_filenames[0] = cap_file_->filename;
|
|
in_filenames[1] = file_name.toUtf8().data();
|
|
merge_status = cf_merge_files(&tmpname, 2, in_filenames, file_type, FALSE);
|
|
} else if (merge_dlg.mergeType() <= 0) {
|
|
/* prepend file */
|
|
in_filenames[0] = file_name.toUtf8().data();
|
|
in_filenames[1] = cap_file_->filename;
|
|
merge_status = cf_merge_files(&tmpname, 2, in_filenames, file_type, TRUE);
|
|
} else {
|
|
/* append file */
|
|
in_filenames[0] = cap_file_->filename;
|
|
in_filenames[1] = file_name.toUtf8().data();
|
|
merge_status = cf_merge_files(&tmpname, 2, in_filenames, file_type, TRUE);
|
|
}
|
|
|
|
if (merge_status != CF_OK) {
|
|
if (rfcode != NULL)
|
|
dfilter_free(rfcode);
|
|
g_free(tmpname);
|
|
continue;
|
|
}
|
|
|
|
cf_close(cap_file_);
|
|
|
|
/* Try to open the merged capture file. */
|
|
cfile.window = this;
|
|
if (cf_open(&cfile, tmpname, TRUE /* temporary file */, &err) != CF_OK) {
|
|
/* We couldn't open it; fail. */
|
|
cfile.window = NULL;
|
|
if (rfcode != NULL)
|
|
dfilter_free(rfcode);
|
|
g_free(tmpname);
|
|
return;
|
|
}
|
|
|
|
/* Attach the new read filter to "cf" ("cf_open()" succeeded, so
|
|
it closed the previous capture file, and thus destroyed any
|
|
previous read filter attached to "cf"). */
|
|
cfile.rfcode = rfcode;
|
|
|
|
switch (cf_read(&cfile, FALSE)) {
|
|
|
|
case CF_READ_OK:
|
|
case CF_READ_ERROR:
|
|
/* Just because we got an error, that doesn't mean we were unable
|
|
to read any of the file; we handle what we could get from the
|
|
file. */
|
|
break;
|
|
|
|
case CF_READ_ABORTED:
|
|
/* The user bailed out of re-reading the capture file; the
|
|
capture file has been closed - just free the capture file name
|
|
string and return (without changing the last containing
|
|
directory). */
|
|
g_free(tmpname);
|
|
return;
|
|
}
|
|
|
|
/* Save the name of the containing directory specified in the path name,
|
|
if any; we can write over cf_merged_name, which is a good thing, given that
|
|
"get_dirname()" does write over its argument. */
|
|
wsApp->setLastOpenDir(get_dirname(tmpname));
|
|
g_free(tmpname);
|
|
df_combo_box_->setEditText(display_filter);
|
|
main_ui_->statusBar->showExpert();
|
|
return;
|
|
}
|
|
|
|
}
|
|
|
|
void MainWindow::importCaptureFile() {
|
|
ImportTextDialog import_dlg;
|
|
|
|
if (!testCaptureFileClose(FALSE, *new QString(tr(" before importing a new capture"))))
|
|
return;
|
|
|
|
import_dlg.exec();
|
|
|
|
if (import_dlg.result() != QDialog::Accepted) {
|
|
main_ui_->mainStack->setCurrentWidget(main_welcome_);
|
|
return;
|
|
}
|
|
|
|
openCaptureFile(import_dlg.capfileName());
|
|
}
|
|
|
|
void MainWindow::saveCaptureFile(capture_file *cf, bool stay_closed) {
|
|
QString file_name;
|
|
gboolean discard_comments;
|
|
|
|
if (cf->is_tempfile) {
|
|
/* This is a temporary capture file, so saving it means saving
|
|
it to a permanent file. Prompt the user for a location
|
|
to which to save it. Don't require that the file format
|
|
support comments - if it's a temporary capture file, it's
|
|
probably pcap-ng, which supports comments and, if it's
|
|
not pcap-ng, let the user decide what they want to do
|
|
if they've added comments. */
|
|
saveAsCaptureFile(cf, FALSE, stay_closed);
|
|
} else {
|
|
if (cf->unsaved_changes) {
|
|
cf_write_status_t status;
|
|
|
|
/* This is not a temporary capture file, but it has unsaved
|
|
changes, so saving it means doing a "safe save" on top
|
|
of the existing file, in the same format - no UI needed
|
|
unless the file has comments and the file's format doesn't
|
|
support them.
|
|
|
|
If the file has comments, does the file's format support them?
|
|
If not, ask the user whether they want to discard the comments
|
|
or choose a different format. */
|
|
switch (CaptureFileDialog::checkSaveAsWithComments(this, cf, cf->cd_t)) {
|
|
|
|
case SAVE:
|
|
/* The file can be saved in the specified format as is;
|
|
just drive on and save in the format they selected. */
|
|
discard_comments = FALSE;
|
|
break;
|
|
|
|
case SAVE_WITHOUT_COMMENTS:
|
|
/* The file can't be saved in the specified format as is,
|
|
but it can be saved without the comments, and the user
|
|
said "OK, discard the comments", so save it in the
|
|
format they specified without the comments. */
|
|
discard_comments = TRUE;
|
|
break;
|
|
|
|
case SAVE_IN_ANOTHER_FORMAT:
|
|
/* There are file formats in which we can save this that
|
|
support comments, and the user said not to delete the
|
|
comments. Do a "Save As" so the user can select
|
|
one of those formats and choose a file name. */
|
|
saveAsCaptureFile(cf, TRUE, stay_closed);
|
|
return;
|
|
|
|
case CANCELLED:
|
|
/* The user said "forget it". Just return. */
|
|
return;
|
|
|
|
default:
|
|
/* Squelch warnings that discard_comments is being used
|
|
uninitialized. */
|
|
g_assert_not_reached();
|
|
return;
|
|
}
|
|
|
|
/* XXX - cf->filename might get freed out from under us, because
|
|
the code path through which cf_save_packets() goes currently
|
|
closes the current file and then opens and reloads the saved file,
|
|
so make a copy and free it later. */
|
|
file_name = cf->filename;
|
|
status = cf_save_packets(cf, file_name.toUtf8().constData(), cf->cd_t, cf->iscompressed,
|
|
discard_comments, stay_closed);
|
|
switch (status) {
|
|
|
|
case CF_WRITE_OK:
|
|
/* The save succeeded; we're done.
|
|
If we discarded comments, redraw the packet list to reflect
|
|
any packets that no longer have comments. */
|
|
if (discard_comments)
|
|
packet_list_queue_draw();
|
|
|
|
cf->unsaved_changes = false; //we just saved so we signal that we have no unsaved changes
|
|
updateForUnsavedChanges(); // we update the title bar to remove the *
|
|
break;
|
|
|
|
case CF_WRITE_ERROR:
|
|
/* The write failed.
|
|
XXX - OK, what do we do now? Let them try a
|
|
"Save As", in case they want to try to save to a
|
|
different directory r file system? */
|
|
break;
|
|
|
|
case CF_WRITE_ABORTED:
|
|
/* The write was aborted; just drive on. */
|
|
break;
|
|
}
|
|
}
|
|
/* Otherwise just do nothing. */
|
|
}
|
|
}
|
|
|
|
void MainWindow::saveAsCaptureFile(capture_file *cf, bool must_support_comments, bool stay_closed) {
|
|
QString file_name = "";
|
|
int file_type;
|
|
gboolean compressed;
|
|
cf_write_status_t status;
|
|
gchar *dirname;
|
|
gboolean discard_comments = FALSE;
|
|
|
|
if (!cf) {
|
|
return;
|
|
}
|
|
|
|
for (;;) {
|
|
CaptureFileDialog save_as_dlg(this, cf);
|
|
|
|
switch (prefs.gui_fileopen_style) {
|
|
|
|
case FO_STYLE_LAST_OPENED:
|
|
/* The user has specified that we should start out in the last directory
|
|
we looked in. If we've already opened a file, use its containing
|
|
directory, if we could determine it, as the directory, otherwise
|
|
use the "last opened" directory saved in the preferences file if
|
|
there was one. */
|
|
/* This is now the default behaviour in file_selection_new() */
|
|
break;
|
|
|
|
case FO_STYLE_SPECIFIED:
|
|
/* The user has specified that we should always start out in a
|
|
specified directory; if they've specified that directory,
|
|
start out by showing the files in that dir. */
|
|
if (prefs.gui_fileopen_dir[0] != '\0')
|
|
save_as_dlg.setDirectory(prefs.gui_fileopen_dir);
|
|
break;
|
|
}
|
|
|
|
/* If the file has comments, does the format the user selected
|
|
support them? If not, ask the user whether they want to
|
|
discard the comments or choose a different format. */
|
|
switch(save_as_dlg.saveAs(file_name, must_support_comments)) {
|
|
|
|
case SAVE:
|
|
/* The file can be saved in the specified format as is;
|
|
just drive on and save in the format they selected. */
|
|
discard_comments = FALSE;
|
|
break;
|
|
|
|
case SAVE_WITHOUT_COMMENTS:
|
|
/* The file can't be saved in the specified format as is,
|
|
but it can be saved without the comments, and the user
|
|
said "OK, discard the comments", so save it in the
|
|
format they specified without the comments. */
|
|
discard_comments = TRUE;
|
|
break;
|
|
|
|
case SAVE_IN_ANOTHER_FORMAT:
|
|
/* There are file formats in which we can save this that
|
|
support comments, and the user said not to delete the
|
|
comments. The combo box of file formats has had the
|
|
formats that don't support comments trimmed from it,
|
|
so run the dialog again, to let the user decide
|
|
whether to save in one of those formats or give up. */
|
|
discard_comments = FALSE;
|
|
must_support_comments = TRUE;
|
|
continue;
|
|
|
|
case CANCELLED:
|
|
/* The user said "forget it". Just get rid of the dialog box
|
|
and return. */
|
|
return;
|
|
}
|
|
file_type = save_as_dlg.selectedFileType();
|
|
compressed = save_as_dlg.isCompressed();
|
|
|
|
fileAddExtension(file_name, file_type, compressed);
|
|
|
|
//#ifndef _WIN32
|
|
// /* If the file exists and it's user-immutable or not writable,
|
|
// ask the user whether they want to override that. */
|
|
// if (!file_target_unwritable_ui(top_level, file_name.toUtf8().constData())) {
|
|
// /* They don't. Let them try another file name or cancel. */
|
|
// continue;
|
|
// }
|
|
//#endif
|
|
|
|
/* Attempt to save the file */
|
|
status = cf_save_packets(cf, file_name.toUtf8().constData(), file_type, compressed,
|
|
discard_comments, stay_closed);
|
|
switch (status) {
|
|
|
|
case CF_WRITE_OK:
|
|
/* The save succeeded; we're done. */
|
|
/* Save the directory name for future file dialogs. */
|
|
dirname = get_dirname(file_name.toUtf8().data()); /* Overwrites cf_name */
|
|
set_last_open_dir(dirname);
|
|
/* If we discarded comments, redraw the packet list to reflect
|
|
any packets that no longer have comments. */
|
|
if (discard_comments)
|
|
packet_list_queue_draw();
|
|
|
|
cf->unsaved_changes = false; //we just saved so we signal that we have no unsaved changes
|
|
updateForUnsavedChanges(); // we update the title bar to remove the *
|
|
return;
|
|
|
|
case CF_WRITE_ERROR:
|
|
/* The save failed; let the user try again. */
|
|
continue;
|
|
|
|
case CF_WRITE_ABORTED:
|
|
/* The user aborted the save; just return. */
|
|
return;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
void MainWindow::exportSelectedPackets() {
|
|
QString file_name = "";
|
|
int file_type;
|
|
gboolean compressed;
|
|
packet_range_t range;
|
|
cf_write_status_t status;
|
|
gchar *dirname;
|
|
gboolean discard_comments = FALSE;
|
|
|
|
if (!cap_file_)
|
|
return;
|
|
|
|
/* Init the packet range */
|
|
packet_range_init(&range, cap_file_);
|
|
range.process_filtered = TRUE;
|
|
range.include_dependents = TRUE;
|
|
|
|
for (;;) {
|
|
CaptureFileDialog esp_dlg(this, cap_file_);
|
|
|
|
switch (prefs.gui_fileopen_style) {
|
|
|
|
case FO_STYLE_LAST_OPENED:
|
|
/* The user has specified that we should start out in the last directory
|
|
we looked in. If we've already opened a file, use its containing
|
|
directory, if we could determine it, as the directory, otherwise
|
|
use the "last opened" directory saved in the preferences file if
|
|
there was one. */
|
|
/* This is now the default behaviour in file_selection_new() */
|
|
break;
|
|
|
|
case FO_STYLE_SPECIFIED:
|
|
/* The user has specified that we should always start out in a
|
|
specified directory; if they've specified that directory,
|
|
start out by showing the files in that dir. */
|
|
if (prefs.gui_fileopen_dir[0] != '\0')
|
|
esp_dlg.setDirectory(prefs.gui_fileopen_dir);
|
|
break;
|
|
}
|
|
|
|
/* If the file has comments, does the format the user selected
|
|
support them? If not, ask the user whether they want to
|
|
discard the comments or choose a different format. */
|
|
switch(esp_dlg.exportSelectedPackets(file_name, &range)) {
|
|
|
|
case SAVE:
|
|
/* The file can be saved in the specified format as is;
|
|
just drive on and save in the format they selected. */
|
|
discard_comments = FALSE;
|
|
break;
|
|
|
|
case SAVE_WITHOUT_COMMENTS:
|
|
/* The file can't be saved in the specified format as is,
|
|
but it can be saved without the comments, and the user
|
|
said "OK, discard the comments", so save it in the
|
|
format they specified without the comments. */
|
|
discard_comments = TRUE;
|
|
break;
|
|
|
|
case SAVE_IN_ANOTHER_FORMAT:
|
|
/* There are file formats in which we can save this that
|
|
support comments, and the user said not to delete the
|
|
comments. The combo box of file formats has had the
|
|
formats that don't support comments trimmed from it,
|
|
so run the dialog again, to let the user decide
|
|
whether to save in one of those formats or give up. */
|
|
discard_comments = FALSE;
|
|
continue;
|
|
|
|
case CANCELLED:
|
|
/* The user said "forget it". Just get rid of the dialog box
|
|
and return. */
|
|
return;
|
|
}
|
|
|
|
/*
|
|
* Check that we're not going to save on top of the current
|
|
* capture file.
|
|
* We do it here so we catch all cases ...
|
|
* Unfortunately, the file requester gives us an absolute file
|
|
* name and the read file name may be relative (if supplied on
|
|
* the command line). From Joerg Mayer.
|
|
*/
|
|
if (files_identical(cap_file_->filename, file_name.toUtf8().constData())) {
|
|
QMessageBox msg_box;
|
|
gchar *display_basename = g_filename_display_basename(file_name.toUtf8().constData());
|
|
|
|
msg_box.setIcon(QMessageBox::Critical);
|
|
msg_box.setText(QString(tr("Unable to export to \"%1\".").arg(display_basename)));
|
|
msg_box.setInformativeText(tr("You cannot export packets to the current capture file."));
|
|
msg_box.setStandardButtons(QMessageBox::Ok);
|
|
msg_box.setDefaultButton(QMessageBox::Ok);
|
|
msg_box.exec();
|
|
g_free(display_basename);
|
|
continue;
|
|
}
|
|
|
|
file_type = esp_dlg.selectedFileType();
|
|
compressed = esp_dlg.isCompressed();
|
|
fileAddExtension(file_name, file_type, compressed);
|
|
|
|
//#ifndef _WIN32
|
|
// /* If the file exists and it's user-immutable or not writable,
|
|
// ask the user whether they want to override that. */
|
|
// if (!file_target_unwritable_ui(top_level, file_name.toUtf8().constData())) {
|
|
// /* They don't. Let them try another file name or cancel. */
|
|
// continue;
|
|
// }
|
|
//#endif
|
|
|
|
/* Attempt to save the file */
|
|
status = cf_export_specified_packets(cap_file_, file_name.toUtf8().constData(), &range, file_type, compressed);
|
|
switch (status) {
|
|
|
|
case CF_WRITE_OK:
|
|
/* The save succeeded; we're done. */
|
|
/* Save the directory name for future file dialogs. */
|
|
dirname = get_dirname(file_name.toUtf8().data()); /* Overwrites cf_name */
|
|
set_last_open_dir(dirname);
|
|
/* If we discarded comments, redraw the packet list to reflect
|
|
any packets that no longer have comments. */
|
|
if (discard_comments)
|
|
packet_list_queue_draw();
|
|
return;
|
|
|
|
case CF_WRITE_ERROR:
|
|
/* The save failed; let the user try again. */
|
|
continue;
|
|
|
|
case CF_WRITE_ABORTED:
|
|
/* The user aborted the save; just return. */
|
|
return;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
void MainWindow::exportDissections(export_type_e export_type) {
|
|
ExportDissectionDialog ed_dlg(this, cap_file_, export_type);
|
|
packet_range_t range;
|
|
|
|
if (!cap_file_)
|
|
return;
|
|
|
|
/* Init the packet range */
|
|
packet_range_init(&range, cap_file_);
|
|
range.process_filtered = TRUE;
|
|
range.include_dependents = TRUE;
|
|
|
|
ed_dlg.exec();
|
|
}
|
|
|
|
void MainWindow::fileAddExtension(QString &file_name, int file_type, bool compressed) {
|
|
QString file_name_lower;
|
|
QString file_suffix;
|
|
GSList *extensions_list;
|
|
gboolean add_extension;
|
|
|
|
/*
|
|
* Append the default file extension if there's none given by
|
|
* the user or if they gave one that's not one of the valid
|
|
* extensions for the file type.
|
|
*/
|
|
file_name_lower = file_name.toLower();
|
|
extensions_list = wtap_get_file_extensions_list(file_type, FALSE);
|
|
if (extensions_list != NULL) {
|
|
GSList *extension;
|
|
|
|
/* We have one or more extensions for this file type.
|
|
Start out assuming we need to add the default one. */
|
|
add_extension = TRUE;
|
|
|
|
/* OK, see if the file has one of those extensions. */
|
|
for (extension = extensions_list; extension != NULL;
|
|
extension = g_slist_next(extension)) {
|
|
file_suffix += tr(".") + (char *)extension->data;
|
|
if (file_name_lower.endsWith(file_suffix)) {
|
|
/*
|
|
* The file name has one of the extensions for
|
|
* this file type.
|
|
*/
|
|
add_extension = FALSE;
|
|
break;
|
|
}
|
|
file_suffix += ".gz";
|
|
if (file_name_lower.endsWith(file_suffix)) {
|
|
/*
|
|
* The file name has one of the extensions for
|
|
* this file type.
|
|
*/
|
|
add_extension = FALSE;
|
|
break;
|
|
}
|
|
}
|
|
} else {
|
|
/* We have no extensions for this file type. Don't add one. */
|
|
add_extension = FALSE;
|
|
}
|
|
if (add_extension) {
|
|
if (wtap_default_file_extension(file_type) != NULL) {
|
|
file_name += tr(".") + wtap_default_file_extension(file_type);
|
|
if (compressed) {
|
|
file_name += ".gz";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
bool MainWindow::testCaptureFileClose(bool from_quit, QString &before_what) {
|
|
bool capture_in_progress = FALSE;
|
|
|
|
if (!cap_file_ || cap_file_->state == FILE_CLOSED)
|
|
return true; /* Already closed, nothing to do */
|
|
|
|
#ifdef HAVE_LIBPCAP
|
|
if (cap_file_->state == FILE_READ_IN_PROGRESS) {
|
|
/* This is true if we're reading a capture file *or* if we're doing
|
|
a live capture. If we're reading a capture file, the main loop
|
|
is busy reading packets, and only accepting input from the
|
|
progress dialog, so we can't get here, so this means we're
|
|
doing a capture. */
|
|
capture_in_progress = TRUE;
|
|
}
|
|
#endif
|
|
|
|
if (prefs.gui_ask_unsaved) {
|
|
if (cf_has_unsaved_data(cap_file_) || capture_in_progress) {
|
|
QMessageBox msg_dialog;
|
|
QString question;
|
|
QPushButton *saveButton;
|
|
QPushButton *discardButton;
|
|
|
|
msg_dialog.setIcon(QMessageBox::Question);
|
|
|
|
/* This file has unsaved data or there's a capture in
|
|
progress; ask the user whether to save the data. */
|
|
if (cap_file_->is_tempfile) {
|
|
|
|
msg_dialog.setText(tr("You have unsaved packets"));
|
|
msg_dialog.setInformativeText(tr("They will be lost if you don't save them."));
|
|
|
|
if (capture_in_progress) {
|
|
question.append(tr("Do you want to stop the capture and save the captured packets"));
|
|
} else {
|
|
question.append(tr("Do you want to save the captured packets"));
|
|
}
|
|
question.append(before_what).append(tr("?"));
|
|
msg_dialog.setInformativeText(question);
|
|
|
|
|
|
} else {
|
|
/*
|
|
* Format the message.
|
|
*/
|
|
if (capture_in_progress) {
|
|
question.append(tr("Do you want to stop the capture and save the captured packets"));
|
|
question.append(before_what).append(tr("?"));
|
|
msg_dialog.setInformativeText(tr("Your captured packets will be lost if you don't save them."));
|
|
} else {
|
|
gchar *display_basename = g_filename_display_basename(cap_file_->filename);
|
|
question.append(QString(tr("Do you want to save the changes you've made to the capture file \"%1\"%2?"))
|
|
.arg(display_basename)
|
|
.arg(before_what)
|
|
);
|
|
g_free(display_basename);
|
|
msg_dialog.setInformativeText(tr("Your changes will be lost if you don't save them."));
|
|
}
|
|
}
|
|
|
|
// XXX Text comes from ui/gtk/stock_icons.[ch]
|
|
// Note that the button roles differ from the GTK+ version.
|
|
// Cancel = RejectRole
|
|
// Save = AcceptRole
|
|
// Don't Save = DestructiveRole
|
|
msg_dialog.addButton(QMessageBox::Cancel);
|
|
|
|
if (capture_in_progress) {
|
|
saveButton = msg_dialog.addButton(tr("Stop and Save"), QMessageBox::AcceptRole);
|
|
} else {
|
|
saveButton = msg_dialog.addButton(QMessageBox::Save);
|
|
}
|
|
msg_dialog.setDefaultButton(saveButton);
|
|
|
|
if (from_quit) {
|
|
if (cap_file_->state == FILE_READ_IN_PROGRESS) {
|
|
discardButton = msg_dialog.addButton(tr("Stop and Quit without Saving"),
|
|
QMessageBox::DestructiveRole);
|
|
} else {
|
|
discardButton = msg_dialog.addButton(tr("Quit without Saving"),
|
|
QMessageBox::DestructiveRole);
|
|
}
|
|
} else {
|
|
if (capture_in_progress) {
|
|
discardButton = msg_dialog.addButton(tr("Stop and Continue without Saving"),
|
|
QMessageBox::DestructiveRole);
|
|
} else {
|
|
discardButton = msg_dialog.addButton(QMessageBox::Discard);
|
|
}
|
|
}
|
|
|
|
msg_dialog.exec();
|
|
/* According to the Qt doc:
|
|
* when using QMessageBox with custom buttons, exec() function returns an opaque value.
|
|
*
|
|
* Therefore we should use clickedButton() to determine which button was clicked. */
|
|
|
|
if(msg_dialog.clickedButton() == saveButton)
|
|
{
|
|
#ifdef HAVE_LIBPCAP
|
|
/* If there's a capture in progress, we have to stop the capture
|
|
and then do the save. */
|
|
if (capture_in_progress)
|
|
captureStop();
|
|
#endif
|
|
/* Save the file and close it */
|
|
saveCaptureFile(cap_file_, TRUE);
|
|
}
|
|
else if(msg_dialog.clickedButton() == discardButton)
|
|
{
|
|
#ifdef HAVE_LIBPCAP
|
|
/*
|
|
* If there's a capture in progress; we have to stop the capture
|
|
* and then do the close.
|
|
*/
|
|
if (capture_in_progress)
|
|
captureStop();
|
|
#endif
|
|
/* Just close the file, discarding changes */
|
|
cf_close(cap_file_);
|
|
return true;
|
|
}
|
|
else //cancelButton or some other unspecified button
|
|
{
|
|
return false;
|
|
}
|
|
|
|
} else {
|
|
/* Unchanged file, just close it */
|
|
cf_close(cap_file_);
|
|
}
|
|
} else {
|
|
/* User asked not to be bothered by those prompts, just close it.
|
|
XXX - should that apply only to saving temporary files? */
|
|
#ifdef HAVE_LIBPCAP
|
|
/* If there's a capture in progress, we have to stop the capture
|
|
and then do the close. */
|
|
if (capture_in_progress)
|
|
captureStop();
|
|
#endif
|
|
cf_close(cap_file_);
|
|
}
|
|
|
|
return true; /* File closed */
|
|
}
|
|
|
|
void MainWindow::captureStop() {
|
|
stopCapture();
|
|
|
|
while(cap_file_ && cap_file_->state == FILE_READ_IN_PROGRESS) {
|
|
WiresharkApplication::processEvents();
|
|
}
|
|
}
|
|
|
|
// Titlebar
|
|
void MainWindow::setTitlebarForCaptureFile()
|
|
{
|
|
if (cap_file_ && cap_file_->filename) {
|
|
//
|
|
// Qt *REALLY* doesn't like windows that sometimes have a
|
|
// title set with setWindowTitle() and other times have a
|
|
// file path set; apparently, once you've set the title
|
|
// with setWindowTitle(), it sticks, and setWindowFilePath()
|
|
// has no effect. It appears to can clear the title with
|
|
// setWindowTitle(NULL), but that clears the actual title in
|
|
// the title bar, and setWindowFilePath() then, I guess, sees
|
|
// that there's already a file path, and does nothing, leaving
|
|
// the title bar empty. So you then have to clear the file path
|
|
// with setWindowFilePath(NULL), and then set it.
|
|
//
|
|
// Maybe there's a #include "you're holding it wrong" here.
|
|
// However, I really don't want to hear from people who think
|
|
// that a window can never be associated with something other
|
|
// than a user file at time T1 and with a user file at time T2,
|
|
// given that, in Wireshark, a window can be associated with a
|
|
// live capture at time T1 and then, after you've saved the live
|
|
// capture to a user file, associated with a user file at time T2.
|
|
//
|
|
if (cap_file_->is_tempfile) {
|
|
//
|
|
// For a temporary file, put the source of the data
|
|
// in the window title, not whatever random pile
|
|
// of characters is the last component of the path
|
|
// name.
|
|
//
|
|
// XXX - on non-Mac platforms, put in the application
|
|
// name?
|
|
//
|
|
gchar *window_name;
|
|
setWindowFilePath(NULL);
|
|
window_name = g_strdup_printf("Capturing from %s[*]", cf_get_tempfile_source(cap_file_)); //TODO : Fix Translate
|
|
setWindowTitle(window_name);
|
|
g_free(window_name);
|
|
} else {
|
|
//
|
|
// For a user file, set the full path; that way,
|
|
// for OS X, it'll set the "proxy icon". Qt
|
|
// handles extracting the last component.
|
|
//
|
|
// Sadly, some UN*Xes don't necessarily use UTF-8
|
|
// for their file names, so we have to map the
|
|
// file path to UTF-8. If that fails, we're somewhat
|
|
// stuck.
|
|
//
|
|
char *utf8_filename = g_filename_to_utf8(cap_file_->filename,
|
|
-1,
|
|
NULL,
|
|
NULL,
|
|
NULL);
|
|
if (utf8_filename == NULL) {
|
|
// So what the heck else can we do here?
|
|
setWindowTitle(tr("(File name can't be mapped to UTF-8)"));
|
|
} else {
|
|
setWindowTitle(NULL);
|
|
setWindowFilePath(NULL);
|
|
setWindowFilePath(utf8_filename);
|
|
g_free(utf8_filename);
|
|
}
|
|
}
|
|
setWindowModified(cf_has_unsaved_data(cap_file_));
|
|
} else {
|
|
/* We have no capture file. */
|
|
setWindowFilePath(NULL);
|
|
setWindowTitle(tr("The Wireshark Network Analyzer"));
|
|
}
|
|
}
|
|
|
|
void MainWindow::setTitlebarForSelectedTreeRow()
|
|
{
|
|
setWindowTitle(tr("The Wireshark Network Analyzer"));
|
|
}
|
|
|
|
|
|
void MainWindow::setTitlebarForCaptureInProgress()
|
|
{
|
|
gchar *window_name;
|
|
|
|
setWindowFilePath(NULL);
|
|
if (cap_file_) {
|
|
window_name = g_strdup_printf("Capturing from %s", cf_get_tempfile_source(cap_file_)); //TODO : Fix Translate
|
|
setWindowTitle(window_name);
|
|
g_free(window_name);
|
|
} else {
|
|
/* We have no capture in progress. */
|
|
setWindowTitle(tr("The Wireshark Network Analyzer"));
|
|
}
|
|
}
|
|
|
|
// Menu state
|
|
|
|
void MainWindow::setMenusForFollowStream()
|
|
{
|
|
if (!cap_file_)
|
|
return;
|
|
|
|
if (!cap_file_->edt)
|
|
return;
|
|
|
|
main_ui_->actionAnalyzeFollowTCPStream->setEnabled(false);
|
|
main_ui_->actionAnalyzeFollowUDPStream->setEnabled(false);
|
|
main_ui_->actionAnalyzeFollowSSLStream->setEnabled(false);
|
|
|
|
if (cap_file_->edt->pi.ipproto == IP_PROTO_TCP)
|
|
{
|
|
main_ui_->actionAnalyzeFollowTCPStream->setEnabled(true);
|
|
}
|
|
|
|
if (cap_file_->edt->pi.ipproto == IP_PROTO_UDP)
|
|
{
|
|
main_ui_->actionAnalyzeFollowUDPStream->setEnabled(true);
|
|
}
|
|
|
|
if ( epan_dissect_packet_contains_field(cap_file_->edt, "ssl") )
|
|
{
|
|
main_ui_->actionAnalyzeFollowSSLStream->setEnabled(true);
|
|
}
|
|
}
|
|
|
|
/* Enable or disable menu items based on whether you have a capture file
|
|
you've finished reading and, if you have one, whether it's been saved
|
|
and whether it could be saved except by copying the raw packet data. */
|
|
void MainWindow::setMenusForCaptureFile(bool force_disable)
|
|
{
|
|
if (force_disable || cap_file_ == NULL || cap_file_->state == FILE_READ_IN_PROGRESS) {
|
|
/* We have no capture file or we're currently reading a file */
|
|
main_ui_->actionFileMerge->setEnabled(false);
|
|
main_ui_->actionFileClose->setEnabled(false);
|
|
main_ui_->actionFileSave->setEnabled(false);
|
|
main_ui_->actionFileSaveAs->setEnabled(false);
|
|
main_ui_->actionSummary->setEnabled(false);
|
|
main_ui_->actionFileExportPackets->setEnabled(false);
|
|
main_ui_->menuFileExportPacketDissections->setEnabled(false);
|
|
main_ui_->actionFileExportPacketBytes->setEnabled(false);
|
|
main_ui_->actionFileExportSSLSessionKeys->setEnabled(false);
|
|
main_ui_->menuFileExportObjects->setEnabled(false);
|
|
main_ui_->actionViewReload->setEnabled(false);
|
|
} else {
|
|
main_ui_->actionFileMerge->setEnabled(cf_can_write_with_wiretap(cap_file_));
|
|
|
|
main_ui_->actionFileClose->setEnabled(true);
|
|
main_ui_->actionFileSave->setEnabled(cf_can_save(cap_file_));
|
|
main_ui_->actionFileSaveAs->setEnabled(cf_can_save_as(cap_file_));
|
|
main_ui_->actionSummary->setEnabled(true);
|
|
/*
|
|
* "Export Specified Packets..." should be available only if
|
|
* we can write the file out in at least one format.
|
|
*/
|
|
main_ui_->actionFileExportPackets->setEnabled(cf_can_write_with_wiretap(cap_file_));
|
|
main_ui_->menuFileExportPacketDissections->setEnabled(true);
|
|
main_ui_->actionFileExportPacketBytes->setEnabled(true);
|
|
main_ui_->actionFileExportSSLSessionKeys->setEnabled(true);
|
|
main_ui_->menuFileExportObjects->setEnabled(true);
|
|
main_ui_->actionViewReload->setEnabled(true);
|
|
}
|
|
}
|
|
|
|
void MainWindow::setMenusForCaptureInProgress(bool capture_in_progress) {
|
|
/* Either a capture was started or stopped; in either case, it's not
|
|
in the process of stopping, so allow quitting. */
|
|
|
|
main_ui_->actionFileOpen->setEnabled(!capture_in_progress);
|
|
main_ui_->menuOpenRecentCaptureFile->setEnabled(!capture_in_progress);
|
|
main_ui_->menuFileExportPacketDissections->setEnabled(capture_in_progress);
|
|
main_ui_->actionFileExportPacketBytes->setEnabled(capture_in_progress);
|
|
main_ui_->actionFileExportSSLSessionKeys->setEnabled(capture_in_progress);
|
|
main_ui_->menuFileExportObjects->setEnabled(capture_in_progress);
|
|
main_ui_->menuFileSet->setEnabled(!capture_in_progress);
|
|
main_ui_->actionFileQuit->setEnabled(true);
|
|
|
|
main_ui_->actionSummary->setEnabled(capture_in_progress);
|
|
|
|
qDebug() << "FIX: packet list heading menu sensitivity";
|
|
// set_menu_sensitivity(ui_manager_packet_list_heading, "/PacketListHeadingPopup/SortAscending",
|
|
// !capture_in_progress);
|
|
// set_menu_sensitivity(ui_manager_packet_list_heading, "/PacketListHeadingPopup/SortDescending",
|
|
// !capture_in_progress);
|
|
// set_menu_sensitivity(ui_manager_packet_list_heading, "/PacketListHeadingPopup/NoSorting",
|
|
// !capture_in_progress);
|
|
|
|
#ifdef HAVE_LIBPCAP
|
|
main_ui_->actionCaptureOptions->setEnabled(!capture_in_progress);
|
|
main_ui_->actionStartCapture->setEnabled(!capture_in_progress);
|
|
main_ui_->actionStartCapture->setChecked(capture_in_progress);
|
|
main_ui_->actionStopCapture->setEnabled(capture_in_progress);
|
|
main_ui_->actionCaptureRestart->setEnabled(capture_in_progress);
|
|
#endif /* HAVE_LIBPCAP */
|
|
|
|
}
|
|
|
|
void MainWindow::setMenusForCaptureStopping() {
|
|
main_ui_->actionFileQuit->setEnabled(false);
|
|
main_ui_->actionSummary->setEnabled(false);
|
|
#ifdef HAVE_LIBPCAP
|
|
main_ui_->actionStartCapture->setChecked(false);
|
|
main_ui_->actionStopCapture->setEnabled(false);
|
|
main_ui_->actionCaptureRestart->setEnabled(false);
|
|
#endif /* HAVE_LIBPCAP */
|
|
}
|
|
|
|
void MainWindow::setForCapturedPackets(bool have_captured_packets)
|
|
{
|
|
main_ui_->actionFilePrint->setEnabled(have_captured_packets);
|
|
|
|
// set_menu_sensitivity(ui_manager_packet_list_menu, "/PacketListMenuPopup/Print",
|
|
// have_captured_packets);
|
|
|
|
main_ui_->actionEditFindPacket->setEnabled(have_captured_packets);
|
|
main_ui_->actionEditFindNext->setEnabled(have_captured_packets);
|
|
main_ui_->actionEditFindPrevious->setEnabled(have_captured_packets);
|
|
// set_menu_sensitivity(ui_manager_main_menubar, "/Menubar/ViewMenu/ZoomIn",
|
|
// have_captured_packets);
|
|
// set_menu_sensitivity(ui_manager_main_menubar, "/Menubar/ViewMenu/ZoomOut",
|
|
// have_captured_packets);
|
|
// set_menu_sensitivity(ui_manager_main_menubar, "/Menubar/ViewMenu/NormalSize",
|
|
// have_captured_packets);
|
|
|
|
main_ui_->actionGoGoToPacket->setEnabled(have_captured_packets);
|
|
main_ui_->actionGoPreviousPacket->setEnabled(have_captured_packets);
|
|
main_ui_->actionGoNextPacket->setEnabled(have_captured_packets);
|
|
main_ui_->actionGoFirstPacket->setEnabled(have_captured_packets);
|
|
main_ui_->actionGoLastPacket->setEnabled(have_captured_packets);
|
|
|
|
// set_menu_sensitivity(ui_manager_main_menubar, "/Menubar/GoMenu/PreviousPacketInConversation",
|
|
// have_captured_packets);
|
|
// set_menu_sensitivity(ui_manager_main_menubar, "/Menubar/GoMenu/NextPacketInConversation",
|
|
// have_captured_packets);
|
|
// set_menu_sensitivity(ui_manager_main_menubar, "/Menubar/StatisticsMenu/Summary",
|
|
// have_captured_packets);
|
|
// set_menu_sensitivity(ui_manager_main_menubar, "/Menubar/StatisticsMenu/ProtocolHierarchy",
|
|
// have_captured_packets);
|
|
}
|
|
|
|
void MainWindow::setMenusForFileSet(bool enable_list_files) {
|
|
bool enable_next = fileset_get_next() != NULL && enable_list_files;
|
|
bool enable_prev = fileset_get_previous() != NULL && enable_list_files;
|
|
|
|
main_ui_->actionFileSetListFiles->setEnabled(enable_list_files);
|
|
main_ui_->actionFileSetNextFile->setEnabled(enable_next);
|
|
main_ui_->actionFileSetPreviousFile->setEnabled(enable_prev);
|
|
}
|
|
|
|
void MainWindow::updateForUnsavedChanges() {
|
|
setTitlebarForCaptureFile();
|
|
setMenusForCaptureFile();
|
|
// set_toolbar_for_capture_file(cf);
|
|
|
|
}
|
|
|
|
/* Update main window items based on whether there's a capture in progress. */
|
|
void MainWindow::setForCaptureInProgress(gboolean capture_in_progress)
|
|
{
|
|
setMenusForCaptureInProgress(capture_in_progress);
|
|
|
|
//#ifdef HAVE_LIBPCAP
|
|
// set_toolbar_for_capture_in_progress(capture_in_progress);
|
|
|
|
// set_capture_if_dialog_for_capture_in_progress(capture_in_progress);
|
|
//#endif
|
|
}
|
|
|
|
/*
|
|
* Editor modelines
|
|
*
|
|
* Local Variables:
|
|
* c-basic-offset: 4
|
|
* tab-width: 8
|
|
* indent-tabs-mode: nil
|
|
* End:
|
|
*
|
|
* ex: set shiftwidth=4 tabstop=8 expandtab:
|
|
* :indentSize=4:tabSize=8:noTabs=true:
|
|
*/
|