Amazing-Mirror/fragattacks
1
0
Fork 0
mirror of https://github.com/vanhoefm/fragattacks.git synced 2025-02-26 13:49:35 -05:00
Code Issues Projects Releases Wiki Activity

fragattack: reorder some code

Browse Source
This commit is contained in:
Mathy Vanhoef 2020-08-12 12:31:03 +04:00
parent 7eed86283b
commit 0792f321bc
1 changed files with 20 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
research/fragattack.py
View File

@ -54,6 +54,8 @@ def str2actions(stractions, default):
return default return default
def prepare_tests(opt): def prepare_tests(opt):
# --------------- Main Tests ---------------
stractions = opt.actions stractions = opt.actions
if opt.testname == "ping": if opt.testname == "ping":
actions = str2actions(stractions, actions = str2actions(stractions,
@ -74,25 +76,6 @@ def prepare_tests(opt):
Action(Action.Connected, enc=True, inc_pn=0)], Action(Action.Connected, enc=True, inc_pn=0)],
separate_with=separator, opt=opt) separate_with=separator, opt=opt)
elif opt.testname == "wep-mixed-key":
log(WARNING, "Cannot predict WEP key reotation. Fragment may time out, use very short key rotation!", color="orange")
test = PingTest(REQ_ICMP,
[Action(Action.Connected, action=Action.GetIp),
Action(Action.Connected, enc=True),
# On a WEP key rotation we get a Connected event. So wait for that.
Action(Action.AfterAuth, enc=True),
])
elif opt.testname == "cache-poison":
# Cache poison attack. Worked against Linux Hostapd and RT-AC51U.
test = PingTest(REQ_ICMP,
[Action(Action.Connected, enc=True),
Action(Action.Connected, action=Action.Reconnect),
Action(Action.AfterAuth, enc=True)])
elif opt.testname == "forward":
test = ForwardTest(eapol=False, dst=stractions)
elif opt.testname in ["eapol-inject", "eapol-inject-large"]: elif opt.testname in ["eapol-inject", "eapol-inject-large"]:
large = opt.testname.endswith("-large") large = opt.testname.endswith("-large")
test = ForwardTest(eapol=True, dst=stractions, large=large) test = ForwardTest(eapol=True, dst=stractions, large=large)
@ -108,12 +91,30 @@ def prepare_tests(opt):
decoy_tid = None if stractions == None else int(stractions) decoy_tid = None if stractions == None else int(stractions)
test = LinuxTest(REQ_ICMP, decoy_tid) test = LinuxTest(REQ_ICMP, decoy_tid)
elif opt.testname in ["amsdu-inject", "amsdu-inject-bad"]:
malformed = opt.testname.endswith("-bad")
test = AmsduInject(REQ_ICMP, malformed)
elif opt.testname == "eapfrag": elif opt.testname == "eapfrag":
actions = str2actions(stractions, actions = str2actions(stractions,
[Action(Action.StartAuth, enc=False), [Action(Action.StartAuth, enc=False),
Action(Action.StartAuth, enc=False)]) Action(Action.StartAuth, enc=False)])
test = BcastEapFragTest(REQ_ICMP, actions, opt.bcast_dst) test = BcastEapFragTest(REQ_ICMP, actions, opt.bcast_dst)
elif opt.testname == "wep-mixed-key":
log(WARNING, "Cannot predict WEP key reotation. Fragment may time out, use very short key rotation!", color="orange")
test = PingTest(REQ_ICMP,
[Action(Action.Connected, action=Action.GetIp),
Action(Action.Connected, enc=True),
# On a WEP key rotation we get a Connected event. So wait for that.
Action(Action.AfterAuth, enc=True),
])
# --------------- Research Tests ---------------
elif opt.testname == "forward":
test = ForwardTest(eapol=False, dst=stractions)
elif opt.testname == "qca-test": elif opt.testname == "qca-test":
test = QcaDriverTest() test = QcaDriverTest()
@ -123,10 +124,6 @@ def prepare_tests(opt):
elif opt.testname == "qca-rekey": elif opt.testname == "qca-rekey":
test = QcaDriverRekey() test = QcaDriverRekey()
elif opt.testname in ["amsdu-inject", "amsdu-inject-bad"]:
malformed = opt.testname.endswith("-bad")
test = AmsduInject(REQ_ICMP, malformed)
# No valid test ID/name was given # No valid test ID/name was given
else: return None else: return None