From 49762fa21fe4848d78439e812e1b8e8ba6998463 Mon Sep 17 00:00:00 2001
From: nirekh01 <nir.ekhauz@arm.com>
Date: Mon, 25 Dec 2017 06:46:48 +0000
Subject: [PATCH] Add 'MBEDTLS_DHM_ALT' #DEFINE to library/config.h

Add 'MBEDTLS_DHM_ALT' #DEFINE to library/config.h to support alternate DHM
---
 include/mbedtls/config.h   | 2 ++
 library/dhm.c              | 3 ++-
 library/version_features.c | 3 +++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/include/mbedtls/config.h b/include/mbedtls/config.h
index 69e997f85e..f8594b841a 100644
--- a/include/mbedtls/config.h
+++ b/include/mbedtls/config.h
@@ -275,6 +275,8 @@
 //#define MBEDTLS_SHA1_ALT
 //#define MBEDTLS_SHA256_ALT
 //#define MBEDTLS_SHA512_ALT
+//#define MBEDTLS_DHM_ALT
+
 /*
  * When replacing the elliptic curve module, pleace consider, that it is
  * implemented with two .c files:
diff --git a/library/dhm.c b/library/dhm.c
index 6f8f021e5c..882b306744 100644
--- a/library/dhm.c
+++ b/library/dhm.c
@@ -580,7 +580,8 @@ int mbedtls_dhm_parse_dhmfile( mbedtls_dhm_context *dhm, const char *path )
 }
 #endif /* MBEDTLS_FS_IO */
 #endif /* MBEDTLS_ASN1_PARSE_C */
-#endif/*MBEDTLS_DHM_ALT*/
+#endif /* MBEDTLS_DHM_ALT */
+
 #if defined(MBEDTLS_SELF_TEST)
 
 static const char mbedtls_test_dhm_params[] =
diff --git a/library/version_features.c b/library/version_features.c
index 5cbe8aca37..000246edcd 100644
--- a/library/version_features.c
+++ b/library/version_features.c
@@ -123,6 +123,9 @@ static const char *features[] = {
 #if defined(MBEDTLS_SHA512_ALT)
     "MBEDTLS_SHA512_ALT",
 #endif /* MBEDTLS_SHA512_ALT */
+#if defined(MBEDTLS_DHM_ALT)
+    "MBEDTLS_DHM_ALT",
+#endif /* MBEDTLS_DHM_ALT */
 #if defined(MBEDTLS_ECP_ALT)
     "MBEDTLS_ECP_ALT",
 #endif /* MBEDTLS_ECP_ALT */