mirror of
https://github.com/netbirdio/netbird.git
synced 2024-11-30 03:54:35 +01:00
38 lines
713 B
Go
38 lines
713 B
Go
//go:build windows
|
|
// +build windows
|
|
|
|
package routemanager
|
|
|
|
import (
|
|
"net"
|
|
"net/netip"
|
|
|
|
"github.com/yusufpapurcu/wmi"
|
|
)
|
|
|
|
type Win32_IP4RouteTable struct {
|
|
Destination string
|
|
Mask string
|
|
}
|
|
|
|
func existsInRouteTable(prefix netip.Prefix) (bool, error) {
|
|
var routes []Win32_IP4RouteTable
|
|
query := "SELECT Destination, Mask FROM Win32_IP4RouteTable"
|
|
|
|
err := wmi.Query(query, &routes)
|
|
if err != nil {
|
|
return true, err
|
|
}
|
|
|
|
for _, route := range routes {
|
|
ip := net.ParseIP(route.Mask)
|
|
ip = ip.To4()
|
|
mask := net.IPv4Mask(ip[0], ip[1], ip[2], ip[3])
|
|
cidr, _ := mask.Size()
|
|
if route.Destination == prefix.Addr().String() && cidr == prefix.Bits() {
|
|
return true, nil
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|