From 2b6c4b66e5df5c82e5197191d32ac7b81e29abcb Mon Sep 17 00:00:00 2001
From: Maxim Moskalets <maximmosk4@gmail.com>
Date: Wed, 15 May 2024 23:13:30 +0300
Subject: [PATCH] dm: use newly added linux/compat alloc functions

Signed-off-by: Maxim Moskalets <maximmosk4@gmail.com>
---
 include/dm/devres.h | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/include/dm/devres.h b/include/dm/devres.h
index 697534aa5b..27761deb6d 100644
--- a/include/dm/devres.h
+++ b/include/dm/devres.h
@@ -266,17 +266,13 @@ static inline void *devm_kzalloc(struct udevice *dev, size_t size, gfp_t gfp)
 static inline void *devm_kmalloc_array(struct udevice *dev,
 				       size_t n, size_t size, gfp_t flags)
 {
-	/* TODO: add kmalloc_array() to linux/compat.h */
-	if (size != 0 && n > SIZE_MAX / size)
-		return NULL;
-	return kmalloc(n * size, flags);
+	return kmalloc_array(n, size, flags);
 }
 
 static inline void *devm_kcalloc(struct udevice *dev,
 				 size_t n, size_t size, gfp_t flags)
 {
-	/* TODO: add kcalloc() to linux/compat.h */
-	return kmalloc(n * size, flags | __GFP_ZERO);
+	return kcalloc(n, size, flags);
 }
 
 static inline void devm_kfree(struct udevice *dev, void *ptr)
-- 
2.39.5