diff options
| -rw-r--r-- | dhcpv6/dhcpv6_test.go | 6 | ||||
| -rw-r--r-- | dhcpv6/dhcpv6message.go | 5 |
2 files changed, 9 insertions, 2 deletions
diff --git a/dhcpv6/dhcpv6_test.go b/dhcpv6/dhcpv6_test.go index 6840252..0b7c895 100644 --- a/dhcpv6/dhcpv6_test.go +++ b/dhcpv6/dhcpv6_test.go @@ -174,6 +174,12 @@ func TestNewReplyFromDHCPv6Message(t *testing.T) { require.Equal(t, rep.(*DHCPv6Message).TransactionID(), msg.TransactionID()) require.Equal(t, rep.Type(), MessageTypeReply) + msg.SetMessage(MessageTypeInformationRequest) + rep, err = NewReplyFromDHCPv6Message(&msg, WithServerID(duid)) + require.NoError(t, err) + require.Equal(t, rep.(*DHCPv6Message).TransactionID(), msg.TransactionID()) + require.Equal(t, rep.Type(), MessageTypeReply) + msg.SetMessage(MessageTypeSolicit) rep, err = NewReplyFromDHCPv6Message(&msg) require.Error(t, err) diff --git a/dhcpv6/dhcpv6message.go b/dhcpv6/dhcpv6message.go index 48b82e9..cb17624 100644 --- a/dhcpv6/dhcpv6message.go +++ b/dhcpv6/dhcpv6message.go @@ -201,13 +201,14 @@ func NewRequestFromAdvertise(advertise DHCPv6, modifiers ...Modifier) (DHCPv6, e // NewReplyFromDHCPv6Message creates a new REPLY packet based on a // DHCPv6Message. The function is to be used when generating a reply to -// REQUEST, CONFIRM, RENEW, REBIND and RELEASE packets. +// REQUEST, CONFIRM, RENEW, REBIND, RELEASE and INFORMATION-REQUEST packets. func NewReplyFromDHCPv6Message(message DHCPv6, modifiers ...Modifier) (DHCPv6, error) { if message == nil { return nil, errors.New("DHCPv6Message cannot be nil") } switch message.Type() { - case MessageTypeRequest, MessageTypeConfirm, MessageTypeRenew, MessageTypeRebind, MessageTypeRelease: + case MessageTypeRequest, MessageTypeConfirm, MessageTypeRenew, + MessageTypeRebind, MessageTypeRelease, MessageTypeInformationRequest: default: return nil, errors.New("Cannot create REPLY from the passed message type set") } |