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

fragattacks: avoid result from being printed twice

Browse Source
This commit is contained in:
Mathy 2020-06-07 13:30:45 -04:00 committed by Mathy Vanhoef
parent b93e980af6
commit 86fe4c7641
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
research/fraginternals.py
View File

@ -213,6 +213,7 @@ class Test(metaclass=abc.ABCMeta):
self.inc_pn = None
self.check_fn = None
self.time_completed = None
self.done = False
def next_trigger_is(self, trigger):
if len(self.actions) == 0:
@ -238,6 +239,11 @@ class Test(metaclass=abc.ABCMeta):
act = self.actions[0]
del self.actions[0]
return act
def check_finished(self):
if self.done:
return
# If this was the last action, record the time
if len(self.actions) == 0:
@ -245,8 +251,7 @@ class Test(metaclass=abc.ABCMeta):
self.time_completed = time.time()
else:
log(STATUS, "All frames sent. You must manually check if the test succeeded (see README).", color="green")
return act
self.done = True
def get_actions(self, action):
return [act for act in self.actions if act.action == action]
@ -560,6 +565,7 @@ class Station():
# Stop processing actions if requested
if act.wait: break
self.test.check_finished()
return result
def update_keys(self):