811d7e375d
It is common for networking tests creating its netns and making its own setting under this new netns (e.g. changing tcp sysctl). If the test forgot to restore to the original netns, it would affect the result of other tests. This patch saves the original netns at the beginning and then restores it after every test. Since the restore "setns()" is not expensive, it does it on all tests without tracking if a test has created a new netns or not. The new restore_netns() could also be done in test__end_subtest() such that each subtest will get an automatic netns reset. However, the individual test would lose flexibility to have total control on netns for its own subtests. In some cases, forcing a test to do unnecessary netns re-configure for each subtest is time consuming. e.g. In my vm, forcing netns re-configure on each subtest in sk_assign.c increased the runtime from 1s to 8s. On top of that, test_progs.c is also doing per-test (instead of per-subtest) cleanup for cgroup. Thus, this patch also does per-test restore_netns(). The only existing per-subtest cleanup is reset_affinity() and no test is depending on this. Thus, it is removed from test__end_subtest() to give a consistent expectation to the individual tests. test_progs.c only ensures any affinity/netns/cgroup change made by an earlier test does not affect the following tests. Signed-off-by: Martin KaFai Lau <kafai@fb.com> Signed-off-by: Daniel Borkmann <daniel@iogearbox.net> Acked-by: Andrii Nakryiko <andriin@fb.com> Link: https://lore.kernel.org/bpf/20200702004858.2103728-1-kafai@fb.com
149 lines
3.6 KiB
C
149 lines
3.6 KiB
C
/* SPDX-License-Identifier: GPL-2.0 */
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
#include <time.h>
|
|
#include <signal.h>
|
|
|
|
#include <linux/types.h>
|
|
typedef __u16 __sum16;
|
|
#include <arpa/inet.h>
|
|
#include <linux/if_ether.h>
|
|
#include <linux/if_packet.h>
|
|
#include <linux/ip.h>
|
|
#include <linux/ipv6.h>
|
|
#include <netinet/tcp.h>
|
|
#include <linux/filter.h>
|
|
#include <linux/perf_event.h>
|
|
#include <linux/socket.h>
|
|
#include <linux/unistd.h>
|
|
|
|
#include <sys/ioctl.h>
|
|
#include <sys/wait.h>
|
|
#include <sys/types.h>
|
|
#include <sys/time.h>
|
|
#include <fcntl.h>
|
|
#include <pthread.h>
|
|
#include <linux/bpf.h>
|
|
#include <linux/err.h>
|
|
#include <bpf/bpf.h>
|
|
#include <bpf/libbpf.h>
|
|
|
|
#include "test_iptunnel_common.h"
|
|
#include "bpf_util.h"
|
|
#include <bpf/bpf_endian.h>
|
|
#include "trace_helpers.h"
|
|
#include "testing_helpers.h"
|
|
#include "flow_dissector_load.h"
|
|
|
|
enum verbosity {
|
|
VERBOSE_NONE,
|
|
VERBOSE_NORMAL,
|
|
VERBOSE_VERY,
|
|
VERBOSE_SUPER,
|
|
};
|
|
|
|
struct str_set {
|
|
const char **strs;
|
|
int cnt;
|
|
};
|
|
|
|
struct test_selector {
|
|
struct str_set whitelist;
|
|
struct str_set blacklist;
|
|
bool *num_set;
|
|
int num_set_len;
|
|
};
|
|
|
|
struct test_env {
|
|
struct test_selector test_selector;
|
|
struct test_selector subtest_selector;
|
|
bool verifier_stats;
|
|
enum verbosity verbosity;
|
|
|
|
bool jit_enabled;
|
|
bool get_test_cnt;
|
|
bool list_test_names;
|
|
|
|
struct prog_test_def *test;
|
|
FILE *stdout;
|
|
FILE *stderr;
|
|
char *log_buf;
|
|
size_t log_cnt;
|
|
int nr_cpus;
|
|
|
|
int succ_cnt; /* successful tests */
|
|
int sub_succ_cnt; /* successful sub-tests */
|
|
int fail_cnt; /* total failed tests + sub-tests */
|
|
int skip_cnt; /* skipped tests */
|
|
|
|
int saved_netns_fd;
|
|
};
|
|
|
|
extern struct test_env env;
|
|
|
|
extern void test__force_log();
|
|
extern bool test__start_subtest(const char *name);
|
|
extern void test__skip(void);
|
|
extern void test__fail(void);
|
|
extern int test__join_cgroup(const char *path);
|
|
|
|
#define PRINT_FAIL(format...) \
|
|
({ \
|
|
test__fail(); \
|
|
fprintf(stdout, "%s:FAIL:%d ", __func__, __LINE__); \
|
|
fprintf(stdout, ##format); \
|
|
})
|
|
|
|
#define _CHECK(condition, tag, duration, format...) ({ \
|
|
int __ret = !!(condition); \
|
|
int __save_errno = errno; \
|
|
if (__ret) { \
|
|
test__fail(); \
|
|
fprintf(stdout, "%s:FAIL:%s ", __func__, tag); \
|
|
fprintf(stdout, ##format); \
|
|
} else { \
|
|
fprintf(stdout, "%s:PASS:%s %d nsec\n", \
|
|
__func__, tag, duration); \
|
|
} \
|
|
errno = __save_errno; \
|
|
__ret; \
|
|
})
|
|
|
|
#define CHECK_FAIL(condition) ({ \
|
|
int __ret = !!(condition); \
|
|
int __save_errno = errno; \
|
|
if (__ret) { \
|
|
test__fail(); \
|
|
fprintf(stdout, "%s:FAIL:%d\n", __func__, __LINE__); \
|
|
} \
|
|
errno = __save_errno; \
|
|
__ret; \
|
|
})
|
|
|
|
#define CHECK(condition, tag, format...) \
|
|
_CHECK(condition, tag, duration, format)
|
|
#define CHECK_ATTR(condition, tag, format...) \
|
|
_CHECK(condition, tag, tattr.duration, format)
|
|
|
|
static inline __u64 ptr_to_u64(const void *ptr)
|
|
{
|
|
return (__u64) (unsigned long) ptr;
|
|
}
|
|
|
|
int bpf_find_map(const char *test, struct bpf_object *obj, const char *name);
|
|
int compare_map_keys(int map1_fd, int map2_fd);
|
|
int compare_stack_ips(int smap_fd, int amap_fd, int stack_trace_len);
|
|
int extract_build_id(char *build_id, size_t size);
|
|
|
|
#ifdef __x86_64__
|
|
#define SYS_NANOSLEEP_KPROBE_NAME "__x64_sys_nanosleep"
|
|
#elif defined(__s390x__)
|
|
#define SYS_NANOSLEEP_KPROBE_NAME "__s390x_sys_nanosleep"
|
|
#else
|
|
#define SYS_NANOSLEEP_KPROBE_NAME "sys_nanosleep"
|
|
#endif
|