90 lines
2.2 KiB
Python
90 lines
2.2 KiB
Python
|
#!/usr/bin/python
|
||
|
#
|
||
|
# AP WPA2-PSK tests
|
||
|
# 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 re
|
||
|
import sys
|
||
|
import time
|
||
|
|
||
|
import logging
|
||
|
|
||
|
from wpasupplicant import WpaSupplicant
|
||
|
|
||
|
def main():
|
||
|
idx = 1
|
||
|
if len(sys.argv) > 1 and sys.argv[1] == '-d':
|
||
|
logging.basicConfig(level=logging.DEBUG)
|
||
|
idx = idx + 1
|
||
|
elif len(sys.argv) > 1 and sys.argv[1] == '-q':
|
||
|
logging.basicConfig(level=logging.WARNING)
|
||
|
idx = idx + 1
|
||
|
else:
|
||
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
||
|
if len(sys.argv) > idx:
|
||
|
test_filter = sys.argv[idx]
|
||
|
else:
|
||
|
test_filter = None
|
||
|
|
||
|
dev0 = WpaSupplicant('wlan0')
|
||
|
dev1 = WpaSupplicant('wlan1')
|
||
|
dev = [ dev0, dev1 ]
|
||
|
|
||
|
for d in dev:
|
||
|
if not d.ping():
|
||
|
print d.ifname + ": No response from wpa_supplicant"
|
||
|
return
|
||
|
d.reset()
|
||
|
print "DEV: " + d.ifname + ": " + d.p2p_dev_addr()
|
||
|
|
||
|
tests = []
|
||
|
for t in os.listdir("."):
|
||
|
m = re.match(r'(test_ap_wpa2.*)\.py$', t)
|
||
|
if m:
|
||
|
print "Import test cases from " + t
|
||
|
mod = __import__(m.group(1))
|
||
|
mod.add_tests(tests)
|
||
|
|
||
|
passed = []
|
||
|
failed = []
|
||
|
|
||
|
for t in tests:
|
||
|
if test_filter:
|
||
|
#if test_filter not in t.__name__:
|
||
|
if test_filter != t.__name__:
|
||
|
continue
|
||
|
for d in dev:
|
||
|
d.reset()
|
||
|
print "START " + t.__name__
|
||
|
if t.__doc__:
|
||
|
print "Test: " + t.__doc__
|
||
|
for d in dev:
|
||
|
d.request("NOTE TEST-START " + t.__name__)
|
||
|
try:
|
||
|
t(dev)
|
||
|
passed.append(t.__name__)
|
||
|
print "PASS " + t.__name__
|
||
|
except Exception, e:
|
||
|
print e
|
||
|
failed.append(t.__name__)
|
||
|
print "FAIL " + t.__name__
|
||
|
for d in dev:
|
||
|
d.request("NOTE TEST-STOP " + t.__name__)
|
||
|
|
||
|
if not test_filter:
|
||
|
for d in dev:
|
||
|
d.reset()
|
||
|
|
||
|
print "passed tests: " + str(passed)
|
||
|
print "failed tests: " + str(failed)
|
||
|
if len(failed):
|
||
|
sys.exit(1)
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
main()
|