From Thomas ERSFELD (GSoC13)

Add TCP/UDP/SSL Follow feature to QtShark

Known issue :
* Duplicate code with GTK (function need follow_info_t struct but in GTK there is some GWidget variable in struct)
* Sometimes TCP Follow fail...

svn path=/trunk/; revision=51883

1 files changed, 161 insertions, 0 deletions

diff --git a/ui/qt/follow_stream_dialog.h b/ui/qt/follow_stream_dialog.h
new file mode 100644
index 0000000000..c977f0cfa6
--- /dev/null
+++ b/ui/qt/follow_stream_dialog.h
@@ -0,0 +1,161 @@
+/* follow_stream_dialog.h
+ *
+ * $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.
+ */
+
+#ifndef FOLLOW_STREAM_DIALOG_H
+#define FOLLOW_STREAM_DIALOG_H
+
+#include <QDialog>
+#include <QMessageBox>
+
+#include <unistd.h>
+
+#include "config.h"
+
+#include "qt_ui_utils.h"
+
+#include <epan/follow.h>
+#include <epan/dissectors/packet-ipv6.h>
+#include <epan/dissectors/packet-tcp.h>
+#include <epan/prefs.h>
+#include <epan/addr_resolv.h>
+#include <epan/charsets.h>
+#include <epan/epan_dissect.h>
+#include <epan/filesystem.h>
+#include <epan/ipproto.h>
+#include <epan/charsets.h>
+#include <epan/plugins.h>
+#include <epan/tap.h>
+
+#include "../file.h"
+#include "ui/alert_box.h"
+#include "ui/simple_dialog.h"
+#include "ui/utf8_entities.h"
+#include "wsutil/tempfile.h"
+#include <wsutil/file_util.h>
+#include "ws_symbol_export.h"
+
+
+#include "globals.h"
+#include "file.h"
+
+#include "version_info.h"
+
+
+#include <QtGui>
+#include "../follow.h"
+
+WS_DLL_PUBLIC FILE *data_out_file;
+
+typedef struct {
+    follow_type_t   follow_type;
+    show_stream_t   show_stream;
+    show_type_t     show_type;
+    char            *data_out_filename;
+    gboolean        is_ipv6;
+    char            *filter_out_filter;
+    GList           *payload;
+    guint           bytes_written[2]; /* Index with FROM_CLIENT or FROM_SERVER for readability. */
+    guint           client_port;
+    address         client_ip;
+} follow_info_t;
+
+namespace Ui {
+class FollowStreamDialog;
+}
+
+class FollowStreamDialog : public QDialog
+{
+    Q_OBJECT
+
+public:
+    explicit FollowStreamDialog(QWidget *parent = 0);
+    ~FollowStreamDialog();
+
+    bool Follow(QString previous_filter_, follow_type_t type);
+
+    frs_return_t
+    follow_show(char *buffer, size_t nchars, gboolean is_from_server,
+            guint32 *global_pos, guint32 *server_packet_count,
+            guint32 *client_packet_count);
+
+    frs_return_t
+    follow_read_stream();
+
+    frs_return_t
+    follow_read_tcp_stream();
+
+    frs_return_t
+    follow_read_udp_stream();
+
+    frs_return_t
+    follow_read_ssl_stream();
+
+    void
+    follow_stream();
+
+    void add_text(char *buffer, size_t nchars, gboolean is_from_server);
+
+private slots:
+    void on_cbCharset_currentIndexChanged(int index);
+    void on_cbDirections_currentIndexChanged(int index);
+    void HelpButton();
+    void FilterOut();
+    void FindText();
+    void SaveAs();
+    void Print();
+//    void on_bNext_clicked();
+//    void on_bPrevious_clicked();
+
+signals:
+    void updateFilter(QString &filter, bool force);
+
+private:
+    Ui::FollowStreamDialog  *ui;
+
+    QPushButton             *bFilterOut;
+    QPushButton             *bFind;
+    QPushButton             *bPrint;
+    QPushButton             *bSave;
+
+    follow_info_t           *follow_info;
+
+    bool                    save_as;
+    QFile                   file;
+};
+
+
+
+#endif // FOLLOW_STREAM_DIALOG_H
+
+/*
+ * Editor modelines  -  http://www.wireshark.org/tools/modelines.html
+ *
+ * Local variables:
+ * c-basic-offset: 4
+ * tab-width: 8
+ * indent-tabs-mode: nil
+ * End:
+ *
+ * vi: set shiftwidth=4 tabstop=8 expandtab:
+ * :indentSize=4:tabSize=8:noTabs=true:
+ */