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
|
package dhcpv6
import (
"fmt"
"net"
"time"
)
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
)
var AllDHCPRelayAgentsAndServers = net.ParseIP("ff02::1:2")
var AllDHCPServers = net.ParseIP("ff05::1:3")
type Client struct {
Dialer *net.Dialer
ReadTimeout *time.Duration
WriteTimeout *time.Duration
LocalAddr net.Addr
RemoteAddr net.Addr
}
// Make a stateful DHCPv6 request
func (c *Client) Exchange(ifname string, solicit DHCPv6) ([]DHCPv6, error) {
conversation := make([]DHCPv6, 0)
var err error
// Solicit
if solicit == nil {
solicit, err = NewSolicitForInterface(ifname)
if err != nil {
return conversation, err
}
}
conversation = append(conversation, solicit)
advertise, err := c.Solicit(ifname, solicit)
if err != nil {
return conversation, err
}
conversation = append(conversation, advertise)
// TODO request/reply
return conversation, nil
}
// send a SOLICIT, return a response, or nil
func (c *Client) Solicit(ifname string, solicit DHCPv6) (DHCPv6, error) {
// 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()
// set WriteTimeout to DefaultWriteTimeout if no other timeout is specified
var wtimeout time.Duration
if c.WriteTimeout == nil {
wtimeout = DefaultWriteTimeout
} else {
wtimeout = *c.WriteTimeout
}
conn.SetWriteDeadline(time.Now().Add(wtimeout))
// send the SOLICIT packet out
_, err = conn.WriteTo(solicit.ToBytes(), &raddr)
if err != nil {
return nil, err
}
// set ReadTimeout to DefaultReadTimeout if no other timeout is specified
var rtimeout time.Duration
if c.ReadTimeout == nil {
rtimeout = DefaultReadTimeout
} else {
rtimeout = *c.ReadTimeout
}
conn.SetReadDeadline(time.Now().Add(rtimeout))
// wait for an ADVERTISE response
buf := make([]byte, maxUDPReceivedPacketSize)
oobdata := []byte{} // ignoring oob data
n, _, _, _, err := conn.ReadMsgUDP(buf, oobdata)
if err != nil {
return nil, err
}
adv, err := FromBytes(buf[:n])
if err != nil {
return nil, err
}
return adv, nil
}
|
