def __init_core(self): log.ThugOpts = ThugOpts() log.ThugVulnModules = ThugVulnModules() log.MIMEHandler = MIMEHandler() log.SchemeHandler = SchemeHandler() log.Encoding = Encoding() log.WebTracking = WebTracking()
def __init__(self, configuration_path = thug.__configuration_path__): log.configuration_path = configuration_path log.personalities_path = os.path.join(configuration_path, "personalities") if configuration_path else None log.ThugOpts = ThugOpts() log.ThugVulnModules = ThugVulnModules() log.WebTracking = WebTracking() log.MIMEHandler = MIMEHandler() log.SchemeHandler = SchemeHandler() log.JSClassifier = JSClassifier() log.URLClassifier = URLClassifier() log.SampleClassifier = SampleClassifier() log.Encoding = Encoding()
from thug.Logging.modules.MongoDB import MongoDB from thug.ThugAPI.ThugVulnModules import ThugVulnModules from thug.Logging.ThugLogging import ThugLogging from thug.Encoding.Encoding import Encoding from thug.DOM.HTTPSession import HTTPSession configuration_path = thug.__configuration_path__ log = logging.getLogger("Thug") log.personalities_path = os.path.join(configuration_path, "personalities") if configuration_path else None log.ThugOpts = ThugOpts() log.configuration_path = configuration_path log.ThugLogging = ThugLogging(thug.__version__) log.ThugVulnModules = ThugVulnModules() log.Encoding = Encoding() log.HTTPSession = HTTPSession() class TestMongoDB: cve = "CVE-XXXX" url = "www.example.com" data = b"sample-data" desc = "sample-desc" cert = "sample-cert" file_data = {'sha1': 'b13d13733c4c9406fd0e01485bc4a34170b7d326', 'data': data, 'ssdeep': u'24:9EGtDqSyDVHNkCq4LOmvmuS+MfTAPxokCOB:97tG5DjQ4LDs+sTAPxLT', 'sha256': '459bf0aeda19633c8e757c05ee06b8121a51217cea69ce60819bb34092a296a0', 'type': 'JAR',
# coding=utf-8 import pytest from thug.Encoding.Encoding import Encoding encoding = Encoding() class TestEncoding: def test_string(self): result = encoding.detect('sample-content') assert result['encoding'] in ('ASCII', ) def test_unicode(self): result = encoding.detect(u'sample-content') assert result['encoding'] in ('ASCII', ) def test_utf8_bom(self): result = encoding.detect(b'\xEF\xBB\xBF') assert result['encoding'] in ('UTF-8-SIG', ) def test_unicode_utf8(self): with pytest.raises(Exception): encoding.detect(u'í')