arno/kube-bench
1
0
Fork 0
mirror of https://github.com/aquasecurity/kube-bench.git synced 2024-11-26 01:49:28 +00:00
Code Issues Releases Wiki Activity

Add default version if version check fails.

Browse Source
This commit is contained in:
Abubakr-Sadik Nii Nai Davis 2017-11-13 15:25:34 +00:00
parent f90dd925b8
commit 42a1068964
2 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
cmd/root.go
View File

@ -27,6 +27,7 @@ import (
var (
envVarsPrefix = "KUBE_BENCH"
cfgDir = "./cfg"
defaultKubeVersion = "1.6"
cfgFile string
jsonFmt bool
pgSql bool

22
cmd/util.go
View File

@ -214,25 +214,35 @@ func multiWordReplace(s string, subname string, sub string) string {
}
func getKubeVersion() string {
var ver string
var matched bool
failmsg := "kubernetes version check failed"
// These executables might not be on the user's path.
_, err := exec.LookPath("kubectl")
if err != nil {
exitWithError(fmt.Errorf("%s: %s", failmsg, err))
continueWithError(err, failmsg)
}
cmd := exec.Command("kubectl", "version", "--short")
out, err := cmd.CombinedOutput()
if err != nil {
exitWithError(fmt.Errorf("%s, %s", failmsg, out))
continueWithError(fmt.Errorf("%s", out), "")
}
validVersionPttn := `\d.\d`
serverVersionRe := regexp.MustCompile(`Server Version: v(\d+.\d+)`)
ver := serverVersionRe.FindStringSubmatch(string(out))[1]
subs := serverVersionRe.FindStringSubmatch(string(out))
if len(subs) > 2 {
ver = string(subs[1])
validVersionPttn := `\d.\d`
if matched, _ = regexp.MatchString(validVersionPttn, ver); !matched {
continueWithError(fmt.Errorf("%s: invalid server version ", ver), failmsg)
}
}
if matched, _ := regexp.MatchString(validVersionPttn, ver); !matched {
exitWithError(fmt.Errorf("%s: invalid server version ", failmsg, ver))
if ver == "" || !matched {
printlnWarn(fmt.Sprintf("Unable to get kubectl version, using default version: %s", defaultKubeVersion))
ver = defaultKubeVersion
}
return ver