From: Mathias Hall-Andersen <mathias@hall-andersen.dk>
Date: Thu, 13 Jul 2017 19:29:22 +0000 (+0200)
Subject: Endpoint discovery from handshake initiation
X-Git-Tag: 0.0.20180514~141
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=0043008ad06bbeda4b037f2959f2c9cf1b55d65b;p=thirdparty%2Fwireguard-go.git

Endpoint discovery from handshake initiation
---

diff --git a/src/receive.go b/src/receive.go
index 3e649b6..f2bf70b 100644
--- a/src/receive.go
+++ b/src/receive.go
@@ -356,6 +356,12 @@ func (device *Device) RoutineHandshake() {
 					return
 				}
 
+				// update endpoint
+
+				peer.mutex.Lock()
+				peer.endpoint = elem.source
+				peer.mutex.Unlock()
+
 				// create response
 
 				response, err := device.CreateMessageResponse(peer)