mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-12-25 15:57:49 +08:00
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
|
package mfa
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/base64"
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/1Panel-dev/1Panel/core/global"
|
||
|
|
"github.com/skip2/go-qrcode"
|
||
|
|
"github.com/xlzd/gotp"
|
||
|
|
)
|
||
|
|
|
||
|
|
const secretLength = 16
|
||
|
|
|
||
|
|
type Otp struct {
|
||
|
|
Secret string `json:"secret"`
|
||
|
|
QrImage string `json:"qrImage"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetOtp(username, title string, interval int) (otp Otp, err error) {
|
||
|
|
secret := gotp.RandomSecret(secretLength)
|
||
|
|
otp.Secret = secret
|
||
|
|
totp := gotp.NewTOTP(secret, 6, interval, nil)
|
||
|
|
uri := totp.ProvisioningUri(username, title)
|
||
|
|
subImg, err := qrcode.Encode(uri, qrcode.Medium, 256)
|
||
|
|
dist := make([]byte, 3000)
|
||
|
|
base64.StdEncoding.Encode(dist, subImg)
|
||
|
|
index := bytes.IndexByte(dist, 0)
|
||
|
|
baseImage := dist[0:index]
|
||
|
|
otp.QrImage = "data:image/png;base64," + string(baseImage)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func ValidCode(code, intervalStr, secret string) bool {
|
||
|
|
interval, err := strconv.Atoi(intervalStr)
|
||
|
|
if err != nil {
|
||
|
|
global.LOG.Errorf("type conversion failed, err: %v", err)
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
totp := gotp.NewTOTP(secret, 6, interval, nil)
|
||
|
|
now := time.Now().Unix()
|
||
|
|
strInt64 := strconv.FormatInt(now, 10)
|
||
|
|
id16, _ := strconv.Atoi(strInt64)
|
||
|
|
return totp.Verify(code, int64(id16))
|
||
|
|
}
|