From: Jouni Malinen <j@w1.fi>
Date: Sat, 5 Dec 2009 19:40:44 +0000 (+0200)
Subject: Include needed header files to avoid compiler warnings
X-Git-Tag: hostap_0_7_1~412
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=841b52b403d404625542ef460c4c5078e7ee3046;p=thirdparty%2Fhostap.git

Include needed header files to avoid compiler warnings
---

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;
 }