lee/backrest
1
0
Fork 0
mirror of https://github.com/garethgeorge/backrest.git synced 2025-12-12 16:55:39 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: validate plan ID and repo ID

Browse Source
This commit is contained in:
Gareth
2024-05-05 07:59:53 -07:00
parent 8448f4cc3a
commit f314c7cced
3 changed files with 21 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/config/stringutil/stringutil.go
View File

@@ -2,7 +2,15 @@ package stringutil
import "regexp"
var (
sanitizeIDRegex = regexp.MustCompile(`[^a-zA-Z0-9_\-\.]+`) // matches invalid characters in an ID
idRegex = regexp.MustCompile(`[a-zA-Z0-9_\-\.]*`) // matches a valid ID (including empty string)
)
func SanitizeID(id string) string {
reg := regexp.MustCompile(`[^a-zA-Z0-9_\-\.]+`)
return reg.ReplaceAllString(id, "_")
return sanitizeIDRegex.ReplaceAllString(id, "_")
}
func ValidateID(id string) bool {
return idRegex.MatchString(id)
}

9
internal/config/validate.go
View File

@@ -63,8 +63,9 @@ func ValidateConfig(c *v1.Config) error {
func validateRepo(repo *v1.Repo) error {
var err error
if repo.Id == "" {
err = multierror.Append(err, errors.New("id is required"))
if repo.Id == "" || !stringutil.ValidateID(repo.Id) {
err = multierror.Append(err, fmt.Errorf("id %q contains invalid characters (or empty)", repo.Id))
}
if repo.Uri == "" {
@@ -88,6 +89,10 @@ func validatePlan(plan *v1.Plan, repos map[string]*v1.Repo) error {
err = multierror.Append(err, fmt.Errorf("path is required"))
}
if plan.Id == "" || !stringutil.ValidateID(plan.Id) {
err = multierror.Append(err, fmt.Errorf("id %q contains invalid characters (or empty)", plan.Id))
}
for idx, p := range plan.Paths {
if p == "" {
err = multierror.Append(err, fmt.Errorf("path[%d] cannot be empty", idx))

12
internal/orchestrator/repo/repo.go
View File

@@ -35,8 +35,10 @@ func NewRepoOrchestrator(config *v1.Config, repoConfig *v1.Repo, resticPath stri
opts = append(opts, restic.WithEnviron())
opts = append(opts, restic.WithEnv("RESTIC_PROGRESS_FPS=0.5"))
if len(repoConfig.GetEnv()) > 0 {
opts = append(opts, restic.WithEnv(repoConfig.GetEnv()...))
if env := repoConfig.GetEnv(); len(env) != 0 {
for _, e := range env {
opts = append(opts, restic.WithEnv(ExpandEnv(e)))
}
}
if p := repoConfig.GetPassword(); p != "" {
@@ -58,12 +60,6 @@ func NewRepoOrchestrator(config *v1.Config, repoConfig *v1.Repo, resticPath stri
opts = append(opts, restic.WithFlags("-o", "sftp.args=-oBatchMode=yes"))
}
if env := repoConfig.GetEnv(); len(env) != 0 {
for _, e := range env {
opts = append(opts, restic.WithEnv(ExpandEnv(e)))
}
}
repo := restic.NewRepo(resticPath, repoConfig.GetUri(), opts...)
return &RepoOrchestrator{