Skip to content

Commit

Permalink
move hostname check to util
Browse files Browse the repository at this point in the history
  • Loading branch information
iwilltry42 committed May 9, 2019
1 parent 4c14a8e commit fd720b0
Show file tree
Hide file tree
Showing 2 changed files with 35 additions and 33 deletions.
35 changes: 2 additions & 33 deletions cli/commands.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,37 +24,6 @@ import (
)

const defaultRegistry = "docker.io"
const clusterNameMaxSize int = 35

// Make sure a cluster name is also a valid host name according to RFC 1123.
// We further restrict the length of the cluster name to shorter than 'clusterNameMaxSize'
// so that we can construct the host names based on the cluster name, and still stay
// within the 64 characters limit.
func checkClusterName(name string) error {
if len(name) > clusterNameMaxSize {
return fmt.Errorf("cluster name is too long")
}

if name[0] == '-' || name[len(name) - 1] == '-' {
return fmt.Errorf("cluster name can not start or end with - (dash)")
}


for _ , c := range name {
switch {
case '0' <= c && c <= '9':
case 'a' <= c && c <= 'z':
case 'A' <= c && c <= 'Z':
case c == '-':
break;
default:
return fmt.Errorf("cluster name contains charaters other than 'Aa-Zz', '0-9' or '-'")

}
}

return nil
}

// CheckTools checks if the docker API server is responding
func CheckTools(c *cli.Context) error {
Expand All @@ -76,8 +45,8 @@ func CheckTools(c *cli.Context) error {
// CreateCluster creates a new single-node cluster container and initializes the cluster directory
func CreateCluster(c *cli.Context) error {

if err := checkClusterName(c.String("name")); err != nil {
return err;
if err := CheckClusterName(c.String("name")); err != nil {
return err
}

// define image
Expand Down
33 changes: 33 additions & 0 deletions cli/util.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package run

import (
"fmt"
"math/rand"
"strings"
"time"
Expand Down Expand Up @@ -36,3 +37,35 @@ func GenerateRandomString(n int) string {

return sb.String()
}

/*** Cluster Name Validation ***/
const clusterNameMaxSize int = 35

// CheckClusterName ensures that a cluster name is also a valid host name according to RFC 1123.
// We further restrict the length of the cluster name to maximum 'clusterNameMaxSize'
// so that we can construct the host names based on the cluster name, and still stay
// within the 64 characters limit.
func CheckClusterName(name string) error {
if len(name) > clusterNameMaxSize {
return fmt.Errorf("[ERROR] Cluster name is too long")
}

if name[0] == '-' || name[len(name)-1] == '-' {
return fmt.Errorf("[ERROR] Cluster name can not start or end with - (dash)")
}

for _, c := range name {
switch {
case '0' <= c && c <= '9':
case 'a' <= c && c <= 'z':
case 'A' <= c && c <= 'Z':
case c == '-':
break
default:
return fmt.Errorf("[ERROR] Cluster name contains characters other than 'Aa-Zz', '0-9' or '-'")

}
}

return nil
}

0 comments on commit fd720b0

Please sign in to comment.