mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-11-28 18:08:51 +03:00
Pull request: 2846 cover aghnet vol.3
Merge in DNS/adguard-home from 2846-cover-aghnet-vol.3 to master Updates #2846. Squashed commit of the following: commit cb22987c43c17bbc8d098e65639cc84e2284bc7b Merge: cf995e9df31ffcc5
Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Wed Mar 30 15:01:10 2022 +0300 Merge branch 'master' into 2846-cover-aghnet-vol.3 commit cf995e9dce635f16e10406a61e2ab12f06407f1f Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Mon Mar 28 18:29:50 2022 +0300 aghnet: imp tests commit bc225fe8800633b29216840bc7d5b82d7c2d2bfb Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Mon Mar 28 18:03:06 2022 +0300 aghnet: imp tests commit a82eb6045495b94a2e81ced9a3ef5bfe65788e56 Merge: f80812490d562a7b
Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Mon Mar 28 17:39:13 2022 +0300 Merge branch 'master' into 2846-cover-aghnet-vol.3 commit f80812490c49f69655d409c6f015b069affa2f19 Merge: edccaa793603b1fc
Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Mon Mar 28 17:29:20 2022 +0300 Merge branch 'master' into 2846-cover-aghnet-vol.3 commit edccaa79fca061ffeea1985c293eed123b16a09c Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Mon Mar 28 13:53:40 2022 +0300 aghnet: imp tests commit 7c5028c92f0a6680516bda67c73e794182c9b825 Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Fri Mar 25 18:01:28 2022 +0300 aghnet: imp code & docs commit 7897c6b13e9be340ae8a71947a8a0bab82c682eb Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Fri Mar 25 17:11:46 2022 +0300 aghnet: imp coverage commit 1eef110af3bf721a0275c695bf27c31815abff04 Author: Eugene Burkov <E.Burkov@AdGuard.COM> Date: Wed Mar 23 21:10:29 2022 +0300 all: return byte slice
This commit is contained in:
parent
f31ffcc5d1
commit
5e71f5df6a
21 changed files with 789 additions and 391 deletions
|
@ -2,13 +2,11 @@ package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
|
||||||
"net"
|
"net"
|
||||||
"strings"
|
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/AdguardTeam/AdGuardHome/internal/aghos"
|
|
||||||
"github.com/AdguardTeam/golibs/errors"
|
"github.com/AdguardTeam/golibs/errors"
|
||||||
"github.com/AdguardTeam/golibs/netutil"
|
"github.com/AdguardTeam/golibs/netutil"
|
||||||
)
|
)
|
||||||
|
@ -116,50 +114,33 @@ func (ns *neighs) reset(with []Neighbor) {
|
||||||
// of Neighbors.
|
// of Neighbors.
|
||||||
type parseNeighsFunc func(sc *bufio.Scanner, lenHint int) (ns []Neighbor)
|
type parseNeighsFunc func(sc *bufio.Scanner, lenHint int) (ns []Neighbor)
|
||||||
|
|
||||||
// runCmdFunc is the function that runs some command and returns its output
|
|
||||||
// wrapped to be a io.Reader.
|
|
||||||
type runCmdFunc func() (r io.Reader, err error)
|
|
||||||
|
|
||||||
// cmdARPDB is the implementation of the ARPDB that uses command line to
|
// cmdARPDB is the implementation of the ARPDB that uses command line to
|
||||||
// retrieve data.
|
// retrieve data.
|
||||||
type cmdARPDB struct {
|
type cmdARPDB struct {
|
||||||
parse parseNeighsFunc
|
parse parseNeighsFunc
|
||||||
runcmd runCmdFunc
|
|
||||||
ns *neighs
|
ns *neighs
|
||||||
|
cmd string
|
||||||
|
args []string
|
||||||
}
|
}
|
||||||
|
|
||||||
// type check
|
// type check
|
||||||
var _ ARPDB = (*cmdARPDB)(nil)
|
var _ ARPDB = (*cmdARPDB)(nil)
|
||||||
|
|
||||||
// runCmd runs the cmd with it's args and returns the result wrapped to be an
|
|
||||||
// io.Reader. The error is returned either if the exit code retured by command
|
|
||||||
// not equals 0 or the execution itself failed.
|
|
||||||
func runCmd(cmd string, args ...string) (r io.Reader, err error) {
|
|
||||||
var code int
|
|
||||||
var out string
|
|
||||||
code, out, err = aghos.RunCommand(cmd, args...)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
} else if code != 0 {
|
|
||||||
return nil, fmt.Errorf("unexpected exit code %d", code)
|
|
||||||
}
|
|
||||||
|
|
||||||
return strings.NewReader(out), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Refresh implements the ARPDB interface for *cmdARPDB.
|
// Refresh implements the ARPDB interface for *cmdARPDB.
|
||||||
func (arp *cmdARPDB) Refresh() (err error) {
|
func (arp *cmdARPDB) Refresh() (err error) {
|
||||||
defer func() { err = errors.Annotate(err, "cmd arpdb: %w") }()
|
defer func() { err = errors.Annotate(err, "cmd arpdb: %w") }()
|
||||||
|
|
||||||
var r io.Reader
|
code, out, err := aghosRunCommand(arp.cmd, arp.args...)
|
||||||
r, err = arp.runcmd()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("running command: %w", err)
|
return fmt.Errorf("running command: %w", err)
|
||||||
|
} else if code != 0 {
|
||||||
|
return fmt.Errorf("running command: unexpected exit code %d", code)
|
||||||
}
|
}
|
||||||
|
|
||||||
sc := bufio.NewScanner(r)
|
sc := bufio.NewScanner(bytes.NewReader(out))
|
||||||
ns := arp.parse(sc, arp.ns.len())
|
ns := arp.parse(sc, arp.ns.len())
|
||||||
if err = sc.Err(); err != nil {
|
if err = sc.Err(); err != nil {
|
||||||
|
// TODO(e.burkov): This error seems unreachable. Investigate.
|
||||||
return fmt.Errorf("scanning the output: %w", err)
|
return fmt.Errorf("scanning the output: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -180,8 +161,7 @@ func (arp *cmdARPDB) Neighbors() (ns []Neighbor) {
|
||||||
type arpdbs struct {
|
type arpdbs struct {
|
||||||
// arps is the set of ARPDB implementations to range through.
|
// arps is the set of ARPDB implementations to range through.
|
||||||
arps []ARPDB
|
arps []ARPDB
|
||||||
// last is the last succeeded ARPDB index.
|
neighs
|
||||||
last int
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// newARPDBs returns a properly initialized *arpdbs. It begins refreshing from
|
// newARPDBs returns a properly initialized *arpdbs. It begins refreshing from
|
||||||
|
@ -189,7 +169,10 @@ type arpdbs struct {
|
||||||
func newARPDBs(arps ...ARPDB) (arp *arpdbs) {
|
func newARPDBs(arps ...ARPDB) (arp *arpdbs) {
|
||||||
return &arpdbs{
|
return &arpdbs{
|
||||||
arps: arps,
|
arps: arps,
|
||||||
last: 0,
|
neighs: neighs{
|
||||||
|
mu: &sync.RWMutex{},
|
||||||
|
ns: make([]Neighbor, 0),
|
||||||
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -199,20 +182,18 @@ var _ ARPDB = (*arpdbs)(nil)
|
||||||
// Refresh implements the ARPDB interface for *arpdbs.
|
// Refresh implements the ARPDB interface for *arpdbs.
|
||||||
func (arp *arpdbs) Refresh() (err error) {
|
func (arp *arpdbs) Refresh() (err error) {
|
||||||
var errs []error
|
var errs []error
|
||||||
l := len(arp.arps)
|
|
||||||
// Start from the last succeeded implementation.
|
|
||||||
for i := 0; i < l; i++ {
|
|
||||||
cur := (arp.last + i) % l
|
|
||||||
err = arp.arps[cur].Refresh()
|
|
||||||
if err == nil {
|
|
||||||
// The succeeded implementation found so update the last succeeded
|
|
||||||
// index.
|
|
||||||
arp.last = cur
|
|
||||||
|
|
||||||
return nil
|
for _, a := range arp.arps {
|
||||||
|
err = a.Refresh()
|
||||||
|
if err != nil {
|
||||||
|
errs = append(errs, err)
|
||||||
|
|
||||||
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
errs = append(errs, err)
|
arp.reset(a.Neighbors())
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(errs) > 0 {
|
if len(errs) > 0 {
|
||||||
|
@ -223,10 +204,8 @@ func (arp *arpdbs) Refresh() (err error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Neighbors implements the ARPDB interface for *arpdbs.
|
// Neighbors implements the ARPDB interface for *arpdbs.
|
||||||
|
//
|
||||||
|
// TODO(e.burkov): Think of a way to avoid cloning the slice twice.
|
||||||
func (arp *arpdbs) Neighbors() (ns []Neighbor) {
|
func (arp *arpdbs) Neighbors() (ns []Neighbor) {
|
||||||
if l := len(arp.arps); l > 0 && arp.last < l {
|
return arp.clone()
|
||||||
return arp.arps[arp.last].Neighbors()
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -16,11 +16,12 @@ import (
|
||||||
func newARPDB() *cmdARPDB {
|
func newARPDB() *cmdARPDB {
|
||||||
return &cmdARPDB{
|
return &cmdARPDB{
|
||||||
parse: parseArpA,
|
parse: parseArpA,
|
||||||
runcmd: rcArpA,
|
|
||||||
ns: &neighs{
|
ns: &neighs{
|
||||||
mu: &sync.RWMutex{},
|
mu: &sync.RWMutex{},
|
||||||
ns: make([]Neighbor, 0),
|
ns: make([]Neighbor, 0),
|
||||||
},
|
},
|
||||||
|
cmd: "arp",
|
||||||
|
args: []string{"-a"},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -6,7 +6,6 @@ package aghnet
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"net"
|
"net"
|
||||||
"strings"
|
"strings"
|
||||||
|
@ -34,11 +33,25 @@ func newARPDB() (arp *arpdbs) {
|
||||||
|
|
||||||
return newARPDBs(
|
return newARPDBs(
|
||||||
// Try /proc/net/arp first.
|
// Try /proc/net/arp first.
|
||||||
&fsysARPDB{ns: ns, fsys: aghos.RootDirFS(), filename: "proc/net/arp"},
|
&fsysARPDB{
|
||||||
// Try "arp -a" then.
|
ns: ns,
|
||||||
&cmdARPDB{parse: parseF, runcmd: rcArpA, ns: ns},
|
fsys: rootDirFS,
|
||||||
// Try "ip neigh" finally.
|
filename: "proc/net/arp",
|
||||||
&cmdARPDB{parse: parseIPNeigh, runcmd: rcIPNeigh, ns: ns},
|
},
|
||||||
|
// Then, try "arp -a".
|
||||||
|
&cmdARPDB{
|
||||||
|
parse: parseF,
|
||||||
|
ns: ns,
|
||||||
|
cmd: "arp",
|
||||||
|
args: []string{"-a"},
|
||||||
|
},
|
||||||
|
// Finally, try "ip neigh".
|
||||||
|
&cmdARPDB{
|
||||||
|
parse: parseIPNeigh,
|
||||||
|
ns: ns,
|
||||||
|
cmd: "ip",
|
||||||
|
args: []string{"neigh"},
|
||||||
|
},
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -187,11 +200,6 @@ func parseArpA(sc *bufio.Scanner, lenHint int) (ns []Neighbor) {
|
||||||
return ns
|
return ns
|
||||||
}
|
}
|
||||||
|
|
||||||
// rcIPNeigh runs "ip neigh".
|
|
||||||
func rcIPNeigh() (r io.Reader, err error) {
|
|
||||||
return runCmd("ip", "neigh")
|
|
||||||
}
|
|
||||||
|
|
||||||
// parseIPNeigh parses the output of the "ip neigh" command on Linux. The
|
// parseIPNeigh parses the output of the "ip neigh" command on Linux. The
|
||||||
// expected input format:
|
// expected input format:
|
||||||
//
|
//
|
||||||
|
|
|
@ -4,11 +4,10 @@
|
||||||
package aghnet
|
package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io"
|
|
||||||
"net"
|
"net"
|
||||||
"strings"
|
|
||||||
"sync"
|
"sync"
|
||||||
"testing"
|
"testing"
|
||||||
|
"testing/fstest"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
|
@ -43,6 +42,8 @@ var wantNeighs = []Neighbor{{
|
||||||
}}
|
}}
|
||||||
|
|
||||||
func TestFSysARPDB(t *testing.T) {
|
func TestFSysARPDB(t *testing.T) {
|
||||||
|
require.NoError(t, fstest.TestFS(testdata, "proc_net_arp"))
|
||||||
|
|
||||||
a := &fsysARPDB{
|
a := &fsysARPDB{
|
||||||
ns: &neighs{
|
ns: &neighs{
|
||||||
mu: &sync.RWMutex{},
|
mu: &sync.RWMutex{},
|
||||||
|
@ -59,10 +60,18 @@ func TestFSysARPDB(t *testing.T) {
|
||||||
assert.Equal(t, wantNeighs, ns)
|
assert.Equal(t, wantNeighs, ns)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCmdARPDB_arpawrt(t *testing.T) {
|
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{
|
a := &cmdARPDB{
|
||||||
parse: parseArpAWrt,
|
parse: parseArpAWrt,
|
||||||
runcmd: func() (r io.Reader, err error) { return strings.NewReader(arpAOutputWrt), nil },
|
cmd: "arp",
|
||||||
|
args: []string{"-a"},
|
||||||
ns: &neighs{
|
ns: &neighs{
|
||||||
mu: &sync.RWMutex{},
|
mu: &sync.RWMutex{},
|
||||||
ns: make([]Neighbor, 0),
|
ns: make([]Neighbor, 0),
|
||||||
|
@ -73,12 +82,13 @@ func TestCmdARPDB_arpawrt(t *testing.T) {
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
assert.Equal(t, wantNeighs, a.Neighbors())
|
assert.Equal(t, wantNeighs, a.Neighbors())
|
||||||
}
|
})
|
||||||
|
|
||||||
func TestCmdARPDB_ipneigh(t *testing.T) {
|
t.Run("ip_neigh", func(t *testing.T) {
|
||||||
a := &cmdARPDB{
|
a := &cmdARPDB{
|
||||||
parse: parseIPNeigh,
|
parse: parseIPNeigh,
|
||||||
runcmd: func() (r io.Reader, err error) { return strings.NewReader(ipNeighOutput), nil },
|
cmd: "ip",
|
||||||
|
args: []string{"neigh"},
|
||||||
ns: &neighs{
|
ns: &neighs{
|
||||||
mu: &sync.RWMutex{},
|
mu: &sync.RWMutex{},
|
||||||
ns: make([]Neighbor, 0),
|
ns: make([]Neighbor, 0),
|
||||||
|
@ -88,4 +98,5 @@ func TestCmdARPDB_ipneigh(t *testing.T) {
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
assert.Equal(t, wantNeighs, a.Neighbors())
|
assert.Equal(t, wantNeighs, a.Neighbors())
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
|
@ -14,12 +14,13 @@ import (
|
||||||
|
|
||||||
func newARPDB() *cmdARPDB {
|
func newARPDB() *cmdARPDB {
|
||||||
return &cmdARPDB{
|
return &cmdARPDB{
|
||||||
runcmd: rcArpA,
|
|
||||||
parse: parseArpA,
|
parse: parseArpA,
|
||||||
ns: &neighs{
|
ns: &neighs{
|
||||||
mu: &sync.RWMutex{},
|
mu: &sync.RWMutex{},
|
||||||
ns: make([]Neighbor, 0),
|
ns: make([]Neighbor, 0),
|
||||||
},
|
},
|
||||||
|
cmd: "arp",
|
||||||
|
args: []string{"-a"},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,12 +1,9 @@
|
||||||
package aghnet
|
package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io"
|
|
||||||
"net"
|
"net"
|
||||||
"strings"
|
|
||||||
"sync"
|
"sync"
|
||||||
"testing"
|
"testing"
|
||||||
"testing/iotest"
|
|
||||||
|
|
||||||
"github.com/AdguardTeam/golibs/errors"
|
"github.com/AdguardTeam/golibs/errors"
|
||||||
"github.com/AdguardTeam/golibs/testutil"
|
"github.com/AdguardTeam/golibs/testutil"
|
||||||
|
@ -16,9 +13,7 @@ import (
|
||||||
|
|
||||||
func TestNewARPDB(t *testing.T) {
|
func TestNewARPDB(t *testing.T) {
|
||||||
var a ARPDB
|
var a ARPDB
|
||||||
require.NotPanics(t, func() {
|
require.NotPanics(t, func() { a = NewARPDB() })
|
||||||
a = NewARPDB()
|
|
||||||
})
|
|
||||||
|
|
||||||
assert.NotNil(t, a)
|
assert.NotNil(t, a)
|
||||||
}
|
}
|
||||||
|
@ -135,11 +130,11 @@ func TestARPDBS(t *testing.T) {
|
||||||
assert.Equal(t, 1, succRefrCount)
|
assert.Equal(t, 1, succRefrCount)
|
||||||
assert.NotEmpty(t, a.Neighbors())
|
assert.NotEmpty(t, a.Neighbors())
|
||||||
|
|
||||||
// Only the last succeeded ARPDB should be used.
|
// Unstable ARPDB should refresh successfully again.
|
||||||
err = a.Refresh()
|
err = a.Refresh()
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
assert.Equal(t, 2, succRefrCount)
|
assert.Equal(t, 1, succRefrCount)
|
||||||
assert.NotEmpty(t, a.Neighbors())
|
assert.NotEmpty(t, a.Neighbors())
|
||||||
})
|
})
|
||||||
|
|
||||||
|
@ -153,6 +148,7 @@ func TestARPDBS(t *testing.T) {
|
||||||
|
|
||||||
func TestCmdARPDB_arpa(t *testing.T) {
|
func TestCmdARPDB_arpa(t *testing.T) {
|
||||||
a := &cmdARPDB{
|
a := &cmdARPDB{
|
||||||
|
cmd: "cmd",
|
||||||
parse: parseArpA,
|
parse: parseArpA,
|
||||||
ns: &neighs{
|
ns: &neighs{
|
||||||
mu: &sync.RWMutex{},
|
mu: &sync.RWMutex{},
|
||||||
|
@ -161,7 +157,8 @@ func TestCmdARPDB_arpa(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
t.Run("arp_a", func(t *testing.T) {
|
t.Run("arp_a", func(t *testing.T) {
|
||||||
a.runcmd = func() (r io.Reader, err error) { return strings.NewReader(arpAOutput), nil }
|
sh := theOnlyCmd("cmd", 0, arpAOutput, nil)
|
||||||
|
substShell(t, sh.RunCmd)
|
||||||
|
|
||||||
err := a.Refresh()
|
err := a.Refresh()
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
@ -170,32 +167,20 @@ func TestCmdARPDB_arpa(t *testing.T) {
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("runcmd_error", func(t *testing.T) {
|
t.Run("runcmd_error", func(t *testing.T) {
|
||||||
a.runcmd = func() (r io.Reader, err error) { return nil, errors.Error("can't run") }
|
sh := theOnlyCmd("cmd", 0, "", errors.Error("can't run"))
|
||||||
|
substShell(t, sh.RunCmd)
|
||||||
|
|
||||||
err := a.Refresh()
|
err := a.Refresh()
|
||||||
testutil.AssertErrorMsg(t, "cmd arpdb: running command: can't run", err)
|
testutil.AssertErrorMsg(t, "cmd arpdb: running command: can't run", err)
|
||||||
})
|
})
|
||||||
}
|
|
||||||
|
|
||||||
func TestCmdARPDB_errors(t *testing.T) {
|
t.Run("bad_code", func(t *testing.T) {
|
||||||
const errRead errors.Error = "can't read"
|
sh := theOnlyCmd("cmd", 1, "", nil)
|
||||||
|
substShell(t, sh.RunCmd)
|
||||||
|
|
||||||
badReaderRunCmd := runCmdFunc(func() (r io.Reader, err error) {
|
err := a.Refresh()
|
||||||
return iotest.ErrReader(errRead), nil
|
testutil.AssertErrorMsg(t, "cmd arpdb: running command: unexpected exit code 1", err)
|
||||||
})
|
})
|
||||||
|
|
||||||
a := &cmdARPDB{
|
|
||||||
runcmd: badReaderRunCmd,
|
|
||||||
parse: parseArpA,
|
|
||||||
ns: &neighs{
|
|
||||||
mu: &sync.RWMutex{},
|
|
||||||
ns: make([]Neighbor, 0),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
const wantErrMsg string = "cmd arpdb: scanning the output: " + string(errRead)
|
|
||||||
|
|
||||||
testutil.AssertErrorMsg(t, wantErrMsg, a.Refresh())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestEmptyARPDB(t *testing.T) {
|
func TestEmptyARPDB(t *testing.T) {
|
||||||
|
|
|
@ -1,13 +0,0 @@
|
||||||
//go:build !windows
|
|
||||||
// +build !windows
|
|
||||||
|
|
||||||
package aghnet
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io"
|
|
||||||
)
|
|
||||||
|
|
||||||
// rcArpA runs "arp -a".
|
|
||||||
func rcArpA() (r io.Reader, err error) {
|
|
||||||
return runCmd("arp", "-a")
|
|
||||||
}
|
|
|
@ -5,7 +5,6 @@ package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"io"
|
|
||||||
"net"
|
"net"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
|
@ -13,20 +12,16 @@ import (
|
||||||
|
|
||||||
func newARPDB() *cmdARPDB {
|
func newARPDB() *cmdARPDB {
|
||||||
return &cmdARPDB{
|
return &cmdARPDB{
|
||||||
runcmd: rcArpA,
|
parse: parseArpA,
|
||||||
ns: &neighs{
|
ns: &neighs{
|
||||||
mu: &sync.RWMutex{},
|
mu: &sync.RWMutex{},
|
||||||
ns: make([]Neighbor, 0),
|
ns: make([]Neighbor, 0),
|
||||||
},
|
},
|
||||||
parse: parseArpA,
|
cmd: "arp",
|
||||||
|
args: []string{"/a"},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// rcArpA runs "arp /a".
|
|
||||||
func rcArpA() (r io.Reader, err error) {
|
|
||||||
return runCmd("arp", "/a")
|
|
||||||
}
|
|
||||||
|
|
||||||
// parseArpA parses the output of the "arp /a" command on Windows. The expected
|
// parseArpA parses the output of the "arp /a" command on Windows. The expected
|
||||||
// input format (the first line is empty):
|
// input format (the first line is empty):
|
||||||
//
|
//
|
||||||
|
|
|
@ -286,6 +286,8 @@ func TestHostsContainer_Translate(t *testing.T) {
|
||||||
OnClose: func() (err error) { panic("not implemented") },
|
OnClose: func() (err error) { panic("not implemented") },
|
||||||
}
|
}
|
||||||
|
|
||||||
|
require.NoError(t, fstest.TestFS(testdata, "etc_hosts"))
|
||||||
|
|
||||||
hc, err := NewHostsContainer(0, testdata, &stubWatcher, "etc_hosts")
|
hc, err := NewHostsContainer(0, testdata, &stubWatcher, "etc_hosts")
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
testutil.CleanupAndRequireSuccess(t, hc.Close)
|
testutil.CleanupAndRequireSuccess(t, hc.Close)
|
||||||
|
@ -358,6 +360,8 @@ func TestHostsContainer_Translate(t *testing.T) {
|
||||||
func TestHostsContainer(t *testing.T) {
|
func TestHostsContainer(t *testing.T) {
|
||||||
const listID = 1234
|
const listID = 1234
|
||||||
|
|
||||||
|
require.NoError(t, fstest.TestFS(testdata, "etc_hosts"))
|
||||||
|
|
||||||
testCases := []struct {
|
testCases := []struct {
|
||||||
want []*rules.DNSRewrite
|
want []*rules.DNSRewrite
|
||||||
name string
|
name string
|
||||||
|
|
|
@ -2,19 +2,27 @@
|
||||||
package aghnet
|
package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
"os/exec"
|
|
||||||
"strings"
|
|
||||||
"syscall"
|
"syscall"
|
||||||
|
|
||||||
|
"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/log"
|
||||||
"github.com/AdguardTeam/golibs/netutil"
|
"github.com/AdguardTeam/golibs/netutil"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// aghosRunCommand is the function to run shell commands. It's an unexported
|
||||||
|
// variable instead of a direct call to make it substitutable in tests.
|
||||||
|
var aghosRunCommand = aghos.RunCommand
|
||||||
|
|
||||||
|
// rootDirFS is the filesystem pointing to the root directory. It's an
|
||||||
|
// unexported variable instead to make it substitutable in tests.
|
||||||
|
var rootDirFS = aghos.RootDirFS()
|
||||||
|
|
||||||
// ErrNoStaticIPInfo is returned by IfaceHasStaticIP when no information about
|
// ErrNoStaticIPInfo is returned by IfaceHasStaticIP when no information about
|
||||||
// the IP being static is available.
|
// the IP being static is available.
|
||||||
const ErrNoStaticIPInfo errors.Error = "no information about static ip"
|
const ErrNoStaticIPInfo errors.Error = "no information about static ip"
|
||||||
|
@ -32,22 +40,29 @@ func IfaceSetStaticIP(ifaceName string) (err error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// GatewayIP returns IP address of interface's gateway.
|
// GatewayIP returns IP address of interface's gateway.
|
||||||
func GatewayIP(ifaceName string) net.IP {
|
//
|
||||||
cmd := exec.Command("ip", "route", "show", "dev", ifaceName)
|
// TODO(e.burkov): Investigate if the gateway address may be fetched in another
|
||||||
log.Tracef("executing %s %v", cmd.Path, cmd.Args)
|
// way since not every machine has the software installed.
|
||||||
d, err := cmd.Output()
|
func GatewayIP(ifaceName string) (ip net.IP) {
|
||||||
if err != nil || cmd.ProcessState.ExitCode() != 0 {
|
code, out, err := aghosRunCommand("ip", "route", "show", "dev", ifaceName)
|
||||||
|
if err != nil {
|
||||||
|
log.Debug("%s", err)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
} else if code != 0 {
|
||||||
|
log.Debug("fetching gateway ip: unexpected exit code: %d", code)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
fields := strings.Fields(string(d))
|
fields := bytes.Fields(out)
|
||||||
// The meaningful "ip route" command output should contain the word
|
// The meaningful "ip route" command output should contain the word
|
||||||
// "default" at first field and default gateway IP address at third field.
|
// "default" at first field and default gateway IP address at third field.
|
||||||
if len(fields) < 3 || fields[0] != "default" {
|
if len(fields) < 3 || string(fields[0]) != "default" {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return net.ParseIP(fields[2])
|
return net.ParseIP(string(fields[2]))
|
||||||
}
|
}
|
||||||
|
|
||||||
// CanBindPort checks if we can bind to the given port.
|
// CanBindPort checks if we can bind to the given port.
|
||||||
|
@ -101,7 +116,7 @@ func (iface NetInterface) MarshalJSON() ([]byte, error) {
|
||||||
|
|
||||||
// GetValidNetInterfacesForWeb returns interfaces that are eligible for DNS and WEB only
|
// GetValidNetInterfacesForWeb returns interfaces that are eligible for DNS and WEB only
|
||||||
// we do not return link-local addresses here
|
// we do not return link-local addresses here
|
||||||
func GetValidNetInterfacesForWeb() ([]*NetInterface, error) {
|
func GetValidNetInterfacesForWeb() (netInterfaces []*NetInterface, err error) {
|
||||||
ifaces, err := net.Interfaces()
|
ifaces, err := net.Interfaces()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("couldn't get interfaces: %w", err)
|
return nil, fmt.Errorf("couldn't get interfaces: %w", err)
|
||||||
|
@ -109,8 +124,6 @@ func GetValidNetInterfacesForWeb() ([]*NetInterface, error) {
|
||||||
return nil, errors.Error("couldn't find any legible interface")
|
return nil, errors.Error("couldn't find any legible interface")
|
||||||
}
|
}
|
||||||
|
|
||||||
var netInterfaces []*NetInterface
|
|
||||||
|
|
||||||
for _, iface := range ifaces {
|
for _, iface := range ifaces {
|
||||||
var addrs []net.Addr
|
var addrs []net.Addr
|
||||||
addrs, err = iface.Addrs()
|
addrs, err = iface.Addrs()
|
||||||
|
@ -130,12 +143,14 @@ func GetValidNetInterfacesForWeb() ([]*NetInterface, error) {
|
||||||
ipNet, ok := addr.(*net.IPNet)
|
ipNet, ok := addr.(*net.IPNet)
|
||||||
if !ok {
|
if !ok {
|
||||||
// Should be net.IPNet, this is weird.
|
// Should be net.IPNet, this is weird.
|
||||||
return nil, fmt.Errorf("got iface.Addrs() element %s that is not net.IPNet, it is %T", addr, addr)
|
return nil, fmt.Errorf("got %s that is not net.IPNet, it is %T", addr, addr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ignore link-local.
|
// Ignore link-local.
|
||||||
if ipNet.IP.IsLinkLocalUnicast() {
|
if ipNet.IP.IsLinkLocalUnicast() {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
netIface.Addresses = append(netIface.Addresses, ipNet.IP)
|
netIface.Addresses = append(netIface.Addresses, ipNet.IP)
|
||||||
netIface.Subnets = append(netIface.Subnets, ipNet)
|
netIface.Subnets = append(netIface.Subnets, ipNet)
|
||||||
}
|
}
|
||||||
|
|
|
@ -4,10 +4,11 @@
|
||||||
package aghnet
|
package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bufio"
|
||||||
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"io"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/AdguardTeam/AdGuardHome/internal/aghos"
|
"github.com/AdguardTeam/AdGuardHome/internal/aghos"
|
||||||
"github.com/AdguardTeam/golibs/errors"
|
"github.com/AdguardTeam/golibs/errors"
|
||||||
|
@ -27,7 +28,7 @@ func canBindPrivilegedPorts() (can bool, err error) {
|
||||||
return aghos.HaveAdminRights()
|
return aghos.HaveAdminRights()
|
||||||
}
|
}
|
||||||
|
|
||||||
func ifaceHasStaticIP(ifaceName string) (bool, error) {
|
func ifaceHasStaticIP(ifaceName string) (ok bool, err error) {
|
||||||
portInfo, err := getCurrentHardwarePortInfo(ifaceName)
|
portInfo, err := getCurrentHardwarePortInfo(ifaceName)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
|
@ -36,9 +37,10 @@ func ifaceHasStaticIP(ifaceName string) (bool, error) {
|
||||||
return portInfo.static, nil
|
return portInfo.static, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// getCurrentHardwarePortInfo gets information for the specified network interface.
|
// getCurrentHardwarePortInfo gets information for the specified network
|
||||||
|
// interface.
|
||||||
func getCurrentHardwarePortInfo(ifaceName string) (hardwarePortInfo, error) {
|
func getCurrentHardwarePortInfo(ifaceName string) (hardwarePortInfo, error) {
|
||||||
// First of all we should find hardware port name
|
// First of all we should find hardware port name.
|
||||||
m := getNetworkSetupHardwareReports()
|
m := getNetworkSetupHardwareReports()
|
||||||
hardwarePort, ok := m[ifaceName]
|
hardwarePort, ok := m[ifaceName]
|
||||||
if !ok {
|
if !ok {
|
||||||
|
@ -48,6 +50,10 @@ func getCurrentHardwarePortInfo(ifaceName string) (hardwarePortInfo, error) {
|
||||||
return getHardwarePortInfo(hardwarePort)
|
return getHardwarePortInfo(hardwarePort)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// hardwareReportsReg is the regular expression matching the lines of
|
||||||
|
// networksetup command output lines containing the interface information.
|
||||||
|
var hardwareReportsReg = regexp.MustCompile("Hardware Port: (.*?)\nDevice: (.*?)\n")
|
||||||
|
|
||||||
// getNetworkSetupHardwareReports parses the output of the `networksetup
|
// getNetworkSetupHardwareReports parses the output of the `networksetup
|
||||||
// -listallhardwareports` command it returns a map where the key is the
|
// -listallhardwareports` command it returns a map where the key is the
|
||||||
// interface name, and the value is the "hardware port" returns nil if it fails
|
// interface name, and the value is the "hardware port" returns nil if it fails
|
||||||
|
@ -56,54 +62,44 @@ func getCurrentHardwarePortInfo(ifaceName string) (hardwarePortInfo, error) {
|
||||||
// TODO(e.burkov): There should be more proper approach than parsing the
|
// TODO(e.burkov): There should be more proper approach than parsing the
|
||||||
// command output. For example, see
|
// command output. For example, see
|
||||||
// https://developer.apple.com/documentation/systemconfiguration.
|
// https://developer.apple.com/documentation/systemconfiguration.
|
||||||
func getNetworkSetupHardwareReports() map[string]string {
|
func getNetworkSetupHardwareReports() (reports map[string]string) {
|
||||||
_, out, err := aghos.RunCommand("networksetup", "-listallhardwareports")
|
_, out, err := aghosRunCommand("networksetup", "-listallhardwareports")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
re, err := regexp.Compile("Hardware Port: (.*?)\nDevice: (.*?)\n")
|
reports = make(map[string]string)
|
||||||
if err != nil {
|
|
||||||
return nil
|
matches := hardwareReportsReg.FindAllSubmatch(out, -1)
|
||||||
|
for _, m := range matches {
|
||||||
|
reports[string(m[2])] = string(m[1])
|
||||||
}
|
}
|
||||||
|
|
||||||
m := make(map[string]string)
|
return reports
|
||||||
|
|
||||||
matches := re.FindAllStringSubmatch(out, -1)
|
|
||||||
for i := range matches {
|
|
||||||
port := matches[i][1]
|
|
||||||
device := matches[i][2]
|
|
||||||
m[device] = port
|
|
||||||
}
|
|
||||||
|
|
||||||
return m
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func getHardwarePortInfo(hardwarePort string) (hardwarePortInfo, error) {
|
// hardwarePortReg is the regular expression matching the lines of networksetup
|
||||||
h := hardwarePortInfo{}
|
// command output lines containing the port information.
|
||||||
|
var hardwarePortReg = regexp.MustCompile("IP address: (.*?)\nSubnet mask: (.*?)\nRouter: (.*?)\n")
|
||||||
|
|
||||||
_, out, err := aghos.RunCommand("networksetup", "-getinfo", hardwarePort)
|
func getHardwarePortInfo(hardwarePort string) (h hardwarePortInfo, err error) {
|
||||||
|
_, out, err := aghosRunCommand("networksetup", "-getinfo", hardwarePort)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return h, err
|
return h, err
|
||||||
}
|
}
|
||||||
|
|
||||||
re := regexp.MustCompile("IP address: (.*?)\nSubnet mask: (.*?)\nRouter: (.*?)\n")
|
match := hardwarePortReg.FindSubmatch(out)
|
||||||
|
if len(match) != 4 {
|
||||||
match := re.FindStringSubmatch(out)
|
|
||||||
if len(match) == 0 {
|
|
||||||
return h, errors.Error("could not find hardware port info")
|
return h, errors.Error("could not find hardware port info")
|
||||||
}
|
}
|
||||||
|
|
||||||
h.name = hardwarePort
|
return hardwarePortInfo{
|
||||||
h.ip = match[1]
|
name: hardwarePort,
|
||||||
h.subnet = match[2]
|
ip: string(match[1]),
|
||||||
h.gatewayIP = match[3]
|
subnet: string(match[2]),
|
||||||
|
gatewayIP: string(match[3]),
|
||||||
if strings.Index(out, "Manual Configuration") == 0 {
|
static: bytes.Index(out, []byte("Manual Configuration")) == 0,
|
||||||
h.static = true
|
}, nil
|
||||||
}
|
|
||||||
|
|
||||||
return h, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func ifaceSetStaticIP(ifaceName string) (err error) {
|
func ifaceSetStaticIP(ifaceName string) (err error) {
|
||||||
|
@ -113,7 +109,7 @@ func ifaceSetStaticIP(ifaceName string) (err error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
if portInfo.static {
|
if portInfo.static {
|
||||||
return errors.Error("IP address is already static")
|
return errors.Error("ip address is already static")
|
||||||
}
|
}
|
||||||
|
|
||||||
dnsAddrs, err := getEtcResolvConfServers()
|
dnsAddrs, err := getEtcResolvConfServers()
|
||||||
|
@ -121,50 +117,62 @@ func ifaceSetStaticIP(ifaceName string) (err error) {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
args := make([]string, 0)
|
args := append([]string{"-setdnsservers", portInfo.name}, dnsAddrs...)
|
||||||
args = append(args, "-setdnsservers", portInfo.name)
|
|
||||||
args = append(args, dnsAddrs...)
|
|
||||||
|
|
||||||
// Setting DNS servers is necessary when configuring a static IP
|
// Setting DNS servers is necessary when configuring a static IP
|
||||||
code, _, err := aghos.RunCommand("networksetup", args...)
|
code, _, err := aghosRunCommand("networksetup", args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
} else if code != 0 {
|
||||||
if code != 0 {
|
|
||||||
return fmt.Errorf("failed to set DNS servers, code=%d", code)
|
return fmt.Errorf("failed to set DNS servers, code=%d", code)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Actually configures hardware port to have static IP
|
// Actually configures hardware port to have static IP
|
||||||
code, _, err = aghos.RunCommand("networksetup", "-setmanual",
|
code, _, err = aghosRunCommand(
|
||||||
portInfo.name, portInfo.ip, portInfo.subnet, portInfo.gatewayIP)
|
"networksetup",
|
||||||
|
"-setmanual",
|
||||||
|
portInfo.name,
|
||||||
|
portInfo.ip,
|
||||||
|
portInfo.subnet,
|
||||||
|
portInfo.gatewayIP,
|
||||||
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
} else if code != 0 {
|
||||||
if code != 0 {
|
|
||||||
return fmt.Errorf("failed to set DNS servers, code=%d", code)
|
return fmt.Errorf("failed to set DNS servers, code=%d", code)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// etcResolvConfReg is the regular expression matching the lines of resolv.conf
|
||||||
|
// file containing a name server information.
|
||||||
|
var etcResolvConfReg = regexp.MustCompile("nameserver ([a-zA-Z0-9.:]+)")
|
||||||
|
|
||||||
// getEtcResolvConfServers returns a list of nameservers configured in
|
// getEtcResolvConfServers returns a list of nameservers configured in
|
||||||
// /etc/resolv.conf.
|
// /etc/resolv.conf.
|
||||||
func getEtcResolvConfServers() ([]string, error) {
|
func getEtcResolvConfServers() (addrs []string, err error) {
|
||||||
body, err := os.ReadFile("/etc/resolv.conf")
|
const filename = "etc/resolv.conf"
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
re := regexp.MustCompile("nameserver ([a-zA-Z0-9.:]+)")
|
_, err = aghos.FileWalker(func(r io.Reader) (_ []string, _ bool, err error) {
|
||||||
|
sc := bufio.NewScanner(r)
|
||||||
matches := re.FindAllStringSubmatch(string(body), -1)
|
for sc.Scan() {
|
||||||
|
matches := etcResolvConfReg.FindAllStringSubmatch(sc.Text(), -1)
|
||||||
if len(matches) == 0 {
|
if len(matches) == 0 {
|
||||||
return nil, errors.Error("found no DNS servers in /etc/resolv.conf")
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
addrs := make([]string, 0)
|
for _, m := range matches {
|
||||||
for i := range matches {
|
addrs = append(addrs, m[1])
|
||||||
addrs = append(addrs, matches[i][1])
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, false, sc.Err()
|
||||||
|
}).Walk(rootDirFS, filename)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("parsing etc/resolv.conf file: %w", err)
|
||||||
|
} else if len(addrs) == 0 {
|
||||||
|
return nil, fmt.Errorf("found no dns servers in %s", filename)
|
||||||
}
|
}
|
||||||
|
|
||||||
return addrs, nil
|
return addrs, nil
|
||||||
|
|
261
internal/aghnet/net_darwin_test.go
Normal file
261
internal/aghnet/net_darwin_test.go
Normal file
|
@ -0,0 +1,261 @@
|
||||||
|
package aghnet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io/fs"
|
||||||
|
"testing"
|
||||||
|
"testing/fstest"
|
||||||
|
|
||||||
|
"github.com/AdguardTeam/AdGuardHome/internal/aghtest"
|
||||||
|
"github.com/AdguardTeam/golibs/errors"
|
||||||
|
"github.com/AdguardTeam/golibs/testutil"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestIfaceHasStaticIP(t *testing.T) {
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
shell mapShell
|
||||||
|
ifaceName string
|
||||||
|
wantHas assert.BoolAssertionFunc
|
||||||
|
wantErrMsg string
|
||||||
|
}{{
|
||||||
|
name: "success",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: nil,
|
||||||
|
out: "IP address: 1.2.3.4\nSubnet mask: 255.255.255.0\nRouter: 1.2.3.1\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ifaceName: "en0",
|
||||||
|
wantHas: assert.False,
|
||||||
|
wantErrMsg: ``,
|
||||||
|
}, {
|
||||||
|
name: "success_static",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: nil,
|
||||||
|
out: "Manual Configuration\nIP address: 1.2.3.4\n" +
|
||||||
|
"Subnet mask: 255.255.255.0\nRouter: 1.2.3.1\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ifaceName: "en0",
|
||||||
|
wantHas: assert.True,
|
||||||
|
wantErrMsg: ``,
|
||||||
|
}, {
|
||||||
|
name: "reports_error",
|
||||||
|
shell: theOnlyCmd(
|
||||||
|
"networksetup -listallhardwareports",
|
||||||
|
0,
|
||||||
|
"",
|
||||||
|
errors.Error("can't list"),
|
||||||
|
),
|
||||||
|
ifaceName: "en0",
|
||||||
|
wantHas: assert.False,
|
||||||
|
wantErrMsg: `could not find hardware port for en0`,
|
||||||
|
}, {
|
||||||
|
name: "port_error",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: errors.Error("can't get"),
|
||||||
|
out: ``,
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ifaceName: "en0",
|
||||||
|
wantHas: assert.False,
|
||||||
|
wantErrMsg: `can't get`,
|
||||||
|
}, {
|
||||||
|
name: "port_bad_output",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: nil,
|
||||||
|
out: "nothing meaningful",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ifaceName: "en0",
|
||||||
|
wantHas: assert.False,
|
||||||
|
wantErrMsg: `could not find hardware port info`,
|
||||||
|
}}
|
||||||
|
|
||||||
|
for _, tc := range testCases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
substShell(t, tc.shell.RunCmd)
|
||||||
|
|
||||||
|
has, err := IfaceHasStaticIP(tc.ifaceName)
|
||||||
|
testutil.AssertErrorMsg(t, tc.wantErrMsg, err)
|
||||||
|
|
||||||
|
tc.wantHas(t, has)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIfaceSetStaticIP(t *testing.T) {
|
||||||
|
succFsys := fstest.MapFS{
|
||||||
|
"etc/resolv.conf": &fstest.MapFile{
|
||||||
|
Data: []byte(`nameserver 1.1.1.1`),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
panicFsys := &aghtest.FS{
|
||||||
|
OnOpen: func(name string) (fs.File, error) { panic("not implemented") },
|
||||||
|
}
|
||||||
|
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
shell mapShell
|
||||||
|
fsys fs.FS
|
||||||
|
wantErrMsg string
|
||||||
|
}{{
|
||||||
|
name: "success",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: nil,
|
||||||
|
out: "IP address: 1.2.3.4\nSubnet mask: 255.255.255.0\nRouter: 1.2.3.1\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -setdnsservers hwport 1.1.1.1": {
|
||||||
|
err: nil,
|
||||||
|
out: "",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -setmanual hwport 1.2.3.4 255.255.255.0 1.2.3.1": {
|
||||||
|
err: nil,
|
||||||
|
out: "",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
fsys: succFsys,
|
||||||
|
wantErrMsg: ``,
|
||||||
|
}, {
|
||||||
|
name: "static_already",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: nil,
|
||||||
|
out: "Manual Configuration\nIP address: 1.2.3.4\n" +
|
||||||
|
"Subnet mask: 255.255.255.0\nRouter: 1.2.3.1\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
fsys: panicFsys,
|
||||||
|
wantErrMsg: `ip address is already static`,
|
||||||
|
}, {
|
||||||
|
name: "reports_error",
|
||||||
|
shell: theOnlyCmd(
|
||||||
|
"networksetup -listallhardwareports",
|
||||||
|
0,
|
||||||
|
"",
|
||||||
|
errors.Error("can't list"),
|
||||||
|
),
|
||||||
|
fsys: panicFsys,
|
||||||
|
wantErrMsg: `could not find hardware port for en0`,
|
||||||
|
}, {
|
||||||
|
name: "resolv_conf_error",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: nil,
|
||||||
|
out: "IP address: 1.2.3.4\nSubnet mask: 255.255.255.0\nRouter: 1.2.3.1\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
fsys: fstest.MapFS{
|
||||||
|
"etc/resolv.conf": &fstest.MapFile{
|
||||||
|
Data: []byte("this resolv.conf is invalid"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantErrMsg: `found no dns servers in etc/resolv.conf`,
|
||||||
|
}, {
|
||||||
|
name: "set_dns_error",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: nil,
|
||||||
|
out: "IP address: 1.2.3.4\nSubnet mask: 255.255.255.0\nRouter: 1.2.3.1\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -setdnsservers hwport 1.1.1.1": {
|
||||||
|
err: errors.Error("can't set"),
|
||||||
|
out: "",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
fsys: succFsys,
|
||||||
|
wantErrMsg: `can't set`,
|
||||||
|
}, {
|
||||||
|
name: "set_manual_error",
|
||||||
|
shell: mapShell{
|
||||||
|
"networksetup -listallhardwareports": {
|
||||||
|
err: nil,
|
||||||
|
out: "Hardware Port: hwport\nDevice: en0\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -getinfo hwport": {
|
||||||
|
err: nil,
|
||||||
|
out: "IP address: 1.2.3.4\nSubnet mask: 255.255.255.0\nRouter: 1.2.3.1\n",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -setdnsservers hwport 1.1.1.1": {
|
||||||
|
err: nil,
|
||||||
|
out: "",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
"networksetup -setmanual hwport 1.2.3.4 255.255.255.0 1.2.3.1": {
|
||||||
|
err: errors.Error("can't set"),
|
||||||
|
out: "",
|
||||||
|
code: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
fsys: succFsys,
|
||||||
|
wantErrMsg: `can't set`,
|
||||||
|
}}
|
||||||
|
|
||||||
|
for _, tc := range testCases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
substShell(t, tc.shell.RunCmd)
|
||||||
|
substRootDirFS(t, tc.fsys)
|
||||||
|
|
||||||
|
err := IfaceSetStaticIP("en0")
|
||||||
|
testutil.AssertErrorMsg(t, tc.wantErrMsg, err)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
|
@ -22,7 +22,7 @@ func ifaceHasStaticIP(ifaceName string) (ok bool, err error) {
|
||||||
|
|
||||||
walker := aghos.FileWalker(interfaceName(ifaceName).rcConfStaticConfig)
|
walker := aghos.FileWalker(interfaceName(ifaceName).rcConfStaticConfig)
|
||||||
|
|
||||||
return walker.Walk(aghos.RootDirFS(), rcConfFilename)
|
return walker.Walk(rootDirFS, rcConfFilename)
|
||||||
}
|
}
|
||||||
|
|
||||||
// rcConfStaticConfig checks if the interface is configured by /etc/rc.conf to
|
// rcConfStaticConfig checks if the interface is configured by /etc/rc.conf to
|
||||||
|
|
|
@ -4,56 +4,74 @@
|
||||||
package aghnet
|
package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"strings"
|
"io/fs"
|
||||||
"testing"
|
"testing"
|
||||||
|
"testing/fstest"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestRcConfStaticConfig(t *testing.T) {
|
func TestRcConfStaticConfig(t *testing.T) {
|
||||||
const iface interfaceName = `em0`
|
const (
|
||||||
const nl = "\n"
|
ifaceName = `em0`
|
||||||
|
rcConf = "etc/rc.conf"
|
||||||
|
)
|
||||||
|
|
||||||
testCases := []struct {
|
testCases := []struct {
|
||||||
name string
|
name string
|
||||||
rcconfData string
|
rootFsys fs.FS
|
||||||
wantCont bool
|
wantHas assert.BoolAssertionFunc
|
||||||
}{{
|
}{{
|
||||||
name: "simple",
|
name: "simple",
|
||||||
rcconfData: `ifconfig_em0="inet 127.0.0.253 netmask 0xffffffff"` + nl,
|
rootFsys: fstest.MapFS{rcConf: &fstest.MapFile{
|
||||||
wantCont: false,
|
Data: []byte(`ifconfig_` + ifaceName + `="inet 127.0.0.253 netmask 0xffffffff"` + nl),
|
||||||
|
}},
|
||||||
|
wantHas: assert.True,
|
||||||
}, {
|
}, {
|
||||||
name: "case_insensitiveness",
|
name: "case_insensitiveness",
|
||||||
rcconfData: `ifconfig_em0="InEt 127.0.0.253 NeTmAsK 0xffffffff"` + nl,
|
rootFsys: fstest.MapFS{rcConf: &fstest.MapFile{
|
||||||
wantCont: false,
|
Data: []byte(`ifconfig_` + ifaceName + `="InEt 127.0.0.253 NeTmAsK 0xffffffff"` + nl),
|
||||||
|
}},
|
||||||
|
wantHas: assert.True,
|
||||||
}, {
|
}, {
|
||||||
name: "comments_and_trash",
|
name: "comments_and_trash",
|
||||||
rcconfData: `# comment 1` + nl +
|
rootFsys: fstest.MapFS{rcConf: &fstest.MapFile{
|
||||||
|
Data: []byte(`# comment 1` + nl +
|
||||||
`` + nl +
|
`` + nl +
|
||||||
`# comment 2` + nl +
|
`# comment 2` + nl +
|
||||||
`ifconfig_em0="inet 127.0.0.253 netmask 0xffffffff"` + nl,
|
`ifconfig_` + ifaceName + `="inet 127.0.0.253 netmask 0xffffffff"` + nl,
|
||||||
wantCont: false,
|
),
|
||||||
|
}},
|
||||||
|
wantHas: assert.True,
|
||||||
}, {
|
}, {
|
||||||
name: "aliases",
|
name: "aliases",
|
||||||
rcconfData: `ifconfig_em0_alias="inet 127.0.0.1/24"` + nl +
|
rootFsys: fstest.MapFS{rcConf: &fstest.MapFile{
|
||||||
`ifconfig_em0="inet 127.0.0.253 netmask 0xffffffff"` + nl,
|
Data: []byte(`ifconfig_` + ifaceName + `_alias="inet 127.0.0.1/24"` + nl +
|
||||||
wantCont: false,
|
`ifconfig_` + ifaceName + `="inet 127.0.0.253 netmask 0xffffffff"` + nl,
|
||||||
|
),
|
||||||
|
}},
|
||||||
|
wantHas: assert.True,
|
||||||
}, {
|
}, {
|
||||||
name: "incorrect_config",
|
name: "incorrect_config",
|
||||||
rcconfData: `ifconfig_em0="inet6 127.0.0.253 netmask 0xffffffff"` + nl +
|
rootFsys: fstest.MapFS{rcConf: &fstest.MapFile{
|
||||||
`ifconfig_em0="inet 256.256.256.256 netmask 0xffffffff"` + nl +
|
Data: []byte(
|
||||||
`ifconfig_em0=""` + nl,
|
`ifconfig_` + ifaceName + `="inet6 127.0.0.253 netmask 0xffffffff"` + nl +
|
||||||
wantCont: true,
|
`ifconfig_` + ifaceName + `="inet 256.256.256.256 netmask 0xffffffff"` + nl +
|
||||||
|
`ifconfig_` + ifaceName + `=""` + nl,
|
||||||
|
),
|
||||||
|
}},
|
||||||
|
wantHas: assert.False,
|
||||||
}}
|
}}
|
||||||
|
|
||||||
for _, tc := range testCases {
|
for _, tc := range testCases {
|
||||||
r := strings.NewReader(tc.rcconfData)
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
_, cont, err := iface.rcConfStaticConfig(r)
|
substRootDirFS(t, tc.rootFsys)
|
||||||
|
|
||||||
|
has, err := IfaceHasStaticIP(ifaceName)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
assert.Equal(t, tc.wantCont, cont)
|
tc.wantHas(t, has)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -21,8 +21,7 @@ import (
|
||||||
// have a static IP.
|
// have a static IP.
|
||||||
func (n interfaceName) dhcpcdStaticConfig(r io.Reader) (subsources []string, cont bool, err error) {
|
func (n interfaceName) dhcpcdStaticConfig(r io.Reader) (subsources []string, cont bool, err error) {
|
||||||
s := bufio.NewScanner(r)
|
s := bufio.NewScanner(r)
|
||||||
ifaceFound := findIfaceLine(s, string(n))
|
if !findIfaceLine(s, string(n)) {
|
||||||
if !ifaceFound {
|
|
||||||
return nil, true, s.Err()
|
return nil, true, s.Err()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -61,9 +60,9 @@ func (n interfaceName) ifacesStaticConfig(r io.Reader) (sub []string, cont bool,
|
||||||
fields := strings.Fields(line)
|
fields := strings.Fields(line)
|
||||||
fieldsNum := len(fields)
|
fieldsNum := len(fields)
|
||||||
|
|
||||||
// Man page interfaces(5) declares that interface definition
|
// Man page interfaces(5) declares that interface definition should
|
||||||
// should consist of the key word "iface" followed by interface
|
// consist of the key word "iface" followed by interface name, and
|
||||||
// name, and method at fourth field.
|
// method at fourth field.
|
||||||
if fieldsNum >= 4 &&
|
if fieldsNum >= 4 &&
|
||||||
fields[0] == "iface" && fields[1] == string(n) && fields[3] == "static" {
|
fields[0] == "iface" && fields[1] == string(n) && fields[3] == "static" {
|
||||||
return nil, false, nil
|
return nil, false, nil
|
||||||
|
@ -78,10 +77,10 @@ func (n interfaceName) ifacesStaticConfig(r io.Reader) (sub []string, cont bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
func ifaceHasStaticIP(ifaceName string) (has bool, err error) {
|
func ifaceHasStaticIP(ifaceName string) (has bool, err error) {
|
||||||
// TODO(a.garipov): Currently, this function returns the first
|
// TODO(a.garipov): Currently, this function returns the first definitive
|
||||||
// definitive result. So if /etc/dhcpcd.conf has a static IP while
|
// result. So if /etc/dhcpcd.conf has and /etc/network/interfaces has no
|
||||||
// /etc/network/interfaces doesn't, it will return true. Perhaps this
|
// static IP configuration, it will return true. Perhaps this is not the
|
||||||
// is not the most desirable behavior.
|
// most desirable behavior.
|
||||||
|
|
||||||
iface := interfaceName(ifaceName)
|
iface := interfaceName(ifaceName)
|
||||||
|
|
||||||
|
@ -95,12 +94,10 @@ func ifaceHasStaticIP(ifaceName string) (has bool, err error) {
|
||||||
FileWalker: iface.ifacesStaticConfig,
|
FileWalker: iface.ifacesStaticConfig,
|
||||||
filename: "etc/network/interfaces",
|
filename: "etc/network/interfaces",
|
||||||
}} {
|
}} {
|
||||||
has, err = pair.Walk(aghos.RootDirFS(), pair.filename)
|
has, err = pair.Walk(rootDirFS, pair.filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
} else if has {
|
||||||
|
|
||||||
if has {
|
|
||||||
return true, nil
|
return true, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -141,13 +138,15 @@ func ifaceSetStaticIP(ifaceName string) (err error) {
|
||||||
gatewayIP := GatewayIP(ifaceName)
|
gatewayIP := GatewayIP(ifaceName)
|
||||||
add := dhcpcdConfIface(ifaceName, ipNet, gatewayIP, ipNet.IP)
|
add := dhcpcdConfIface(ifaceName, ipNet, gatewayIP, ipNet.IP)
|
||||||
|
|
||||||
body, err := os.ReadFile("/etc/dhcpcd.conf")
|
const filename = "/etc/dhcpcd.conf"
|
||||||
|
|
||||||
|
body, err := os.ReadFile(filename)
|
||||||
if err != nil && !errors.Is(err, os.ErrNotExist) {
|
if err != nil && !errors.Is(err, os.ErrNotExist) {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
body = append(body, []byte(add)...)
|
body = append(body, []byte(add)...)
|
||||||
err = maybe.WriteFile("/etc/dhcpcd.conf", body, 0o644)
|
err = maybe.WriteFile(filename, body, 0o644)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("writing conf: %w", err)
|
return fmt.Errorf("writing conf: %w", err)
|
||||||
}
|
}
|
||||||
|
|
|
@ -4,122 +4,130 @@
|
||||||
package aghnet
|
package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"io/fs"
|
||||||
"net"
|
"net"
|
||||||
"testing"
|
"testing"
|
||||||
|
"testing/fstest"
|
||||||
|
|
||||||
|
"github.com/AdguardTeam/golibs/testutil"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestDHCPCDStaticConfig(t *testing.T) {
|
func TestHasStaticIP(t *testing.T) {
|
||||||
const iface interfaceName = `wlan0`
|
const ifaceName = "wlan0"
|
||||||
|
|
||||||
|
const (
|
||||||
|
dhcpcd = "etc/dhcpcd.conf"
|
||||||
|
netifaces = "etc/network/interfaces"
|
||||||
|
)
|
||||||
|
|
||||||
testCases := []struct {
|
testCases := []struct {
|
||||||
|
rootFsys fs.FS
|
||||||
name string
|
name string
|
||||||
data []byte
|
wantHas assert.BoolAssertionFunc
|
||||||
wantCont bool
|
wantErrMsg string
|
||||||
}{{
|
}{{
|
||||||
name: "has_not",
|
rootFsys: fstest.MapFS{
|
||||||
data: []byte(`#comment` + nl +
|
dhcpcd: &fstest.MapFile{
|
||||||
|
Data: []byte(`#comment` + nl +
|
||||||
`# comment` + nl +
|
`# comment` + nl +
|
||||||
`interface eth0` + nl +
|
`interface eth0` + nl +
|
||||||
`static ip_address=192.168.0.1/24` + nl +
|
`static ip_address=192.168.0.1/24` + nl +
|
||||||
`# interface ` + iface + nl +
|
`# interface ` + ifaceName + nl +
|
||||||
`static ip_address=192.168.1.1/24` + nl +
|
`static ip_address=192.168.1.1/24` + nl +
|
||||||
`# comment` + nl,
|
`# comment` + nl,
|
||||||
),
|
),
|
||||||
wantCont: true,
|
},
|
||||||
|
},
|
||||||
|
name: "dhcpcd_has_not",
|
||||||
|
wantHas: assert.False,
|
||||||
|
wantErrMsg: `no information about static ip`,
|
||||||
}, {
|
}, {
|
||||||
name: "has",
|
rootFsys: fstest.MapFS{
|
||||||
data: []byte(`#comment` + nl +
|
dhcpcd: &fstest.MapFile{
|
||||||
|
Data: []byte(`#comment` + nl +
|
||||||
`# comment` + nl +
|
`# comment` + nl +
|
||||||
`interface eth0` + nl +
|
`interface ` + ifaceName + nl +
|
||||||
`static ip_address=192.168.0.1/24` + nl +
|
`static ip_address=192.168.0.1/24` + nl +
|
||||||
`# interface ` + iface + nl +
|
`# interface ` + ifaceName + nl +
|
||||||
`static ip_address=192.168.1.1/24` + nl +
|
`static ip_address=192.168.1.1/24` + nl +
|
||||||
`# comment` + nl +
|
`# comment` + nl,
|
||||||
`interface ` + iface + nl +
|
|
||||||
`# comment` + nl +
|
|
||||||
`static ip_address=192.168.2.1/24` + nl,
|
|
||||||
),
|
),
|
||||||
wantCont: false,
|
},
|
||||||
}}
|
},
|
||||||
|
name: "dhcpcd_has",
|
||||||
for _, tc := range testCases {
|
wantHas: assert.True,
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
wantErrMsg: ``,
|
||||||
r := bytes.NewReader(tc.data)
|
}, {
|
||||||
_, cont, err := iface.dhcpcdStaticConfig(r)
|
rootFsys: fstest.MapFS{
|
||||||
require.NoError(t, err)
|
netifaces: &fstest.MapFile{
|
||||||
|
Data: []byte(`allow-hotplug ` + ifaceName + nl +
|
||||||
assert.Equal(t, tc.wantCont, cont)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestIfacesStaticConfig(t *testing.T) {
|
|
||||||
const iface interfaceName = `enp0s3`
|
|
||||||
|
|
||||||
testCases := []struct {
|
|
||||||
name string
|
|
||||||
data []byte
|
|
||||||
wantCont bool
|
|
||||||
wantPatterns []string
|
|
||||||
}{{
|
|
||||||
name: "has_not",
|
|
||||||
data: []byte(`allow-hotplug ` + iface + nl +
|
|
||||||
`#iface enp0s3 inet static` + nl +
|
`#iface enp0s3 inet static` + nl +
|
||||||
`# address 192.168.0.200` + nl +
|
`# address 192.168.0.200` + nl +
|
||||||
`# netmask 255.255.255.0` + nl +
|
`# netmask 255.255.255.0` + nl +
|
||||||
`# gateway 192.168.0.1` + nl +
|
`# gateway 192.168.0.1` + nl +
|
||||||
`iface ` + iface + ` inet dhcp` + nl,
|
`iface ` + ifaceName + ` inet dhcp` + nl,
|
||||||
),
|
),
|
||||||
wantCont: true,
|
},
|
||||||
wantPatterns: []string{},
|
},
|
||||||
|
name: "netifaces_has_not",
|
||||||
|
wantHas: assert.False,
|
||||||
|
wantErrMsg: `no information about static ip`,
|
||||||
}, {
|
}, {
|
||||||
name: "has",
|
rootFsys: fstest.MapFS{
|
||||||
data: []byte(`allow-hotplug ` + iface + nl +
|
netifaces: &fstest.MapFile{
|
||||||
`iface ` + iface + ` inet static` + nl +
|
Data: []byte(`allow-hotplug ` + ifaceName + nl +
|
||||||
|
`iface ` + ifaceName + ` inet static` + nl +
|
||||||
` address 192.168.0.200` + nl +
|
` address 192.168.0.200` + nl +
|
||||||
` netmask 255.255.255.0` + nl +
|
` netmask 255.255.255.0` + nl +
|
||||||
` gateway 192.168.0.1` + nl +
|
` gateway 192.168.0.1` + nl +
|
||||||
`#iface ` + iface + ` inet dhcp` + nl,
|
`#iface ` + ifaceName + ` inet dhcp` + nl,
|
||||||
),
|
),
|
||||||
wantCont: false,
|
},
|
||||||
wantPatterns: []string{},
|
},
|
||||||
|
name: "netifaces_has",
|
||||||
|
wantHas: assert.True,
|
||||||
|
wantErrMsg: ``,
|
||||||
}, {
|
}, {
|
||||||
name: "return_patterns",
|
rootFsys: fstest.MapFS{
|
||||||
data: []byte(`source hello` + nl +
|
netifaces: &fstest.MapFile{
|
||||||
`source world` + nl +
|
Data: []byte(`source hello` + nl +
|
||||||
`#iface ` + iface + ` inet static` + nl,
|
`#iface ` + ifaceName + ` inet static` + nl,
|
||||||
),
|
),
|
||||||
wantCont: true,
|
},
|
||||||
wantPatterns: []string{"hello", "world"},
|
"hello": &fstest.MapFile{
|
||||||
|
Data: []byte(`iface ` + ifaceName + ` inet static` + nl),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
name: "netifaces_another_file",
|
||||||
|
wantHas: assert.True,
|
||||||
|
wantErrMsg: ``,
|
||||||
}, {
|
}, {
|
||||||
// This one tests if the first found valid interface prevents
|
rootFsys: fstest.MapFS{
|
||||||
// checking files under the `source` directive.
|
netifaces: &fstest.MapFile{
|
||||||
name: "ignore_patterns",
|
Data: []byte(`source hello` + nl +
|
||||||
data: []byte(`source hello` + nl +
|
`iface ` + ifaceName + ` inet static` + nl,
|
||||||
`source world` + nl +
|
|
||||||
`iface ` + iface + ` inet static` + nl,
|
|
||||||
),
|
),
|
||||||
wantCont: false,
|
},
|
||||||
wantPatterns: []string{},
|
},
|
||||||
|
name: "netifaces_ignore_another",
|
||||||
|
wantHas: assert.True,
|
||||||
|
wantErrMsg: ``,
|
||||||
}}
|
}}
|
||||||
|
|
||||||
for _, tc := range testCases {
|
for _, tc := range testCases {
|
||||||
r := bytes.NewReader(tc.data)
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
patterns, has, err := iface.ifacesStaticConfig(r)
|
substRootDirFS(t, tc.rootFsys)
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
assert.Equal(t, tc.wantCont, has)
|
has, err := IfaceHasStaticIP(ifaceName)
|
||||||
assert.ElementsMatch(t, tc.wantPatterns, patterns)
|
testutil.AssertErrorMsg(t, tc.wantErrMsg, err)
|
||||||
|
|
||||||
|
tc.wantHas(t, has)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSetStaticIPdhcpcdConf(t *testing.T) {
|
func TestSetStaticIP_dhcpcdConfIface(t *testing.T) {
|
||||||
testCases := []struct {
|
testCases := []struct {
|
||||||
name string
|
name string
|
||||||
dhcpcdConf string
|
dhcpcdConf string
|
||||||
|
|
|
@ -20,7 +20,7 @@ func canBindPrivilegedPorts() (can bool, err error) {
|
||||||
func ifaceHasStaticIP(ifaceName string) (ok bool, err error) {
|
func ifaceHasStaticIP(ifaceName string) (ok bool, err error) {
|
||||||
filename := fmt.Sprintf("etc/hostname.%s", ifaceName)
|
filename := fmt.Sprintf("etc/hostname.%s", ifaceName)
|
||||||
|
|
||||||
return aghos.FileWalker(hostnameIfStaticConfig).Walk(aghos.RootDirFS(), filename)
|
return aghos.FileWalker(hostnameIfStaticConfig).Walk(rootDirFS, filename)
|
||||||
}
|
}
|
||||||
|
|
||||||
// hostnameIfStaticConfig checks if the interface is configured by
|
// hostnameIfStaticConfig checks if the interface is configured by
|
||||||
|
|
|
@ -4,49 +4,69 @@
|
||||||
package aghnet
|
package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"strings"
|
"fmt"
|
||||||
|
"io/fs"
|
||||||
"testing"
|
"testing"
|
||||||
|
"testing/fstest"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestHostnameIfStaticConfig(t *testing.T) {
|
func TestIfaceHasStaticIP(t *testing.T) {
|
||||||
const nl = "\n"
|
const ifaceName = "em0"
|
||||||
|
|
||||||
|
confFile := fmt.Sprintf("etc/hostname.%s", ifaceName)
|
||||||
|
|
||||||
testCases := []struct {
|
testCases := []struct {
|
||||||
name string
|
name string
|
||||||
rcconfData string
|
rootFsys fs.FS
|
||||||
wantHas bool
|
wantHas assert.BoolAssertionFunc
|
||||||
}{{
|
}{{
|
||||||
name: "simple",
|
name: "simple",
|
||||||
rcconfData: `inet 127.0.0.253` + nl,
|
rootFsys: fstest.MapFS{
|
||||||
wantHas: true,
|
confFile: &fstest.MapFile{
|
||||||
|
Data: []byte(`inet 127.0.0.253` + nl),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantHas: assert.True,
|
||||||
}, {
|
}, {
|
||||||
name: "case_sensitiveness",
|
name: "case_sensitiveness",
|
||||||
rcconfData: `InEt 127.0.0.253` + nl,
|
rootFsys: fstest.MapFS{
|
||||||
wantHas: false,
|
confFile: &fstest.MapFile{
|
||||||
|
Data: []byte(`InEt 127.0.0.253` + nl),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantHas: assert.False,
|
||||||
}, {
|
}, {
|
||||||
name: "comments_and_trash",
|
name: "comments_and_trash",
|
||||||
rcconfData: `# comment 1` + nl +
|
rootFsys: fstest.MapFS{
|
||||||
`` + nl +
|
confFile: &fstest.MapFile{
|
||||||
|
Data: []byte(`# comment 1` + nl + nl +
|
||||||
`# inet 127.0.0.253` + nl +
|
`# inet 127.0.0.253` + nl +
|
||||||
`inet` + nl,
|
`inet` + nl,
|
||||||
wantHas: false,
|
),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantHas: assert.False,
|
||||||
}, {
|
}, {
|
||||||
name: "incorrect_config",
|
name: "incorrect_config",
|
||||||
rcconfData: `inet6 127.0.0.253` + nl +
|
rootFsys: fstest.MapFS{
|
||||||
`inet 256.256.256.256` + nl,
|
confFile: &fstest.MapFile{
|
||||||
wantHas: false,
|
Data: []byte(`inet6 127.0.0.253` + nl + `inet 256.256.256.256` + nl),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantHas: assert.False,
|
||||||
}}
|
}}
|
||||||
|
|
||||||
for _, tc := range testCases {
|
for _, tc := range testCases {
|
||||||
r := strings.NewReader(tc.rcconfData)
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
_, has, err := hostnameIfStaticConfig(r)
|
substRootDirFS(t, tc.rootFsys)
|
||||||
|
|
||||||
|
has, err := IfaceHasStaticIP(ifaceName)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
assert.Equal(t, tc.wantHas, has)
|
tc.wantHas(t, has)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,9 +1,11 @@
|
||||||
package aghnet
|
package aghnet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"net"
|
"net"
|
||||||
"os"
|
"os"
|
||||||
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/AdguardTeam/AdGuardHome/internal/aghtest"
|
"github.com/AdguardTeam/AdGuardHome/internal/aghtest"
|
||||||
|
@ -14,11 +16,102 @@ import (
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func TestMain(m *testing.M) {
|
||||||
|
aghtest.DiscardLogOutput(m)
|
||||||
|
}
|
||||||
|
|
||||||
// testdata is the filesystem containing data for testing the package.
|
// testdata is the filesystem containing data for testing the package.
|
||||||
var testdata fs.FS = os.DirFS("./testdata")
|
var testdata fs.FS = os.DirFS("./testdata")
|
||||||
|
|
||||||
func TestMain(m *testing.M) {
|
// substRootDirFS replaces the aghos.RootDirFS function used throughout the
|
||||||
aghtest.DiscardLogOutput(m)
|
// package with fsys for tests ran under t.
|
||||||
|
func substRootDirFS(t testing.TB, fsys fs.FS) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
prev := rootDirFS
|
||||||
|
t.Cleanup(func() { rootDirFS = prev })
|
||||||
|
rootDirFS = fsys
|
||||||
|
}
|
||||||
|
|
||||||
|
// RunCmdFunc is the signature of aghos.RunCommand function.
|
||||||
|
type RunCmdFunc func(cmd string, args ...string) (code int, out []byte, err error)
|
||||||
|
|
||||||
|
// substShell replaces the the aghos.RunCommand function used throughout the
|
||||||
|
// package with rc for tests ran under t.
|
||||||
|
func substShell(t testing.TB, rc RunCmdFunc) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
prev := aghosRunCommand
|
||||||
|
t.Cleanup(func() { aghosRunCommand = prev })
|
||||||
|
aghosRunCommand = rc
|
||||||
|
}
|
||||||
|
|
||||||
|
// mapShell is a substitution of aghos.RunCommand that maps the command to it's
|
||||||
|
// execution result. It's only needed to simplify testing.
|
||||||
|
//
|
||||||
|
// TODO(e.burkov): Perhaps put all the shell interactions behind an interface.
|
||||||
|
type mapShell map[string]struct {
|
||||||
|
err error
|
||||||
|
out string
|
||||||
|
code int
|
||||||
|
}
|
||||||
|
|
||||||
|
// theOnlyCmd returns s that only handles a single command and arguments
|
||||||
|
// combination from cmd.
|
||||||
|
func theOnlyCmd(cmd string, code int, out string, err error) (s mapShell) {
|
||||||
|
return mapShell{cmd: {code: code, out: out, err: err}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// RunCmd is a RunCmdFunc handled by s.
|
||||||
|
func (s mapShell) RunCmd(cmd string, args ...string) (code int, out []byte, err error) {
|
||||||
|
key := strings.Join(append([]string{cmd}, args...), " ")
|
||||||
|
ret, ok := s[key]
|
||||||
|
if !ok {
|
||||||
|
return 0, nil, fmt.Errorf("unexpected shell command %q", key)
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret.code, []byte(ret.out), ret.err
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGatewayIP(t *testing.T) {
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
shell mapShell
|
||||||
|
want net.IP
|
||||||
|
}{{
|
||||||
|
name: "success_v4",
|
||||||
|
shell: theOnlyCmd("ip route show dev ifaceName", 0, `default via 1.2.3.4 onlink`, nil),
|
||||||
|
want: net.IP{1, 2, 3, 4}.To16(),
|
||||||
|
}, {
|
||||||
|
name: "success_v6",
|
||||||
|
shell: theOnlyCmd("ip route show dev ifaceName", 0, `default via ::ffff onlink`, nil),
|
||||||
|
want: net.IP{
|
||||||
|
0x0, 0x0, 0x0, 0x0,
|
||||||
|
0x0, 0x0, 0x0, 0x0,
|
||||||
|
0x0, 0x0, 0x0, 0x0,
|
||||||
|
0x0, 0x0, 0xFF, 0xFF,
|
||||||
|
},
|
||||||
|
}, {
|
||||||
|
name: "bad_output",
|
||||||
|
shell: theOnlyCmd("ip route show dev ifaceName", 0, `non-default via 1.2.3.4 onlink`, nil),
|
||||||
|
want: nil,
|
||||||
|
}, {
|
||||||
|
name: "err_runcmd",
|
||||||
|
shell: theOnlyCmd("ip route show dev ifaceName", 0, "", errors.Error("can't run command")),
|
||||||
|
want: nil,
|
||||||
|
}, {
|
||||||
|
name: "bad_code",
|
||||||
|
shell: theOnlyCmd("ip route show dev ifaceName", 1, "", nil),
|
||||||
|
want: nil,
|
||||||
|
}}
|
||||||
|
|
||||||
|
for _, tc := range testCases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
substShell(t, tc.shell.RunCmd)
|
||||||
|
|
||||||
|
assert.Equal(t, tc.want, GatewayIP("ifaceName"))
|
||||||
|
})
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestGetInterfaceByIP(t *testing.T) {
|
func TestGetInterfaceByIP(t *testing.T) {
|
||||||
|
@ -133,6 +226,8 @@ func TestCheckPort(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCollectAllIfacesAddrs(t *testing.T) {
|
func TestCollectAllIfacesAddrs(t *testing.T) {
|
||||||
|
t.Skip("TODO(e.burkov): Substitute the net.Interfaces.")
|
||||||
|
|
||||||
addrs, err := CollectAllIfacesAddrs()
|
addrs, err := CollectAllIfacesAddrs()
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
|
|
@ -57,20 +57,22 @@ func HaveAdminRights() (bool, error) {
|
||||||
const MaxCmdOutputSize = 64 * 1024
|
const MaxCmdOutputSize = 64 * 1024
|
||||||
|
|
||||||
// RunCommand runs shell command.
|
// RunCommand runs shell command.
|
||||||
func RunCommand(command string, arguments ...string) (code int, output string, err error) {
|
func RunCommand(command string, arguments ...string) (code int, output []byte, err error) {
|
||||||
cmd := exec.Command(command, arguments...)
|
cmd := exec.Command(command, arguments...)
|
||||||
out, err := cmd.Output()
|
out, err := cmd.Output()
|
||||||
if len(out) > MaxCmdOutputSize {
|
if len(out) > MaxCmdOutputSize {
|
||||||
out = out[:MaxCmdOutputSize]
|
out = out[:MaxCmdOutputSize]
|
||||||
}
|
}
|
||||||
|
|
||||||
if errors.As(err, new(*exec.ExitError)) {
|
if err != nil {
|
||||||
return cmd.ProcessState.ExitCode(), string(out), nil
|
if eerr := new(exec.ExitError); errors.As(err, &eerr) {
|
||||||
} else if err != nil {
|
return eerr.ExitCode(), eerr.Stderr, nil
|
||||||
return 1, "", fmt.Errorf("command %q failed: %w: %s", command, err, out)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return cmd.ProcessState.ExitCode(), string(out), nil
|
return 1, nil, fmt.Errorf("command %q failed: %w: %s", command, err, out)
|
||||||
|
}
|
||||||
|
|
||||||
|
return cmd.ProcessState.ExitCode(), out, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// PIDByCommand searches for process named command and returns its PID ignoring
|
// PIDByCommand searches for process named command and returns its PID ignoring
|
||||||
|
|
|
@ -314,12 +314,13 @@ func (s *openbsdRunComService) runCom(cmd string) (out string, err error) {
|
||||||
// TODO(e.burkov): It's possible that os.ErrNotExist is caused by
|
// TODO(e.burkov): It's possible that os.ErrNotExist is caused by
|
||||||
// something different than the service script's non-existence. Keep it
|
// something different than the service script's non-existence. Keep it
|
||||||
// in mind, when replace the aghos.RunCommand.
|
// in mind, when replace the aghos.RunCommand.
|
||||||
_, out, err = aghos.RunCommand(scriptPath, cmd)
|
var outData []byte
|
||||||
|
_, outData, err = aghos.RunCommand(scriptPath, cmd)
|
||||||
if errors.Is(err, os.ErrNotExist) {
|
if errors.Is(err, os.ErrNotExist) {
|
||||||
return "", service.ErrNotInstalled
|
return "", service.ErrNotInstalled
|
||||||
}
|
}
|
||||||
|
|
||||||
return out, err
|
return string(outData), err
|
||||||
}
|
}
|
||||||
|
|
||||||
// Status implements service.Service interface for *openbsdRunComService.
|
// Status implements service.Service interface for *openbsdRunComService.
|
||||||
|
|
Loading…
Reference in a new issue