traffic_tab.h

Go to the documentation of this file.

 
#ifndef TRAFFIC_TAB_H
#define TRAFFIC_TAB_H
 
#include "config.h"
 
#include <ui/qt/models/atap_data_model.h>
#include <ui/qt/filter_action.h>
#include <ui/qt/widgets/traffic_tree.h>
#include <ui/qt/widgets/detachable_tabwidget.h>
#include <ui/qt/widgets/traffic_types_list.h>
 
#include <QTabWidget>
#include <QTreeView>
#include <QFile>
#include <QUrl>
#include <QAbstractItemDelegate>
#include <QSortFilterProxyModel>

typedef ATapDataModel * (*ATapModelCallback)(int protoId, QString filter);

typedef QAbstractItemDelegate * (*ATapCreateDelegate)(QWidget * parent);

class TabData
{
public:
    TabData();

    TabData(const TabData &) = default;

    TabData &operator=(const TabData &) = default;

    TabData(QString name, int proto);

    QString name() const;

    int protoId() const;
 
private:
    QString _name;

    int _protoId;
};
 
Q_DECLARE_METATYPE(TabData)
 

class TrafficTab : public DetachableTabWidget
{
    Q_OBJECT
 
public:
    TrafficTab(QWidget *parent = nullptr);

    virtual ~TrafficTab();

    void setProtocolInfo(QString tableName, TrafficTypesList * trafficList, GList ** recentList, GList ** recentColumnList, ATapModelCallback createModel);

    void setDelegate(ATapCreateDelegate createDelegate);

    void setFilter(QString filter = QString());

    void setNameResolution(bool checked);

    void disableTap();

    QMenu * createCopyMenu(QWidget * parent = nullptr);

    bool hasNameResolution(int tabIdx = -1);
 
#ifdef HAVE_MAXMINDDB
    bool hasGeoIPData(int tabIdx = -1);

    QUrl createGeoIPMap(bool onlyJSON, int tabIdx = -1);
#endif

    QVariant currentItemData(int role = Qt::DisplayRole);

    qlonglong countSelectedItems(int role = Qt::DisplayRole);

    QList<QList<QVariant> > selectedItemsIOGData();

    void useNanosecondTimestamps(bool useNSTime);

    ATapDataModel * dataModelForTabIndex(int tabIdx = -1);
 
public slots:

    void useAbsoluteTime(bool absolute);

    void limitToDisplayFilter(bool limit);

    void setMachineReadable(bool machine);

    void setOpenTabs(QList<int> protocols);
 
signals:
    void filterAction(QString filter, FilterAction::Action action, FilterAction::ActionType type);

    void tabDataChanged(int idx, int selcounter);

    void retapRequired();

    void disablingTaps();

    void tabsChanged(QList<int> protocols);

    void columnsHaveChanged(QList<int> columns);
 
protected slots:

    virtual void detachTab(int idx, QPoint pos) override;

    virtual void attachTab(QWidget * content, QString name) override;
 
private:
    QList<int> _allProtocols;

    QMap<int, int> _tabs;

    ATapModelCallback _createModel;

    ATapCreateDelegate _createDelegate;

    GList ** _recentList;

    GList ** _recentColumnList;

    bool _disableTaps;

    bool _nameResolution;

    bool _absoluteTime;

    bool _limitToDisplayFilter;

    bool _nanoseconds;

    bool _machineReadable;

    QTreeView * createTree(int protoId);

    TrafficDataFilterProxy * modelForTabIndex(int tabIdx = -1);

    TrafficDataFilterProxy * modelForWidget(QWidget * widget);

    ATapDataModel * dataModelForWidget(QWidget * widget);

    void insertProtoTab(int protoId, bool emitSignals = true);

    void removeProtoTab(int protoId, bool emitSignals = true);
 
#ifdef HAVE_MAXMINDDB
    bool writeGeoIPMapFile(QFile * fp, bool json_only, TrafficDataFilterProxy * model);
#endif
 
private slots:
    void modelReset();

    void doCurrentIndexChange(const QModelIndex & cur, const QModelIndex & prev);

    void doSelectionChange(const QItemSelection &selected, const QItemSelection &deselected);
};
 
#endif // TRAFFIC_TAB_H