X-Git-Url: https://git.decadent.org.uk/gitweb/?p=odhcp6c.git;a=blobdiff_plain;f=src%2Fdhcpv6.c;h=4b1ba957b21fcbc3d18c0030a0bc2acb915ec07b;hp=7416f6103eac1a9cbc77023e4447231cf70c062d;hb=c291def1cadf58c0aa10c18a53c2fc4d2dee1ad6;hpb=bbcc9cfa44372f58cb33d556d9b7c57f6ee96b61

diff --git a/src/dhcpv6.c b/src/dhcpv6.c
index 7416f61..4b1ba95 100644
--- a/src/dhcpv6.c
+++ b/src/dhcpv6.c
@@ -1,5 +1,5 @@
 /**
- * Copyright (C) 2012-2013 Steven Barth <steven@midlink.org>
+ * Copyright (C) 2012-2014 Steven Barth <steven@midlink.org>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License v2 as published by
@@ -105,10 +105,13 @@ static bool accept_reconfig = false;
 // Reconfigure key
 static uint8_t reconf_key[16];
 
+// client options
+static unsigned int client_options = 0;
 
-int init_dhcpv6(const char *ifname, int request_pd, bool strict_options, int sol_timeout)
+
+int init_dhcpv6(const char *ifname, unsigned int options, int sol_timeout)
 {
-	request_prefix = request_pd;
+	client_options = options;
 	dhcpv6_retx[DHCPV6_MSG_SOLICIT].max_timeo = sol_timeout;
 
 	sock = socket(AF_INET6, SOCK_DGRAM | SOCK_CLOEXEC, IPPROTO_UDP);
@@ -158,7 +161,7 @@ int init_dhcpv6(const char *ifname, int request_pd, bool strict_options, int sol
 	}
 
 	// Create ORO
-	if (!strict_options) {
+	if (!(client_options & DHCPV6_STRICT_OPTIONS)) {
 		uint16_t oro[] = {
 			htons(DHCPV6_OPT_SIP_SERVER_D),
 			htons(DHCPV6_OPT_SIP_SERVER_A),
@@ -201,7 +204,6 @@ void dhcpv6_set_ia_mode(enum odhcp6c_ia_mode na, enum odhcp6c_ia_mode pd)
 	pd_mode = pd;
 }
 
-
 static void dhcpv6_send(enum dhcpv6_msg type, uint8_t trid[3], uint32_t ecs)
 {
 	// Build FQDN
@@ -230,62 +232,112 @@ static void dhcpv6_send(enum dhcpv6_msg type, uint8_t trid[3], uint32_t ecs)
 
 	// Build IA_PDs
 	size_t ia_pd_entries, ia_pd_len = 0;
-	struct odhcp6c_entry *e = odhcp6c_get_state(STATE_IA_PD, &ia_pd_entries);
-	ia_pd_entries /= sizeof(*e);
-	struct dhcpv6_ia_hdr hdr_ia_pd = {
-		htons(DHCPV6_OPT_IA_PD),
-		htons(sizeof(hdr_ia_pd) - 4),
-		1, 0, 0
-	};
+	uint8_t *ia_pd;
 
+	if (type == DHCPV6_MSG_SOLICIT) {
+		odhcp6c_clear_state(STATE_IA_PD);
+		size_t n_prefixes;
+		struct odhcp6c_request_prefix *request_prefixes = odhcp6c_get_state(STATE_IA_PD_INIT, &n_prefixes);
+		n_prefixes /= sizeof(struct odhcp6c_request_prefix);
+
+		ia_pd = alloca(n_prefixes * (sizeof(struct dhcpv6_ia_hdr) + sizeof(struct dhcpv6_ia_prefix)));
+
+		for (size_t i = 0; i < n_prefixes; i++) {
+			struct dhcpv6_ia_hdr hdr_ia_pd = {
+				htons(DHCPV6_OPT_IA_PD),
+				htons(sizeof(hdr_ia_pd) - 4 + sizeof(struct dhcpv6_ia_prefix)),
+				request_prefixes[i].iaid, 0, 0
+			};
+			struct dhcpv6_ia_prefix pref = {
+				.type = htons(DHCPV6_OPT_IA_PREFIX),
+				.len = htons(25), .prefix = request_prefixes[i].length
+			};
+			memcpy(ia_pd + ia_pd_len, &hdr_ia_pd, sizeof(hdr_ia_pd));
+			ia_pd_len += sizeof(hdr_ia_pd);
+			memcpy(ia_pd + ia_pd_len, &pref, sizeof(pref));
+			ia_pd_len += sizeof(pref);
+		}
+	} else {
+		struct odhcp6c_entry *e = odhcp6c_get_state(STATE_IA_PD, &ia_pd_entries);
+		ia_pd_entries /= sizeof(*e);
+
+		// we're too lazy to count our distinct IAIDs,
+		// so just allocate maximally needed space
+		ia_pd = alloca(ia_pd_entries * (sizeof(struct dhcpv6_ia_prefix) + 10 +
+					sizeof(struct dhcpv6_ia_hdr)));
+
+		for (size_t i = 0; i < ia_pd_entries; ++i) {
+			uint32_t iaid = e[i].iaid;
+
+			// check if this is an unprocessed IAID and skip if not.
+			int new_iaid = 1;
+			for (int j = i-1; j >= 0; j--) {
+				if (e[j].iaid == iaid) {
+					new_iaid = 0;
+					break;
+				}
+			}
 
-	uint8_t *ia_pd = alloca(ia_pd_entries * (sizeof(struct dhcpv6_ia_prefix) + 10));
-	for (size_t i = 0; i < ia_pd_entries; ++i) {
-		uint8_t ex_len = 0;
-		if (e[i].priority > 0)
-			ex_len = ((e[i].priority - e[i].length - 1) / 8) + 6;
+			if (!new_iaid)
+				continue;
 
-		struct dhcpv6_ia_prefix p = {
-			.type = htons(DHCPV6_OPT_IA_PREFIX),
-			.len = htons(sizeof(p) - 4U + ex_len),
-			.prefix = e[i].length,
-			.addr = e[i].target
-		};
+			// construct header
+			struct dhcpv6_ia_hdr hdr_ia_pd = {
+				htons(DHCPV6_OPT_IA_PD),
+				htons(sizeof(hdr_ia_pd) - 4),
+				iaid, 0, 0
+			};
 
-		memcpy(ia_pd + ia_pd_len, &p, sizeof(p));
-		ia_pd_len += sizeof(p);
+			memcpy(ia_pd + ia_pd_len, &hdr_ia_pd, sizeof(hdr_ia_pd));
+			struct dhcpv6_ia_hdr *hdr = (struct dhcpv6_ia_hdr *) (ia_pd + ia_pd_len);
+			ia_pd_len += sizeof(hdr_ia_pd);
 
-		if (ex_len) {
-			ia_pd[ia_pd_len++] = 0;
-			ia_pd[ia_pd_len++] = DHCPV6_OPT_PD_EXCLUDE;
-			ia_pd[ia_pd_len++] = 0;
-			ia_pd[ia_pd_len++] = ex_len - 4;
-			ia_pd[ia_pd_len++] = e[i].priority;
+			for (size_t j = i; j < ia_pd_entries; j++) {
+				if (e[j].iaid != iaid)
+					continue;
 
-			uint32_t excl = ntohl(e[i].router.s6_addr32[1]);
-			excl >>= (64 - e[i].priority);
-			excl <<= 8 - ((e[i].priority - e[i].length) % 8);
+				uint8_t ex_len = 0;
+				if (e[j].priority > 0)
+					ex_len = ((e[j].priority - e[j].length - 1) / 8) + 6;
+
+				struct dhcpv6_ia_prefix p = {
+					.type = htons(DHCPV6_OPT_IA_PREFIX),
+					.len = htons(sizeof(p) - 4U + ex_len),
+					.prefix = e[j].length,
+					.addr = e[j].target
+				};
+
+				memcpy(ia_pd + ia_pd_len, &p, sizeof(p));
+				ia_pd_len += sizeof(p);
+
+				if (ex_len) {
+					ia_pd[ia_pd_len++] = 0;
+					ia_pd[ia_pd_len++] = DHCPV6_OPT_PD_EXCLUDE;
+					ia_pd[ia_pd_len++] = 0;
+					ia_pd[ia_pd_len++] = ex_len - 4;
+					ia_pd[ia_pd_len++] = e[j].priority;
+
+					uint32_t excl = ntohl(e[j].router.s6_addr32[1]);
+					excl >>= (64 - e[j].priority);
+					excl <<= 8 - ((e[j].priority - e[j].length) % 8);
+
+					for (size_t i = ex_len - 5; i > 0; --i, excl >>= 8)
+						ia_pd[ia_pd_len + i] = excl & 0xff;
+					ia_pd_len += ex_len - 5;
+				}
 
-			for (size_t i = ex_len - 5; i > 0; --i, excl >>= 8)
-				ia_pd[ia_pd_len + i] = excl & 0xff;
-			ia_pd_len += ex_len - 5;
+				hdr->len = htons(ntohs(hdr->len) + ntohs(p.len) + 4U);
+			}
 		}
 	}
 
-	struct dhcpv6_ia_prefix pref = {
-		.type = htons(DHCPV6_OPT_IA_PREFIX),
-		.len = htons(25), .prefix = request_prefix
-	};
-	if (request_prefix > 0 && ia_pd_len == 0 && type == DHCPV6_MSG_SOLICIT) {
-		ia_pd = (uint8_t*)&pref;
-		ia_pd_len = sizeof(pref);
-	}
-	hdr_ia_pd.len = htons(ntohs(hdr_ia_pd.len) + ia_pd_len);
+	if (ia_pd_entries > 0)
+		request_prefix = 1;
 
 	// Build IA_NAs
 	size_t ia_na_entries, ia_na_len = 0;
 	void *ia_na = NULL;
-	e = odhcp6c_get_state(STATE_IA_NA, &ia_na_entries);
+	struct odhcp6c_entry *e = odhcp6c_get_state(STATE_IA_NA, &ia_na_entries);
 	ia_na_entries /= sizeof(*e);
 
 	struct dhcpv6_ia_hdr hdr_ia_na = {
@@ -316,6 +368,21 @@ static void dhcpv6_send(enum dhcpv6_msg type, uint8_t trid[3], uint32_t ecs)
 	// Request Information Refresh
 	uint16_t oro_refresh = htons(DHCPV6_OPT_INFO_REFRESH);
 
+	// Build vendor-class option
+	size_t vendor_class_len, user_class_len;
+	struct dhcpv6_vendorclass *vendor_class = odhcp6c_get_state(STATE_VENDORCLASS, &vendor_class_len);
+	void *user_class = odhcp6c_get_state(STATE_USERCLASS, &user_class_len);
+
+	struct {
+		uint16_t type;
+		uint16_t length;
+	} vendor_class_hdr = {htons(DHCPV6_OPT_VENDOR_CLASS), htons(vendor_class_len)};
+
+	struct {
+		uint16_t type;
+		uint16_t length;
+	} user_class_hdr = {htons(DHCPV6_OPT_USER_CLASS), htons(user_class_len)};
+
 	// Prepare Header
 	size_t oro_len;
 	void *oro = odhcp6c_get_state(STATE_ORO, &oro_len);
@@ -340,35 +407,42 @@ static void dhcpv6_send(enum dhcpv6_msg type, uint8_t trid[3], uint32_t ecs)
 		{&oro_refresh, 0},
 		{cl_id, cl_id_len},
 		{srv_id, srv_id_len},
+		{&vendor_class_hdr, vendor_class_len ? sizeof(vendor_class_hdr) : 0},
+		{vendor_class, vendor_class_len},
+		{&user_class_hdr, user_class_len ? sizeof(user_class_hdr) : 0},
+		{user_class, user_class_len},
 		{&reconf_accept, sizeof(reconf_accept)},
 		{&fqdn, fqdn_len},
 		{&hdr_ia_na, sizeof(hdr_ia_na)},
 		{ia_na, ia_na_len},
-		{&hdr_ia_pd, sizeof(hdr_ia_pd)},
 		{ia_pd, ia_pd_len},
 	};
 
 	size_t cnt = ARRAY_SIZE(iov);
 	if (type == DHCPV6_MSG_INFO_REQ) {
-		cnt = 5;
+		cnt = 9;
 		iov[2].iov_len = sizeof(oro_refresh);
 		hdr.oro_len = htons(oro_len + sizeof(oro_refresh));
 	} else if (!request_prefix) {
-		cnt = 9;
+		cnt = 13;
 	}
 
 	// Disable IAs if not used
 	if (type != DHCPV6_MSG_SOLICIT) {
-		iov[5].iov_len = 0;
+		iov[7].iov_len = 0;
 		if (ia_na_len == 0)
-			iov[7].iov_len = 0;
-		if (ia_pd_len == 0)
 			iov[9].iov_len = 0;
 	}
 
 	if (na_mode == IA_MODE_NONE)
+		iov[9].iov_len = 0;
+
+	if (!(client_options & DHCPV6_ACCEPT_RECONFIGURE))
 		iov[7].iov_len = 0;
 
+	if (!(client_options & DHCPV6_CLIENT_FQDN))
+		iov[8].iov_len = 0;
+
 	struct sockaddr_in6 srv = {AF_INET6, htons(DHCPV6_SERVER_PORT),
 		0, ALL_DHCPV6_RELAYS, ifindex};
 	struct msghdr msg = {&srv, sizeof(srv), iov, cnt, NULL, 0, 0};
@@ -400,9 +474,9 @@ int dhcpv6_request(enum dhcpv6_msg type)
 	if (type == DHCPV6_MSG_UNKNOWN)
 		timeout = t1;
 	else if (type == DHCPV6_MSG_RENEW)
-		timeout = (t2 > t1) ? t2 - t1 : 0;
+		timeout = (t2 > t1) ? t2 - t1 : ((t1 == UINT32_MAX) ? UINT32_MAX : 0);
 	else if (type == DHCPV6_MSG_REBIND)
-		timeout = (t3 > t2) ? t3 - t2 : 0;
+		timeout = (t3 > t2) ? t3 - t2 : ((t2 == UINT32_MAX) ? UINT32_MAX : 0);
 
 	if (timeout == 0)
 		return -1;
@@ -440,8 +514,8 @@ int dhcpv6_request(enum dhcpv6_msg type)
 		uint64_t round_end = round_start + rto;
 		elapsed = round_start - start;
 
-		// Don't wait too long
-		if (round_end - start > timeout * 1000)
+		// Don't wait too long if timeout differs from infinite
+		if ((timeout != UINT32_MAX) && (round_end - start > timeout * 1000))
 			round_end = timeout * 1000 + start;
 
 		// Built and send package
@@ -468,9 +542,9 @@ int dhcpv6_request(enum dhcpv6_msg type)
 
 			// Set timeout for receiving
 			uint64_t t = round_end - round_start;
-			struct timeval timeout = {t / 1000, (t % 1000) * 1000};
+			struct timeval tv = {t / 1000, (t % 1000) * 1000};
 			setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO,
-					&timeout, sizeof(timeout));
+					&tv, sizeof(tv));
 
 			// Receive cycle
 			len = recvmsg(sock, &msg, 0);
@@ -515,8 +589,8 @@ int dhcpv6_request(enum dhcpv6_msg type)
 		// Allow
 		if (retx->handler_finish)
 			len = retx->handler_finish();
-	} while (len < 0 && ((elapsed / 1000 < timeout) && (!retx->max_rc || rc < retx->max_rc)));
-
+	} while (len < 0 && ((timeout == UINT32_MAX) || (elapsed / 1000 < timeout)) && 
+			(!retx->max_rc || rc < retx->max_rc));
 	return len;
 }
 
@@ -787,9 +861,14 @@ static int dhcpv6_handle_reply(enum dhcpv6_msg orig, _unused const int rc,
 		uint32_t elapsed = (last_update > 0) ? now - last_update : 0;
 		last_update = now;
 
-		t1 -= elapsed;
-		t2 -= elapsed;
-		t3 -= elapsed;
+		if (t1 != UINT32_MAX)
+			t1 -= elapsed;
+
+		if (t2 != UINT32_MAX)
+			t2 -= elapsed;
+
+		if (t3 != UINT32_MAX)
+			t3 -= elapsed;
 
 		if (t1 < 0)
 			t1 = 0;
@@ -825,7 +904,7 @@ static int dhcpv6_handle_reply(enum dhcpv6_msg orig, _unused const int rc,
 			struct dhcpv6_ia_hdr *ia_hdr = (void*)(&odata[-4]);
 
 			// Test ID
-			if (ia_hdr->iaid != 1)
+			if (ia_hdr->iaid != 1 && otype == DHCPV6_OPT_IA_NA)
 				continue;
 
 			uint16_t code = DHCPV6_Success;
@@ -975,7 +1054,9 @@ static int dhcpv6_parse_ia(void *opt, void *end)
 	// Update address IA
 	dhcpv6_for_each_option(&ia_hdr[1], end, otype, olen, odata) {
 		struct odhcp6c_entry entry = {IN6ADDR_ANY_INIT, 0, 0,
-				IN6ADDR_ANY_INIT, 0, 0, 0, 0, 0};
+				IN6ADDR_ANY_INIT, 0, 0, 0, 0, 0, 0};
+
+		entry.iaid = ia_hdr->iaid;
 
 		if (otype == DHCPV6_OPT_IA_PREFIX) {
 			struct dhcpv6_ia_prefix *prefix = (void*)&odata[-4];