mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-16 11:51:30 +01:00
4fec709bb1
* compile client under freebsd (#1620) Compile netbird client under freebsd and now support netstack and userspace modes. Refactoring linux specific code to share same code with FreeBSD, move to *_unix.go files. Not implemented yet: Kernel mode not supported DNS probably does not work yet Routing also probably does not work yet SSH support did not tested yet Lack of test environment for freebsd (dedicated VM for github runners under FreeBSD required) Lack of tests for freebsd specific code info reporting need to review and also implement, for example OS reported as GENERIC instead of FreeBSD (lack of FreeBSD icon in management interface) Lack of proper client setup under FreeBSD Lack of FreeBSD port/package * Add DNS routes (#1943) Given domains are resolved periodically and resolved IPs are replaced with the new ones. Unless the flag keep_route is set to true, then only new ones are added. This option is helpful if there are long-running connections that might still point to old IP addresses from changed DNS records. * Add process posture check (#1693) Introduces a process posture check to validate the existence and active status of specific binaries on peer systems. The check ensures that files are present at specified paths, and that corresponding processes are running. This check supports Linux, Windows, and macOS systems. Co-authored-by: Evgenii <mail@skillcoder.com> Co-authored-by: Pascal Fischer <pascal@netbird.io> Co-authored-by: Zoltan Papp <zoltan.pmail@gmail.com> Co-authored-by: Viktor Liu <17948409+lixmal@users.noreply.github.com> Co-authored-by: Bethuel Mmbaga <bethuelmbaga12@gmail.com>
356 lines
7.6 KiB
Go
356 lines
7.6 KiB
Go
package routemanager
|
|
|
|
import (
|
|
"net/netip"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/netbirdio/netbird/client/internal/routemanager/static"
|
|
"github.com/netbirdio/netbird/route"
|
|
)
|
|
|
|
func TestGetBestrouteFromStatuses(t *testing.T) {
|
|
|
|
testCases := []struct {
|
|
name string
|
|
statuses map[route.ID]routerPeerStatus
|
|
expectedRouteID route.ID
|
|
currentRoute route.ID
|
|
existingRoutes map[route.ID]*route.Route
|
|
}{
|
|
{
|
|
name: "one route",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "route1",
|
|
},
|
|
{
|
|
name: "one connected routes with relayed and direct",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: true,
|
|
direct: true,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "route1",
|
|
},
|
|
{
|
|
name: "one connected routes with relayed and no direct",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: true,
|
|
direct: false,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "route1",
|
|
},
|
|
{
|
|
name: "no connected peers",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: false,
|
|
relayed: false,
|
|
direct: false,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "",
|
|
},
|
|
{
|
|
name: "multiple connected peers with different metrics",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
},
|
|
"route2": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: 9000,
|
|
Peer: "peer1",
|
|
},
|
|
"route2": {
|
|
ID: "route2",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer2",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "route1",
|
|
},
|
|
{
|
|
name: "multiple connected peers with one relayed",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
},
|
|
"route2": {
|
|
connected: true,
|
|
relayed: true,
|
|
direct: true,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
"route2": {
|
|
ID: "route2",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer2",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "route1",
|
|
},
|
|
{
|
|
name: "multiple connected peers with one direct",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
},
|
|
"route2": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: false,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
"route2": {
|
|
ID: "route2",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer2",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "route1",
|
|
},
|
|
{
|
|
name: "multiple connected peers with different latencies",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
latency: 300 * time.Millisecond,
|
|
},
|
|
"route2": {
|
|
connected: true,
|
|
latency: 10 * time.Millisecond,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
"route2": {
|
|
ID: "route2",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer2",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "route2",
|
|
},
|
|
{
|
|
name: "should ignore routes with latency 0",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
latency: 0 * time.Millisecond,
|
|
},
|
|
"route2": {
|
|
connected: true,
|
|
latency: 10 * time.Millisecond,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
"route2": {
|
|
ID: "route2",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer2",
|
|
},
|
|
},
|
|
currentRoute: "",
|
|
expectedRouteID: "route2",
|
|
},
|
|
{
|
|
name: "current route with similar score and similar but slightly worse latency should not change",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
latency: 15 * time.Millisecond,
|
|
},
|
|
"route2": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
latency: 10 * time.Millisecond,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
"route2": {
|
|
ID: "route2",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer2",
|
|
},
|
|
},
|
|
currentRoute: "route1",
|
|
expectedRouteID: "route1",
|
|
},
|
|
{
|
|
name: "current route with bad score should be changed to route with better score",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
latency: 200 * time.Millisecond,
|
|
},
|
|
"route2": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
latency: 10 * time.Millisecond,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
"route2": {
|
|
ID: "route2",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer2",
|
|
},
|
|
},
|
|
currentRoute: "route1",
|
|
expectedRouteID: "route2",
|
|
},
|
|
{
|
|
name: "current chosen route doesn't exist anymore",
|
|
statuses: map[route.ID]routerPeerStatus{
|
|
"route1": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
latency: 20 * time.Millisecond,
|
|
},
|
|
"route2": {
|
|
connected: true,
|
|
relayed: false,
|
|
direct: true,
|
|
latency: 10 * time.Millisecond,
|
|
},
|
|
},
|
|
existingRoutes: map[route.ID]*route.Route{
|
|
"route1": {
|
|
ID: "route1",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer1",
|
|
},
|
|
"route2": {
|
|
ID: "route2",
|
|
Metric: route.MaxMetric,
|
|
Peer: "peer2",
|
|
},
|
|
},
|
|
currentRoute: "routeDoesntExistAnymore",
|
|
expectedRouteID: "route2",
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
currentRoute := &route.Route{
|
|
ID: "routeDoesntExistAnymore",
|
|
}
|
|
if tc.currentRoute != "" {
|
|
currentRoute = tc.existingRoutes[tc.currentRoute]
|
|
}
|
|
|
|
// create new clientNetwork
|
|
client := &clientNetwork{
|
|
handler: static.NewRoute(&route.Route{Network: netip.MustParsePrefix("192.168.0.0/24")}, nil, nil),
|
|
routes: tc.existingRoutes,
|
|
currentChosen: currentRoute,
|
|
}
|
|
|
|
chosenRoute := client.getBestRouteFromStatuses(tc.statuses)
|
|
if chosenRoute != tc.expectedRouteID {
|
|
t.Errorf("expected routeID %s, got %s", tc.expectedRouteID, chosenRoute)
|
|
}
|
|
})
|
|
}
|
|
}
|