diff --git a/p2p/net/transport/tcp.go b/p2p/net/transport/tcp.go index 2b64854c..018cd53b 100644 --- a/p2p/net/transport/tcp.go +++ b/p2p/net/transport/tcp.go @@ -8,6 +8,7 @@ import ( lgbl "github.com/ipfs/go-libp2p/loggables" manet "gx/ipfs/QmQB7mNP3QE7b4zP2MQmsyJDqG5hzYE2CL8k1VyLWky2Ed/go-multiaddr-net" + mafmt "gx/ipfs/QmWLfU4tstw2aNcTykDm44xbSTCYJ9pUJwfhQCKGwckcHx/mafmt" context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context" reuseport "gx/ipfs/QmaaC9QMYTQHCbMq3Ebr3uMaAR2ev4AVqMmsJpgQijAZbJ/go-reuseport" ma "gx/ipfs/QmcobAGsCjYt5DXoq9et9L8yR8er7o7Cu3DTvpaq12jYSz/go-multiaddr" @@ -103,7 +104,7 @@ func manetListen(addr ma.Multiaddr) (manet.Listener, error) { } func (t *TcpTransport) Matches(a ma.Multiaddr) bool { - return IsTcpMultiaddr(a) + return mafmt.TCP.Matches(a) } type tcpDialer struct { @@ -199,7 +200,7 @@ func (d *tcpDialer) reuseDial(raddr ma.Multiaddr) (manet.Conn, error) { } func (d *tcpDialer) Matches(a ma.Multiaddr) bool { - return IsTcpMultiaddr(a) + return mafmt.TCP.Matches(a) } type tcpListener struct { diff --git a/p2p/net/transport/transport.go b/p2p/net/transport/transport.go index 783f0205..833e7374 100644 --- a/p2p/net/transport/transport.go +++ b/p2p/net/transport/transport.go @@ -49,13 +49,3 @@ type TimeoutOpt time.Duration type ReuseportOpt bool var ReusePorts ReuseportOpt = true - -func IsTcpMultiaddr(a ma.Multiaddr) bool { - p := a.Protocols() - return len(p) == 2 && (p[0].Name == "ip4" || p[0].Name == "ip6") && p[1].Name == "tcp" -} - -func IsUtpMultiaddr(a ma.Multiaddr) bool { - p := a.Protocols() - return len(p) == 3 && p[2].Name == "utp" -} diff --git a/p2p/net/transport/utp.go b/p2p/net/transport/utp.go index dd402448..3cf22e2d 100644 --- a/p2p/net/transport/utp.go +++ b/p2p/net/transport/utp.go @@ -7,6 +7,7 @@ import ( manet "gx/ipfs/QmQB7mNP3QE7b4zP2MQmsyJDqG5hzYE2CL8k1VyLWky2Ed/go-multiaddr-net" mautp "gx/ipfs/QmQB7mNP3QE7b4zP2MQmsyJDqG5hzYE2CL8k1VyLWky2Ed/go-multiaddr-net/utp" utp "gx/ipfs/QmVs3wq4cN64TFCxANzgSHjGPrjMnRnwPrxU8bqc7YP42s/utp" + mafmt "gx/ipfs/QmWLfU4tstw2aNcTykDm44xbSTCYJ9pUJwfhQCKGwckcHx/mafmt" ma "gx/ipfs/QmcobAGsCjYt5DXoq9et9L8yR8er7o7Cu3DTvpaq12jYSz/go-multiaddr" ) @@ -22,8 +23,7 @@ func NewUtpTransport() *UtpTransport { } func (d *UtpTransport) Matches(a ma.Multiaddr) bool { - p := a.Protocols() - return len(p) == 3 && p[2].Name == "utp" + return mafmt.UTP.Matches(a) } type UtpSocket struct { @@ -129,8 +129,7 @@ func (s *UtpSocket) Accept() (Conn, error) { } func (s *UtpSocket) Matches(a ma.Multiaddr) bool { - p := a.Protocols() - return len(p) == 3 && p[2].Name == "utp" + return mafmt.UTP.Matches(a) } func (t *UtpSocket) Close() error { diff --git a/package.json b/package.json index 01daac40..24974775 100644 --- a/package.json +++ b/package.json @@ -127,6 +127,12 @@ "name": "go-peerstream", "hash": "QmQDPXRFzRcCGPbPViQCKjzbQBkZGpLV1f9KwXnksSNcTK", "version": "0.0.0" + }, + { + "author": "whyrusleeping", + "name": "mafmt", + "hash": "QmWLfU4tstw2aNcTykDm44xbSTCYJ9pUJwfhQCKGwckcHx", + "version": "0.0.0" } ], "language": "go",