lzc256
/
AquaDX
mirror of https://github.com/hykilpikonna/AquaDX
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

[+] Allow ftk as an auth token

pull/23/head
Azalea 2024-03-23 12:22:32 -04:00
parent 54e865feb2
commit 89461893a4
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/java/icu/samnyan/aqua/sega/allnet/AllNetSecure.kt
View File

@ -1,6 +1,7 @@
package icu.samnyan.aqua.sega.allnet package icu.samnyan.aqua.sega.allnet
import ext.Str import ext.Str
import icu.samnyan.aqua.net.FrontierProps
import jakarta.servlet.http.HttpServletRequest import jakarta.servlet.http.HttpServletRequest
import jakarta.servlet.http.HttpServletRequestWrapper import jakarta.servlet.http.HttpServletRequestWrapper
import jakarta.servlet.http.HttpServletResponse import jakarta.servlet.http.HttpServletResponse
@ -25,6 +26,7 @@ import org.springframework.web.servlet.config.annotation.WebMvcConfigurer
class TokenChecker( class TokenChecker(
val keyChipRepo: KeyChipRepo, val keyChipRepo: KeyChipRepo,
val keychipSessionService: KeychipSessionService, val keychipSessionService: KeychipSessionService,
val frontierProps: FrontierProps,
) : HandlerInterceptor { ) : HandlerInterceptor {
val log = LoggerFactory.getLogger(TokenChecker::class.java) val log = LoggerFactory.getLogger(TokenChecker::class.java)
@ -41,8 +43,10 @@ class TokenChecker(
// Check whether the token exists in the database // Check whether the token exists in the database
// The token can either be a keychip id (old method) or a session id (new method) // The token can either be a keychip id (old method) or a session id (new method)
// Or the frontier token
val session = keychipSessionService.find(token) val session = keychipSessionService.find(token)
if (token.isNotBlank() && (keyChipRepo.existsByKeychipId(token) || session != null)) if (token.isNotBlank() && (keyChipRepo.existsByKeychipId(token) || session != null
|| (frontierProps.enabled && frontierProps.ftk == token)))
{ {
// Forward the request // Forward the request
val w = RewriteWrapper(req, token).apply { setAttribute("token", token) } val w = RewriteWrapper(req, token).apply { setAttribute("token", token) }