Merge pull request OpenMathLib#4523 from martin-frbg/gemmtstack

Fix a potential buffer overflow in GEMMT
martin-frbg · Feb 25, 2024 · 07e62a4 · 07e62a4
2 parents f860e82 + baf8856
commit 07e62a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/interface/gemmt.c b/interface/gemmt.c
@@ -522,7 +522,7 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo,
 
 			IDEBUG_START;
 
-			buffer_size = j + k + 128 / sizeof(FLOAT);
+			buffer_size = 2 * (j + k) + 128 / sizeof(FLOAT);
 #ifdef WINDOWS_ABI
 			buffer_size += 160 / sizeof(FLOAT);
 #endif
@@ -611,7 +611,7 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo,
 #endif
 			IDEBUG_START;
 
-			buffer_size = j + k + 128 / sizeof(FLOAT);
+			buffer_size = 2 * (j + k) + 128 / sizeof(FLOAT);
 #ifdef WINDOWS_ABI
 			buffer_size += 160 / sizeof(FLOAT);
 #endif