summaryrefslogtreecommitdiff
path: root/pkg/p2p/ip.go
blob: c82b861e6726ca79b9136947f2a2b7f3439e834c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
package p2p

import (
	"fmt"
	"math/rand"
	"net"

	"github.com/boozec/rahanna/internal/logger"
)

// Connect a DNS to get the address
func GetOutboundIP() net.IP {
	log, _ := logger.GetLogger()
	conn, err := net.Dial("udp", "8.8.8.8:80")
	if err != nil {
		log.Sugar().Error("err", err)
	}
	defer conn.Close()

	localAddr := conn.LocalAddr().(*net.UDPAddr)

	return localAddr.IP
}

// Returns a random available port on the node
func GetRandomAvailablePort() (int, error) {
	for i := 0; i < 100; i += 1 {
		port := rand.Intn(65535-1024) + 1024
		addr := fmt.Sprintf(":%d", port)
		ln, err := net.Listen("tcp", addr)
		if err == nil {
			defer ln.Close()
			return port, nil
		}
	}
	return 0, fmt.Errorf("failed to find an available port after multiple attempts")
}