mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-11-21 20:45:33 +03:00
6fea7099a2
Squashed commit of the following: commit 6f6f6cc5d9b9ae04e369e0b789aaab74f234e6a0 Merge: 9aa3ac58c8fb76701fAuthor: Stanislav Chzhen <s.chzhen@adguard.com> Date: Thu Aug 24 13:29:47 2023 +0300 Merge branch 'master' into AG-24794-imp-arpdb commit 9aa3ac58c76fc4b2f950a988d63dfebd0652e507 Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Aug 23 16:14:02 2023 +0300 scripts: gocognit: add arpdb commit e99b0534be1891de1c13f4010beeedb4459ccd7c Merge: 84893bc2d3722c2846Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Aug 23 16:08:25 2023 +0300 Merge branch 'master' into AG-24794-imp-arpdb commit 84893bc2d3018c9ee1e411578b33cdb6ba6d3d81 Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Aug 23 16:07:43 2023 +0300 arpdb: add todo commit ad4b3689b51324521bf47c478c61b6008332b4f5 Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Aug 23 14:02:07 2023 +0300 arpdb: imp code commit 9cdd17dadbb91ccc3f8e79ba7a21bc365647e089 Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Fri Aug 18 19:05:10 2023 +0300 all: imp arpdb
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
//go:build linux
|
|
|
|
package arpdb
|
|
|
|
import (
|
|
"net"
|
|
"net/netip"
|
|
"sync"
|
|
"testing"
|
|
"testing/fstest"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const arpAOutputWrt = `
|
|
IP address HW type Flags HW address Mask Device
|
|
1.2.3.4.5 0x1 0x2 aa:bb:cc:dd:ee:ff * wan
|
|
1.2.3.4 0x1 0x2 12:34:56:78:910 * wan
|
|
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 = `
|
|
invalid.mac (1.2.3.4) at 12:34:56:78:910 on el0 ifscope [ethernet]
|
|
invalid.ip (1.2.3.4.5) at ab:cd:ef:ab:cd:12 on ek0 ifscope [ethernet]
|
|
invalid.fmt 1 at 12:cd:ef:ab:cd:ef on er0 ifscope [ethernet]
|
|
? (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 = `
|
|
1.2.3.4.5 dev enp0s3 lladdr aa:bb:cc:dd:ee:ff DELAY
|
|
1.2.3.4 dev enp0s3 lladdr 12:34:56:78:910 DELAY
|
|
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: netip.MustParseAddr("192.168.1.2"),
|
|
MAC: net.HardwareAddr{0xAB, 0xCD, 0xEF, 0xAB, 0xCD, 0xEF},
|
|
}, {
|
|
IP: netip.MustParseAddr("::ffff:ffff"),
|
|
MAC: net.HardwareAddr{0xEF, 0xCD, 0xAB, 0xEF, 0xCD, 0xAB},
|
|
}}
|
|
|
|
func TestFSysARPDB(t *testing.T) {
|
|
require.NoError(t, fstest.TestFS(testdata, "proc_net_arp"))
|
|
|
|
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_linux(t *testing.T) {
|
|
sh := mapShell{
|
|
"arp -a": {err: nil, out: arpAOutputWrt, code: 0},
|
|
"ip neigh": {err: nil, out: ipNeighOutput, code: 0},
|
|
}
|
|
substShell(t, sh.RunCmd)
|
|
|
|
t.Run("wrt", func(t *testing.T) {
|
|
a := &cmdARPDB{
|
|
parse: parseArpAWrt,
|
|
cmd: "arp",
|
|
args: []string{"-a"},
|
|
ns: &neighs{
|
|
mu: &sync.RWMutex{},
|
|
ns: make([]Neighbor, 0),
|
|
},
|
|
}
|
|
|
|
err := a.Refresh()
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, wantNeighs, a.Neighbors())
|
|
})
|
|
|
|
t.Run("ip_neigh", func(t *testing.T) {
|
|
a := &cmdARPDB{
|
|
parse: parseIPNeigh,
|
|
cmd: "ip",
|
|
args: []string{"neigh"},
|
|
ns: &neighs{
|
|
mu: &sync.RWMutex{},
|
|
ns: make([]Neighbor, 0),
|
|
},
|
|
}
|
|
err := a.Refresh()
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, wantNeighs, a.Neighbors())
|
|
})
|
|
}
|