添加idkey源码

idkey.go

@@ -0,0 +1,115 @@
+package main
+
+import (
+	"bytes"
+	"crypto/rand"
+	"encoding/base64"
+	"errors"
+	"fmt"
+	"golang.org/x/crypto/argon2"
+	"strings"
+)
+
+type Data struct {
+	Hash    []byte // 密码密文
+	Salt    []byte // 加密盐值
+	Time    uint32 // 时间参数
+	Memory  uint32 // 内存参数
+	Threads uint8  // 线程参数
+	KeyLen  uint32 // 密文长度
+}
+
+func Encode(password string) string {
+	salt := generateSalt(16)
+	data := &Data{
+		Hash:    nil,
+		Salt:    salt,
+		Time:    1,
+		Memory:  64 * 1024,
+		Threads: 4,
+		KeyLen:  32,
+	}
+	hash := argon2.IDKey(
+		[]byte(password),
+		data.Salt,
+		data.Time,
+		data.Memory,
+		data.Threads,
+		data.KeyLen,
+	)
+	return fmt.Sprintf(
+		"$argon2id$v=%d$m=%d,t=%d,p=%d$%s$%s",
+		argon2.Version,
+		data.Memory,
+		data.Time,
+		data.Threads,
+		base64.RawStdEncoding.EncodeToString(salt),
+		base64.RawStdEncoding.EncodeToString(hash),
+	)
+}
+
+func Decode(passwordHash string) (data *Data, err error) {
+	data = &Data{}
+	params := strings.Split(passwordHash, "$")
+	if len(params) != 6 {
+		err = errors.New("input error")
+		return
+	}
+	var version int
+	_, err = fmt.Sscanf(
+		params[2],
+		"v=%d",
+		&version,
+	)
+	if err != nil {
+		return
+	}
+	if version != argon2.Version {
+		err = errors.New("not support")
+		return
+	}
+	_, err = fmt.Sscanf(
+		params[3],
+		"m=%d,t=%d,p=%d",
+		&data.Memory,
+		&data.Time,
+		&data.Threads,
+	)
+	if err != nil {
+		return
+	}
+	salt, err := base64.RawStdEncoding.DecodeString(params[4])
+	if err != nil {
+		return
+	}
+	data.Salt = salt
+	hash, err := base64.RawStdEncoding.DecodeString(params[5])
+	if err != nil {
+		return
+	}
+	data.Hash = hash
+	data.KeyLen = uint32(len(hash))
+	return
+}
+
+func Verify(password, passwordHash string) bool {
+	data, err := Decode(passwordHash)
+	if err != nil {
+		return false
+	}
+	hash := argon2.IDKey(
+		[]byte(password),
+		data.Salt,
+		data.Time,
+		data.Memory,
+		data.Threads,
+		data.KeyLen,
+	)
+	return bytes.Equal(hash, data.Hash)
+}
+
+func generateSalt(l int) (b []byte) {
+	b = make([]byte, l)
+	_, _ = rand.Read(b)
+	return
+}