mirror of
https://blitiri.com.ar/repos/chasquid
synced 2025-12-16 14:27:01 +00:00
029bca7013
This patch adds a new test, which verifies the TLS tracking. Because we need to simulate SPF records, and Go does not support fully intercepting DNS lookups yet, this test relies on dnsmasq to provide a DNS resolver. In the future, once Go supports DNS lookup interception, we can get rid of this additional dependency.
37 lines
881 B
Go
37 lines
881 B
Go
// Support for overriding DNS lookups, for testing purposes.
|
|
// This is only used in tests, when the "dnsoverride" tag is active.
|
|
// It requires Go >= 1.8.
|
|
//
|
|
// +build dnsoverride
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
dnsAddr = flag.String("testing__dns_addr", "127.0.0.1:9053",
|
|
"DNS server address to use, for testing purposes only")
|
|
)
|
|
|
|
var dialer = &net.Dialer{
|
|
// We're going to talk to localhost, so have a short timeout so we fail
|
|
// fast. Otherwise the callers might hang indefinitely when trying to
|
|
// dial the DNS server.
|
|
Timeout: 2 * time.Second,
|
|
}
|
|
|
|
func dial(ctx context.Context, network, address string) (net.Conn, error) {
|
|
return dialer.DialContext(ctx, network, *dnsAddr)
|
|
}
|
|
|
|
func init() {
|
|
// Override the resolver to talk with our local server for testing.
|
|
net.DefaultResolver.PreferGo = true
|
|
net.DefaultResolver.Dial = dial
|
|
}
|