mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-11-25 14:35:48 +03:00
e8c1f5c8d3
Merge in DNS/adguard-home from 2508-ip-conversion to master Updates #2508. Squashed commit of the following: commit 3f64709fbc73ef74c11b910997be1e9bc337193c Merge: 5ac7faaaa0d67aa251Author: Eugene Burkov <e.burkov@adguard.com> Date: Wed Jan 13 16:21:34 2021 +0300 Merge branch 'master' into 2508-ip-conversion commit 5ac7faaaa9dda570fdb872acad5d13d078f46b64 Author: Eugene Burkov <e.burkov@adguard.com> Date: Wed Jan 13 12:00:11 2021 +0300 all: replace conditions with appropriate functions in tests commit 9e3fa9a115ed23024c57dd5192d5173477ddbf71 Merge: db992a42abba74859eAuthor: Eugene Burkov <e.burkov@adguard.com> Date: Wed Jan 13 10:47:10 2021 +0300 Merge branch 'master' into 2508-ip-conversion commit db992a42a2c6f315421e78a6a0492e2bfb3ce89d Author: Eugene Burkov <e.burkov@adguard.com> Date: Tue Jan 12 18:55:53 2021 +0300 sysutil: fix linux tests commit f629b15d62349323ce2da05e68dc9cc0b5f6e194 Author: Eugene Burkov <e.burkov@adguard.com> Date: Tue Jan 12 18:41:20 2021 +0300 all: improve code quality commit 3bf03a75524040738562298bd1de6db536af130f Author: Eugene Burkov <e.burkov@adguard.com> Date: Tue Jan 12 17:33:26 2021 +0300 sysutil: fix linux net.IP conversion commit 5d5b6994916923636e635588631b63b7e7b74e5f Author: Eugene Burkov <e.burkov@adguard.com> Date: Tue Jan 12 14:57:26 2021 +0300 dnsforward: remove redundant net.IP <-> string conversion commit 0b955d99b7fad40942f21d1dd8734adb99126195 Author: Eugene Burkov <e.burkov@adguard.com> Date: Mon Jan 11 18:04:25 2021 +0300 dhcpd: remove net.IP <-> string conversion
130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
// +build aix darwin dragonfly freebsd linux netbsd openbsd solaris
|
|
|
|
package dhcpd
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
testutil.DiscardLogOutput(m)
|
|
}
|
|
|
|
func testNotify(flags uint32) {
|
|
}
|
|
|
|
// Leases database store/load
|
|
func TestDB(t *testing.T) {
|
|
var err error
|
|
s := Server{}
|
|
s.conf.DBFilePath = dbFilename
|
|
|
|
conf := V4ServerConf{
|
|
Enabled: true,
|
|
RangeStart: net.IP{192, 168, 10, 100},
|
|
RangeEnd: net.IP{192, 168, 10, 200},
|
|
GatewayIP: net.IP{192, 168, 10, 1},
|
|
SubnetMask: net.IP{255, 255, 255, 0},
|
|
notify: testNotify,
|
|
}
|
|
s.srv4, err = v4Create(conf)
|
|
assert.Nil(t, err)
|
|
|
|
s.srv6, err = v6Create(V6ServerConf{})
|
|
assert.Nil(t, err)
|
|
|
|
l := Lease{}
|
|
l.IP = net.IP{192, 168, 10, 100}
|
|
l.HWAddr, _ = net.ParseMAC("aa:aa:aa:aa:aa:aa")
|
|
exp1 := time.Now().Add(time.Hour)
|
|
l.Expiry = exp1
|
|
s.srv4.(*v4Server).addLease(&l)
|
|
|
|
l2 := Lease{}
|
|
l2.IP = net.IP{192, 168, 10, 101}
|
|
l2.HWAddr, _ = net.ParseMAC("aa:aa:aa:aa:aa:bb")
|
|
s.srv4.AddStaticLease(l2)
|
|
|
|
_ = os.Remove("leases.db")
|
|
s.dbStore()
|
|
s.srv4.ResetLeases(nil)
|
|
|
|
s.dbLoad()
|
|
|
|
ll := s.srv4.GetLeases(LeasesAll)
|
|
|
|
assert.Equal(t, "aa:aa:aa:aa:aa:bb", ll[0].HWAddr.String())
|
|
assert.Equal(t, "192.168.10.101", ll[0].IP.String())
|
|
assert.EqualValues(t, leaseExpireStatic, ll[0].Expiry.Unix())
|
|
|
|
assert.Equal(t, "aa:aa:aa:aa:aa:aa", ll[1].HWAddr.String())
|
|
assert.Equal(t, "192.168.10.100", ll[1].IP.String())
|
|
assert.Equal(t, exp1.Unix(), ll[1].Expiry.Unix())
|
|
|
|
_ = os.Remove("leases.db")
|
|
}
|
|
|
|
func TestIsValidSubnetMask(t *testing.T) {
|
|
assert.True(t, isValidSubnetMask([]byte{255, 255, 255, 0}))
|
|
assert.True(t, isValidSubnetMask([]byte{255, 255, 254, 0}))
|
|
assert.True(t, isValidSubnetMask([]byte{255, 255, 252, 0}))
|
|
assert.False(t, isValidSubnetMask([]byte{255, 255, 253, 0}))
|
|
assert.False(t, isValidSubnetMask([]byte{255, 255, 255, 1}))
|
|
}
|
|
|
|
func TestNormalizeLeases(t *testing.T) {
|
|
dynLeases := []*Lease{}
|
|
staticLeases := []*Lease{}
|
|
|
|
lease := &Lease{}
|
|
lease.HWAddr = []byte{1, 2, 3, 4}
|
|
dynLeases = append(dynLeases, lease)
|
|
lease = new(Lease)
|
|
lease.HWAddr = []byte{1, 2, 3, 5}
|
|
dynLeases = append(dynLeases, lease)
|
|
|
|
lease = new(Lease)
|
|
lease.HWAddr = []byte{1, 2, 3, 4}
|
|
lease.IP = []byte{0, 2, 3, 4}
|
|
staticLeases = append(staticLeases, lease)
|
|
lease = new(Lease)
|
|
lease.HWAddr = []byte{2, 2, 3, 4}
|
|
staticLeases = append(staticLeases, lease)
|
|
|
|
leases := normalizeLeases(staticLeases, dynLeases)
|
|
|
|
assert.Len(t, leases, 3)
|
|
assert.True(t, bytes.Equal(leases[0].HWAddr, []byte{1, 2, 3, 4}))
|
|
assert.True(t, bytes.Equal(leases[0].IP, []byte{0, 2, 3, 4}))
|
|
assert.True(t, bytes.Equal(leases[1].HWAddr, []byte{2, 2, 3, 4}))
|
|
assert.True(t, bytes.Equal(leases[2].HWAddr, []byte{1, 2, 3, 5}))
|
|
}
|
|
|
|
func TestOptions(t *testing.T) {
|
|
code, val := parseOptionString(" 12 hex abcdef ")
|
|
assert.EqualValues(t, 12, code)
|
|
assert.True(t, bytes.Equal([]byte{0xab, 0xcd, 0xef}, val))
|
|
|
|
code, _ = parseOptionString(" 12 hex abcdef1 ")
|
|
assert.EqualValues(t, 0, code)
|
|
|
|
code, val = parseOptionString("123 ip 1.2.3.4")
|
|
assert.EqualValues(t, 123, code)
|
|
assert.Equal(t, "1.2.3.4", net.IP(string(val)).String())
|
|
|
|
code, _ = parseOptionString("256 ip 1.1.1.1")
|
|
assert.EqualValues(t, 0, code)
|
|
code, _ = parseOptionString("-1 ip 1.1.1.1")
|
|
assert.EqualValues(t, 0, code)
|
|
code, _ = parseOptionString("12 ip 1.1.1.1x")
|
|
assert.EqualValues(t, 0, code)
|
|
code, _ = parseOptionString("12 x 1.1.1.1")
|
|
assert.EqualValues(t, 0, code)
|
|
}
|