AdGuardHome/internal/dhcpsvc/server.go

package dhcpsvc

import (
	"fmt"
	"sync/atomic"
	"time"

	"golang.org/x/exp/maps"
	"golang.org/x/exp/slices"
)

// DHCPServer is a DHCP server for both IPv4 and IPv6 address families.
type DHCPServer struct {
	// enabled indicates whether the DHCP server is enabled and can provide
	// information about its clients.
	enabled *atomic.Bool

	// localTLD is the top-level domain name to use for resolving DHCP
	// clients' hostnames.
	localTLD string

	// interfaces4 is the set of IPv4 interfaces sorted by interface name.
	interfaces4 []*iface4

	// interfaces6 is the set of IPv6 interfaces sorted by interface name.
	interfaces6 []*iface6

	// leases is the set of active DHCP leases.
	leases []*Lease

	// icmpTimeout is the timeout for checking another DHCP server's presence.
	icmpTimeout time.Duration
}

// New creates a new DHCP server with the given configuration.  It returns an
// error if the given configuration can't be used.
//
// TODO(e.burkov):  Use.
func New(conf *Config) (srv *DHCPServer, err error) {
	if !conf.Enabled {
		// TODO(e.burkov):  Perhaps return [Empty]?
		return nil, nil
	}

	ifaces4 := make([]*iface4, len(conf.Interfaces))
	ifaces6 := make([]*iface6, len(conf.Interfaces))

	ifaceNames := maps.Keys(conf.Interfaces)
	slices.Sort(ifaceNames)

	var i4 *iface4
	var i6 *iface6

	for _, ifaceName := range ifaceNames {
		iface := conf.Interfaces[ifaceName]

		i4, err = newIface4(ifaceName, iface.IPv4)
		if err != nil {
			return nil, fmt.Errorf("interface %q: ipv4: %w", ifaceName, err)
		} else if i4 != nil {
			ifaces4 = append(ifaces4, i4)
		}

		i6 = newIface6(ifaceName, iface.IPv6)
		if i6 != nil {
			ifaces6 = append(ifaces6, i6)
		}
	}

	enabled := &atomic.Bool{}
	enabled.Store(conf.Enabled)

	return &DHCPServer{
		enabled:     enabled,
		interfaces4: ifaces4,
		interfaces6: ifaces6,
		localTLD:    conf.LocalDomainName,
		icmpTimeout: conf.ICMPTimeout,
	}, nil
}

// type check
//
// TODO(e.burkov):  Uncomment when the [Interface] interface is implemented.
// var _ Interface = (*DHCPServer)(nil)

// Enabled implements the [Interface] interface for *DHCPServer.
func (srv *DHCPServer) Enabled() (ok bool) {
	return srv.enabled.Load()
}

// Leases implements the [Interface] interface for *DHCPServer.
func (srv *DHCPServer) Leases() (leases []*Lease) {
	leases = make([]*Lease, 0, len(srv.leases))
	for _, lease := range srv.leases {
		leases = append(leases, lease.Clone())
	}

	return leases
}