DnsServer/DnsServerApp/Program.cs

/*
Technitium DNS Server
Copyright (C) 2026  Shreyas Zare ([email protected])

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

*/

using DnsServerCore;
using System;
using System.Threading;
using System.Threading.Tasks;
using System.Runtime.InteropServices;

namespace DnsServerApp
{
    class Program
    {
        static async Task Main(string[] args)
        {
            bool throwIfBindFails = false;
            bool isPortableApp = false; //true if logs folder should be inside the config folder else will use OS specific log folder path
            string? configFolder = null;

            foreach (string arg in args)
            {
                switch (arg)
                {
                    case "--icu-test":
                        _ = System.Globalization.CultureInfo.CurrentCulture;
                        return;

                    case "--stop-if-bind-fails":
                        throwIfBindFails = true;
                        break;

                    case "--portable-app":
                        isPortableApp = true;
                        break;

                    default:
                        configFolder = arg;
                        break;
                }
            }

            ManualResetEvent waitHandle = new ManualResetEvent(false);
            ManualResetEvent exitHandle = new ManualResetEvent(false);
            DnsWebService? service = null;
            PosixSignalRegistration? psr = null;

            try
            {
                if (!isPortableApp)
                    isPortableApp = configFolder is null; //no config folder specified so run as portable app

                Uri updateCheckUri;

                switch (Environment.OSVersion.Platform)
                {
                    case PlatformID.Win32NT:
                        updateCheckUri = new Uri("https://go.technitium.com/?id=41");
                        break;

                    default:
                        updateCheckUri = new Uri("https://go.technitium.com/?id=42");
                        break;
                }

                service = new DnsWebService(isPortableApp, configFolder, updateCheckUri);
                await service.StartAsync(throwIfBindFails);

                Console.CancelKeyPress += delegate (object? sender, ConsoleCancelEventArgs e)
                {
                    e.Cancel = true;
                    waitHandle.Set();
                };

                AppDomain.CurrentDomain.ProcessExit += delegate (object? sender, EventArgs e)
                {
                    waitHandle.Set();
                    exitHandle.WaitOne();
                };

                if (Environment.OSVersion.Platform == PlatformID.Unix)
                {
                    psr = PosixSignalRegistration.Create(PosixSignal.SIGTERM, delegate (PosixSignalContext context)
                    {
                        waitHandle.Set();
                        exitHandle.WaitOne();
                    });
                }

                string serviceUrl;

                if (service.IsWebServiceTlsEnabled)
                    serviceUrl = $"https://{Environment.MachineName.ToLowerInvariant()}:{service.WebServiceTlsPort}";
                else
                    serviceUrl = $"http://{Environment.MachineName.ToLowerInvariant()}:{service.WebServiceHttpPort}";

                Console.WriteLine($"Technitium DNS Server v{service.Version} was started successfully.\r\nStarted on: {service.UpTimeStamp.ToLocalTime()} Local Time\r\nUsing config folder: {service.ConfigFolder}\r\n\r\nNote: Open {serviceUrl} in web browser to access web console.\r\n\r\nPress [CTRL + C] to stop...");

                waitHandle.WaitOne();
            }
            catch (Exception ex)
            {
                Console.WriteLine(ex.ToString());
            }
            finally
            {
                Console.WriteLine("\r\nTechnitium DNS Server is stopping...");

                service?.Dispose();
                psr?.Dispose();

                Console.WriteLine("Technitium DNS Server was stopped successfully.");
                exitHandle.Set();
            }
        }
    }
}