mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-16 07:16:38 +00:00
3ce5d6a4f8
Introduced the NETBIRD_IDP_TIMEOUT environment variable to the management service. This allows configuring a timeout for supported IDPs. If the variable is unset or contains an invalid value, a default timeout of 10 seconds is used as a fallback. This is needed for larger IDP environments where 10s is just not enough time.
95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package idp
|
|
|
|
import (
|
|
"encoding/json"
|
|
"math/rand"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
lowerCharSet = "abcdedfghijklmnopqrst"
|
|
upperCharSet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
specialCharSet = "!@#$%&*"
|
|
numberSet = "0123456789"
|
|
allCharSet = lowerCharSet + upperCharSet + specialCharSet + numberSet
|
|
)
|
|
|
|
type JsonParser struct{}
|
|
|
|
func (JsonParser) Marshal(v interface{}) ([]byte, error) {
|
|
return json.Marshal(v)
|
|
}
|
|
|
|
func (JsonParser) Unmarshal(data []byte, v interface{}) error {
|
|
return json.Unmarshal(data, v)
|
|
}
|
|
|
|
// GeneratePassword generates user password
|
|
func GeneratePassword(passwordLength, minSpecialChar, minNum, minUpperCase int) string {
|
|
var password strings.Builder
|
|
|
|
//Set special character
|
|
for i := 0; i < minSpecialChar; i++ {
|
|
random := rand.Intn(len(specialCharSet))
|
|
password.WriteString(string(specialCharSet[random]))
|
|
}
|
|
|
|
//Set numeric
|
|
for i := 0; i < minNum; i++ {
|
|
random := rand.Intn(len(numberSet))
|
|
password.WriteString(string(numberSet[random]))
|
|
}
|
|
|
|
//Set uppercase
|
|
for i := 0; i < minUpperCase; i++ {
|
|
random := rand.Intn(len(upperCharSet))
|
|
password.WriteString(string(upperCharSet[random]))
|
|
}
|
|
|
|
remainingLength := passwordLength - minSpecialChar - minNum - minUpperCase
|
|
for i := 0; i < remainingLength; i++ {
|
|
random := rand.Intn(len(allCharSet))
|
|
password.WriteString(string(allCharSet[random]))
|
|
}
|
|
inRune := []rune(password.String())
|
|
rand.Shuffle(len(inRune), func(i, j int) {
|
|
inRune[i], inRune[j] = inRune[j], inRune[i]
|
|
})
|
|
return string(inRune)
|
|
}
|
|
|
|
// baseURL returns the base url by concatenating
|
|
// the scheme and host components of the parsed URL.
|
|
func baseURL(rawURL string) string {
|
|
parsedURL, err := url.Parse(rawURL)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return parsedURL.Scheme + "://" + parsedURL.Host
|
|
}
|
|
|
|
const (
|
|
// Provides the env variable name for use with idpTimeout function
|
|
idpTimeoutEnv = "NB_IDP_TIMEOUT"
|
|
// Sets the defaultTimeout to 10s.
|
|
defaultTimeout = 10 * time.Second
|
|
)
|
|
|
|
// idpTimeout returns a timeout value for the IDP
|
|
func idpTimeout() time.Duration {
|
|
timeoutStr, ok := os.LookupEnv(idpTimeoutEnv)
|
|
if !ok || timeoutStr == "" {
|
|
return defaultTimeout
|
|
}
|
|
|
|
timeout, err := time.ParseDuration(timeoutStr)
|
|
if err != nil {
|
|
return defaultTimeout
|
|
}
|
|
return timeout
|
|
}
|