mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-12-15 19:31:45 +03:00
89bf3721b5
Merge in DNS/adguard-home from AG-20352-dhcpd-lease-is-static to master Squashed commit of the following: commit e4f4278a7ffa0f084ed41472dd3e7de4466c9f50 Merge: b6e3b62e15bba281
Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Mon Apr 10 11:13:05 2023 +0300 Merge branch 'master' into AG-20352-dhcpd-lease-is-static commit b6e3b62ed4e7bc17fc3fdd2f4faa940a7f4334c2 Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Thu Apr 6 15:20:42 2023 +0300 dhcpd: imp naming commit e2d9ed0832b329f4cebcf8cbfcfadc5755fe441b Merge: ecd244a610bffd89
Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Apr 5 16:53:55 2023 +0300 Merge branch 'master' into AG-20352-dhcpd-lease-is-static commit ecd244a60841d3cb96d292da688e353650baf645 Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Apr 5 16:53:15 2023 +0300 dhcpd: add lease json form commit 9ebc246ed4711ff46091326a7cb35ea90b880cff Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Apr 5 13:11:25 2023 +0300 dhcpd: imp code commit fc3d0cdaebf9e32e73d57e80296f9891896259cf Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Apr 5 13:00:59 2023 +0300 all: fix json commit d722578543b98b1fefabecd6486f3bc102263d71 Author: Stanislav Chzhen <s.chzhen@adguard.com> Date: Wed Apr 5 12:23:34 2023 +0300 dhcpd: add lease is static
194 lines
3.9 KiB
Go
194 lines
3.9 KiB
Go
// On-disk database for lease table
|
|
|
|
package dhcpd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net"
|
|
"net/netip"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/AdguardTeam/golibs/errors"
|
|
"github.com/AdguardTeam/golibs/log"
|
|
"github.com/google/renameio/maybe"
|
|
)
|
|
|
|
const dbFilename = "leases.db"
|
|
|
|
type leaseJSON struct {
|
|
HWAddr []byte `json:"mac"`
|
|
IP []byte `json:"ip"`
|
|
Hostname string `json:"host"`
|
|
Expiry int64 `json:"exp"`
|
|
}
|
|
|
|
func normalizeIP(ip net.IP) net.IP {
|
|
ip4 := ip.To4()
|
|
if ip4 != nil {
|
|
return ip4
|
|
}
|
|
return ip
|
|
}
|
|
|
|
// Load lease table from DB
|
|
//
|
|
// TODO(s.chzhen): Decrease complexity.
|
|
func (s *server) dbLoad() (err error) {
|
|
dynLeases := []*Lease{}
|
|
staticLeases := []*Lease{}
|
|
v6StaticLeases := []*Lease{}
|
|
v6DynLeases := []*Lease{}
|
|
|
|
data, err := os.ReadFile(s.conf.DBFilePath)
|
|
if err != nil {
|
|
if !errors.Is(err, os.ErrNotExist) {
|
|
return fmt.Errorf("reading db: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
obj := []leaseJSON{}
|
|
err = json.Unmarshal(data, &obj)
|
|
if err != nil {
|
|
return fmt.Errorf("decoding db: %w", err)
|
|
}
|
|
|
|
numLeases := len(obj)
|
|
for i := range obj {
|
|
obj[i].IP = normalizeIP(obj[i].IP)
|
|
|
|
ip, ok := netip.AddrFromSlice(obj[i].IP)
|
|
if !ok {
|
|
log.Info("dhcp: invalid IP: %s", obj[i].IP)
|
|
continue
|
|
}
|
|
|
|
lease := Lease{
|
|
HWAddr: obj[i].HWAddr,
|
|
IP: ip,
|
|
Hostname: obj[i].Hostname,
|
|
Expiry: time.Unix(obj[i].Expiry, 0),
|
|
IsStatic: obj[i].Expiry == leaseExpireStatic,
|
|
}
|
|
|
|
if len(obj[i].IP) == 16 {
|
|
if lease.IsStatic {
|
|
v6StaticLeases = append(v6StaticLeases, &lease)
|
|
} else {
|
|
v6DynLeases = append(v6DynLeases, &lease)
|
|
}
|
|
} else {
|
|
if lease.IsStatic {
|
|
staticLeases = append(staticLeases, &lease)
|
|
} else {
|
|
dynLeases = append(dynLeases, &lease)
|
|
}
|
|
}
|
|
}
|
|
|
|
leases4 := normalizeLeases(staticLeases, dynLeases)
|
|
err = s.srv4.ResetLeases(leases4)
|
|
if err != nil {
|
|
return fmt.Errorf("resetting dhcpv4 leases: %w", err)
|
|
}
|
|
|
|
leases6 := normalizeLeases(v6StaticLeases, v6DynLeases)
|
|
if s.srv6 != nil {
|
|
err = s.srv6.ResetLeases(leases6)
|
|
if err != nil {
|
|
return fmt.Errorf("resetting dhcpv6 leases: %w", err)
|
|
}
|
|
}
|
|
|
|
log.Info("dhcp: loaded leases v4:%d v6:%d total-read:%d from DB",
|
|
len(leases4), len(leases6), numLeases)
|
|
|
|
return nil
|
|
}
|
|
|
|
// Skip duplicate leases
|
|
// Static leases have a priority over dynamic leases
|
|
func normalizeLeases(staticLeases, dynLeases []*Lease) []*Lease {
|
|
leases := []*Lease{}
|
|
index := map[string]int{}
|
|
|
|
for i, lease := range staticLeases {
|
|
_, ok := index[lease.HWAddr.String()]
|
|
if ok {
|
|
continue // skip the lease with the same HW address
|
|
}
|
|
index[lease.HWAddr.String()] = i
|
|
leases = append(leases, lease)
|
|
}
|
|
|
|
for i, lease := range dynLeases {
|
|
_, ok := index[lease.HWAddr.String()]
|
|
if ok {
|
|
continue // skip the lease with the same HW address
|
|
}
|
|
index[lease.HWAddr.String()] = i
|
|
leases = append(leases, lease)
|
|
}
|
|
|
|
return leases
|
|
}
|
|
|
|
// Store lease table in DB
|
|
func (s *server) dbStore() (err error) {
|
|
// Use an empty slice here as opposed to nil so that it doesn't write
|
|
// "null" into the database file if leases are empty.
|
|
leases := []leaseJSON{}
|
|
|
|
leases4 := s.srv4.getLeasesRef()
|
|
for _, l := range leases4 {
|
|
if l.Expiry.Unix() == 0 {
|
|
continue
|
|
}
|
|
|
|
lease := leaseJSON{
|
|
HWAddr: l.HWAddr,
|
|
IP: l.IP.AsSlice(),
|
|
Hostname: l.Hostname,
|
|
Expiry: l.Expiry.Unix(),
|
|
}
|
|
|
|
leases = append(leases, lease)
|
|
}
|
|
|
|
if s.srv6 != nil {
|
|
leases6 := s.srv6.getLeasesRef()
|
|
for _, l := range leases6 {
|
|
if l.Expiry.Unix() == 0 {
|
|
continue
|
|
}
|
|
|
|
lease := leaseJSON{
|
|
HWAddr: l.HWAddr,
|
|
IP: l.IP.AsSlice(),
|
|
Hostname: l.Hostname,
|
|
Expiry: l.Expiry.Unix(),
|
|
}
|
|
|
|
leases = append(leases, lease)
|
|
}
|
|
}
|
|
|
|
var data []byte
|
|
data, err = json.Marshal(leases)
|
|
if err != nil {
|
|
return fmt.Errorf("encoding db: %w", err)
|
|
}
|
|
|
|
err = maybe.WriteFile(s.conf.DBFilePath, data, 0o644)
|
|
if err != nil {
|
|
return fmt.Errorf("writing db: %w", err)
|
|
}
|
|
|
|
log.Info("dhcp: stored %d leases in db", len(leases))
|
|
|
|
return nil
|
|
}
|