From: AeonLucid <aeonlucid@gmail.com>
Date: Sun, 20 Sep 2020 22:45:16 +0000 (+0200)
Subject: Kick/ban proper messages
X-Git-Tag: v1.0.0~11
X-Git-Url: https://git.deb.at/?a=commitdiff_plain;h=f5dc13f775ada1854e6bb4afb028ace76dab720d;p=rhonda%2Fimpostor.git

Kick/ban proper messages
---

diff --git a/src/AmongUs.Server/Net/Game.cs b/src/AmongUs.Server/Net/Game.cs
index 6fdfee7..97f9624 100644
--- a/src/AmongUs.Server/Net/Game.cs
+++ b/src/AmongUs.Server/Net/Game.cs
@@ -160,7 +160,7 @@ namespace AmongUs.Server.Net
 
             using (var packet = MessageWriter.Get(SendOption.Reliable))
             {
-                WriteRemovePlayerMessage(packet, false, playerId, reason);
+                WriteRemovePlayerMessage(packet, false, playerId, (DisconnectReason) reason);
                 SendToAllExcept(packet, player);
             }
         }
@@ -177,7 +177,9 @@ namespace AmongUs.Server.Net
                     player.Game = null;
                 }
                 
-                WriteRemovePlayerMessage(message, true, playerId, 0);
+                WriteRemovePlayerMessage(message, true, playerId, isBan 
+                    ? DisconnectReason.Banned 
+                    : DisconnectReason.Kicked);
                 SendToAllExcept(message, player);
             }
         }
@@ -264,8 +266,11 @@ namespace AmongUs.Server.Net
             }
         }
 
-        private void WriteRemovePlayerMessage(MessageWriter message, bool clear, int playerId, byte reason)
+        private void WriteRemovePlayerMessage(MessageWriter message, bool clear, int playerId, DisconnectReason reason)
         {
+            // Only a subset of DisconnectReason shows an unique message.
+            // ExitGame, Banned and Kicked.
+            
             if (clear)
             {
                 message.Clear(SendOption.Reliable);
@@ -275,7 +280,7 @@ namespace AmongUs.Server.Net
             message.Write(Code);
             message.Write(playerId);
             message.Write(HostId);
-            message.Write(reason);
+            message.Write((byte) reason);
             message.EndMessage();
         }