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
|
package async
import (
"context"
"net"
"testing"
"time"
"github.com/insomniacslk/dhcp/dhcpv4"
"github.com/stretchr/testify/require"
)
// server creates a server which responds with a predefined response
func serve(ctx context.Context, addr *net.UDPAddr, response *dhcpv4.DHCPv4) error {
conn, err := net.ListenUDP("udp4", addr)
if err != nil {
return err
}
go func() {
defer conn.Close()
oobdata := []byte{}
buffer := make([]byte, dhcpv4.MaxUDPReceivedPacketSize)
for {
select {
case <-ctx.Done():
return
default:
conn.SetReadDeadline(time.Now().Add(1 * time.Second))
n, _, _, src, err := conn.ReadMsgUDP(buffer, oobdata)
if err != nil {
continue
}
_, err = dhcpv4.FromBytes(buffer[:n])
if err != nil {
continue
}
conn.SetWriteDeadline(time.Now().Add(1 * time.Second))
_, err = conn.WriteTo(response.ToBytes(), src)
if err != nil {
continue
}
}
}
}()
return nil
}
func TestNewClient(t *testing.T) {
c := NewClient()
require.NotNil(t, c)
require.Equal(t, c.ReadTimeout, dhcpv4.DefaultReadTimeout)
require.Equal(t, c.ReadTimeout, dhcpv4.DefaultWriteTimeout)
}
func TestOpenInvalidAddrFailes(t *testing.T) {
c := NewClient()
err := c.Open(512)
require.Error(t, err)
}
// This test uses port 15438 so please make sure its not used before running
func TestOpenClose(t *testing.T) {
c := NewClient()
addr, err := net.ResolveUDPAddr("udp4", "127.0.0.1:15438")
require.NoError(t, err)
c.LocalAddr = addr
err = c.Open(512)
require.NoError(t, err)
defer c.Close()
}
// This test uses ports 15438 and 15439 so please make sure they are not used
// before running
func TestSendTimeout(t *testing.T) {
c := NewClient()
addr, err := net.ResolveUDPAddr("udp4", "127.0.0.1:15438")
require.NoError(t, err)
remote, err := net.ResolveUDPAddr("udp4", "127.0.0.1:15439")
require.NoError(t, err)
c.ReadTimeout = 50 * time.Millisecond
c.WriteTimeout = 50 * time.Millisecond
c.LocalAddr = addr
c.RemoteAddr = remote
err = c.Open(512)
require.NoError(t, err)
defer c.Close()
m, err := dhcpv4.New()
require.NoError(t, err)
_, err, timeout := c.Send(m).GetOrTimeout(200)
require.NoError(t, err)
require.True(t, timeout)
}
// This test uses ports 15438 and 15439 so please make sure they are not used
// before running
func TestSend(t *testing.T) {
m, err := dhcpv4.New()
require.NoError(t, err)
require.NotNil(t, m)
c := NewClient()
addr, err := net.ResolveUDPAddr("udp4", "127.0.0.1:15438")
require.NoError(t, err)
remote, err := net.ResolveUDPAddr("udp4", "127.0.0.1:15439")
require.NoError(t, err)
c.LocalAddr = addr
c.RemoteAddr = remote
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
err = serve(ctx, remote, m)
require.NoError(t, err)
err = c.Open(16)
require.NoError(t, err)
defer c.Close()
f := c.Send(m)
response, err, timeout := f.GetOrTimeout(2000)
r, ok := response.(*dhcpv4.DHCPv4)
require.True(t, ok)
require.False(t, timeout)
require.NoError(t, err)
require.Equal(t, m.TransactionID(), r.TransactionID())
}
|
