snow/wireguard-go
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

device: recheck counters while holding write lock

Browse source
This commit is contained in:
Jason A. Donenfeld 2019-10-17 15:43:06 +02:00
parent 95fbfccf60
commit ae492d1b35
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
device/noise-protocol.go
View file

@ -315,8 +315,13 @@ func (device *Device) ConsumeMessageInitiation(msg *MessageInitiation) *Peer {
handshake.chainKey = chainKey
handshake.remoteIndex = msg.Sender
handshake.remoteEphemeral = msg.Ephemeral
handshake.lastTimestamp = timestamp
handshake.lastInitiationConsumption = time.Now()
if timestamp.After(handshake.lastTimestamp) {
handshake.lastTimestamp = timestamp
}
now := time.Now()
if now.After(handshake.lastInitiationConsumption) {
handshake.lastInitiationConsumption = now
}
handshake.state = HandshakeInitiationConsumed
handshake.mutex.Unlock()