netbird/client/internal/routemanager/systemops_windows.go

38 lines
700 B
Go
Raw Normal View History

//go:build windows
// +build windows
package routemanager
2023-06-09 19:15:39 +02:00
import (
2023-06-12 11:43:18 +02:00
"net"
2023-06-09 19:15:39 +02:00
"net/netip"
2023-06-09 19:17:26 +02:00
"github.com/yusufpapurcu/wmi"
2023-06-09 19:15:39 +02:00
)
type Win32_IP4RouteTable struct {
2023-06-12 11:06:49 +02:00
Destination string
Mask string
NextHop string
2023-06-09 19:15:39 +02:00
}
func existsInRouteTable(prefix netip.Prefix) (bool, error) {
var routes []Win32_IP4RouteTable
2023-06-12 11:06:49 +02:00
query := "SELECT Destination, Mask, NextHop FROM Win32_IP4RouteTable"
2023-06-09 19:15:39 +02:00
err := wmi.Query(query, &routes)
if err != nil {
return true, err
}
for _, route := range routes {
2023-06-12 11:43:18 +02:00
ip := net.ParseIP(route.Mask)
mask := net.IPMask(ip)
cidr, _ := mask.Size()
if route.Destination == prefix.Addr().String() && cidr == prefix.Bits() {
return true, nil
}
2023-06-09 19:15:39 +02:00
}
return false, nil
}