utils.go => validations.go

utils/validations.go

@@ -0,0 +1,43 @@
+package utils
+
+import (
+	"encoding/binary"
+	"regexp"
+	"strings"
+	"unicode"
+
+	"github.com/acarl005/stripansi"
+)
+
+func ClearUnprintableChars(s string, allowNewlines bool) string {
+	// This will remove ANSI color codes.
+	s = stripansi.Strip(s)
+
+	return strings.Map(func(r rune) rune {
+		if unicode.IsPrint(r) || (allowNewlines && r == '\n') {
+			return r
+		}
+		return -1
+	}, s)
+}
+
+var devNameRegexp = regexp.MustCompile(`^[0-9a-z_-]+$`)
+
+func ValidateDevName(s string) bool {
+	// Allow mapped devices.
+	s = strings.TrimPrefix(s, "mapper/")
+
+	return devNameRegexp.MatchString(s)
+}
+
+var unixUsernameRegexp = regexp.MustCompile(`^[a-z_]([a-z0-9_-]{0,31}|[a-z0-9_-]{0,30}\$)$`)
+
+func ValidateUnixUsername(s string) bool {
+	return unixUsernameRegexp.MatchString(s)
+}
+
+func Uint16ToBytesBE(v uint16) []byte {
+	b := make([]byte, 2)
+	binary.BigEndian.PutUint16(b, v)
+	return b
+}