116 lines
3.6 KiB
Go
116 lines
3.6 KiB
Go
package dhcpd
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/krolaw/dhcp4"
|
|
)
|
|
|
|
func check(t *testing.T, result bool, msg string) {
|
|
if !result {
|
|
t.Fatal(msg)
|
|
}
|
|
}
|
|
|
|
// Tests performed:
|
|
// . Handle Discover message (lease reserve)
|
|
// . Handle Request message (lease commit)
|
|
func TestDHCP(t *testing.T) {
|
|
var s = Server{}
|
|
var p, p2 dhcp4.Packet
|
|
var hw net.HardwareAddr
|
|
var lease *Lease
|
|
var opt dhcp4.Options
|
|
|
|
s.leaseStart = []byte{1, 1, 1, 1}
|
|
s.leaseStop = []byte{1, 1, 1, 2}
|
|
s.leaseTime = 5 * time.Second
|
|
s.leaseOptions = dhcp4.Options{}
|
|
s.ipnet = &net.IPNet{
|
|
IP: []byte{1, 2, 3, 4},
|
|
Mask: []byte{0xff, 0xff, 0xff, 0xff},
|
|
}
|
|
|
|
p = make(dhcp4.Packet, 241)
|
|
|
|
// Reserve an IP
|
|
hw = []byte{1, 2, 3, 4, 5, 6}
|
|
p.SetCHAddr(hw)
|
|
lease, _ = s.reserveLease(p)
|
|
check(t, bytes.Equal(lease.HWAddr, hw), "lease.HWAddr")
|
|
check(t, bytes.Equal(lease.IP, []byte{1, 1, 1, 1}), "lease.IP")
|
|
|
|
// Try to reserve another IP for the same machine - no new IP must be reserved
|
|
hw = []byte{1, 2, 3, 4, 5, 6}
|
|
p.SetCHAddr(hw)
|
|
lease, _ = s.reserveLease(p)
|
|
check(t, bytes.Equal(lease.HWAddr, hw), "lease.HWAddr")
|
|
check(t, bytes.Equal(lease.IP, []byte{1, 1, 1, 1}), "lease.IP")
|
|
|
|
// Reserve an IP - the next IP from the range
|
|
hw = []byte{2, 2, 3, 4, 5, 6}
|
|
p.SetCHAddr(hw)
|
|
lease, _ = s.reserveLease(p)
|
|
check(t, bytes.Equal(lease.HWAddr, hw), "lease.HWAddr")
|
|
check(t, bytes.Equal(lease.IP, []byte{1, 1, 1, 2}), "lease.IP")
|
|
|
|
// Reserve an IP - we have no more available IPs
|
|
p.SetCHAddr([]byte{3, 2, 3, 4, 5, 6})
|
|
lease, _ = s.reserveLease(p)
|
|
check(t, lease == nil, "lease == nil")
|
|
|
|
// Decline request for a lease which doesn't match our internal state
|
|
hw = []byte{1, 2, 3, 4, 5, 6}
|
|
p.SetCHAddr(hw)
|
|
p.SetCIAddr([]byte{0, 0, 0, 0})
|
|
opt = make(dhcp4.Options, 10)
|
|
// ask a different IP
|
|
opt[dhcp4.OptionRequestedIPAddress] = []byte{1, 1, 1, 2}
|
|
p2 = s.handleDHCP4Request(p, opt)
|
|
opt = p2.ParseOptions()
|
|
check(t, bytes.Equal(opt[dhcp4.OptionDHCPMessageType], []byte{byte(dhcp4.NAK)}), "dhcp4.NAK")
|
|
|
|
// Commit the previously reserved lease
|
|
hw = []byte{1, 2, 3, 4, 5, 6}
|
|
p.SetCHAddr(hw)
|
|
p.SetCIAddr([]byte{0, 0, 0, 0})
|
|
opt = make(dhcp4.Options, 10)
|
|
opt[dhcp4.OptionRequestedIPAddress] = []byte{1, 1, 1, 1}
|
|
p2 = s.handleDHCP4Request(p, opt)
|
|
opt = p2.ParseOptions()
|
|
check(t, bytes.Equal(opt[dhcp4.OptionDHCPMessageType], []byte{byte(dhcp4.ACK)}), "dhcp4.ACK")
|
|
check(t, bytes.Equal(p2.YIAddr(), []byte{1, 1, 1, 1}), "p2.YIAddr")
|
|
check(t, bytes.Equal(p2.CHAddr(), hw), "p2.CHAddr")
|
|
check(t, bytes.Equal(opt[dhcp4.OptionIPAddressLeaseTime], dhcp4.OptionsLeaseTime(5*time.Second)), "OptionIPAddressLeaseTime")
|
|
check(t, bytes.Equal(opt[dhcp4.OptionServerIdentifier], s.ipnet.IP), "OptionServerIdentifier")
|
|
|
|
s.reset()
|
|
misc(t, &s)
|
|
}
|
|
|
|
// Small tests that don't require a static server's state
|
|
func misc(t *testing.T, s *Server) {
|
|
var p, p2 dhcp4.Packet
|
|
var hw net.HardwareAddr
|
|
var opt dhcp4.Options
|
|
|
|
p = make(dhcp4.Packet, 241)
|
|
|
|
// Commit a lease for an IP without prior Discover-Offer packets
|
|
hw = []byte{2, 2, 3, 4, 5, 6}
|
|
p.SetCHAddr(hw)
|
|
p.SetCIAddr([]byte{0, 0, 0, 0})
|
|
opt = make(dhcp4.Options, 10)
|
|
opt[dhcp4.OptionRequestedIPAddress] = []byte{1, 1, 1, 1}
|
|
p2 = s.handleDHCP4Request(p, opt)
|
|
opt = p2.ParseOptions()
|
|
check(t, bytes.Equal(opt[dhcp4.OptionDHCPMessageType], []byte{byte(dhcp4.ACK)}), "dhcp4.ACK")
|
|
check(t, bytes.Equal(p2.YIAddr(), []byte{1, 1, 1, 1}), "p2.YIAddr")
|
|
check(t, bytes.Equal(p2.CHAddr(), hw), "p2.CHAddr")
|
|
check(t, bytes.Equal(opt[dhcp4.OptionIPAddressLeaseTime], dhcp4.OptionsLeaseTime(5*time.Second)), "OptionIPAddressLeaseTime")
|
|
check(t, bytes.Equal(opt[dhcp4.OptionServerIdentifier], s.ipnet.IP), "OptionServerIdentifier")
|
|
}
|