yggdrasil-network/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2025-04-30 07:05:06 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix goroutine leak

Browse source
This commit is contained in:
Neil Alexander 2020-05-10 12:08:48 +01:00
parent cd93bf9e7b
commit e93612b522
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/mdns/mdns.go
View file

@ -197,7 +197,10 @@ func (m *MDNS) _updateInterfaces() {
// Work out which interfaces are new.
for n, addrs := range interfaces {
if _, ok := m._servers[n]; !ok {
for addr, intf := range addrs {
m._servers[n] = make(map[string]*mDNSServer)
}
for addr, intf := range addrs {
if _, ok := m._servers[n][addr]; !ok {
if err := m._startInterface(intf, addr); err != nil {
m.log.Errorf("Failed to start mDNS interface %s on address %s: %s", n, addr, err)
} else {
@ -236,6 +239,11 @@ func (m *MDNS) _updateInterfaces() {
}
func (m *MDNS) _startInterface(intf net.Interface, addr string) error {
// Don't start a new interface if it is already alive.
if _, ok := m._servers[intf.Name][addr]; ok {
return errors.New("already started")
}
// Construct a listener on this address.
// Work out what the listen address of the new TCP listener should be.
ip := net.ParseIP(addr)
@ -328,9 +336,8 @@ func (s *mDNSServer) listen() {
incoming := make(chan *mdns.ServiceEntry)
go func() {
defer close(s.stop)
if err := mdns.Lookup(MDNSService, MDNSDomain, incoming); err != nil {
s.mdns.log.Errorln("Failed to initialize resolver:", err.Error())
s.mdns.log.Println("Failed to initialize resolver:", err.Error())
}
}()
@ -338,7 +345,7 @@ func (s *mDNSServer) listen() {
select {
case <-s.stop:
s.mdns.log.Debugln("Stopped listening for mDNS on", s.intf.Name)
break
return
case entry := <-incoming:
if bytes.Equal(entry.Addr, s.ourIP) {
continue