Use File.SetUnixFileMode

SteamRE · Aug 27, 2024 · fec2fe7 · fec2fe7
1 parent a1f9811
commit fec2fe7
Showing 1 changed file with 6 additions and 124 deletions.
diff --git a/DepotDownloader/PlatformUtilities.cs b/DepotDownloader/PlatformUtilities.cs
@@ -2,147 +2,29 @@
 // in file 'LICENSE', which is part of this source code package.
 
 using System;
+using System.IO;
 using System.Runtime.InteropServices;
 
 namespace DepotDownloader
 {
     static class PlatformUtilities
     {
-        private const int ModeExecuteOwner = 0x0040;
-        private const int ModeExecuteGroup = 0x0008;
-        private const int ModeExecuteOther = 0x0001;
-        private const int ModeExecute = ModeExecuteOwner | ModeExecuteGroup | ModeExecuteOther;
-
-        [StructLayout(LayoutKind.Explicit, Size = 144)]
-        private readonly struct StatLinuxX64
-        {
-            [FieldOffset(24)] public readonly uint st_mode;
-        }
-
-        [StructLayout(LayoutKind.Explicit, Size = 104)]
-        private readonly struct StatLinuxArm32
-        {
-            [FieldOffset(16)] public readonly uint st_mode;
-        }
-
-        [StructLayout(LayoutKind.Explicit, Size = 128)]
-        private readonly struct StatLinuxArm64
-        {
-            [FieldOffset(16)] public readonly uint st_mode;
-        }
-
-        [StructLayout(LayoutKind.Explicit, Size = 144)]
-        private readonly struct StatOSX
-        {
-            [FieldOffset(4)] public readonly ushort st_mode;
-        }
-
-        [StructLayout(LayoutKind.Explicit, Size = 224)]
-        private readonly struct StatFBSDX64
-        {
-            [FieldOffset(24)] public readonly ushort st_mode;
-        }
-
-        [DllImport("libc", EntryPoint = "__xstat", SetLastError = true)]
-        private static extern int statLinuxX64(int version, string path, out StatLinuxX64 statLinux);
-
-        [DllImport("libc", EntryPoint = "__xstat", SetLastError = true)]
-        private static extern int statLinuxArm32(int version, string path, out StatLinuxArm32 statLinux);
-
-        [DllImport("libc", EntryPoint = "__xstat", SetLastError = true)]
-        private static extern int statLinuxArm64(int version, string path, out StatLinuxArm64 statLinux);
-
-        [DllImport("libc", EntryPoint = "stat", SetLastError = true)]
-        private static extern int statOSX(string path, out StatOSX stat);
-
-        [DllImport("libc", EntryPoint = "stat$INODE64", SetLastError = true)]
-        private static extern int statOSXCompat(string path, out StatOSX stat);
-
-        [DllImport("libc", EntryPoint = "stat", SetLastError = true)]
-        private static extern int statFBSDX64(string path, out StatFBSDX64 stat);
-
-        [DllImport("libc", SetLastError = true)]
-        private static extern int chmod(string path, uint mode);
-
-        [DllImport("libc", CallingConvention = CallingConvention.Cdecl, SetLastError = true)]
-        private static extern IntPtr strerror(int errno);
-
-        private static void ThrowIf(int i)
-        {
-            if (i == -1)
-            {
-                var errno = Marshal.GetLastWin32Error();
-                throw new Exception(Marshal.PtrToStringAnsi(strerror(errno)));
-            }
-        }
-
-        private static uint GetFileMode(string path)
-        {
-            if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
-            {
-                switch (RuntimeInformation.ProcessArchitecture)
-                {
-                    case Architecture.X64:
-                    {
-                        ThrowIf(statLinuxX64(1, path, out var stat));
-                        return stat.st_mode;
-                    }
-                    case Architecture.Arm:
-                    {
-                        ThrowIf(statLinuxArm32(3, path, out var stat));
-                        return stat.st_mode;
-                    }
-                    case Architecture.Arm64:
-                    {
-                        ThrowIf(statLinuxArm64(0, path, out var stat));
-                        return stat.st_mode;
-                    }
-                }
-            }
-            else if (RuntimeInformation.IsOSPlatform(OSPlatform.FreeBSD))
-            {
-                switch (RuntimeInformation.ProcessArchitecture)
-                {
-                    case Architecture.X64:
-                    {
-                        ThrowIf(statFBSDX64(path, out var stat));
-                        return stat.st_mode;
-                    }
-                }
-            }
-            else if (RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
-            {
-                switch (RuntimeInformation.ProcessArchitecture)
-                {
-                    case Architecture.X64:
-                    {
-                        ThrowIf(statOSXCompat(path, out var stat));
-                        return stat.st_mode;
-                    }
-                    case Architecture.Arm64:
-                    {
-                        ThrowIf(statOSX(path, out var stat));
-                        return stat.st_mode;
-                    }
-                }
-            }
-            throw new PlatformNotSupportedException();
-        }
-
         public static void SetExecutable(string path, bool value)
         {
             if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
             {
                 return;
             }
 
-            var mode = GetFileMode(path);
+            const UnixFileMode ModeExecute = UnixFileMode.UserExecute | UnixFileMode.GroupExecute | UnixFileMode.OtherExecute;
+
+            var mode = File.GetUnixFileMode(path);
             var hasExecuteMask = (mode & ModeExecute) == ModeExecute;
             if (hasExecuteMask != value)
             {
-                ThrowIf(chmod(path, (uint)(value
+                File.SetUnixFileMode(path, value
                     ? mode | ModeExecute
-                    : mode & ~ModeExecute)));
+                    : mode & ~ModeExecute);
             }
         }
     }