diff options
-rw-r--r-- | conn.go | 6 | ||||
-rw-r--r-- | device.go | 2 | ||||
-rw-r--r-- | receive.go | 4 |
3 files changed, 8 insertions, 4 deletions
@@ -75,6 +75,7 @@ func unsafeCloseBind(device *Device) error { err = netc.bind.Close() netc.bind = nil } + netc.stopping.Wait() return err } @@ -162,10 +163,11 @@ func (device *Device) BindUpdate() error { // start receiving routines - device.state.starting.Add(ConnRoutineNumber) - device.state.stopping.Add(ConnRoutineNumber) + device.net.starting.Add(ConnRoutineNumber) + device.net.stopping.Add(ConnRoutineNumber) go device.RoutineReceiveIncoming(ipv4.Version, netc.bind) go device.RoutineReceiveIncoming(ipv6.Version, netc.bind) + device.net.starting.Wait() device.log.Debug.Println("UDP bind has been updated") } @@ -35,6 +35,8 @@ type Device struct { } net struct { + starting sync.WaitGroup + stopping sync.WaitGroup mutex sync.RWMutex bind Bind // bind interface port uint16 // listening port @@ -125,11 +125,11 @@ func (device *Device) RoutineReceiveIncoming(IP int, bind Bind) { logDebug := device.log.Debug defer func() { logDebug.Println("Routine: receive incoming IPv" + strconv.Itoa(IP) + " - stopped") - device.state.stopping.Done() + device.net.stopping.Done() }() logDebug.Println("Routine: receive incoming IPv" + strconv.Itoa(IP) + " - starting") - device.state.starting.Done() + device.net.starting.Done() // receive datagrams until conn is closed |