refactor(netx): towards removing connid, dialid, etc (#369)
I have verified that experiment/tor does not depend on this functionality, therefore we can safely remove it. Part of https://github.com/ooni/probe-engine/issues/897
This commit is contained in:
parent
5b73230a6d
commit
3672e14d3e
|
@ -66,7 +66,7 @@ func maybeWithMeasurementRoot(
|
||||||
// If you have others needs, manually build the chain you need.
|
// If you have others needs, manually build the chain you need.
|
||||||
func newDNSDialer(resolver dialer.Resolver) dialer.DNSDialer {
|
func newDNSDialer(resolver dialer.Resolver) dialer.DNSDialer {
|
||||||
return dialer.DNSDialer{
|
return dialer.DNSDialer{
|
||||||
Dialer: dialer.EmitterDialer{
|
Dialer: EmitterDialer{
|
||||||
Dialer: dialer.ErrorWrapperDialer{
|
Dialer: dialer.ErrorWrapperDialer{
|
||||||
Dialer: dialer.ByteCounterDialer{
|
Dialer: dialer.ByteCounterDialer{
|
||||||
Dialer: dialer.Default,
|
Dialer: dialer.Default,
|
||||||
|
|
|
@ -1,18 +1,20 @@
|
||||||
package dialer
|
package netx
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"net"
|
"net"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/connid"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/dialid"
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/dialid"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/modelx"
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/modelx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/transactionid"
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/transactionid"
|
||||||
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/dialer"
|
||||||
)
|
)
|
||||||
|
|
||||||
// EmitterDialer is a Dialer that emits events
|
// EmitterDialer is a Dialer that emits events
|
||||||
type EmitterDialer struct {
|
type EmitterDialer struct {
|
||||||
Dialer
|
dialer.Dialer
|
||||||
}
|
}
|
||||||
|
|
||||||
// DialContext implements Dialer.DialContext
|
// DialContext implements Dialer.DialContext
|
||||||
|
@ -101,3 +103,14 @@ func (c EmitterConn) Close() (err error) {
|
||||||
})
|
})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func safeLocalAddress(conn net.Conn) (s string) {
|
||||||
|
if conn != nil && conn.LocalAddr() != nil {
|
||||||
|
s = conn.LocalAddr().String()
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func safeConnID(network string, conn net.Conn) int64 {
|
||||||
|
return connid.Compute(network, safeLocalAddress(conn))
|
||||||
|
}
|
|
@ -1,4 +1,4 @@
|
||||||
package dialer_test
|
package netx
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
@ -12,7 +12,6 @@ import (
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/handlers"
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/handlers"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/modelx"
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/modelx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/transactionid"
|
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/transactionid"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/dialer"
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -24,7 +23,7 @@ func TestEmitterFailure(t *testing.T) {
|
||||||
Handler: saver,
|
Handler: saver,
|
||||||
})
|
})
|
||||||
ctx = transactionid.WithTransactionID(ctx)
|
ctx = transactionid.WithTransactionID(ctx)
|
||||||
d := dialer.EmitterDialer{Dialer: mockablex.Dialer{
|
d := EmitterDialer{Dialer: mockablex.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return nil, io.EOF
|
return nil, io.EOF
|
||||||
},
|
},
|
||||||
|
@ -83,7 +82,7 @@ func TestEmitterSuccess(t *testing.T) {
|
||||||
Handler: saver,
|
Handler: saver,
|
||||||
})
|
})
|
||||||
ctx = transactionid.WithTransactionID(ctx)
|
ctx = transactionid.WithTransactionID(ctx)
|
||||||
d := dialer.EmitterDialer{Dialer: mockablex.Dialer{
|
d := EmitterDialer{Dialer: mockablex.Dialer{
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
||||||
return &mockablex.Conn{
|
return &mockablex.Conn{
|
||||||
MockRead: func(b []byte) (int, error) {
|
MockRead: func(b []byte) (int, error) {
|
|
@ -6,10 +6,7 @@ import (
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/handlers"
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/legacy/netx/modelx"
|
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/dialer"
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/dialer"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/errorx"
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/errorx"
|
||||||
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/mockablex"
|
||||||
|
@ -117,46 +114,8 @@ func TestDNSDialerDialForManyIPSuccess(t *testing.T) {
|
||||||
conn.Close()
|
conn.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDNSDialerDialSetsDialID(t *testing.T) {
|
// TODO(bassosimone): remove the dialID etc since the only
|
||||||
saver := &handlers.SavingHandler{}
|
// test still using legacy/netx does not care.
|
||||||
ctx := modelx.WithMeasurementRoot(context.Background(), &modelx.MeasurementRoot{
|
|
||||||
Beginning: time.Now(),
|
|
||||||
Handler: saver,
|
|
||||||
})
|
|
||||||
dialer := dialer.DNSDialer{Dialer: dialer.EmitterDialer{
|
|
||||||
Dialer: mockablex.Dialer{
|
|
||||||
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
|
||||||
return &mockablex.Conn{
|
|
||||||
MockClose: func() error {
|
|
||||||
return nil
|
|
||||||
},
|
|
||||||
MockLocalAddr: func() net.Addr {
|
|
||||||
return &net.TCPAddr{}
|
|
||||||
},
|
|
||||||
}, nil
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}, Resolver: MockableResolver{
|
|
||||||
Addresses: []string{"1.1.1.1", "8.8.8.8"},
|
|
||||||
}}
|
|
||||||
conn, err := dialer.DialContext(ctx, "tcp", "dot.dns:853")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("expected nil error here")
|
|
||||||
}
|
|
||||||
if conn == nil {
|
|
||||||
t.Fatal("expected non-nil conn")
|
|
||||||
}
|
|
||||||
conn.Close()
|
|
||||||
events := saver.Read()
|
|
||||||
if len(events) != 2 {
|
|
||||||
t.Fatal("unexpected number of events")
|
|
||||||
}
|
|
||||||
for _, ev := range events {
|
|
||||||
if ev.Connect != nil && ev.Connect.DialID == 0 {
|
|
||||||
t.Fatal("unexpected DialID")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReduceErrors(t *testing.T) {
|
func TestReduceErrors(t *testing.T) {
|
||||||
t.Run("no errors", func(t *testing.T) {
|
t.Run("no errors", func(t *testing.T) {
|
||||||
|
|
Loading…
Reference in New Issue
Block a user