LXR linux/tools/testing/selftests/bpf/test

   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>
  13typedef __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
  42enum verbosity {
  43        VERBOSE_NONE,
  44        VERBOSE_NORMAL,
  45        VERBOSE_VERY,
  46        VERBOSE_SUPER,
  47};
  48
  49struct str_set {
  50        const char **strs;
  51        int cnt;
  52};
  53
  54struct test_selector {
  55        struct str_set whitelist;
  56        struct str_set blacklist;
  57        bool *num_set;
  58        int num_set_len;
  59};
  60
  61struct 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
  81extern struct test_env env;
  82
  83extern void test__force_log();
  84extern bool test__start_subtest(const char *name);
  85extern void test__skip(void);
  86extern void test__fail(void);
  87extern int test__join_cgroup(const char *path);
  88
  89#define MAGIC_BYTES 123
  90
  91/* ipv4 test vector */
  92struct ipv4_packet {
  93        struct ethhdr eth;
  94        struct iphdr iph;
  95        struct tcphdr tcp;
  96} __packed;
  97extern struct ipv4_packet pkt_v4;
  98
  99/* ipv6 test vector */
 100struct ipv6_packet {
 101        struct ethhdr eth;
 102        struct ipv6hdr iph;
 103        struct tcphdr tcp;
 104} __packed;
 105extern 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
 142static inline __u64 ptr_to_u64(const void *ptr)
 143{
 144        return (__u64) (unsigned long) ptr;
 145}
 146
 147int bpf_find_map(const char *test, struct bpf_object *obj, const char *name);
 148int compare_map_keys(int map1_fd, int map2_fd);
 149int compare_stack_ips(int smap_fd, int amap_fd, int stack_trace_len);
 150int extract_build_id(char *build_id, size_t size);
 151void *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
 160