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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
package dhcpv6
import (
"bytes"
"net"
"testing"
"github.com/stretchr/testify/require"
)
func TestRelayMessage(t *testing.T) {
ll := net.IP{0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xaa, 0xbb, 0xcc, 0xff, 0xfe, 0xdd, 0xee, 0xff}
ma := net.IP{0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01}
r := RelayMessage{
MessageType: MessageTypeRelayForward,
HopCount: 10,
LinkAddr: ll,
PeerAddr: ma,
// options is left empty here for testing purposes, even if it's
// mandatory to have at least a relay message option
}
if mt := r.Type(); mt != MessageTypeRelayForward {
t.Fatalf("Invalid message type. Expected %v, got %v", MessageTypeRelayForward, mt)
}
if hc := r.HopCount; hc != 10 {
t.Fatalf("Invalid hop count. Expected 10, got %v", hc)
}
if la := r.LinkAddr; !la.Equal(ll) {
t.Fatalf("Invalid link address. Expected %v, got %v", ll, la)
}
if pa := r.PeerAddr; !pa.Equal(ma) {
t.Fatalf("Invalid peer address. Expected %v, got %v", ma, pa)
}
if opts := r.Options; len(opts) != 0 {
t.Fatalf("Invalid options. Expected none, got %v", opts)
}
}
func TestRelayMessageToBytesDefault(t *testing.T) {
want := []byte{
12, // MessageTypeRelayForward
0, // hop count
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // link addr
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // peer addr
}
r := RelayMessage{MessageType: MessageTypeRelayForward}
require.Equal(t, r.ToBytes(), want)
}
func TestRelayMessageToBytes(t *testing.T) {
expected := []byte{
12, // MessageTypeRelayForward
1, // hop count
0xff, 0x01, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0x01, // link addr
0xff, 0x02, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0x02, // peer addr
// option relay message
0, 9, // relay msg
0, 10, // option length
// inner dhcp solicit
1, // MessageTypeSolicit
0xaa, 0xbb, 0xcc, // transaction ID
// inner option - elapsed time
0, 8, // elapsed time
0, 2, // length
0, 0,
}
r := RelayMessage{
MessageType: MessageTypeRelayForward,
HopCount: 1,
LinkAddr: net.IPv6interfacelocalallnodes,
PeerAddr: net.IPv6linklocalallrouters,
}
opt := OptRelayMsg{
relayMessage: &Message{
MessageType: MessageTypeSolicit,
TransactionID: TransactionID{0xaa, 0xbb, 0xcc},
Options: MessageOptions{
Options: []Option{
OptElapsedTime(0),
},
},
},
}
r.AddOption(&opt)
relayBytes := r.ToBytes()
if !bytes.Equal(expected, relayBytes) {
t.Fatalf("Invalid ToBytes result. Expected %v, got %v", expected, relayBytes)
}
}
func TestNewRelayRepFromRelayForw(t *testing.T) {
// create a new relay forward
rf := RelayMessage{}
rf.MessageType = MessageTypeRelayForward
rf.PeerAddr = net.IPv6linklocalallrouters
rf.LinkAddr = net.IPv6interfacelocalallnodes
rf.AddOption(&OptInterfaceId{})
rf.AddOption(&OptRemoteId{})
// create the inner message
s, err := NewMessage()
require.NoError(t, err)
s.AddOption(OptClientID(Duid{}))
orm := OptRelayMsg{}
orm.SetRelayMessage(s)
rf.AddOption(&orm)
a, err := NewAdvertiseFromSolicit(s)
require.NoError(t, err)
rr, err := NewRelayReplFromRelayForw(&rf, a)
require.NoError(t, err)
relay := rr.(*RelayMessage)
require.Equal(t, rr.Type(), MessageTypeRelayReply)
require.Equal(t, relay.HopCount, rf.HopCount)
require.Equal(t, relay.PeerAddr, rf.PeerAddr)
require.Equal(t, relay.LinkAddr, rf.LinkAddr)
require.NotNil(t, rr.GetOneOption(OptionInterfaceID))
require.NotNil(t, rr.GetOneOption(OptionRemoteID))
m, err := relay.GetInnerMessage()
require.NoError(t, err)
require.Equal(t, m, a)
_, err = NewRelayReplFromRelayForw(nil, a)
require.Error(t, err)
_, err = NewRelayReplFromRelayForw(&rf, nil)
require.Error(t, err)
}
|
