SessionControllerFactory.cs

Go to the documentation of this file.

using Byond.TopicSender;
using Microsoft.Extensions.Logging;
using System;
using System.Globalization;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Tgstation.Server.Api;
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Internal;
using Tgstation.Server.Host.Components.Byond;
using Tgstation.Server.Host.Components.Chat;
using Tgstation.Server.Host.Components.Deployment;
using Tgstation.Server.Host.Components.Interop;
using Tgstation.Server.Host.Components.Interop.Bridge;
using Tgstation.Server.Host.Core;
using Tgstation.Server.Host.IO;
using Tgstation.Server.Host.Jobs;
using Tgstation.Server.Host.Security;
using Tgstation.Server.Host.System;

namespace Tgstation.Server.Host.Components.Session
{
    sealed class SessionControllerFactory : ISessionControllerFactory
    {
        readonly IProcessExecutor processExecutor;

        readonly IByondManager byond;

        readonly ITopicClient byondTopicSender;

        readonly ICryptographySuite cryptographySuite;

        readonly IAssemblyInformationProvider assemblyInformationProvider;

        readonly IIOManager ioManager;

        readonly IChatManager chat;

        readonly INetworkPromptReaper networkPromptReaper;

        readonly IPlatformIdentifier platformIdentifier;

        readonly IBridgeRegistrar bridgeRegistrar;

        readonly IServerPortProvider serverPortProvider;

        readonly ILoggerFactory loggerFactory;

        readonly ILogger<SessionControllerFactory> logger;

        readonly Api.Models.Instance instance;

        static string SecurityWord(DreamDaemonSecurity securityLevel)
        {
            return securityLevel switch
            {
                DreamDaemonSecurity.Safe => "safe",
                DreamDaemonSecurity.Trusted => "trusted",
                DreamDaemonSecurity.Ultrasafe => "ultrasafe",
                _ => throw new ArgumentOutOfRangeException(nameof(securityLevel), securityLevel, String.Format(CultureInfo.InvariantCulture, "Bad DreamDaemon security level: {0}", securityLevel)),
            };
        }

        static void PortBindTest(ushort port)
        {
            using var socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);

            try
            {
                socket.Bind(new IPEndPoint(IPAddress.Any, port));
            }
            catch (Exception ex)
            {
                throw new JobException(ErrorCode.DreamDaemonPortInUse, ex);
            }
        }

        public SessionControllerFactory(
            IProcessExecutor processExecutor,
            IByondManager byond,
            ITopicClient byondTopicSender,
            ICryptographySuite cryptographySuite,
            IAssemblyInformationProvider assemblyInformationProvider,
            IIOManager ioManager,
            IChatManager chat,
            INetworkPromptReaper networkPromptReaper,
            IPlatformIdentifier platformIdentifier,
            IBridgeRegistrar bridgeRegistrar,
            IServerPortProvider serverPortProvider,
            ILoggerFactory loggerFactory,
            ILogger<SessionControllerFactory> logger,
            Api.Models.Instance instance)
        {
            this.processExecutor = processExecutor ?? throw new ArgumentNullException(nameof(processExecutor));
            this.byond = byond ?? throw new ArgumentNullException(nameof(byond));
            this.byondTopicSender = byondTopicSender ?? throw new ArgumentNullException(nameof(byondTopicSender));
            this.cryptographySuite = cryptographySuite ?? throw new ArgumentNullException(nameof(cryptographySuite));
            this.assemblyInformationProvider = assemblyInformationProvider ?? throw new ArgumentNullException(nameof(assemblyInformationProvider));
            this.instance = instance ?? throw new ArgumentNullException(nameof(instance));
            this.ioManager = ioManager ?? throw new ArgumentNullException(nameof(ioManager));
            this.chat = chat ?? throw new ArgumentNullException(nameof(chat));
            this.networkPromptReaper = networkPromptReaper ?? throw new ArgumentNullException(nameof(networkPromptReaper));
            this.platformIdentifier = platformIdentifier ?? throw new ArgumentNullException(nameof(platformIdentifier));
            this.bridgeRegistrar = bridgeRegistrar ?? throw new ArgumentNullException(nameof(bridgeRegistrar));
            this.serverPortProvider = serverPortProvider ?? throw new ArgumentNullException(nameof(serverPortProvider));
            this.loggerFactory = loggerFactory ?? throw new ArgumentNullException(nameof(loggerFactory));
            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
        }

