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
85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
//go:build linux
|
|
// +build linux
|
|
|
|
package aghnet
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const arpAOutputWrt = `
|
|
IP address HW type Flags HW address Mask Device
|
|
192.168.1.2 0x1 0x2 ab:cd:ef:ab:cd:ef * wan
|
|
::ffff:ffff 0x1 0x2 ef:cd:ab:ef:cd:ab * wan`
|
|
|
|
const arpAOutput = `
|
|
? (192.168.1.2) at ab:cd:ef:ab:cd:ef on en0 ifscope [ethernet]
|
|
? (::ffff:ffff) at ef:cd:ab:ef:cd:ab on em0 expires in 100 seconds [ethernet]`
|
|
|
|
const ipNeighOutput = `
|
|
192.168.1.2 dev enp0s3 lladdr ab:cd:ef:ab:cd:ef DELAY
|
|
::ffff:ffff dev enp0s3 lladdr ef:cd:ab:ef:cd:ab router STALE`
|
|
|
|
var wantNeighs = []Neighbor{{
|
|
IP: net.IPv4(192, 168, 1, 2),
|
|
MAC: net.HardwareAddr{0xAB, 0xCD, 0xEF, 0xAB, 0xCD, 0xEF},
|
|
}, {
|
|
IP: net.ParseIP("::ffff:ffff"),
|
|
MAC: net.HardwareAddr{0xEF, 0xCD, 0xAB, 0xEF, 0xCD, 0xAB},
|
|
}}
|
|
|
|
func TestFSysARPDB(t *testing.T) {
|
|
a := &fsysARPDB{
|
|
ns: &neighs{
|
|
mu: &sync.RWMutex{},
|
|
ns: make([]Neighbor, 0),
|
|
},
|
|
fsys: testdata,
|
|
filename: "proc_net_arp",
|
|
}
|
|
|
|
err := a.Refresh()
|
|
require.NoError(t, err)
|
|
|
|
ns := a.Neighbors()
|
|
assert.Equal(t, wantNeighs, ns)
|
|
}
|
|
|
|
func TestCmdARPDB_arpawrt(t *testing.T) {
|
|
a := &cmdARPDB{
|
|
parse: parseArpAWrt,
|
|
runcmd: func() (r io.Reader, err error) { return strings.NewReader(arpAOutputWrt), nil },
|
|
ns: &neighs{
|
|
mu: &sync.RWMutex{},
|
|
ns: make([]Neighbor, 0),
|
|
},
|
|
}
|
|
|
|
err := a.Refresh()
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, wantNeighs, a.Neighbors())
|
|
}
|
|
|
|
func TestCmdARPDB_ipneigh(t *testing.T) {
|
|
a := &cmdARPDB{
|
|
parse: parseIPNeigh,
|
|
runcmd: func() (r io.Reader, err error) { return strings.NewReader(ipNeighOutput), nil },
|
|
ns: &neighs{
|
|
mu: &sync.RWMutex{},
|
|
ns: make([]Neighbor, 0),
|
|
},
|
|
}
|
|
err := a.Refresh()
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, wantNeighs, a.Neighbors())
|
|
}
|