From: Matt Joiner <anacrolix@gmail.com>
Date: Sun, 8 Mar 2015 06:44:08 +0000 (+1100)
Subject: util.AddrIP, TCPAddr
X-Git-Tag: v1.0.0~1294
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=0c7c2e195558b9480d9e50a24a711ae5d0a86ef2;p=btrtrc.git

util.AddrIP, TCPAddr
---

diff --git a/util/addr.go b/util/addr.go
index b6ec49e1..44354dab 100644
--- a/util/addr.go
+++ b/util/addr.go
@@ -27,6 +27,8 @@ func AddrIP(addr net.Addr) net.IP {
 	switch raw := addr.(type) {
 	case *net.UDPAddr:
 		return raw.IP
+	case *net.TCPAddr:
+		return raw.IP
 	default:
 		host, _, err := net.SplitHostPort(addr.String())
 		if err != nil {