Amazing-Mirror/fragattacks
1
0
Fork 0
mirror of https://github.com/vanhoefm/fragattacks.git synced 2024-11-25 00:38:24 -05:00
Code Issues Projects Releases Wiki Activity

fragattack: only import tests currently in the directory

Browse Source
This commit is contained in:
Mathy Vanhoef 2020-06-28 09:49:07 +04:00 committed by Mathy Vanhoef
parent edaf1abf56
commit 3c772c7df8
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
research/fragattack.py
View File

@ -1,8 +1,6 @@
#!/usr/bin/env python3
import glob, importlib
from fraginternals import *
from tests_common import *
from tests_qca import *
from tests_attacks import *
#TODO: Check that modified drivers are used using debugfs
#TODO: Check that atheros is using patched firmware using debugfs
@ -250,6 +248,15 @@ if __name__ == "__main__":
else:
options.inject_test_postauth = False
# Dynamically import tests depending on their availability in the directory
for test in glob("tests_*.py"):
module = importlib.import_module(test[:-3])
globals().update(
{n: getattr(module, n) for n in module.__all__} if hasattr(module, '__all__')
else
{k: v for (k, v) in module.__dict__.items() if not k.startswith('_')
})
# Construct the test
options.test = prepare_tests(options)
if options.test == None: