path: root/ui/qt/models/filter_list_model.cpp

/* filter_list_model.cpp
 * Model for all filter types
 *
 * Wireshark - Network traffic analyzer
 * By Gerald Combs <gerald@wireshark.org>
 * Copyright 1998 Gerald Combs
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 */

#include <glib.h>

#include <wsutil/filesystem.h>

#include <ui/qt/utils/qt_ui_utils.h>
#include <ui/qt/models/filter_list_model.h>
#include <ui/qt/models/profile_model.h>

#include <QFile>
#include <QTextStream>
#include <QRegExp>
#include <QDir>

/*
 * Old filter file name.
 */
#define FILTER_FILE_NAME      "filters"

/*
 * Capture filter file name.
 */
#define CFILTER_FILE_NAME     "cfilters"

/*
 * Display filter file name.
 */
#define DFILTER_FILE_NAME     "dfilters"

FilterListModel::FilterListModel(QObject * parent) :
    QAbstractListModel(parent),
    type_(FilterListModel::Display)
{
    reload();
}

FilterListModel::FilterListModel(FilterListModel::FilterListType type, QObject * parent) :
    QAbstractListModel(parent),
    type_(type)
{
    reload();
}

void FilterListModel::reload()
{
    QFile file;

    storage.clear();

    /* Try personal config file first */
    file.setFileName(qstring_strdup(get_persconffile_path(FilterListModel::Capture ? CFILTER_FILE_NAME : DFILTER_FILE_NAME, TRUE)));
    /* Try personal old-style config file next */
    if ( ! file.exists() )
        file.setFileName(qstring_strdup(get_persconffile_path(FILTER_FILE_NAME, TRUE)));
    /* Last but not least, try the global file */
    if ( ! file.exists() )
        file.setFileName(qstring_strdup(get_datafile_path(FilterListModel::Capture ? CFILTER_FILE_NAME : DFILTER_FILE_NAME)));

    /* Still can use the model, just have to start from an empty set */
    if ( ! file.exists() || ! file.open(QIODevice::ReadOnly | QIODevice::Text) )
        return;

    QTextStream in(&file);
    QRegExp rx("\\s*\\\"(.*)\\\"\\s(.*)");
    while (!in.atEnd())
    {
        QString line = in.readLine().trimmed();
        if ( line.startsWith("#") || line.indexOf("\"") <= -1 )
            continue;

        rx.indexIn(line);
        QStringList groups = rx.capturedTexts();
        if ( groups.count() != 3 )
            continue;
        addFilter(groups.at(1), groups.at(2));
    }
}

void FilterListModel::setFilterType(FilterListModel::FilterListType type)
{
    type_ = type;
    reload();
}

FilterListModel::FilterListType FilterListModel::filterType() const
{
    return type_;
}

int FilterListModel::rowCount(const QModelIndex &/* parent */) const
{
    return storage.count();
}

int FilterListModel::columnCount(const QModelIndex &/* parent */) const
{
    return 2;
}

QVariant FilterListModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if ( section >= columnCount() || section < 0 || orientation != Qt::Horizontal )
        return QVariant();

    if ( role == Qt::DisplayRole )
    {
        switch ( section ) {
            case ColumnName:
                return tr("Filter Name");
                break;
            case ColumnExpression:
                return tr("Filter Expression");
                break;
        }
    }

    return QVariant();
}

QVariant FilterListModel::data(const QModelIndex &index, int role) const
{
    if ( ! index.isValid() || index.row() >= rowCount() )
        return QVariant();

    QStringList row = storage.at(index.row()).split("\n");
    if ( role == Qt::DisplayRole )
        return row.at(index.column());

    return QVariant();
}

bool FilterListModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
    if ( ! index.isValid() || index.row() >= rowCount() || role != Qt::EditRole )
        return false;

    QStringList row = storage.at(index.row()).split("\n");
    if ( row.count() <= index.column() )
        return false;

    row[index.column()] = value.toString();
    storage[index.row()] = row.join("\n");

    return true;
}

Qt::ItemFlags FilterListModel::flags(const QModelIndex &index) const
{
    Qt::ItemFlags fl = QAbstractListModel::flags(index);
    if ( ! index.isValid() || index.row() >= rowCount() )
        return fl;

    QStringList row = storage.at(index.row()).split("\n");

    fl |= Qt::ItemIsEditable;

    return fl;
}
QModelIndex FilterListModel::addFilter(QString name, QString expression)
{
    if ( name.length() == 0 || expression.length() == 0 )
        return QModelIndex();

    beginInsertRows(QModelIndex(), rowCount(), rowCount());
    storage << QString("%1\n%2").arg(name).arg(expression);
    endInsertRows();

    return index(rowCount() - 1, 0);
}

QModelIndex FilterListModel::findByName(QString name)
{
    if ( name.length() == 0 )
        return QModelIndex();

    for ( int cnt = 0; cnt < rowCount(); cnt++ )
    {
        if ( storage.at(cnt).startsWith(QString("%1\n").arg(name)) )
            return index(cnt, 0);
    }

    return QModelIndex();
}

QModelIndex FilterListModel::findByExpression(QString expression)
{
    if ( expression.length() == 0 )
        return QModelIndex();

    for ( int cnt = 0; cnt < rowCount(); cnt++ )
    {
        if ( storage.at(cnt).endsWith(QString("\n%1").arg(expression)) )
            return index(cnt, 0);
    }

    return QModelIndex();
}

void FilterListModel::removeFilter(QModelIndex idx)
{
    if ( ! idx.isValid() || idx.row() >= rowCount() )
        return;

    beginRemoveRows(QModelIndex(), idx.row(), idx.row());
    storage.removeAt(idx.row());
    endRemoveRows();
}

void FilterListModel::saveList()
{
    QString filename = FilterListModel::Capture ? CFILTER_FILE_NAME : DFILTER_FILE_NAME;

    filename = QString("%1%2%3").arg(ProfileModel::activeProfilePath()).arg(QDir::separator()).arg(filename);
    QFile file(filename);

    if ( ! file.open(QIODevice::WriteOnly | QIODevice::Text) )
        return;

    QTextStream out(&file);
    for ( int row = 0; row < rowCount(); row++ )
    {
        QString line = QString("\"%1\"").arg(index(row, ColumnName).data().toString().replace("\\", "\\\\").replace("\"", "\\\""));
        line.append(QString(" %1").arg(index(row, ColumnExpression).data().toString()));

#ifdef _WIN32
        line = line.append("\r\n");
#else
        line = line.append("\n");
#endif
        out << line;
    }

    file.close();
}

/*
 * Editor modelines  -  https://www.wireshark.org/tools/modelines.html
 *
 * Local Variables:
 * c-basic-offset: 2
 * tab-width: 8
 * indent-tabs-mode: nil
 * End:
 *
 * vi: set shiftwidth=2 tabstop=8 expandtab:
 * :indentSize=2:tabSize=8:noTabs=true:
 */