TrustMemoryStorage: Add security policies

2 files changed, 24 insertions, 0 deletions

diff --git a/src/client/QXmppTrustMemoryStorage.cpp b/src/client/QXmppTrustMemoryStorage.cpp
index 1d9d1909..c6d80179 100644
--- a/src/client/QXmppTrustMemoryStorage.cpp
+++ b/src/client/QXmppTrustMemoryStorage.cpp
@@ -72,6 +72,9 @@ struct UnprocessedKey
 class QXmppTrustMemoryStoragePrivate
 {
 public:
+    // encryption protocols mapped to security policies
+    QMap<QString, QXmppTrustStorage::SecurityPolicy> securityPolicies;
+
     // encryption protocols mapped to keys of this client instance
     QMap<QString, QString> ownKeys;
 
@@ -94,6 +97,24 @@ QXmppTrustMemoryStorage::QXmppTrustMemoryStorage()
 QXmppTrustMemoryStorage::~QXmppTrustMemoryStorage() = default;
 
 /// \cond
+QFuture<void> QXmppTrustMemoryStorage::setSecurityPolicies(const QString &encryption, const QXmppTrustStorage::SecurityPolicy securityPolicy)
+{
+    if (encryption.isEmpty()) {
+        d->securityPolicies.clear();
+    } else if (securityPolicy == QXmppTrustStorage::NoSecurityPolicy) {
+        d->securityPolicies.remove(encryption);
+    } else {
+        d->securityPolicies.insert(encryption, securityPolicy);
+    }
+
+    return makeReadyFuture();
+}
+
+QFuture<QXmppTrustStorage::SecurityPolicy> QXmppTrustMemoryStorage::securityPolicy(const QString &encryption)
+{
+    return makeReadyFuture(std::move(d->securityPolicies.value(encryption)));
+}
+
 QFuture<void> QXmppTrustMemoryStorage::addOwnKey(const QString &encryption, const QString &keyId)
 {
     d->ownKeys.insert(encryption, keyId);
diff --git a/src/client/QXmppTrustMemoryStorage.h b/src/client/QXmppTrustMemoryStorage.h
index 6c1eb668..eecbaa33 100644
--- a/src/client/QXmppTrustMemoryStorage.h
+++ b/src/client/QXmppTrustMemoryStorage.h
@@ -38,6 +38,9 @@ public:
     ~QXmppTrustMemoryStorage();
 
     /// \cond
+    QFuture<void> setSecurityPolicies(const QString &encryption = {}, SecurityPolicy securityPolicy = QXmppTrustStorage::NoSecurityPolicy) override;
+    QFuture<SecurityPolicy> securityPolicy(const QString &encryption) override;
+
     QFuture<void> addOwnKey(const QString &encryption, const QString &keyId) override;
     QFuture<void> removeOwnKey(const QString &encryption) override;
     QFuture<QString> ownKey(const QString &encryption) const override;