cmdline: check size of result buffer to avoid overflow

[libcmdline.git] / src / lib / cmdline_parse_etheraddr.c

diff --git a/src/lib/cmdline_parse_etheraddr.c b/src/lib/cmdline_parse_etheraddr.c
index cc2a4f4..7a50e02 100644 (file)
--- a/src/lib/cmdline_parse_etheraddr.c
+++ b/src/lib/cmdline_parse_etheraddr.c
@@ -65,11 +65,7 @@
 #include <ctype.h>
 #include <string.h>
 #include <sys/types.h>
-#include <sys/socket.h>
 #include <net/ethernet.h>
-#ifdef __linux__
-#include <netinet/ether.h>
-#endif
 
 #include "cmdline_parse.h"
 #include "cmdline_parse_etheraddr.h"
@@ -84,28 +80,52 @@ struct cmdline_token_ops cmdline_token_etheraddr_ops = {
 
 #define ETHER_ADDRSTRLEN 18
 
+#ifdef __linux__
+#define ea_oct ether_addr_octet
+#else
+#define ea_oct octet
+#endif
+
+static struct ether_addr *
+my_ether_aton(const char *a)
+{
+       int i;
+       static struct ether_addr ether_addr;
+       unsigned int o0, o1, o2, o3, o4, o5;
+
+       i = sscanf(a, "%x:%x:%x:%x:%x:%x", &o0, &o1, &o2, &o3, &o4, &o5);
+
+       if (i != ETHER_ADDR_LEN)
+               return NULL;
+
+       ether_addr.ea_oct[0] = (uint8_t)o0;
+       ether_addr.ea_oct[1] = (uint8_t)o1;
+       ether_addr.ea_oct[2] = (uint8_t)o2;
+       ether_addr.ea_oct[3] = (uint8_t)o3;
+       ether_addr.ea_oct[4] = (uint8_t)o4;
+       ether_addr.ea_oct[5] = (uint8_t)o5;
+
+       return (struct ether_addr *)&ether_addr;
+}
+
 int
 cmdline_parse_etheraddr(__attribute__((unused)) cmdline_parse_token_hdr_t *tk,
-                       const char *buf, void *res)
+                       const char *buf, void *res, unsigned ressize)
 {
        unsigned int token_len = 0;
        char ether_str[ETHER_ADDRSTRLEN];
        struct ether_addr *etheraddr = res;
        struct ether_addr *tmp;
 
-       if (! *buf)
+       if (res && ressize < sizeof(struct ether_addr))
                return -1;
 
-       while (!cmdline_isendoftoken(buf[token_len]))
-               token_len++;
-
        /* if token is too big... */
-       if (token_len >= ETHER_ADDRSTRLEN)
+       token_len = snprintf(ether_str, sizeof(ether_str), "%s", buf);
+       if (token_len >= sizeof(ether_str))
                return -1;
 
-       snprintf(ether_str, token_len+1, "%s", buf);
-
-       tmp = ether_aton(ether_str);
+       tmp = my_ether_aton(ether_str);
        if (tmp == NULL)
                return -1;