[crypto] update mbedtls psa crypto context structs

Author: silabs-HarshaK

src/core/config/platform.h

@@ -177,6 +177,15 @@
 #define OPENTHREAD_CONFIG_PLATFORM_POWER_CALIBRATION_ENABLE 0
 #endif
 
+/**
+ * @def OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
+ *
+ * Define to 1 to enable platform allocation of crypto contexts.
+ */
+#ifndef OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
+#define OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS 0
+#endif
+
 #if OPENTHREAD_CONFIG_PLATFORM_RADIO_PROPRIETARY_SUPPORT
 #if (!defined(OPENTHREAD_CONFIG_PLATFORM_RADIO_PROPRIETARY_CHANNEL_PAGE) || \
      !defined(OPENTHREAD_CONFIG_PLATFORM_RADIO_PROPRIETARY_CHANNEL_MIN) ||  \

src/core/crypto/aes_ecb.cpp

@@ -40,8 +40,13 @@ namespace Crypto {
 
 AesEcb::AesEcb(void)
 {
+#if OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
+    mContext.mContext     = nullptr;
+    mContext.mContextSize = 0;
+#else
     mContext.mContext     = mContextStorage;
     mContext.mContextSize = sizeof(mContextStorage);
+#endif
     SuccessOrAssert(otPlatCryptoAesInit(&mContext));
 }
 

src/core/crypto/aes_ecb.hpp

@@ -86,7 +86,9 @@ class AesEcb
 
 private:
     otCryptoContext mContext;
+#if !OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
     OT_DEFINE_ALIGNED_VAR(mContextStorage, kAesContextSize, uint64_t);
+#endif
 };
 
 /**

src/core/crypto/hkdf_sha256.cpp

@@ -44,8 +44,13 @@ namespace Crypto {
 
 HkdfSha256::HkdfSha256(void)
 {
+#if OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
+    mContext.mContext     = nullptr;
+    mContext.mContextSize = 0;
+#else
     mContext.mContext     = mContextStorage;
     mContext.mContextSize = sizeof(mContextStorage);
+#endif
     SuccessOrAssert(otPlatCryptoHkdfInit(&mContext));
 }
 

src/core/crypto/hkdf_sha256.hpp

@@ -94,7 +94,9 @@ class HkdfSha256
 
 private:
     otCryptoContext mContext;
+#if !OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
     OT_DEFINE_ALIGNED_VAR(mContextStorage, kHkdfContextSize, uint64_t);
+#endif
 };
 
 /**

src/core/crypto/hmac_sha256.cpp

@@ -41,8 +41,13 @@ namespace Crypto {
 
 HmacSha256::HmacSha256(void)
 {
+#if OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
+    mContext.mContext     = nullptr;
+    mContext.mContextSize = 0;
+#else
     mContext.mContext     = mContextStorage;
     mContext.mContextSize = sizeof(mContextStorage);
+#endif
 
     SuccessOrAssert(otPlatCryptoHmacSha256Init(&mContext));
 }

src/core/crypto/hmac_sha256.hpp

@@ -124,7 +124,9 @@ class HmacSha256
 
 private:
     otCryptoContext mContext;
+#if !OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
     OT_DEFINE_ALIGNED_VAR(mContextStorage, kHmacSha256ContextSize, uint64_t);
+#endif
 };
 
 /**

src/core/crypto/sha256.cpp

@@ -42,8 +42,14 @@ namespace Crypto {
 
 Sha256::Sha256(void)
 {
+#if OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
+    mContext.mContext     = nullptr;
+    mContext.mContextSize = 0;
+#else
     mContext.mContext     = mContextStorage;
     mContext.mContextSize = sizeof(mContextStorage);
+#endif
+
     SuccessOrAssert(otPlatCryptoSha256Init(&mContext));
 }
 

src/core/crypto/sha256.hpp

@@ -136,7 +136,9 @@ class Sha256
 
 private:
     otCryptoContext mContext;
+#if !OPENTHREAD_CONFIG_PLATFORM_ALLOCS_CRYPTO_CONTEXTS
     OT_DEFINE_ALIGNED_VAR(mContextStorage, kSha256ContextSize, uint64_t);
+#endif
 };
 
 /**