From a47f815ff3484903e5c7cf49a5cb85e06227c420 Mon Sep 17 00:00:00 2001
From: Jouni Malinen <j@w1.fi>
Date: Sun, 8 Jun 2014 18:17:08 +0300
Subject: [PATCH] tests: Extend RADIUS protocol testing coverage

Test RADIUS client behavior with various invalid Access-Accept messages.

Signed-off-by: Jouni Malinen <j@w1.fi>
---
 tests/hwsim/test_radius.py | 68 +++++++++++++++++++++++++++++++++-----
 1 file changed, 59 insertions(+), 9 deletions(-)

diff --git a/tests/hwsim/test_radius.py b/tests/hwsim/test_radius.py
index 5a1b79234..91098f46f 100644
--- a/tests/hwsim/test_radius.py
+++ b/tests/hwsim/test_radius.py
@@ -4,9 +4,11 @@
 # This software may be distributed under the terms of the BSD license.
 # See README for more details.
 
+import hmac
 import logging
 logger = logging.getLogger()
 import select
+import struct
 import subprocess
 import threading
 import time
@@ -593,8 +595,8 @@ def test_radius_failover(dev, apdev):
     if req_e <= req_s:
         raise Exception("Unexpected RADIUS server acct MIB value")
 
-def run_pyrad_server(srv, stop_event):
-    srv.RunWithStop(stop_event)
+def run_pyrad_server(srv, t_events):
+    srv.RunWithStop(t_events)
 
 def test_radius_protocol(dev, apdev):
     """RADIUS Authentication protocol tests with a fake server"""
@@ -611,16 +613,41 @@ def test_radius_protocol(dev, apdev):
             logger.info("Received authentication request")
             reply = self.CreateReplyPacket(pkt)
             reply.code = pyrad.packet.AccessAccept
-            # TODO: Add attributes. For now, this works as a test case for
-            # missing Message-Authenticator.
+            if self.t_events['msg_auth'].is_set():
+                logger.info("Add Message-Authenticator")
+                if self.t_events['wrong_secret'].is_set():
+                    logger.info("Use incorrect RADIUS shared secret")
+                    pw = "incorrect"
+                else:
+                    pw = reply.secret
+                hmac_obj = hmac.new(pw)
+                hmac_obj.update(struct.pack("B", reply.code))
+                hmac_obj.update(struct.pack("B", reply.id))
+
+                # reply attributes
+                reply.AddAttribute("Message-Authenticator",
+                                   "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00")
+                attrs = reply._PktEncodeAttributes()
+
+                # Length
+                flen = 4 + 16 + len(attrs)
+                hmac_obj.update(struct.pack(">H", flen))
+                hmac_obj.update(pkt.authenticator)
+                hmac_obj.update(attrs)
+                if self.t_events['double_msg_auth'].is_set():
+                    logger.info("Include two Message-Authenticator attributes")
+                else:
+                    del reply[80]
+                reply.AddAttribute("Message-Authenticator", hmac_obj.digest())
             self.SendReplyPacket(pkt.fd, reply)
 
-        def RunWithStop(self, stop_event):
+        def RunWithStop(self, t_events):
             self._poll = select.poll()
             self._fdmap = {}
             self._PrepareSockets()
+            self.t_events = t_events
 
-            while not stop_event.is_set():
+            while not t_events['stop'].is_set():
                 for (fd, event) in self._poll.poll(1000):
                     if event == select.POLLIN:
                         try:
@@ -639,8 +666,12 @@ def test_radius_protocol(dev, apdev):
                                                      "radius",
                                                      "localhost")
     srv.BindToAddress("")
-    t_stop = threading.Event()
-    t = threading.Thread(target=run_pyrad_server, args=(srv, t_stop))
+    t_events = {}
+    t_events['stop'] = threading.Event()
+    t_events['msg_auth'] = threading.Event()
+    t_events['wrong_secret'] = threading.Event()
+    t_events['double_msg_auth'] = threading.Event()
+    t = threading.Thread(target=run_pyrad_server, args=(srv, t_events))
     t.start()
 
     try:
@@ -652,6 +683,25 @@ def test_radius_protocol(dev, apdev):
         if ev is None:
             raise Exception("Timeout on EAP start")
         time.sleep(1)
+        dev[0].request("REMOVE_NETWORK all")
+        time.sleep(0.1)
+        dev[0].dump_monitor()
+        t_events['msg_auth'].set()
+        t_events['wrong_secret'].set()
+        connect(dev[0], "radius-test", wait_connect=False)
+        time.sleep(1)
+        dev[0].request("REMOVE_NETWORK all")
+        time.sleep(0.1)
+        dev[0].dump_monitor()
+        t_events['wrong_secret'].clear()
+        connect(dev[0], "radius-test", wait_connect=False)
+        time.sleep(1)
+        dev[0].request("REMOVE_NETWORK all")
+        time.sleep(0.1)
+        dev[0].dump_monitor()
+        t_events['double_msg_auth'].set()
+        connect(dev[0], "radius-test", wait_connect=False)
+        time.sleep(1)
     finally:
-        t_stop.set()
+        t_events['stop'].set()
         t.join()