diff options
Diffstat (limited to 'dhcpv6/option_prefixdelegation_test.go')
-rw-r--r-- | dhcpv6/option_prefixdelegation_test.go | 132 |
1 files changed, 132 insertions, 0 deletions
diff --git a/dhcpv6/option_prefixdelegation_test.go b/dhcpv6/option_prefixdelegation_test.go new file mode 100644 index 0000000..caedd40 --- /dev/null +++ b/dhcpv6/option_prefixdelegation_test.go @@ -0,0 +1,132 @@ +package dhcpv6 + +import ( + "testing" + + "github.com/stretchr/testify/require" +) + +func TestOptIAForPrefixDelegationParseOptIAForPrefixDelegation(t *testing.T) { + data := []byte{ + 1, 0, 0, 0, // IAID + 0, 0, 0, 1, // T1 + 0, 0, 0, 2, // T2 + 0, 26, 0, 25, // 26 = IAPrefix Option, 25 = length + 0xaa, 0xbb, 0xcc, 0xdd, // IAPrefix preferredLifetime + 0xee, 0xff, 0x00, 0x11, // IAPrefix validLifetime + 36, // IAPrefix prefixLength + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, // IAPrefix ipv6Prefix + } + opt, err := ParseOptIAForPrefixDelegation(data) + require.NoError(t, err) + require.Equal(t, len(data), opt.Length()) +} + +func TestOptIAForPrefixDelegationParseOptIAForPrefixDelegationInvalidLength(t *testing.T) { + data := []byte{ + 1, 0, 0, 0, // IAID + 0, 0, 0, 1, // T1 + // truncated from here + } + _, err := ParseOptIAForPrefixDelegation(data) + require.Error(t, err) +} + +func TestOptIAForPrefixDelegationParseOptIAForPrefixDelegationInvalidOptions(t *testing.T) { + data := []byte{ + 1, 0, 0, 0, // IAID + 0, 0, 0, 1, // T1 + 0, 0, 0, 2, // T2 + 0, 26, 0, 25, // 26 = IAPrefix Option, 25 = length + 0xaa, 0xbb, 0xcc, 0xdd, // IAPrefix preferredLifetime + 0xee, 0xff, 0x00, 0x11, // IAPrefix validLifetime + 36, // IAPrefix prefixLength + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // IAPrefix ipv6Prefix missing last byte + } + _, err := ParseOptIAForPrefixDelegation(data) + require.Error(t, err) +} + +func TestOptIAForPrefixDelegationGetOneOption(t *testing.T) { + buf := []byte{ + 0xaa, 0xbb, 0xcc, 0xdd, // preferredLifetime + 0xee, 0xff, 0x00, 0x11, // validLifetime + 36, // prefixLength + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, // ipv6Prefix + } + oaddr, err := ParseOptIAPrefix(buf) + if err != nil { + t.Fatal(err) + } + opt := OptIAForPrefixDelegation{} + opt.SetOptions(oaddr.ToBytes()) + require.Equal(t, oaddr, opt.GetOneOption(OptionIAPrefix)) +} + +func TestOptIAForPrefixDelegationGetOneOptionMissingOpt(t *testing.T) { + buf := []byte{ + 0xaa, 0xbb, 0xcc, 0xdd, // preferredLifetime + 0xee, 0xff, 0x00, 0x11, // validLifetime + 36, // prefixLength + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, // ipv6Prefix + } + oaddr, err := ParseOptIAPrefix(buf) + if err != nil { + t.Fatal(err) + } + opt := OptIAForPrefixDelegation{} + opt.SetOptions(oaddr.ToBytes()) + require.Equal(t, nil, opt.GetOneOption(OptionDNSRecursiveNameServer)) +} + +func TestOptIAForPrefixDelegationDelOption(t *testing.T) { + optiana1 := OptIAForPrefixDelegation{} + optiana2 := OptIAForPrefixDelegation{} + optiaaddr := OptIAPrefix{} + optsc := OptStatusCode{} + + var buf = make([]byte, 0) + + buf = append(buf, optsc.ToBytes()...) + buf = append(buf, optiaaddr.ToBytes()...) + buf = append(buf, optiaaddr.ToBytes()...) + optiana1.SetOptions(buf) + optiana1.DelOption(OptionIAPrefix) + require.Equal(t, optiana1.Options(), optsc.ToBytes()) + + buf = make([]byte, 0) + buf = append(buf, optiaaddr.ToBytes()...) + buf = append(buf, optsc.ToBytes()...) + buf = append(buf, optiaaddr.ToBytes()...) + optiana2.SetOptions(buf) + optiana2.DelOption(OptionIAPrefix) + require.Equal(t, optiana2.Options(), optsc.ToBytes()) +} + +func TestOptIAForPrefixDelegationToBytes(t *testing.T) { + oaddr := OptIAPrefix{} + oaddr.SetPreferredLifetime(0xaabbccdd) + oaddr.SetValidLifetime(0xeeff0011) + oaddr.SetPrefixLength(36) + oaddr.SetIPv6Prefix([16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1}) + + opt := OptIAForPrefixDelegation{ + iaId: [4]byte{1, 2, 3, 4}, + t1: 12345, + t2: 54321, + } + opt.SetOptions(oaddr.ToBytes()) + expected := []byte{ + 0, 25, // OptionIAPD + 0, 41, // length + 1, 2, 3, 4, // IA ID + 0, 0, 0x30, 0x39, // T1 = 12345 + 0, 0, 0xd4, 0x31, // T2 = 54321 + 0, 26, 0, 25, // 26 = IAPrefix Option, 25 = length + 0xaa, 0xbb, 0xcc, 0xdd, // IAPrefix preferredLifetime + 0xee, 0xff, 0x00, 0x11, // IAPrefix validLifetime + 36, // IAPrefix prefixLength + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, // IAPrefix ipv6Prefix + } + require.Equal(t, expected, opt.ToBytes()) +} |