netdata/netdata

// SPDX-License-Identifier: GPL-3.0-or-later

package isc_dhcpd

import (
    "bufio"
    "bytes"
    "net"
    "os"
)

/*
Documentation (v4.4): https://kb.isc.org/docs/en/isc-dhcp-44-manual-pages-dhcpdleases

DHCPv4 prepare declaration:
  prepare ip-address {
    statements...
  }

DHCPv6 prepare declaration:
  ia_ta IAID_DUID {
    cltt date;
    iaaddr ipv6-address {
      statements...
    }
  }
  ia_na IAID_DUID {
    cltt date;
    iaaddr ipv6-address {
      statements...
    }
  }
  ia_pd IAID_DUID {
    cltt date;
    iaprefix ipv6-address/prefix-length {
      statements...
    }
  }
*/

type leaseEntry struct {
    ip           net.IP
    bindingState string
}

func (l leaseEntry) hasIP() bool           { return l.ip != nil }
func (l leaseEntry) hasBindingState() bool { return l.bindingState != "" }

func parseDHCPdLeasesFile(filepath string) ([]leaseEntry, error) {
    f, err := os.Open(filepath)
    if err != nil {
        return nil, err
    }
    defer func() { _ = f.Close() }()

    leasesSet := make(map[string]leaseEntry)
    l := leaseEntry{}
    sc := bufio.NewScanner(f)

    for sc.Scan() {
        bs := bytes.TrimSpace(sc.Bytes())
        switch {
        case !l.hasIP() && bytes.HasPrefix(bs, []byte("lease")):
            // "lease 192.168.0.1 {" => "192.168.0.1"
            s := string(bs)
            l.ip = net.ParseIP(s[6 : len(s)-2])
        case !l.hasIP() && bytes.HasPrefix(bs, []byte("iaaddr")):
            // "iaaddr 1985:470:1f0b:c9a::001 {" =>  "1985:470:1f0b:c9a::001"
            s := string(bs)
            l.ip = net.ParseIP(s[7 : len(s)-2])
        case l.hasIP() && !l.hasBindingState() && bytes.HasPrefix(bs, []byte("binding state")):
            // "binding state active;" => "active"
            s := string(bs)
            l.bindingState = s[14 : len(s)-1]
        case bytes.HasPrefix(bs, []byte("}")):
            if l.hasIP() && l.hasBindingState() {
                leasesSet[l.ip.String()] = l
            }
            l = leaseEntry{}
        }
    }

    if len(leasesSet) == 0 {
        return nil, nil
    }

    leases := make([]leaseEntry, 0, len(leasesSet))
    for _, l := range leasesSet {
        leases = append(leases, l)
    }
    return leases, nil
}