295 lines
8.4 KiB
Go
295 lines
8.4 KiB
Go
package service
|
|
|
|
import (
|
|
"crypto/ed25519"
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
"git.lumeweb.com/LumeWeb/portal/db"
|
|
"git.lumeweb.com/LumeWeb/portal/model"
|
|
"github.com/joomcode/errorx"
|
|
"github.com/kataras/iris/v12"
|
|
"github.com/kataras/jwt"
|
|
"golang.org/x/crypto/bcrypt"
|
|
"time"
|
|
)
|
|
|
|
var sharedKey = []byte("sercrethatmaycontainch@r$32chars")
|
|
|
|
var blocklist *jwt.Blocklist
|
|
|
|
func init() {
|
|
blocklist = jwt.NewBlocklist(1 * time.Hour)
|
|
}
|
|
|
|
type AuthService struct {
|
|
ctx iris.Context
|
|
}
|
|
|
|
type LoginRequest struct {
|
|
Email string `json:"email"`
|
|
Password string `json:"password"`
|
|
}
|
|
|
|
type LoginResponse struct {
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
type LogoutRequest struct {
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
type ChallengeRequest struct {
|
|
Pubkey string `json:"pubkey"`
|
|
}
|
|
|
|
type ChallengeResponse struct {
|
|
Challenge string `json:"challenge"`
|
|
}
|
|
|
|
type PubkeyLoginRequest struct {
|
|
Pubkey string `json:"public_key"`
|
|
Challenge string `json:"challenge"`
|
|
Signature string `json:"signature"`
|
|
}
|
|
|
|
// verifyPassword compares the provided plaintext password with a hashed password and returns an error if they don't match.
|
|
func verifyPassword(hashedPassword, password string) error {
|
|
err := bcrypt.CompareHashAndPassword([]byte(hashedPassword), []byte(password))
|
|
if err != nil {
|
|
return errors.New("invalid email or password")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// generateToken generates a JWT token for the given account ID.
|
|
func generateToken(maxAge time.Duration) (string, error) {
|
|
// Define the JWT claims.
|
|
claim := jwt.Claims{
|
|
Expiry: time.Now().Add(time.Hour * 24).Unix(), // Token expires in 24 hours.
|
|
IssuedAt: time.Now().Unix(),
|
|
}
|
|
|
|
token, err := jwt.Sign(jwt.HS256, sharedKey, claim, jwt.MaxAge(maxAge))
|
|
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return string(token), nil
|
|
}
|
|
|
|
func generateAndSaveLoginToken(accountID uint, maxAge time.Duration) (string, error) {
|
|
// Generate a JWT token for the authenticated user.
|
|
token, err := generateToken(maxAge)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to generate token: %s", err)
|
|
}
|
|
|
|
verifiedToken, _ := jwt.Verify(jwt.HS256, sharedKey, []byte(token), blocklist)
|
|
var claim *jwt.Claims
|
|
|
|
_ = verifiedToken.Claims(&claim)
|
|
|
|
// Save the token to the database.
|
|
session := model.LoginSession{
|
|
Account: model.Account{ID: accountID},
|
|
Token: token,
|
|
Expiration: claim.ExpiresAt(),
|
|
}
|
|
|
|
if err := db.Get().Create(&session).Error; err != nil {
|
|
return "", errorx.Decorate(err, "failed to save token")
|
|
}
|
|
|
|
return token, nil
|
|
}
|
|
|
|
func generateAndSaveChallengeToken(accountID uint, maxAge time.Duration) (string, error) {
|
|
// Generate a JWT token for the authenticated user.
|
|
token, err := generateToken(maxAge)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to generate token: %s", err)
|
|
}
|
|
|
|
verifiedToken, _ := jwt.Verify(jwt.HS256, sharedKey, []byte(token), blocklist)
|
|
var claim *jwt.Claims
|
|
|
|
_ = verifiedToken.Claims(&claim)
|
|
|
|
// Save the token to the database.
|
|
keyChallenge := model.KeyChallenge{
|
|
Account: model.Account{ID: accountID},
|
|
Challenge: token,
|
|
Expiration: claim.ExpiresAt(),
|
|
}
|
|
|
|
if err := db.Get().Create(&keyChallenge).Error; err != nil {
|
|
return "", errorx.Decorate(err, "failed to save token")
|
|
}
|
|
|
|
return token, nil
|
|
}
|
|
|
|
// PostLogin handles the POST /api/auth/login request to authenticate a user and return a JWT token.
|
|
func (a *AuthService) PostLogin() {
|
|
var r LoginRequest
|
|
|
|
// Read the login request from the client.
|
|
if err := a.ctx.ReadJSON(&r); err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
// Retrieve the account for the given email.
|
|
account := model.Account{}
|
|
if err := db.Get().Where("email = ?", r.Email).First(&account).Error; err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errors.New("invalid email or password"))
|
|
return
|
|
}
|
|
|
|
// Verify the provided password against the hashed password stored in the database.
|
|
if err := verifyPassword(*account.Password, r.Password); err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errors.New("invalid email or password"))
|
|
return
|
|
}
|
|
|
|
// Generate a JWT token for the authenticated user.
|
|
token, err := generateAndSaveLoginToken(account.ID, 24*time.Hour)
|
|
if err != nil {
|
|
a.ctx.StopWithError(iris.StatusInternalServerError, fmt.Errorf("failed to generate token: %s", err))
|
|
return
|
|
}
|
|
|
|
// Return the JWT token to the client.
|
|
err = a.ctx.JSON(&LoginResponse{Token: token})
|
|
if err != nil {
|
|
panic(fmt.Errorf("Error with login attempt: %s \n", err))
|
|
}
|
|
}
|
|
|
|
// PostChallenge handles the POST /api/auth/pubkey-challenge request to generate a challenge for a user's public key.
|
|
func (a *AuthService) PostPubkeyChallenge() {
|
|
var r LoginRequest
|
|
|
|
// Read the login request from the client.
|
|
if err := a.ctx.ReadJSON(&r); err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
// Retrieve the account for the given email.
|
|
account := model.Account{}
|
|
if err := db.Get().Where("email = ?", r.Email).First(&account).Error; err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errors.New("invalid email or password"))
|
|
return
|
|
}
|
|
|
|
// Generate a random challenge string.
|
|
challenge, err := generateAndSaveChallengeToken(account.ID, time.Minute)
|
|
if err != nil {
|
|
a.ctx.StopWithError(iris.StatusInternalServerError, errors.New("failed to generate challenge"))
|
|
return
|
|
}
|
|
|
|
// Return the challenge to the client.
|
|
err = a.ctx.JSON(&ChallengeResponse{Challenge: challenge})
|
|
if err != nil {
|
|
panic(fmt.Errorf("Error with challenge request: %s \n", err))
|
|
}
|
|
}
|
|
|
|
// PostKeyLogin handles the POST /api/auth/pubkey-login request to authenticate a user using a public key challenge and return a JWT token.
|
|
func (a *AuthService) PostPubkeyLogin() {
|
|
var r PubkeyLoginRequest
|
|
|
|
// Read the key login request from the client.
|
|
if err := a.ctx.ReadJSON(&r); err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
// Retrieve the key challenge for the given challenge.
|
|
challenge := model.KeyChallenge{}
|
|
if err := db.Get().Where("challenge = ?", r.Challenge).Preload("Key").First(&challenge).Error; err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errorx.RejectedOperation.New("invalid key challenge"))
|
|
return
|
|
}
|
|
|
|
verifiedToken, err := jwt.Verify(jwt.HS256, sharedKey, []byte(r.Challenge), blocklist)
|
|
if err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errorx.RejectedOperation.New("invalid key challenge"))
|
|
return
|
|
}
|
|
|
|
rawPubKey, err := hex.DecodeString(r.Pubkey)
|
|
if err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errorx.RejectedOperation.New("invalid pubkey"))
|
|
return
|
|
}
|
|
|
|
rawSignature, err := hex.DecodeString(r.Signature)
|
|
if err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errorx.RejectedOperation.New("invalid signature"))
|
|
return
|
|
}
|
|
|
|
publicKeyDecoded := ed25519.PublicKey(rawPubKey)
|
|
|
|
// Verify the challenge signature.
|
|
if !ed25519.Verify(publicKeyDecoded, []byte(r.Challenge), rawSignature) {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errorx.RejectedOperation.New("invalid challenge"))
|
|
}
|
|
|
|
// Generate a JWT token for the authenticated user.
|
|
token, err := generateAndSaveLoginToken(challenge.Account.ID, 24*time.Hour)
|
|
if err != nil {
|
|
a.ctx.StopWithError(iris.StatusInternalServerError, errorx.RejectedOperation.Wrap(err, "failed to generate token"))
|
|
return
|
|
}
|
|
|
|
err = blocklist.InvalidateToken(verifiedToken.Token, verifiedToken.StandardClaims)
|
|
if err != nil {
|
|
a.ctx.StopWithError(iris.StatusInternalServerError, errorx.RejectedOperation.Wrap(err, "failed to invalidate token"))
|
|
return
|
|
}
|
|
|
|
if err := db.Get().Delete(&challenge).Error; err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errorx.RejectedOperation.New("failed to delete key challenge"))
|
|
return
|
|
}
|
|
|
|
// Return the JWT token to the client.
|
|
err = a.ctx.JSON(&LoginResponse{Token: token})
|
|
if err != nil {
|
|
panic(fmt.Errorf("Error with login attempt: %s \n", err))
|
|
}
|
|
|
|
}
|
|
|
|
// PostLogout handles the POST /api/auth/logout request to invalidate a JWT token.
|
|
func (a *AuthService) PostLogout() {
|
|
var r LogoutRequest
|
|
|
|
// Read the logout request from the client.
|
|
if err := a.ctx.ReadJSON(&r); err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
// Verify the provided token.
|
|
claims, err := jwt.Verify(jwt.HS256, sharedKey, []byte(r.Token), blocklist)
|
|
if err != nil {
|
|
a.ctx.StopWithError(iris.StatusBadRequest, errors.New("invalid token"))
|
|
return
|
|
}
|
|
|
|
err = blocklist.InvalidateToken(claims.Token, claims.StandardClaims)
|
|
if err != nil {
|
|
panic(fmt.Errorf("Error with logout: %s \n", err))
|
|
}
|
|
|
|
// Return a success response to the client.
|
|
a.ctx.StatusCode(iris.StatusNoContent)
|
|
}
|