packet_list.h

Go to the documentation of this file.

 
#ifndef PACKET_LIST_H
#define PACKET_LIST_H
 
#include "data_source_tab.h"
#include <ui/qt/models/packet_list_model.h>
#include "proto_tree.h"
#include "protocol_preferences_menu.h"
#include <ui/qt/models/related_packet_delegate.h>
#include <ui/qt/models/multi_color_packet_delegate.h>
#include <ui/qt/utils/field_information.h>
 
#include <QMenu>
#include <QTime>
#include <QTreeView>
#include <QPainter>
 
class PacketListHeader;
class OverlayScrollBar;
class ProfileSwitcher;
 
class QAction;
class QTimerEvent;
 
//
// XXX - Wireshark supports up to 2^32-1 packets in a capture, but
// row numbers in a QAbstractItemModel are ints, not unsigned ints,
// so we can only have 2^31-1 rows on ILP32, LP64, and LLP64 platforms.
// Does that mean we're permanently stuck at a maximum of 2^31-1 packets
// per capture?
//
 
class PacketList : public QTreeView
{
    Q_OBJECT
public:
    explicit PacketList(QWidget *parent = 0);
 
    ~PacketList();
 
    enum SummaryCopyType {
        CopyAsText,
        CopyAsCSV,
        CopyAsYAML,
        CopyAsHTML
    };
    Q_ENUM(SummaryCopyType)
 
    
    virtual void scrollTo(const QModelIndex &index, QAbstractItemView::ScrollHint hint = EnsureVisible) override;
 
    QMenu *conversationMenu() { return &conv_menu_; }
 
    QMenu *colorizeMenu() { return &colorize_menu_; }
 
    void setProtoTree(ProtoTree *proto_tree);
 
    bool freeze(bool keep_current_frame = false);
 
    bool thaw(bool restore_selection = false);
 
    void clear();
 
    void writeRecent(FILE *rf);
 
    bool contextMenuActive();
 
    QString getFilterFromRowAndColumn(QModelIndex idx);
 
    void resetColorized();
 
    QString getPacketComment(unsigned c_number);
 
    void addPacketComment(QString new_comment);
 
    void setPacketComment(unsigned c_number, QString new_comment);
 
    QString allPacketComments();
 
    void deleteCommentsFromPackets();
 
    void deleteAllPacketComments();
 
    void setVerticalAutoScroll(bool enabled = true);
 
    void setCaptureInProgress(bool in_progress = false, bool auto_scroll = true) { capture_in_progress_ = in_progress; tail_at_end_ = in_progress && auto_scroll; }
 
    void captureFileReadFinished();
 
    void setColumnDelegate();
 
    void resetColumns();
 
    bool haveNextHistory(bool update_cur = false);
 
    bool havePreviousHistory(bool update_cur = false);
 
    void setProfileSwitcher(ProfileSwitcher *profile_switcher);
 
    frame_data * getFDataForRow(int row) const;
 
    bool uniqueSelectActive();
 
    bool multiSelectActive();
 
    QList<int> selectedRows(bool useFrameNum = false);
 
    QString createSummaryText(QModelIndex idx, SummaryCopyType type);
 
    QString createHeaderSummaryText(SummaryCopyType type);
 
    QStringList createHeaderPartsForAligned();
 
    QList<int> createAlignmentPartsForAligned();
 
    QList<int> createSizePartsForAligned(bool useHeader, QStringList hdr_parts, QList<int> rows);
 
    QString createHeaderSummaryForAligned(QStringList hdr_parts, QList<int> align_parts, QList<int> size_parts);
 
    QString createSummaryForAligned(QModelIndex idx, QList<int> align_parts, QList<int> size_parts);
 
    QString createDefaultStyleForHtml();
 
    QString createOpeningTagForHtml();
 
    QString createHeaderSummaryForHtml();
 
    QString createSummaryForHtml(QModelIndex idx);
 
    QString createClosingTagForHtml();
 
    void resizeAllColumns(bool onlyTimeFormatted = false);
 
protected:
    void selectionChanged(const QItemSelection & selected, const QItemSelection & deselected) override;
 
    virtual void contextMenuEvent(QContextMenuEvent *event) override;
 
    void timerEvent(QTimerEvent *event) override;
 
    void paintEvent(QPaintEvent *event) override;
 
