Mirrors/coredns
1
0
Fork 0
mirror of https://github.com/coredns/coredns.git synced 2025-11-03 10:43:20 -05:00
Code Issues Packages Projects Releases Wiki Activity

plugin/forward: check message Id when reading from upstream server (#2637)

Browse Source
This commit is contained in:
Ruslan Drozhdzh
2019-03-01 17:40:52 +03:00
committed by Miek Gieben
parent 352a042362
commit e3e3249016
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
plugin/forward/connect.go
View File

@@ -103,14 +103,21 @@ func (p *Proxy) Connect(ctx context.Context, state request.Request, opts options
return nil, err return nil, err
} }
var ret *dns.Msg
conn.SetReadDeadline(time.Now().Add(readTimeout)) conn.SetReadDeadline(time.Now().Add(readTimeout))
ret, err := conn.ReadMsg() for {
if err != nil { ret, err = conn.ReadMsg()
conn.Close() // not giving it back if err != nil {
if err == io.EOF && cached { conn.Close() // not giving it back
return nil, ErrCachedClosed if err == io.EOF && cached {
return nil, ErrCachedClosed
}
return ret, err
}
// drop out-of-order responses
if state.Req.Id == ret.Id {
break
} }
return ret, err
} }
p.transport.Yield(conn) p.transport.Yield(conn)