From 1640a2e42415056613bc9fb3bee8fd84872fff70 Mon Sep 17 00:00:00 2001
From: Jouni Malinen <j@w1.fi>
Date: Sat, 2 Nov 2013 18:10:17 +0200
Subject: [PATCH] tests: Add test cases for SAE

Signed-hostap: Jouni Malinen <j@w1.fi>
---
 tests/hwsim/test_sae.py | 65 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 tests/hwsim/test_sae.py

diff --git a/tests/hwsim/test_sae.py b/tests/hwsim/test_sae.py
new file mode 100644
index 000000000..39d745f3b
--- /dev/null
+++ b/tests/hwsim/test_sae.py
@@ -0,0 +1,65 @@
+#!/usr/bin/python
+#
+# Test cases for SAE
+# 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 time
+import subprocess
+import logging
+logger = logging.getLogger()
+
+import hwsim_utils
+import hostapd
+
+def test_sae(dev, apdev):
+    """SAE with default group"""
+    params = hostapd.wpa2_params(ssid="test-sae",
+                                 passphrase="12345678")
+    params['wpa_key_mgmt'] = 'SAE'
+    hostapd.add_ap(apdev[0]['ifname'], params)
+
+    dev[0].request("SET sae_groups ")
+    id = dev[0].connect("test-sae", psk="12345678", key_mgmt="SAE",
+                        scan_freq="2412")
+    if dev[0].get_status_field('sae_group') != '19':
+            raise Exception("Expected default SAE group not used")
+
+def test_sae_groups(dev, apdev):
+    """SAE with all supported groups"""
+    # This would be the full list of supported groups, but groups 14-16
+    # (2048-4096 bit MODP) are a bit too slow on some VMs and can result in
+    # hitting mac80211 authentication timeout, so skip them for now.
+    #sae_groups = [ 19, 25, 26, 20, 21, 2, 5, 14, 15, 16, 22, 23, 24 ]
+    sae_groups = [ 19, 25, 26, 20, 21, 2, 5, 22, 23, 24 ]
+    groups = [str(g) for g in sae_groups]
+    params = hostapd.wpa2_params(ssid="test-sae-groups",
+                                 passphrase="12345678")
+    params['wpa_key_mgmt'] = 'SAE'
+    params['sae_groups'] = ' '.join(groups)
+    hostapd.add_ap(apdev[0]['ifname'], params)
+
+    for g in groups:
+        logger.info("Testing SAE group " + g)
+        dev[0].request("SET sae_groups " + g)
+        id = dev[0].connect("test-sae-groups", psk="12345678", key_mgmt="SAE",
+                            scan_freq="2412")
+        if dev[0].get_status_field('sae_group') != g:
+            raise Exception("Expected SAE group not used")
+        dev[0].remove_network(id)
+
+def test_sae_group_nego(dev, apdev):
+    """SAE group negotiation"""
+    params = hostapd.wpa2_params(ssid="test-sae-group-nego",
+                                 passphrase="12345678")
+    params['wpa_key_mgmt'] = 'SAE'
+    params['sae_groups'] = '19'
+    hostapd.add_ap(apdev[0]['ifname'], params)
+
+    dev[0].request("SET sae_groups 25 26 20 19")
+    dev[0].connect("test-sae-group-nego", psk="12345678", key_mgmt="SAE",
+                   scan_freq="2412")
+    if dev[0].get_status_field('sae_group') != '19':
+        raise Exception("Expected SAE group not used")