1 files changed, 205 insertions, 0 deletions
diff --git a/ui/qt/main_window.cpp b/ui/qt/main_window.cpp
new file mode 100644
index 00000000..0f54995f
--- /dev/null
+++ b/ui/qt/main_window.cpp
@@ -0,0 +1,205 @@
+/* main_window.cpp
+ *
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#include "config.h"
+
+#include <glib.h>
+
+#include "ui/preference_utils.h"
+
+#include "main_window.h"
+
+#include "funnel_statistics.h"
+#include "packet_list.h"
+#include "widgets/display_filter_combo.h"
+
+// Packet Menu actions
+static QList<QAction *> dynamic_packet_menu_actions;
+
+MainWindow::MainWindow(QWidget *parent) :
+    QMainWindow(parent),
+    main_stack_(nullptr),
+    welcome_page_(nullptr),
+    cur_layout_(QVector<unsigned>()),
+    packet_list_(nullptr),
+    proto_tree_(nullptr),
+    byte_view_tab_(nullptr),
+    packet_diagram_(nullptr),
+    df_combo_box_(nullptr),
+    main_status_bar_(nullptr)
+{
+
+}
+
+MainWindow::~MainWindow()
+{
+    clearAddedPacketMenus();
+}
+
+bool MainWindow::hasSelection()
+{
+    if (packet_list_)
+        return packet_list_->multiSelectActive();
+    return false;
+}
+
+/*
+ * As hasSelection() is not looking for one single packet
+ * selection, but at least 2, this method returns TRUE in
+ * this specific case.
+ */
+bool MainWindow::hasUniqueSelection()
+{
+    if (packet_list_)
+        return packet_list_->uniqueSelectActive();
+    return false;
+}
+
+QList<int> MainWindow::selectedRows(bool useFrameNum)
+{
+    if (packet_list_)
+        return packet_list_->selectedRows(useFrameNum);
+    return QList<int>();
+}
+
+frame_data* MainWindow::frameDataForRow(int row) const
+{
+    if (packet_list_)
+        return packet_list_->getFDataForRow(row);
+
+    return Q_NULLPTR;
+}
+
+void MainWindow::insertColumn(QString name, QString abbrev, gint pos)
+{
+    gint colnr = 0;
+    if (name.length() > 0 && abbrev.length() > 0)
+    {
+        colnr = column_prefs_add_custom(COL_CUSTOM, name.toStdString().c_str(), abbrev.toStdString().c_str(), pos);
+        packet_list_->columnsChanged();
+        packet_list_->resizeColumnToContents(colnr);
+        prefs_main_write();
+    }
+}
+
+void MainWindow::gotoFrame(int packet_num)
+{
+    if (packet_num > 0) {
+        packet_list_->goToPacket(packet_num);
+    }
+}
+
+QString MainWindow::getFilter()
+{
+    return df_combo_box_->currentText();
+}
+
+MainStatusBar *MainWindow::statusBar()
+{
+    return main_status_bar_;
+}
+
+void MainWindow::setDisplayFilter(QString filter, FilterAction::Action action, FilterAction::ActionType filterType)
+{
+    emit filterAction(filter, action, filterType);
+}
+
+/*
+ * Used for registering custom packet menus
+ *
+ * @param funnel_action a custom packet menu action
+ */
+void MainWindow::appendPacketMenu(QAction* funnel_action)
+{
+    dynamic_packet_menu_actions.append(funnel_action);
+    connect(funnel_action, SIGNAL(triggered(bool)), funnel_action, SLOT(triggerPacketCallback()));
+}
+
+/*
+ * Returns the list of registered packet menu actions
+ *
+ * After ensuring that all stored custom packet menu actions
+ * are registered with the Wireshark GUI, it returns them as a list
+ * so that they can potentially be displayed to a user.
+ *
+ * @return the list of registered packet menu actions
+ */
+QList<QAction *> MainWindow::getPacketMenuActions()
+{
+    if (funnel_statistics_packet_menus_modified()) {
+        // If the packet menus were modified, we need to clear the already
+        // loaded packet menus to avoid duplicates
+        this->clearAddedPacketMenus();
+        funnel_statistics_load_packet_menus();
+    }
+    return dynamic_packet_menu_actions;
+}
+
+/*
+ * Clears the list of registered packet menu actions
+ *
+ * Clears the list of registered packet menu actions
+ * and frees all associated memory.
+ */
+void MainWindow::clearAddedPacketMenus()
+{
+    for( int i=0; i<dynamic_packet_menu_actions.count(); ++i )
+    {
+        delete dynamic_packet_menu_actions[i];
+    }
+    dynamic_packet_menu_actions.clear();
+}
+
+
+/*
+ * Adds the custom packet menus to the supplied QMenu
+ *
+ * This method takes in QMenu and the selected packet's data
+ * and adds all applicable custom packet menus to it.
+ *
+ * @param ctx_menu The menu to add the packet menu entries to
+ * @param finfo_array The data in the selected packet
+ * @return true if a packet menu was added to the ctx_menu
+ */
+bool MainWindow::addPacketMenus(QMenu * ctx_menu, GPtrArray *finfo_array)
+{
+    bool insertedPacketMenu = false;
+    QList<QAction *> myPacketMenuActions = this->getPacketMenuActions();
+    if (myPacketMenuActions.isEmpty()) {
+        return insertedPacketMenu;
+    }
+
+    // Build a set of fields present for efficient lookups
+    QSet<QString> fieldsPresent = QSet<QString>();
+    for (guint fieldInfoIndex = 0; fieldInfoIndex < finfo_array->len; fieldInfoIndex++) {
+        field_info *fi = (field_info *)g_ptr_array_index (finfo_array, fieldInfoIndex);
+        fieldsPresent.insert(QString(fi->hfinfo->abbrev));
+    }
+
+    // Place actions in the relevant (sub)menu
+    // The 'root' menu is the ctx_menu, so map NULL to that
+    QHash<QString, QMenu *> menuTextToMenus;
+    menuTextToMenus.insert(NULL, ctx_menu);
+    foreach (QAction * action, myPacketMenuActions) {
+        if (! qobject_cast<FunnelAction *>(action)) {
+            continue;
+        }
+        FunnelAction * packetAction = qobject_cast<FunnelAction *>(action);
+
+        // Only display a menu if all required fields are present
+        if (!fieldsPresent.contains(packetAction->getPacketRequiredFields())) {
+            continue;
+        }
+
+        packetAction->setPacketData(finfo_array);
+        packetAction->addToMenu(ctx_menu, menuTextToMenus);
+        insertedPacketMenu = true;
+    }
+    return insertedPacketMenu;
+}