        #pragma warning disable CA1506 // TODO: Decomplexify
        public async Task<ISessionController> LaunchNew(
            IDmbProvider dmbProvider,
            IByondExecutableLock currentByondLock,
            DreamDaemonLaunchParameters launchParameters,
            bool primaryPort,
            bool primaryDirectory,
            bool apiValidate,
            CancellationToken cancellationToken)
        {
            var portToUse = primaryPort ? launchParameters.PrimaryPort : launchParameters.SecondaryPort;
            if (!portToUse.HasValue)
                throw new InvalidOperationException("Given port is null!");

            var basePath = primaryDirectory ? dmbProvider.PrimaryDirectory : dmbProvider.SecondaryDirectory;

            switch (dmbProvider.CompileJob.MinimumSecurityLevel)
            {
                case DreamDaemonSecurity.Ultrasafe:
                    break;
                case DreamDaemonSecurity.Safe:
                    if (launchParameters.SecurityLevel == DreamDaemonSecurity.Ultrasafe)
                        launchParameters.SecurityLevel = DreamDaemonSecurity.Safe;
                    break;
                case DreamDaemonSecurity.Trusted:
                    launchParameters.SecurityLevel = DreamDaemonSecurity.Trusted;
                    break;
                default:
                    throw new InvalidOperationException(String.Format(CultureInfo.InvariantCulture, "Invalid DreamDaemonSecurity value: {0}", dmbProvider.CompileJob.MinimumSecurityLevel));
            }

            var chatTrackingContext = chat.CreateTrackingContext();
            try
            {
                // get the byond lock
                var byondLock = currentByondLock ?? await byond.UseExecutables(Version.Parse(dmbProvider.CompileJob.ByondVersion), cancellationToken).ConfigureAwait(false);
                try
                {
                    if (launchParameters.SecurityLevel == DreamDaemonSecurity.Trusted)
                        await byondLock.TrustDmbPath(ioManager.ConcatPath(basePath, dmbProvider.DmbName), cancellationToken).ConfigureAwait(false);

                    PortBindTest(portToUse.Value);
                    await CheckPagerIsNotRunning(cancellationToken).ConfigureAwait(false);

                    var accessIdentifier = cryptographySuite.GetSecureString();

                    // set command line options
                    // more sanitization here cause it uses the same scheme
                    var parameters = $"{DMApiConstants.ParamApiVersion}={byondTopicSender.SanitizeString(DMApiConstants.Version.Semver().ToString())}&{byondTopicSender.SanitizeString(DMApiConstants.ParamServerPort)}={serverPortProvider.HttpApiPort}&{byondTopicSender.SanitizeString(DMApiConstants.ParamAccessIdentifier)}={byondTopicSender.SanitizeString(accessIdentifier)}";

                    var visibility = apiValidate ? "invisible" : "public";

                    // important to run on all ports to allow port changing
                    Guid? logFileGuid = null;
                    var arguments = String.Format(
                        CultureInfo.InvariantCulture,
                        "{0} -port {1} -ports 1-65535 {2}-close -{3} -{4}{5} -public -params \"{6}\"",
                        dmbProvider.DmbName,
                        portToUse,
                        launchParameters.AllowWebClient.Value ? "-webclient " : String.Empty,
                        SecurityWord(launchParameters.SecurityLevel.Value),
                        visibility,
                        platformIdentifier.IsWindows
                            ? $" -log {logFileGuid = Guid.NewGuid()}"
                            : String.Empty, // Just use stdout on linux
                        parameters);

                    // See https://github.com/tgstation/tgstation-server/issues/719
                    var noShellExecute = !platformIdentifier.IsWindows;

                    // launch dd
                    var process = processExecutor.LaunchProcess(
                        byondLock.DreamDaemonPath,
                        basePath,
                        arguments,
                        noShellExecute,
                        noShellExecute,
                        noShellExecute: noShellExecute);

                    async Task<string> GetDDOutput()
                    {
                        if (!platformIdentifier.IsWindows)
                            return process.GetCombinedOutput();

                        var logFilePath = ioManager.ConcatPath(basePath, logFileGuid.ToString());
                        try
                        {
                            var dreamDaemonLogBytes = await ioManager.ReadAllBytes(
                                logFilePath,
                                default)
                                .ConfigureAwait(false);

                            return Encoding.UTF8.GetString(dreamDaemonLogBytes);
                        }
                        finally
                        {
                            try
                            {
                                await ioManager.DeleteFile(logFilePath, default).ConfigureAwait(false);
                            }
                            catch (Exception ex)
                            {
                                logger.LogWarning("Failed to delete DreamDaemon log file {0}: {1}", logFilePath, ex);
                            }
                        }
                    }

                    // Log DD output
                    _ = process.Lifetime.ContinueWith(
                            async x =>
                            {
                                try
                                {
                                    var ddOutput = await GetDDOutput().ConfigureAwait(false);
                                    logger.LogTrace(
                                        "DreamDaemon Output:{0}{1}",
                                        Environment.NewLine, ddOutput);
                                }
                                catch (Exception ex)
                                {
                                    logger.LogWarning("Error reading DreamDaemon output: {0}", ex);
                                }
                            },
                            TaskScheduler.Current);

                    try
                    {
                        networkPromptReaper.RegisterProcess(process);

                        var runtimeInformation = CreateRuntimeInformation(
                            dmbProvider,
                            chatTrackingContext,
                            launchParameters.SecurityLevel.Value,
                            apiValidate);

                        var reattachInformation = new ReattachInformation(
                            dmbProvider,
                            process,
                            runtimeInformation,
                            accessIdentifier,
                            portToUse.Value,
                            primaryDirectory);

                        var sessionController = new SessionController(
                            reattachInformation,
                            instance,
                            process,
                            byondLock,
                            byondTopicSender,
                            chatTrackingContext,
                            bridgeRegistrar,
                            chat,
                            assemblyInformationProvider,
                            loggerFactory.CreateLogger<SessionController>(),
                            launchParameters.StartupTimeout,
                            false);

                        return sessionController;
                    }
                    catch
                    {
                        process.Terminate();
                        process.Dispose();
                        throw;
                    }
                }
                catch
                {
                    if (currentByondLock == null)
                        byondLock.Dispose();
                    throw;
                }
            }
            catch
            {
                chatTrackingContext.Dispose();
                throw;
            }
        }
        #pragma warning restore CA1506

