mirror of
https://github.com/BaaRai/ARTEX.git
synced 2026-11-13 11:40:53 +08:00
BaaRai
3528287d61
doHTTP now connects through intercept.PinnedDialer under the target scope of the request's context, so the probe uses the same pinned-address dial as the target gate. ProbeSite takes the context of the call that asks for the probe. A probe without a scope is refused; with the gate on, the probe bypasses the recording proxy so the dialer decides on the target itself. The path stays dormant: nothing calls ProbeSite on main.
34 lines
950 B
Go
34 lines
950 B
Go
package intercept
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
// DialGate exposes the pinned-dial fixture of target_dial_test.go to the tests
|
|
// of package intercept_test, which may import the packages that import
|
|
// intercept (enrich).
|
|
type DialGate struct {
|
|
g *gateFixture
|
|
rec *dialRecorder
|
|
}
|
|
|
|
func NewDialGate(t *testing.T) *DialGate {
|
|
g, rec := dialGate(t)
|
|
return &DialGate{g: g, rec: rec}
|
|
}
|
|
|
|
// Ctx returns a context carrying the fixture's target scope, gate on.
|
|
func (d *DialGate) Ctx() context.Context { return d.g.ctx() }
|
|
|
|
func (d *DialGate) Pin(host string, port int, ips ...string) { d.g.pin(host, port, ips...) }
|
|
|
|
func (d *DialGate) Resolve(host string, ips ...string) { d.g.res.set(host, ips...) }
|
|
|
|
func (d *DialGate) Exempt(urls ...string) {
|
|
d.g.ic.SetDerivedExemptions(func() []string { return urls })
|
|
}
|
|
|
|
// Calls returns every address the base dialer was asked to connect to.
|
|
func (d *DialGate) Calls() []string { return d.rec.calls() }
|