def test_find_elfs():
a = Analysis.Analysis('/usr/bin')
elfs = a.find_elfs()
assert len(elfs) > 0
def test_scan_elfs():
a = Analysis.Analysis('/usr/bin')
b = a.scan_elfs(['/usr/bin/yes'])
assert b.has_key('yes')
def test_get_complexity():
a = Analysis.Analysis('/usr/bin')
b = a.get_complexity('/usr/bin/yes')
assert b.has_key('r2aa')
#!/usr/bin/env python
from cybertestlab import Analysis
import json
import sys
from elasticsearch import Elasticsearch
es = Elasticsearch(['http://localhost:9200'])
if len(sys.argv) > 1:
a = Analysis.Analysis(sys.argv[1], debug=True)
else:
a = Analysis.Analysis('/usr/bin', debug=True)
elfs = a.find_elfs()
# Let's loop this so we don't chew up all the RAM
for one_elf in elfs:
b = a.scan_elfs([one_elf])
for i in b.keys():
try:
es.update(id=i, index="ctl", doc_type='doc', body={'doc' :b[i], 'doc_as_upsert': True})
except:
# We'll figure this out someday
pass