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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
|
package dhcpv6
import (
"fmt"
"net"
"time"
)
// Client constants
const (
DefaultWriteTimeout = 3 * time.Second // time to wait for write calls
DefaultReadTimeout = 3 * time.Second // time to wait for read calls
DefaultInterfaceUpTimeout = 3 * time.Second // time to wait before a network interface goes up
maxUDPReceivedPacketSize = 8192 // arbitrary size. Theoretically could be up to 65kb
)
// Broadcast destination IP addresses as defined by RFC 3315
var (
AllDHCPRelayAgentsAndServers = net.ParseIP("ff02::1:2")
AllDHCPServers = net.ParseIP("ff05::1:3")
)
// Client implements a DHCPv6 client
type Client struct {
ReadTimeout time.Duration
WriteTimeout time.Duration
LocalAddr net.Addr
RemoteAddr net.Addr
}
// NewClient returns a Client with default settings
func NewClient() *Client {
return &Client{
ReadTimeout: DefaultReadTimeout,
WriteTimeout: DefaultWriteTimeout,
}
}
// Exchange executes a 4-way DHCPv6 request (SOLICIT, ADVERTISE, REQUEST,
// REPLY). If the SOLICIT packet is nil, defaults are used. The modifiers will
// be applied to the Request packet. A common use is to make sure that the
// Request packet has the right options, see modifiers.go
func (c *Client) Exchange(ifname string, solicit DHCPv6, modifiers ...Modifier) ([]DHCPv6, error) {
conversation := make([]DHCPv6, 0)
var err error
// Solicit
if solicit == nil {
solicit, err = NewSolicitForInterface(ifname)
if err != nil {
return conversation, err
}
}
solicit, advertise, err := c.Solicit(ifname, solicit, modifiers...)
conversation = append(conversation, solicit)
if err != nil {
return conversation, err
}
conversation = append(conversation, advertise)
request, reply, err := c.Request(ifname, advertise, nil, modifiers...)
if request != nil {
conversation = append(conversation, request)
}
if err != nil {
return conversation, err
}
conversation = append(conversation, reply)
return conversation, nil
}
func (c *Client) sendReceive(ifname string, packet DHCPv6, expectedType MessageType) (DHCPv6, error) {
if packet == nil {
return nil, fmt.Errorf("Packet to send cannot be nil")
}
if expectedType == MSGTYPE_NONE {
// infer the expected type from the packet being sent
if packet.Type() == SOLICIT {
expectedType = ADVERTISE
} else if packet.Type() == REQUEST {
expectedType = REPLY
} else if packet.Type() == RELAY_FORW {
expectedType = RELAY_REPL
} else if packet.Type() == LEASEQUERY {
expectedType = LEASEQUERY_REPLY
} // and probably more
}
// if no LocalAddr is specified, get the interface's link-local address
var laddr net.UDPAddr
if c.LocalAddr == nil {
llAddr, err := GetLinkLocalAddr(ifname)
if err != nil {
return nil, err
}
laddr = net.UDPAddr{IP: *llAddr, Port: DefaultClientPort, Zone: ifname}
} else {
if addr, ok := c.LocalAddr.(*net.UDPAddr); ok {
laddr = *addr
} else {
return nil, fmt.Errorf("Invalid local address: not a net.UDPAddr: %v", c.LocalAddr)
}
}
// if no RemoteAddr is specified, use AllDHCPRelayAgentsAndServers
var raddr net.UDPAddr
if c.RemoteAddr == nil {
raddr = net.UDPAddr{IP: AllDHCPRelayAgentsAndServers, Port: DefaultServerPort}
} else {
if addr, ok := c.RemoteAddr.(*net.UDPAddr); ok {
raddr = *addr
} else {
return nil, fmt.Errorf("Invalid remote address: not a net.UDPAddr: %v", c.RemoteAddr)
}
}
// prepare the socket to listen on for replies
conn, err := net.ListenUDP("udp6", &laddr)
if err != nil {
return nil, err
}
defer conn.Close()
// send the packet out
conn.SetWriteDeadline(time.Now().Add(c.WriteTimeout))
_, err = conn.WriteTo(packet.ToBytes(), &raddr)
if err != nil {
return nil, err
}
// wait for a reply
oobdata := []byte{} // ignoring oob data
conn.SetReadDeadline(time.Now().Add(c.ReadTimeout))
var (
adv DHCPv6
isMessage bool
)
defer conn.Close()
msg, ok := packet.(*DHCPv6Message)
if ok {
isMessage = true
}
for {
buf := make([]byte, maxUDPReceivedPacketSize)
n, _, _, _, err := conn.ReadMsgUDP(buf, oobdata)
if err != nil {
return nil, err
}
adv, err = FromBytes(buf[:n])
if err != nil {
// skip non-DHCP packets
continue
}
if recvMsg, ok := adv.(*DHCPv6Message); ok && isMessage {
// if a regular message, check the transaction ID first
// XXX should this unpack relay messages and check the XID of the
// inner packet too?
if msg.TransactionID() != recvMsg.TransactionID() {
// different XID, we don't want this packet for sure
continue
}
}
if expectedType == MSGTYPE_NONE {
// just take whatever arrived
break
} else if adv.Type() == expectedType {
break
}
}
return adv, nil
}
// Solicit sends a SOLICIT, return the solicit, an ADVERTISE (if not nil), and
// an error if any
func (c *Client) Solicit(ifname string, solicit DHCPv6, modifiers ...Modifier) (DHCPv6, DHCPv6, error) {
var err error
if solicit == nil {
solicit, err = NewSolicitForInterface(ifname)
if err != nil {
return nil, nil, err
}
}
for _, mod := range modifiers {
solicit = mod(solicit)
}
advertise, err := c.sendReceive(ifname, solicit, MSGTYPE_NONE)
return solicit, advertise, err
}
// Request sends a REQUEST built from an ADVERTISE if no REQUEST is specified.
// It returns the request, a reply if not nil, and an error if any
func (c *Client) Request(ifname string, advertise, request DHCPv6, modifiers ...Modifier) (DHCPv6, DHCPv6, error) {
if request == nil {
var err error
request, err = NewRequestFromAdvertise(advertise)
if err != nil {
return nil, nil, err
}
}
for _, mod := range modifiers {
request = mod(request)
}
reply, err := c.sendReceive(ifname, request, MSGTYPE_NONE)
return request, reply, err
}
|
