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

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