mm/hugetlb: Check nid when alloc hugepage

ascend inclusion
category: bugfix
bugzilla: NA
CVE: NA

------------

Check nid when alloc hugepage.
Signed-off-by: NZhou Guanghui <zhouguanghui1@huawei.com>
Signed-off-by: NLijun Fang <fanglijun3@huawei.com>
Reviewed-by: NKefeng Wang <wangkefeng.wang@huawei.com>
Signed-off-by: NYang Yingliang <yangyingliang@huawei.com>

mm/hugetlb.c

@@ -5136,6 +5136,11 @@ EXPORT_SYMBOL_GPL(hugetlb_get_hstate);
  */
 struct page *hugetlb_alloc_hugepage(int nid)
 {
+	VM_WARN_ON(nid < 0 || nid >= MAX_NUMNODES);
+
+	if (nid == NUMA_NO_NODE)
+		nid = numa_mem_id();
+
 	return alloc_huge_page_node(&default_hstate, nid);
 }
 EXPORT_SYMBOL_GPL(hugetlb_alloc_hugepage);