From 06facb4b5200aef88f5143cbd0da60ac51d2ac7d Mon Sep 17 00:00:00 2001
From: zfx <zfx@coder.com>
Date: Mon, 7 Jun 2021 17:47:58 +0800
Subject: [PATCH] check the max size of memory

---
 osal/src/osal_mem.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/osal/src/osal_mem.c b/osal/src/osal_mem.c
index 23886ce..22024ab 100644
--- a/osal/src/osal_mem.c
+++ b/osal/src/osal_mem.c
@@ -47,6 +47,10 @@ static void *osal_mem_alloc(size_t size, uint32_t *type)
 {
 	char *base = NULL;
 	const uint32_t mng_size = sizeof(struct mem_block);
+	if (size > (SIZE_MAX - mng_size)) {
+		HDF_LOGE("%s invalid param %d", __func__, size);
+		return NULL;
+	}
 
 	if (size > (KMALLOC_SIZE - mng_size)) {
 		base = (char *)vmalloc(size + mng_size);
-- 
Gitee