Endpoint discovery from handshake initiation

diff --git a/src/receive.go b/src/receive.go
index 3e649b6c0dc861444641eb5d2b6d3bb0caeab14b..f2bf70bf462f86bac8eb24bcecaa8000c6a9d07b 100644 (file)
--- 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)