From 72afa0502990b12e6a172090c3621cf35fc9f3de Mon Sep 17 00:00:00 2001
From: Arceliar <Arceliar@users.noreply.github.com>
Date: Sat, 2 May 2020 10:01:09 -0500
Subject: [PATCH] test dial/listen in the sim

---
 build                    |  2 +-
 cmd/yggdrasilsim/dial.go | 60 ++++++++++++++++++++++++++++++++++++++++
 cmd/yggdrasilsim/main.go |  4 +--
 src/yggdrasil/simlink.go |  3 +-
 4 files changed, 63 insertions(+), 6 deletions(-)
 create mode 100644 cmd/yggdrasilsim/dial.go

diff --git a/build b/build
index 6b93ca77..0d0da7e2 100755
--- a/build
+++ b/build
@@ -45,7 +45,7 @@ elif [ $ANDROID ]; then
     github.com/yggdrasil-network/yggdrasil-extras/src/mobile \
     github.com/yggdrasil-network/yggdrasil-extras/src/dummy
 else
-  for CMD in yggdrasil yggdrasilctl yggdrasilsim; do
+  for CMD in yggdrasil yggdrasilctl; do
     echo "Building: $CMD"
     go build $ARGS -ldflags="$LDFLAGS" -gcflags="$GCFLAGS" ./cmd/$CMD
 
diff --git a/cmd/yggdrasilsim/dial.go b/cmd/yggdrasilsim/dial.go
new file mode 100644
index 00000000..5713fdd7
--- /dev/null
+++ b/cmd/yggdrasilsim/dial.go
@@ -0,0 +1,60 @@
+package main
+
+import (
+	"fmt"
+	"sort"
+	"time"
+
+	"github.com/yggdrasil-network/yggdrasil-go/src/crypto"
+)
+
+func doListen(recvNode *simNode) {
+	for {
+		c, err := recvNode.listener.Accept()
+		if err != nil {
+			panic(err)
+		}
+		c.Close()
+	}
+}
+
+func dialTest(sendNode, recvNode *simNode) {
+	if sendNode.id == recvNode.id {
+		fmt.Println("Skipping dial to self")
+		return
+	}
+	var mask crypto.NodeID
+	for idx := range mask {
+		mask[idx] = 0xff
+	}
+	for {
+		c, err := sendNode.dialer.DialByNodeIDandMask(nil, &recvNode.nodeID, &mask)
+		if c != nil {
+			c.Close()
+			return
+		}
+		if err != nil {
+			fmt.Println("Dial failed:", err)
+		}
+		time.Sleep(time.Second)
+	}
+}
+
+func dialStore(store nodeStore) {
+	var nodeIdxs []int
+	for idx, n := range store {
+		nodeIdxs = append(nodeIdxs, idx)
+		go doListen(n)
+	}
+	sort.Slice(nodeIdxs, func(i, j int) bool {
+		return nodeIdxs[i] < nodeIdxs[j]
+	})
+	for _, idx := range nodeIdxs {
+		sendNode := store[idx]
+		for _, jdx := range nodeIdxs {
+			recvNode := store[jdx]
+			fmt.Printf("Dialing from node %d to node %d / %d...\n", idx, jdx, len(store))
+			dialTest(sendNode, recvNode)
+		}
+	}
+}
diff --git a/cmd/yggdrasilsim/main.go b/cmd/yggdrasilsim/main.go
index 40fd9ce4..fcbcfc97 100644
--- a/cmd/yggdrasilsim/main.go
+++ b/cmd/yggdrasilsim/main.go
@@ -9,7 +9,5 @@ import (
 
 func main() {
 	store := makeStoreSquareGrid(4)
-	var block chan struct{}
-	<-block
-	panic(store)
+	dialStore(store)
 }
diff --git a/src/yggdrasil/simlink.go b/src/yggdrasil/simlink.go
index 33332645..e846f3ba 100644
--- a/src/yggdrasil/simlink.go
+++ b/src/yggdrasil/simlink.go
@@ -3,7 +3,6 @@ package yggdrasil
 import (
 	"errors"
 	"github.com/Arceliar/phony"
-	"github.com/yggdrasil-network/yggdrasil-go/src/util"
 )
 
 type Simlink struct {
@@ -44,7 +43,7 @@ func (s *Simlink) writeMsgs(msgs [][]byte) (int, error) {
 	var size int
 	for _, msg := range msgs {
 		size += len(msg)
-		bs := append(util.GetBytes(), msg...)
+		bs := append([]byte(nil), msg...)
 		phony.Block(s, func() {
 			s.dest.Act(s, func() {
 				defer func() { recover() }()