feat: add support to differentiate specific hypervisors on s390x

Signed-off-by: Ching Han Chen <[email protected]>

Author: routerhan

source/cpu/cpuid_amd64.go

@@ -36,3 +36,5 @@ func getCpuidAttributes() map[string]string {
 
 	return ret
 }
+
+func getHypervisor() string { return "" }

source/cpu/cpuid_linux_arm.go

@@ -121,3 +121,5 @@ func getCpuidFlags() []string {
 }
 
 func getCpuidAttributes() map[string]string { return nil }
+
+func getHypervisor() string { return "" }

source/cpu/cpuid_linux_arm64.go

@@ -201,3 +201,5 @@ func getCpuidFlags() []string {
 }
 
 func getCpuidAttributes() map[string]string { return nil }
+
+func getHypervisor() string { return "" }

source/cpu/cpuid_linux_ppc64le.go

@@ -155,3 +155,5 @@ func getCpuidFlags() []string {
 }
 
 func getCpuidAttributes() map[string]string { return nil }
+
+func getHypervisor() string { return "" }

source/cpu/cpuid_linux_s390x.go

@@ -24,6 +24,12 @@ unsigned long gethwcap() {
 }
 */
 import "C"
+import (
+	"os"
+	"strings"
+
+	"k8s.io/klog/v2"
+)
 
 /*
 all special features for s390x should be defined here; canonical list:
@@ -96,4 +102,29 @@ func getCpuidFlags() []string {
 	return r
 }
 
-func getCpuidAttributes() map[string]string { return nil }
+func getCpuidAttributes() map[string]string {
+	ret := make(map[string]string)
+	ret["hypervisor"] = getHypervisor()
+
+	return ret
+}
+
+func getHypervisor() string {
+	data, err := os.ReadFile("/proc/sysinfo")
+	if err != nil {
+		klog.ErrorS(err, "failed to read /proc/sysinfo")
+		return ""
+	}
+
+	for _, line := range strings.Split(string(data), "\n") {
+		if strings.Contains(line, "Control Program:") {
+			parts := strings.SplitN(line, ":", 2)
+			if len(parts) == 2 {
+				return strings.TrimSpace(parts[1])
+			}
+		}
+	}
+
+	// if Control Program not found, we determine that the node is PR/SM(LPAR)
+	return "PR/SM"
+}

source/cpu/cpuid_stub.go

@@ -22,3 +22,5 @@ package cpu
 func getCpuidFlags() []string { return nil }
 
 func getCpuidAttributes() map[string]string { return nil }
+
+func getHypervisor() string { return "" }