MeshCentral/MeshAgent
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshAgent synced 2025-12-24 12:13:16 +00:00
Code Issues Releases Wiki Activity

Added timeout functionality to async firewall enumeration

Browse Source
This commit is contained in:
Bryan Roe
2022-04-16 12:45:36 -07:00
parent ce2aca195a
commit e57acc526d
2 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
modules/win-firewall.js
View File

@@ -294,6 +294,13 @@ function getFirewallRulesAsync2(p)
var rule, tmp;
OleAut.VariantClear(p.vvar);
if (p.options.timeout != null && (Date.now() - p.startTime) > p.options.timeout)
{
p.NetFwPolicy2.funcs.Release(p.NetFwPolicy2);
p.reject();
return;
}
hr = p.enumerator.funcs.Next(p.enumerator.Deref(), 1, p.vvar, p.fetched);
if (hr.Val == 0)
{
@@ -321,8 +328,6 @@ function getFirewallRulesAsync2(p)
&& p.options.program.toLowerCase() == p.val.Deref().Wide2UTF8.toLowerCase()) || !p.options || !p.options.program)
{
obj = {};
obj._rule = rule;
obj._rule._i = p.NetFwPolicy2;
if (p.val.Deref().Val != 0)
{
obj.Program = p.val.Deref().Wide2UTF8;
@@ -364,10 +369,12 @@ function getFirewallRulesAsync2(p)
if (p.options.noResult != true) { p.arr.push(obj); }
p.arr.push(obj);
}
rule.funcs.Release(rule.Deref());
setImmediate(getFirewallRulesAsync2, p);
}
else
{
p.NetFwPolicy2.funcs.Release(p.NetFwPolicy2);
p.resolve(p.options.noResult === true ? null : p.arr);
}
}
@@ -403,6 +410,7 @@ function getFirewallRulesAsync(options)
var count = GM.CreateVariable(4);
ret.rules.funcs.get_Count(ret.rules.Deref(), count).Val;
ret.count = count.toBuffer().readInt32LE();
ret.startTime = Date.now();
setImmediate(getFirewallRulesAsync2, ret);