1 files changed, 4 insertions, 17 deletions
diff --git a/ui/qt/follow_stream_dialog.h b/ui/qt/follow_stream_dialog.h
index da0457175c..8508034fe8 100644
--- a/ui/qt/follow_stream_dialog.h
+++ b/ui/qt/follow_stream_dialog.h
@@ -34,7 +34,7 @@
 
 #include "file.h"
 
-#include "ui/follow.h"
+#include "epan/follow.h"
 
 #include "wireshark_dialog.h"
 
@@ -42,21 +42,6 @@
 #include <QMap>
 #include <QPushButton>
 
-extern "C" {
-WS_DLL_PUBLIC FILE *data_out_file;
-}
-
-// Shouldn't these be member variables?
-typedef struct {
-    show_stream_t   show_stream;
-    show_type_t     show_type;
-    gboolean        is_ipv6;
-    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;
 }
@@ -69,7 +54,7 @@ public:
     explicit FollowStreamDialog(QWidget &parent, CaptureFile &cf, follow_type_t type = FOLLOW_TCP);
     ~FollowStreamDialog();
 
-    bool follow(QString previous_filter = QString(), bool use_stream_index = false);
+    bool follow(QString previous_filter = QString(), bool use_stream_index = false, int stream_num = -1);
 
 public slots:
     void captureFileClosing();
@@ -125,6 +110,8 @@ private:
 
     follow_type_t           follow_type_;
     follow_info_t           follow_info_;
+    register_follow_t*      follower_;
+    show_type_t             show_type_;
     QString                 data_out_filename_;
     static const int        max_document_length_;
     bool                    truncated_;