feat: invalidate token on logout

server/common/auth.go

@@ -3,6 +3,7 @@ package common
 import (
 	"time"
 
+	"github.com/Xhofe/go-cache"
 	"github.com/alist-org/alist/v3/internal/conf"
 	"github.com/alist-org/alist/v3/internal/model"
 	"github.com/golang-jwt/jwt/v4"
@@ -17,6 +18,8 @@ type UserClaims struct {
 	jwt.RegisteredClaims
 }
 
+var validTokenCache = cache.NewMemCache[bool]()
+
 func GenerateToken(user *model.User) (tokenString string, err error) {
 	claim := UserClaims{
 		Username: user.Username,
@@ -28,13 +31,20 @@ func GenerateToken(user *model.User) (tokenString string, err error) {
 		}}
 	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claim)
 	tokenString, err = token.SignedString(SecretKey)
+	if err != nil {
+		return "", err
+	}
+	validTokenCache.Set(tokenString, true)
 	return tokenString, err
 }
 
 func ParseToken(tokenString string) (*UserClaims, error) {
 	token, err := jwt.ParseWithClaims(tokenString, &UserClaims{}, func(token *jwt.Token) (interface{}, error) {
 		return SecretKey, nil
 	})
+	if IsTokenInvalidated(tokenString) {
+		return nil, errors.New("token is invalidated")
+	}
 	if err != nil {
 		if ve, ok := err.(*jwt.ValidationError); ok {
 			if ve.Errors&jwt.ValidationErrorMalformed != 0 {
@@ -53,3 +63,16 @@ func ParseToken(tokenString string) (*UserClaims, error) {
 	}
 	return nil, errors.New("couldn't handle this token")
 }
+
+func InvalidateToken(tokenString string) error {
+	if tokenString == "" {
+		return nil // don't invalidate empty guest token
+	}
+	validTokenCache.Del(tokenString)
+	return nil
+}
+
+func IsTokenInvalidated(tokenString string) bool {
+	_, ok := validTokenCache.Get(tokenString)
+	return !ok
+}

server/handles/auth.go

@@ -181,3 +181,12 @@ func Verify2FA(c *gin.Context) {
 		common.SuccessResp(c)
 	}
 }
+
+func LogOut(c *gin.Context) {
+	err := common.InvalidateToken(c.GetHeader("Authorization"))
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+	} else {
+		common.SuccessResp(c)
+	}
+}

server/router.go

@@ -54,6 +54,7 @@ func Init(e *gin.Engine) {
 	auth.POST("/me/update", handles.UpdateCurrent)
 	auth.POST("/auth/2fa/generate", handles.Generate2FA)
 	auth.POST("/auth/2fa/verify", handles.Verify2FA)
+	auth.GET("/auth/logout", handles.LogOut)
 
 	// auth
 	api.GET("/auth/sso", handles.SSOLoginRedirect)