diff options
author | Jiong Wang <jiong.wang@netronome.com> | 2019-02-11 07:01:20 -0500 |
---|---|---|
committer | Alexei Starovoitov <ast@kernel.org> | 2019-02-11 23:31:38 -0500 |
commit | bd4aed0ee73ca873bef3cb3ec746dd796f03df28 (patch) | |
tree | 1dcd4de735f8ca3dbf672dc5d48d064ea1dea5dc /tools/testing/selftests/bpf/progs/sendmsg4_prog.c | |
parent | 4836b4637ef080c2764c44ee40ed354cdb991d79 (diff) |
selftests: bpf: centre kernel bpf objects under new subdir "progs"
At the moment, all kernel bpf objects are listed under BPF_OBJ_FILES.
Listing them manually sometimes causing patch conflict when people are
adding new testcases simultaneously.
It is better to centre all the related source files under a subdir
"progs", then auto-generate the object file list.
Suggested-by: Alexei Starovoitov <ast@kernel.org>
Reviewed-by: Jakub Kicinski <jakub.kicinski@netronome.com>
Signed-off-by: Jiong Wang <jiong.wang@netronome.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Diffstat (limited to 'tools/testing/selftests/bpf/progs/sendmsg4_prog.c')
-rw-r--r-- | tools/testing/selftests/bpf/progs/sendmsg4_prog.c | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/tools/testing/selftests/bpf/progs/sendmsg4_prog.c b/tools/testing/selftests/bpf/progs/sendmsg4_prog.c new file mode 100644 index 000000000000..a91536b1c47e --- /dev/null +++ b/tools/testing/selftests/bpf/progs/sendmsg4_prog.c | |||
@@ -0,0 +1,49 @@ | |||
1 | // SPDX-License-Identifier: GPL-2.0 | ||
2 | // Copyright (c) 2018 Facebook | ||
3 | |||
4 | #include <linux/stddef.h> | ||
5 | #include <linux/bpf.h> | ||
6 | #include <sys/socket.h> | ||
7 | |||
8 | #include "bpf_helpers.h" | ||
9 | #include "bpf_endian.h" | ||
10 | |||
11 | #define SRC1_IP4 0xAC100001U /* 172.16.0.1 */ | ||
12 | #define SRC2_IP4 0x00000000U | ||
13 | #define SRC_REWRITE_IP4 0x7f000004U | ||
14 | #define DST_IP4 0xC0A801FEU /* 192.168.1.254 */ | ||
15 | #define DST_REWRITE_IP4 0x7f000001U | ||
16 | #define DST_PORT 4040 | ||
17 | #define DST_REWRITE_PORT4 4444 | ||
18 | |||
19 | int _version SEC("version") = 1; | ||
20 | |||
21 | SEC("cgroup/sendmsg4") | ||
22 | int sendmsg_v4_prog(struct bpf_sock_addr *ctx) | ||
23 | { | ||
24 | if (ctx->type != SOCK_DGRAM) | ||
25 | return 0; | ||
26 | |||
27 | /* Rewrite source. */ | ||
28 | if (ctx->msg_src_ip4 == bpf_htonl(SRC1_IP4) || | ||
29 | ctx->msg_src_ip4 == bpf_htonl(SRC2_IP4)) { | ||
30 | ctx->msg_src_ip4 = bpf_htonl(SRC_REWRITE_IP4); | ||
31 | } else { | ||
32 | /* Unexpected source. Reject sendmsg. */ | ||
33 | return 0; | ||
34 | } | ||
35 | |||
36 | /* Rewrite destination. */ | ||
37 | if ((ctx->user_ip4 >> 24) == (bpf_htonl(DST_IP4) >> 24) && | ||
38 | ctx->user_port == bpf_htons(DST_PORT)) { | ||
39 | ctx->user_ip4 = bpf_htonl(DST_REWRITE_IP4); | ||
40 | ctx->user_port = bpf_htons(DST_REWRITE_PORT4); | ||
41 | } else { | ||
42 | /* Unexpected source. Reject sendmsg. */ | ||
43 | return 0; | ||
44 | } | ||
45 | |||
46 | return 1; | ||
47 | } | ||
48 | |||
49 | char _license[] SEC("license") = "GPL"; | ||