chunyu/sing-box
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-box.git synced 2025-06-13 21:54:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix UDP DNS not saving connections

Browse Source
This commit is contained in:
世界 2025-02-05 09:27:11 +08:00
parent 6158af1858
commit bc200170b2
No known key found for this signature in database
GPG Key ID: CD109927C34A63C4
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
dns/transport/udp.go
View File

@ -144,6 +144,13 @@ func (t *UDPTransport) exchange(ctx context.Context, message *mDNS.Msg) (*mDNS.M
func (t *UDPTransport) open(ctx context.Context) (*dnsConnection, error) { func (t *UDPTransport) open(ctx context.Context) (*dnsConnection, error) {
t.access.Lock() t.access.Lock()
defer t.access.Unlock() defer t.access.Unlock()
if t.conn != nil {
select {
case <-t.conn.done:
default:
return t.conn, nil
}
}
conn, err := t.dialer.DialContext(ctx, N.NetworkUDP, t.serverAddr) conn, err := t.dialer.DialContext(ctx, N.NetworkUDP, t.serverAddr)
if err != nil { if err != nil {
return nil, err return nil, err
@ -154,6 +161,7 @@ func (t *UDPTransport) open(ctx context.Context) (*dnsConnection, error) {
callbacks: make(map[uint16]*dnsCallback), callbacks: make(map[uint16]*dnsCallback),
} }
go t.recvLoop(dnsConn) go t.recvLoop(dnsConn)
t.conn = dnsConn
return dnsConn, nil return dnsConn, nil
} }
@ -201,8 +209,6 @@ type dnsConnection struct {
} }
func (c *dnsConnection) Close(err error) { func (c *dnsConnection) Close(err error) {
c.access.Lock()
defer c.access.Unlock()
c.closeOnce.Do(func() { c.closeOnce.Do(func() {
close(c.done) close(c.done)
c.err = err c.err = err