1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
package dhcpv6
import (
"net"
"testing"
"github.com/stretchr/testify/require"
)
func TestOptIANAParseOptIANA(t *testing.T) {
data := []byte{
1, 0, 0, 0, // IAID
0, 0, 0, 1, // T1
0, 0, 0, 2, // T2
0, 5, 0, 0x18, 0x24, 1, 0xdb, 0, 0x30, 0x10, 0xc0, 0x8f, 0xfa, 0xce, 0, 0, 0, 0x44, 0, 0, 0, 0, 0xb2, 0x7a, 0, 0, 0xc0, 0x8a, // options
}
opt, err := ParseOptIANA(data)
require.NoError(t, err)
require.Equal(t, len(data), opt.Length())
}
func TestOptIANAParseOptIANAInvalidLength(t *testing.T) {
data := []byte{
1, 0, 0, 0, // IAID
0, 0, 0, 1, // T1
// truncated from here
}
_, err := ParseOptIANA(data)
require.Error(t, err)
}
func TestOptIANAParseOptIANAInvalidOptions(t *testing.T) {
data := []byte{
1, 0, 0, 0, // IAID
0, 0, 0, 1, // T1
0, 0, 0, 2, // T2
0, 5, 0, 0x18, 0x24, 1, 0xdb, 0, 0x30, 0x10, 0xc0, 0x8f, 0xfa, 0xce, 0, 0, 0, 0x44, 0, 0, 0, 0, 0xb2, 0x7a, // truncated options
}
_, err := ParseOptIANA(data)
require.Error(t, err)
}
func TestOptIANAGetOneOption(t *testing.T) {
oaddr := &OptIAAddress{
IPv6Addr: net.ParseIP("::1"),
}
opt := OptIANA{
Options: []Option{&OptElapsedTime{}, oaddr},
}
require.Equal(t, oaddr, opt.GetOneOption(OPTION_IAADDR))
}
func TestOptIANAGetOneOptionMissingOpt(t *testing.T) {
oaddr := &OptIAAddress{
IPv6Addr: net.ParseIP("::1"),
}
opt := OptIANA{
Options: []Option{&OptElapsedTime{}, oaddr},
}
require.Equal(t, nil, opt.GetOneOption(DNS_RECURSIVE_NAME_SERVER))
}
func TestOptIANADelOption(t *testing.T) {
optiana1 := OptIANA{}
optiana2 := OptIANA{}
optiaaddr := OptIAAddress{}
optsc := OptStatusCode{}
optiana1.Options = append(optiana1.Options, &optsc)
optiana1.Options = append(optiana1.Options, &optiaaddr)
optiana1.Options = append(optiana1.Options, &optiaaddr)
optiana1.DelOption(OPTION_IAADDR)
require.Equal(t, optiana1.Options, []Option{&optsc})
optiana2.Options = append(optiana2.Options, &optiaaddr)
optiana2.Options = append(optiana2.Options, &optsc)
optiana2.Options = append(optiana2.Options, &optiaaddr)
optiana2.DelOption(OPTION_IAADDR)
require.Equal(t, optiana2.Options, []Option{&optsc})
}
func TestOptIANAToBytes(t *testing.T) {
opt := OptIANA{
IaId: [4]byte{1, 2, 3, 4},
T1: 12345,
T2: 54321,
Options: []Option{
&OptElapsedTime{
ElapsedTime: 0xaabb,
},
},
}
expected := []byte{
0, 3, // OPTION_IA_NA
0, 18, // length
1, 2, 3, 4, // IA ID
0, 0, 0x30, 0x39, // T1 = 12345
0, 0, 0xd4, 0x31, // T2 = 54321
0, 8, 0, 2, 0xaa, 0xbb,
}
require.Equal(t, expected, opt.ToBytes())
}
|
