54 lines
1.6 KiB
Go
54 lines
1.6 KiB
Go
package flow
|
|
|
|
import (
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
const CodePasswordSeparator = ";"
|
|
|
|
var alphaNum = regexp.MustCompile(`^[a-zA-Z0-9]*$`)
|
|
|
|
// Sets the secret answers for the flow executor for protocols that only support username/password
|
|
// according to used options
|
|
func (fe *FlowExecutor) SetSecrets(password string, mfaCodeBased bool) {
|
|
if fe.Answers[StageAuthenticatorValidate] != "" || fe.Answers[StagePassword] != "" {
|
|
return
|
|
}
|
|
fe.Answers[StagePassword] = password
|
|
if !mfaCodeBased {
|
|
// If code-based MFA is disabled StageAuthenticatorValidate answer is set to password.
|
|
// This allows flows with a mfa stage only.
|
|
fe.Answers[StageAuthenticatorValidate] = password
|
|
return
|
|
}
|
|
// password doesn't contain the separator
|
|
if !strings.Contains(password, CodePasswordSeparator) {
|
|
return
|
|
}
|
|
// password ends with the separator, so it won't contain an answer
|
|
if strings.HasSuffix(password, CodePasswordSeparator) {
|
|
return
|
|
}
|
|
idx := strings.LastIndex(password, CodePasswordSeparator)
|
|
authenticator := password[idx+1:]
|
|
// Authenticator is either 6 chars (totp code) or 8 chars (long totp or static)
|
|
if len(authenticator) == 6 {
|
|
// authenticator answer isn't purely numerical, so won't be value
|
|
if _, err := strconv.Atoi(authenticator); err != nil {
|
|
return
|
|
}
|
|
} else if len(authenticator) == 8 {
|
|
// 8 chars can be a long totp or static token, so it needs to be alphanumerical
|
|
if !alphaNum.MatchString(authenticator) {
|
|
return
|
|
}
|
|
} else {
|
|
// Any other length, doesn't contain an answer
|
|
return
|
|
}
|
|
fe.Answers[StagePassword] = password[:idx]
|
|
fe.Answers[StageAuthenticatorValidate] = authenticator
|
|
}
|