diff options
Diffstat (limited to 'dhcpv6/modifiers.go')
| -rw-r--r-- | dhcpv6/modifiers.go | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/dhcpv6/modifiers.go b/dhcpv6/modifiers.go index 4d00a0c..be4c367 100644 --- a/dhcpv6/modifiers.go +++ b/dhcpv6/modifiers.go @@ -8,19 +8,21 @@ import ( "github.com/insomniacslk/dhcp/rfc1035label" ) -// WithClientID adds a client ID option to a DHCPv6 packet -func WithClientID(duid Duid) Modifier { +// WithOption adds the specific option to the DHCPv6 message. +func WithOption(o Option) Modifier { return func(d DHCPv6) { - d.UpdateOption(OptClientID(duid)) + d.UpdateOption(o) } } +// WithClientID adds a client ID option to a DHCPv6 packet +func WithClientID(duid Duid) Modifier { + return WithOption(OptClientID(duid)) +} + // WithServerID adds a client ID option to a DHCPv6 packet func WithServerID(duid Duid) Modifier { - return func(d DHCPv6) { - sid := OptServerId{Sid: duid} - d.UpdateOption(&sid) - } + return WithOption(OptServerID(duid)) } // WithNetboot adds bootfile URL and bootfile param options to a DHCPv6 packet. |