413 lines
16 KiB
Python
413 lines
16 KiB
Python
# Cipher suite tests
|
|
# Copyright (c) 2013-2015, Jouni Malinen <j@w1.fi>
|
|
#
|
|
# This software may be distributed under the terms of the BSD license.
|
|
# See README for more details.
|
|
|
|
from remotehost import remote_compatible
|
|
import time
|
|
import logging
|
|
logger = logging.getLogger()
|
|
import os
|
|
import subprocess
|
|
|
|
import hwsim_utils
|
|
import hostapd
|
|
from utils import HwsimSkip, skip_with_fips, require_under_vm
|
|
from wlantest import Wlantest
|
|
|
|
def check_cipher(dev, ap, cipher):
|
|
if cipher not in dev.get_capability("pairwise"):
|
|
raise HwsimSkip("Cipher %s not supported" % cipher)
|
|
params = { "ssid": "test-wpa2-psk",
|
|
"wpa_passphrase": "12345678",
|
|
"wpa": "2",
|
|
"wpa_key_mgmt": "WPA-PSK",
|
|
"rsn_pairwise": cipher }
|
|
hapd = hostapd.add_ap(ap, params)
|
|
dev.connect("test-wpa2-psk", psk="12345678",
|
|
pairwise=cipher, group=cipher, scan_freq="2412")
|
|
hwsim_utils.test_connectivity(dev, hapd)
|
|
|
|
def check_group_mgmt_cipher(dev, ap, cipher, sta_req_cipher=None):
|
|
if cipher not in dev.get_capability("group_mgmt"):
|
|
raise HwsimSkip("Cipher %s not supported" % cipher)
|
|
params = { "ssid": "test-wpa2-psk-pmf",
|
|
"wpa_passphrase": "12345678",
|
|
"wpa": "2",
|
|
"ieee80211w": "2",
|
|
"wpa_key_mgmt": "WPA-PSK-SHA256",
|
|
"rsn_pairwise": "CCMP",
|
|
"group_mgmt_cipher": cipher }
|
|
hapd = hostapd.add_ap(ap, params)
|
|
|
|
Wlantest.setup(hapd)
|
|
wt = Wlantest()
|
|
wt.flush()
|
|
wt.add_passphrase("12345678")
|
|
|
|
dev.connect("test-wpa2-psk-pmf", psk="12345678", ieee80211w="2",
|
|
key_mgmt="WPA-PSK-SHA256", group_mgmt=sta_req_cipher,
|
|
pairwise="CCMP", group="CCMP", scan_freq="2412")
|
|
hwsim_utils.test_connectivity(dev, hapd)
|
|
hapd.request("DEAUTHENTICATE ff:ff:ff:ff:ff:ff")
|
|
dev.wait_disconnected()
|
|
if wt.get_bss_counter('valid_bip_mmie', ap['bssid']) < 1:
|
|
raise Exception("No valid BIP MMIE seen")
|
|
if wt.get_bss_counter('bip_deauth', ap['bssid']) < 1:
|
|
raise Exception("No valid BIP deauth seen")
|
|
|
|
if cipher == "AES-128-CMAC":
|
|
group_mgmt = "BIP"
|
|
else:
|
|
group_mgmt = cipher
|
|
res = wt.info_bss('group_mgmt', ap['bssid']).strip()
|
|
if res != group_mgmt:
|
|
raise Exception("Unexpected group mgmt cipher: " + res)
|
|
|
|
@remote_compatible
|
|
def test_ap_cipher_tkip(dev, apdev):
|
|
"""WPA2-PSK/TKIP connection"""
|
|
skip_with_fips(dev[0])
|
|
check_cipher(dev[0], apdev[0], "TKIP")
|
|
|
|
@remote_compatible
|
|
def test_ap_cipher_tkip_countermeasures_ap(dev, apdev):
|
|
"""WPA-PSK/TKIP countermeasures (detected by AP)"""
|
|
skip_with_fips(dev[0])
|
|
testfile = "/sys/kernel/debug/ieee80211/%s/netdev:%s/tkip_mic_test" % (dev[0].get_driver_status_field("phyname"), dev[0].ifname)
|
|
if dev[0].cmd_execute([ "ls", testfile ])[0] != 0:
|
|
raise HwsimSkip("tkip_mic_test not supported in mac80211")
|
|
|
|
params = { "ssid": "tkip-countermeasures",
|
|
"wpa_passphrase": "12345678",
|
|
"wpa": "1",
|
|
"wpa_key_mgmt": "WPA-PSK",
|
|
"wpa_pairwise": "TKIP" }
|
|
hapd = hostapd.add_ap(apdev[0], params)
|
|
|
|
dev[0].connect("tkip-countermeasures", psk="12345678",
|
|
pairwise="TKIP", group="TKIP", scan_freq="2412")
|
|
|
|
dev[0].dump_monitor()
|
|
dev[0].cmd_execute([ "echo", "-n", apdev[0]['bssid'], ">", testfile ],
|
|
shell=True)
|
|
ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=1)
|
|
if ev is not None:
|
|
raise Exception("Unexpected disconnection on first Michael MIC failure")
|
|
|
|
dev[0].cmd_execute([ "echo", "-n", "ff:ff:ff:ff:ff:ff", ">", testfile ],
|
|
shell=True)
|
|
ev = dev[0].wait_disconnected(timeout=10,
|
|
error="No disconnection after two Michael MIC failures")
|
|
if "reason=14" not in ev:
|
|
raise Exception("Unexpected disconnection reason: " + ev)
|
|
ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=1)
|
|
if ev is not None:
|
|
raise Exception("Unexpected connection during TKIP countermeasures")
|
|
|
|
@remote_compatible
|
|
def test_ap_cipher_tkip_countermeasures_sta(dev, apdev):
|
|
"""WPA-PSK/TKIP countermeasures (detected by STA)"""
|
|
skip_with_fips(dev[0])
|
|
params = { "ssid": "tkip-countermeasures",
|
|
"wpa_passphrase": "12345678",
|
|
"wpa": "1",
|
|
"wpa_key_mgmt": "WPA-PSK",
|
|
"wpa_pairwise": "TKIP" }
|
|
hapd = hostapd.add_ap(apdev[0], params)
|
|
|
|
testfile = "/sys/kernel/debug/ieee80211/%s/netdev:%s/tkip_mic_test" % (hapd.get_driver_status_field("phyname"), apdev[0]['ifname'])
|
|
if hapd.cmd_execute([ "ls", testfile ])[0] != 0:
|
|
raise HwsimSkip("tkip_mic_test not supported in mac80211")
|
|
|
|
dev[0].connect("tkip-countermeasures", psk="12345678",
|
|
pairwise="TKIP", group="TKIP", scan_freq="2412")
|
|
|
|
dev[0].dump_monitor()
|
|
hapd.cmd_execute([ "echo", "-n", dev[0].own_addr(), ">", testfile ],
|
|
shell=True)
|
|
ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=1)
|
|
if ev is not None:
|
|
raise Exception("Unexpected disconnection on first Michael MIC failure")
|
|
|
|
hapd.cmd_execute([ "echo", "-n", "ff:ff:ff:ff:ff:ff", ">", testfile ],
|
|
shell=True)
|
|
ev = dev[0].wait_disconnected(timeout=10,
|
|
error="No disconnection after two Michael MIC failures")
|
|
if "reason=14 locally_generated=1" not in ev:
|
|
raise Exception("Unexpected disconnection reason: " + ev)
|
|
ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=1)
|
|
if ev is not None:
|
|
raise Exception("Unexpected connection during TKIP countermeasures")
|
|
|
|
def test_ap_cipher_tkip_countermeasures_sta2(dev, apdev, params):
|
|
"""WPA-PSK/TKIP countermeasures (detected by two STAs) [long]"""
|
|
if not params['long']:
|
|
raise HwsimSkip("Skip test case with long duration due to --long not specified")
|
|
skip_with_fips(dev[0])
|
|
params = { "ssid": "tkip-countermeasures",
|
|
"wpa_passphrase": "12345678",
|
|
"wpa": "1",
|
|
"wpa_key_mgmt": "WPA-PSK",
|
|
"wpa_pairwise": "TKIP" }
|
|
hapd = hostapd.add_ap(apdev[0], params)
|
|
|
|
testfile = "/sys/kernel/debug/ieee80211/%s/netdev:%s/tkip_mic_test" % (hapd.get_driver_status_field("phyname"), apdev[0]['ifname'])
|
|
if hapd.cmd_execute([ "ls", testfile ])[0] != 0:
|
|
raise HwsimSkip("tkip_mic_test not supported in mac80211")
|
|
|
|
dev[0].connect("tkip-countermeasures", psk="12345678",
|
|
pairwise="TKIP", group="TKIP", scan_freq="2412")
|
|
dev[0].dump_monitor()
|
|
id = dev[1].connect("tkip-countermeasures", psk="12345678",
|
|
pairwise="TKIP", group="TKIP", scan_freq="2412")
|
|
dev[1].dump_monitor()
|
|
|
|
hapd.cmd_execute([ "echo", "-n", "ff:ff:ff:ff:ff:ff", ">", testfile ],
|
|
shell=True)
|
|
ev = dev[0].wait_disconnected(timeout=10,
|
|
error="No disconnection after two Michael MIC failure")
|
|
if "reason=14" not in ev:
|
|
raise Exception("Unexpected disconnection reason: " + ev)
|
|
ev = dev[1].wait_disconnected(timeout=5,
|
|
error="No disconnection after two Michael MIC failure")
|
|
if "reason=14" not in ev:
|
|
raise Exception("Unexpected disconnection reason: " + ev)
|
|
ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=1)
|
|
if ev is not None:
|
|
raise Exception("Unexpected connection during TKIP countermeasures")
|
|
ev = dev[1].wait_event(["CTRL-EVENT-CONNECTED"], timeout=1)
|
|
if ev is not None:
|
|
raise Exception("Unexpected connection during TKIP countermeasures")
|
|
|
|
dev[0].request("REMOVE_NETWORK all")
|
|
logger.info("Waiting for TKIP countermeasures to end")
|
|
connected = False
|
|
start = os.times()[4]
|
|
while True:
|
|
now = os.times()[4]
|
|
if start + 70 < now:
|
|
break
|
|
dev[0].connect("tkip-countermeasures", psk="12345678",
|
|
pairwise="TKIP", group="TKIP", scan_freq="2412",
|
|
wait_connect=False)
|
|
ev = dev[0].wait_event(["CTRL-EVENT-AUTH-REJECT",
|
|
"CTRL-EVENT-CONNECTED"], timeout=10)
|
|
if ev is None:
|
|
raise Exception("No connection result")
|
|
if "CTRL-EVENT-CONNECTED" in ev:
|
|
connected = True
|
|
break
|
|
if "status_code=14" not in ev:
|
|
raise Exception("Unexpected connection failure reason during TKIP countermeasures: " + ev)
|
|
dev[0].request("REMOVE_NETWORK all")
|
|
time.sleep(1)
|
|
dev[0].dump_monitor()
|
|
dev[1].dump_monitor()
|
|
if not connected:
|
|
raise Exception("No connection after TKIP countermeasures terminated")
|
|
|
|
ev = dev[1].wait_event(["CTRL-EVENT-CONNECTED"], timeout=1)
|
|
if ev is None:
|
|
dev[1].request("DISCONNECT")
|
|
dev[1].select_network(id)
|
|
dev[1].wait_connected()
|
|
|
|
@remote_compatible
|
|
def test_ap_cipher_ccmp(dev, apdev):
|
|
"""WPA2-PSK/CCMP connection"""
|
|
check_cipher(dev[0], apdev[0], "CCMP")
|
|
|
|
def test_ap_cipher_gcmp(dev, apdev):
|
|
"""WPA2-PSK/GCMP connection"""
|
|
check_cipher(dev[0], apdev[0], "GCMP")
|
|
|
|
def test_ap_cipher_ccmp_256(dev, apdev):
|
|
"""WPA2-PSK/CCMP-256 connection"""
|
|
check_cipher(dev[0], apdev[0], "CCMP-256")
|
|
|
|
def test_ap_cipher_gcmp_256(dev, apdev):
|
|
"""WPA2-PSK/GCMP-256 connection"""
|
|
check_cipher(dev[0], apdev[0], "GCMP-256")
|
|
|
|
@remote_compatible
|
|
def test_ap_cipher_mixed_wpa_wpa2(dev, apdev):
|
|
"""WPA2-PSK/CCMP/ and WPA-PSK/TKIP mixed configuration"""
|
|
skip_with_fips(dev[0])
|
|
ssid = "test-wpa-wpa2-psk"
|
|
passphrase = "12345678"
|
|
params = { "ssid": ssid,
|
|
"wpa_passphrase": passphrase,
|
|
"wpa": "3",
|
|
"wpa_key_mgmt": "WPA-PSK",
|
|
"rsn_pairwise": "CCMP",
|
|
"wpa_pairwise": "TKIP" }
|
|
hapd = hostapd.add_ap(apdev[0], params)
|
|
dev[0].connect(ssid, psk=passphrase, proto="WPA2",
|
|
pairwise="CCMP", group="TKIP", scan_freq="2412")
|
|
status = dev[0].get_status()
|
|
if status['key_mgmt'] != 'WPA2-PSK':
|
|
raise Exception("Incorrect key_mgmt reported")
|
|
if status['pairwise_cipher'] != 'CCMP':
|
|
raise Exception("Incorrect pairwise_cipher reported")
|
|
if status['group_cipher'] != 'TKIP':
|
|
raise Exception("Incorrect group_cipher reported")
|
|
bss = dev[0].get_bss(apdev[0]['bssid'])
|
|
if bss['ssid'] != ssid:
|
|
raise Exception("Unexpected SSID in the BSS entry")
|
|
if "[WPA-PSK-TKIP]" not in bss['flags']:
|
|
raise Exception("Missing BSS flag WPA-PSK-TKIP")
|
|
if "[WPA2-PSK-CCMP]" not in bss['flags']:
|
|
raise Exception("Missing BSS flag WPA2-PSK-CCMP")
|
|
hwsim_utils.test_connectivity(dev[0], hapd)
|
|
|
|
dev[1].connect(ssid, psk=passphrase, proto="WPA",
|
|
pairwise="TKIP", group="TKIP", scan_freq="2412")
|
|
status = dev[1].get_status()
|
|
if status['key_mgmt'] != 'WPA-PSK':
|
|
raise Exception("Incorrect key_mgmt reported")
|
|
if status['pairwise_cipher'] != 'TKIP':
|
|
raise Exception("Incorrect pairwise_cipher reported")
|
|
if status['group_cipher'] != 'TKIP':
|
|
raise Exception("Incorrect group_cipher reported")
|
|
hwsim_utils.test_connectivity(dev[1], hapd)
|
|
hwsim_utils.test_connectivity(dev[0], dev[1])
|
|
|
|
@remote_compatible
|
|
def test_ap_cipher_bip(dev, apdev):
|
|
"""WPA2-PSK with BIP"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "AES-128-CMAC")
|
|
|
|
def test_ap_cipher_bip_req(dev, apdev):
|
|
"""WPA2-PSK with BIP required"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "AES-128-CMAC", "AES-128-CMAC")
|
|
|
|
def test_ap_cipher_bip_req2(dev, apdev):
|
|
"""WPA2-PSK with BIP required (2)"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "AES-128-CMAC",
|
|
"AES-128-CMAC BIP-GMAC-128 BIP-GMAC-256 BIP-CMAC-256")
|
|
|
|
def test_ap_cipher_bip_gmac_128(dev, apdev):
|
|
"""WPA2-PSK with BIP-GMAC-128"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "BIP-GMAC-128")
|
|
|
|
def test_ap_cipher_bip_gmac_128_req(dev, apdev):
|
|
"""WPA2-PSK with BIP-GMAC-128 required"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "BIP-GMAC-128", "BIP-GMAC-128")
|
|
|
|
def test_ap_cipher_bip_gmac_256(dev, apdev):
|
|
"""WPA2-PSK with BIP-GMAC-256"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "BIP-GMAC-256")
|
|
|
|
def test_ap_cipher_bip_gmac_256_req(dev, apdev):
|
|
"""WPA2-PSK with BIP-GMAC-256 required"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "BIP-GMAC-256", "BIP-GMAC-256")
|
|
|
|
def test_ap_cipher_bip_cmac_256(dev, apdev):
|
|
"""WPA2-PSK with BIP-CMAC-256"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "BIP-CMAC-256")
|
|
|
|
def test_ap_cipher_bip_cmac_256_req(dev, apdev):
|
|
"""WPA2-PSK with BIP-CMAC-256 required"""
|
|
check_group_mgmt_cipher(dev[0], apdev[0], "BIP-CMAC-256", "BIP-CMAC-256")
|
|
|
|
def test_ap_cipher_bip_req_mismatch(dev, apdev):
|
|
"""WPA2-PSK with BIP cipher mismatch"""
|
|
group_mgmt = dev[0].get_capability("group_mgmt")
|
|
for cipher in [ "AES-128-CMAC", "BIP-GMAC-256" ]:
|
|
if cipher not in group_mgmt:
|
|
raise HwsimSkip("Cipher %s not supported" % cipher)
|
|
|
|
params = { "ssid": "test-wpa2-psk-pmf",
|
|
"wpa_passphrase": "12345678",
|
|
"wpa": "2",
|
|
"ieee80211w": "2",
|
|
"wpa_key_mgmt": "WPA-PSK-SHA256",
|
|
"rsn_pairwise": "CCMP",
|
|
"group_mgmt_cipher": "AES-128-CMAC" }
|
|
hapd = hostapd.add_ap(apdev[0], params)
|
|
|
|
dev[0].scan_for_bss(hapd.own_addr(), 2412)
|
|
id = dev[0].connect("test-wpa2-psk-pmf", psk="12345678", ieee80211w="2",
|
|
key_mgmt="WPA-PSK-SHA256", group_mgmt="BIP-GMAC-256",
|
|
pairwise="CCMP", group="CCMP", scan_freq="2412",
|
|
wait_connect=False)
|
|
ev = dev[0].wait_event(["CTRL-EVENT-NETWORK-NOT-FOUND",
|
|
"CTRL-EVENT-CONNECTED"], timeout=10)
|
|
if ev is None:
|
|
raise Exception("Network selection result not indicated")
|
|
if "CTRL-EVENT-CONNECTED" in ev:
|
|
raise Exception("Unexpected connection")
|
|
|
|
dev[0].request("DISCONNECT")
|
|
dev[0].set_network(id, "group_mgmt", "AES-128-CMAC")
|
|
dev[0].select_network(id)
|
|
dev[0].wait_connected()
|
|
|
|
def get_rx_spec(phy, gtk=False):
|
|
keys = "/sys/kernel/debug/ieee80211/%s/keys" % (phy)
|
|
for key in os.listdir(keys):
|
|
keydir = keys + "/" + key
|
|
files = os.listdir(keydir)
|
|
if not gtk and "station" not in files:
|
|
continue
|
|
if gtk and "station" in files:
|
|
continue
|
|
with open(keydir + "/rx_spec") as f:
|
|
return f.read()
|
|
return None
|
|
|
|
def test_ap_wpa2_delayed_group_m1_retransmission(dev, apdev):
|
|
"""Delayed group M1 retransmission"""
|
|
require_under_vm()
|
|
try:
|
|
subprocess.call(['sysctl', '-w', 'net.ipv6.conf.all.disable_ipv6=1'],
|
|
stdout=open('/dev/null', 'w'))
|
|
subprocess.call(['sysctl', '-w',
|
|
'net.ipv6.conf.default.disable_ipv6=1'],
|
|
stdout=open('/dev/null', 'w'))
|
|
run_ap_wpa2_delayed_group_m1_retransmission(dev, apdev)
|
|
finally:
|
|
subprocess.call(['sysctl', '-w', 'net.ipv6.conf.all.disable_ipv6=0'],
|
|
stdout=open('/dev/null', 'w'))
|
|
subprocess.call(['sysctl', '-w',
|
|
'net.ipv6.conf.default.disable_ipv6=0'],
|
|
stdout=open('/dev/null', 'w'))
|
|
|
|
def run_ap_wpa2_delayed_group_m1_retransmission(dev, apdev):
|
|
params = hostapd.wpa2_params(ssid="test-wpa2-psk", passphrase="12345678")
|
|
hapd = hostapd.add_ap(apdev[0], params)
|
|
|
|
Wlantest.setup(hapd)
|
|
wt = Wlantest()
|
|
wt.flush()
|
|
wt.add_passphrase("12345678")
|
|
|
|
phy = dev[0].get_driver_status_field("phyname")
|
|
dev[0].connect("test-wpa2-psk", psk="12345678", scan_freq="2412")
|
|
|
|
for i in range(5):
|
|
hwsim_utils.test_connectivity(dev[0], hapd)
|
|
|
|
time.sleep(0.1)
|
|
before = get_rx_spec(phy, gtk=True).splitlines()
|
|
addr = dev[0].own_addr()
|
|
if "OK" not in hapd.request("RESEND_GROUP_M1 " + addr):
|
|
raise Exception("RESEND_GROUP_M1 failed")
|
|
time.sleep(0.1)
|
|
after = get_rx_spec(phy, gtk=True).splitlines()
|
|
|
|
if "OK" not in hapd.request("RESET_PN " + addr):
|
|
raise Exception("RESET_PN failed")
|
|
time.sleep(0.1)
|
|
hwsim_utils.test_connectivity(dev[0], hapd, timeout=1,
|
|
success_expected=False)
|
|
dev[0].request("DISCONNECT")
|
|
dev[0].wait_disconnected()
|
|
|
|
for i in range(len(before)):
|
|
b = int(before[i], 16)
|
|
a = int(after[i], 16)
|
|
if a < b:
|
|
raise Exception("RX counter decreased: idx=%d before=%d after=%d" % (i, b, a))
|