Detect when EXE is launched through a symlink (#125)

bflattened · Nov 5, 2023 · 623d32a · 623d32a
1 parent f81771e
commit 623d32a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/bflat/Program.cs b/src/bflat/Program.cs
@@ -19,6 +19,8 @@
 using System.CommandLine.Builder;
 using System.CommandLine.Help;
 using System.CommandLine.Parsing;
+using System.Diagnostics;
+using System.IO;
 using System.Reflection;
 
 class Program
@@ -27,6 +29,18 @@ class Program
 
     private static int Main(string[] args)
     {
+        string exeName = Environment.ProcessPath;
+        if (exeName != null)
+        {
+            FileSystemInfo actualExeInfo = File.ResolveLinkTarget(exeName, returnFinalTarget: true);
+            if (actualExeInfo != null)
+            {
+                Process p = Process.Start(actualExeInfo.FullName, args);
+                p.WaitForExit();
+                return p.ExitCode;
+            }
+        }
+
         using PerfWatch total = new PerfWatch("Total");
 
         var root = new RootCommand(