atap_data_model.h

Go to the documentation of this file.

 
#ifndef ATAP_DATA_MODEL_H
#define ATAP_DATA_MODEL_H
 
#include "config.h"
 
#include <epan/tap.h>
#include <epan/conversation.h>
#include <epan/conversation_table.h>
 
#include <QAbstractListModel>
 
class ATapDataModel : public QAbstractListModel
{
    Q_OBJECT
public:
 
    enum {
        DISPLAY_FILTER = Qt::UserRole,
        UNFORMATTED_DISPLAYDATA,
#ifdef HAVE_MAXMINDDB
        GEODATA_AVAILABLE,
        GEODATA_LOOKUPTABLE,
        GEODATA_ADDRESS,
#endif
        TIMELINE_DATA,
        ENDPOINT_DATATYPE,
        PROTO_ID,
        CONVERSATION_ID,
        ROW_IS_FILTERED,
        DATA_ADDRESS_TYPE,
        DATA_IPV4_INTEGER,
        DATA_IPV6_LIST,
    };
 
    typedef enum {
        DATAMODEL_ENDPOINT,
        DATAMODEL_CONVERSATION,
        DATAMODEL_UNKNOWN
    } dataModelType;
 
    conv_hash_t hash_;
 
    explicit ATapDataModel(dataModelType type, int protoId, QString filter, QObject *parent = nullptr);
    virtual ~ATapDataModel();
 
    int rowCount(const QModelIndex &parent = QModelIndex()) const;
 
    virtual int columnCount(const QModelIndex &parent = QModelIndex()) const = 0;
    virtual QVariant headerData(int section, Qt::Orientation orientation = Qt::Horizontal, int role = Qt::DisplayRole) const = 0;
    virtual QVariant data(const QModelIndex &idx, int role = Qt::DisplayRole) const = 0;
 
    QString tap() const;
 
    int protoId() const;
 
    void setFilter(QString filter);
 
    QString filter() const;
 
    bool resolveNames() const;
 
    void setResolveNames(bool resolve);
 
    bool allowsNameResolution() const;
 
    void useAbsoluteTime(bool absolute);
 
    void useNanosecondTimestamps(bool nanoseconds);
 
    bool portsAreHidden() const;
 
    bool showTotalColumn() const;
 
    bool enableTap();
 
    void disableTap();
 
    dataModelType modelType() const;
 
    conv_hash_t * hash();
 
    void updateFlags(unsigned flag);
 
#ifdef HAVE_MAXMINDDB
    bool hasGeoIPData();
#endif
 
signals:
    void tapListenerChanged(bool enable);
 
protected:
 
    static void tapReset(void *tapdata);
    static void tapDraw(void *tap_data);
 
    virtual tap_packet_cb conversationPacketHandler();
 
    void resetData();
    void updateData(GArray * data);
 
    dataModelType _type;
    GArray * storage_;
    QString _filter;
 
    bool _absoluteTime;
    bool _nanoseconds;
    bool _resolveNames;
    bool _disableTap;
 
    double _minRelStartTime;
    double _maxRelStopTime;
 
    unsigned _tapFlags;
 
    register_ct_t* registerTable() const;
 
private:
    int _protoId;
 
};
 
class EndpointDataModel : public ATapDataModel
{
    Q_OBJECT
public:
 
    typedef enum
    {
        ENDP_COLUMN_ADDR,
        ENDP_COLUMN_PORT,
        ENDP_COLUMN_PACKETS,
        ENDP_COLUMN_BYTES,
        ENDP_COLUMN_PACKETS_TOTAL,
        ENDP_COLUMN_BYTES_TOTAL,
        ENDP_COLUMN_PKT_AB,
        ENDP_COLUMN_BYTES_AB,
        ENDP_COLUMN_PKT_BA,
        ENDP_COLUMN_BYTES_BA,
        ENDP_NUM_COLUMNS,
        ENDP_COLUMN_GEO_COUNTRY = ENDP_NUM_COLUMNS,
        ENDP_COLUMN_GEO_CITY,
        ENDP_COLUMN_GEO_LATITUDE,
        ENDP_COLUMN_GEO_LONGITUDE,
        ENDP_COLUMN_GEO_AS_NUM,
        ENDP_COLUMN_GEO_AS_ORG,
        ENDP_NUM_GEO_COLUMNS
    } endpoint_column_type_e;
 
    explicit EndpointDataModel(int protoId, QString filter, QObject *parent = nullptr);
 
    int columnCount(const QModelIndex &parent = QModelIndex()) const;
    QVariant headerData(int section, Qt::Orientation orientation = Qt::Horizontal, int role = Qt::DisplayRole) const;
    QVariant data(const QModelIndex &idx, int role = Qt::DisplayRole) const;
 
};
 
class ConversationDataModel : public ATapDataModel
{
    Q_OBJECT
public:
 
    typedef enum {
        CONV_COLUMN_SRC_ADDR,
        CONV_COLUMN_SRC_PORT,
        CONV_COLUMN_DST_ADDR,
        CONV_COLUMN_DST_PORT,
        CONV_COLUMN_PACKETS,
        CONV_COLUMN_BYTES,
        CONV_COLUMN_CONV_ID,
        CONV_COLUMN_PACKETS_TOTAL,
        CONV_COLUMN_BYTES_TOTAL,
        CONV_COLUMN_PKT_AB,
        CONV_COLUMN_BYTES_AB,
        CONV_COLUMN_PKT_BA,
        CONV_COLUMN_BYTES_BA,
        CONV_COLUMN_START,
        CONV_COLUMN_DURATION,
        CONV_COLUMN_BPS_AB,
        CONV_COLUMN_BPS_BA,
        CONV_NUM_COLUMNS,
        CONV_INDEX_COLUMN = CONV_NUM_COLUMNS
    } conversation_column_type_e;
 
    typedef enum {
        CONV_TCP_EXT_COLUMN_A = CONV_INDEX_COLUMN,
        CONV_TCP_EXT_NUM_COLUMNS,
        CONV_TCP_EXT_INDEX_COLUMN = CONV_TCP_EXT_NUM_COLUMNS
    } conversation_tcp_ext_column_type_e;
 
    explicit ConversationDataModel(int protoId, QString filter, QObject *parent = nullptr);
 
    int columnCount(const QModelIndex &parent = QModelIndex()) const;
    QVariant headerData(int section, Qt::Orientation orientation = Qt::Horizontal, int role = Qt::DisplayRole) const;
    QVariant data(const QModelIndex &idx, int role = Qt::DisplayRole) const;
 
    void doDataUpdate();
 
    conv_item_t * itemForRow(int row);
 
    bool showConversationId(int row = 0) const;
 
};
 
#endif // ATAP_DATA_MODEL_H