tests/hwsim/test_sae.py

   1 # Test cases for SAE
   2 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
   3 #
   4 # This software may be distributed under the terms of the BSD license.
   5 # See README for more details.
   6
   7 import time
   8 import subprocess
   9 import logging
  10 logger = logging.getLogger()
  11
  12 import hwsim_utils
  13 import hostapd
  14
  15 def test_sae(dev, apdev):
  16     """SAE with default group"""
  17     params = hostapd.wpa2_params(ssid="test-sae",
  18                                  passphrase="12345678")
  19     params['wpa_key_mgmt'] = 'SAE'
  20     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
  21     key_mgmt = hapd.get_config()['key_mgmt']
  22     if key_mgmt.split(' ')[0] != "SAE":
  23         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
  24
  25     dev[0].request("SET sae_groups ")
  26     id = dev[0].connect("test-sae", psk="12345678", key_mgmt="SAE",
  27                         scan_freq="2412")
  28     if dev[0].get_status_field('sae_group') != '19':
  29             raise Exception("Expected default SAE group not used")
  30
  31 def test_sae_groups(dev, apdev):
  32     """SAE with all supported groups"""
  33     # This would be the full list of supported groups, but groups 14-16
  34     # (2048-4096 bit MODP) are a bit too slow on some VMs and can result in
  35     # hitting mac80211 authentication timeout, so skip them for now.
  36     #sae_groups = [ 19, 25, 26, 20, 21, 2, 5, 14, 15, 16, 22, 23, 24 ]
  37     sae_groups = [ 19, 25, 26, 20, 21, 2, 5, 22, 23, 24 ]
  38     groups = [str(g) for g in sae_groups]
  39     params = hostapd.wpa2_params(ssid="test-sae-groups",
  40                                  passphrase="12345678")
  41     params['wpa_key_mgmt'] = 'SAE'
  42     params['sae_groups'] = ' '.join(groups)
  43     hostapd.add_ap(apdev[0]['ifname'], params)
  44
  45     for g in groups:
  46         logger.info("Testing SAE group " + g)
  47         dev[0].request("SET sae_groups " + g)
  48         id = dev[0].connect("test-sae-groups", psk="12345678", key_mgmt="SAE",
  49                             scan_freq="2412")
  50         if dev[0].get_status_field('sae_group') != g:
  51             raise Exception("Expected SAE group not used")
  52         dev[0].remove_network(id)
  53
  54 def test_sae_group_nego(dev, apdev):
  55     """SAE group negotiation"""
  56     params = hostapd.wpa2_params(ssid="test-sae-group-nego",
  57                                  passphrase="12345678")
  58     params['wpa_key_mgmt'] = 'SAE'
  59     params['sae_groups'] = '19'
  60     hostapd.add_ap(apdev[0]['ifname'], params)
  61
  62     dev[0].request("SET sae_groups 25 26 20 19")
  63     dev[0].connect("test-sae-group-nego", psk="12345678", key_mgmt="SAE",
  64                    scan_freq="2412")
  65     if dev[0].get_status_field('sae_group') != '19':
  66         raise Exception("Expected SAE group not used")
  67
  68 def test_sae_anti_clogging(dev, apdev):
  69     """SAE anti clogging"""
  70     params = hostapd.wpa2_params(ssid="test-sae", passphrase="12345678")
  71     params['wpa_key_mgmt'] = 'SAE'
  72     params['sae_anti_clogging_threshold'] = '1'
  73     hostapd.add_ap(apdev[0]['ifname'], params)
  74
  75     dev[0].request("SET sae_groups ")
  76     dev[1].request("SET sae_groups ")
  77     id = {}
  78     for i in range(0, 2):
  79         dev[i].scan(freq="2412")
  80         id[i] = dev[i].connect("test-sae", psk="12345678", key_mgmt="SAE",
  81                                scan_freq="2412", only_add_network=True)
  82     for i in range(0, 2):
  83         dev[i].select_network(id[i])
  84     for i in range(0, 2):
  85         ev = dev[i].wait_event(["CTRL-EVENT-CONNECTED"], timeout=10)
  86         if ev is None:
  87             raise Exception("Association with the AP timed out")
  88
  89 def test_sae_forced_anti_clogging(dev, apdev):
  90     """SAE anti clogging (forced)"""
  91     params = hostapd.wpa2_params(ssid="test-sae", passphrase="12345678")
  92     params['wpa_key_mgmt'] = 'SAE'
  93     params['sae_anti_clogging_threshold'] = '0'
  94     hostapd.add_ap(apdev[0]['ifname'], params)
  95     for i in range(0, 2):
  96         dev[i].request("SET sae_groups ")
  97         dev[i].connect("test-sae", psk="12345678", key_mgmt="SAE",
  98                        scan_freq="2412")
  99
 100 def test_sae_mixed(dev, apdev):
 101     """Mixed SAE and non-SAE network"""
 102     params = hostapd.wpa2_params(ssid="test-sae", passphrase="12345678")
 103     params['wpa_key_mgmt'] = 'SAE WPA-PSK'
 104     params['sae_anti_clogging_threshold'] = '0'
 105     hostapd.add_ap(apdev[0]['ifname'], params)
 106
 107     dev[2].connect("test-sae", psk="12345678", scan_freq="2412")
 108     for i in range(0, 2):
 109         dev[i].request("SET sae_groups ")
 110         dev[i].connect("test-sae", psk="12345678", key_mgmt="SAE",
 111                        scan_freq="2412")