diff options
| -rw-r--r-- | conf/conf.go | 63 | ||||
| -rw-r--r-- | main.go | 2 | ||||
| -rw-r--r-- | openbills.toml | 2 |
3 files changed, 58 insertions, 9 deletions
diff --git a/conf/conf.go b/conf/conf.go index 26a5d90..6cf0d99 100644 --- a/conf/conf.go +++ b/conf/conf.go @@ -19,9 +19,60 @@ package conf import ( "github.com/spf13/viper" + "golang.org/x/crypto/bcrypt" "log" ) +func validateConf() { + ok := true + log.Println("\x1b[46m\x1b[30m[info]\x1b[0m Checking errors in config file...") + + port := viper.GetInt("port") + if port < 1024 || port > 65535 { + log.Println("\x1b[41m\x1b[30m[err]\x1b[0m Invalid port number.") + ok = false + } + + if viper.GetInt("username.min_username_length") < 1 { + log.Println("\x1b[41m\x1b[30m[err]\x1b[0m Minimum username length must be greater than 0.") + ok = false + } + + minPassLen := viper.GetInt("security.min_password_length") + maxPassLen := viper.GetInt("security.max_password_length") + + if minPassLen < 1 { + log.Println("\x1b[41m\x1b[30m[err]\x1b[0m Minimum password length must be greater than 0.") + ok = false + } + + if maxPassLen > 72 { + log.Println("\x1b[41m\x1b[30m[err]\x1b[0m Maximum password length can't exceed 72 characters.") + ok = false + } + + if minPassLen > maxPassLen { + log.Println("\x1b[41m\x1b[30m[err]\x1b[0m Minimum password length can't be greater than maximum password length.") + ok = false + } + + hashingCost := viper.GetInt("cryptography.password_hashing_cost") + + if hashingCost > bcrypt.MaxCost { + log.Printf("\x1b[41m\x1b[30m[err]\x1b[0m Password hashing cost can't be greater than %d.\n", bcrypt.MaxCost) + ok = false + } + + if hashingCost < bcrypt.MinCost { + log.Printf("\x1b[41m\x1b[30m[err]\x1b[0m Password hashing cost can't be less than %d.\n", bcrypt.MinCost) + ok = false + } + + if !ok { + log.Fatalln("\x1b[41m\x1b[30m[err]\x1b[0m Config file has errors.") + } +} + // TODO: validate config func init() { viper.SetConfigName("openbills") @@ -29,11 +80,8 @@ func init() { viper.AddConfigPath(".") if err := viper.ReadInConfig(); err != nil { - if _, ok := err.(viper.ConfigFileNotFoundError); ok { - log.Printf("\x1b[43m\x1b[30m[warn]\x1b[0m config file not found, skipping...\n") - } else { - log.Fatalf("\x1b[41m\x1b[30m[err]\x1b[0m cannot read config file. %v\n", err.Error()) - } + log.Printf("\x1b[41m\x1b[30m[err]\x1b[0m Failed to read config file:\n") + log.Fatalf("%v\n", err.Error()) } viper.SetDefault("port", "8765") @@ -51,7 +99,8 @@ func init() { viper.SetDefault("username.min_username_length", 2) viper.SetDefault("username.max_username_length", 20) - viper.SetDefault("cryptography.password_hashing_cost", 14) + viper.SetDefault("cryptography.password_hashing_cost", bcrypt.DefaultCost) - log.Printf("\x1b[46m\x1b[30m[info]\x1b[0m Loaded Config File \"%s\"\n", viper.ConfigFileUsed()) + validateConf() + log.Printf("\x1b[46m\x1b[30m[info]\x1b[0m Loaded Config \"%s\"\n", viper.ConfigFileUsed()) } @@ -37,7 +37,7 @@ import ( "log" ) -const OPENBILLS_VERSION = "v0.0.1" +const OPENBILLS_VERSION = "v0.0.2" func init() { if viper.GetBool("production_mode") { diff --git a/openbills.toml b/openbills.toml index d830544..dd37e22 100644 --- a/openbills.toml +++ b/openbills.toml @@ -1,5 +1,5 @@ port = 8765 -production_mode = true +production_mode = false [database] username = "openbills" |