summaryrefslogtreecommitdiffhomepage
path: root/dhcpv6
diff options
context:
space:
mode:
Diffstat (limited to 'dhcpv6')
-rw-r--r--dhcpv6/utils.go17
-rw-r--r--dhcpv6/utils_test.go11
2 files changed, 24 insertions, 4 deletions
diff --git a/dhcpv6/utils.go b/dhcpv6/utils.go
index 1681661..d1715da 100644
--- a/dhcpv6/utils.go
+++ b/dhcpv6/utils.go
@@ -11,16 +11,25 @@ import (
// if the "boot file" option is included in the packet, which is useful for
// ADVERTISE/REPLY packet.
func IsNetboot(msg DHCPv6) bool {
+ if IsRequested(msg, OptionBootfileURL) {
+ return true
+ }
+ if optbf := msg.GetOneOption(OptionBootfileURL); optbf != nil {
+ return true
+ }
+ return false
+}
+
+// IsRequested function takes a DHCPv6 message and an OptionCode, and returns
+// true if that option is within the requested options of the DHCPv6 message.
+func IsRequested(msg DHCPv6, requested OptionCode) bool {
for _, optoro := range msg.GetOption(OptionORO) {
for _, o := range optoro.(*OptRequestedOption).RequestedOptions() {
- if o == OptionBootfileURL {
+ if o == requested {
return true
}
}
}
- if optbf := msg.GetOneOption(OptionBootfileURL); optbf != nil {
- return true
- }
return false
}
diff --git a/dhcpv6/utils_test.go b/dhcpv6/utils_test.go
index f3b53f0..779d55c 100644
--- a/dhcpv6/utils_test.go
+++ b/dhcpv6/utils_test.go
@@ -22,6 +22,17 @@ func TestIsNetboot(t *testing.T) {
require.True(t, IsNetboot(&msg3))
}
+func TestIsRequetsed(t *testing.T) {
+ msg1 := DHCPv6Message{}
+ require.False(t, IsRequested(&msg1, OptionDNSRecursiveNameServer))
+
+ msg2 := DHCPv6Message{}
+ optro := OptRequestedOption{}
+ optro.AddRequestedOption(OptionDNSRecursiveNameServer)
+ msg2.AddOption(&optro)
+ require.True(t, IsRequested(&msg2, OptionDNSRecursiveNameServer))
+}
+
func TestIsUsingUEFIArchTypeTrue(t *testing.T) {
msg := DHCPv6Message{}
opt := OptClientArchType{ArchType: EFI_BC}