DotNetCliInvoker.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using Semmle.Util;
using Semmle.Util.Logging;

namespace Semmle.Extraction.CSharp.DependencyFetching
{
    /// <summary>
    /// Low level utilities to run the "dotnet" command.
    /// </summary>
    internal sealed class DotNetCliInvoker : IDotNetCliInvoker
    {
        private readonly ILogger logger;
        private readonly IRegistryProxy? proxy;

        public string Exec { get; }

        public DotNetCliInvoker(ILogger logger, string exec, IRegistryProxy? registryProxy)
        {
            this.logger = logger;
            this.proxy = registryProxy;
            this.Exec = exec;
            logger.LogInfo($"Using .NET CLI executable: '{Exec}'");
        }

        private ProcessStartInfo MakeDotnetStartInfo(List<string> args, string? workingDirectory)
        {
            var startInfo = new ProcessStartInfo(Exec, args)
            {
                UseShellExecute = false,
                RedirectStandardOutput = true,
                RedirectStandardError = true
            };
            if (!string.IsNullOrWhiteSpace(workingDirectory))
            {
                startInfo.WorkingDirectory = workingDirectory;
            }

            // Set minimal environment variables.
            foreach (var kvp in IDotNetCliInvoker.MinimalEnvironment)
            {
                startInfo.EnvironmentVariables[kvp.Key] = kvp.Value;
            }

            // Configure the proxy settings, if applicable.
            proxy?.SetProcessEnvironment(startInfo);

            return startInfo;
        }

        private int RunCommandExitCodeAux(List<string> args, string? workingDirectory, out IList<string> output, out string dirLog, bool silent)
        {
            dirLog = string.IsNullOrWhiteSpace(workingDirectory) ? "" : $" in {workingDirectory}";
            var pi = MakeDotnetStartInfo(args, workingDirectory);
            var threadId = Environment.CurrentManagedThreadId;
            void onOut(string s) => logger.Log(silent ? Severity.Debug : Severity.Info, s, threadId);
            void onError(string s) => logger.LogError(s, threadId);
            logger.LogInfo($"Running '{Exec} {string.Join(" ", args)}'{dirLog}");
            var exitCode = pi.ReadOutput(out output, onOut, onError);
            return exitCode;
        }

        private bool RunCommandAux(List<string> args, string? workingDirectory, out IList<string> output, bool silent)
        {
            var exitCode = RunCommandExitCodeAux(args, workingDirectory, out output, out var dirLog, silent);
            if (exitCode != 0)
            {
                logger.LogError($"Command '{Exec} {string.Join(" ", args)}'{dirLog} failed with exit code {exitCode}");
                return false;
            }
            return true;
        }

        public bool RunCommand(List<string> args, bool silent = true) =>
            RunCommandAux(args, null, out _, silent);

        public int RunCommandExitCode(List<string> args, bool silent = true) =>
            RunCommandExitCodeAux(args, null, out _, out _, silent);

        public bool RunCommand(List<string> args, out IList<string> output, bool silent = true) =>
            RunCommandAux(args, null, out output, silent);

        public bool RunCommand(List<string> args, string? workingDirectory, out IList<string> output, bool silent = true) =>
            RunCommandAux(args, workingDirectory, out output, silent);
    }
}