From eae8f9a666ce0b8e59deb1b5fa335e892d136df0 Mon Sep 17 00:00:00 2001
From: Neil Alexander <neilalexander@users.noreply.github.com>
Date: Wed, 5 Dec 2018 22:39:04 +0000
Subject: [PATCH] Try to SO_REUSEPORT on UNIX platforms

---
 src/yggdrasil/multicast.go       |  6 +++++-
 src/yggdrasil/multicast_other.go |  9 +++++++++
 src/yggdrasil/multicast_unix.go  | 22 ++++++++++++++++++++++
 3 files changed, 36 insertions(+), 1 deletion(-)
 create mode 100644 src/yggdrasil/multicast_other.go
 create mode 100644 src/yggdrasil/multicast_unix.go

diff --git a/src/yggdrasil/multicast.go b/src/yggdrasil/multicast.go
index 697744c..749dfcd 100644
--- a/src/yggdrasil/multicast.go
+++ b/src/yggdrasil/multicast.go
@@ -1,6 +1,7 @@
 package yggdrasil
 
 import (
+	"context"
 	"fmt"
 	"net"
 	"time"
@@ -35,7 +36,10 @@ func (m *multicast) start() error {
 			return err
 		}
 		listenString := fmt.Sprintf("[::]:%v", addr.Port)
-		conn, err := net.ListenPacket("udp6", listenString)
+		lc := net.ListenConfig{
+			Control: multicastReuse,
+		}
+		conn, err := lc.ListenPacket(context.Background(), "udp6", listenString)
 		if err != nil {
 			return err
 		}
diff --git a/src/yggdrasil/multicast_other.go b/src/yggdrasil/multicast_other.go
new file mode 100644
index 0000000..98fe2df
--- /dev/null
+++ b/src/yggdrasil/multicast_other.go
@@ -0,0 +1,9 @@
+// +build !linux,!darwin,!netbsd,!freebsd,!openbsd,!dragonflybsd
+
+package yggdrasil
+
+import "syscall"
+
+func multicastReuse(network string, address string, c syscall.RawConn) error {
+	return nil
+}
diff --git a/src/yggdrasil/multicast_unix.go b/src/yggdrasil/multicast_unix.go
new file mode 100644
index 0000000..9c6d1f1
--- /dev/null
+++ b/src/yggdrasil/multicast_unix.go
@@ -0,0 +1,22 @@
+// +build linux darwin netbsd freebsd openbsd dragonflybsd
+
+package yggdrasil
+
+import "syscall"
+import "golang.org/x/sys/unix"
+
+func multicastReuse(network string, address string, c syscall.RawConn) error {
+	var control error
+	var reuseport error
+
+	control = c.Control(func(fd uintptr) {
+		reuseport = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_REUSEPORT, 1)
+	})
+
+	switch {
+	case reuseport != nil:
+		return reuseport
+	default:
+		return control
+	}
+}