mirror of
https://github.com/AuxXxilium/linux_dsm_epyc7002.git
synced 2025-01-16 11:36:36 +07:00
3e689141e6
Fix all selftests to include libbpf header files with the bpf/ prefix, to
be consistent with external users of the library. Also ensure that all
includes of exported libbpf header files (those that are exported on 'make
install' of the library) use bracketed includes instead of quoted.
To not break the build, keep the old include path until everything has been
changed to the new one; a subsequent patch will remove that.
Fixes: 6910d7d386 ("selftests/bpf: Ensure bpf_helper_defs.h are taken from selftests dir")
Signed-off-by: Toke Høiland-Jørgensen <toke@redhat.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Acked-by: Andrii Nakryiko <andriin@fb.com>
Link: https://lore.kernel.org/bpf/157952560568.1683545.9649335788846513446.stgit@toke.dk
96 lines
2.4 KiB
C
96 lines
2.4 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
// Copyright (c) 2018 Facebook
|
|
|
|
#include <string.h>
|
|
|
|
#include <linux/stddef.h>
|
|
#include <linux/bpf.h>
|
|
#include <linux/in.h>
|
|
#include <linux/in6.h>
|
|
#include <sys/socket.h>
|
|
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <bpf/bpf_endian.h>
|
|
|
|
#define SRC_REWRITE_IP6_0 0
|
|
#define SRC_REWRITE_IP6_1 0
|
|
#define SRC_REWRITE_IP6_2 0
|
|
#define SRC_REWRITE_IP6_3 6
|
|
|
|
#define DST_REWRITE_IP6_0 0
|
|
#define DST_REWRITE_IP6_1 0
|
|
#define DST_REWRITE_IP6_2 0
|
|
#define DST_REWRITE_IP6_3 1
|
|
|
|
#define DST_REWRITE_PORT6 6666
|
|
|
|
int _version SEC("version") = 1;
|
|
|
|
SEC("cgroup/connect6")
|
|
int connect_v6_prog(struct bpf_sock_addr *ctx)
|
|
{
|
|
struct bpf_sock_tuple tuple = {};
|
|
struct sockaddr_in6 sa;
|
|
struct bpf_sock *sk;
|
|
|
|
/* Verify that new destination is available. */
|
|
memset(&tuple.ipv6.saddr, 0, sizeof(tuple.ipv6.saddr));
|
|
memset(&tuple.ipv6.sport, 0, sizeof(tuple.ipv6.sport));
|
|
|
|
tuple.ipv6.daddr[0] = bpf_htonl(DST_REWRITE_IP6_0);
|
|
tuple.ipv6.daddr[1] = bpf_htonl(DST_REWRITE_IP6_1);
|
|
tuple.ipv6.daddr[2] = bpf_htonl(DST_REWRITE_IP6_2);
|
|
tuple.ipv6.daddr[3] = bpf_htonl(DST_REWRITE_IP6_3);
|
|
|
|
tuple.ipv6.dport = bpf_htons(DST_REWRITE_PORT6);
|
|
|
|
if (ctx->type != SOCK_STREAM && ctx->type != SOCK_DGRAM)
|
|
return 0;
|
|
else if (ctx->type == SOCK_STREAM)
|
|
sk = bpf_sk_lookup_tcp(ctx, &tuple, sizeof(tuple.ipv6),
|
|
BPF_F_CURRENT_NETNS, 0);
|
|
else
|
|
sk = bpf_sk_lookup_udp(ctx, &tuple, sizeof(tuple.ipv6),
|
|
BPF_F_CURRENT_NETNS, 0);
|
|
|
|
if (!sk)
|
|
return 0;
|
|
|
|
if (sk->src_ip6[0] != tuple.ipv6.daddr[0] ||
|
|
sk->src_ip6[1] != tuple.ipv6.daddr[1] ||
|
|
sk->src_ip6[2] != tuple.ipv6.daddr[2] ||
|
|
sk->src_ip6[3] != tuple.ipv6.daddr[3] ||
|
|
sk->src_port != DST_REWRITE_PORT6) {
|
|
bpf_sk_release(sk);
|
|
return 0;
|
|
}
|
|
|
|
bpf_sk_release(sk);
|
|
|
|
/* Rewrite destination. */
|
|
ctx->user_ip6[0] = bpf_htonl(DST_REWRITE_IP6_0);
|
|
ctx->user_ip6[1] = bpf_htonl(DST_REWRITE_IP6_1);
|
|
ctx->user_ip6[2] = bpf_htonl(DST_REWRITE_IP6_2);
|
|
ctx->user_ip6[3] = bpf_htonl(DST_REWRITE_IP6_3);
|
|
|
|
ctx->user_port = bpf_htons(DST_REWRITE_PORT6);
|
|
|
|
/* Rewrite source. */
|
|
memset(&sa, 0, sizeof(sa));
|
|
|
|
sa.sin6_family = AF_INET6;
|
|
sa.sin6_port = bpf_htons(0);
|
|
|
|
sa.sin6_addr.s6_addr32[0] = bpf_htonl(SRC_REWRITE_IP6_0);
|
|
sa.sin6_addr.s6_addr32[1] = bpf_htonl(SRC_REWRITE_IP6_1);
|
|
sa.sin6_addr.s6_addr32[2] = bpf_htonl(SRC_REWRITE_IP6_2);
|
|
sa.sin6_addr.s6_addr32[3] = bpf_htonl(SRC_REWRITE_IP6_3);
|
|
|
|
if (bpf_bind(ctx, (struct sockaddr *)&sa, sizeof(sa)) != 0)
|
|
return 0;
|
|
|
|
return 1;
|
|
}
|
|
|
|
char _license[] SEC("license") = "GPL";
|