Fix address filtering for /ip6, add tests

License: MIT
Signed-off-by: Lars Gierth <larsg@systemli.org>

p2p/net/filter/filter.go

@@ -2,7 +2,6 @@ package filter
 
 import (
 	"net"
-	"strings"
 	"sync"
 
 	manet "gx/ipfs/QmYVqhVfbK4BKvbW88Lhm26b3ud14sTBvcm1H7uWUx1Fkp/go-multiaddr-net"
@@ -27,14 +26,13 @@ func (fs *Filters) AddDialFilter(f *net.IPNet) {
 }
 
 func (f *Filters) AddrBlocked(a ma.Multiaddr) bool {
-	_, addr, err := manet.DialArgs(a)
+	ipstr, err := manet.ToNetAddr(a)
 	if err != nil {
 		// if we cant parse it, its probably not blocked
 		return false
 	}
 
-	ipstr := strings.Split(addr, ":")[0]
-	ip := net.ParseIP(ipstr)
+	ip := net.ParseIP(ipstr.String())
 	f.mu.RLock()
 	defer f.mu.RUnlock()
 	for _, ft := range f.filters {

p2p/net/filter/filter_test.go

@@ -0,0 +1,51 @@
+package filter
+
+import (
+	"net"
+	"testing"
+
+	ma "gx/ipfs/QmcobAGsCjYt5DXoq9et9L8yR8er7o7Cu3DTvpaq12jYSz/go-multiaddr"
+)
+
+func TestFilter(t *testing.T) {
+	f := NewFilters()
+	for _, cidr := range []string{
+		"1.2.3.0/24",
+		"4.3.2.1/32",
+		"fd00::/8",
+		"fc00::1/128",
+	} {
+		_, ipnet, _ := net.ParseCIDR(cidr)
+		f.AddDialFilter(ipnet)
+	}
+
+	for _, blocked := range []string{
+		"/ip4/1.2.3.4",
+		"/ip4/4.3.2.1",
+		"/ip6/fd00::2",
+		"/ip6/fc00::1",
+	} {
+		maddr, err := ma.NewMultiaddr(blocked)
+		if err != nil {
+			t.Error(err)
+		}
+		if !f.AddrBlocked(maddr) {
+			t.Fatalf("expected %s to be blocked", blocked)
+		}
+	}
+
+	for _, notBlocked := range []string{
+		"/ip4/1.2.4.1",
+		"/ip4/4.3.2.2",
+		"/ip6/fe00::1",
+		"/ip6/fc00::2",
+	} {
+		maddr, err := ma.NewMultiaddr(notBlocked)
+		if err != nil {
+			t.Error(err)
+		}
+		if f.AddrBlocked(maddr) {
+			t.Fatalf("expected %s to not be blocked", notBlocked)
+		}
+	}
+}