mirror of
				https://source.quilibrium.com/quilibrium/ceremonyclient.git
				synced 2025-11-04 07:07:26 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			80 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			80 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
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
 | 
						|
}
 |