Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) client-ip=2620:137:e000::1:20;
From:   Li kunyu <kunyu@nfschina.com>
To:     shuah@kernel.org, trenn@suse.com
Cc:     ray.huang@amd.com, linux-pm@vger.kernel.org,
        linux-kernel@vger.kernel.org, Li kunyu <kunyu@nfschina.com>
Subject: [v2 PATCH] power: cpupower: utils: Add malloc return value check
Date:   Wed, 19 Oct 2022 09:46:14 +0800
Message-Id: <20221019014614.3120-1-kunyu@nfschina.com>
Precedence: bulk

Add the check and judgment statement of malloc return value.

Signed-off-by: Li kunyu <kunyu@nfschina.com>
---
 v2: Remove unlikely() check

 tools/power/cpupower/utils/helpers/misc.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tools/power/cpupower/utils/helpers/misc.c b/tools/power/cpupower/utils/helpers/misc.c
index 36dbd0562240..083d5a6bb74e 100644
--- a/tools/power/cpupower/utils/helpers/misc.c
+++ b/tools/power/cpupower/utils/helpers/misc.c
@@ -139,6 +139,8 @@ void print_online_cpus(void)
 
 	str_len = online_cpus->size * 5;
 	online_cpus_str = malloc(sizeof(char) * str_len);
+	if (!online_cpus_str)
+		return;
 
 	if (!bitmask_isallclear(online_cpus)) {
 		bitmask_displaylist(online_cpus_str, str_len, online_cpus);
@@ -157,6 +159,8 @@ void print_offline_cpus(void)
 
 	str_len = offline_cpus->size * 5;
 	offline_cpus_str = malloc(sizeof(char) * str_len);
+	if (!offline_cpus_str)
+		return;
 
 	if (!bitmask_isallclear(offline_cpus)) {
 		bitmask_displaylist(offline_cpus_str, str_len, offline_cpus);
-- 
2.18.2