From 9c4acb9f358f576479975f7baaf6b4ed5ec932a8 Mon Sep 17 00:00:00 2001
From: Mathias Hall-Andersen <mathias@hall-andersen.dk>
Date: Mon, 10 Jul 2017 17:20:43 +0200
Subject: [PATCH] Fixed incomming initiation bug

---
 src/main.go    | 7 +++----
 src/receive.go | 6 ++++--
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/main.go b/src/main.go
index b89af17..50140e3 100644
--- a/src/main.go
+++ b/src/main.go
@@ -18,9 +18,7 @@ func main() {
 	}
 	deviceName := os.Args[1]
 
-	// Open TUN device
-
-	// TODO: Fix capabilities
+	// open TUN device
 
 	tun, err := CreateTUN(deviceName)
 	log.Println(tun, err)
@@ -29,8 +27,9 @@ func main() {
 	}
 
 	device := NewDevice(tun, LogLevelDebug)
+	device.log.Info.Println("Starting device")
 
-	// Start configuration lister
+	// start configuration lister
 
 	socketPath := fmt.Sprintf("/var/run/wireguard/%s.sock", deviceName)
 	l, err := net.Listen("unix", socketPath)
diff --git a/src/receive.go b/src/receive.go
index 6530c47..e9c14f5 100644
--- a/src/receive.go
+++ b/src/receive.go
@@ -358,11 +358,13 @@ func (device *Device) RoutineHandshake() {
 					return
 				}
 
+				logDebug.Println("Creating response...")
+
 				outElem := device.NewOutboundElement()
 				writer := bytes.NewBuffer(outElem.data[:0])
 				binary.Write(writer, binary.LittleEndian, response)
-				elem.packet = writer.Bytes()
-				peer.mac.AddMacs(elem.packet)
+				outElem.packet = writer.Bytes()
+				peer.mac.AddMacs(outElem.packet)
 				addToOutboundQueue(peer.queue.outbound, outElem)
 
 			case MessageResponseType: