diff options
| author |   Mathias Hall-Andersen <mathias@hall-andersen.dk> | 2019-08-05 22:08:18 +0200 |
|---|---|---|
| committer | Mathias Hall-Andersen <mathias@hall-andersen.dk> | 2019-08-05 22:08:18 +0200 |
| commit | 8f488882be26badbe6ed6070e3923e857751cf24 (patch) | |
| tree | 4f841f6d31d8fd8bdf72b8f0c2a973a3c1520e2e /src/handshake/noise.rs | |
| parent | Multiple mac2 can be checked concurrently (diff) | |
| download | wireguard-rs-8f488882be26badbe6ed6070e3923e857751cf24.tar.xz wireguard-rs-8f488882be26badbe6ed6070e3923e857751cf24.zip | |
Prepare for resuse of message buffers for response
Diffstat (limited to '')
| -rw-r--r-- | src/handshake/noise.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/handshake/noise.rs b/src/handshake/noise.rs index d66304c..6532f4d 100644 --- a/src/handshake/noise.rs +++ b/src/handshake/noise.rs @@ -16,6 +16,7 @@ use generic_array::GenericArray; use super::device::Device; use super::messages::{NoiseInitiation, NoiseResponse}; +use super::messages::{TYPE_INITIATION, TYPE_RESPONSE}; use super::peer::{Peer, State}; use super::timestamp; use super::types::*; @@ -178,6 +179,7 @@ pub fn create_initiation<T: Copy, R: RngCore + CryptoRng>( let hs = INITIAL_HS; let hs = HASH!(&hs, peer.pk.as_bytes()); + msg.f_type.set(TYPE_INITIATION as u32); msg.f_sender.set(sender); // (E_priv, E_pub) := DH-Generate() @@ -325,6 +327,8 @@ pub fn create_response<T: Copy, R: RngCore + CryptoRng>( // unpack state let (receiver, eph_r_pk, hs, ck) = state; + + msg.f_type.set(TYPE_RESPONSE as u32); msg.f_sender.set(sender); msg.f_receiver.set(receiver); |