MISP
/
PyMISPWarningLists
mirror of https://github.com/MISP/PyMISPWarningLists
2
0
Fork 0
Code Issues Releases Wiki Activity

Merge branch 'master' of github.com:MISP/PyMISPWarningLists

pull/14/head
Raphaël Vinot 2020-06-03 12:22:08 +02:00
parent 45aa17b2e4 50f3778593
commit 7c1e213192
3 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pymispwarninglists/api.py
View File

@ -102,7 +102,7 @@ class WarningList():
parsed_url = urlparse(value)
if parsed_url.hostname:
value = parsed_url.hostname
return any(value.endswith(v) for v in self.list)
return any(value == v or value.endswith("."+v.lstrip(".")) for v in self.list)
elif self.type == 'cidr':
try:
value = ip_address(value)

2
pyproject.toml
View File

@ -23,9 +23,9 @@ include = ["pymispwarninglists/data/misp-warninglists/schema.json",
[tool.poetry.dependencies]
python = "^3.6"
coveralls = "^2.0.0"
[tool.poetry.dev-dependencies]
coveralls = "^2.0.0"
codecov = "^2.0.22"
jsonschema = "^3.2.0"
nose = "^1.3.7"

8
tests/tests.py
View File

@ -43,6 +43,10 @@ class TestPyMISPWarningLists(unittest.TestCase):
results = self.warninglists.search('2001:DB8::34:1')
self.assertEqual(results[0].name, 'List of RFC 3849 CIDR blocks')
results = self.warninglists.search('1e100.net')
self.assertEqual(results[0].name, 'Top 1,000,000 most-used sites from Tranco')
self.assertTrue('List of known google domains' in [r.name for r in results])
results = self.warninglists.search('something.files.1drv.com')
self.assertEqual(results[0].name, 'Top 1,000,000 most-used sites from Tranco')
self.assertTrue('List of known microsoft domains' in [r.name for r in results])
results = self.warninglists.search('arbitrary-domain-1e100.net')
self.assertEqual(results, [])
results = self.warninglists.search('phishing.co.uk')
self.assertEqual(results, [])