mirror of
https://github.com/coredhcp/coredhcp.git
synced 2026-03-31 19:11:04 +02:00
130 lines
3.2 KiB
Go
130 lines
3.2 KiB
Go
// Copyright 2018-present the CoreDHCP Authors. All rights reserved
|
|
// This source code is licensed under the MIT license found in the
|
|
// LICENSE file in the root directory of this source tree.
|
|
|
|
package serverid
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/insomniacslk/dhcp/dhcpv6"
|
|
)
|
|
|
|
func makeTestDUID(uuid string) *dhcpv6.Duid {
|
|
return &dhcpv6.Duid{
|
|
Type: dhcpv6.DUID_UUID,
|
|
Uuid: []byte(uuid),
|
|
}
|
|
}
|
|
|
|
func TestRejectBadServerIDV6(t *testing.T) {
|
|
req, err := dhcpv6.NewMessage()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
v6ServerID = makeTestDUID("0000000000000000")
|
|
|
|
req.MessageType = dhcpv6.MessageTypeRenew
|
|
dhcpv6.WithClientID(*makeTestDUID("1000000000000000"))(req)
|
|
dhcpv6.WithServerID(*makeTestDUID("0000000000000001"))(req)
|
|
|
|
stub, err := dhcpv6.NewReplyFromMessage(req)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
resp, stop := Handler6(req, stub)
|
|
if resp != nil {
|
|
t.Error("server_id is sending a response message to a request with mismatched ServerID")
|
|
}
|
|
if !stop {
|
|
t.Error("server_id did not interrupt processing on a request with mismatched ServerID")
|
|
}
|
|
}
|
|
|
|
func TestRejectUnexpectedServerIDV6(t *testing.T) {
|
|
req, err := dhcpv6.NewMessage()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
v6ServerID = makeTestDUID("0000000000000000")
|
|
|
|
req.MessageType = dhcpv6.MessageTypeSolicit
|
|
dhcpv6.WithClientID(*makeTestDUID("1000000000000000"))(req)
|
|
dhcpv6.WithServerID(*makeTestDUID("0000000000000000"))(req)
|
|
|
|
stub, err := dhcpv6.NewAdvertiseFromSolicit(req)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
resp, stop := Handler6(req, stub)
|
|
if resp != nil {
|
|
t.Error("server_id is sending a response message to a solicit with a ServerID")
|
|
}
|
|
if !stop {
|
|
t.Error("server_id did not interrupt processing on a solicit with a ServerID")
|
|
}
|
|
}
|
|
|
|
func TestAddServerIDV6(t *testing.T) {
|
|
req, err := dhcpv6.NewMessage()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
v6ServerID = makeTestDUID("0000000000000000")
|
|
|
|
req.MessageType = dhcpv6.MessageTypeRebind
|
|
dhcpv6.WithClientID(*makeTestDUID("1000000000000000"))(req)
|
|
|
|
stub, err := dhcpv6.NewReplyFromMessage(req)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
resp, _ := Handler6(req, stub)
|
|
if resp == nil {
|
|
t.Fatal("plugin did not return an answer")
|
|
}
|
|
|
|
if opt := resp.GetOneOption(dhcpv6.OptionServerID); opt == nil {
|
|
t.Fatal("plugin did not add a ServerID option")
|
|
} else {
|
|
sid := opt.(*dhcpv6.OptServerId)
|
|
if !sid.Sid.Equal(*v6ServerID) {
|
|
t.Fatalf("Got unexpected DUID: expected %v, got %v", v6ServerID, sid.Sid)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRejectInnerMessageServerID(t *testing.T) {
|
|
req, err := dhcpv6.NewMessage()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
v6ServerID = makeTestDUID("0000000000000000")
|
|
|
|
req.MessageType = dhcpv6.MessageTypeSolicit
|
|
dhcpv6.WithClientID(*makeTestDUID("1000000000000000"))(req)
|
|
dhcpv6.WithServerID(*makeTestDUID("0000000000000000"))(req)
|
|
|
|
stub, err := dhcpv6.NewAdvertiseFromSolicit(req)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
relayedRequest, err := dhcpv6.EncapsulateRelay(req, dhcpv6.MessageTypeRelayForward, net.IPv6loopback, net.IPv6loopback)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
resp, stop := Handler6(relayedRequest, stub)
|
|
if resp != nil {
|
|
t.Error("server_id is sending a response message to a relayed solicit with a ServerID")
|
|
}
|
|
if !stop {
|
|
t.Error("server_id did not interrupt processing on a relayed solicit with a ServerID")
|
|
}
|
|
}
|