blah

RAGECOOP · Aug 18, 2022 · c4312fa · c4312fa
1 parent 7566423
commit c4312fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/RageCoop.Client/Networking/Networking.cs b/RageCoop.Client/Networking/Networking.cs
@@ -21,6 +21,7 @@ internal static partial class Networking
         internal static readonly Dictionary<PacketType, Func<byte[], Packet>> RequestHandlers = new Dictionary<PacketType, Func<byte[], Packet>>();
         internal static float SimulatedLatency=0;
         public static bool IsConnecting { get; private set; }
+        public static IPEndPoint _targetServerEP;
         static Networking()
         {
             Security=new Security(Main.Logger);
@@ -77,7 +78,7 @@ public static void ToggleConnection(string address, string username = null, stri
 
                 PlayerList.Cleanup();
                 EntityPool.AddPlayer();
-                if (publicKey==null && !string.IsNullOrEmpty(password) && !Menus.CoopMenu.ShowPopUp("WARNING", "WARNING", "Server's IP can be spoofed when using direct connection, do you wish to continue?", "", true))
+                if (publicKey==null && !string.IsNullOrEmpty(password) && !Menus.CoopMenu.ShowPopUp("", "WARNING", "Server's IP can be spoofed when using direct connection, do you wish to continue?", "", true))
                 {
                     IsConnecting=false;
                     return;
@@ -86,6 +87,7 @@ public static void ToggleConnection(string address, string username = null, stri
                 {
                     try
                     {
+                        _targetServerEP=CoreUtils.StringToEndPoint(address);
                         DownloadManager.Cleanup();
                         Peer = new CoopPeer(config);
                         Peer.OnMessageReceived+= (s, m) =>

diff --git a/RageCoop.Client/Networking/Receive.cs b/RageCoop.Client/Networking/Receive.cs
@@ -197,7 +197,7 @@ public static void ProcessMessage(NetIncomingMessage message)
                                 }
                             case PacketType.PublicKeyResponse:
                                 { 
-                                    if(Security.ServerRSA!=null){break;}
+                                    if(message.SenderEndPoint.ToString()!=_targetServerEP.ToString() ||!IsConnecting){break;}
                                     var packet = data.GetPacket<Packets.PublicKeyResponse>();
                                     Security.SetServerPublicKey(packet.Modulus, packet.Exponent);
                                     _publicKeyReceived.Set();