76 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Diff
		
	
	
	
	
	
			
		
		
	
	
			76 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Diff
		
	
	
	
	
	
--- a/networking/udhcp/packet.c
 | 
						|
+++ b/networking/udhcp/packet.c
 | 
						|
@@ -114,6 +114,10 @@ uint16_t FAST_FUNC udhcp_checksum(void *
 | 
						|
 	return ~sum;
 | 
						|
 }
 | 
						|
 
 | 
						|
+int udhcp_get_payload_len(struct dhcpMessage *payload)
 | 
						|
+{
 | 
						|
+	return sizeof(struct dhcpMessage) - DHCP_OPTIONS_BUFSIZE + end_option(payload->options) + sizeof(payload->options[0]);
 | 
						|
+}
 | 
						|
 
 | 
						|
 /* Construct a ip/udp header for a packet, send packet */
 | 
						|
 int FAST_FUNC udhcp_send_raw_packet(struct dhcpMessage *payload,
 | 
						|
@@ -126,11 +130,7 @@ int FAST_FUNC udhcp_send_raw_packet(stru
 | 
						|
 	int fd;
 | 
						|
 	int result = -1;
 | 
						|
 	const char *msg;
 | 
						|
-
 | 
						|
-	enum {
 | 
						|
-		IP_UPD_DHCP_SIZE = sizeof(struct udp_dhcp_packet) - CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS,
 | 
						|
-		UPD_DHCP_SIZE    = IP_UPD_DHCP_SIZE - offsetof(struct udp_dhcp_packet, udp),
 | 
						|
-	};
 | 
						|
+	int p_len = udhcp_get_payload_len(payload);
 | 
						|
 
 | 
						|
 	fd = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_IP));
 | 
						|
 	if (fd < 0) {
 | 
						|
@@ -140,7 +140,7 @@ int FAST_FUNC udhcp_send_raw_packet(stru
 | 
						|
 
 | 
						|
 	memset(&dest, 0, sizeof(dest));
 | 
						|
 	memset(&packet, 0, sizeof(packet));
 | 
						|
-	packet.data = *payload; /* struct copy */
 | 
						|
+	memcpy(&(packet.data), payload, p_len);
 | 
						|
 
 | 
						|
 	dest.sll_family = AF_PACKET;
 | 
						|
 	dest.sll_protocol = htons(ETH_P_IP);
 | 
						|
@@ -157,23 +157,18 @@ int FAST_FUNC udhcp_send_raw_packet(stru
 | 
						|
 	packet.ip.daddr = dest_ip;
 | 
						|
 	packet.udp.source = htons(source_port);
 | 
						|
 	packet.udp.dest = htons(dest_port);
 | 
						|
-	/* size, excluding IP header: */
 | 
						|
-	packet.udp.len = htons(UPD_DHCP_SIZE);
 | 
						|
-	/* for UDP checksumming, ip.len is set to UDP packet len */
 | 
						|
+	p_len += sizeof(packet.udp);
 | 
						|
+	packet.udp.len = htons(p_len);
 | 
						|
 	packet.ip.tot_len = packet.udp.len;
 | 
						|
-	packet.udp.check = udhcp_checksum(&packet, IP_UPD_DHCP_SIZE);
 | 
						|
-	/* but for sending, it is set to IP packet len */
 | 
						|
-	packet.ip.tot_len = htons(IP_UPD_DHCP_SIZE);
 | 
						|
+	p_len += sizeof(packet.ip);
 | 
						|
+	packet.udp.check = udhcp_checksum(&packet, p_len);
 | 
						|
+	packet.ip.tot_len = htons(p_len);
 | 
						|
 	packet.ip.ihl = sizeof(packet.ip) >> 2;
 | 
						|
 	packet.ip.version = IPVERSION;
 | 
						|
 	packet.ip.ttl = IPDEFTTL;
 | 
						|
 	packet.ip.check = udhcp_checksum(&packet.ip, sizeof(packet.ip));
 | 
						|
 
 | 
						|
-	/* Currently we send full-sized DHCP packets (zero padded).
 | 
						|
-	 * If you need to change this: last byte of the packet is
 | 
						|
-	 * packet.data.options[end_option(packet.data.options)]
 | 
						|
-	 */
 | 
						|
-	result = sendto(fd, &packet, IP_UPD_DHCP_SIZE, 0,
 | 
						|
+	result = sendto(fd, &packet, p_len, 0,
 | 
						|
 				(struct sockaddr *) &dest, sizeof(dest));
 | 
						|
 	msg = "sendto";
 | 
						|
  ret_close:
 | 
						|
@@ -225,8 +220,7 @@ int FAST_FUNC udhcp_send_kernel_packet(s
 | 
						|
 		goto ret_close;
 | 
						|
 	}
 | 
						|
 
 | 
						|
-	/* Currently we send full-sized DHCP packets (see above) */
 | 
						|
-	result = safe_write(fd, payload, DHCP_SIZE);
 | 
						|
+	result = safe_write(fd, payload, udhcp_get_payload_len(payload));
 | 
						|
 	msg = "write";
 | 
						|
  ret_close:
 | 
						|
 	close(fd);
 |