mirror of
https://github.com/vanhoefm/fragattacks.git
synced 2024-11-25 08:48:31 -05:00
3ed2814a1e
print and logger.info() were directing output to different locations (stdout and stderr, respectively) which resulted in buildbot showing reordered entries. Use logger consistently to avoid that. Signed-hostap: Jouni Malinen <j@w1.fi>
132 lines
3.8 KiB
Python
Executable File
132 lines
3.8 KiB
Python
Executable File
#!/usr/bin/python
|
|
#
|
|
# AP 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
|
|
logger = logging.getLogger(__name__)
|
|
|
|
from wpasupplicant import WpaSupplicant
|
|
from hostapd import HostapdGlobal
|
|
|
|
def reset_devs(dev, apdev):
|
|
hapd = HostapdGlobal()
|
|
for d in dev:
|
|
d.reset()
|
|
for ap in apdev:
|
|
hapd.remove(ap['ifname'])
|
|
|
|
def main():
|
|
test_file = None
|
|
error_file = None
|
|
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 + 1 and sys.argv[idx] == '-e':
|
|
error_file = sys.argv[idx + 1]
|
|
idx = idx + 2
|
|
|
|
if len(sys.argv) > idx + 1 and sys.argv[idx] == '-f':
|
|
test_file = sys.argv[idx + 1]
|
|
idx = idx + 2
|
|
|
|
if len(sys.argv) > idx:
|
|
test_filter = sys.argv[idx]
|
|
else:
|
|
test_filter = None
|
|
|
|
dev0 = WpaSupplicant('wlan0', '/tmp/wpas-wlan0')
|
|
dev1 = WpaSupplicant('wlan1', '/tmp/wpas-wlan1')
|
|
dev2 = WpaSupplicant('wlan2', '/tmp/wpas-wlan2')
|
|
dev = [ dev0, dev1, dev2 ]
|
|
apdev = [ ]
|
|
apdev.append({"ifname": 'wlan3', "bssid": "02:00:00:00:03:00"})
|
|
apdev.append({"ifname": 'wlan4', "bssid": "02:00:00:00:04:00"})
|
|
|
|
for d in dev:
|
|
if not d.ping():
|
|
logger.info(d.ifname + ": No response from wpa_supplicant")
|
|
return
|
|
logger.info("DEV: " + d.ifname + ": " + d.p2p_dev_addr())
|
|
for ap in apdev:
|
|
logger.info("APDEV: " + ap['ifname'])
|
|
|
|
tests = []
|
|
for t in os.listdir("."):
|
|
m = re.match(r'(test_.*)\.py$', t)
|
|
if m:
|
|
if test_file and test_file not in t:
|
|
continue
|
|
logger.info("Import test cases from " + t)
|
|
mod = __import__(m.group(1))
|
|
for s in dir(mod):
|
|
if s.startswith("test_"):
|
|
func = mod.__dict__.get(s)
|
|
tests.append(func)
|
|
|
|
passed = []
|
|
failed = []
|
|
|
|
for t in tests:
|
|
if test_filter:
|
|
if test_filter != t.__name__:
|
|
continue
|
|
reset_devs(dev, apdev)
|
|
logger.info("START " + t.__name__)
|
|
if t.__doc__:
|
|
logger.info("Test: " + t.__doc__)
|
|
for d in dev:
|
|
try:
|
|
d.request("NOTE TEST-START " + t.__name__)
|
|
except Exception, e:
|
|
logger.info("Failed to issue TEST-START before " + t.__name__ + " for " + d.ifname)
|
|
logger.info(e)
|
|
try:
|
|
if t.func_code.co_argcount > 1:
|
|
t(dev, apdev)
|
|
else:
|
|
t(dev)
|
|
passed.append(t.__name__)
|
|
logger.info("PASS " + t.__name__)
|
|
except Exception, e:
|
|
logger.info(e)
|
|
failed.append(t.__name__)
|
|
logger.info("FAIL " + t.__name__)
|
|
for d in dev:
|
|
try:
|
|
d.request("NOTE TEST-STOP " + t.__name__)
|
|
except Exception, e:
|
|
logger.info("Failed to issue TEST-STOP after " + t.__name__ + " for " + d.ifname)
|
|
logger.info(e)
|
|
|
|
if not test_filter:
|
|
reset_devs(dev, apdev)
|
|
|
|
if len(failed):
|
|
logger.info("passed " + str(len(passed)) + " test case(s)")
|
|
logger.info("failed tests: " + str(failed))
|
|
if error_file:
|
|
f = open(error_file, 'w')
|
|
f.write(str(failed) + '\n')
|
|
f.close()
|
|
sys.exit(1)
|
|
logger.info("passed all " + str(len(passed)) + " test case(s)")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|