1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
package virtualnetwork
import (
"context"
"errors"
"net"
"strconv"
"gvisor.dev/gvisor/pkg/tcpip"
"gvisor.dev/gvisor/pkg/tcpip/adapters/gonet"
"gvisor.dev/gvisor/pkg/tcpip/network/ipv4"
)
func (n *VirtualNetwork) Dial(network, addr string) (net.Conn, error) {
ip, port, err := splitIPPort(network, addr)
if err != nil {
return nil, err
}
return gonet.DialTCP(n.stack, tcpip.FullAddress{
NIC: 1,
Addr: tcpip.AddrFrom4Slice(ip.To4()),
Port: uint16(port),
}, ipv4.ProtocolNumber)
}
func (n *VirtualNetwork) DialContextTCP(ctx context.Context, addr string) (net.Conn, error) {
ip, port, err := splitIPPort("tcp", addr)
if err != nil {
return nil, err
}
return gonet.DialContextTCP(ctx, n.stack,
tcpip.FullAddress{
NIC: 1,
Addr: tcpip.AddrFrom4Slice(ip.To4()),
Port: uint16(port),
}, ipv4.ProtocolNumber)
}
func (n *VirtualNetwork) Listen(network, addr string) (net.Listener, error) {
ip, port, err := splitIPPort(network, addr)
if err != nil {
return nil, err
}
return gonet.ListenTCP(n.stack, tcpip.FullAddress{
NIC: 1,
Addr: tcpip.AddrFrom4Slice(ip.To4()),
Port: uint16(port),
}, ipv4.ProtocolNumber)
}
func splitIPPort(network string, addr string) (net.IP, uint64, error) {
if network != "tcp" {
return nil, 0, errors.New("only tcp is supported")
}
host, portString, err := net.SplitHostPort(addr)
if err != nil {
return nil, 0, err
}
port, err := strconv.ParseUint(portString, 10, 16)
if err != nil {
return nil, 0, err
}
ip := net.ParseIP(host)
if ip == nil {
return nil, 0, errors.New("invalid address, must be an IP")
}
return ip, port, nil
}
|
