test/security/k8s/src/check/cmd/check/check.go

   1 package main
   2
   3 import (
   4         "flag"
   5         "log"
   6
   7         "check/rancher"
   8         "check/raw"
   9         "check/validators/master"
  10 )
  11
  12 var (
  13         ranchercli = flag.Bool("ranchercli", false, "use rancher utility for accessing cluster nodes")
  14         rke        = flag.Bool("rke", false, "use RKE cluster definition and ssh for accessing cluster nodes (default)")
  15 )
  16
  17 func main() {
  18         flag.Parse()
  19         if *ranchercli && *rke {
  20                 log.Fatal("Not supported.")
  21         }
  22
  23         // Use default cluster access method if none was declared explicitly.
  24         if !(*ranchercli || *rke) {
  25                 *rke = true
  26         }
  27
  28         var (
  29                 k8sParams []string
  30                 err       error
  31         )
  32
  33         switch {
  34         case *ranchercli:
  35                 k8sParams, err = rancher.GetK8sParams()
  36         case *rke:
  37                 k8sParams, err = raw.GetK8sParams()
  38         default:
  39                 log.Fatal("Missing cluster access method.")
  40         }
  41
  42         if err != nil {
  43                 log.Fatal(err)
  44         }
  45
  46         master.Check(k8sParams)
  47 }