follow_stream_dialog.h

Go to the documentation of this file.

 
#ifndef FOLLOW_STREAM_DIALOG_H
#define FOLLOW_STREAM_DIALOG_H
 
#include <config.h>
 
#include <stdio.h>
 
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
 
#include "file.h"
 
#include "epan/follow.h"
 
#include "wireshark_dialog.h"
 
#include <QFile>
#include <QMap>
#include <QPushButton>
 
namespace Ui {
class FollowStreamDialog;
}

class FollowStreamDialog : public WiresharkDialog
{
    Q_OBJECT
 
public:
    explicit FollowStreamDialog(QWidget &parent, CaptureFile &cf, int proto_id, const QString& previous_filter);

    ~FollowStreamDialog();

    void addCodecs(const QMap<QString, const char *> &codecMap);

    bool follow(bool use_stream_index = false, unsigned stream_num = 0, unsigned sub_stream_num = 0);
 
protected:
    bool eventFilter(QObject *obj, QEvent *event) override;

    void keyPressEvent(QKeyEvent *event) override;
 

    void endRetapPackets() override;

    void captureFileClosed() override;

    virtual QString labelHint(int pkt = 0);

    int client_packet_count() const { return client_packet_count_; }

    int server_packet_count() const { return server_packet_count_; }

    int turns() const { return turns_; }

    const follow_info_t& followInfo() const { return follow_info_; }

    virtual QString serverToClientString() const;

    virtual QString clientToServerString() const;

    virtual QString bothDirectionsString() const;
 
 
private slots:
    void cbCharsetCurrentIndexChanged(int idx);

    void deltaComboBoxCurrentIndexChanged(int idx);

    void cbDirectionsCurrentIndexChanged(int idx);

    void bFindClicked();

    void leFindReturnPressed();

    void helpButton();

    void backButton();

    void close();

    void useRegexFind(bool use_regex);

    void findText(bool go_back = true);

    void saveAs();

    void printStream();

    void fillHintLabel(int pkt = 0);

    void goToPacketForTextPos(int pkt = 0);

    void streamNumberSpinBoxValueChanged(int stream_num);

    void subStreamNumberSpinBoxValueChanged(int sub_stream_num);

    void buttonBoxRejected();
 
signals:
    void updateFilter(QString filter, bool force);

    void goToPacket(int packet_num);
 
private:
    static void resetStream(void *tapData);

    void removeStreamControls();

    void resetStream(void);

    void updateWidgets(bool follow_in_progress);

    void updateWidgets() override { updateWidgets(false); } // Needed for WiresharkDialog?

    void showBuffer(QByteArray &buffer, size_t nchars, bool is_from_server,
                uint32_t packet_num, nstime_t abs_ts, uint32_t *global_pos);

    void readStream();

    void readFollowStream();

    void followStream();

    void addText(QString text, bool is_from_server, uint32_t packet_num, bool colorize = true);
 
    void filterMenuAboutToShow(QMenu *, bool);

    Ui::FollowStreamDialog  *ui;

    QPushButton             *b_filter_prepare_;

    QPushButton             *b_filter_apply_;

    QPushButton             *b_find_;

    QPushButton             *b_print_;

    QPushButton             *b_save_;

    QPushButton             *b_back_;

    follow_info_t           follow_info_;

    register_follow_t*      follower_;

    QString                 previous_filter_;

    QString                 output_filter_;

    QString                 follow_filter_;

    int                     client_buffer_count_;

    int                     server_buffer_count_;

    int                     client_packet_count_;

    int                     server_packet_count_;

    uint32_t                last_packet_;

    bool                    last_from_server_;

    nstime_t                last_ts_;

    int                     turns_;

    bool                    use_regex_find_;

    bool                    terminating_;

    int                     previous_sub_stream_num_;
};
 
#endif // FOLLOW_STREAM_DIALOG_H