rixxc/wireguard-rs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Test for confirmation of key using staged packet

Browse Source
This commit is contained in:
Mathias Hall-Andersen
2019-09-11 16:36:38 +02:00
parent de48bec958
commit c3ad827197
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/router/tests.rs
View File

@@ -454,11 +454,18 @@ mod tests {
fn test_outbound_inbound() { fn test_outbound_inbound() {
init(); init();
let tests = [( let tests = [
false, (
false, // confirm with keepalive
("192.168.1.0", 24, "192.168.1.20", true), ("192.168.1.0", 24, "192.168.1.20", true),
("172.133.133.133", 32, "172.133.133.133", true), ("172.133.133.133", 32, "172.133.133.133", true),
)]; ),
(
true, // confirm with staged packet
("192.168.1.0", 24, "192.168.1.20", true),
("172.133.133.133", 32, "172.133.133.133", true),
),
];
for (stage, p1, p2) in tests.iter() { for (stage, p1, p2) in tests.iter() {
let (bind1, bind2) = bind_pair(); let (bind1, bind2) = bind_pair();
@@ -533,7 +540,7 @@ mod tests {
assert!(opaq1.is_empty(), "nothing should happend on peer1"); assert!(opaq1.is_empty(), "nothing should happend on peer1");
// read confirming message received by the other end ("across the internet") // read confirming message received by the other end ("across the internet")
let mut buf = vec![0u8; 1024]; let mut buf = vec![0u8; 2048];
let (len, from) = bind1.recv(&mut buf).unwrap(); let (len, from) = bind1.recv(&mut buf).unwrap();
buf.truncate(len); buf.truncate(len);
router1.recv(from, buf).unwrap(); router1.recv(from, buf).unwrap();