of/numa: add nid check for memory block

If the numa-id which was configured in memory@ devicetree node is greater
than MAX_NUMNODES, we should report a warning. We have done this for cpus
and distance-map dt nodes, this patch help them to be consistent.

Acked-by: Rob Herring <robh@kernel.org>
Signed-off-by: Zhen Lei <thunder.leizhen@huawei.com>
Signed-off-by: Will Deacon <will.deacon@arm.com>

1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/of/of_numa.c b/drivers/of/of_numa.c
index 7b3fbdc0b278..c1bd62ce3144 100644
--- a/drivers/of/of_numa.c
+++ b/drivers/of/of_numa.c
@@ -75,6 +75,11 @@ static int __init of_numa_parse_memory_nodes(void)
                          */
                         continue;
 
+                if (nid >= MAX_NUMNODES) {
+                        pr_warn("NUMA: Node id %u exceeds maximum value\n", nid);
+                        r = -EINVAL;
+                }
+
                 for (i = 0; !r && !of_address_to_resource(np, i, &rsrc); i++)
                         r = numa_add_memblk(nid, rsrc.start, rsrc.end + 1);