tools/testing/selftests/bpf/test_progs.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #include <stdio.h>
   3 #include <unistd.h>
   4 #include <errno.h>
   5 #include <string.h>
   6 #include <assert.h>
   7 #include <stdlib.h>
   8 #include <stdarg.h>
   9 #include <time.h>
  10 #include <signal.h>
  11
  12 #include <linux/types.h>
  13 typedef __u16 __sum16;
  14 #include <arpa/inet.h>
  15 #include <linux/if_ether.h>
  16 #include <linux/if_packet.h>
  17 #include <linux/ip.h>
  18 #include <linux/ipv6.h>
  19 #include <netinet/tcp.h>
  20 #include <linux/filter.h>
  21 #include <linux/perf_event.h>
  22 #include <linux/socket.h>
  23 #include <linux/unistd.h>
  24
  25 #include <sys/ioctl.h>
  26 #include <sys/wait.h>
  27 #include <sys/types.h>
  28 #include <sys/time.h>
  29 #include <fcntl.h>
  30 #include <pthread.h>
  31 #include <linux/bpf.h>
  32 #include <linux/err.h>
  33 #include <bpf/bpf.h>
  34 #include <bpf/libbpf.h>
  35
  36 #include "test_iptunnel_common.h"
  37 #include "bpf_util.h"
  38 #include <bpf/bpf_endian.h>
  39 #include "trace_helpers.h"
  40 #include "flow_dissector_load.h"
  41
  42 enum verbosity {
  43         VERBOSE_NONE,
  44         VERBOSE_NORMAL,
  45         VERBOSE_VERY,
  46         VERBOSE_SUPER,
  47 };
  48
  49 struct str_set {
  50         const char **strs;
  51         int cnt;
  52 };
  53
  54 struct test_selector {
  55         struct str_set whitelist;
  56         struct str_set blacklist;
  57         bool *num_set;
  58         int num_set_len;
  59 };
  60
  61 struct test_env {
  62         struct test_selector test_selector;
  63         struct test_selector subtest_selector;
  64         bool verifier_stats;
  65         enum verbosity verbosity;
  66
  67         bool jit_enabled;
  68
  69         struct prog_test_def *test;
  70         FILE *stdout;
  71         FILE *stderr;
  72         char *log_buf;
  73         size_t log_cnt;
  74
  75         int succ_cnt; /* successful tests */
  76         int sub_succ_cnt; /* successful sub-tests */
  77         int fail_cnt; /* total failed tests + sub-tests */
  78         int skip_cnt; /* skipped tests */
  79 };
  80
  81 extern struct test_env env;
  82
  83 extern void test__force_log();
  84 extern bool test__start_subtest(const char *name);
  85 extern void test__skip(void);
  86 extern void test__fail(void);
  87 extern int test__join_cgroup(const char *path);
  88
  89 #define MAGIC_BYTES 123
  90
  91 /* ipv4 test vector */
  92 struct ipv4_packet {
  93         struct ethhdr eth;
  94         struct iphdr iph;
  95         struct tcphdr tcp;
  96 } __packed;
  97 extern struct ipv4_packet pkt_v4;
  98
  99 /* ipv6 test vector */
 100 struct ipv6_packet {
 101         struct ethhdr eth;
 102         struct ipv6hdr iph;
 103         struct tcphdr tcp;
 104 } __packed;
 105 extern struct ipv6_packet pkt_v6;
 106
 107 #define _CHECK(condition, tag, duration, format...) ({                  \
 108         int __ret = !!(condition);                                      \
 109         int __save_errno = errno;                                       \
 110         if (__ret) {                                                    \
 111                 test__fail();                                           \
 112                 printf("%s:FAIL:%s ", __func__, tag);                   \
 113                 printf(format);                                         \
 114         } else {                                                        \
 115                 printf("%s:PASS:%s %d nsec\n",                          \
 116                        __func__, tag, duration);                        \
 117         }                                                               \
 118         errno = __save_errno;                                           \
 119         __ret;                                                          \
 120 })
 121
 122 #define CHECK_FAIL(condition) ({                                        \
 123         int __ret = !!(condition);                                      \
 124         int __save_errno = errno;                                       \
 125         if (__ret) {                                                    \
 126                 test__fail();                                           \
 127                 printf("%s:FAIL:%d\n", __func__, __LINE__);             \
 128         }                                                               \
 129         errno = __save_errno;                                           \
 130         __ret;                                                          \
 131 })
 132
 133 #define CHECK(condition, tag, format...) \
 134         _CHECK(condition, tag, duration, format)
 135 #define CHECK_ATTR(condition, tag, format...) \
 136         _CHECK(condition, tag, tattr.duration, format)
 137
 138 #define MAGIC_VAL 0x1234
 139 #define NUM_ITER 100000
 140 #define VIP_NUM 5
 141
 142 static inline __u64 ptr_to_u64(const void *ptr)
 143 {
 144         return (__u64) (unsigned long) ptr;
 145 }
 146
 147 int bpf_find_map(const char *test, struct bpf_object *obj, const char *name);
 148 int compare_map_keys(int map1_fd, int map2_fd);
 149 int compare_stack_ips(int smap_fd, int amap_fd, int stack_trace_len);
 150 int extract_build_id(char *build_id, size_t size);
 151 void *spin_lock_thread(void *arg);
 152
 153 #ifdef __x86_64__
 154 #define SYS_NANOSLEEP_KPROBE_NAME "__x64_sys_nanosleep"
 155 #elif defined(__s390x__)
 156 #define SYS_NANOSLEEP_KPROBE_NAME "__s390x_sys_nanosleep"
 157 #else
 158 #define SYS_NANOSLEEP_KPROBE_NAME "sys_nanosleep"
 159 #endif