Leahnaya/wfc-server
1
0
Fork 0
mirror of https://github.com/WiiLink24/wfc-server.git synced 2026-07-13 07:01:02 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #23 from MikeIsAStar/prevent-invalid-packets-from-stopping-server

Browse Source 
QR2: Prevent invalid packets from stopping the server
This commit is contained in:
Sketch 2023-12-09 22:40:29 -05:00 committed by GitHub
commit ea8af08097
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
qr2/main.go
View File

@ -54,16 +54,19 @@ func StartServer() {
}
func handleConnection(conn net.PacketConn, addr net.Addr, buffer []byte) {
if buffer[0] == AvailableRequest {
logging.Notice("QR2", "Command:", aurora.Yellow("AVAILABLE"))
conn.WriteTo(createResponseHeader(AvailableRequest, 0), addr)
return
}
packetType := buffer[0]
sessionId := binary.BigEndian.Uint32(buffer[1:5])
session, ok := sessions[sessionId]
moduleName := "QR2:" + strconv.FormatInt(int64(sessionId), 10)
switch buffer[0] {
if packetType != HeartbeatRequest && packetType != AvailableRequest {
if !ok {
logging.Error(moduleName, "Invalid session")
return
}
}
switch packetType {
case QueryRequest:
logging.Notice(moduleName, "Command:", aurora.Yellow("QUERY"))
break
@ -72,9 +75,9 @@ func handleConnection(conn net.PacketConn, addr net.Addr, buffer []byte) {
logging.Notice(moduleName, "Command:", aurora.Yellow("CHALLENGE"))
mutex.Lock()
if sessions[sessionId].Challenge != "" {
if session.Challenge != "" {
// TODO: Verify the challenge
sessions[sessionId].Authenticated = true
session.Authenticated = true
mutex.Unlock()
conn.WriteTo(createResponseHeader(ClientRegisteredReply, sessionId), addr)
@ -110,13 +113,14 @@ func handleConnection(conn net.PacketConn, addr net.Addr, buffer []byte) {
case KeepAliveRequest:
logging.Notice(moduleName, "Command:", aurora.Yellow("KEEPALIVE"))
sessionId := binary.BigEndian.Uint32(buffer[1:5])
mutex.Lock()
sessions[sessionId].LastKeepAlive = time.Now().Unix()
session.LastKeepAlive = time.Now().Unix()
mutex.Unlock()
return
case AvailableRequest:
logging.Notice("QR2", "Command:", aurora.Yellow("AVAILABLE"))
conn.WriteTo(createResponseHeader(AvailableRequest, 0), addr)
return
case ClientRegisteredReply: