diff options
Diffstat (limited to 'middleware')
| -rw-r--r-- | middleware/proxy/client.go | 6 | ||||
| -rw-r--r-- | middleware/proxy/proxy_test.go | 3 |
2 files changed, 6 insertions, 3 deletions
diff --git a/middleware/proxy/client.go b/middleware/proxy/client.go index b02e8864b..56accd659 100644 --- a/middleware/proxy/client.go +++ b/middleware/proxy/client.go @@ -83,8 +83,14 @@ func (c *client) exchange(m *dns.Msg, co net.Conn) (dns.Msg, error) { dnsco := &dns.Conn{Conn: co, UDPSize: udpsize} + writeDeadline := time.Now().Add(defaultTimeout) + dnsco.SetWriteDeadline(writeDeadline) dnsco.WriteMsg(m) + + readDeadline := time.Now().Add(defaultTimeout) + co.SetReadDeadline(readDeadline) r, err := dnsco.ReadMsg() + dnsco.Close() if r == nil { return dns.Msg{}, err diff --git a/middleware/proxy/proxy_test.go b/middleware/proxy/proxy_test.go deleted file mode 100644 index faeef1858..000000000 --- a/middleware/proxy/proxy_test.go +++ /dev/null @@ -1,3 +0,0 @@ -package proxy - -/* TODO */ |