mirror of
https://github.com/hacdias/webdav.git
synced 2026-09-23 11:41:54 +08:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8c49af0b68 |
+2
-2
@@ -160,8 +160,8 @@ func (c *Config) Validate() error {
|
|||||||
return fmt.Errorf("invalid config: %w", err)
|
return fmt.Errorf("invalid config: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, u := range c.Users {
|
for i := range c.Users {
|
||||||
err := u.Validate(c.NoPassword)
|
err := c.Users[i].Validate(c.NoPassword)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("invalid config: %w", err)
|
return fmt.Errorf("invalid config: %w", err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,6 +22,17 @@ func writeAndParseConfig(t *testing.T, content, extension string) *Config {
|
|||||||
return cfg
|
return cfg
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func writeAndParseConfigWithError(t *testing.T, content, extension, error string) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
tmpFile := filepath.Join(tmpDir, "config"+extension)
|
||||||
|
|
||||||
|
err := os.WriteFile(tmpFile, []byte(content), 0666)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
_, err = ParseConfig(tmpFile, nil)
|
||||||
|
require.ErrorContains(t, err, error)
|
||||||
|
}
|
||||||
|
|
||||||
func TestConfigDefaults(t *testing.T) {
|
func TestConfigDefaults(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
@@ -224,3 +235,33 @@ func TestConfigEnv(t *testing.T) {
|
|||||||
require.NoError(t, os.Setenv("WD_PERMISSIONS", ""))
|
require.NoError(t, os.Setenv("WD_PERMISSIONS", ""))
|
||||||
require.NoError(t, os.Setenv("WD_DIRECTORY", ""))
|
require.NoError(t, os.Setenv("WD_DIRECTORY", ""))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestConfigParseUserPasswordEnvironment(t *testing.T) {
|
||||||
|
content := `
|
||||||
|
directory: /
|
||||||
|
users:
|
||||||
|
- username: '{env}USER1_USERNAME'
|
||||||
|
password: '{env}USER1_PASSWORD'
|
||||||
|
- username: basic
|
||||||
|
password: basic
|
||||||
|
`
|
||||||
|
|
||||||
|
writeAndParseConfigWithError(t, content, ".yml", "username environment variable is empty")
|
||||||
|
|
||||||
|
err := os.Setenv("USER1_USERNAME", "admin")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
writeAndParseConfigWithError(t, content, ".yml", "password environment variable is empty")
|
||||||
|
|
||||||
|
err = os.Setenv("USER1_PASSWORD", "admin")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
cfg := writeAndParseConfig(t, content, ".yaml")
|
||||||
|
require.NoError(t, cfg.Validate())
|
||||||
|
|
||||||
|
require.Equal(t, "admin", cfg.Users[0].Username)
|
||||||
|
require.Equal(t, "basic", cfg.Users[1].Username)
|
||||||
|
|
||||||
|
require.True(t, cfg.Users[0].checkPassword("admin"))
|
||||||
|
require.True(t, cfg.Users[1].checkPassword("basic"))
|
||||||
|
}
|
||||||
|
|||||||
+10
-1
@@ -27,12 +27,21 @@ func (u User) checkPassword(input string) bool {
|
|||||||
func (u *User) Validate(noPassword bool) error {
|
func (u *User) Validate(noPassword bool) error {
|
||||||
if u.Username == "" {
|
if u.Username == "" {
|
||||||
return errors.New("invalid user: username must be set")
|
return errors.New("invalid user: username must be set")
|
||||||
|
} else if strings.HasPrefix(u.Username, "{env}") {
|
||||||
|
env := strings.TrimPrefix(u.Username, "{env}")
|
||||||
|
if env == "" {
|
||||||
|
return fmt.Errorf("invalid user %q: username environment variable not set", u.Username)
|
||||||
|
}
|
||||||
|
|
||||||
|
u.Username = os.Getenv(env)
|
||||||
|
if u.Username == "" {
|
||||||
|
return fmt.Errorf("invalid user %q: username environment variable is empty", u.Username)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if u.Password == "" && !noPassword {
|
if u.Password == "" && !noPassword {
|
||||||
return fmt.Errorf("invalid user %q: password must be set", u.Username)
|
return fmt.Errorf("invalid user %q: password must be set", u.Username)
|
||||||
} else if strings.HasPrefix(u.Password, "{env}") {
|
} else if strings.HasPrefix(u.Password, "{env}") {
|
||||||
|
|
||||||
env := strings.TrimPrefix(u.Password, "{env}")
|
env := strings.TrimPrefix(u.Password, "{env}")
|
||||||
if env == "" {
|
if env == "" {
|
||||||
return fmt.Errorf("invalid user %q: password environment variable not set", u.Username)
|
return fmt.Errorf("invalid user %q: password environment variable not set", u.Username)
|
||||||
|
|||||||
Reference in New Issue
Block a user