mirrors
/
AdGuardHome
mirror of https://github.com/AdguardTeam/AdGuardHome.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

all: slog arpdb

This commit is contained in:
Stanislav Chzhen 2024-08-26 19:11:10 +03:00
parent 738958d90a
commit 2738383303
8 changed files with 106 additions and 142 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
internal/arpdb/arpdb.go
View File

@ -5,6 +5,7 @@ import (
"bufio" "bufio"
"bytes" "bytes"
"fmt" "fmt"
"log/slog"
"net" "net"
"net/netip" "net/netip"
"slices" "slices"
@ -12,7 +13,7 @@ import (
"github.com/AdguardTeam/AdGuardHome/internal/aghos" "github.com/AdguardTeam/AdGuardHome/internal/aghos"
"github.com/AdguardTeam/golibs/errors" "github.com/AdguardTeam/golibs/errors"
"github.com/AdguardTeam/golibs/log" "github.com/AdguardTeam/golibs/logutil/slogutil"
"github.com/AdguardTeam/golibs/netutil" "github.com/AdguardTeam/golibs/netutil"
"github.com/AdguardTeam/golibs/osutil" "github.com/AdguardTeam/golibs/osutil"
) )
@ -38,8 +39,8 @@ type Interface interface {
} }
// New returns the [Interface] properly initialized for the OS. // New returns the [Interface] properly initialized for the OS.
func New() (arp Interface) { func New(logger *slog.Logger) (arp Interface) {
return newARPDB() return newARPDB(logger)
} }
// Empty is the [Interface] implementation that does nothing. // Empty is the [Interface] implementation that does nothing.
@ -69,6 +70,28 @@ type Neighbor struct {
MAC net.HardwareAddr MAC net.HardwareAddr
} }
// newNeighbor returns the new initialized [Neighbor] by parsing string
// representations of IP and MAC addresses.
func newNeighbor(host, ipStr, macStr string) (n *Neighbor, err error) {
defer func() { err = errors.Annotate(err, "getting arp neighbor: %w") }()
ip, err := netip.ParseAddr(ipStr)
if err != nil {
return nil, err
}
mac, err := net.ParseMAC(macStr)
if err != nil {
return nil, err
}
return &Neighbor{
Name: host,
IP: ip,
MAC: mac,
}, nil
}
// Clone returns the deep copy of n. // Clone returns the deep copy of n.
func (n Neighbor) Clone() (clone Neighbor) { func (n Neighbor) Clone() (clone Neighbor) {
return Neighbor{ return Neighbor{
@ -80,10 +103,10 @@ func (n Neighbor) Clone() (clone Neighbor) {
// validatedHostname returns h if it's a valid hostname, or an empty string // validatedHostname returns h if it's a valid hostname, or an empty string
// otherwise, logging the validation error. // otherwise, logging the validation error.
func validatedHostname(h string) (host string) { func validatedHostname(logger *slog.Logger, h string) (host string) {
err := netutil.ValidateHostname(h) err := netutil.ValidateHostname(h)
if err != nil { if err != nil {
log.Debug("arpdb: parsing arp output: host: %s", err) logger.Debug("parsing host of arp output", slogutil.KeyError, err)
return "" return ""
} }
@ -132,15 +155,16 @@ func (ns *neighs) reset(with []Neighbor) {
// parseNeighsFunc parses the text from sc as if it'd be an output of some // parseNeighsFunc parses the text from sc as if it'd be an output of some
// ARP-related command. lenHint is a hint for the size of the allocated slice // ARP-related command. lenHint is a hint for the size of the allocated slice
// of Neighbors. // of Neighbors.
type parseNeighsFunc func(sc *bufio.Scanner, lenHint int) (ns []Neighbor) type parseNeighsFunc func(logger *slog.Logger, sc *bufio.Scanner, lenHint int) (ns []Neighbor)
// cmdARPDB is the implementation of the [Interface] that uses command line to // cmdARPDB is the implementation of the [Interface] that uses command line to
// retrieve data. // retrieve data.
type cmdARPDB struct { type cmdARPDB struct {
parse parseNeighsFunc logger *slog.Logger
ns *neighs parse parseNeighsFunc
cmd string ns *neighs
args []string cmd string
args []string
} }
// type check // type check
@ -158,7 +182,7 @@ func (arp *cmdARPDB) Refresh() (err error) {
} }
sc := bufio.NewScanner(bytes.NewReader(out)) sc := bufio.NewScanner(bytes.NewReader(out))
ns := arp.parse(sc, arp.ns.len()) ns := arp.parse(arp.logger, sc, arp.ns.len())
if err = sc.Err(); err != nil { if err = sc.Err(); err != nil {
// TODO(e.burkov): This error seems unreachable. Investigate. // TODO(e.burkov): This error seems unreachable. Investigate.
return fmt.Errorf("scanning the output: %w", err) return fmt.Errorf("scanning the output: %w", err)

31
internal/arpdb/arpdb_bsd.go
View File

@ -4,17 +4,17 @@ package arpdb
import ( import (
"bufio" "bufio"
"net" "log/slog"
"net/netip"
"strings" "strings"
"sync" "sync"
"github.com/AdguardTeam/golibs/log" "github.com/AdguardTeam/golibs/logutil/slogutil"
) )
func newARPDB() (arp *cmdARPDB) { func newARPDB(logger *slog.Logger) (arp *cmdARPDB) {
return &cmdARPDB{ return &cmdARPDB{
parse: parseArpA, logger: logger,
parse: parseArpA,
ns: &neighs{ ns: &neighs{
mu: &sync.RWMutex{}, mu: &sync.RWMutex{},
ns: make([]Neighbor, 0), ns: make([]Neighbor, 0),
@ -33,7 +33,7 @@ func newARPDB() (arp *cmdARPDB) {
// The expected input format: // The expected input format:
// //
// host.name (192.168.0.1) at ff:ff:ff:ff:ff:ff on en0 ifscope [ethernet] // host.name (192.168.0.1) at ff:ff:ff:ff:ff:ff on en0 ifscope [ethernet]
func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) { func parseArpA(logger *slog.Logger, sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
ns = make([]Neighbor, 0, lenHint) ns = make([]Neighbor, 0, lenHint)
for sc.Scan() { for sc.Scan() {
ln := sc.Text() ln := sc.Text()
@ -48,26 +48,15 @@ func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
continue continue
} }
ip, err := netip.ParseAddr(ipStr[1 : len(ipStr)-1]) host := validatedHostname(logger, fields[0])
n, err := newNeighbor(host, ipStr[1:len(ipStr)-1], fields[3])
if err != nil { if err != nil {
log.Debug("arpdb: parsing arp output: ip: %s", err) logger.Debug("parsing arp output", "line", ln, slogutil.KeyError, err)
continue continue
} }
hwStr := fields[3] ns = append(ns, *n)
mac, err := net.ParseMAC(hwStr)
if err != nil {
log.Debug("arpdb: parsing arp output: mac: %s", err)
continue
}
ns = append(ns, Neighbor{
IP: ip,
MAC: mac,
Name: validatedHostname(fields[0]),
})
} }
return ns return ns

8
internal/arpdb/arpdb_internal_test.go
View File

@ -11,6 +11,7 @@ import (
"testing" "testing"
"github.com/AdguardTeam/golibs/errors" "github.com/AdguardTeam/golibs/errors"
"github.com/AdguardTeam/golibs/logutil/slogutil"
"github.com/AdguardTeam/golibs/testutil" "github.com/AdguardTeam/golibs/testutil"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
@ -61,7 +62,7 @@ func (s mapShell) RunCmd(cmd string, args ...string) (code int, out []byte, err
func Test_New(t *testing.T) { func Test_New(t *testing.T) {
var a Interface var a Interface
require.NotPanics(t, func() { a = New() }) require.NotPanics(t, func() { a = New(slogutil.NewDiscardLogger()) })
assert.NotNil(t, a) assert.NotNil(t, a)
} }
@ -201,8 +202,9 @@ func Test_NewARPDBs(t *testing.T) {
func TestCmdARPDB_arpa(t *testing.T) { func TestCmdARPDB_arpa(t *testing.T) {
a := &cmdARPDB{ a := &cmdARPDB{
cmd: "cmd", logger: slogutil.NewDiscardLogger(),
parse: parseArpA, cmd: "cmd",
parse: parseArpA,
ns: &neighs{ ns: &neighs{
mu: &sync.RWMutex{}, mu: &sync.RWMutex{},
ns: make([]Neighbor, 0), ns: make([]Neighbor, 0),

82
internal/arpdb/arpdb_linux.go
View File

@ -6,17 +6,18 @@ import (
"bufio" "bufio"
"fmt" "fmt"
"io/fs" "io/fs"
"log/slog"
"net" "net"
"net/netip" "net/netip"
"strings" "strings"
"sync" "sync"
"github.com/AdguardTeam/AdGuardHome/internal/aghos" "github.com/AdguardTeam/AdGuardHome/internal/aghos"
"github.com/AdguardTeam/golibs/log" "github.com/AdguardTeam/golibs/logutil/slogutil"
"github.com/AdguardTeam/golibs/stringutil" "github.com/AdguardTeam/golibs/stringutil"
) )
func newARPDB() (arp *arpdbs) { func newARPDB(logger *slog.Logger) (arp *arpdbs) {
// Use the common storage among the implementations. // Use the common storage among the implementations.
ns := &neighs{ ns := &neighs{
mu: &sync.RWMutex{}, mu: &sync.RWMutex{},
@ -39,9 +40,10 @@ func newARPDB() (arp *arpdbs) {
}, },
// Then, try "arp -a -n". // Then, try "arp -a -n".
&cmdARPDB{ &cmdARPDB{
parse: parseF, logger: logger,
ns: ns, parse: parseF,
cmd: "arp", ns: ns,
cmd: "arp",
// Use -n flag to avoid resolving the hostnames of the neighbors. // Use -n flag to avoid resolving the hostnames of the neighbors.
// By default ARP attempts to resolve the hostnames via DNS. See // By default ARP attempts to resolve the hostnames via DNS. See
// man 8 arp. // man 8 arp.
@ -51,10 +53,11 @@ func newARPDB() (arp *arpdbs) {
}, },
// Finally, try "ip neigh". // Finally, try "ip neigh".
&cmdARPDB{ &cmdARPDB{
parse: parseIPNeigh, logger: logger,
ns: ns, parse: parseIPNeigh,
cmd: "ip", ns: ns,
args: []string{"neigh"}, cmd: "ip",
args: []string{"neigh"},
}, },
) )
} }
@ -131,7 +134,7 @@ func (arp *fsysARPDB) Neighbors() (ns []Neighbor) {
// //
// IP address HW type Flags HW address Mask Device // IP address HW type Flags HW address Mask Device
// 192.168.11.98 0x1 0x2 5a:92:df:a9:7e:28 * wan // 192.168.11.98 0x1 0x2 5a:92:df:a9:7e:28 * wan
func parseArpAWrt(sc *bufio.Scanner, lenHint int) (ns []Neighbor) { func parseArpAWrt(logger *slog.Logger, sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
if !sc.Scan() { if !sc.Scan() {
// Skip the header. // Skip the header.
return return
@ -146,25 +149,14 @@ func parseArpAWrt(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
continue continue
} }
ip, err := netip.ParseAddr(fields[0]) n, err := newNeighbor("", fields[0], fields[3])
if err != nil { if err != nil {
log.Debug("arpdb: parsing arp output: ip: %s", err) logger.Debug("parsing arp output", "line", ln, slogutil.KeyError, err)
continue continue
} }
hwStr := fields[3] ns = append(ns, *n)
mac, err := net.ParseMAC(hwStr)
if err != nil {
log.Debug("arpdb: parsing arp output: mac: %s", err)
continue
}
ns = append(ns, Neighbor{
IP: ip,
MAC: mac,
})
} }
return ns return ns
@ -174,7 +166,7 @@ func parseArpAWrt(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
// expected input format: // expected input format:
// //
// hostname (192.168.1.1) at ab:cd:ef:ab:cd:ef [ether] on enp0s3 // hostname (192.168.1.1) at ab:cd:ef:ab:cd:ef [ether] on enp0s3
func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) { func parseArpA(logger *slog.Logger, sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
ns = make([]Neighbor, 0, lenHint) ns = make([]Neighbor, 0, lenHint)
for sc.Scan() { for sc.Scan() {
ln := sc.Text() ln := sc.Text()
@ -189,26 +181,15 @@ func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
continue continue
} }
ip, err := netip.ParseAddr(ipStr[1 : len(ipStr)-1]) host := validatedHostname(logger, fields[0])
n, err := newNeighbor(host, ipStr[1:len(ipStr)-1], fields[3])
if err != nil { if err != nil {
log.Debug("arpdb: parsing arp output: ip: %s", err) logger.Debug("parsing arp output", "line", ln, slogutil.KeyError, err)
continue continue
} }
hwStr := fields[3] ns = append(ns, *n)
mac, err := net.ParseMAC(hwStr)
if err != nil {
log.Debug("arpdb: parsing arp output: mac: %s", err)
continue
}
ns = append(ns, Neighbor{
IP: ip,
MAC: mac,
Name: validatedHostname(fields[0]),
})
} }
return ns return ns
@ -218,7 +199,7 @@ func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
// expected input format: // expected input format:
// //
// 192.168.1.1 dev enp0s3 lladdr ab:cd:ef:ab:cd:ef REACHABLE // 192.168.1.1 dev enp0s3 lladdr ab:cd:ef:ab:cd:ef REACHABLE
func parseIPNeigh(sc *bufio.Scanner, lenHint int) (ns []Neighbor) { func parseIPNeigh(logger *slog.Logger, sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
ns = make([]Neighbor, 0, lenHint) ns = make([]Neighbor, 0, lenHint)
for sc.Scan() { for sc.Scan() {
ln := sc.Text() ln := sc.Text()
@ -228,27 +209,14 @@ func parseIPNeigh(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
continue continue
} }
n := Neighbor{} n, err := newNeighbor("", fields[0], fields[4])
ip, err := netip.ParseAddr(fields[0])
if err != nil { if err != nil {
log.Debug("arpdb: parsing arp output: ip: %s", err) logger.Debug("parsing arp output", "line", ln, slogutil.KeyError, err)
continue continue
} else {
n.IP = ip
} }
mac, err := net.ParseMAC(fields[4]) ns = append(ns, *n)
if err != nil {
log.Debug("arpdb: parsing arp output: mac: %s", err)
continue
} else {
n.MAC = mac
}
ns = append(ns, n)
} }
return ns return ns

15
internal/arpdb/arpdb_linux_internal_test.go
View File

@ -9,6 +9,7 @@ import (
"testing" "testing"
"testing/fstest" "testing/fstest"
"github.com/AdguardTeam/golibs/logutil/slogutil"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
) )
@ -69,9 +70,10 @@ func TestCmdARPDB_linux(t *testing.T) {
t.Run("wrt", func(t *testing.T) { t.Run("wrt", func(t *testing.T) {
a := &cmdARPDB{ a := &cmdARPDB{
parse: parseArpAWrt, logger: slogutil.NewDiscardLogger(),
cmd: "arp", parse: parseArpAWrt,
args: []string{"-a"}, cmd: "arp",
args: []string{"-a"},
ns: &neighs{ ns: &neighs{
mu: &sync.RWMutex{}, mu: &sync.RWMutex{},
ns: make([]Neighbor, 0), ns: make([]Neighbor, 0),
@ -86,9 +88,10 @@ func TestCmdARPDB_linux(t *testing.T) {
t.Run("ip_neigh", func(t *testing.T) { t.Run("ip_neigh", func(t *testing.T) {
a := &cmdARPDB{ a := &cmdARPDB{
parse: parseIPNeigh, logger: slogutil.NewDiscardLogger(),
cmd: "ip", parse: parseIPNeigh,
args: []string{"neigh"}, cmd: "ip",
args: []string{"neigh"},
ns: &neighs{ ns: &neighs{
mu: &sync.RWMutex{}, mu: &sync.RWMutex{},
ns: make([]Neighbor, 0), ns: make([]Neighbor, 0),

31
internal/arpdb/arpdb_openbsd.go
View File

@ -4,17 +4,17 @@ package arpdb
import ( import (
"bufio" "bufio"
"net" "log/slog"
"net/netip"
"strings" "strings"
"sync" "sync"
"github.com/AdguardTeam/golibs/log" "github.com/AdguardTeam/golibs/logutil/slogutil"
) )
func newARPDB() (arp *cmdARPDB) { func newARPDB(logger *slog.Logger) (arp *cmdARPDB) {
return &cmdARPDB{ return &cmdARPDB{
parse: parseArpA, logger: logger,
parse: parseArpA,
ns: &neighs{ ns: &neighs{
mu: &sync.RWMutex{}, mu: &sync.RWMutex{},
ns: make([]Neighbor, 0), ns: make([]Neighbor, 0),
@ -34,7 +34,7 @@ func newARPDB() (arp *cmdARPDB) {
// //
// Host Ethernet Address Netif Expire Flags // Host Ethernet Address Netif Expire Flags
// 192.168.1.1 ab:cd:ef:ab:cd:ef em0 19m59s // 192.168.1.1 ab:cd:ef:ab:cd:ef em0 19m59s
func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) { func parseArpA(logger *slog.Logger, sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
// Skip the header. // Skip the header.
if !sc.Scan() { if !sc.Scan() {
return nil return nil
@ -49,27 +49,14 @@ func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
continue continue
} }
n := Neighbor{} n, err := newNeighbor("", fields[0], fields[1])
ip, err := netip.ParseAddr(fields[0])
if err != nil { if err != nil {
log.Debug("arpdb: parsing arp output: ip: %s", err) logger.Debug("parsing arp output", "line", ln, slogutil.KeyError, err)
continue continue
} else {
n.IP = ip
} }
mac, err := net.ParseMAC(fields[1]) ns = append(ns, *n)
if err != nil {
log.Debug("arpdb: parsing arp output: mac: %s", err)
continue
} else {
n.MAC = mac
}
ns = append(ns, n)
} }
return ns return ns

28
internal/arpdb/arpdb_windows.go
View File

@ -4,17 +4,17 @@ package arpdb
import ( import (
"bufio" "bufio"
"net" "log/slog"
"net/netip"
"strings" "strings"
"sync" "sync"
"github.com/AdguardTeam/golibs/log" "github.com/AdguardTeam/golibs/logutil/slogutil"
) )
func newARPDB() (arp *cmdARPDB) { func newARPDB(logger *slog.Logger) (arp *cmdARPDB) {
return &cmdARPDB{ return &cmdARPDB{
parse: parseArpA, logger: logger,
parse: parseArpA,
ns: &neighs{ ns: &neighs{
mu: &sync.RWMutex{}, mu: &sync.RWMutex{},
ns: make([]Neighbor, 0), ns: make([]Neighbor, 0),
@ -31,7 +31,7 @@ func newARPDB() (arp *cmdARPDB) {
// Internet Address Physical Address Type // Internet Address Physical Address Type
// 192.168.56.1 0a-00-27-00-00-00 dynamic // 192.168.56.1 0a-00-27-00-00-00 dynamic
// 192.168.56.255 ff-ff-ff-ff-ff-ff static // 192.168.56.255 ff-ff-ff-ff-ff-ff static
func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) { func parseArpA(logger *slog.Logger, sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
ns = make([]Neighbor, 0, lenHint) ns = make([]Neighbor, 0, lenHint)
for sc.Scan() { for sc.Scan() {
ln := sc.Text() ln := sc.Text()
@ -44,24 +44,14 @@ func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
continue continue
} }
ip, err := netip.ParseAddr(fields[0]) n, err := newNeighbor("", fields[0], fields[1])
if err != nil { if err != nil {
log.Debug("arpdb: parsing arp output: ip: %s", err) logger.Debug("parsing arp output", "line", ln, slogutil.KeyError, err)
continue continue
} }
mac, err := net.ParseMAC(fields[1]) ns = append(ns, *n)
if err != nil {
log.Debug("arpdb: parsing arp output: mac: %s", err)
continue
}
ns = append(ns, Neighbor{
IP: ip,
MAC: mac,
})
} }
return ns return ns

7
internal/home/home.go
View File

@ -39,6 +39,7 @@ import (
"github.com/AdguardTeam/golibs/errors" "github.com/AdguardTeam/golibs/errors"
"github.com/AdguardTeam/golibs/hostsfile" "github.com/AdguardTeam/golibs/hostsfile"
"github.com/AdguardTeam/golibs/log" "github.com/AdguardTeam/golibs/log"
"github.com/AdguardTeam/golibs/logutil/slogutil"
"github.com/AdguardTeam/golibs/netutil" "github.com/AdguardTeam/golibs/netutil"
"github.com/AdguardTeam/golibs/osutil" "github.com/AdguardTeam/golibs/osutil"
) )
@ -276,7 +277,7 @@ func setupOpts(opts options) (err error) {
} }
// initContextClients initializes Context clients and related fields. // initContextClients initializes Context clients and related fields.
func initContextClients() (err error) { func initContextClients(logger *slog.Logger) (err error) {
err = setupDNSFilteringConf(config.Filtering) err = setupDNSFilteringConf(config.Filtering)
if err != nil { if err != nil {
// Don't wrap the error, because it's informative enough as is. // Don't wrap the error, because it's informative enough as is.
@ -300,7 +301,7 @@ func initContextClients() (err error) {
var arpDB arpdb.Interface var arpDB arpdb.Interface
if config.Clients.Sources.ARP { if config.Clients.Sources.ARP {
arpDB = arpdb.New() arpDB = arpdb.New(logger.With(slogutil.KeyError, "arpdb"))
} }
return Context.clients.Init( return Context.clients.Init(
@ -582,7 +583,7 @@ func run(opts options, clientBuildFS fs.FS, done chan struct{}) {
// data first, but also to avoid relying on automatic Go init() function. // data first, but also to avoid relying on automatic Go init() function.
filtering.InitModule() filtering.InitModule()
err = initContextClients() err = initContextClients(slogLogger)
fatalOnError(err) fatalOnError(err)
err = setupOpts(opts) err = setupOpts(opts)