selftests/net: replace manual array size calc with ARRAYSIZE macro.

[platform/kernel/linux-rpi.git] / tools / testing / selftests / net / csum.c

diff --git a/tools/testing/selftests/net/csum.c b/tools/testing/selftests/net/csum.c
index 82a1c18..90eb06f 100644 (file)
--- a/tools/testing/selftests/net/csum.c
+++ b/tools/testing/selftests/net/csum.c
@@ -91,6 +91,8 @@
 #include <sys/types.h>
 #include <unistd.h>
 
+#include "kselftest.h"
+
 static bool cfg_bad_csum;
 static int cfg_family = PF_INET6;
 static int cfg_num_pkt = 4;
@@ -450,7 +452,7 @@ static void send_packet(int fd, const char *buf, int len)
        iov[2].iov_len = len;
 
        msg.msg_iov = iov;
-       msg.msg_iovlen = sizeof(iov) / sizeof(iov[0]);
+       msg.msg_iovlen = ARRAY_SIZE(iov);
 
        msg.msg_name = &addr;
        msg.msg_namelen = sizeof(addr);
@@ -505,7 +507,7 @@ static void __recv_prepare_packet_filter(int fd, int off_nexthdr, int off_dport)
        struct sock_fprog prog = {};
 
        prog.filter = filter;
-       prog.len = sizeof(filter) / sizeof(struct sock_filter);
+       prog.len = ARRAY_SIZE(filter);
        if (setsockopt(fd, SOL_SOCKET, SO_ATTACH_FILTER, &prog, sizeof(prog)))
                error(1, errno, "setsockopt filter");
 }