mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-11-22 21:15:35 +03:00
573cbafe3f
Merge in DNS/adguard-home from 3597-wrt-netlink to master
Updates #3597.
Squashed commit of the following:
commit 1709582cd204bb80c84775feabae8723ed3340f6
Merge: 0507b6ed e7b3c996
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Tue Mar 15 20:25:18 2022 +0300
Merge branch 'master' into 3597-wrt-netlink
commit 0507b6ede1162554ca8ac7399d827264aca64f98
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Tue Mar 15 20:21:29 2022 +0300
all: imp code
commit 71f9758d854b3e2cf90cbd3655ae4818cfbcf528
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 9 18:03:48 2022 +0500
aghnet: imp naming
commit c949e765104f130aa3e5ba465bdebc3286bebe44
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 9 17:26:30 2022 +0500
all: imp code, docs
commit cf605ddb401b6e7b0a7a4bb1b175a4dc588d253a
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Tue Mar 8 15:33:52 2022 +0500
all: imp code, docs
commit 2960c6549a7e4944cc0072ca47a0cd4e82ec850e
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Sun Mar 6 21:34:58 2022 +0500
all: imp code & docs, fix tests
commit 29c049f3aee91a826c3416961686396d562a7066
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 2 20:45:34 2022 +0300
all: add arpdb
358 lines
8 KiB
Go
358 lines
8 KiB
Go
package home
|
|
|
|
import (
|
|
"net"
|
|
"os"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/dhcpd"
|
|
"github.com/AdguardTeam/golibs/testutil"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestClients(t *testing.T) {
|
|
clients := clientsContainer{}
|
|
clients.testing = true
|
|
|
|
clients.Init(nil, nil, nil, nil)
|
|
|
|
t.Run("add_success", func(t *testing.T) {
|
|
c := &Client{
|
|
IDs: []string{"1.1.1.1", "1:2:3::4", "aa:aa:aa:aa:aa:aa"},
|
|
Name: "client1",
|
|
}
|
|
|
|
ok, err := clients.Add(c)
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ok)
|
|
|
|
c = &Client{
|
|
IDs: []string{"2.2.2.2"},
|
|
Name: "client2",
|
|
}
|
|
|
|
ok, err = clients.Add(c)
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ok)
|
|
|
|
c, ok = clients.Find("1.1.1.1")
|
|
require.True(t, ok)
|
|
|
|
assert.Equal(t, "client1", c.Name)
|
|
|
|
c, ok = clients.Find("1:2:3::4")
|
|
require.True(t, ok)
|
|
|
|
assert.Equal(t, "client1", c.Name)
|
|
|
|
c, ok = clients.Find("2.2.2.2")
|
|
require.True(t, ok)
|
|
|
|
assert.Equal(t, "client2", c.Name)
|
|
|
|
assert.False(t, clients.Exists(net.IP{1, 2, 3, 4}, ClientSourceHostsFile))
|
|
assert.True(t, clients.Exists(net.IP{1, 1, 1, 1}, ClientSourceHostsFile))
|
|
assert.True(t, clients.Exists(net.IP{2, 2, 2, 2}, ClientSourceHostsFile))
|
|
})
|
|
|
|
t.Run("add_fail_name", func(t *testing.T) {
|
|
ok, err := clients.Add(&Client{
|
|
IDs: []string{"1.2.3.5"},
|
|
Name: "client1",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.False(t, ok)
|
|
})
|
|
|
|
t.Run("add_fail_ip", func(t *testing.T) {
|
|
ok, err := clients.Add(&Client{
|
|
IDs: []string{"2.2.2.2"},
|
|
Name: "client3",
|
|
})
|
|
require.Error(t, err)
|
|
assert.False(t, ok)
|
|
})
|
|
|
|
t.Run("update_fail_name", func(t *testing.T) {
|
|
err := clients.Update("client3", &Client{
|
|
IDs: []string{"1.2.3.0"},
|
|
Name: "client3",
|
|
})
|
|
require.Error(t, err)
|
|
|
|
err = clients.Update("client3", &Client{
|
|
IDs: []string{"1.2.3.0"},
|
|
Name: "client2",
|
|
})
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
t.Run("update_fail_ip", func(t *testing.T) {
|
|
err := clients.Update("client1", &Client{
|
|
IDs: []string{"2.2.2.2"},
|
|
Name: "client1",
|
|
})
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
t.Run("update_success", func(t *testing.T) {
|
|
err := clients.Update("client1", &Client{
|
|
IDs: []string{"1.1.1.2"},
|
|
Name: "client1",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
assert.False(t, clients.Exists(net.IP{1, 1, 1, 1}, ClientSourceHostsFile))
|
|
assert.True(t, clients.Exists(net.IP{1, 1, 1, 2}, ClientSourceHostsFile))
|
|
|
|
err = clients.Update("client1", &Client{
|
|
IDs: []string{"1.1.1.2"},
|
|
Name: "client1-renamed",
|
|
UseOwnSettings: true,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
c, ok := clients.Find("1.1.1.2")
|
|
require.True(t, ok)
|
|
|
|
assert.Equal(t, "client1-renamed", c.Name)
|
|
assert.True(t, c.UseOwnSettings)
|
|
|
|
nilCli, ok := clients.list["client1"]
|
|
require.False(t, ok)
|
|
|
|
assert.Nil(t, nilCli)
|
|
|
|
require.Len(t, c.IDs, 1)
|
|
|
|
assert.Equal(t, "1.1.1.2", c.IDs[0])
|
|
})
|
|
|
|
t.Run("del_success", func(t *testing.T) {
|
|
ok := clients.Del("client1-renamed")
|
|
require.True(t, ok)
|
|
|
|
assert.False(t, clients.Exists(net.IP{1, 1, 1, 2}, ClientSourceHostsFile))
|
|
})
|
|
|
|
t.Run("del_fail", func(t *testing.T) {
|
|
ok := clients.Del("client3")
|
|
assert.False(t, ok)
|
|
})
|
|
|
|
t.Run("addhost_success", func(t *testing.T) {
|
|
ip := net.IP{1, 1, 1, 1}
|
|
|
|
ok, err := clients.AddHost(ip, "host", ClientSourceARP)
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ok)
|
|
|
|
ok, err = clients.AddHost(ip, "host2", ClientSourceARP)
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ok)
|
|
|
|
ok, err = clients.AddHost(ip, "host3", ClientSourceHostsFile)
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ok)
|
|
|
|
assert.True(t, clients.Exists(ip, ClientSourceHostsFile))
|
|
})
|
|
|
|
t.Run("dhcp_replaces_arp", func(t *testing.T) {
|
|
ip := net.IP{1, 2, 3, 4}
|
|
|
|
ok, err := clients.AddHost(ip, "from_arp", ClientSourceARP)
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ok)
|
|
assert.True(t, clients.Exists(ip, ClientSourceARP))
|
|
|
|
ok, err = clients.AddHost(ip, "from_dhcp", ClientSourceDHCP)
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ok)
|
|
assert.True(t, clients.Exists(ip, ClientSourceDHCP))
|
|
})
|
|
|
|
t.Run("addhost_fail", func(t *testing.T) {
|
|
ok, err := clients.AddHost(net.IP{1, 1, 1, 1}, "host1", ClientSourceRDNS)
|
|
require.NoError(t, err)
|
|
assert.False(t, ok)
|
|
})
|
|
}
|
|
|
|
func TestClientsWHOIS(t *testing.T) {
|
|
clients := clientsContainer{
|
|
testing: true,
|
|
}
|
|
clients.Init(nil, nil, nil, nil)
|
|
whois := &RuntimeClientWHOISInfo{
|
|
Country: "AU",
|
|
Orgname: "Example Org",
|
|
}
|
|
|
|
t.Run("new_client", func(t *testing.T) {
|
|
ip := net.IP{1, 1, 1, 255}
|
|
clients.SetWHOISInfo(ip, whois)
|
|
v, _ := clients.ipToRC.Get(ip)
|
|
require.NotNil(t, v)
|
|
|
|
rc, ok := v.(*RuntimeClient)
|
|
require.True(t, ok)
|
|
require.NotNil(t, rc)
|
|
|
|
assert.Equal(t, rc.WHOISInfo, whois)
|
|
})
|
|
|
|
t.Run("existing_auto-client", func(t *testing.T) {
|
|
ip := net.IP{1, 1, 1, 1}
|
|
ok, err := clients.AddHost(ip, "host", ClientSourceRDNS)
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ok)
|
|
|
|
clients.SetWHOISInfo(ip, whois)
|
|
v, _ := clients.ipToRC.Get(ip)
|
|
require.NotNil(t, v)
|
|
|
|
rc, ok := v.(*RuntimeClient)
|
|
require.True(t, ok)
|
|
require.NotNil(t, rc)
|
|
|
|
assert.Equal(t, rc.WHOISInfo, whois)
|
|
})
|
|
|
|
t.Run("can't_set_manually-added", func(t *testing.T) {
|
|
ip := net.IP{1, 1, 1, 2}
|
|
|
|
ok, err := clients.Add(&Client{
|
|
IDs: []string{"1.1.1.2"},
|
|
Name: "client1",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.True(t, ok)
|
|
|
|
clients.SetWHOISInfo(ip, whois)
|
|
v, _ := clients.ipToRC.Get(ip)
|
|
require.Nil(t, v)
|
|
|
|
assert.True(t, clients.Del("client1"))
|
|
})
|
|
}
|
|
|
|
func TestClientsAddExisting(t *testing.T) {
|
|
clients := clientsContainer{
|
|
testing: true,
|
|
}
|
|
clients.Init(nil, nil, nil, nil)
|
|
|
|
t.Run("simple", func(t *testing.T) {
|
|
ip := net.IP{1, 1, 1, 1}
|
|
|
|
// Add a client.
|
|
ok, err := clients.Add(&Client{
|
|
IDs: []string{ip.String(), "1:2:3::4", "aa:aa:aa:aa:aa:aa", "2.2.2.0/24"},
|
|
Name: "client1",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.True(t, ok)
|
|
|
|
// Now add an auto-client with the same IP.
|
|
ok, err = clients.AddHost(ip, "test", ClientSourceRDNS)
|
|
require.NoError(t, err)
|
|
assert.True(t, ok)
|
|
})
|
|
|
|
t.Run("complicated", func(t *testing.T) {
|
|
// TODO(a.garipov): Properly decouple the DHCP server from the client
|
|
// storage.
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip("skipping dhcp test on windows")
|
|
}
|
|
|
|
var err error
|
|
|
|
ip := net.IP{1, 2, 3, 4}
|
|
|
|
// First, init a DHCP server with a single static lease.
|
|
config := &dhcpd.ServerConfig{
|
|
Enabled: true,
|
|
DBFilePath: "leases.db",
|
|
Conf4: dhcpd.V4ServerConf{
|
|
Enabled: true,
|
|
GatewayIP: net.IP{1, 2, 3, 1},
|
|
SubnetMask: net.IP{255, 255, 255, 0},
|
|
RangeStart: net.IP{1, 2, 3, 2},
|
|
RangeEnd: net.IP{1, 2, 3, 10},
|
|
},
|
|
}
|
|
|
|
clients.dhcpServer, err = dhcpd.Create(config)
|
|
require.NoError(t, err)
|
|
testutil.CleanupAndRequireSuccess(t, func() (err error) {
|
|
return os.Remove("leases.db")
|
|
})
|
|
|
|
err = clients.dhcpServer.AddStaticLease(&dhcpd.Lease{
|
|
HWAddr: net.HardwareAddr{0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA},
|
|
IP: ip,
|
|
Hostname: "testhost",
|
|
Expiry: time.Now().Add(time.Hour),
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// Add a new client with the same IP as for a client with MAC.
|
|
ok, err := clients.Add(&Client{
|
|
IDs: []string{ip.String()},
|
|
Name: "client2",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.True(t, ok)
|
|
|
|
// Add a new client with the IP from the first client's IP range.
|
|
ok, err = clients.Add(&Client{
|
|
IDs: []string{"2.2.2.2"},
|
|
Name: "client3",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.True(t, ok)
|
|
})
|
|
}
|
|
|
|
func TestClientsCustomUpstream(t *testing.T) {
|
|
clients := clientsContainer{
|
|
testing: true,
|
|
}
|
|
clients.Init(nil, nil, nil, nil)
|
|
|
|
// Add client with upstreams.
|
|
ok, err := clients.Add(&Client{
|
|
IDs: []string{"1.1.1.1", "1:2:3::4", "aa:aa:aa:aa:aa:aa"},
|
|
Name: "client1",
|
|
Upstreams: []string{
|
|
"1.1.1.1",
|
|
"[/example.org/]8.8.8.8",
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
assert.True(t, ok)
|
|
|
|
config, err := clients.findUpstreams("1.2.3.4")
|
|
assert.Nil(t, config)
|
|
assert.NoError(t, err)
|
|
|
|
config, err = clients.findUpstreams("1.1.1.1")
|
|
require.NotNil(t, config)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, config.Upstreams, 1)
|
|
assert.Len(t, config.DomainReservedUpstreams, 1)
|
|
}
|