mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-13 18:31:18 +01:00
33 lines
612 B
Go
33 lines
612 B
Go
|
package netstack
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
"golang.zx2c4.com/wireguard/tun/netstack"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Dialer interface {
|
||
|
|
Dial(ctx context.Context, network, addr string) (net.Conn, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
type NSDialer struct {
|
||
|
|
net *netstack.Net
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewNSDialer(net *netstack.Net) *NSDialer {
|
||
|
|
return &NSDialer{
|
||
|
|
net: net,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *NSDialer) Dial(ctx context.Context, network, addr string) (net.Conn, error) {
|
||
|
|
log.Debugf("dialing %s %s", network, addr)
|
||
|
|
conn, err := d.net.Dial(network, addr)
|
||
|
|
if err != nil {
|
||
|
|
log.Debugf("failed to deal connection: %s", err)
|
||
|
|
}
|
||
|
|
return conn, err
|
||
|
|
}
|