1Panel/core/init/validator/validator.go at 18690411a679f6855f3d73d8fdd806430f8b2d46 · 1Panel-dev/1Panel · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package validator

import (
	"regexp"
	"unicode"

	"github.com/1Panel-dev/1Panel/core/global"

	"github.com/go-playground/validator/v10"
)

func Init() {
	validator := validator.New()
	if err := validator.RegisterValidation("name", checkNamePattern); err != nil {
		panic(err)
	}
	if err := validator.RegisterValidation("ip", checkIpPattern); err != nil {
		panic(err)
	}
	if err := validator.RegisterValidation("password", checkPasswordPattern); err != nil {
		panic(err)
	}
	if err := validator.RegisterValidation("base_setting_key", checkBaseSettingKey); err != nil {
		panic(err)
	}
	global.VALID = validator
}

var baseSettingKeys = map[string]struct{}{
	"PanelName":                  {},
	"Theme":                      {},
	"MenuTabs":                   {},
	"Language":                   {},
	"DeveloperMode":              {},
	"UpgradeBackupCopies":        {},
	"SecurityEntrance":           {},
	"BindDomain":                 {},
	"AllowIPs":                   {},
	"PasskeyTrustedProxies":      {},
	"ComplexityVerification":     {},
	"NoAuthSetting":              {},
	"DashboardMemoVisible":       {},
	"DashboardSimpleNodeVisible": {},
	"Edition":                    {},
	"DocSource":                  {},
	"IsOffline":                  {},
	"AppStoreLastModified":       {},
	"ScriptSync":                 {},
	"HideMenu":                   {},
	"OpsReportExportFormat":      {},
	"OpsReportSchedule":          {},
	"OpsReportSavePath":          {},
	"OpsReportThreshold":         {},
}

func checkNamePattern(fl validator.FieldLevel) bool {
	value := fl.Field().String()
	result, err := regexp.MatchString("^[a-zA-Z\u4e00-\u9fa5]{1}[a-zA-Z0-9_\u4e00-\u9fa5]{0,30}$", value)
	if err != nil {
		global.LOG.Errorf("regexp matchString failed, %v", err)
	}
	return result
}

func checkIpPattern(fl validator.FieldLevel) bool {
	value := fl.Field().String()
	result, err := regexp.MatchString(`^((2(5[0-5]|[0-4]\d))|[0-1]?\d{1,2})(\.((2(5[0-5]|[0-4]\d))|[0-1]?\d{1,2})){3}$`, value)
	if err != nil {
		global.LOG.Errorf("regexp check ip matchString failed, %v", err)
	}
	return result
}

func checkPasswordPattern(fl validator.FieldLevel) bool {
	value := fl.Field().String()
	if len(value) < 8 || len(value) > 30 {
		return false
	}

	hasNum := false
	hasLetter := false
	for _, r := range value {
		if unicode.IsLetter(r) && !hasLetter {
			hasLetter = true
		}
		if unicode.IsNumber(r) && !hasNum {
			hasNum = true
		}
		if hasLetter && hasNum {
			return true
		}
	}

	return false
}

func checkBaseSettingKey(fl validator.FieldLevel) bool {
	_, ok := baseSettingKeys[fl.Field().String()]
	return ok
}