From fb339bcc7637d127b29b55444d14b36ef0dd2e5d Mon Sep 17 00:00:00 2001
From: Rahul Rahul <rahul.rahul@intel.com>
Date: Wed, 22 Jun 2016 15:49:15 -0700
Subject: [PATCH] eapol: helper functions for REKEY_OFFLOAD

---
 src/eapol.c | 26 ++++++++++++++++++++++++++
 src/eapol.h |  8 ++++++++
 2 files changed, 34 insertions(+)

diff --git a/src/eapol.c b/src/eapol.c
index 00dbe145..bb020041 100644
--- a/src/eapol.c
+++ b/src/eapol.c
@@ -47,6 +47,7 @@ eapol_get_nonce_func_t get_nonce = NULL;
 eapol_install_tk_func_t install_tk = NULL;
 eapol_install_gtk_func_t install_gtk = NULL;
 eapol_deauthenticate_func_t deauthenticate = NULL;
+eapol_rekey_offload_func_t rekey_offload = NULL;
 enum eapol_protocol_version protocol_version = EAPOL_PROTOCOL_VERSION_2004;
 
 #define VERIFY_IS_ZERO(field)					\
@@ -1199,6 +1200,10 @@ static void eapol_handle_ptk_3_of_4(uint32_t ifindex,
 				ek->key_rsc, 6, cipher, sm->user_data);
 	}
 
+	if (rekey_offload)
+		rekey_offload(sm->ifindex, ptk->kek, ptk->kck,
+			sm->replay_counter, sm->user_data);
+
 fail:
 	l_free(step4);
 }
@@ -1528,6 +1533,22 @@ void __eapol_rx_packet(uint32_t ifindex, const uint8_t *spa, const uint8_t *aa,
 	}
 }
 
+void __eapol_update_replay_counter(uint32_t ifindex, const uint8_t *spa,
+				const uint8_t *aa, uint64_t replay_counter)
+{
+	struct eapol_sm *sm;
+
+	sm = eapol_find_sm(ifindex, spa, aa);
+
+	if (!sm)
+		return;
+
+	if (sm->replay_counter >= replay_counter)
+		return;
+
+	sm->replay_counter = replay_counter;
+}
+
 void __eapol_set_tx_packet_func(eapol_tx_packet_func_t func)
 {
 	tx_packet = func;
@@ -1558,6 +1579,11 @@ void __eapol_set_deauthenticate_func(eapol_deauthenticate_func_t func)
 	deauthenticate = func;
 }
 
+void __eapol_set_rekey_offload_func(eapol_rekey_offload_func_t func)
+{
+	rekey_offload = func;
+}
+
 struct l_io *eapol_open_pae(uint32_t index)
 {
 	struct l_io *io;
diff --git a/src/eapol.h b/src/eapol.h
index 5410af6e..d5d48f9c 100644
--- a/src/eapol.h
+++ b/src/eapol.h
@@ -120,6 +120,11 @@ typedef void (*eapol_deauthenticate_func_t)(uint32_t ifindex, const uint8_t *aa,
 						const uint8_t *spa,
 						uint16_t reason_code,
 						void *user_data);
+typedef void (*eapol_rekey_offload_func_t)(uint32_t ifindex,
+					const uint8_t *kek,
+					const uint8_t *kck,
+					uint64_t replay_counter,
+					void *user_data);
 
 bool eapol_calculate_mic(const uint8_t *kck, const struct eapol_key *frame,
 				uint8_t *mic);
@@ -168,6 +173,9 @@ void __eapol_set_protocol_version(enum eapol_protocol_version version);
 void __eapol_set_install_tk_func(eapol_install_tk_func_t func);
 void __eapol_set_install_gtk_func(eapol_install_gtk_func_t func);
 void __eapol_set_deauthenticate_func(eapol_deauthenticate_func_t func);
+void __eapol_set_rekey_offload_func(eapol_rekey_offload_func_t func);
+void __eapol_update_replay_counter(uint32_t ifindex, const uint8_t *spa,
+				const uint8_t *aa, uint64_t replay_counter);
 
 struct eapol_sm *eapol_sm_new();
 void eapol_sm_free(struct eapol_sm *sm);