You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
1.9 KiB
Python
83 lines
1.9 KiB
Python
#!/usr/bin/python
|
|
#
|
|
# Test script for wpaspy
|
|
# Copyright (c) 2013, Jouni Malinen <j@w1.fi>
|
|
#
|
|
# This software may be distributed under the terms of the BSD license.
|
|
# See README for more details.
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
import wpaspy
|
|
|
|
wpas_ctrl = '/var/run/wpa_supplicant'
|
|
|
|
def wpas_connect(host=None, port=9877):
|
|
ifaces = []
|
|
|
|
if host != None:
|
|
try:
|
|
wpas = wpaspy.Ctrl(host, port)
|
|
return wpas
|
|
except:
|
|
print("Could not connect to host: ", host)
|
|
return None
|
|
|
|
if os.path.isdir(wpas_ctrl):
|
|
try:
|
|
ifaces = [os.path.join(wpas_ctrl, i) for i in os.listdir(wpas_ctrl)]
|
|
except OSError as error:
|
|
print("Could not find wpa_supplicant: ", error)
|
|
return None
|
|
|
|
if len(ifaces) < 1:
|
|
print("No wpa_supplicant control interface found")
|
|
return None
|
|
|
|
for ctrl in ifaces:
|
|
try:
|
|
wpas = wpaspy.Ctrl(ctrl)
|
|
return wpas
|
|
except Exception as e:
|
|
pass
|
|
return None
|
|
|
|
|
|
def main(host=None, port=9877):
|
|
print("Testing wpa_supplicant control interface connection")
|
|
wpas = wpas_connect(host, port)
|
|
if wpas is None:
|
|
return
|
|
print("Connected to wpa_supplicant")
|
|
print(wpas.request('PING'))
|
|
|
|
mon = wpas_connect(host, port)
|
|
if mon is None:
|
|
print("Could not open event monitor connection")
|
|
return
|
|
|
|
mon.attach()
|
|
print("Scan")
|
|
print(wpas.request('SCAN'))
|
|
|
|
count = 0
|
|
while count < 10:
|
|
count += 1
|
|
time.sleep(1)
|
|
while mon.pending():
|
|
ev = mon.recv()
|
|
print(ev)
|
|
if 'CTRL-EVENT-SCAN-RESULTS' in ev:
|
|
print('Scan completed')
|
|
print(wpas.request('SCAN_RESULTS'))
|
|
count = 10
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 2:
|
|
main(host=sys.argv[1], port=int(sys.argv[2]))
|
|
else:
|
|
main()
|