Feature: host support centos 6 (#406)

pkg/util/fileutils2/fileutils.go

@@ -97,6 +97,14 @@ func Zerofiles(sPath string) error {
 	return nil
 }
 
+func FileSetContents(filename string, content string) error {
+	return FilePutContents(filename, content, false)
+}
+
+func FileAppendContents(filename string, content string) error {
+	return FilePutContents(filename, content, true)
+}
+
 func FilePutContents(filename string, content string, modAppend bool) error {
 	var mode = os.O_WRONLY | os.O_CREATE
 	if modAppend {

pkg/util/sysutils/hostname.go

@@ -0,0 +1,30 @@
+package sysutils
+
+import (
+	"fmt"
+
+	"yunion.io/x/onecloud/pkg/util/fileutils2"
+	"yunion.io/x/onecloud/pkg/util/procutils"
+)
+
+func setHostname7(name string) error {
+	_, err := procutils.NewCommand("hostnamectl", "set-hostname", name).Run()
+	return err
+}
+
+func setHostname6(name string) error {
+	_, err := procutils.NewCommand("hostname", name).Run()
+	if err != nil {
+		return err
+	}
+	cont := fmt.Sprintf("NETWORKING=yes\nHOSTNAME=%s\n", name)
+	return fileutils2.FileSetContents("/etc/sysconfig/network", cont)
+}
+
+func SetHostname(name string) error {
+	err := setHostname7(name)
+	if err != nil {
+		err = setHostname6(name)
+	}
+	return err
+}

pkg/util/sysutils/kvm.go

@@ -0,0 +1,179 @@
+package sysutils
+
+import (
+	"os"
+	"path"
+	"regexp"
+	"strings"
+
+	"yunion.io/x/log"
+	"yunion.io/x/pkg/utils"
+
+	"yunion.io/x/onecloud/pkg/util/fileutils2"
+	"yunion.io/x/onecloud/pkg/util/procutils"
+)
+
+const (
+	KVM_MODULE_INTEL     = "kvm-intel"
+	KVM_MODULE_AMD       = "kvm-amd"
+	KVM_MODULE_UNSUPPORT = "unsupport"
+
+	HOST_NEST_UNSUPPORT = "0"
+	HOST_NEST_SUPPORT   = "1"
+	HOST_NEST_ENABLE    = "3"
+)
+
+var (
+	kvmModuleSupport string
+	nestStatus       string
+)
+
+func GetKVMModuleSupport() string {
+	if len(kvmModuleSupport) == 0 {
+		kvmModuleSupport = detectiveKVMModuleSupport()
+	}
+	return kvmModuleSupport
+}
+
+func IsKvmSupport() bool {
+	GetKVMModuleSupport()
+	if kvmModuleSupport == KVM_MODULE_UNSUPPORT {
+		return false
+	}
+	return true
+}
+
+func IsProcessorIntel() bool {
+	GetKVMModuleSupport()
+	if kvmModuleSupport == KVM_MODULE_INTEL {
+		return true
+	}
+	return false
+}
+
+func IsProcessorAmd() bool {
+	GetKVMModuleSupport()
+	if kvmModuleSupport == KVM_MODULE_AMD {
+		return true
+	}
+	return false
+}
+
+func detectiveKVMModuleSupport() string {
+	var km = KVM_MODULE_UNSUPPORT
+	if ModprobeKvmModule(KVM_MODULE_INTEL, false, false) {
+		km = KVM_MODULE_INTEL
+	} else if ModprobeKvmModule(KVM_MODULE_AMD, false, false) {
+		km = KVM_MODULE_AMD
+	}
+	return km
+}
+
+func ModprobeKvmModule(name string, remove, nest bool) bool {
+	var params = []string{"modprobe"}
+	if remove {
+		params = append(params, "-r")
+	}
+	params = append(params, name)
+	if nest {
+		params = append(params, "nested=1")
+	}
+	if _, err := procutils.NewCommand(params[0], params[1:]...).Run(); err != nil {
+		return false
+	}
+	return true
+}
+
+func IsNestEnabled() bool {
+	return GetNestSupport() == HOST_NEST_ENABLE
+}
+
+func GetNestSupport() string {
+	if len(nestStatus) == 0 {
+		nestStatus = detectNestSupport()
+	}
+	return nestStatus
+}
+
+func detectNestSupport() string {
+	moduleName := GetKVMModuleSupport()
+	nestStatus := HOST_NEST_UNSUPPORT
+
+	if moduleName != KVM_MODULE_UNSUPPORT && isNestSupport(moduleName) {
+		nestStatus = HOST_NEST_SUPPORT
+	}
+
+	if nestStatus == HOST_NEST_SUPPORT && loadKvmModuleWithNest(moduleName) {
+		nestStatus = HOST_NEST_ENABLE
+	}
+	return nestStatus
+}
+
+func isNestSupport(name string) bool {
+	output, err := procutils.NewCommand("modinfo", name).Run()
+	if err != nil {
+		log.Errorln(err)
+		return false
+	}
+
+	// TODO Test
+	var re = regexp.MustCompile(`parm:\s*nested:`)
+	for _, line := range strings.Split(string(output), "\n") {
+		if re.MatchString(line) {
+			return true
+		}
+	}
+	return false
+}
+
+func loadKvmModuleWithNest(name string) bool {
+	var notload = true
+	if IsKernelModuleLoaded(name) {
+		nest := GetKernelModuleParameter(name, "nested")
+		if nest == "Y" {
+			return true
+		}
+		notload = unloadKvmModule(name)
+	}
+	if notload {
+		if ModprobeKvmModule(name, false, true) {
+			return true
+		}
+	}
+	return false
+}
+
+func unloadKvmModule(name string) bool {
+	return ModprobeKvmModule(name, true, false)
+}
+
+func GetKernelModuleParameter(name, moduel string) string {
+	pa := path.Join("/sys/module/", strings.Replace(name, "-", "_", -1), "/parameters/", moduel)
+	if f, err := os.Stat(pa); err == nil {
+		if f.IsDir() {
+			return ""
+		}
+		cont, err := fileutils2.FileGetContents(pa)
+		if err != nil {
+			log.Errorln(err)
+			return ""
+		}
+		return strings.TrimSpace(cont)
+	}
+	return ""
+}
+
+func IsKernelModuleLoaded(name string) bool {
+	output, err := procutils.NewCommand("lsmod").Run()
+	if err != nil {
+		log.Errorln(err)
+		return false
+	}
+	for _, line := range strings.Split(string(output), "\n") {
+		lm := strings.Split(line, " ")
+		if len(lm) > 0 && utils.IsInStringArray(strings.Replace(name, "-", "_", -1), lm) {
+			return true
+		}
+	}
+	return false
+}