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
|
package dhcpv6
import (
"bytes"
"net"
"testing"
)
func TestDHCPv6Relay(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 := DHCPv6Relay{
messageType: RELAY_FORW,
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.MessageType(); mt != RELAY_FORW {
t.Fatalf("Invalid message type. Expected %v, got %v", RELAY_FORW, mt)
}
if hc := r.HopCount(); hc != 10 {
t.Fatalf("Invalid hop count. Expected 10, got %v", hc)
}
if la := r.LinkAddr(); !bytes.Equal(la, ll) {
t.Fatalf("Invalid link address. Expected %v, got %v", ll, la)
}
if pa := r.PeerAddr(); !bytes.Equal(pa, 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 TestDHCPv6RelaySettersAndGetters(t *testing.T) {
r := DHCPv6Relay{}
// set and get message type
r.SetMessageType(RELAY_REPL)
if mt := r.MessageType(); mt != RELAY_REPL {
t.Fatalf("Invalid message type. Expected %v, got %v", RELAY_REPL, mt)
}
// set and get hop count
r.SetHopCount(5)
if hc := r.HopCount(); hc != 5 {
t.Fatalf("Invalid hop count. Expected 5, got %v", hc)
}
// set and get link address
r.SetLinkAddr(net.IPv6loopback)
if la := r.LinkAddr(); !bytes.Equal(la, net.IPv6loopback) {
t.Fatalf("Invalid link address. Expected %v, got %v", net.IPv6loopback, la)
}
// set and get peer address
r.SetPeerAddr(net.IPv6loopback)
if pa := r.PeerAddr(); !bytes.Equal(pa, net.IPv6loopback) {
t.Fatalf("Invalid peer address. Expected %v, got %v", net.IPv6loopback, pa)
}
// set and get options
oneOpt := []Option{&OptRelayMsg{relayMessage: &DHCPv6Message{}}}
r.SetOptions(oneOpt)
if opts := r.Options(); len(opts) != 1 || opts[0] != oneOpt[0] {
t.Fatalf("Invalid options. Expected %v, got %v", oneOpt, opts)
}
}
func TestDHCPv6RelayToBytes(t *testing.T) {
expected := []byte{
12, // RELAY_FORW
1, // hop count
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, // link addr
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, // peer addr
// option relay message
0, 9, // relay msg
0, 10, // option length
// inner dhcp solicit
1, // SOLICIT
0xaa, 0xbb, 0xcc, // transaction ID
// inner option - elapsed time
0, 8, // elapsed time
0, 2, // length
0, 0,
}
r := DHCPv6Relay{
messageType: RELAY_FORW,
hopCount: 1,
linkAddr: net.IPv6loopback,
peerAddr: net.IPv6loopback,
}
opt := OptRelayMsg{
relayMessage: &DHCPv6Message{
messageType: SOLICIT,
transactionID: 0xaabbcc,
options: []Option{
&OptElapsedTime{
elapsedTime: 0,
},
},
},
}
r.AddOption(&opt)
relayBytes := r.ToBytes()
if !bytes.Equal(expected, relayBytes) {
t.Fatalf("Invalid ToBytes result. Expected %v, got %v", expected, relayBytes)
}
}
|
