From cb5939f94121013fe2dc9a537769bc2540ca2ae1 Mon Sep 17 00:00:00 2001
From: Denis Kenzior <denkenz@gmail.com>
Date: Mon, 12 Jul 2021 16:44:53 -0500
Subject: [PATCH] network: Set SAE-PT into handshake

If the authenticator is SAE H2E capable, add the relevant PTs into the
handshake_state to enable use of SAE H2E.
---
 src/network.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/network.c b/src/network.c
index ce24c3ab..717e7316 100644
--- a/src/network.c
+++ b/src/network.c
@@ -425,6 +425,13 @@ static int network_set_handshake_secrets_psk(struct network *network,
 			return -ENOKEY;
 
 		handshake_state_set_passphrase(hs, network->passphrase);
+
+		if (ie_rsnxe_capable(hs->authenticator_rsnxe,
+							IE_RSNX_SAE_H2E)) {
+			l_debug("Authenticator is SAE H2E capable");
+			handshake_state_add_ecc_sae_pt(hs, network->sae_pt_19);
+			handshake_state_add_ecc_sae_pt(hs, network->sae_pt_20);
+		}
 	} else {
 		const uint8_t *psk = network_get_psk(network);