From b5c839d7f28cc635376d5bee36de0e898ad8a1b8 Mon Sep 17 00:00:00 2001
From: Owen Mooney <owen@owenmooney.com>
Date: Sun, 29 Jul 2018 14:14:31 +0100
Subject: Add support for modifiers to dhcpv4

---
 dhcpv4/modifiers.go | 11 +++++++++++
 1 file changed, 11 insertions(+)
 create mode 100644 dhcpv4/modifiers.go

(limited to 'dhcpv4/modifiers.go')

diff --git a/dhcpv4/modifiers.go b/dhcpv4/modifiers.go
new file mode 100644
index 0000000..bc19219
--- /dev/null
+++ b/dhcpv4/modifiers.go
@@ -0,0 +1,11 @@
+package dhcpv4
+
+// WithUserClass adds a user class option to the packet
+func WithUserClass(uc []byte) Modifier {
+	// TODO let the user specify multiple user classes
+	return func(d *DHCPv4) *DHCPv4 {
+		ouc := OptUserClass{UserClasses: [][]byte{uc}}
+		d.AddOption(&ouc)
+		return d
+	}
+}
\ No newline at end of file
-- 
cgit v1.2.3