scsusers/2fa.go

63 lines
1.1 KiB
Go

package scsusers
import (
"crypto/rand"
"errors"
"fmt"
"math/big"
"strconv"
"time"
)
func generate2fa() string {
num, err := rand.Int(rand.Reader, big.NewInt(999999))
if err != nil {
return "918273"
}
return fmt.Sprintf("%06d", num)
}
func Validate2FA(u *UserData, challenge string) bool {
defer func() {
u.Delete("2fa")
u.Delete("2faexpires")
}()
v, ok := u.Get("2fa")
if !ok {
return false
}
exp, ok := u.Get("2faexpires")
if !ok {
return false
}
tmp, err := strconv.ParseInt(exp, 10, 64)
if err != nil {
return false
}
if int64(time.Now().Unix()) > tmp {
return false
}
return ok && v == challenge
}
func Send2FA(u *UserData) error {
code := generate2fa()
u.Set("2fa", code)
u.Set("2faexpires", fmt.Sprintf("%d", time.Now().Add(15*time.Minute).Unix()))
firstname, ok := u.Get("firstname")
if !ok {
return errors.New("send2fa: no firstname")
}
lastname, ok := u.Get("lastname")
if !ok {
return errors.New("send2fa: no lastname")
}
fullname := fmt.Sprintf("%s %s", firstname, lastname)
return Send2faEmail(u.Username, fullname, code)
}