70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
//go:build windows
|
||
|
||
package adminreset
|
||
|
||
import (
|
||
"bufio"
|
||
"errors"
|
||
"fmt"
|
||
"os"
|
||
"strings"
|
||
|
||
"git.pcux.cn/xuans/Smail/utils/password"
|
||
|
||
"golang.org/x/term"
|
||
)
|
||
|
||
func ReadProtectedPassword() (string, error) {
|
||
if !term.IsTerminal(int(os.Stdin.Fd())) {
|
||
return "", errors.New("受保护的密码输入需要交互式控制台")
|
||
}
|
||
first, err := readWindowsPassword("请输入新的超级管理员密码")
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
second, err := readWindowsPassword("请再次输入新的超级管理员密码")
|
||
if err != nil {
|
||
clearPassword(&first)
|
||
return "", err
|
||
}
|
||
if first == "" || first != second {
|
||
clearPassword(&first)
|
||
clearPassword(&second)
|
||
return "", errors.New("密码为空或两次输入不一致")
|
||
}
|
||
clearPassword(&second)
|
||
return first, nil
|
||
}
|
||
|
||
func readWindowsPassword(prompt string) (string, error) {
|
||
if !term.IsTerminal(int(os.Stdin.Fd())) {
|
||
return "", errors.New("受保护的密码输入需要交互式控制台")
|
||
}
|
||
fmt.Fprint(os.Stderr, prompt+": ")
|
||
valueBytes, err := term.ReadPassword(int(os.Stdin.Fd()))
|
||
fmt.Fprintln(os.Stderr)
|
||
if err != nil {
|
||
return "", errors.New("受保护的密码输入失败")
|
||
}
|
||
value := string(valueBytes)
|
||
for index := range valueBytes {
|
||
valueBytes[index] = 0
|
||
}
|
||
if password.ValidateUserPassword(value) != nil {
|
||
return "", errors.New("密码长度必须为 8 到 12 个字符")
|
||
}
|
||
return value, nil
|
||
}
|
||
|
||
func ReadVisibleInput(prompt string) (string, error) {
|
||
if !term.IsTerminal(int(os.Stdin.Fd())) {
|
||
return "", errors.New("安全验证需要交互式控制台")
|
||
}
|
||
fmt.Fprint(os.Stderr, strings.TrimRight(prompt, ":： ")+"： ")
|
||
value, err := bufio.NewReader(os.Stdin).ReadString('\n')
|
||
if err != nil {
|
||
return "", errors.New("读取安全验证输入失败")
|
||
}
|
||
return strings.TrimSuffix(strings.TrimSuffix(value, "\n"), "\r"), nil
|
||
}
|