mem-cred: Replace existing equal private keys during add_key()

diff --git a/src/libstrongswan/credentials/sets/mem_cred.c b/src/libstrongswan/credentials/sets/mem_cred.c
index b8da3f620903aba790c2a42d8e11a017d21da34a..def017255a105ab9eae60f43da8e9e0a391725e1 100644 (file)
--- a/src/libstrongswan/credentials/sets/mem_cred.c
+++ b/src/libstrongswan/credentials/sets/mem_cred.c
@@ -307,8 +307,25 @@ METHOD(credential_set_t, create_private_enumerator, enumerator_t*,
 METHOD(mem_cred_t, add_key, void,
        private_mem_cred_t *this, private_key_t *key)
 {
+       enumerator_t *enumerator;
+       private_key_t *current;
+
        this->lock->write_lock(this->lock);
+
+       enumerator = this->keys->create_enumerator(this->keys);
+       while (enumerator->enumerate(enumerator, &current))
+       {
+               if (current->equals(current, key))
+               {
+                       this->keys->remove_at(this->keys, enumerator);
+                       current->destroy(current);
+                       break;
+               }
+       }
+       enumerator->destroy(enumerator);
+
        this->keys->insert_first(this->keys, key);
+
        this->lock->unlock(this->lock);
 }