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 <linux/filter.h>
  20 #include <linux/perf_event.h>
  21 #include <linux/socket.h>
  22 #include <linux/unistd.h>
  23
  24 #include <sys/ioctl.h>
  25 #include <sys/wait.h>
  26 #include <sys/types.h>
  27 #include <sys/time.h>
  28 #include <fcntl.h>
  29 #include <pthread.h>
  30 #include <linux/bpf.h>
  31 #include <linux/err.h>
  32 #include <bpf/bpf.h>
  33 #include <bpf/libbpf.h>
  34
  35 #include "test_iptunnel_common.h"
  36 #include "bpf_util.h"
  37 #include <bpf/bpf_endian.h>
  38 #include "trace_helpers.h"
  39 #include "testing_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         bool has_testmod;
  69         bool get_test_cnt;
  70         bool list_test_names;
  71
  72         struct prog_test_def *test;
  73         FILE *stdout;
  74         FILE *stderr;
  75         char *log_buf;
  76         size_t log_cnt;
  77         int nr_cpus;
  78
  79         int succ_cnt; /* successful tests */
  80         int sub_succ_cnt; /* successful sub-tests */
  81         int fail_cnt; /* total failed tests + sub-tests */
  82         int skip_cnt; /* skipped tests */
  83
  84         int saved_netns_fd;
  85 };
  86
  87 extern struct test_env env;
  88
  89 extern void test__force_log();
  90 extern bool test__start_subtest(const char *name);
  91 extern void test__skip(void);
  92 extern void test__fail(void);
  93 extern int test__join_cgroup(const char *path);
  94
  95 #define PRINT_FAIL(format...)                                                  \
  96         ({                                                                     \
  97                 test__fail();                                                  \
  98                 fprintf(stdout, "%s:FAIL:%d ", __func__, __LINE__);            \
  99                 fprintf(stdout, ##format);                                     \
 100         })
 101
 102 #define _CHECK(condition, tag, duration, format...) ({                  \
 103         int __ret = !!(condition);                                      \
 104         int __save_errno = errno;                                       \
 105         if (__ret) {                                                    \
 106                 test__fail();                                           \
 107                 fprintf(stdout, "%s:FAIL:%s ", __func__, tag);          \
 108                 fprintf(stdout, ##format);                              \
 109         } else {                                                        \
 110                 fprintf(stdout, "%s:PASS:%s %d nsec\n",                 \
 111                        __func__, tag, duration);                        \
 112         }                                                               \
 113         errno = __save_errno;                                           \
 114         __ret;                                                          \
 115 })
 116
 117 #define CHECK_FAIL(condition) ({                                        \
 118         int __ret = !!(condition);                                      \
 119         int __save_errno = errno;                                       \
 120         if (__ret) {                                                    \
 121                 test__fail();                                           \
 122                 fprintf(stdout, "%s:FAIL:%d\n", __func__, __LINE__);    \
 123         }                                                               \
 124         errno = __save_errno;                                           \
 125         __ret;                                                          \
 126 })
 127
 128 #define CHECK(condition, tag, format...) \
 129         _CHECK(condition, tag, duration, format)
 130 #define CHECK_ATTR(condition, tag, format...) \
 131         _CHECK(condition, tag, tattr.duration, format)
 132
 133 #define ASSERT_TRUE(actual, name) ({                                    \
 134         static int duration = 0;                                        \
 135         bool ___ok = (actual);                                          \
 136         CHECK(!___ok, (name), "unexpected %s: got FALSE\n", (name));    \
 137         ___ok;                                                          \
 138 })
 139
 140 #define ASSERT_FALSE(actual, name) ({                                   \
 141         static int duration = 0;                                        \
 142         bool ___ok = !(actual);                                         \
 143         CHECK(!___ok, (name), "unexpected %s: got TRUE\n", (name));     \
 144         ___ok;                                                          \
 145 })
 146
 147 #define ASSERT_EQ(actual, expected, name) ({                            \
 148         static int duration = 0;                                        \
 149         typeof(actual) ___act = (actual);                               \
 150         typeof(expected) ___exp = (expected);                           \
 151         bool ___ok = ___act == ___exp;                                  \
 152         CHECK(!___ok, (name),                                           \
 153               "unexpected %s: actual %lld != expected %lld\n",          \
 154               (name), (long long)(___act), (long long)(___exp));        \
 155         ___ok;                                                          \
 156 })
 157
 158 #define ASSERT_NEQ(actual, expected, name) ({                           \
 159         static int duration = 0;                                        \
 160         typeof(actual) ___act = (actual);                               \
 161         typeof(expected) ___exp = (expected);                           \
 162         bool ___ok = ___act != ___exp;                                  \
 163         CHECK(!___ok, (name),                                           \
 164               "unexpected %s: actual %lld == expected %lld\n",          \
 165               (name), (long long)(___act), (long long)(___exp));        \
 166         ___ok;                                                          \
 167 })
 168
 169 #define ASSERT_LT(actual, expected, name) ({                            \
 170         static int duration = 0;                                        \
 171         typeof(actual) ___act = (actual);                               \
 172         typeof(expected) ___exp = (expected);                           \
 173         bool ___ok = ___act < ___exp;                                   \
 174         CHECK(!___ok, (name),                                           \
 175               "unexpected %s: actual %lld >= expected %lld\n",          \
 176               (name), (long long)(___act), (long long)(___exp));        \
 177         ___ok;                                                          \
 178 })
 179
 180 #define ASSERT_LE(actual, expected, name) ({                            \
 181         static int duration = 0;                                        \
 182         typeof(actual) ___act = (actual);                               \
 183         typeof(expected) ___exp = (expected);                           \
 184         bool ___ok = ___act <= ___exp;                                  \
 185         CHECK(!___ok, (name),                                           \
 186               "unexpected %s: actual %lld > expected %lld\n",           \
 187               (name), (long long)(___act), (long long)(___exp));        \
 188         ___ok;                                                          \
 189 })
 190
 191 #define ASSERT_GT(actual, expected, name) ({                            \
 192         static int duration = 0;                                        \
 193         typeof(actual) ___act = (actual);                               \
 194         typeof(expected) ___exp = (expected);                           \
 195         bool ___ok = ___act > ___exp;                                   \
 196         CHECK(!___ok, (name),                                           \
 197               "unexpected %s: actual %lld <= expected %lld\n",          \
 198               (name), (long long)(___act), (long long)(___exp));        \
 199         ___ok;                                                          \
 200 })
 201
 202 #define ASSERT_GE(actual, expected, name) ({                            \
 203         static int duration = 0;                                        \
 204         typeof(actual) ___act = (actual);                               \
 205         typeof(expected) ___exp = (expected);                           \
 206         bool ___ok = ___act >= ___exp;                                  \
 207         CHECK(!___ok, (name),                                           \
 208               "unexpected %s: actual %lld < expected %lld\n",           \
 209               (name), (long long)(___act), (long long)(___exp));        \
 210         ___ok;                                                          \
 211 })
 212
 213 #define ASSERT_STREQ(actual, expected, name) ({                         \
 214         static int duration = 0;                                        \
 215         const char *___act = actual;                                    \
 216         const char *___exp = expected;                                  \
 217         bool ___ok = strcmp(___act, ___exp) == 0;                       \
 218         CHECK(!___ok, (name),                                           \
 219               "unexpected %s: actual '%s' != expected '%s'\n",          \
 220               (name), ___act, ___exp);                                  \
 221         ___ok;                                                          \
 222 })
 223
 224 #define ASSERT_OK(res, name) ({                                         \
 225         static int duration = 0;                                        \
 226         long long ___res = (res);                                       \
 227         bool ___ok = ___res == 0;                                       \
 228         CHECK(!___ok, (name), "unexpected error: %lld (errno %d)\n",    \
 229               ___res, errno);                                           \
 230         ___ok;                                                          \
 231 })
 232
 233 #define ASSERT_ERR(res, name) ({                                        \
 234         static int duration = 0;                                        \
 235         long long ___res = (res);                                       \
 236         bool ___ok = ___res < 0;                                        \
 237         CHECK(!___ok, (name), "unexpected success: %lld\n", ___res);    \
 238         ___ok;                                                          \
 239 })
 240
 241 #define ASSERT_NULL(ptr, name) ({                                       \
 242         static int duration = 0;                                        \
 243         const void *___res = (ptr);                                     \
 244         bool ___ok = !___res;                                           \
 245         CHECK(!___ok, (name), "unexpected pointer: %p\n", ___res);      \
 246         ___ok;                                                          \
 247 })
 248
 249 #define ASSERT_OK_PTR(ptr, name) ({                                     \
 250         static int duration = 0;                                        \
 251         const void *___res = (ptr);                                     \
 252         bool ___ok = !IS_ERR_OR_NULL(___res);                           \
 253         CHECK(!___ok, (name),                                           \
 254               "unexpected error: %ld\n", PTR_ERR(___res));              \
 255         ___ok;                                                          \
 256 })
 257
 258 #define ASSERT_ERR_PTR(ptr, name) ({                                    \
 259         static int duration = 0;                                        \
 260         const void *___res = (ptr);                                     \
 261         bool ___ok = IS_ERR(___res);                                    \
 262         CHECK(!___ok, (name), "unexpected pointer: %p\n", ___res);      \
 263         ___ok;                                                          \
 264 })
 265
 266 static inline __u64 ptr_to_u64(const void *ptr)
 267 {
 268         return (__u64) (unsigned long) ptr;
 269 }
 270
 271 static inline void *u64_to_ptr(__u64 ptr)
 272 {
 273         return (void *) (unsigned long) ptr;
 274 }
 275
 276 int bpf_find_map(const char *test, struct bpf_object *obj, const char *name);
 277 int compare_map_keys(int map1_fd, int map2_fd);
 278 int compare_stack_ips(int smap_fd, int amap_fd, int stack_trace_len);
 279 int extract_build_id(char *build_id, size_t size);
 280 int kern_sync_rcu(void);
 281
 282 #ifdef __x86_64__
 283 #define SYS_NANOSLEEP_KPROBE_NAME "__x64_sys_nanosleep"
 284 #elif defined(__s390x__)
 285 #define SYS_NANOSLEEP_KPROBE_NAME "__s390x_sys_nanosleep"
 286 #else
 287 #define SYS_NANOSLEEP_KPROBE_NAME "sys_nanosleep"
 288 #endif