add IsRequested

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}