/
dnstransport.go
51 lines (40 loc) · 1.17 KB
/
dnstransport.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package netxlite
//
// Generic DNS transport code.
//
import (
"context"
"github.com/ooni/probe-engine/pkg/model"
)
// wrapDNSTransport wraps a DNSTransport to provide error wrapping.
func wrapDNSTransport(txp model.DNSTransport) (out model.DNSTransport) {
out = &dnsTransportErrWrapper{
DNSTransport: txp,
}
return
}
// dnsTransportErrWrapper wraps DNSTransport to provide error wrapping.
type dnsTransportErrWrapper struct {
DNSTransport model.DNSTransport
}
var _ model.DNSTransport = &dnsTransportErrWrapper{}
func (t *dnsTransportErrWrapper) RoundTrip(
ctx context.Context, query model.DNSQuery) (model.DNSResponse, error) {
resp, err := t.DNSTransport.RoundTrip(ctx, query)
if err != nil {
return nil, NewErrWrapper(ClassifyResolverError, DNSRoundTripOperation, err)
}
return resp, nil
}
func (t *dnsTransportErrWrapper) RequiresPadding() bool {
return t.DNSTransport.RequiresPadding()
}
func (t *dnsTransportErrWrapper) Network() string {
return t.DNSTransport.Network()
}
func (t *dnsTransportErrWrapper) Address() string {
return t.DNSTransport.Address()
}
func (t *dnsTransportErrWrapper) CloseIdleConnections() {
t.DNSTransport.CloseIdleConnections()
}