From a3bdfccb049109cdfecae792f87a1c75031bf801 Mon Sep 17 00:00:00 2001
From: Matt Johnston <matt@ucc.asn.au>
Date: Thu, 28 Dec 2023 22:52:09 +0800
Subject: [PATCH] Fix tests after adding strict kex
---
src/encrypt.rs | 1 +
src/kex.rs | 4 ++--
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/encrypt.rs b/src/encrypt.rs
index fb1c633..5d5891f 100644
--- a/src/encrypt.rs
+++ b/src/encrypt.rs
@@ -749,6 +749,7 @@ mod tests {
discard_next: false,
is_client: false,
send_ext_info: true,
+ strict_kex: false,
})
})
// and plaintext
diff --git a/src/kex.rs b/src/kex.rs
index 69d5d31..f9e81f0 100644
--- a/src/kex.rs
+++ b/src/kex.rs
@@ -991,8 +991,8 @@ mod tests {
let serv_init = if let Packet::KexInit(k) = serv_init { k } else { panic!() };
assert!(ts.next().is_none());
- serv.handle_kexinit(cli_init, false, &serv_conf, &version, &mut ts.sender()).unwrap();
- cli.handle_kexinit(serv_init, true, &cli_conf, &version, &mut tc.sender()).unwrap();
+ serv.handle_kexinit(cli_init, false, &serv_conf, &version, true, &mut ts.sender()).unwrap();
+ cli.handle_kexinit(serv_init, true, &cli_conf, &version, true, &mut tc.sender()).unwrap();
let cli_dhinit = tc.next().unwrap();
let cli_dhinit = if let Packet::KexDHInit(k) = cli_dhinit { k } else { panic!() };
--
GitLab