package holepunch import ( "context" "github.com/libp2p/go-libp2p/core/host" "github.com/libp2p/go-libp2p/core/network" "github.com/libp2p/go-libp2p/core/peer" ma "github.com/multiformats/go-multiaddr" manet "github.com/multiformats/go-multiaddr/net" ) func containsPublicAddr(addrs []ma.Multiaddr) bool { for _, addr := range addrs { if isRelayAddress(addr) || !manet.IsPublicAddr(addr) { continue } return true } return false } func removeRelayAddrs(addrs []ma.Multiaddr) []ma.Multiaddr { result := make([]ma.Multiaddr, 0, len(addrs)) for _, addr := range addrs { if !isRelayAddress(addr) { result = append(result, addr) } } return result } func isRelayAddress(a ma.Multiaddr) bool { _, err := a.ValueForProtocol(ma.P_CIRCUIT) return err == nil } func addrsToBytes(as []ma.Multiaddr) [][]byte { bzs := make([][]byte, 0, len(as)) for _, a := range as { bzs = append(bzs, a.Bytes()) } return bzs } func addrsFromBytes(bzs [][]byte) []ma.Multiaddr { addrs := make([]ma.Multiaddr, 0, len(bzs)) for _, bz := range bzs { a, err := ma.NewMultiaddrBytes(bz) if err == nil { addrs = append(addrs, a) } } return addrs } func getDirectConnection(h host.Host, p peer.ID) network.Conn { for _, c := range h.Network().ConnsToPeer(p) { if !isRelayAddress(c.RemoteMultiaddr()) { return c } } return nil } func holePunchConnect(ctx context.Context, host host.Host, pi peer.AddrInfo, isClient bool) error { holePunchCtx := network.WithSimultaneousConnect(ctx, isClient, "hole-punching") forceDirectConnCtx := network.WithForceDirectDial(holePunchCtx, "hole-punching") dialCtx, cancel := context.WithTimeout(forceDirectConnCtx, dialTimeout) defer cancel() if err := host.Connect(dialCtx, pi); err != nil { log.Debugw("hole punch attempt with peer failed", "peer ID", pi.ID, "error", err) return err } log.Debugw("hole punch successful", "peer", pi.ID) return nil }