status-im
/
go-libp2p
mirror of https://github.com/status-im/go-libp2p.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

cleanup addrToIP logic

This commit is contained in:
Will Scott 2020-02-25 10:24:47 -08:00
parent 10b0a942b8
commit 426d729c51
2 changed files with 16 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
p2p/host/autonat/svc.go
View File

@ -3,7 +3,7 @@ package autonat
import (
"bytes"
"context"
"fmt"
"errors"
"math/rand"
"net"
"strings"
@ -106,31 +106,25 @@ func (as *AutoNATService) handleStream(s network.Stream) {
}
// Optimistically extract the net.IP host from a multiaddress.
func addrToIP(addr ma.Multiaddr) net.IP {
func addrToIP(addr ma.Multiaddr) (net.IP, error) {
n, ip, err := manet.DialArgs(addr)
if err != nil {
return nil
return nil, err
}
// if no port:
if n == "ip" || n == "ip4" || n == "ip6" {
// Strip v6 zone if it's there.
if strings.Contains(ip, "%") {
ip = ip[:strings.Index(ip, "%")]
}
return net.ParseIP(ip)
if strings.HasPrefix(n, "tcp") || strings.HasPrefix(n, "udp") {
ip, _, err = net.SplitHostPort(ip)
} else if !strings.HasPrefix(n, "ip") {
return nil, errors.New("non-ip multiaddr")
}
ip, _, err = net.SplitHostPort(ip)
if err != nil {
fmt.Printf("failed to split: %v", err)
return nil
return nil, err
}
// Strip v6 zone if it's there.
if strings.Contains(ip, "%") {
ip = ip[:strings.Index(ip, "%")]
}
return net.ParseIP(ip)
return net.ParseIP(ip), nil
}
func (as *AutoNATService) handleDial(p peer.ID, obsaddr ma.Multiaddr, mpi *pb.Message_PeerInfo) *pb.Message_DialResponse {
@ -158,7 +152,7 @@ func (as *AutoNATService) handleDial(p peer.ID, obsaddr ma.Multiaddr, mpi *pb.Me
if !as.skipDial(obsaddr) {
addrs = append(addrs, obsaddr)
seen[obsaddr.String()] = struct{}{}
obsHost = addrToIP(obsaddr)
obsHost, _ = addrToIP(obsaddr)
}
for _, maddr := range mpi.GetAddrs() {
@ -172,7 +166,7 @@ func (as *AutoNATService) handleDial(p peer.ID, obsaddr ma.Multiaddr, mpi *pb.Me
continue
}
if !bytes.Equal(obsHost, addrToIP(addr)) {
if ip, err := addrToIP(addr); err != nil || !bytes.Equal(obsHost, ip) {
continue
}

10
p2p/host/autonat/svc_test.go
View File

@ -139,26 +139,26 @@ func TestAutoNATServiceDialRateLimiter(t *testing.T) {
func TestAddrToIP(t *testing.T) {
addr, _ := ma.NewMultiaddr("/ip4/127.0.0.1/tcp/0")
if !addrToIP(addr).Equal(net.IPv4(127, 0, 0, 1)) {
if ip, err := addrToIP(addr); err != nil || !ip.Equal(net.IPv4(127, 0, 0, 1)) {
t.Fatal("addrToIP of ipv4 localhost incorrect!")
}
addr, _ = ma.NewMultiaddr("/ip4/192.168.0.1/tcp/6")
if !addrToIP(addr).Equal(net.IPv4(192, 168, 0, 1)) {
if ip, err := addrToIP(addr); err != nil || !ip.Equal(net.IPv4(192, 168, 0, 1)) {
t.Fatal("addrToIP of ipv4 incorrect!")
}
addr, _ = ma.NewMultiaddr("/ip6/::ffff:127.0.0.1/tcp/111")
if !addrToIP(addr).Equal(net.ParseIP("::ffff:127.0.0.1")) {
if ip, err := addrToIP(addr); err != nil || !ip.Equal(net.ParseIP("::ffff:127.0.0.1")) {
t.Fatal("addrToIP of ipv6 incorrect!")
}
addr, _ = ma.NewMultiaddr("/ip6zone/eth0/ip6/fe80::1")
if !addrToIP(addr).Equal(net.ParseIP("fe80::1")) {
if ip, err := addrToIP(addr); err != nil || !ip.Equal(net.ParseIP("fe80::1")) {
t.Fatal("addrToIP of ip6zone incorrect!")
}
addr, _ = ma.NewMultiaddr("/unix/a/b/c/d")
if addrToIP(addr) != nil {
if _, err := addrToIP(addr); err == nil {
t.Fatal("invalid addrToIP populates")
}
}