You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.8 KiB
73 lines
2.8 KiB
using IPA.Logging;
|
|
using System;
|
|
using System.Diagnostics;
|
|
using System.Diagnostics.CodeAnalysis;
|
|
using System.IO;
|
|
using System.Security.AccessControl;
|
|
using System.Security.Principal;
|
|
using System.Threading.Tasks;
|
|
#if NET3
|
|
using Net3_Proxy;
|
|
#endif
|
|
|
|
namespace IPA.Injector
|
|
{
|
|
internal static class PermissionFix
|
|
{
|
|
[SuppressMessage("Reliability", "CA2008:Do not create tasks without passing a TaskScheduler",
|
|
Justification = "I very explicitly want the default scheduler")]
|
|
public static Task FixPermissions(DirectoryInfo root)
|
|
{
|
|
if (!root.Exists) return new Task(() => { });
|
|
|
|
return Task.Factory.StartNew(() =>
|
|
{
|
|
var sw = Stopwatch.StartNew();
|
|
try
|
|
{
|
|
var acl = root.GetAccessControl();
|
|
|
|
var rules = acl.GetAccessRules(true, true, typeof(SecurityIdentifier));
|
|
|
|
var requestedRights = FileSystemRights.Modify;
|
|
var requestedInheritance = InheritanceFlags.ContainerInherit | InheritanceFlags.ObjectInherit;
|
|
var requestedPropagation = PropagationFlags.InheritOnly;
|
|
|
|
bool hasRule = false;
|
|
for (var i = 0; i < rules.Count; i++)
|
|
{
|
|
var rule = rules[i];
|
|
|
|
if (rule is FileSystemAccessRule fsrule
|
|
&& fsrule.AccessControlType == AccessControlType.Allow
|
|
&& fsrule.InheritanceFlags.HasFlag(requestedInheritance)
|
|
&& fsrule.PropagationFlags == requestedPropagation
|
|
&& fsrule.FileSystemRights.HasFlag(requestedRights))
|
|
{ hasRule = true; break; }
|
|
}
|
|
|
|
if (!hasRule)
|
|
{ // this is *sooo* fucking slow on first run
|
|
acl.AddAccessRule(
|
|
new FileSystemAccessRule(
|
|
new SecurityIdentifier(WellKnownSidType.WorldSid, null),
|
|
requestedRights,
|
|
requestedInheritance,
|
|
requestedPropagation,
|
|
AccessControlType.Allow
|
|
)
|
|
);
|
|
root.SetAccessControl(acl);
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.log.Warn("Error configuring permissions in the game install dir");
|
|
Logger.log.Warn(e);
|
|
}
|
|
sw.Stop();
|
|
Logger.log.Info($"Configuring permissions took {sw.Elapsed}");
|
|
});
|
|
}
|
|
}
|
|
}
|