RAGECOOP-V/Client/Scripts/Networking/Networking.cs

using System;
using System.Collections.Generic;
using System.Net;
using System.Security.Cryptography;
using System.Threading;
using System.Threading.Tasks;
using GTA.UI;
using Lidgren.Network;
using RageCoop.Client.Menus;
using RageCoop.Client.Scripting;
using RageCoop.Core;

namespace RageCoop.Client
{
    internal static partial class Networking
    {
        public static IPEndPoint _targetServerEP;

        public static CoopPeer Peer;
        public static bool ShowNetworkInfo = false;
        public static Security Security = new();
        public static NetConnection ServerConnection;

        private static readonly Dictionary<int, Action<PacketType, NetIncomingMessage>> PendingResponses = new();

        internal static readonly Dictionary<PacketType, Func<NetIncomingMessage, Packet>> RequestHandlers = new();

        internal static float SimulatedLatency = 0;

        public static float Latency => ServerConnection.AverageRoundtripTime / 2;
        public static bool IsConnecting { get; private set; }
        public static bool IsOnServer => ServerConnection?.Status == NetConnectionStatus.Connected;

        public static void ToggleConnection(string address, string username = null, string password = null,
            PublicKey publicKey = null)
        {
            CoopMenu.Menu.Visible = false;

            if (IsConnecting)
            {
                _publicKeyReceived.Set();
                IsConnecting = false;
                API.QueueAction(() =>
                Notification.Show("Connection has been canceled"));

                Peer.Shutdown("bye");
            }
            else if (IsOnServer)
            {
                Peer.Shutdown("bye");
            }
            else 
            {
                IsConnecting = true;
                password ??= Settings.Password;
                username ??= Settings.Username;

                // 623c92c287cc392406e7aaaac1c0f3b0 = RAGECOOP
                var config = new NetPeerConfiguration("623c92c287cc392406e7aaaac1c0f3b0")
                {
                    AutoFlushSendQueue = false,
                    AcceptIncomingConnections = true,
                    MaximumConnections = 32,
                    PingInterval = 5
                };
#if DEBUG
                config.SimulatedMinimumLatency = SimulatedLatency;
                config.SimulatedRandomLatency = 0;
#endif

                config.EnableMessageType(NetIncomingMessageType.UnconnectedData);
                config.EnableMessageType(NetIncomingMessageType.NatIntroductionSuccess);

                var ip = new string[2];

                var idx = address.LastIndexOf(':');
                if (idx != -1)
                {
                    ip[0] = address.Substring(0, idx);
                    ip[1] = address.Substring(idx + 1);
                }

                if (ip.Length != 2) throw new Exception("Malformed URL");

                PlayerList.Cleanup();
                EntityPool.AddPlayer();
                if (publicKey == null && !string.IsNullOrEmpty(password) && !CoopMenu.ShowPopUp("", "WARNING",
                        "Server's IP can be spoofed when using direct connection, do you wish to continue?", "", true))
                {
                    IsConnecting = false;
                    return;
                }

                ThreadManager.CreateThread(() =>
                {
                    try
                    {
                        _targetServerEP = CoreUtils.StringToEndPoint(address);

                        // Ensure static constructor invocation
                        DownloadManager.Cleanup();
                        Peer = new CoopPeer(config,Log);
                        Peer.OnMessageReceived += (s, m) =>
                        {
                            try
                            {
                                ProcessMessage(m);
                            }
                            catch (Exception ex)
                            {
                                Log.Error(ex);
                            }
                        };
                        API.QueueAction(() => { Notification.Show("~y~Trying to connect..."); });
                        CoopMenu._serverConnectItem.Enabled = false;
                        Security.Regen();
                        if (publicKey == null)
                        {
                            if (!GetServerPublicKey(ip[0], int.Parse(ip[1])))
                            {
                                CoopMenu._serverConnectItem.Enabled = true;
                                throw new TimeoutException("Failed to retrive server's public key");
                            }
                        }
                        else
                        {
                            Security.SetServerPublicKey(publicKey.Modulus, publicKey.Exponent);
                        }

                        // Send handshake packet
                        var outgoingMessage = Peer.CreateMessage();
                        var handshake = new Packets.Handshake
                        {
                            PedID = LocalPlayerID,
                            Username = username,
                            ModVersion = Main.ModVersion.ToString(),
                            PasswordEncrypted = Security.Encrypt(password.GetBytes()),
                            InternalEndPoint = new IPEndPoint(CoreUtils.GetLocalAddress(ip[0]), Peer.Port)
                        };

                        Security.GetSymmetricKeysCrypted(out handshake.AesKeyCrypted, out handshake.AesIVCrypted);
                        handshake.Pack(outgoingMessage);
                        ServerConnection = Peer.Connect(ip[0], short.Parse(ip[1]), outgoingMessage);
                    }
                    catch (Exception ex)
                    {
                        Log.Error("Cannot connect to server: ", ex);
                        API.QueueAction(() => Notification.Show("~r~Cannot connect to server: " + ex.Message));
                    }

                    IsConnecting = false;
                }, "Connect");
            }
        }

        private static int NewRequestID()
        {
            var ID = 0;
            while (ID == 0 || PendingResponses.ContainsKey(ID))
            {
                var rngBytes = new byte[4];

                RandomNumberGenerator.Create().GetBytes(rngBytes);

                // Convert the bytes into an integer
                ID = BitConverter.ToInt32(rngBytes, 0);
            }

            return ID;
        }

        #region -- PLAYER --

        private static void PlayerConnect(Packets.PlayerConnect packet)
        {
            var p = new Player
            {
                ID = packet.PedID,
                Username = packet.Username
            };
            PlayerList.SetPlayer(packet.PedID, packet.Username);

            Log.Debug($"player connected:{p.Username}");
            API.QueueAction(() =>
                Notification.Show($"~h~{p.Username}~h~ connected."));
        }

        private static void PlayerDisconnect(Packets.PlayerDisconnect packet)
        {
            var player = PlayerList.GetPlayer(packet.PedID);
            if (player == null) return;
            PlayerList.RemovePlayer(packet.PedID);
            API.QueueAction(() =>
            {
                EntityPool.RemoveAllFromPlayer(packet.PedID);
                Notification.Show($"~h~{player.Username}~h~ left.");
            });
        }

        #endregion // -- PLAYER --

        #region -- GET --

        private static bool GetServerPublicKey(string host, int port, int timeout = 10000)
        {
            Security.ServerRSA = null;
            var msg = Peer.CreateMessage();
            new Packets.PublicKeyRequest().Pack(msg);
            Peer.SendUnconnectedMessage(msg, host, port);
            return _publicKeyReceived.WaitOne(timeout) && Security.ServerRSA != null;
        }

        public static void GetResponse<T>(Packet request, Action<T> callback,
            ConnectionChannel channel = ConnectionChannel.RequestResponse) where T : Packet, new()
        {
            var received = new AutoResetEvent(false);
            var id = NewRequestID();
            PendingResponses.Add(id, (type, p) =>
            {
                var result = new T();
                result.Deserialize(p);
                callback(result);
            });
            var msg = Peer.CreateMessage();
            msg.Write((byte)PacketType.Request);
            msg.Write(id);
            request.Pack(msg);
            Peer.SendMessage(msg, ServerConnection, NetDeliveryMethod.ReliableOrdered, (int)channel);
        }

        #endregion
    }
}