d57c78bc71
This is how I did it: 1. `git clone https://github.com/ooni/probe-engine internal/engine` 2. ``` (cd internal/engine && git describe --tags) v0.23.0 ``` 3. `nvim go.mod` (merging `go.mod` with `internal/engine/go.mod` 4. `rm -rf internal/.git internal/engine/go.{mod,sum}` 5. `git add internal/engine` 6. `find . -type f -name \*.go -exec sed -i 's@/ooni/probe-engine@/ooni/probe-cli/v3/internal/engine@g' {} \;` 7. `go build ./...` (passes) 8. `go test -race ./...` (temporary failure on RiseupVPN) 9. `go mod tidy` 10. this commit message Once this piece of work is done, we can build a new version of `ooniprobe` that is using `internal/engine` directly. We need to do more work to ensure all the other functionality in `probe-engine` (e.g. making mobile packages) are still WAI. Part of https://github.com/ooni/probe/issues/1335
143 lines
2.6 KiB
Go
143 lines
2.6 KiB
Go
package resolver
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/ooni/probe-cli/v3/internal/engine/atomicx"
|
|
)
|
|
|
|
type FakeDialer struct {
|
|
Conn net.Conn
|
|
Err error
|
|
}
|
|
|
|
func (d FakeDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
|
|
time.Sleep(10 * time.Microsecond)
|
|
return d.Conn, d.Err
|
|
}
|
|
|
|
type FakeConn struct {
|
|
ReadError error
|
|
ReadData []byte
|
|
SetDeadlineError error
|
|
SetReadDeadlineError error
|
|
SetWriteDeadlineError error
|
|
WriteError error
|
|
}
|
|
|
|
func (c *FakeConn) Read(b []byte) (int, error) {
|
|
if len(c.ReadData) > 0 {
|
|
n := copy(b, c.ReadData)
|
|
c.ReadData = c.ReadData[n:]
|
|
return n, nil
|
|
}
|
|
if c.ReadError != nil {
|
|
return 0, c.ReadError
|
|
}
|
|
return 0, io.EOF
|
|
}
|
|
|
|
func (c *FakeConn) Write(b []byte) (n int, err error) {
|
|
if c.WriteError != nil {
|
|
return 0, c.WriteError
|
|
}
|
|
n = len(b)
|
|
return
|
|
}
|
|
|
|
func (*FakeConn) Close() (err error) {
|
|
return
|
|
}
|
|
|
|
func (*FakeConn) LocalAddr() net.Addr {
|
|
return &net.TCPAddr{}
|
|
}
|
|
|
|
func (*FakeConn) RemoteAddr() net.Addr {
|
|
return &net.TCPAddr{}
|
|
}
|
|
|
|
func (c *FakeConn) SetDeadline(t time.Time) (err error) {
|
|
return c.SetDeadlineError
|
|
}
|
|
|
|
func (c *FakeConn) SetReadDeadline(t time.Time) (err error) {
|
|
return c.SetReadDeadlineError
|
|
}
|
|
|
|
func (c *FakeConn) SetWriteDeadline(t time.Time) (err error) {
|
|
return c.SetWriteDeadlineError
|
|
}
|
|
|
|
type FakeTransport struct {
|
|
Data []byte
|
|
Err error
|
|
}
|
|
|
|
func (ft FakeTransport) RoundTrip(ctx context.Context, query []byte) ([]byte, error) {
|
|
return ft.Data, ft.Err
|
|
}
|
|
|
|
func (ft FakeTransport) RequiresPadding() bool {
|
|
return false
|
|
}
|
|
|
|
func (ft FakeTransport) Address() string {
|
|
return ""
|
|
}
|
|
|
|
func (ft FakeTransport) Network() string {
|
|
return "fake"
|
|
}
|
|
|
|
type FakeEncoder struct {
|
|
Data []byte
|
|
Err error
|
|
}
|
|
|
|
func (fe FakeEncoder) Encode(domain string, qtype uint16, padding bool) ([]byte, error) {
|
|
return fe.Data, fe.Err
|
|
}
|
|
|
|
type FakeResolver struct {
|
|
NumFailures *atomicx.Int64
|
|
Err error
|
|
Result []string
|
|
}
|
|
|
|
func NewFakeResolverThatFails() FakeResolver {
|
|
return FakeResolver{NumFailures: atomicx.NewInt64(), Err: errNotFound}
|
|
}
|
|
|
|
func NewFakeResolverWithResult(r []string) FakeResolver {
|
|
return FakeResolver{NumFailures: atomicx.NewInt64(), Result: r}
|
|
}
|
|
|
|
var errNotFound = &net.DNSError{
|
|
Err: "no such host",
|
|
}
|
|
|
|
func (c FakeResolver) LookupHost(ctx context.Context, hostname string) ([]string, error) {
|
|
time.Sleep(10 * time.Microsecond)
|
|
if c.Err != nil {
|
|
if c.NumFailures != nil {
|
|
c.NumFailures.Add(1)
|
|
}
|
|
return nil, c.Err
|
|
}
|
|
return c.Result, nil
|
|
}
|
|
|
|
func (c FakeResolver) Network() string {
|
|
return "fake"
|
|
}
|
|
|
|
func (c FakeResolver) Address() string {
|
|
return ""
|
|
}
|
|
|
|
var _ Resolver = FakeResolver{}
|