Changes in uspace/srv/net/inetsrv/icmp.c [9749e47:13be2583] in mainline

File:

• uspace/srv/net/inetsrv/icmp.c (modified) (4 diffs)

uspace/srv/net/inetsrv/icmp.c

@@ -41 +41 @@
 #include <stdlib.h>
 #include <types/inetping.h>
+#include <net/socket_codes.h>
 #include "icmp.h"
 #include "icmp_std.h"
@@ -122 +123 @@
 {
         log_msg(LOG_DEFAULT, LVL_DEBUG, "icmp_recv_echo_reply()");
-
+        
         if (dgram->size < sizeof(icmp_echo_t))
                 return EINVAL;
-
+        
         icmp_echo_t *reply = (icmp_echo_t *) dgram->data;
-
+        
         inetping_sdu_t sdu;
-
-        sdu.src = dgram->src;
-        sdu.dest = dgram->dest;
+        
+        uint16_t family = inet_addr_get(&dgram->src, &sdu.src, NULL);
+        if (family != AF_INET)
+                return EINVAL;
+        
+        family = inet_addr_get(&dgram->dest, &sdu.dest, NULL);
+        if (family != AF_INET)
+                return EINVAL;
+        
         sdu.seq_no = uint16_t_be2host(reply->seq_no);
         sdu.data = reply + sizeof(icmp_echo_t);
         sdu.size = dgram->size - sizeof(icmp_echo_t);
-
+        
         uint16_t ident = uint16_t_be2host(reply->ident);
 
@@ -147 +154 @@
         if (rdata == NULL)
                 return ENOMEM;
-
+        
         icmp_echo_t *request = (icmp_echo_t *) rdata;
-
+        
         request->type = ICMP_ECHO_REQUEST;
         request->code = 0;
@@ -155 +162 @@
         request->ident = host2uint16_t_be(ident);
         request->seq_no = host2uint16_t_be(sdu->seq_no);
-
+        
         memcpy(rdata + sizeof(icmp_echo_t), sdu->data, sdu->size);
-
+        
         uint16_t checksum = inet_checksum_calc(INET_CHECKSUM_INIT, rdata, rsize);
         request->checksum = host2uint16_t_be(checksum);
-
+        
         inet_dgram_t dgram;
-
-        dgram.src = sdu->src;
-        dgram.dest = sdu->dest;
+        
+        inet_addr_set(sdu->src, &dgram.src);
+        inet_addr_set(sdu->dest, &dgram.dest);
+        
         dgram.iplink = 0;
         dgram.tos = ICMP_TOS;
         dgram.data = rdata;
         dgram.size = rsize;
-
+        
         int rc = inet_route_packet(&dgram, IP_PROTO_ICMP, INET_TTL_MAX, 0);
-
+        
         free(rdata);
         return rc;