[Build] Prevent module loading if detected CPU count > CORE_COUNT

Author: cyring

aarch64/corefreqk.c

@@ -6138,6 +6138,15 @@ static int CoreFreqK_Scale_And_Compute_Level_Up(INIT_ARG *pArg)
 					COREFREQ_REV	);
 
 	PUBLIC(RO(Proc))->CPU.Count = pArg->SMT_Count;
+
+    if (PUBLIC(RO(Proc))->CPU.Count > CORE_COUNT) {
+	pr_warn("CoreFreq: Detected %u CPUs, but built with CORE_COUNT=%u\n",
+		PUBLIC(RO(Proc))->CPU.Count, CORE_COUNT);
+	pr_warn("CoreFreq: Run 'make help' for instructions "	\
+		"on setting CORE_COUNT.\n");
+
+	return -ENOMEM;
+    } else {
 	/* PreCompute SysGate memory allocation. */
 	PUBLIC(RO(Proc))->Gate.ReqMem.Size = sizeof(SYSGATE_RO);
 
@@ -6159,7 +6168,9 @@ static int CoreFreqK_Scale_And_Compute_Level_Up(INIT_ARG *pArg)
 				PUBLIC(RO(Proc))->Features.Info.Vendor.ID;
 	/* Initialize with any hypervisor found so far.			*/
 	PUBLIC(RO(Proc))->HypervisorID = pArg->HypervisorID;
+
 	return 0;
+    }
 }
 
 static void CoreFreqK_Alloc_Public_Cache_Level_Down(void)

ppc64le/corefreqk.c

@@ -4491,6 +4491,15 @@ static int CoreFreqK_Scale_And_Compute_Level_Up(INIT_ARG *pArg)
 					COREFREQ_REV	);
 
 	PUBLIC(RO(Proc))->CPU.Count = pArg->SMT_Count;
+
+    if (PUBLIC(RO(Proc))->CPU.Count > CORE_COUNT) {
+	pr_warn("CoreFreq: Detected %u CPUs, but built with CORE_COUNT=%u\n",
+		PUBLIC(RO(Proc))->CPU.Count, CORE_COUNT);
+	pr_warn("CoreFreq: Run 'make help' for instructions "	\
+		"on setting CORE_COUNT.\n");
+
+	return -ENOMEM;
+    } else {
 	/* PreCompute SysGate memory allocation. */
 	PUBLIC(RO(Proc))->Gate.ReqMem.Size = sizeof(SYSGATE_RO);
 
@@ -4509,7 +4518,9 @@ static int CoreFreqK_Scale_And_Compute_Level_Up(INIT_ARG *pArg)
 
 	/* Initialize with any hypervisor found so far.			*/
 	PUBLIC(RO(Proc))->HypervisorID = pArg->HypervisorID;
+
 	return 0;
+    }
 }
 
 static void CoreFreqK_Alloc_Public_Cache_Level_Down(void)

riscv64/corefreqk.c

@@ -4482,6 +4482,15 @@ static int CoreFreqK_Scale_And_Compute_Level_Up(INIT_ARG *pArg)
 					COREFREQ_REV	);
 
 	PUBLIC(RO(Proc))->CPU.Count = pArg->SMT_Count;
+
+    if (PUBLIC(RO(Proc))->CPU.Count > CORE_COUNT) {
+	pr_warn("CoreFreq: Detected %u CPUs, but built with CORE_COUNT=%u\n",
+		PUBLIC(RO(Proc))->CPU.Count, CORE_COUNT);
+	pr_warn("CoreFreq: Run 'make help' for instructions "	\
+		"on setting CORE_COUNT.\n");
+
+	return -ENOMEM;
+    } else {
 	/* PreCompute SysGate memory allocation. */
 	PUBLIC(RO(Proc))->Gate.ReqMem.Size = sizeof(SYSGATE_RO);
 
@@ -4500,7 +4509,9 @@ static int CoreFreqK_Scale_And_Compute_Level_Up(INIT_ARG *pArg)
 
 	/* Initialize with any hypervisor found so far.			*/
 	PUBLIC(RO(Proc))->HypervisorID = pArg->HypervisorID;
+
 	return 0;
+    }
 }
 
 static void CoreFreqK_Alloc_Public_Cache_Level_Down(void)

x86_64/corefreqk.c

@@ -24639,6 +24639,15 @@ static int CoreFreqK_Scale_And_Compute_Level_Up(INIT_ARG *pArg)
 					COREFREQ_REV	);
 
 	PUBLIC(RO(Proc))->CPU.Count = pArg->SMT_Count;
+
+    if (PUBLIC(RO(Proc))->CPU.Count > CORE_COUNT) {
+	pr_warn("CoreFreq: Detected %u CPUs, but built with CORE_COUNT=%u\n",
+		PUBLIC(RO(Proc))->CPU.Count, CORE_COUNT);
+	pr_warn("CoreFreq: Run 'make help' for instructions "	\
+		"on setting CORE_COUNT.\n");
+
+	return -ENOMEM;
+    } else {
 	/* PreCompute SysGate memory allocation. */
 	PUBLIC(RO(Proc))->Gate.ReqMem.Size = sizeof(SYSGATE_RO);
 
@@ -24659,6 +24668,7 @@ static int CoreFreqK_Scale_And_Compute_Level_Up(INIT_ARG *pArg)
 	Arch[GenuineArch].Architecture[0] = \
 				PUBLIC(RO(Proc))->Features.Info.Vendor.ID;
 	return 0;
+    }
 }
 
 static void CoreFreqK_Alloc_Public_Cache_Level_Down(void)