util.AddrIP, TCPAddr

diff --git a/util/addr.go b/util/addr.go
index b6ec49e13bf1bb8cc3e6b85a184a88f634d5f622..44354dab465cf419782573f6ae79f6f05a4ceed3 100644 (file)
--- 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 {