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;
 
    virtual void setResolveNames(bool resolve) = 0;
 
    bool allowsNameResolution() const;
 
    virtual void useAbsoluteTime(bool absolute) = 0;
 
    virtual void useNanosecondTimestamps(bool nanoseconds) = 0;
 
    void setMachineReadable(bool machineReadable);
 
    void limitToDisplayFilter(bool limit);
 
    bool portsAreHidden() const;
 
    bool isFilterApplied() 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; 
    // XXX - There are other possible time precisions besides
    // microseconds and nanoseconds; e.g., Netmon 2.3 uses
    // 100 ns, and pcapng can have one of many values.
    bool _nanoseconds; 
    bool _resolveNames; 
    bool _machineReadable; 
    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 override;
 
    QVariant headerData(int section, Qt::Orientation orientation = Qt::Horizontal,
                        int role = Qt::DisplayRole) const override;
 
    QVariant data(const QModelIndex &idx, int role = Qt::DisplayRole) const override;
 
    void setResolveNames(bool resolve) override;
 
    void useAbsoluteTime(bool absolute) override;
 
    void useNanosecondTimestamps(bool nanoseconds) override;
};
 
 
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 override;
 
    QVariant headerData(int section, Qt::Orientation orientation = Qt::Horizontal,
                        int role = Qt::DisplayRole) const override;
 
    QVariant data(const QModelIndex &idx, int role = Qt::DisplayRole) const override;
 
    void doDataUpdate();
 
    conv_item_t *itemForRow(int row);
 
    bool showConversationId(int row = 0) const;
 
    void setResolveNames(bool resolve) override;
 
    void useAbsoluteTime(bool absolute) override;
 
    void useNanosecondTimestamps(bool nanoseconds) override;
};
 
#endif // ATAP_DATA_MODEL_H