1 files changed, 41 insertions, 32 deletions

diff --git a/networking/udhcp/packet.c b/networking/udhcp/packet.c
index 22c18deaf..443fea6dd 100644
--- a/networking/udhcp/packet.c
+++ b/networking/udhcp/packet.c
@@ -41,20 +41,14 @@ void udhcp_init_header(struct dhcpMessage *packet, char type)
 /* read a packet from socket fd, return -1 on read error, -2 on packet error */
 int udhcp_recv_packet(struct dhcpMessage *packet, int fd)
 {
-#if 0
-        static const char broken_vendors[][8] = {
-                "MSFT 98",
-                ""
-        };
-#endif
         int bytes;
         unsigned char *vendor;
 
         memset(packet, 0, sizeof(*packet));
-        bytes = read(fd, packet, sizeof(*packet));
+        bytes = safe_read(fd, packet, sizeof(*packet));
         if (bytes < 0) {
                 DEBUG("cannot read on listening socket, ignoring");
-                return -1;
+                return bytes; /* returns -1 */
         }
 
         if (packet->cookie != htonl(DHCP_MAGIC)) {
@@ -67,6 +61,10 @@ int udhcp_recv_packet(struct dhcpMessage *packet, int fd)
                 vendor = get_option(packet, DHCP_VENDOR);
                 if (vendor) {
 #if 0
+                        static const char broken_vendors[][8] = {
+                                "MSFT 98",
+                                ""
+                        };
                         int i;
                         for (i = 0; broken_vendors[i][0]; i++) {
                                 if (vendor[OPT_LEN - 2] == (uint8_t)strlen(broken_vendors[i])
@@ -127,10 +125,11 @@ int udhcp_send_raw_packet(struct dhcpMessage *payload,
                 uint32_t source_ip, int source_port,
                 uint32_t dest_ip, int dest_port, const uint8_t *dest_arp, int ifindex)
 {
-        int fd;
-        int result;
         struct sockaddr_ll dest;
         struct udp_dhcp_packet packet;
+        int fd;
+        int result = -1;
+        const char *msg;
 
         enum {
                 IP_UPD_DHCP_SIZE = sizeof(struct udp_dhcp_packet) - CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS,
@@ -139,8 +138,8 @@ int udhcp_send_raw_packet(struct dhcpMessage *payload,
 
         fd = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_IP));
         if (fd < 0) {
-                bb_perror_msg("socket");
-                return -1;
+                msg = "socket(%s)";
+                goto ret_msg;
         }
 
         memset(&dest, 0, sizeof(dest));
@@ -152,10 +151,9 @@ int udhcp_send_raw_packet(struct dhcpMessage *payload,
         dest.sll_ifindex = ifindex;
         dest.sll_halen = 6;
         memcpy(dest.sll_addr, dest_arp, 6);
-        if (bind(fd, (struct sockaddr *)&dest, sizeof(struct sockaddr_ll)) < 0) {
-                bb_perror_msg("bind");
-                close(fd);
-                return -1;
+        if (bind(fd, (struct sockaddr *)&dest, sizeof(dest)) < 0) {
+                msg = "bind(%s)";
+                goto ret_close;
         }
 
         packet.ip.protocol = IPPROTO_UDP;
@@ -179,11 +177,15 @@ int udhcp_send_raw_packet(struct dhcpMessage *payload,
          * 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, (struct sockaddr *) &dest, sizeof(dest));
-        if (result <= 0) {
-                bb_perror_msg("sendto");
-        }
+        result = sendto(fd, &packet, IP_UPD_DHCP_SIZE, 0,
+                                (struct sockaddr *) &dest, sizeof(dest));
+        msg = "sendto";
+ ret_close:
         close(fd);
+        if (result < 0) {
+ ret_msg:
+                bb_perror_msg(msg, "PACKET");
+        }
         return result;
 }
 
@@ -193,41 +195,48 @@ int udhcp_send_kernel_packet(struct dhcpMessage *payload,
                 uint32_t source_ip, int source_port,
                 uint32_t dest_ip, int dest_port)
 {
-        int fd, result;
         struct sockaddr_in client;
+        int fd;
+        int result = -1;
+        const char *msg;
 
         enum {
                 DHCP_SIZE = sizeof(struct dhcpMessage) - CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS,
         };
 
         fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
-        if (fd < 0)
-                return -1;
-
+        if (fd < 0) {
+                msg = "socket(%s)";
+                goto ret_msg;
+        }
         setsockopt_reuseaddr(fd);
 
         memset(&client, 0, sizeof(client));
         client.sin_family = AF_INET;
         client.sin_port = htons(source_port);
         client.sin_addr.s_addr = source_ip;
-
         if (bind(fd, (struct sockaddr *)&client, sizeof(client)) == -1) {
-                close(fd);
-                return -1;
+                msg = "bind(%s)";
+                goto ret_close;
         }
 
         memset(&client, 0, sizeof(client));
         client.sin_family = AF_INET;
         client.sin_port = htons(dest_port);
         client.sin_addr.s_addr = dest_ip;
-
-        if (connect(fd, (struct sockaddr *)&client, sizeof(struct sockaddr)) == -1) {
-                close(fd);
-                return -1;
+        if (connect(fd, (struct sockaddr *)&client, sizeof(client)) == -1) {
+                msg = "connect";
+                goto ret_close;
         }
 
         /* Currently we send full-sized DHCP packets (see above) */
-        result = write(fd, payload, DHCP_SIZE);
+        result = safe_write(fd, payload, DHCP_SIZE);
+        msg = "write";
+ ret_close:
         close(fd);
+        if (result < 0) {
+ ret_msg:
+                bb_perror_msg(msg, "UDP");
+        }
         return result;
 }