        public async Task<ISessionController> Reattach(
            ReattachInformation reattachInformation,
            CancellationToken cancellationToken)
        {
            if (reattachInformation == null)
                throw new ArgumentNullException(nameof(reattachInformation));

            var chatTrackingContext = chat.CreateTrackingContext();
            try
            {
                var byondLock = await byond.UseExecutables(Version.Parse(reattachInformation.Dmb.CompileJob.ByondVersion), cancellationToken).ConfigureAwait(false);
                try
                {
                    var process = processExecutor.GetProcess(reattachInformation.ProcessId);
                    if (process == null)
                        return null;

                    try
                    {
                        networkPromptReaper.RegisterProcess(process);
                        var runtimeInformation = CreateRuntimeInformation(
                            reattachInformation.Dmb,
                            chatTrackingContext,
                            null,
                            false);
                        reattachInformation.SetRuntimeInformation(runtimeInformation);

                        var controller = new SessionController(
                            reattachInformation,
                            instance,
                            process,
                            byondLock,
                            byondTopicSender,
                            chatTrackingContext,
                            bridgeRegistrar,
                            chat,
                            assemblyInformationProvider,
                            loggerFactory.CreateLogger<SessionController>(),
                            null,
                            true);

                        process = null;
                        byondLock = null;
                        chatTrackingContext = null;

                        return controller;
                    }
                    finally
                    {
                        process?.Dispose();
                    }
                }
                finally
                {
                    byondLock?.Dispose();
                }
            }
            finally
            {
                chatTrackingContext?.Dispose();
            }
        }

        public ISessionController CreateDeadSession(IDmbProvider dmbProvider) => new DeadSessionController(dmbProvider);

        RuntimeInformation CreateRuntimeInformation(
            IDmbProvider dmbProvider,
            IChatTrackingContext chatTrackingContext,
            DreamDaemonSecurity? securityLevel,
            bool apiValidateOnly)
        {
            var revisionInfo = new Api.Models.Internal.RevisionInformation
            {
                CommitSha = dmbProvider.CompileJob.RevisionInformation.CommitSha,
                OriginCommitSha = dmbProvider.CompileJob.RevisionInformation.OriginCommitSha
            };

            var testMerges = dmbProvider
                .CompileJob
                .RevisionInformation
                .ActiveTestMerges?
                .Select(x => x.TestMerge)
                .Select(x => new TestMergeInformation(x, revisionInfo))
                ?? Enumerable.Empty<TestMergeInformation>();

            return new RuntimeInformation(
                assemblyInformationProvider,
                serverPortProvider,
                testMerges,
                chatTrackingContext.Channels,
                instance,
                revisionInfo,
                securityLevel,
                apiValidateOnly);
        }

        async Task CheckPagerIsNotRunning(CancellationToken cancellationToken)
        {
            if (!platformIdentifier.IsWindows)
                return;

            using var otherProcess = processExecutor.GetProcessByName("byond");
            if (otherProcess == null)
                return;

            var otherUsernameTask = otherProcess.GetExecutingUsername(cancellationToken);
            using var ourProcess = processExecutor.GetCurrentProcess();
            var ourUserName = await ourProcess.GetExecutingUsername(cancellationToken).ConfigureAwait(false);
            var otherUserName = await otherUsernameTask.ConfigureAwait(false);

            if(otherUserName.Equals(ourUserName, StringComparison.Ordinal))
                throw new JobException(ErrorCode.DeploymentPagerRunning);
        }
    }
}