From 35be8bedd07d323c2bae9395e1a5d7d518d990ea Mon Sep 17 00:00:00 2001
From: ChsBuffer <33744752+chsbuffer@users.noreply.github.com>
Date: Mon, 5 Jul 2021 01:20:08 +0800
Subject: [PATCH] Update MainController Async

---
 Netch/Controllers/MainController.cs | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/Netch/Controllers/MainController.cs b/Netch/Controllers/MainController.cs
index 9c13d736..2b960552 100644
--- a/Netch/Controllers/MainController.cs
+++ b/Netch/Controllers/MainController.cs
@@ -28,20 +28,24 @@ namespace Netch.Controllers
 
             Mode = mode;
 
-            NativeMethods.RefreshDNSCache();
-            Firewall.AddNetchFwRules();
-            Task.Run(() =>
-            {
-                if (Log.IsEnabled(LogEventLevel.Debug))
+            await Task.WhenAll(
+                Task.Run(NativeMethods.RefreshDNSCache),
+                Task.Run(Firewall.AddNetchFwRules)
+            );
+
+            if (Log.IsEnabled(LogEventLevel.Debug))
+                Task.Run(() =>
+                {
+                    // TODO log level setting
                     Log.Debug("Running Processes: \n{Processes}", string.Join("\n", SystemInfo.Processes(false)));
-            }).Forget();
+                }).Forget();
 
             try
             {
                 if (!ModeHelper.SkipServerController(server, mode))
-                    server = StartServer(server);
+                    server = await Task.Run(() => StartServer(server));
 
-                StartMode(server, mode);
+                await Task.Run(() => StartMode(server, mode));
             }
             catch (Exception e)
             {