From a1d7733b950c46e1526722c090fba12f8af88882 Mon Sep 17 00:00:00 2001
From: Pol Henarejos <pol.henarejos@cttc.es>
Date: Tue, 2 Jan 2024 02:44:59 +0100
Subject: [PATCH] Fix key exchange.

Signed-off-by: Pol Henarejos <pol.henarejos@cttc.es>
---
 src/hsm/cmd_decrypt_asym.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/hsm/cmd_decrypt_asym.c b/src/hsm/cmd_decrypt_asym.c
index 690e0b2..4f4dc5b 100644
--- a/src/hsm/cmd_decrypt_asym.c
+++ b/src/hsm/cmd_decrypt_asym.c
@@ -111,6 +111,7 @@ int cmd_decrypt_asym() {
         }
         r = -1;
         if (p2 == ALGO_EC_DH) {
+            *(apdu.data - 1) = (uint8_t)apdu.nc;
             r = mbedtls_ecdh_read_public(&ctx, apdu.data - 1, apdu.nc + 1);
         }
         else if (p2 == ALGO_EC_DH_XKEK) {
@@ -119,7 +120,9 @@ int cmd_decrypt_asym() {
             if (pub) {
                 uint16_t t86_len = 0;
                 const uint8_t *t86 = cvc_get_field(pub, pub_len, &t86_len, 0x86);
+                uint8_t *t86w = (uint8_t *)t86;
                 if (t86) {
+                    *(t86w - 1) = (uint8_t)t86_len;
                     r = mbedtls_ecdh_read_public(&ctx, t86 - 1, t86_len + 1);
                 }
             }