mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 11:34:57 +02:00
293 lines
9.9 KiB
C#
293 lines
9.9 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
|
|
/*
|
|
* Important - Even though this will compile in the shared projects, it will cause build failures within the mono runtime
|
|
*/
|
|
namespace MediaBrowser.ServerApplication.Native
|
|
{
|
|
/// <summary>
|
|
/// http://blogs.msdn.com/b/fiddler/archive/2011/12/10/fiddler-windows-8-apps-enable-LoopUtil-network-isolation-exemption.aspx
|
|
/// </summary>
|
|
public class LoopUtil
|
|
{
|
|
//http://msdn.microsoft.com/en-us/library/windows/desktop/aa379595(v=vs.85).aspx
|
|
[StructLayout(LayoutKind.Sequential)]
|
|
internal struct SID_AND_ATTRIBUTES
|
|
{
|
|
public IntPtr Sid;
|
|
public uint Attributes;
|
|
}
|
|
|
|
[StructLayoutAttribute(LayoutKind.Sequential)]
|
|
internal struct INET_FIREWALL_AC_CAPABILITIES
|
|
{
|
|
public uint count;
|
|
public IntPtr capabilities; //SID_AND_ATTRIBUTES
|
|
}
|
|
|
|
[StructLayoutAttribute(LayoutKind.Sequential)]
|
|
internal struct INET_FIREWALL_AC_BINARIES
|
|
{
|
|
public uint count;
|
|
public IntPtr binaries;
|
|
}
|
|
|
|
[StructLayoutAttribute(LayoutKind.Sequential)]
|
|
internal struct INET_FIREWALL_APP_CONTAINER
|
|
{
|
|
internal IntPtr appContainerSid;
|
|
internal IntPtr userSid;
|
|
[MarshalAs(UnmanagedType.LPWStr)]
|
|
public string appContainerName;
|
|
[MarshalAs(UnmanagedType.LPWStr)]
|
|
public string displayName;
|
|
[MarshalAs(UnmanagedType.LPWStr)]
|
|
public string description;
|
|
internal INET_FIREWALL_AC_CAPABILITIES capabilities;
|
|
internal INET_FIREWALL_AC_BINARIES binaries;
|
|
[MarshalAs(UnmanagedType.LPWStr)]
|
|
public string workingDirectory;
|
|
[MarshalAs(UnmanagedType.LPWStr)]
|
|
public string packageFullName;
|
|
}
|
|
|
|
|
|
// Call this API to load the current list of LoopUtil-enabled AppContainers
|
|
[DllImport("FirewallAPI.dll")]
|
|
internal static extern uint NetworkIsolationGetAppContainerConfig(out uint pdwCntACs, out IntPtr appContainerSids);
|
|
|
|
// Call this API to set the LoopUtil-exemption list
|
|
[DllImport("FirewallAPI.dll")]
|
|
private static extern uint NetworkIsolationSetAppContainerConfig(uint pdwCntACs, SID_AND_ATTRIBUTES[] appContainerSids);
|
|
|
|
// Use this API to convert a string SID into an actual SID
|
|
[DllImport("advapi32.dll", SetLastError = true)]
|
|
internal static extern bool ConvertStringSidToSid(string strSid, out IntPtr pSid);
|
|
|
|
[DllImport("advapi32", /*CharSet = CharSet.Auto,*/ SetLastError = true)]
|
|
static extern bool ConvertSidToStringSid(IntPtr pSid, out string strSid);
|
|
|
|
// Call this API to enumerate all of the AppContainers on the system
|
|
[DllImport("FirewallAPI.dll")]
|
|
internal static extern uint NetworkIsolationEnumAppContainers(uint Flags, out uint pdwCntPublicACs, out IntPtr ppACs);
|
|
// DWORD NetworkIsolationEnumAppContainers(
|
|
// _In_ DWORD Flags,
|
|
// _Out_ DWORD *pdwNumPublicAppCs,
|
|
// _Out_ PINET_FIREWALL_APP_CONTAINER *ppPublicAppCs
|
|
//);
|
|
|
|
//http://msdn.microsoft.com/en-gb/library/windows/desktop/hh968116.aspx
|
|
enum NETISO_FLAG
|
|
{
|
|
NETISO_FLAG_FORCE_COMPUTE_BINARIES = 0x1,
|
|
NETISO_FLAG_MAX = 0x2
|
|
}
|
|
|
|
|
|
public class AppContainer
|
|
{
|
|
public String appContainerName { get; set; }
|
|
public String displayName { get; set; }
|
|
public String workingDirectory { get; set; }
|
|
public String StringSid { get; set; }
|
|
public List<uint> capabilities { get; set; }
|
|
public bool LoopUtil { get; set; }
|
|
|
|
public AppContainer(String _appContainerName, String _displayName, String _workingDirectory, IntPtr _sid)
|
|
{
|
|
this.appContainerName = _appContainerName;
|
|
this.displayName = _displayName;
|
|
this.workingDirectory = _workingDirectory;
|
|
String tempSid;
|
|
ConvertSidToStringSid(_sid, out tempSid);
|
|
this.StringSid = tempSid;
|
|
}
|
|
}
|
|
|
|
internal List<LoopUtil.INET_FIREWALL_APP_CONTAINER> _AppList;
|
|
internal List<LoopUtil.SID_AND_ATTRIBUTES> _AppListConfig;
|
|
public List<AppContainer> Apps = new List<AppContainer>();
|
|
internal IntPtr _pACs;
|
|
|
|
public LoopUtil()
|
|
{
|
|
LoadApps();
|
|
}
|
|
|
|
public void LoadApps()
|
|
{
|
|
Apps.Clear();
|
|
_pACs = IntPtr.Zero;
|
|
//Full List of Apps
|
|
_AppList = PI_NetworkIsolationEnumAppContainers();
|
|
//List of Apps that have LoopUtil enabled.
|
|
_AppListConfig = PI_NetworkIsolationGetAppContainerConfig();
|
|
foreach (var PI_app in _AppList)
|
|
{
|
|
AppContainer app = new AppContainer(PI_app.appContainerName, PI_app.displayName, PI_app.workingDirectory, PI_app.appContainerSid);
|
|
|
|
app.LoopUtil = CheckLoopback(PI_app.appContainerSid);
|
|
Apps.Add(app);
|
|
}
|
|
}
|
|
private bool CheckLoopback(IntPtr intPtr)
|
|
{
|
|
foreach (SID_AND_ATTRIBUTES item in _AppListConfig)
|
|
{
|
|
string left, right;
|
|
ConvertSidToStringSid(item.Sid, out left);
|
|
ConvertSidToStringSid(intPtr, out right);
|
|
|
|
if (left == right)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private bool CreateExcemptions(string appName)
|
|
{
|
|
var hasChanges = false;
|
|
|
|
foreach (var app in Apps)
|
|
{
|
|
if ((app.appContainerName ?? string.Empty).IndexOf(appName, StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
(app.displayName ?? string.Empty).IndexOf(appName, StringComparison.OrdinalIgnoreCase) != -1)
|
|
{
|
|
if (!app.LoopUtil)
|
|
{
|
|
app.LoopUtil = true;
|
|
hasChanges = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return hasChanges;
|
|
}
|
|
|
|
public static void Run(string appName)
|
|
{
|
|
var util = new LoopUtil();
|
|
util.LoadApps();
|
|
|
|
var hasChanges = util.CreateExcemptions(appName);
|
|
|
|
if (hasChanges)
|
|
{
|
|
util.SaveLoopbackState();
|
|
}
|
|
}
|
|
|
|
private static List<SID_AND_ATTRIBUTES> PI_NetworkIsolationGetAppContainerConfig()
|
|
{
|
|
|
|
IntPtr arrayValue = IntPtr.Zero;
|
|
uint size = 0;
|
|
var list = new List<SID_AND_ATTRIBUTES>();
|
|
|
|
// Pin down variables
|
|
GCHandle handle_pdwCntPublicACs = GCHandle.Alloc(size, GCHandleType.Pinned);
|
|
GCHandle handle_ppACs = GCHandle.Alloc(arrayValue, GCHandleType.Pinned);
|
|
|
|
uint retval = NetworkIsolationGetAppContainerConfig(out size, out arrayValue);
|
|
|
|
var structSize = Marshal.SizeOf(typeof(SID_AND_ATTRIBUTES));
|
|
for (var i = 0; i < size; i++)
|
|
{
|
|
var cur = (SID_AND_ATTRIBUTES)Marshal.PtrToStructure(arrayValue, typeof(SID_AND_ATTRIBUTES));
|
|
list.Add(cur);
|
|
arrayValue = new IntPtr((long)(arrayValue) + (long)(structSize));
|
|
}
|
|
|
|
//release pinned variables.
|
|
handle_pdwCntPublicACs.Free();
|
|
handle_ppACs.Free();
|
|
|
|
return list;
|
|
|
|
|
|
}
|
|
|
|
private List<INET_FIREWALL_APP_CONTAINER> PI_NetworkIsolationEnumAppContainers()
|
|
{
|
|
|
|
IntPtr arrayValue = IntPtr.Zero;
|
|
uint size = 0;
|
|
var list = new List<INET_FIREWALL_APP_CONTAINER>();
|
|
|
|
// Pin down variables
|
|
GCHandle handle_pdwCntPublicACs = GCHandle.Alloc(size, GCHandleType.Pinned);
|
|
GCHandle handle_ppACs = GCHandle.Alloc(arrayValue, GCHandleType.Pinned);
|
|
|
|
//uint retval2 = NetworkIsolationGetAppContainerConfig( out size, out arrayValue);
|
|
|
|
uint retval = NetworkIsolationEnumAppContainers((Int32)NETISO_FLAG.NETISO_FLAG_MAX, out size, out arrayValue);
|
|
_pACs = arrayValue; //store the pointer so it can be freed when we close the form
|
|
|
|
var structSize = Marshal.SizeOf(typeof(INET_FIREWALL_APP_CONTAINER));
|
|
for (var i = 0; i < size; i++)
|
|
{
|
|
var cur = (INET_FIREWALL_APP_CONTAINER)Marshal.PtrToStructure(arrayValue, typeof(INET_FIREWALL_APP_CONTAINER));
|
|
list.Add(cur);
|
|
arrayValue = new IntPtr((long)(arrayValue) + (long)(structSize));
|
|
}
|
|
|
|
//release pinned variables.
|
|
handle_pdwCntPublicACs.Free();
|
|
handle_ppACs.Free();
|
|
|
|
return list;
|
|
|
|
|
|
}
|
|
|
|
public bool SaveLoopbackState()
|
|
{
|
|
var countEnabled = CountEnabledLoopUtil();
|
|
SID_AND_ATTRIBUTES[] arr = new SID_AND_ATTRIBUTES[countEnabled];
|
|
int count = 0;
|
|
|
|
for (int i = 0; i < Apps.Count; i++)
|
|
{
|
|
if (Apps[i].LoopUtil)
|
|
{
|
|
arr[count].Attributes = 0;
|
|
//TO DO:
|
|
IntPtr ptr;
|
|
ConvertStringSidToSid(Apps[i].StringSid, out ptr);
|
|
arr[count].Sid = ptr;
|
|
count++;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
if (NetworkIsolationSetAppContainerConfig((uint)countEnabled, arr) == 0)
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{ return false; }
|
|
|
|
}
|
|
|
|
private int CountEnabledLoopUtil()
|
|
{
|
|
var count = 0;
|
|
for (int i = 0; i < Apps.Count; i++)
|
|
{
|
|
if (Apps[i].LoopUtil)
|
|
{
|
|
count++;
|
|
}
|
|
|
|
}
|
|
return count;
|
|
}
|
|
}
|
|
}
|