52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package security
|
||
|
||
import (
|
||
"encoding/base64"
|
||
"errors"
|
||
"math/rand"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
const passwordLength = 256
|
||
|
||
type password [passwordLength]byte
|
||
|
||
func init() {
|
||
// 更新随机种子，防止生成一样的随机密码
|
||
rand.Seed(time.Now().Unix())
|
||
}
|
||
|
||
// 采用base64编码把密码转换为字符串
|
||
func (password *password) String() string {
|
||
return base64.StdEncoding.EncodeToString(password[:])
|
||
}
|
||
|
||
// ParseLittlePassword 解析采用base64编码的字符串获取密码
|
||
func ParseLittlePassword(passwordString string) (*password, error) {
|
||
bs, err := base64.StdEncoding.DecodeString(strings.TrimSpace(passwordString))
|
||
if err != nil || len(bs) != passwordLength {
|
||
return nil, errors.New("不合法的密码")
|
||
}
|
||
password := password{}
|
||
copy(password[:], bs)
|
||
//bs = nil
|
||
return &password, nil
|
||
}
|
||
|
||
// RandLittlePassword 产生 256个byte随机组合的 密码，最后会使用base64编码为字符串存储在配置文件中
|
||
// 不能出现任何一个重复的byte位，必须又 0-255 组成，并且都需要包含
|
||
func RandLittlePassword() string {
|
||
// 随机生成一个由 0~255 组成的 byte 数组
|
||
intArr := rand.Perm(passwordLength)
|
||
password := &password{}
|
||
for i, v := range intArr {
|
||
password[i] = byte(v)
|
||
if i == v {
|
||
// 确保不会出现如何一个byte位出现重复
|
||
return RandLittlePassword()
|
||
}
|
||
}
|
||
return password.String()
|
||
}
|