|
@@ -6,6 +6,7 @@ package udp
|
|
|
import (
|
|
import (
|
|
|
"io"
|
|
"io"
|
|
|
"net/netip"
|
|
"net/netip"
|
|
|
|
|
+ "os"
|
|
|
"sync/atomic"
|
|
"sync/atomic"
|
|
|
|
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/sirupsen/logrus"
|
|
@@ -106,11 +107,11 @@ func (u *TesterConn) WriteTo(b []byte, addr netip.AddrPort) error {
|
|
|
return nil
|
|
return nil
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func (u *TesterConn) ListenOut(r EncReader) {
|
|
|
|
|
|
|
+func (u *TesterConn) ListenOut(r EncReader) error {
|
|
|
for {
|
|
for {
|
|
|
p, ok := <-u.RxPackets
|
|
p, ok := <-u.RxPackets
|
|
|
if !ok {
|
|
if !ok {
|
|
|
- return
|
|
|
|
|
|
|
+ return os.ErrClosed
|
|
|
}
|
|
}
|
|
|
r(p.From, p.Data)
|
|
r(p.From, p.Data)
|
|
|
}
|
|
}
|