1 files changed, 97 insertions, 0 deletions
diff --git a/src/test.h b/src/test.h
new file mode 100644
index 0000000000..92315858ba
--- /dev/null
+++ b/src/test.h
@@ -0,0 +1,97 @@
+#pragma once
+
+/*
+ * Test Helpers
+ * Bunch of helpers to setup the environment for networking tests. This
+ * includes net-namespace setups, veth setups, and more.
+ */
+
+#include <assert.h>
+#include <endian.h>
+#include <errno.h>
+#include <net/ethernet.h>
+#include <net/if.h>
+#include <netinet/in.h>
+#include <poll.h>
+#include <sched.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+#include "n-acd.h"
+
+static inline void test_if_query(const char *name, int *indexp, struct ether_addr *macp) {
+        struct ifreq ifr = {};
+        size_t l;
+        int r, s;
+
+        l = strlen(name);
+        assert(l <= IF_NAMESIZE);
+
+        if (indexp) {
+                *indexp = if_nametoindex(name);
+                assert(*indexp > 0);
+        }
+
+        if (macp) {
+                s = socket(AF_INET, SOCK_DGRAM, 0);
+                assert(s >= 0);
+
+                strncpy(ifr.ifr_name, name, l);
+                r = ioctl(s, SIOCGIFHWADDR, &ifr);
+                assert(r >= 0);
+
+                memcpy(macp->ether_addr_octet, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
+
+                close(s);
+        }
+}
+
+static inline void test_veth_cmd(int ifindex, const char *cmd) {
+        char *p, name[IF_NAMESIZE + 1] = {};
+        int r;
+
+        p = if_indextoname(ifindex, name);
+        assert(p);
+
+        r = asprintf(&p, "ip link set %s %s", name, cmd);
+        assert(r >= 0);
+
+        /* Again: Ewwww... */
+        r = system(p);
+        assert(r == 0);
+
+        free(p);
+}
+
+static inline void test_veth_new(int *parent_indexp,
+                                 struct ether_addr *parent_macp,
+                                 int *child_indexp,
+                                 struct ether_addr *child_macp) {
+        int r;
+
+        /* Eww... but it works. */
+        r = system("ip link add type veth");
+        assert(r == 0);
+        r = system("ip link set veth0 up");
+        assert(r == 0);
+        r = system("ip link set veth1 up");
+        assert(r == 0);
+
+        test_if_query("veth0", parent_indexp, parent_macp);
+        test_if_query("veth1", child_indexp, child_macp);
+}
+
+static inline int test_setup(void) {
+        int r;
+
+        r = unshare(CLONE_NEWNET);
+        if (r < 0) {
+                assert(errno == EPERM);
+                return 77;
+        }
+
+        return 0;
+}