From 841b52b403d404625542ef460c4c5078e7ee3046 Mon Sep 17 00:00:00 2001
From: Jouni Malinen <j@w1.fi>
Date: Sat, 5 Dec 2009 21:40:44 +0200
Subject: [PATCH] Include needed header files to avoid compiler warnings
---
tests/test-milenage.c | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/tests/test-milenage.c b/tests/test-milenage.c
index d5485ae1e..b41e1a75c 100644
--- a/tests/test-milenage.c
+++ b/tests/test-milenage.c
@@ -1,6 +1,7 @@
#include "includes.h"
#include "common.h"
+#include "crypto/aes_wrap.h"
#include "crypto/milenage.h"
@@ -13,13 +14,15 @@ extern int wpa_debug_level;
* @k: K = 128-bit subscriber key
* @opc: Buffer for OPc = 128-bit value derived from OP and K
*/
-static void milenage_opc(const u8 *op, const u8 *k, u8 *opc)
+static int milenage_opc(const u8 *op, const u8 *k, u8 *opc)
{
int i;
/* OP_C = OP XOR E_K(OP) */
- aes_128_encrypt_block(k, op, opc);
+ if (aes_128_encrypt_block(k, op, opc) < 0)
+ return -1;
for (i = 0; i < 16; i++)
opc[i] ^= op[i];
+ return 0;
}