keymat: Log nonce and DH context ids
[strongswan.git] / src / charon-tkm / src / tkm / tkm_keymat.c
index 186f67b..e3fcfa0 100644 (file)
@@ -17,6 +17,8 @@
 #include <daemon.h>
 #include <sa/ikev2/keymat_v2.h>
 
+#include "tkm.h"
+#include "tkm_diffie_hellman.h"
 #include "tkm_keymat.h"
 
 typedef struct private_tkm_keymat_t private_tkm_keymat_t;
@@ -36,6 +38,11 @@ struct private_tkm_keymat_t {
         */
        keymat_v2_t *proxy;
 
+       /**
+        * IKE_SA Role, initiator or responder
+        */
+       bool initiator;
+
 };
 
 METHOD(keymat_t, get_version, ike_version_t,
@@ -61,9 +68,25 @@ METHOD(tkm_keymat_t, derive_ike_keys, bool,
        chunk_t nonce_i, chunk_t nonce_r, ike_sa_id_t *id,
        pseudo_random_function_t rekey_function, chunk_t rekey_skd)
 {
-       DBG1(DBG_IKE, "deriving IKE keys");
-       return this->proxy->derive_ike_keys(this->proxy, proposal, dh, nonce_i,
-                       nonce_r, id, rekey_function, rekey_skd);
+       tkm_diffie_hellman_t * const tkm_dh = (tkm_diffie_hellman_t *)dh;
+       chunk_t * const nonce = this->initiator ? &nonce_i : &nonce_r;
+
+       const uint64_t nc_id = tkm->chunk_map->get_id(tkm->chunk_map, nonce);
+       if (!nc_id)
+       {
+               DBG1(DBG_IKE, "unable to acquire context id for nonce");
+               return FALSE;
+       }
+
+       DBG1(DBG_IKE, "deriving IKE keys (nc: %llu, dh: %llu)", nc_id,
+                       tkm_dh->get_id(tkm_dh));
+       if (this->proxy->derive_ike_keys(this->proxy, proposal, dh, nonce_i,
+                               nonce_r, id, rekey_function, rekey_skd))
+       {
+               tkm->chunk_map->remove(tkm->chunk_map, nonce);
+               return TRUE;
+       }
+       return FALSE;
 }
 
 METHOD(tkm_keymat_t, derive_child_keys, bool,
@@ -136,6 +159,7 @@ tkm_keymat_t *tkm_keymat_create(bool initiator)
                        .get_auth_octets = _get_auth_octets,
                        .get_psk_sig = _get_psk_sig,
                },
+               .initiator = initiator,
                .proxy = keymat_v2_create(initiator),
        );