Leahnaya/wfc-server
1
0
Fork 0
mirror of https://github.com/WiiLink24/wfc-server.git synced 2026-03-21 17:44:58 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Database: Always create new auth token

Browse Source
This commit is contained in:
mkwcat 2023-11-05 15:06:34 -05:00
parent 18620a9220
commit 90f706c0b6
No known key found for this signature in database
GPG Key ID: 7A505679CE9E7AA9
1 changed files with 9 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
database/login.go
View File

@ -20,34 +20,29 @@ const (
)
func GenerateAuthToken(pool *pgxpool.Pool, ctx context.Context, userId int64, gsbrcd string) string {
var authToken string
err := pool.QueryRow(ctx, GetUserAuthToken, userId, gsbrcd).Scan(&authToken)
if err == nil {
// Temporary(?) workaround for multiple sessions with the same user ID (i.e. multiple Dolphin instances).
// Just don't change the user's auth token... ever.
// TODO: What do we actually do here? Do we even care about proper authentication at this stage?
return authToken
var userExists bool
err := pool.QueryRow(ctx, DoesNASUserExist, userId, gsbrcd).Scan(&userExists)
if err != nil {
panic(err)
}
exists := false
authToken = "NDS" + common.RandomString(80)
authToken := "NDS" + common.RandomString(80)
for {
// We must make sure that the auth token doesn't exist before attempting to insert it into the database.
var exists bool
err := pool.QueryRow(ctx, DoesAuthTokenExist, authToken).Scan(&exists)
var tokenExists bool
err := pool.QueryRow(ctx, DoesAuthTokenExist, authToken).Scan(&tokenExists)
if err != nil {
panic(err)
}
if !exists {
if !tokenExists {
break
}
authToken = "NDS" + common.RandomString(80)
}
if exists {
if userExists {
// UPDATE rather than INSERT
_, err = pool.Exec(ctx, UpdateUserLogin, authToken, userId, gsbrcd)
if err != nil {