X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=src%2Fdhcpv6.c;h=785b57b03aedf1543dea2ff08a08594599873ceb;hb=8e50917e51f7f21c53f9021063670065c8fa06d8;hp=3f5bc83094d6589ce2dc509d77f0989ab0ac73c2;hpb=7a0a5a7a6378b958f1dde4841504401d9bc0273c;p=odhcp6c.git

diff --git a/src/dhcpv6.c b/src/dhcpv6.c
index 3f5bc83..785b57b 100644
--- a/src/dhcpv6.c
+++ b/src/dhcpv6.c
@@ -142,8 +142,12 @@ int init_dhcpv6(const char *ifname, int request_pd)
 			htons(DHCPV6_OPT_DNS_SERVERS),
 			htons(DHCPV6_OPT_DNS_DOMAIN),
 			htons(DHCPV6_OPT_NTP_SERVER),
+			htons(DHCPV6_OPT_SIP_SERVER_A),
 			htons(DHCPV6_OPT_AFTR_NAME),
 			htons(DHCPV6_OPT_PD_EXCLUDE),
+#ifdef EXT_PREFIX_CLASS
+			htons(DHCPV6_OPT_PREFIX_CLASS),
+#endif
 	};
 	odhcp6c_add_state(STATE_ORO, oro, sizeof(oro));
 
@@ -827,11 +831,11 @@ static uint32_t dhcpv6_parse_ia(void *opt, void *end)
 	uint16_t otype, olen;
 	uint8_t *odata;
 
-	struct odhcp6c_entry entry = {IN6ADDR_ANY_INIT,
-			0, 0, IN6ADDR_ANY_INIT, 0, 0};
-
 	// Update address IA
 	dhcpv6_for_each_option(opt, end, otype, olen, odata) {
+		struct odhcp6c_entry entry = {IN6ADDR_ANY_INIT, 0, 0,
+				IN6ADDR_ANY_INIT, 0, 0, 0};
+
 		if (otype == DHCPV6_OPT_IA_PREFIX) {
 			struct dhcpv6_ia_prefix *prefix = (void*)&odata[-4];
 			if (olen + 4U < sizeof(*prefix))
@@ -845,11 +849,19 @@ static uint32_t dhcpv6_parse_ia(void *opt, void *end)
 
 			entry.length = prefix->prefix;
 			entry.target = prefix->addr;
+			uint16_t stype, slen;
+			uint8_t *sdata;
+
+#ifdef EXT_PREFIX_CLASS
+                        // Find prefix class, if any
+			dhcpv6_for_each_option(&prefix[1], odata + olen,
+					stype, slen, sdata)
+				if (stype == DHCPV6_OPT_PREFIX_CLASS && slen == 2)
+					entry.class = sdata[0] << 8 | sdata[1];
+#endif
 
 			// Parse PD-exclude
 			bool ok = true;
-			uint16_t stype, slen;
-			uint8_t *sdata;
 			dhcpv6_for_each_option(odata + sizeof(*prefix) - 4U,
 					odata + olen, stype, slen, sdata) {
 				if (stype != DHCPV6_OPT_PD_EXCLUDE || slen < 2)
@@ -904,9 +916,18 @@ static uint32_t dhcpv6_parse_ia(void *opt, void *end)
 			entry.length = 128;
 			entry.target = addr->addr;
 
+#ifdef EXT_PREFIX_CLASS
+			uint16_t stype, slen;
+			uint8_t *sdata;
+			// Find prefix class, if any
+			dhcpv6_for_each_option(&addr[1], odata + olen,
+					stype, slen, sdata)
+				if (stype == DHCPV6_OPT_PREFIX_CLASS && slen == 2)
+					entry.class = sdata[0] << 8 | sdata[1];
+#endif
+
 			odhcp6c_update_entry(STATE_IA_NA, &entry);
 		}
-
 		if (entry.valid > 0 && timeout > entry.valid)
 			timeout = entry.valid;
 	}