1 files changed, 155 insertions, 0 deletions
diff --git a/ui/qt/models/credentials_model.cpp b/ui/qt/models/credentials_model.cpp
new file mode 100644
index 00000000..8bc649f1
--- /dev/null
+++ b/ui/qt/models/credentials_model.cpp
@@ -0,0 +1,155 @@
+/*
+ * credentials_model.h
+ *
+ * Copyright 2019 - Dario Lombardo <lomato@gmail.com>
+ *
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#include "credentials_model.h"
+
+#include <file.h>
+#include <ui/qt/utils/qt_ui_utils.h>
+
+CredentialsModel::CredentialsModel(QObject *parent)
+    :QAbstractListModel(parent)
+{
+}
+
+CredentialsModel::~CredentialsModel()
+{
+    clear();
+}
+
+int CredentialsModel::rowCount(const QModelIndex &) const
+{
+    return static_cast<int>(credentials_.count());
+}
+
+int CredentialsModel::columnCount(const QModelIndex &) const
+{
+    return COL_INFO + 1;
+}
+
+QVariant CredentialsModel::data(const QModelIndex &index, int role) const
+{
+    if (!index.isValid())
+        return QVariant();
+
+    tap_credential_t * auth = credentials_.at(index.row());
+    if (!auth)
+        return QVariant();
+
+
+    if (role == Qt::DisplayRole) {
+        switch (index.column()) {
+            case COL_NUM:
+                return QVariant::fromValue(auth->num);
+            case COL_PROTO:
+                return QString(auth->proto);
+            case COL_USERNAME:
+                return QString(auth->username);
+            case COL_INFO:
+                return QString(auth->info);
+            default:
+                return QVariant();
+        }
+    }
+
+    if (role == Qt::UserRole) {
+        switch (index.column()) {
+            case COL_NUM:
+                if (auth->num > 0)
+                    return QVariant::fromValue(auth->num);
+                break;
+            case COL_USERNAME:
+                if (auth->username_num > 0)
+                    return QVariant::fromValue(auth->username_num);
+                break;
+            default:
+                return QVariant();
+        }
+    }
+
+    if (role == CredentialsModel::ColumnHFID)
+        return QVariant::fromValue(auth->password_hf_id);
+
+    if (role == Qt::ToolTipRole) {
+        const QString select_msg(tr("Click to select the packet"));
+        switch (index.column()) {
+            case COL_NUM:
+                if (auth->num > 0)
+                    return select_msg;
+                break;
+            case  COL_USERNAME:
+                if (auth->username_num > 0) {
+                    if (auth->username_num != auth->num)
+                        return QString(tr("Click to select the packet with username"));
+                    else
+                        return select_msg;
+                } else {
+                    return QString(tr("Username not available"));
+                }
+                break;
+            default:
+                return QVariant();
+        }
+    }
+
+    return QVariant();
+}
+
+void CredentialsModel::addRecord(const tap_credential_t* auth)
+{
+    emit beginInsertRows(QModelIndex(), rowCount(), rowCount() + 1);
+
+    tap_credential_t* clone = new tap_credential_t;
+    clone->num = auth->num;
+    clone->username_num = auth->username_num;
+    clone->password_hf_id = auth->password_hf_id;
+    clone->username = qstring_strdup(auth->username);
+    clone->proto = auth->proto;
+    clone->info = qstring_strdup(auth->info);
+    credentials_.append(clone);
+
+    emit endInsertRows();
+}
+
+void CredentialsModel::clear()
+{
+    if (!credentials_.isEmpty()) {
+        emit beginRemoveRows(QModelIndex(), 0, rowCount() - 1);
+        for (QList<tap_credential_t*>::iterator itr = credentials_.begin(); itr != credentials_.end(); ++itr) {
+            g_free((*itr)->username);
+            g_free((*itr)->info);
+            delete *itr;
+        }
+        credentials_.clear();
+        emit endRemoveRows();
+    }
+}
+
+QVariant CredentialsModel::headerData(int section, Qt::Orientation orientation, int role) const
+{
+    if (role != Qt::DisplayRole)
+        return QVariant();
+
+    if (orientation == Qt::Horizontal) {
+        switch (section) {
+            case COL_NUM:
+                return QString(tr("Packet No."));
+            case COL_PROTO:
+                return QString(tr("Protocol"));
+            case COL_USERNAME:
+                return QString(tr("Username"));
+            case COL_INFO:
+                return QString(tr("Additional Info"));
+        }
+    }
+
+    return QVariant();
+}