Failover: Support v6 gateway

Useful for v6 only VPNs

1 files changed, 18 insertions, 2 deletions

diff --git a/internal/failover/ping_windows.go b/internal/failover/ping_windows.go
index 3f181f5..eb801d8 100644
--- a/internal/failover/ping_windows.go
+++ b/internal/failover/ping_windows.go
@@ -8,13 +8,29 @@ import (
 )
 
 func NewPinger(gateway string, size int) (*Pinger, error) {
-	l, err := icmp.ListenPacket("ip4:icmp", "0.0.0.0")
+	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("ip4:icmp", "0.0.0.0")
+	} else {
+		l, err = icmp.ListenPacket("ip6:icmp", "::")
+	}
 	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.IPAddr{IP: net.ParseIP(gateway)},
+		gateway:  &net.IPAddr{IP: gip},
+		v4:       isV4,
 	}, nil
 }