tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
fork
Configure Feed
Select the types of activity you want to include in your feed.
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