aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--conn.go6
-rw-r--r--device.go2
-rw-r--r--receive.go4
3 files changed, 8 insertions, 4 deletions
diff --git a/conn.go b/conn.go
index b79b9cd..ded7419 100644
--- a/conn.go
+++ b/conn.go
@@ -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")
}
diff --git a/device.go b/device.go
index 6758cbc..7a23350 100644
--- a/device.go
+++ b/device.go
@@ -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
diff --git a/receive.go b/receive.go
index 0f8554f..29fe5e9 100644
--- a/receive.go
+++ b/receive.go
@@ -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