76 lines
1.8 KiB
Python
76 lines
1.8 KiB
Python
#!/usr/bin/python2.7
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from ..util.process import Process
|
|
import json
|
|
|
|
class Wash(object):
|
|
''' Wrapper for Wash program. '''
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@staticmethod
|
|
def exists():
|
|
return Process.exists('wash')
|
|
|
|
@staticmethod
|
|
def check_for_wps_and_update_targets(capfile, targets):
|
|
if not Wash.exists():
|
|
return
|
|
|
|
command = [
|
|
'wash',
|
|
'-f', capfile,
|
|
'-j' # json
|
|
]
|
|
|
|
p = Process(command)
|
|
try:
|
|
p.wait()
|
|
lines = p.stdout()
|
|
except:
|
|
# Failure is acceptable
|
|
return
|
|
|
|
# Find all BSSIDs
|
|
bssids = set()
|
|
for line in lines.split('\n'):
|
|
try:
|
|
obj = json.loads(line)
|
|
bssid = obj['bssid']
|
|
locked = obj['wps_locked']
|
|
if locked != True:
|
|
bssids.add(bssid)
|
|
except:
|
|
pass
|
|
|
|
# Update targets
|
|
for t in targets:
|
|
t.wps = t.bssid.upper() in bssids
|
|
|
|
if __name__ == '__main__':
|
|
test_file = './tests/files/contains_wps_network.cap'
|
|
|
|
target_bssid = 'A4:2B:8C:16:6B:3A'
|
|
from ..model.target import Target
|
|
fields = [
|
|
'A4:2B:8C:16:6B:3A', # BSSID
|
|
'2015-05-27 19:28:44', '2015-05-27 19:28:46', # Dates
|
|
'11', # Channel
|
|
'54', # throughput
|
|
'WPA2', 'CCMP TKIP', 'PSK', # AUTH
|
|
'-58', '2', '0', '0.0.0.0', '9', # ???
|
|
'Test Router Please Ignore', # SSID
|
|
]
|
|
t = Target(fields)
|
|
targets = [t]
|
|
|
|
# Should update 'wps' field of a target
|
|
Wash.check_for_wps_and_update_targets(test_file, targets)
|
|
|
|
print('Target(BSSID={}).wps = {} (Expected: True)'.format(targets[0].bssid, targets[0].wps))
|
|
|
|
assert targets[0].wps == True
|
|
|