1 // SPDX-License-Identifier: GPL-2.0
5 #include <linux/stddef.h>
9 #include <sys/socket.h>
10 #include <netinet/tcp.h>
14 #include <bpf/bpf_helpers.h>
15 #include <bpf/bpf_endian.h>
17 #define SERV4_IP 0xc0a801feU /* 192.168.1.254 */
18 #define SERV4_PORT 4040
19 #define SERV4_REWRITE_IP 0x7f000001U /* 127.0.0.1 */
20 #define SERV4_REWRITE_PORT 4444
26 static __inline int bind_to_device(struct bpf_sock_addr *ctx)
28 char veth1[IFNAMSIZ] = "test_sock_addr1";
29 char veth2[IFNAMSIZ] = "test_sock_addr2";
30 char missing[IFNAMSIZ] = "nonexistent_dev";
31 char del_bind[IFNAMSIZ] = "";
32 int veth1_idx, veth2_idx;
34 if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,
35 &veth1, sizeof(veth1)))
37 if (bpf_getsockopt(ctx, SOL_SOCKET, SO_BINDTOIFINDEX,
38 &veth1_idx, sizeof(veth1_idx)) || !veth1_idx)
40 if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,
41 &veth2, sizeof(veth2)))
43 if (bpf_getsockopt(ctx, SOL_SOCKET, SO_BINDTOIFINDEX,
44 &veth2_idx, sizeof(veth2_idx)) || !veth2_idx ||
45 veth1_idx == veth2_idx)
47 if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,
48 &missing, sizeof(missing)) != -ENODEV)
50 if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTOIFINDEX,
51 &veth1_idx, sizeof(veth1_idx)))
53 if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,
54 &del_bind, sizeof(del_bind)))
60 static __inline int misc_opts(struct bpf_sock_addr *ctx, int opt)
62 int old, tmp, new = 0xeb9f;
64 /* Socket in test case has guarantee that old never equals to new. */
65 if (bpf_getsockopt(ctx, SOL_SOCKET, opt, &old, sizeof(old)) ||
68 if (bpf_setsockopt(ctx, SOL_SOCKET, opt, &new, sizeof(new)))
70 if (bpf_getsockopt(ctx, SOL_SOCKET, opt, &tmp, sizeof(tmp)) ||
73 if (bpf_setsockopt(ctx, SOL_SOCKET, opt, &old, sizeof(old)))
80 int bind_v4_prog(struct bpf_sock_addr *ctx)
90 if (sk->family != AF_INET)
93 if (ctx->type != SOCK_STREAM && ctx->type != SOCK_DGRAM)
96 if (ctx->user_ip4 != bpf_htonl(SERV4_IP) ||
97 ctx->user_port != bpf_htons(SERV4_PORT))
102 user_ip4 |= ((volatile __u8 *)&ctx->user_ip4)[0] << 0;
103 user_ip4 |= ((volatile __u8 *)&ctx->user_ip4)[1] << 8;
104 user_ip4 |= ((volatile __u8 *)&ctx->user_ip4)[2] << 16;
105 user_ip4 |= ((volatile __u8 *)&ctx->user_ip4)[3] << 24;
106 if (ctx->user_ip4 != user_ip4)
110 user_port |= ((volatile __u8 *)&ctx->user_port)[0] << 0;
111 user_port |= ((volatile __u8 *)&ctx->user_port)[1] << 8;
112 if (ctx->user_port != user_port)
117 user_ip4 |= ((volatile __u16 *)&ctx->user_ip4)[0] << 0;
118 user_ip4 |= ((volatile __u16 *)&ctx->user_ip4)[1] << 16;
119 if (ctx->user_ip4 != user_ip4)
122 /* Bind to device and unbind it. */
123 if (bind_to_device(ctx))
126 /* Test for misc socket options. */
127 if (misc_opts(ctx, SO_MARK) || misc_opts(ctx, SO_PRIORITY))
130 ctx->user_ip4 = bpf_htonl(SERV4_REWRITE_IP);
131 ctx->user_port = bpf_htons(SERV4_REWRITE_PORT);
136 char _license[] SEC("license") = "GPL";