From 019c2a7817c702f5d7826bc14badf2a5c7a36c4d Mon Sep 17 00:00:00 2001
From: Chien Wong <m@xv97.com>
Date: Tue, 23 Jan 2024 21:38:06 +0800
Subject: [PATCH] Handle sizeof(size_t) > sizeof(uint64_t)

Signed-off-by: Chien Wong <m@xv97.com>
---
 library/gcm.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/library/gcm.c b/library/gcm.c
index 337145b718..033cb59017 100644
--- a/library/gcm.c
+++ b/library/gcm.c
@@ -358,7 +358,12 @@ int mbedtls_gcm_update_ad(mbedtls_gcm_context *ctx,
 
     /* AD is limited to 2^64 bits, ie 2^61 bytes
      * Also check for possible overflow */
-    new_add_len = ctx->add_len + add_len;
+#if SIZE_MAX > 0xFFFFFFFFFFFFFFFFULL
+    if (add_len > 0xFFFFFFFFFFFFFFFFULL) {
+        return MBEDTLS_ERR_GCM_BAD_INPUT;
+    }
+#endif
+    new_add_len = ctx->add_len + (uint64_t) add_len;
     if (new_add_len < ctx->add_len || new_add_len >> 61 != 0) {
         return MBEDTLS_ERR_GCM_BAD_INPUT;
     }