Fixed bh_owned error on non-existing values

If a computer on the network has been compromised but is not listed in the Neo4j database. CME won't crash any more.
main
Shariq Malik 2022-11-25 16:58:52 +05:00 committed by mpgn
parent 4bab776011
commit b14fe7f94f
1 changed files with 4 additions and 1 deletions

View File

@ -75,7 +75,10 @@ class CMEModule:
result = tx.run(
"MATCH (c:Computer {{name:\"{}\"}}) SET c.owned=True RETURN c.name AS name".format(host_fqdn))
record = result.single()
value = record.value()
try:
value = record.value()
except AttributeError:
value = []
if len(value) > 0:
context.log.success("Node {} successfully set as owned in BloodHound".format(host_fqdn))
else: