40 lines
782 B
Go
40 lines
782 B
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/libp2p/go-libp2p/core/host"
|
|
pp "github.com/libp2p/go-libp2p/core/peer"
|
|
"github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
func PeerIsAlive(h host.Host, ad pp.AddrInfo) bool {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
|
|
err := h.Connect(ctx, ad)
|
|
return err == nil
|
|
}
|
|
|
|
func ExtractIP(addr string) (net.IP, error) {
|
|
ma, err := multiaddr.NewMultiaddr(addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ipStr, err := ma.ValueForProtocol(multiaddr.P_IP4)
|
|
if err != nil {
|
|
ipStr, err = ma.ValueForProtocol(multiaddr.P_IP6)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
ip := net.ParseIP(ipStr)
|
|
if ip == nil {
|
|
return nil, fmt.Errorf("invalid IP: %s", ipStr)
|
|
}
|
|
return ip, nil
|
|
}
|