tools/smp-tester: Add support for generating the LTK

1 files changed, 22 insertions, 1 deletions

diff --git a/tools/smp-tester.c b/tools/smp-tester.c
index 0398ad4c1..2793188ac 100644
--- a/tools/smp-tester.c
+++ b/tools/smp-tester.c
@@ -101,6 +101,7 @@ struct test_data {
 	uint8_t smp_pcnf[16];
 	uint8_t smp_preq[7];
 	uint8_t smp_prsp[7];
+	uint8_t smp_ltk[16];
 };
 
 struct smp_req_rsp {
@@ -278,6 +279,16 @@ static int smp_c1(uint8_t r[16], uint8_t res[16])
 	return smp_e(data->smp_tk, res, res);
 }
 
+static int smp_s1(uint8_t r1[16], uint8_t r2[16], uint8_t res[16])
+{
+	struct test_data *data = tester_get_data();
+
+	memcpy(res, r1 + 8, 8);
+	memcpy(res + 8, r2 + 8, 8);
+
+	return smp_e(data->smp_tk, res, res);
+}
+
 static void mgmt_debug(const char *str, void *user_data)
 {
 	const char *prefix = user_data;
@@ -619,7 +630,7 @@ static const void *get_pdu(const uint8_t *data)
 static bool verify_random(const uint8_t rnd[16])
 {
 	struct test_data *data = tester_get_data();
-	uint8_t confirm[16], res[16];
+	uint8_t confirm[16], res[16], key[16];
 	int err;
 
 	err = smp_c1(data->smp_rrnd, res);
@@ -633,6 +644,16 @@ static bool verify_random(const uint8_t rnd[16])
 		return false;
 	}
 
+	if (data->out) {
+		struct bthost *bthost = hciemu_client_get_host(data->hciemu);
+		smp_s1(data->smp_rrnd, data->smp_prnd, key);
+		swap128(key, data->smp_ltk);
+		bthost_le_start_encrypt(bthost, data->handle, data->smp_ltk);
+	} else {
+		smp_s1(data->smp_prnd, data->smp_rrnd, key);
+		swap128(key, data->smp_ltk);
+	}
+
 	return true;
 }