extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2024-10-05 09:42:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

test windows

Browse Source
This commit is contained in:
Pascal Fischer 2023-06-12 16:22:53 +02:00
parent f45eb1a1da
commit b5d8142705
2 changed files with 6 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
client/internal/routemanager/systemops_nonandroid_test.go
View File

@ -163,8 +163,6 @@ func TestAddExistAndRemoveRouteNonAndroid(t *testing.T) {
},
}
// MOCK_ADDR := "127.0.0.1"
for n, testCase := range testCases {
var buf bytes.Buffer
log.SetOutput(&buf)
@ -183,16 +181,16 @@ func TestAddExistAndRemoveRouteNonAndroid(t *testing.T) {
err = wgInterface.Create()
require.NoError(t, err, "should create testing wireguard interface")
MOCK_ADDR := wgInterface.Address().IP.String()
MockAddr := wgInterface.Address().IP.String()
// Prepare the environment
if testCase.preExistingPrefix.IsValid() {
err := addToRouteTableIfNoExists(testCase.preExistingPrefix, MOCK_ADDR)
err := addToRouteTableIfNoExists(testCase.preExistingPrefix, MockAddr)
require.NoError(t, err, "should not return err when adding pre-existing route")
}
// Add the route
err = addToRouteTableIfNoExists(testCase.prefix, MOCK_ADDR)
err = addToRouteTableIfNoExists(testCase.prefix, MockAddr)
require.NoError(t, err, "should not return err when adding route")
if testCase.shouldAddRoute {
@ -202,7 +200,7 @@ func TestAddExistAndRemoveRouteNonAndroid(t *testing.T) {
require.True(t, ok, "route should exist")
// remove route again if added
err = removeFromRouteTableIfNonSystem(testCase.prefix, MOCK_ADDR)
err = removeFromRouteTableIfNonSystem(testCase.prefix, MockAddr)
require.NoError(t, err, "should not return err")
}

7
client/internal/routemanager/systemops_windows.go
View File

@ -4,7 +4,6 @@
package routemanager
import (
"fmt"
"net"
"net/netip"
@ -14,12 +13,11 @@ import (
type Win32_IP4RouteTable struct {
Destination string
Mask string
NextHop string
}
func existsInRouteTable(prefix netip.Prefix) (bool, error) {
var routes []Win32_IP4RouteTable
query := "SELECT Destination, Mask, NextHop FROM Win32_IP4RouteTable"
query := "SELECT Destination, Mask FROM Win32_IP4RouteTable"
err := wmi.Query(query, &routes)
if err != nil {
@ -27,12 +25,11 @@ func existsInRouteTable(prefix netip.Prefix) (bool, error) {
}
for _, route := range routes {
fmt.Println(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() && false {
if route.Destination == prefix.Addr().String() && cidr == prefix.Bits() {
return true, nil
}
}