add missing utils

reg-chen · Dec 27, 2021 · d504414 · d504414
1 parent d8e8109
commit d504414
Showing 1 changed file with 87 additions and 17 deletions.
diff --git a/ControllerCommon/Utils.cs b/ControllerCommon/Utils.cs
@@ -5,6 +5,7 @@
 using System.Management;
 using System.Net;
 using System.Runtime.InteropServices;
+using System.Security.AccessControl;
 using System.Security.Principal;
 using System.Text;
 
@@ -132,29 +133,40 @@ public static void OpenUrl(string url)
             }
         }
 
+        public static bool IsFileWritable(string filePath)
+        {
+            try
+            {
+                if (File.Exists(filePath))
+                {
+                    using (var fs = new FileStream(filePath, FileMode.Open))
+                        return fs.CanWrite;
+                }
+                else
+                {
+                    bool CanWrite = false;
+                    using (var fs = new FileStream(filePath, FileMode.Create))
+                        CanWrite = fs.CanWrite;
+                    File.Delete(filePath);
+                    return CanWrite;
+                }
+            }
+            catch(Exception)
+            {
+                return false;
+            }
+        }
 
-        public static bool IsDirectoryWritable(string dirPath, bool throwIfFails = false)
+        public static bool IsDirectoryWritable(string dirPath)
         {
             try
             {
-                using (FileStream fs = File.Create(
-                    Path.Combine(
-                        dirPath,
-                        Path.GetRandomFileName()
-                    ),
-                    1,
-                    FileOptions.DeleteOnClose)
-                )
-                { }
-                return true;
+                using (FileStream fs = File.Create(Path.Combine(dirPath,Path.GetRandomFileName()), 1, FileOptions.DeleteOnClose)          )
+                    return true;
             }
-            catch (Exception ex)
+            catch (Exception)
             {
-                Console.WriteLine(ex.Message);
-                if (throwIfFails)
-                    throw;
-                else
-                    return false;
+                return false;
             }
         }
 
@@ -182,5 +194,63 @@ public void SetFilterAttrs(double minCutoff, double beta)
                 axis1Filter.Beta = axis2Filter.Beta = axis3Filter.Beta = beta;
             }
         }
+
+        public static void SetDirectoryWritable(string processpath)
+        {
+            var rootDirectory = new DirectoryInfo(processpath);
+            var directorySecurity = rootDirectory.GetAccessControl();
+
+            SecurityIdentifier everyone = new SecurityIdentifier(WellKnownSidType.WorldSid, null);
+            SecurityIdentifier adminitrators = new SecurityIdentifier(WellKnownSidType.BuiltinAdministratorsSid, null);
+
+            directorySecurity.AddAccessRule(
+                new FileSystemAccessRule(
+                        everyone,
+                        FileSystemRights.FullControl,
+                        InheritanceFlags.None,
+                        PropagationFlags.NoPropagateInherit,
+                        AccessControlType.Allow));
+
+            directorySecurity.AddAccessRule(
+                new FileSystemAccessRule(
+                WindowsIdentity.GetCurrent().Name,
+                FileSystemRights.FullControl,
+                InheritanceFlags.None,
+                PropagationFlags.NoPropagateInherit,
+                AccessControlType.Allow));
+
+            directorySecurity.SetAccessRuleProtection(isProtected: true, preserveInheritance: false);
+
+            rootDirectory.SetAccessControl(directorySecurity);
+        }
+
+        public static void SetFileWritable(string processpath)
+        {
+            var rootFile = new FileInfo(processpath);
+            var fileSecurity = rootFile.GetAccessControl();
+
+            SecurityIdentifier everyone = new SecurityIdentifier(WellKnownSidType.WorldSid, null);
+            SecurityIdentifier adminitrators = new SecurityIdentifier(WellKnownSidType.BuiltinAdministratorsSid, null);
+
+            fileSecurity.AddAccessRule(
+                new FileSystemAccessRule(
+                        everyone,
+                        FileSystemRights.FullControl,
+                        InheritanceFlags.None,
+                        PropagationFlags.NoPropagateInherit,
+                        AccessControlType.Allow));
+
+            fileSecurity.AddAccessRule(
+                new FileSystemAccessRule(
+                WindowsIdentity.GetCurrent().Name,
+                FileSystemRights.FullControl,
+                InheritanceFlags.None,
+                PropagationFlags.NoPropagateInherit,
+                AccessControlType.Allow));
+
+            fileSecurity.SetAccessRuleProtection(isProtected: true, preserveInheritance: false);
+
+            rootFile.SetAccessControl(fileSecurity);
+        }
     }
 }