From e29f700dd639a69e77ca503e64005521d1e4f6f0 Mon Sep 17 00:00:00 2001
From: Arceliar <Arceliar@users.noreply.github.com>
Date: Thu, 7 Jun 2018 22:07:19 -0500
Subject: [PATCH] enable multicast on interfaces that come up after startup

---
 src/yggdrasil/admin.go     |  2 +-
 src/yggdrasil/multicast.go | 65 ++++++++++++++++++++------------------
 2 files changed, 36 insertions(+), 31 deletions(-)

diff --git a/src/yggdrasil/admin.go b/src/yggdrasil/admin.go
index fe9c05e..699bc08 100644
--- a/src/yggdrasil/admin.go
+++ b/src/yggdrasil/admin.go
@@ -175,7 +175,7 @@ func (a *admin) init(c *Core, listenaddr string) {
 	})
 	a.addHandler("getMulticastInterfaces", []string{}, func(in admin_info) (admin_info, error) {
 		var intfs []string
-		for _, v := range a.core.multicast.interfaces {
+		for _, v := range a.core.multicast.interfaces() {
 			intfs = append(intfs, v.Name)
 		}
 		return admin_info{"multicast_interfaces": intfs}, nil
diff --git a/src/yggdrasil/multicast.go b/src/yggdrasil/multicast.go
index c86159c..8211e42 100644
--- a/src/yggdrasil/multicast.go
+++ b/src/yggdrasil/multicast.go
@@ -7,10 +7,9 @@ import "fmt"
 import "golang.org/x/net/ipv6"
 
 type multicast struct {
-	core       *Core
-	sock       *ipv6.PacketConn
-	groupAddr  string
-	interfaces []net.Interface
+	core      *Core
+	sock      *ipv6.PacketConn
+	groupAddr string
 }
 
 func (m *multicast) init(core *Core) {
@@ -21,31 +20,7 @@ func (m *multicast) init(core *Core) {
 		return
 	}
 	// Ask the system for network interfaces
-	allifaces, err := net.Interfaces()
-	if err != nil {
-		panic(err)
-	}
-	// Work out which interfaces to announce on
-	for _, iface := range allifaces {
-		if iface.Flags&net.FlagUp == 0 {
-			// Ignore interfaces that are down
-			continue
-		}
-		if iface.Flags&net.FlagMulticast == 0 {
-			// Ignore non-multicast interfaces
-			continue
-		}
-		if iface.Flags&net.FlagPointToPoint != 0 {
-			// Ignore point-to-point interfaces
-			continue
-		}
-		for _, expr := range m.core.ifceExpr {
-			if expr.MatchString(iface.Name) {
-				m.interfaces = append(m.interfaces, iface)
-			}
-		}
-	}
-	m.core.log.Println("Found", len(m.interfaces), "multicast interface(s)")
+	m.core.log.Println("Found", len(m.interfaces()), "multicast interface(s)")
 }
 
 func (m *multicast) start() error {
@@ -75,6 +50,36 @@ func (m *multicast) start() error {
 	return nil
 }
 
+func (m *multicast) interfaces() []net.Interface {
+	// Ask the system for network interfaces
+	var interfaces []net.Interface
+	allifaces, err := net.Interfaces()
+	if err != nil {
+		panic(err)
+	}
+	// Work out which interfaces to announce on
+	for _, iface := range allifaces {
+		if iface.Flags&net.FlagUp == 0 {
+			// Ignore interfaces that are down
+			continue
+		}
+		if iface.Flags&net.FlagMulticast == 0 {
+			// Ignore non-multicast interfaces
+			continue
+		}
+		if iface.Flags&net.FlagPointToPoint != 0 {
+			// Ignore point-to-point interfaces
+			continue
+		}
+		for _, expr := range m.core.ifceExpr {
+			if expr.MatchString(iface.Name) {
+				interfaces = append(interfaces, iface)
+			}
+		}
+	}
+	return interfaces
+}
+
 func (m *multicast) announce() {
 	groupAddr, err := net.ResolveUDPAddr("udp6", m.groupAddr)
 	if err != nil {
@@ -88,7 +93,7 @@ func (m *multicast) announce() {
 		panic(err)
 	}
 	for {
-		for _, iface := range m.interfaces {
+		for _, iface := range m.interfaces() {
 			m.sock.JoinGroup(&iface, groupAddr)
 			//err := n.sock.JoinGroup(&iface, groupAddr)
 			//if err != nil { panic(err) }