50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
#!/usr/bin/python
|
|
|
|
import time
|
|
|
|
class Attack(object):
|
|
'''
|
|
Contains functionality common to all attacks
|
|
'''
|
|
|
|
target_wait = 20
|
|
|
|
def __init__(self, target):
|
|
self.target = target
|
|
|
|
def run(self):
|
|
raise Exception("Unimplemented method: run")
|
|
|
|
def wait_for_target(self, airodump):
|
|
'''
|
|
Waits for target to appear in airodump
|
|
'''
|
|
start_time = time.time()
|
|
targets = airodump.get_targets()
|
|
while len(targets) == 0:
|
|
# Wait for target to appear in airodump.
|
|
if int(time.time() - start_time) > Attack.target_wait:
|
|
raise Exception(
|
|
"Target did not appear after %d seconds, stopping"
|
|
% Attack.target_wait)
|
|
time.sleep(1)
|
|
targets = airodump.get_targets()
|
|
continue
|
|
|
|
# Ensure this target was seen by airodump
|
|
airodump_target = None
|
|
for t in targets:
|
|
if t.bssid == self.target.bssid:
|
|
airodump_target = t
|
|
break
|
|
|
|
if airodump_target == None:
|
|
raise Exception(
|
|
'Could not find target (%s) in airodump' % self.target.bssid)
|
|
|
|
return airodump_target
|
|
|
|
|
|
if __name__ == '__main__':
|
|
pass
|