[PATCH v3] powerpc/smp: Add check for kcalloc() failure in parse_thread_groups()

[Guangshuo Li]

As kcalloc() may fail, check its return value to avoid a NULL pointer
dereference when passing it to of_property_read_u32_array().

Fixes: 790a1662d3a26 ("powerpc/smp: Parse ibm,thread-groups with multiple 
properties")
Cc: sta...@vger.kernel.org
Reviewed-by: Christophe Leroy <christophe.le...@csgroup.eu>
Signed-off-by: Guangshuo Li <lgs201920130...@gmail.com>
---
changelog:
v3:
- Move Signed-off-by above the '---' separator.
- No code changes.
changelog:
v2:
- Return -ENOMEM directly on allocation failure.
---
 arch/powerpc/kernel/smp.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/arch/powerpc/kernel/smp.c b/arch/powerpc/kernel/smp.c
index 5ac7084eebc0..cfccb9389760 100644
--- a/arch/powerpc/kernel/smp.c
+++ b/arch/powerpc/kernel/smp.c
@@ -822,6 +822,8 @@ static int parse_thread_groups(struct device_node *dn,
 
        count = of_property_count_u32_elems(dn, "ibm,thread-groups");
        thread_group_array = kcalloc(count, sizeof(u32), GFP_KERNEL);
+       if (!thread_group_array)
+               return -ENOMEM;
        ret = of_property_read_u32_array(dn, "ibm,thread-groups",
                                         thread_group_array, count);
        if (ret)
-- 
2.43.0