update components
This commit is contained in:
Luke Pulverenti
parent
fc2a5acfca
commit
d6b0dcaddb
358
Emby.Server.Implementations/Windows/LoopUtil.cs
Normal file
358
Emby.Server.Implementations/Windows/LoopUtil.cs
Normal file
|
|
@ -0,0 +1,358 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Runtime.InteropServices;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace Emby.Server.Implementations.Windows
|
||||
{
|
||||
/// <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 free the memory returned by the Enumeration API
|
||||
[DllImport("FirewallAPI.dll")]
|
||||
internal static extern void NetworkIsolationFreeAppContainers(IntPtr pACs);
|
||||
|
||||
// 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(
|
||||
[MarshalAs(UnmanagedType.LPArray)] byte[] pSID,
|
||||
out IntPtr ptrSid);
|
||||
|
||||
[DllImport("advapi32", /*CharSet = CharSet.Auto,*/ SetLastError = true)]
|
||||
static extern bool ConvertSidToStringSid(IntPtr pSid, out string strSid);
|
||||
|
||||
// Use this API to convert a string reference (e.g. "@{blah.pri?ms-resource://whatever}") into a plain string
|
||||
[DllImport("shlwapi.dll", CharSet = CharSet.Unicode, ExactSpelling = true)]
|
||||
internal static extern int SHLoadIndirectString(string pszSource, StringBuilder pszOutBuf);
|
||||
|
||||
// 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);
|
||||
|
||||
var app_capabilities = LoopUtil.getCapabilites(PI_app.capabilities);
|
||||
if (app_capabilities.Count > 0)
|
||||
{
|
||||
//var sid = new SecurityIdentifier(app_capabilities[0], 0);
|
||||
|
||||
IntPtr arrayValue = IntPtr.Zero;
|
||||
//var b = LoopUtil.ConvertStringSidToSid(app_capabilities[0].Sid, out arrayValue);
|
||||
//string mysid;
|
||||
//var b = LoopUtil.ConvertSidToStringSid(app_capabilities[0].Sid, out mysid);
|
||||
}
|
||||
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();
|
||||
}
|
||||
util.SaveLoopbackState();
|
||||
}
|
||||
|
||||
private static List<SID_AND_ATTRIBUTES> getCapabilites(INET_FIREWALL_AC_CAPABILITIES cap)
|
||||
{
|
||||
List<SID_AND_ATTRIBUTES> mycap = new List<SID_AND_ATTRIBUTES>();
|
||||
|
||||
IntPtr arrayValue = cap.capabilities;
|
||||
|
||||
var structSize = Marshal.SizeOf(typeof(SID_AND_ATTRIBUTES));
|
||||
for (var i = 0; i < cap.count; i++)
|
||||
{
|
||||
var cur = (SID_AND_ATTRIBUTES)Marshal.PtrToStructure(arrayValue, typeof(SID_AND_ATTRIBUTES));
|
||||
mycap.Add(cur);
|
||||
arrayValue = new IntPtr((long)(arrayValue) + (long)(structSize));
|
||||
}
|
||||
|
||||
return mycap;
|
||||
|
||||
}
|
||||
|
||||
private static List<SID_AND_ATTRIBUTES> getContainerSID(INET_FIREWALL_AC_CAPABILITIES cap)
|
||||
{
|
||||
List<SID_AND_ATTRIBUTES> mycap = new List<SID_AND_ATTRIBUTES>();
|
||||
|
||||
IntPtr arrayValue = cap.capabilities;
|
||||
|
||||
var structSize = Marshal.SizeOf(typeof(SID_AND_ATTRIBUTES));
|
||||
for (var i = 0; i < cap.count; i++)
|
||||
{
|
||||
var cur = (SID_AND_ATTRIBUTES)Marshal.PtrToStructure(arrayValue, typeof(SID_AND_ATTRIBUTES));
|
||||
mycap.Add(cur);
|
||||
arrayValue = new IntPtr((long)(arrayValue) + (long)(structSize));
|
||||
}
|
||||
|
||||
return mycap;
|
||||
|
||||
}
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
public void FreeResources()
|
||||
{
|
||||
NetworkIsolationFreeAppContainers(_pACs);
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user