    virtual void mousePressEvent (QMouseEvent *event) override;
 
    virtual void mouseReleaseEvent (QMouseEvent *event) override;
 
    virtual void mouseMoveEvent (QMouseEvent *event) override;
 
    virtual void resizeEvent(QResizeEvent *event) override;
 
    virtual void keyPressEvent(QKeyEvent *event) override;
 
    virtual void focusInEvent(QFocusEvent *event) override;
 
protected slots:
    void rowsInserted(const QModelIndex &parent, int start, int end) override;
 
    virtual void drawRow(QPainter *painter, const QStyleOptionViewItem &option,
        const QModelIndex &index) const override;
 
private:
    PacketListModel *packet_list_model_;
 
    PacketListHeader * packet_list_header_;
 
    ProtoTree *proto_tree_;
 
    capture_file *cap_file_;
 
    QMenu conv_menu_;
 
    QMenu colorize_menu_;
 
    int ctx_column_;
 
    QByteArray column_state_;
 
    OverlayScrollBar *overlay_sb_;
 
    int overlay_timer_id_;
 
    bool create_near_overlay_;
 
    bool create_far_overlay_;
 
    QVector<QRgb> overlay_colors_;
 
    bool changing_profile_;
 
    QModelIndex mouse_pressed_at_;
 
    RelatedPacketDelegate related_packet_delegate_;
 
    MultiColorPacketDelegate multi_color_delegate_;
 
    QAction *show_hide_separator_;
 
    QList<QAction *>show_hide_actions_;
 
    bool capture_in_progress_;
 
    bool tail_at_end_;
 
    bool columns_changed_;
 
    bool set_column_visibility_;
 
    bool set_style_sheet_;
 
    QModelIndex frozen_current_row_;
 
    QModelIndexList frozen_selected_rows_;
 
    QVector<int> selection_history_;
 
    int cur_history_;
 
    bool in_history_;
 
    GPtrArray *finfo_array;
 
    ProfileSwitcher *profile_switcher_;
 
    void setFrameReftime(bool set, frame_data *fdata);
 
    void setColumnVisibility();
 
    void setRecentColumnWidth(int column);
 
    void drawCurrentPacket();
 
    void applyRecentColumnWidths();
 
    void scrollViewChanged(bool at_end);
 
    QString joinSummaryRow(QStringList col_parts, int row, SummaryCopyType type);
 
signals:
    void packetDissectionChanged();
 
    void showColumnPreferences(QString pane_name);
 
    void editColumn(int column);
 
    void packetListScrolled(bool at_end);
 
    void showProtocolPreferences(const QString module_name);
 
    void editProtocolPreference(pref_t *pref, module_t *module);
 
    void framesSelected(QList<int> frames);
 
    void fieldSelected(FieldInformation *finfo);
 
public slots:
    void setCaptureFile(capture_file *cf);
 
    void setMonospaceFont(const QFont &mono_font);
 
    void setRegularFont(const QFont &regular_font);
 
    void goNextPacket();
 
    void goPreviousPacket();
 
    void goFirstPacket();
 
    void goLastPacket();
 
    void goToPacket(int packet, int hf_id = -1);
 
    void goNextHistoryPacket();
 
    void goPreviousHistoryPacket();
 
    void markFrame();
 
    void markAllDisplayedFrames(bool set);
 
    void ignoreFrame();
 
    void ignoreAllDisplayedFrames(bool set);
 
    void setTimeReference();
 
    void unsetAllTimeReferences();
 
    void applyTimeShift();
 
    void recolorPackets();
 
    void redrawVisiblePackets();
 
    void redrawVisiblePacketsDontSelectCurrent();
 
    void colorsChanged();
 
    void columnsChanged();
 
    void fieldsChanged(capture_file *cf);
 
    void preferencesChanged();
 
    void freezePacketList(bool changing_profile);
 
private slots:
    void columnVisibilityTriggered();
 
    void sectionResized(int col, int, int new_width);
 
    void sectionMoved(int logicalIndex, int oldVisualIndex, int newVisualIndex);
 
    void copySummary();
 
    void vScrollBarActionTriggered(int);
 
    void drawFarOverlay();
 
    void drawNearOverlay();
 
    void updatePackets(bool redraw);
 
    void ctxDecodeAsDialog();
};
 
#endif // PACKET_LIST_H