From 93d0dac77bd2aba285265c2427e9d712e26e4bdd Mon Sep 17 00:00:00 2001
From: Tim Kourt <tim.a.kourt@linux.intel.com>
Date: Wed, 14 Feb 2018 16:14:45 -0800
Subject: [PATCH] peap: handle receive for phase two data

---
 src/eap-peap.c | 63 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)

diff --git a/src/eap-peap.c b/src/eap-peap.c
index 8c9c7397..0e0dfeed 100644
--- a/src/eap-peap.c
+++ b/src/eap-peap.c
@@ -61,6 +61,12 @@ enum peap_flag {
 	PEAP_FLAG_L    = 0x80,
 };
 
+struct databuf {
+	uint8_t *data;
+	size_t len;
+	size_t capacity;
+};
+
 struct eap_peap_state {
 	enum peap_version version;
 	struct l_tls *tunnel;
@@ -68,6 +74,8 @@ struct eap_peap_state {
 	uint8_t *tx_pdu_buf;
 	size_t tx_pdu_buf_len;
 
+	struct databuf *plain_buf;
+
 	uint8_t *rx_pdu_buf;
 	size_t rx_pdu_buf_len;
 	size_t rx_pdu_buf_offset;
@@ -83,6 +91,49 @@ struct eap_peap_state {
 	char *passphrase;
 };
 
+static struct databuf *databuf_new(size_t capacity)
+{
+	struct databuf *databuf;
+
+	if (!capacity)
+		return NULL;
+
+	databuf = l_new(struct databuf, 1);
+	databuf->data = l_malloc(capacity);
+	databuf->capacity = capacity;
+
+	return databuf;
+}
+
+static void databuf_append(struct databuf *databuf, const uint8_t *data,
+								size_t data_len)
+{
+	size_t new_len;
+
+	if (!databuf)
+		return;
+
+	new_len = databuf->len + data_len;
+
+	if (new_len > databuf->capacity) {
+		databuf->capacity = new_len * 2;
+		databuf->data = l_realloc(databuf->data, databuf->capacity);
+	}
+
+	memcpy(databuf->data + databuf->len, data, data_len);
+
+	databuf->len = new_len;
+}
+
+static void databuf_free(struct databuf *databuf)
+{
+	if (!databuf)
+		return;
+
+	l_free(databuf->data);
+	l_free(databuf);
+}
+
 static void eap_peap_free_tx_buffer(struct eap_state *eap)
 {
 	struct eap_peap_state *peap = eap_get_data(eap);
@@ -119,6 +170,11 @@ static void eap_peap_free(struct eap_state *eap)
 
 	eap_peap_free_tx_buffer(eap);
 
+	if (peap->plain_buf) {
+		databuf_free(peap->plain_buf);
+		peap->plain_buf = NULL;
+	}
+
 	eap_peap_free_rx_buffer(eap);
 
 	eap_set_data(eap, NULL);
@@ -199,6 +255,13 @@ static void eap_peap_tunnel_data_send(const uint8_t *data, size_t data_len,
 static void eap_peap_tunnel_data_received(const uint8_t *data, size_t data_len,
 								void *user_data)
 {
+	struct eap_state *eap = user_data;
+	struct eap_peap_state *peap = eap_get_data(eap);
+
+	if (!peap->plain_buf)
+		peap->plain_buf = databuf_new(data_len);
+
+	databuf_append(peap->plain_buf, data, data_len);
 }
 
 static void eap_peap_tunnel_ready(const char *peer_identity, void *user_data)