1 files changed, 128 insertions, 0 deletions
diff --git a/dhcpv4/server4/server_test.go b/dhcpv4/server4/server_test.go
new file mode 100644
index 0000000..f6ce18e
--- /dev/null
+++ b/dhcpv4/server4/server_test.go
@@ -0,0 +1,128 @@
+// +build integration
+
+package server4
+
+import (
+	"log"
+	"math/rand"
+	"net"
+	"testing"
+	"time"
+
+	"github.com/insomniacslk/dhcp/dhcpv4"
+	"github.com/insomniacslk/dhcp/dhcpv4/client4"
+	"github.com/insomniacslk/dhcp/interfaces"
+	"github.com/stretchr/testify/require"
+)
+
+func init() {
+	// initialize seed. This is generally bad, but "good enough"
+	// to generate random ports for these tests
+	rand.Seed(time.Now().UTC().UnixNano())
+}
+
+func randPort() int {
+	// can't use port 0 with raw sockets, so until we implement
+	// a non-raw-sockets client for non-static ports, we have to
+	// deal with this "randomness"
+	return 1024 + rand.Intn(65536-1024)
+}
+
+// DORAHandler is a server handler suitable for DORA transactions
+func DORAHandler(conn net.PacketConn, peer net.Addr, m *dhcpv4.DHCPv4) {
+	if m == nil {
+		log.Printf("Packet is nil!")
+		return
+	}
+	if m.OpCode != dhcpv4.OpcodeBootRequest {
+		log.Printf("Not a BootRequest!")
+		return
+	}
+	reply, err := dhcpv4.NewReplyFromRequest(m)
+	if err != nil {
+		log.Printf("NewReplyFromRequest failed: %v", err)
+		return
+	}
+	reply.UpdateOption(dhcpv4.OptServerIdentifier(net.IP{1, 2, 3, 4}))
+	switch mt := m.MessageType(); mt {
+	case dhcpv4.MessageTypeDiscover:
+		reply.UpdateOption(dhcpv4.OptMessageType(dhcpv4.MessageTypeOffer))
+	case dhcpv4.MessageTypeRequest:
+		reply.UpdateOption(dhcpv4.OptMessageType(dhcpv4.MessageTypeAck))
+	default:
+		log.Printf("Unhandled message type: %v", mt)
+		return
+	}
+
+	if _, err := conn.WriteTo(reply.ToBytes(), peer); err != nil {
+		log.Printf("Cannot reply to client: %v", err)
+	}
+}
+
+// utility function to set up a client and a server instance and run it in
+// background. The caller needs to call Server.Close() once finished.
+func setUpClientAndServer(handler Handler) (*client4.Client, *Server) {
+	// strong assumption, I know
+	loAddr := net.ParseIP("127.0.0.1")
+	laddr := net.UDPAddr{
+		IP:   loAddr,
+		Port: randPort(),
+	}
+	s := NewServer(laddr, handler)
+	go s.ActivateAndServe()
+
+	c := client4.NewClient()
+	// FIXME this doesn't deal well with raw sockets, the actual 0 will be used
+	// in the UDP header as source port
+	c.LocalAddr = &net.UDPAddr{IP: loAddr, Port: randPort()}
+	for {
+		if s.LocalAddr() != nil {
+			break
+		}
+		time.Sleep(10 * time.Millisecond)
+		log.Printf("Waiting for server to run...")
+	}
+	c.RemoteAddr = s.LocalAddr()
+	log.Printf("Client.RemoteAddr: %s", c.RemoteAddr)
+
+	return c, s
+}
+
+func TestNewServer(t *testing.T) {
+	laddr := net.UDPAddr{
+		IP:   net.ParseIP("127.0.0.1"),
+		Port: 0,
+	}
+	s := NewServer(laddr, DORAHandler)
+	defer s.Close()
+
+	require.NotNil(t, s)
+	require.Nil(t, s.conn)
+	require.Equal(t, laddr, s.localAddr)
+	require.NotNil(t, s.Handler)
+}
+
+func TestServerActivateAndServe(t *testing.T) {
+	c, s := setUpClientAndServer(DORAHandler)
+	defer s.Close()
+
+	ifaces, err := interfaces.GetLoopbackInterfaces()
+	require.NoError(t, err)
+	require.NotEqual(t, 0, len(ifaces))
+
+	xid := dhcpv4.TransactionID{0xaa, 0xbb, 0xcc, 0xdd}
+	hwaddr := net.HardwareAddr{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0xa, 0xb, 0xc, 0xd, 0xe, 0xf}
+
+	modifiers := []dhcpv4.Modifier{
+		dhcpv4.WithTransactionID(xid),
+		dhcpv4.WithHwAddr(hwaddr),
+	}
+
+	conv, err := c.Exchange(ifaces[0].Name, modifiers...)
+	require.NoError(t, err)
+	require.Equal(t, 4, len(conv))
+	for _, p := range conv {
+		require.Equal(t, xid, p.TransactionID)
+		require.Equal(t, hwaddr, p.ClientHWAddr)
+	}
+}