add ecc SVDP_DH

adafruit · Feb 13, 2020 · 39844bf · 39844bf
1 parent f5e1e7f
commit 39844bf
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/ecc/nRFCrypto_ECC.cpp b/src/ecc/nRFCrypto_ECC.cpp
@@ -23,6 +23,7 @@
  */
 
 #include "nrf_cc310/include/crys_ecpki_kg.h"
+#include "nrf_cc310/include/crys_ecpki_dh.h"
 
 #include "Adafruit_nRFCrypto.h"
 
@@ -55,6 +56,23 @@ bool nRFCrypto_ECC::genKeyPair(nRFCrypto_ECC_PrivateKey& private_key, nRFCrypto_
   return true;
 }
 
+uint32_t nRFCrypto_ECC::SVDP_DH(nRFCrypto_ECC_PrivateKey& private_key, nRFCrypto_ECC_PublicKey& peer_pubkey, uint8_t* shared_secret, uint32_t bufsize)
+{
+  CRYS_ECDH_TempData_t* tempbuf = (CRYS_ECDH_TempData_t *) rtos_malloc(sizeof(CRYS_ECDH_TempData_t));
+  VERIFY(tempbuf);
+
+  nRFCrypto.enable();
+
+  uint32_t err = CRYS_ECDH_SVDP_DH(&peer_pubkey._key, &private_key._key, shared_secret, &bufsize, tempbuf);
+
+  nRFCrypto.disable();
+
+  rtos_free(tempbuf);
+
+  VERIFY_CRYS(err, 0);
+  return bufsize;
+}
+
 nRFCrypto_ECC::nRFCrypto_ECC(void)
 {
 

diff --git a/src/ecc/nRFCrypto_ECC.h b/src/ecc/nRFCrypto_ECC.h
@@ -37,6 +37,7 @@ class nRFCrypto_ECC
   public:
     //------------- Static Members -------------//
     static bool genKeyPair(nRFCrypto_ECC_PrivateKey& private_key, nRFCrypto_ECC_PublicKey& public_key);
+    static uint32_t SVDP_DH(nRFCrypto_ECC_PrivateKey& private_key, nRFCrypto_ECC_PublicKey& peer_pubkey, uint8_t* shared_secret, uint32_t bufsize);
 
   public:
     nRFCrypto_ECC(void);