From 3175e64b6362ad553e266e877238d0d0f9af5a0d Mon Sep 17 00:00:00 2001
From: Bryan Boreham <bjboreham@gmail.com>
Date: Wed, 8 Oct 2014 14:24:29 +0100
Subject: [PATCH] Shut down multicast listener when client is shut down

---
 server/mdns_client.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/server/mdns_client.go b/server/mdns_client.go
index ffbdf82a..a6d9f544 100644
--- a/server/mdns_client.go
+++ b/server/mdns_client.go
@@ -44,6 +44,7 @@ type inflightQuery struct {
 }
 
 type MDNSClient struct {
+	server    *dns.Server
 	conn      *net.UDPConn
 	addr      *net.UDPAddr
 	inflight  map[string]*inflightQuery
@@ -82,7 +83,8 @@ func (c *MDNSClient) Start(ifi *net.Interface) error {
 		}
 	}
 
-	go dns.ActivateAndServe(nil, multicast, dns.HandlerFunc(handleMDNS))
+	c.server = &dns.Server{Listener: nil, PacketConn: multicast, Handler: dns.HandlerFunc(handleMDNS)}
+	go c.server.ActivateAndServe()
 
 	queryChan := make(chan *MDNSInteraction, 4)
 	c.queryChan = queryChan
@@ -174,6 +176,7 @@ func (c *MDNSClient) queryLoop(queryChan <-chan *MDNSInteraction) {
 			}
 			switch query.code {
 			case CShutdown:
+				c.server.Shutdown()
 				terminate = true
 			case CSendQuery:
 				err = c.handleSendQuery(query.payload.(mDNSQueryInfo))
@@ -186,7 +189,6 @@ func (c *MDNSClient) queryLoop(queryChan <-chan *MDNSInteraction) {
 			run()
 		}
 	}
-	// handle shutdown here
 }
 
 func (c *MDNSClient) handleSendQuery(q mDNSQueryInfo) (err error) {