arm64: armv8_deprecated: Checking return value for memory allocation

There are no return value checking when using kzalloc() and kcalloc() for
memory allocation. so add it.

Signed-off-by: Yunfeng Ye <yeyunfeng@huawei.com>
Signed-off-by: Will Deacon <will@kernel.org>

1 files changed, 5 insertions, 0 deletions

diff --git a/arch/arm64/kernel/armv8_deprecated.c b/arch/arm64/kernel/armv8_deprecated.c
index 2ec09debc2bb..ca158be21f83 100644
--- a/arch/arm64/kernel/armv8_deprecated.c
+++ b/arch/arm64/kernel/armv8_deprecated.c
@@ -174,6 +174,9 @@ static void __init register_insn_emulation(struct insn_emulation_ops *ops)
         struct insn_emulation *insn;
 
         insn = kzalloc(sizeof(*insn), GFP_KERNEL);
+        if (!insn)
+                return;
+
         insn->ops = ops;
         insn->min = INSN_UNDEF;
 
@@ -233,6 +236,8 @@ static void __init register_insn_emulation_sysctl(void)
 
         insns_sysctl = kcalloc(nr_insn_emulated + 1, sizeof(*sysctl),
                                GFP_KERNEL);
+        if (!insns_sysctl)
+                return;
 
         raw_spin_lock_irqsave(&insn_emulation_lock, flags);
         list_for_each_entry(insn, &insn_emulation, node) {