Skip to content

Commit 2127365

Browse files
Rust wrapper: fix intermittent test_dh failure
1 parent 67c2d80 commit 2127365

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

wrapper/rust/wolfssl/tests/test_dh.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,19 +176,23 @@ fn test_dh_shared_secret() {
176176
let mut public0 = [0u8; 256];
177177
let mut public0_size = 0u32;
178178
dh.generate_key_pair(&mut rng, &mut private0, &mut private0_size, &mut public0, &mut public0_size).expect("Error with generate_key_pair()");
179+
let private0 = &private0[0..(private0_size as usize)];
180+
let public0 = &public0[0..(public0_size as usize)];
179181

180182
let mut private1 = [0u8; 256];
181183
let mut private1_size = 0u32;
182184
let mut public1 = [0u8; 256];
183185
let mut public1_size = 0u32;
184186
dh.generate_key_pair(&mut rng, &mut private1, &mut private1_size, &mut public1, &mut public1_size).expect("Error with generate_key_pair()");
187+
let private1 = &private1[0..(private1_size as usize)];
188+
let public1 = &public1[0..(public1_size as usize)];
185189

186190
let mut ss0 = [0u8; 256];
187-
let ss0_size = dh.shared_secret(&mut ss0, &private0, &public1).expect("Error with shared_secret()");
191+
let ss0_size = dh.shared_secret(&mut ss0, private0, public1).expect("Error with shared_secret()");
188192
let ss0 = &ss0[0..ss0_size];
189193

190194
let mut ss1 = [0u8; 256];
191-
let ss1_size = dh.shared_secret(&mut ss1, &private1, &public0).expect("Error with shared_secret()");
195+
let ss1_size = dh.shared_secret(&mut ss1, private1, public0).expect("Error with shared_secret()");
192196
let ss1 = &ss1[0..ss1_size];
193197

194198
assert_eq!(ss0_size, ss1_size);

0 commit comments

Comments
 (0)