summaryrefslogtreecommitdiff
path: root/internal/failover/ping_default.go
blob: 181f3fccba697488599fd74c1df37b8824a7d629 (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
38
39
//go:build !windows

package failover

import (
	"fmt"
	"net"

	"golang.org/x/net/icmp"
)

// NewPinger creates a new pinger with gateway `gateway` and size `size`
func NewPinger(gateway string, size int) (*Pinger, error) {
	gip := net.ParseIP(gateway)
	isV4 := gip.To4() != nil
	mtuOverhead := mtuV6Overhead
	if isV4 {
		mtuOverhead = mtuV4Overhead
	}
	if size < mtuOverhead {
		return nil, fmt.Errorf("invalid MTU size given, MTU has to be at least: %v bytes", mtuOverhead)
	}
	var l *icmp.PacketConn
	var err error
	if isV4 {
		l, err = icmp.ListenPacket("udp4", "0.0.0.0")
	} else {
		l, err = icmp.ListenPacket("udp6", "::")
	}
	if err != nil {
		return nil, fmt.Errorf("failed creating ping with error: %w", err)
	}
	return &Pinger{
		listener: l,
		buffer:   make([]byte, size-mtuOverhead),
		gateway:  &net.UDPAddr{IP: gip},
		v4:       isV4,
	}, nil
}