MdeModulePkg/RegularExpressionDxe: Fix memory assert in FreePool()

Memory buffer that is allocated by malloc() and realloc() will be
shifted by 8 bytes because Oniguruma keeps its memory signature. This 8
bytes shift is not handled while calling free() to release memory. Add
free() function to check Oniguruma signature before release memory
because memory buffer is not touched when using calloc().

Signed-off-by: Nickle Wang <nickle.wang@hpe.com>
Reviewed-by: Liming Gao <gaoliming@byosoft.com.cn>

diff --git a/MdeModulePkg/Universal/RegularExpressionDxe/OnigurumaUefiPort.c b/MdeModulePkg/Universal/RegularExpressionDxe/OnigurumaUefiPort.c
index 9aa7b0a68eefdd203db0b8f3befcc2c98d20ac09..4659c4473e954cf213b0ce921bca31b158e623c2 100644 (file)
--- a/MdeModulePkg/Universal/RegularExpressionDxe/OnigurumaUefiPort.c
+++ b/MdeModulePkg/Universal/RegularExpressionDxe/OnigurumaUefiPort.c
@@ -2,7 +2,7 @@
 \r
   Module to rewrite stdlib references within Oniguruma\r
 \r
-  (C) Copyright 2014-2015 Hewlett Packard Enterprise Development LP<BR>\r
+  (C) Copyright 2014-2021 Hewlett Packard Enterprise Development LP<BR>\r
   Copyright (c) 2020, Intel Corporation. All rights reserved.<BR>\r
 \r
   SPDX-License-Identifier: BSD-2-Clause-Patent\r
@@ -96,3 +96,20 @@ void* memset (void *dest, char ch, unsigned int count)
   return SetMem (dest, count, ch);\r
 }\r
 \r
+void free(void *ptr)\r
+{\r
+  VOID         *EvalOnce;\r
+  ONIGMEM_HEAD *PoolHdr;\r
+\r
+  EvalOnce = ptr;\r
+  if (EvalOnce == NULL) {\r
+    return;\r
+  }\r
+\r
+  PoolHdr = (ONIGMEM_HEAD *)EvalOnce - 1;\r
+  if (PoolHdr->Signature == ONIGMEM_HEAD_SIGNATURE) {\r
+    FreePool (PoolHdr);\r
+  } else {\r
+    FreePool (EvalOnce);\r
+  }\r
+}\r

diff --git a/MdeModulePkg/Universal/RegularExpressionDxe/OnigurumaUefiPort.h b/MdeModulePkg/Universal/RegularExpressionDxe/OnigurumaUefiPort.h
index 20b75c336131f9342818f1a979469b2cfd84df9e..0bdb7be52952379f506c9a2f8f0ed684abcd9ae3 100644 (file)
--- a/MdeModulePkg/Universal/RegularExpressionDxe/OnigurumaUefiPort.h
+++ b/MdeModulePkg/Universal/RegularExpressionDxe/OnigurumaUefiPort.h
@@ -2,7 +2,7 @@
 \r
   Module to rewrite stdlib references within Oniguruma\r
 \r
-  (C) Copyright 2014-2015 Hewlett Packard Enterprise Development LP<BR>\r
+  (C) Copyright 2014-2021 Hewlett Packard Enterprise Development LP<BR>\r
   Copyright (c) 2020, Intel Corporation. All rights reserved.<BR>\r
 \r
   SPDX-License-Identifier: BSD-2-Clause-Patent\r
@@ -46,17 +46,6 @@ typedef INTN        intptr_t;
 #endif\r
 \r
 #define calloc(n,s) AllocateZeroPool((n)*(s))\r
-\r
-#define free(p)             \\r
-  do {                      \\r
-    VOID *EvalOnce;         \\r
-                            \\r
-    EvalOnce = (p);         \\r
-    if (EvalOnce != NULL) { \\r
-      FreePool (EvalOnce);  \\r
-    }                       \\r
-  } while (FALSE)\r
-\r
 #define xmemmove(Dest,Src,Length) CopyMem(Dest,Src,Length)\r
 #define xmemcpy(Dest,Src,Length) CopyMem(Dest,Src,Length)\r
 #define xmemset(Buffer,Value,Length) SetMem(Buffer,Length,Value)\r
@@ -98,6 +87,7 @@ void* malloc(size_t size);
 void* realloc(void *ptr, size_t size);\r
 void* memcpy (void *dest, const void *src, unsigned int count);\r
 void* memset (void *dest, char ch, unsigned int count);\r
+void free(void *ptr);\r
 \r
 #define exit(n) ASSERT(FALSE);\r
 \r