Add QR2 kick order to ensure the peer is disconnected from the game session

Author: EpicUsername12

gpcm/kick.go

@@ -1,6 +1,47 @@
 package gpcm
 
-import "wwfc/common"
+import (
+	"strconv"
+	"time"
+	"wwfc/common"
+	"wwfc/qr2"
+)
+
+// Expects a global mutex to be locked
+// This function will return all sessions that are matched with the player `profileID`
+func getSessionsMatchedWithPlayer(gameName string, profileID uint32) []*GameSpySession {
+
+	var matchedPlayers []*GameSpySession
+
+	// Check if the user is part of a group
+	var foundGroup *qr2.GroupInfo
+	groups := qr2.GetGroups([]string{gameName}, []string{}, false)
+	for _, group := range groups {
+		for _, playerInfo := range group.Players {
+			pid, err := strconv.ParseUint(playerInfo.ProfileID, 10, 32)
+			if err == nil {
+				if uint32(pid) == profileID {
+					foundGroup = &group
+					break
+				}
+			}
+		}
+	}
+
+	// If the user is part of a group, send a kick order to all players in the group
+	if foundGroup != nil {
+		for _, playerInfo := range foundGroup.Players {
+			pid, err := strconv.ParseUint(playerInfo.ProfileID, 10, 32)
+			if err == nil {
+				if session, exists := sessions[uint32(pid)]; exists {
+					matchedPlayers = append(matchedPlayers, session)
+				}
+			}
+		}
+	}
+
+	return matchedPlayers
+}
 
 func kickPlayer(profileID uint32, reason string) {
 	if session, exists := sessions[profileID]; exists {
@@ -31,12 +72,23 @@ func kickPlayer(profileID uint32, reason string) {
 			return
 		}
 
+		players := getSessionsMatchedWithPlayer(session.GameName, profileID)
 		session.replyError(GPError{
 			ErrorCode:   ErrConnectionClosed.ErrorCode,
 			ErrorString: "The player was kicked from the server. Reason: " + reason,
 			Fatal:       true,
 			WWFCMessage: errorMessage,
 		})
+
+		// After 3 seconds, send kick order to all players in the group
+		// This is to prevent the restricted player from staying in the group if he ignores the GPCM kick
+		go func(players []*GameSpySession) {
+			time.AfterFunc(3*time.Second, func() {
+				for _, player := range players {
+					qr2.OrderKickFromGroup(player.User.ProfileId, profileID)
+				}
+			})
+		}(players)
 	}
 }
 
@@ -52,12 +104,23 @@ func KickPlayerCustomMessage(profileID uint32, reason string, message WWFCErrorM
 	defer mutex.Unlock()
 
 	if session, exists := sessions[profileID]; exists {
+		players := getSessionsMatchedWithPlayer(session.GameName, profileID)
 		session.replyError(GPError{
 			ErrorCode:   ErrConnectionClosed.ErrorCode,
 			ErrorString: "The player was kicked from the server. Reason: " + reason,
 			Fatal:       true,
 			WWFCMessage: message,
 			Reason:      reason,
 		})
+
+		// After 3 seconds, send kick order to all players in the group
+		// This is to prevent the restricted player from staying in the group if he ignores the GPCM kick
+		go func(players []*GameSpySession) {
+			time.AfterFunc(3*time.Second, func() {
+				for _, player := range players {
+					qr2.OrderKickFromGroup(player.User.ProfileId, profileID)
+				}
+			})
+		}(players)
 	}
 }

qr2/group.go

@@ -583,3 +583,29 @@ func loadGroups() error {
 
 	return nil
 }
+
+func OrderKickFromGroup(profileID uint32, profileToKick uint32) {
+	moduleName := "QR2:OrderKickFromGroup/" + strconv.FormatUint(uint64(profileID), 10)
+
+	login, exists := logins[profileID]
+	if !exists || login == nil {
+		return
+	}
+
+	session := login.session
+	if session == nil {
+		return
+	}
+
+	mutex.Lock()
+	defer mutex.Unlock()
+
+	message := createResponseHeader(ClientKickPeerOrder, session.SessionID)
+	message = append(message, 0) // padding for 4 byte alignment
+	message = binary.BigEndian.AppendUint32(message, profileToKick)
+	_, err := masterConn.WriteTo(message, &session.Addr)
+	if err != nil {
+		logging.Error(moduleName, "Error sending message:", err.Error())
+	}
+
+}

qr2/main.go

@@ -24,7 +24,8 @@ const (
 	AvailableRequest        = 0x09
 	ClientRegisteredReply   = 0x0A
 
-	ClientExploitReply = 0x10
+	ClientExploitReply  = 0x10
+	ClientKickPeerOrder = 0x11
 )
 
 var (