def _run(self, context, exploit):
sample = os.path.join(MISC, exploit[0])
instance = ThugAPI(None, configuration_path = "/etc/thug")
instance.set_events('click')
instance.set_timeout(1)
instance.log_init(sample)
instance.run_local(sample)
for assertion in exploit[1].split(","):
assert assertion in context.log_capture.getvalue()
def do_perform_test(self, caplog, url, expected):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_broken_url()
thug.log_init(url)
thug.run_remote(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected, type_ = "remote"):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_features_logging()
thug.set_ssl_verify()
thug.log_init(url)
m = getattr(thug, "run_{}".format(type_))
m(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected, type_ = "remote"):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_features_logging()
thug.log_init(url)
m = getattr(thug, "run_{}".format(type_))
m(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.set_useragent('winxpie70')
thug.set_threshold(2)
thug.disable_cert_logging()
thug.set_features_logging()
thug.log_init(sample)
thug.reset_customclassifiers()
thug.add_customclassifier('url', self.catchall)
thug.reset_customclassifiers()
thug.add_customclassifier('html', self.catchall)
thug.add_customclassifier('url', self.catchall)
thug.add_customclassifier('js', self.catchall)
thug.add_customclassifier('vbs', self.catchall)
thug.add_customclassifier('sample', self.catchall)
thug.add_customclassifier('cookie', self.catchall)
thug.add_customclassifier('text', self.catchall)
thug.add_htmlclassifier(
os.path.join(self.signatures_path, "html_signature_1.yar"))
thug.add_jsclassifier(
os.path.join(self.signatures_path, "js_signature_2.yar"))
thug.add_urlclassifier(
os.path.join(self.signatures_path, "url_signature_3.yar"))
thug.add_urlfilter(
os.path.join(self.signatures_path, "url_filter_4.yar"))
thug.add_textclassifier(
os.path.join(self.signatures_path, "text_signature_5.yar"))
thug.add_vbsclassifier(
os.path.join(self.signatures_path, "vbs_signature_6.yar"))
thug.add_urlclassifier(
os.path.join(self.signatures_path, "url_signature_7.yar"))
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, adobe, expected):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_events('click,storage')
thug.disable_cert_logging()
thug.set_features_logging()
if adobe in ('disable', ):
thug.disable_acropdf()
else:
thug.set_acropdf_pdf(adobe)
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_events('click,storage')
thug.set_extensive()
thug.disable_cert_logging()
thug.set_file_logging()
thug.set_json_logging()
thug.set_features_logging()
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.set_useragent('winxpie60')
thug.set_events('click')
thug.set_connect_timeout(2)
thug.disable_cert_logging()
thug.set_features_logging()
thug.set_ssl_verify()
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def __init__(self, context):
ThugAPI.__init__(self)
self.exploits = list()
for row in context.table:
self.exploits.append(row)
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.set_useragent('osx10safari5')
thug.set_events('click,storage')
thug.set_connect_timeout(2)
thug.disable_cert_logging()
thug.set_features_logging()
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def __init__(self, context):
ThugAPI.__init__(self)
self.jquery = list()
for row in context.table:
self.jquery.append(row)
def do_perform_test(self,
caplog,
sample,
expected,
events='',
useragent='win7ie90'):
thug = ThugAPI()
thug.set_useragent(useragent)
thug.set_events(events)
thug.disable_cert_logging()
thug.set_features_logging()
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.log_init(sample)
thug.add_htmlclassifier(os.path.join(self.signatures_path, "html_signature_1.yar"))
thug.add_textclassifier(os.path.join(self.signatures_path, "text_signature_5.yar"))
thug.add_cookieclassifier(os.path.join(self.signatures_path, "cookie_signature_8.yar"))
thug.add_sampleclassifier(os.path.join(self.signatures_path, "sample_signature_10.yar"))
thug.add_htmlfilter(os.path.join(self.signatures_path, "html_filter_2.yar"))
thug.add_jsfilter(os.path.join(self.signatures_path, "js_signature_2.yar"))
thug.add_vbsfilter(os.path.join(self.signatures_path, "vbs_signature_6.yar"))
thug.add_textfilter(os.path.join(self.signatures_path, "text_signature_5.yar"))
thug.add_cookiefilter(os.path.join(self.signatures_path, "cookie_filter_9.yar"))
thug.add_samplefilter(os.path.join(self.signatures_path, "sample_filter_11.yar"))
thug.add_htmlclassifier(os.path.join(self.signatures_path, "not_existing.yar"))
thug.add_htmlfilter(os.path.join(self.signatures_path, "not_existing.yar"))
thug.add_customclassifier('wrong_type', 'wrong_method')
thug.add_customclassifier('url', 'wrong_method')
thug.add_customclassifier('sample', self.sample_passthrough)
with open(os.path.join(self.samples_path, sample), 'rb') as fd:
data = fd.read()
log.HTMLClassifier.classify(os.path.basename(sample), data)
log.TextClassifier.classify(os.path.basename(sample), data)
log.TextClassifier.classify(os.path.basename(sample), data)
log.CookieClassifier.classify(os.path.basename(sample), data)
log.CookieClassifier.classify(os.path.basename(sample), data)
log.SampleClassifier.classify(data, hashlib.md5(data).hexdigest())
log.HTMLClassifier.filter(os.path.basename(sample), data)
log.JSClassifier.filter(os.path.basename(sample), data)
log.VBSClassifier.filter(os.path.basename(sample), data)
log.TextClassifier.filter(os.path.basename(sample), data)
log.CookieClassifier.filter(os.path.basename(sample), data)
log.SampleClassifier.filter(data, hashlib.md5(data).hexdigest())
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected, type_ = "remote"):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_events('click,storage')
thug.set_web_tracking()
thug.disable_cert_logging()
thug.log_init(url)
m = getattr(thug, "run_{}".format(type_))
m(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self,
caplog,
sample,
expected,
useragent='linuxfirefox40'):
thug = ThugAPI()
thug.set_useragent(useragent)
thug.set_events('click,storage')
thug.set_connect_timeout(2)
thug.set_delay(500)
thug.disable_cert_logging()
thug.set_features_logging()
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.disable_acropdf()
thug.disable_shockwave_flash()
thug.disable_javaplugin()
thug.disable_silverlight()
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample):
thug = ThugAPI()
thug.log_init(sample)
thug.set_useragent('win7ie90')
thug.set_verbose()
thug.set_json_logging()
thug.reset_features_logging()
assert thug.get_features_logging() is False
thug.set_features_logging()
assert thug.get_features_logging() is True
thug.log_init(sample)
thug.run_local(sample)
thug.log_event()
for r in caplog.records:
try:
features = json.dumps(r)
except Exception:
continue
if not isinstance(features, dict):
continue
if "html_count" not in features:
continue
for url in self.expected:
if not url.endswith(sample):
continue
for key in features:
assert features[key] == self.expected[url][key]
def __init__(self, context):
ThugAPI.__init__(self)
self.misc = list()
for row in context.table:
self.misc.append(row)
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.set_useragent('winxpie70')
thug.set_ssl_verify()
thug.log_init(sample)
thug.add_htmlclassifier(
os.path.join(self.signatures_path, "inspector.yar"))
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected, type_="remote"):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_ssl_verify()
thug.reset_image_processing()
thug.set_image_processing()
thug.get_image_processing()
thug.register_pyhook("MIMEHandler", "handle_image",
self.handle_image_hook)
thug.set_json_logging()
thug.log_init(url)
m = getattr(thug, "run_{}".format(type_))
m(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
class TestThugAPI:
thug_api = ThugAPI()
cwd_path = os.path.dirname(os.path.realpath(__file__))
samples_path = os.path.join(cwd_path, os.pardir, os.pardir,
"tests/test_files")
yara_file = os.path.join(samples_path, "test_yara")
log_url = os.path.join(samples_path, "../log-dir-example")
log_file = os.path.join(samples_path, "test-filehandler")
def test_version(self):
with pytest.raises(
SystemExit): # TODO: Needs assert statement by mocking print
self.thug_api.version()
def test_useragent(self):
assert self.thug_api.get_useragent() in ('winxpie60', )
self.thug_api.set_useragent('winxpchrome20')
assert self.thug_api.get_useragent() in ('winxpchrome20', )
def test_events(self):
assert self.thug_api.get_events() in ([], )
self.thug_api.set_events('event1,event2,event2,event3')
assert self.thug_api.get_events() in (['event1', 'event2', 'event3'], )
def test_delay(self):
assert self.thug_api.get_delay() in (0, )
self.thug_api.set_delay(10)
assert self.thug_api.get_delay() in (10, )
def test_attachment(self):
assert not self.thug_api.get_attachment()
self.thug_api.set_attachment()
assert self.thug_api.get_attachment()
def test_file_logging(self):
assert not self.thug_api.get_file_logging()
self.thug_api.set_file_logging()
assert self.thug_api.get_file_logging()
def test_json_logging(self):
assert not self.thug_api.get_json_logging()
self.thug_api.set_json_logging()
assert self.thug_api.get_json_logging()
def test_elasticsearch_logging(self):
assert not self.thug_api.get_elasticsearch_logging()
self.thug_api.set_elasticsearch_logging()
assert self.thug_api.get_elasticsearch_logging()
assert logging.getLogger("elasticsearch").getEffectiveLevel() in (
logging.ERROR, )
def test_referer(self):
assert self.thug_api.get_referer() in ('about:blank', )
self.thug_api.set_referer('https://www.example.com')
assert self.thug_api.get_referer() in ('https://www.example.com', )
def test_proxy(self):
assert self.thug_api.get_proxy() is None
self.thug_api.set_proxy('http://www.example.com')
assert self.thug_api.get_proxy() in ('http://www.example.com', )
def test_raise_for_proxy(self):
assert self.thug_api.get_raise_for_proxy()
self.thug_api.set_raise_for_proxy(False)
assert not self.thug_api.get_raise_for_proxy()
def test_no_fetch(self):
assert not log.ThugOpts.no_fetch
self.thug_api.set_no_fetch()
assert log.ThugOpts.no_fetch
def test_verbose(self):
assert not log.ThugOpts.verbose
self.thug_api.set_verbose()
assert log.ThugOpts.verbose
assert log.getEffectiveLevel() in (logging.INFO, )
def test_debug(self):
assert not log.ThugOpts.debug
self.thug_api.set_debug()
assert log.ThugOpts.debug
assert log.getEffectiveLevel() in (logging.DEBUG, )
def test_ast_debug(self):
assert not log.ThugOpts.ast_debug
self.thug_api.set_ast_debug()
assert log.ThugOpts.ast_debug
def test_http_debug(self):
assert log.ThugOpts.http_debug in (0, )
self.thug_api.set_http_debug()
assert log.ThugOpts.http_debug in (1, )
self.thug_api.set_http_debug()
assert log.ThugOpts.http_debug in (2, )
def test_acropdf_pdf(self):
assert log.ThugVulnModules.acropdf_pdf in ('9.1.0', )
self.thug_api.set_acropdf_pdf('1.0.0', )
assert log.ThugVulnModules.acropdf_pdf in ('1.0.0', )
def test_disable_acropdf(self):
assert not log.ThugVulnModules.acropdf_disabled
self.thug_api.disable_acropdf()
assert log.ThugVulnModules.acropdf_disabled
def test_shockwave_flash(self):
assert log.ThugVulnModules.shockwave_flash in ('10.0.64.0', )
self.thug_api.set_shockwave_flash('8.0', )
assert log.ThugVulnModules.shockwave_flash in ('8.0', )
def test_disable_shockwave_flash(self):
assert not log.ThugVulnModules.shockwave_flash_disabled
self.thug_api.disable_shockwave_flash()
assert log.ThugVulnModules.shockwave_flash_disabled
def test_javaplugin(self):
assert log.ThugVulnModules.javaplugin in ('160_32', )
self.thug_api.set_javaplugin('1.0', )
assert log.ThugVulnModules.javaplugin in ('100_00', )
def test_disable_javaplugin(self):
assert not log.ThugVulnModules.javaplugin_disabled
self.thug_api.disable_javaplugin()
assert log.ThugVulnModules.javaplugin_disabled
def test_silverlight(self):
assert log.ThugVulnModules.silverlight in ('4.0.50826.0', )
self.thug_api.set_silverlight('1.0', )
assert log.ThugVulnModules.silverlight in ('1.0', )
def test_disable_silverlight(self):
assert not log.ThugVulnModules.silverlight_disabled
self.thug_api.disable_silverlight()
assert log.ThugVulnModules.silverlight_disabled
def test_threshold(self):
assert self.thug_api.get_threshold() in (0, )
self.thug_api.set_threshold(5)
assert self.thug_api.get_threshold() in (5, )
def test_extensive(self):
assert not self.thug_api.get_extensive()
self.thug_api.set_extensive()
assert self.thug_api.get_extensive()
def test_timeout(self):
assert self.thug_api.get_timeout() in (600, )
self.thug_api.set_timeout(300)
assert self.thug_api.get_timeout() in (300, )
def test_connect_timeout(self):
assert self.thug_api.get_connect_timeout() in (10, )
self.thug_api.set_connect_timeout(20)
assert self.thug_api.get_connect_timeout() in (20, )
def test_broken_url(self):
assert not self.thug_api.get_broken_url()
self.thug_api.set_broken_url()
assert self.thug_api.get_broken_url()
def test_web_tracking(self):
assert not self.thug_api.get_web_tracking()
self.thug_api.set_web_tracking()
assert self.thug_api.get_web_tracking()
def test_honeyagent(self):
assert log.ThugOpts.honeyagent
self.thug_api.disable_honeyagent()
assert not log.ThugOpts.honeyagent
def test_code_logging(self):
self.thug_api.enable_code_logging()
assert log.ThugOpts.code_logging
self.thug_api.disable_code_logging()
assert not log.ThugOpts.code_logging
def test_cert_logging(self):
self.thug_api.enable_cert_logging()
assert log.ThugOpts.cert_logging
self.thug_api.disable_cert_logging()
assert not log.ThugOpts.cert_logging
def test_log_init(self):
log.ThugOpts.file_logging = True
url = "../thugapi-example"
self.thug_api.log_init(url)
base_dir = log.ThugLogging.baseDir
log_path = os.path.dirname(
os.path.dirname(base_dir)) # TODO: Make this neat
assert os.path.isdir(base_dir)
shutil.rmtree(log_path)
assert not os.path.isdir(base_dir)
def test_log_dir(self):
self.thug_api.set_log_dir(self.log_url)
assert os.path.isdir(self.log_url)
shutil.rmtree(self.log_url)
assert not os.path.isdir(self.log_url)
def test_log_output(self):
self.thug_api.set_log_output(self.log_file)
assert isinstance(log.handlers[0], logging.FileHandler)
assert os.path.isfile(self.log_file)
os.remove(self.log_file)
assert not os.path.isfile(self.log_file)
def test_log_quiet(self):
self.thug_api.set_log_quiet()
handler = logging.getLogger().handlers[0]
assert isinstance(handler.filters[0], OpaqueFilter)
def test_vt_query(self):
assert not log.ThugOpts.vt_query
self.thug_api.set_vt_query()
assert log.ThugOpts.vt_query
def test_vt_submit(self):
assert not log.ThugOpts.vt_submit
self.thug_api.set_vt_submit()
assert log.ThugOpts.vt_submit
def test_vt_runtime_apikey(self):
assert self.thug_api.get_vt_runtime_apikey() is None
self.thug_api.set_vt_runtime_apikey('sample-key')
assert self.thug_api.get_vt_runtime_apikey() in ('sample-key', )
def test_mongodb_address(self):
assert self.thug_api.get_mongodb_address() is None
self.thug_api.set_mongodb_address('127.0.0.1:27017')
assert self.thug_api.get_mongodb_address() in ('127.0.0.1:27017', )
def test_add_htmlclassifier(self):
self.thug_api.add_htmlclassifier(self.yara_file)
rules = log.HTMLClassifier._rules
match = False
for ns in rules:
if self.yara_file in rules[ns]:
match = True
assert match is True
def test_add_urlclassifier(self):
self.thug_api.add_urlclassifier(self.yara_file)
rules = log.URLClassifier._rules
match = False
for ns in rules:
if self.yara_file in rules[ns]:
match = True
assert match is True
def test_add_jsclassifier(self):
self.thug_api.add_jsclassifier(self.yara_file)
rules = log.JSClassifier._rules
match = False
for ns in rules:
if self.yara_file in rules[ns]:
match = True
assert match is True
def test_add_vbsclassifier(self):
self.thug_api.add_vbsclassifier(self.yara_file)
rules = log.VBSClassifier._rules
match = False
for ns in rules:
if self.yara_file in rules[ns]:
match = True
assert match is True
def test_add_textclassifier(self):
self.thug_api.add_textclassifier(self.yara_file)
rules = log.TextClassifier._rules
match = False
for ns in rules:
if self.yara_file in rules[ns]:
match = True
assert match is True
def test_add_cookieclassifier(self):
self.thug_api.add_cookieclassifier(self.yara_file)
rules = log.CookieClassifier._rules
match = False
for ns in rules:
if self.yara_file in rules[ns]:
match = True
assert match is True
def test_add_sampleclassifier(self):
self.thug_api.add_sampleclassifier(self.yara_file)
rules = log.SampleClassifier._rules
match = False
for ns in rules:
if self.yara_file in rules[ns]:
match = True
assert match is True
def test_add_htmlfilter(self):
self.thug_api.add_htmlfilter(self.yara_file)
filters = log.HTMLClassifier._filters
match = False
for ns in filters:
if self.yara_file in filters[ns]:
match = True
assert match is True
def test_add_urlfilter(self):
self.thug_api.add_urlfilter(self.yara_file)
filters = log.URLClassifier._filters
match = False
for ns in filters:
if self.yara_file in filters[ns]:
match = True
assert match is True
def test_add_jsfilter(self):
self.thug_api.add_jsfilter(self.yara_file)
filters = log.JSClassifier._filters
match = False
for ns in filters:
if self.yara_file in filters[ns]:
match = True
assert match is True
def test_add_vbsfilter(self):
self.thug_api.add_vbsfilter(self.yara_file)
filters = log.VBSClassifier._filters
match = False
for ns in filters:
if self.yara_file in filters[ns]:
match = True
assert match is True
def test_add_textfilter(self):
self.thug_api.add_textfilter(self.yara_file)
filters = log.TextClassifier._filters
match = False
for ns in filters:
if self.yara_file in filters[ns]:
match = True
assert match is True
def test_add_cookiefilter(self):
self.thug_api.add_cookiefilter(self.yara_file)
filters = log.CookieClassifier._filters
match = False
for ns in filters:
if self.yara_file in filters[ns]:
match = True
assert match is True
def test_add_samplefilter(self):
self.thug_api.add_samplefilter(self.yara_file)
filters = log.SampleClassifier._filters
match = False
for ns in filters:
if self.yara_file in filters[ns]:
match = True
assert match is True
def test_log_event(self, caplog):
caplog.clear()
log.ThugOpts.file_logging = True
self.thug_api.log_event()
assert 'Thug analysis logs saved' in caplog.text
assert os.path.isdir(self.log_url)
shutil.rmtree(self.log_url)
assert not os.path.isdir(self.log_url)
def test_run_local(self):
pass
def test_analyse(self):
with pytest.raises(NotImplementedError):
self.thug_api.analyze()
def do_perform_test(self, caplog, sample, expected):
thug = ThugAPI()
thug.log_init(sample)
thug.add_htmlclassifier(os.path.join(self.signatures_path, "html_signature_1.yar"))
thug.add_textclassifier(os.path.join(self.signatures_path, "text_signature_5.yar"))
thug.add_cookieclassifier(os.path.join(self.signatures_path, "cookie_signature_8.yar"))
thug.add_sampleclassifier(os.path.join(self.signatures_path, "sample_signature_10.yar"))
thug.add_htmlfilter(os.path.join(self.signatures_path, "html_filter_2.yar"))
thug.add_jsfilter(os.path.join(self.signatures_path, "js_signature_2.yar"))
thug.add_vbsfilter(os.path.join(self.signatures_path, "vbs_signature_6.yar"))
thug.add_textfilter(os.path.join(self.signatures_path, "text_signature_5.yar"))
thug.add_cookiefilter(os.path.join(self.signatures_path, "cookie_filter_9.yar"))
thug.add_samplefilter(os.path.join(self.signatures_path, "sample_filter_11.yar"))
thug.add_htmlclassifier(os.path.join(self.signatures_path, "not_existing.yar"))
thug.add_htmlfilter(os.path.join(self.signatures_path, "not_existing.yar"))
thug.add_customclassifier('wrong_type', 'wrong_method')
thug.add_customclassifier('url', 'wrong_method')
thug.add_customclassifier('sample', self.sample_passthrough)
with open(os.path.join(self.samples_path, sample), 'r') as fd:
data = fd.read()
log.HTMLClassifier.classify(os.path.basename(sample), data)
log.TextClassifier.classify(os.path.basename(sample), data)
log.TextClassifier.classify(os.path.basename(sample), data)
log.CookieClassifier.classify(os.path.basename(sample), data)
log.CookieClassifier.classify(os.path.basename(sample), data)
log.SampleClassifier.classify(data, hashlib.md5(data).hexdigest())
log.HTMLClassifier.filter(os.path.basename(sample), data)
log.JSClassifier.filter(os.path.basename(sample), data)
log.VBSClassifier.filter(os.path.basename(sample), data)
log.TextClassifier.filter(os.path.basename(sample), data)
log.CookieClassifier.filter(os.path.basename(sample), data)
log.SampleClassifier.filter(data, hashlib.md5(data).hexdigest())
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, silverlight, expected):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_events('click,storage')
thug.disable_cert_logging()
thug.set_features_logging()
if silverlight in ('disable', ):
thug.disable_silverlight()
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, sample, expected, events = '', useragent = 'win7ie90'):
thug = ThugAPI()
thug.set_useragent(useragent)
thug.set_events(events)
thug.disable_cert_logging()
thug.set_features_logging()
thug.log_init(sample)
thug.run_local(sample)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def __init__(self, context):
ThugAPI.__init__(self)
self.misc = list()
for row in context.table:
self.misc.append(row)
def do_perform_remote_test(self, caplog, url, expected):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_threshold(2)
thug.disable_cert_logging()
thug.set_features_logging()
thug.log_init(url)
thug.add_htmlclassifier(
os.path.join(self.signatures_path, "html_signature_12.yar"))
thug.run_remote(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected, type_ = "local"):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_features_logging()
thug.set_connect_timeout(1)
thug.log_init(url)
thug.register_pyhook("DFT", "do_handle_params", self.do_handle_params_hook)
m = getattr(thug, "run_{}".format(type_))
m(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected, type_="local"):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_features_logging()
thug.set_ssl_verify()
thug.set_connect_timeout(1)
thug.add_urlclassifier(
os.path.join(self.signatures_path, "url_signature_13.yar"))
thug.register_pyhook("DFT", "do_handle_params",
self.do_handle_params_hook)
thug.register_pyhook("ThugLogging", "log_classifier",
self.log_classifier_hook)
thug.log_init(url)
m = getattr(thug, "run_{}".format(type_))
m(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected):
thug = ThugAPI()
thug.set_useragent('win7ie90')
thug.set_vt_query()
thug.set_vt_submit()
thug.disable_cert_logging()
thug.set_features_logging()
thug.set_ssl_verify()
thug.log_init(url)
thug.run_remote(url)
records = [r.message for r in caplog.records]
matches = 0
for e in expected:
for record in records:
if e in record:
matches += 1
assert matches >= len(expected)
