WindowsFirewallHelper.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Logging;
 
namespace Tgstation.Server.Host.System
{
    static class WindowsFirewallHelper
    {
        public static async ValueTask<int> AddFirewallException(
            IProcessExecutor processExecutor,
            ILogger logger,
            string exceptionName,
            string exePath,
            bool lowPriority,
            CancellationToken cancellationToken)
        {
            logger.LogInformation("Adding Windows Firewall exception for {path}...", exePath);
            var arguments = $"advfirewall firewall add rule name=\"{exceptionName}\" program=\"{exePath}\" protocol=tcp dir=in enable=yes action=allow";
            await using var netshProcess = await processExecutor.LaunchProcess(
                "netsh.exe",
                Environment.CurrentDirectory,
                arguments,
                cancellationToken,
                readStandardHandles: true,
                noShellExecute: true);
 
            if (lowPriority)
                netshProcess.AdjustPriority(false);
 
            int exitCode;
            using (cancellationToken.Register(() => netshProcess.Terminate()))
                exitCode = (await netshProcess.Lifetime).Value;
            cancellationToken.ThrowIfCancellationRequested();
 
            logger.LogDebug(
                "netsh.exe output:{newLine}{output}",
                Environment.NewLine,
                await netshProcess.GetCombinedOutput(cancellationToken));
 
            return exitCode;
        }
    }
}