Leahnaya/wfc-server
1
0
Fork 0
mirror of https://github.com/WiiLink24/wfc-server.git synced 2026-09-25 00:35:32 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

GPCM: Return error

Browse Source
This commit is contained in:
Sketch
2023-11-11 13:15:55 -05:00
parent 55c1299ada
commit 91f3d0b06c
1 changed files with 29 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
gpcm/login.go
View File

@@ -4,11 +4,13 @@ import (
"crypto/md5"
"encoding/base64"
"encoding/hex"
"fmt"
"log"
"strconv"
"strings"
"wwfc/common"
"wwfc/database"
"wwfc/logging"
)
func generateResponse(gpcmChallenge, nasChallenge, authToken, clientChallenge string) string {
@@ -38,35 +40,37 @@ func (g *GameSpySession) login(command common.GameSpyCommand) {
authToken := command.OtherValues["authtoken"]
challenge := database.GetChallenge(pool, ctx, authToken)
if challenge == "" {
// TODO: Error out
log.Fatalf("Invalid auth token")
err := g.FormatError("invalid auth token", 256, command.OtherValues["id"])
g.Conn.Write([]byte(err))
return
}
response := generateResponse(g.Challenge, challenge, authToken, command.OtherValues["challenge"])
if response != command.OtherValues["response"] {
// TODO: Return an error
log.Fatalf("response mismatch")
err := g.FormatError("response mismatch", 256, command.OtherValues["id"])
g.Conn.Write([]byte(err))
return
}
proof := generateProof(g.Challenge, challenge, command.OtherValues["authtoken"], command.OtherValues["challenge"])
// Perform the login with the database.
// TODO: Check valid result
user, ok := database.LoginUserToGPCM(pool, ctx, authToken)
if !ok {
// TODO: Return an error
log.Fatalf("GPCM login error")
err := g.FormatError("GPCM login error", 256, command.OtherValues["id"])
g.Conn.Write([]byte(err))
return
}
g.User = user
// Check to see if a session is already open with this profile ID
// TODO: Test this
mutex.Lock()
_, exists := sessions[g.User.ProfileId]
if exists {
mutex.Unlock()
// TODO: Return an error
log.Fatalf("Session with profile ID %d is already open", g.User.ProfileId)
err := g.FormatError(fmt.Sprintf("Session with profile ID %d is already open", g.User.ProfileId), 256, command.OtherValues["id"])
g.Conn.Write([]byte(err))
return
}
sessions[g.User.ProfileId] = g
mutex.Unlock()
@@ -96,6 +100,21 @@ func (g *GameSpySession) login(command common.GameSpyCommand) {
g.Conn.Write([]byte(payload))
}
func (g *GameSpySession) FormatError(err string, code int, id string) string {
logging.Error(g.ModuleName, err)
return common.CreateGameSpyMessage(common.GameSpyCommand{
Command: "error",
CommandValue: "",
OtherValues: map[string]string{
"err": strconv.Itoa(code),
"fatal": "",
"errmsg": err,
"id": id,
},
})
}
func IsLoggedIn(profileID uint32) bool {
mutex.Lock()
defer mutex.Unlock()