public void TestFirewallCollectorLinux()
{
if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
{
var result = ExternalCommandRunner.RunExternalCommand("iptables", "-A INPUT -p tcp --dport 19999 -j DROP");
var fwc = new FirewallCollector();
fwc.Execute();
Assert.IsTrue(fwc.Results.Any(x => x is FirewallObject FWO && FWO.LocalPorts.Contains("19999")));
result = ExternalCommandRunner.RunExternalCommand("iptables", "-D INPUT -p tcp --dport 19999 -j DROP");
}
}
public void TestFirewallCollectorWindows()
{
if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
{
Assert.IsTrue(AsaHelpers.IsAdmin());
var rule = FirewallManager.Instance.CreatePortRule(
@"TestFirewallPortRule",
FirewallAction.Allow,
9999,
FirewallProtocol.TCP
);
FirewallManager.Instance.Rules.Add(rule);
rule = FirewallManager.Instance.CreateApplicationRule(
@"TestFirewallAppRule",
FirewallAction.Allow,
@"C:\MyApp.exe"
);
rule.Direction = FirewallDirection.Outbound;
FirewallManager.Instance.Rules.Add(rule);
var fwc = new FirewallCollector();
fwc.TryExecute();
Assert.IsTrue(fwc.Results.Any(x => x is FirewallObject FWO && FWO.LocalPorts.Contains("9999")));
Assert.IsTrue(fwc.Results.Any(x => x is FirewallObject FWO && FWO.ApplicationName is string && FWO.ApplicationName.Equals(@"C:\MyApp.exe")));
ConcurrentStack <CollectObject> results = new ConcurrentStack <CollectObject>();
fwc = new FirewallCollector(changeHandler: x => results.Push(x));
fwc.TryExecute();
Assert.IsTrue(results.Any(x => x is FirewallObject FWO && FWO.LocalPorts.Contains("9999")));
Assert.IsTrue(results.Any(x => x is FirewallObject FWO && FWO.ApplicationName is string && FWO.ApplicationName.Equals(@"C:\MyApp.exe")));
var rules = FirewallManager.Instance.Rules.Where(r => r.Name == "TestFirewallPortRule");
foreach (var ruleIn in rules)
{
FirewallManager.Instance.Rules.Remove(ruleIn);
}
rules = FirewallManager.Instance.Rules.Where(r => r.Name == "TestFirewallAppRule");
foreach (var ruleIn in rules)
{
FirewallManager.Instance.Rules.Remove(ruleIn);
}
}
}
