From 790d6f76ad24ec8153db9eb8dde21c1c6bc44d3f Mon Sep 17 00:00:00 2001 From: Sam Whited Date: Wed, 16 Mar 2022 11:18:11 -0400 Subject: obfsvpn: allow wrapping existing connections This can be used to add support for additional networks that we don't necessarily want to depend on in the library. Signed-off-by: Sam Whited --- listener.go | 25 ++++++++++++++++--------- 1 file changed, 16 insertions(+), 9 deletions(-) (limited to 'listener.go') diff --git a/listener.go b/listener.go index 4163cb9..593032f 100644 --- a/listener.go +++ b/listener.go @@ -38,20 +38,16 @@ func NewListenConfigCert(cert string) (*ListenConfig, error) { }, nil } -// Listen announces on the local network address. -// -// See func net.Dial for a description of the network and address parameters. -func (lc *ListenConfig) Listen(ctx context.Context, network, address string) (*Listener, error) { - ln, err := lc.ListenConfig.Listen(ctx, network, address) - if err != nil { - return nil, err - } +// Wrap takes an existing net.Listener and wraps it in a listener that is +// configured to perform the ntor handshake. +// Values from the inner net.ListenConfig are ignored. +func (lc *ListenConfig) Wrap(ctx context.Context, ln net.Listener) (*Listener, error) { args := make(pt.Args) args.Add("node-id", lc.NodeID.Hex()) args.Add("private-key", lc.PrivateKey.Hex()) seed := ntor.KeySeed{} if bytes.Equal(lc.Seed[:], seed[:]) { - _, err = rand.Read(seed[:]) + _, err := rand.Read(seed[:]) if err != nil { return nil, err } @@ -66,6 +62,17 @@ func (lc *ListenConfig) Listen(ctx context.Context, network, address string) (*L return &Listener{sf: sf, ln: ln}, nil } +// Listen announces on the local network address. +// +// See func net.Dial for a description of the network and address parameters. +func (lc *ListenConfig) Listen(ctx context.Context, network, address string) (*Listener, error) { + ln, err := lc.ListenConfig.Listen(ctx, network, address) + if err != nil { + return nil, err + } + return lc.Wrap(ctx, ln) +} + // Listener is a network listener that accepts obfuscated connections and // performs the ntor handshake on them. type Listener struct { -- cgit v1.2.3