mirror of
git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2025-08-05 16:54:27 +00:00
e626a13f6f
Drop bpf_iter.h header which uses vmlinux.h but re-defines a bunch of iterator structures and some of BPF constants for use in BPF iterator selftests. None of that is necessary when fresh vmlinux.h header is generated for vmlinux image that matches latest selftests. So drop ugly hacks and have a nice plain vmlinux.h usage everywhere. We could do the same with all the kfunc __ksym redefinitions, but that has dependency on very fresh pahole, so I'm not addressing that here. Signed-off-by: Andrii Nakryiko <andrii@kernel.org> Link: https://lore.kernel.org/r/20241029203919.1948941-1-andrii@kernel.org Signed-off-by: Alexei Starovoitov <ast@kernel.org>
59 lines
1.2 KiB
C
59 lines
1.2 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright (c) 2020 Cloudflare */
|
|
#include <vmlinux.h>
|
|
#include "bpf_tracing_net.h"
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <bpf/bpf_tracing.h>
|
|
#include <errno.h>
|
|
|
|
char _license[] SEC("license") = "GPL";
|
|
|
|
struct {
|
|
__uint(type, BPF_MAP_TYPE_SOCKMAP);
|
|
__uint(max_entries, 64);
|
|
__type(key, __u32);
|
|
__type(value, __u64);
|
|
} sockmap SEC(".maps");
|
|
|
|
struct {
|
|
__uint(type, BPF_MAP_TYPE_SOCKHASH);
|
|
__uint(max_entries, 64);
|
|
__type(key, __u32);
|
|
__type(value, __u64);
|
|
} sockhash SEC(".maps");
|
|
|
|
struct {
|
|
__uint(type, BPF_MAP_TYPE_SOCKHASH);
|
|
__uint(max_entries, 64);
|
|
__type(key, __u32);
|
|
__type(value, __u64);
|
|
} dst SEC(".maps");
|
|
|
|
__u32 elems = 0;
|
|
__u32 socks = 0;
|
|
|
|
SEC("iter/sockmap")
|
|
int copy(struct bpf_iter__sockmap *ctx)
|
|
{
|
|
struct sock *sk = ctx->sk;
|
|
__u32 tmp, *key = ctx->key;
|
|
int ret;
|
|
|
|
if (!key)
|
|
return 0;
|
|
|
|
elems++;
|
|
|
|
/* We need a temporary buffer on the stack, since the verifier doesn't
|
|
* let us use the pointer from the context as an argument to the helper.
|
|
*/
|
|
tmp = *key;
|
|
|
|
if (sk) {
|
|
socks++;
|
|
return bpf_map_update_elem(&dst, &tmp, sk, 0) != 0;
|
|
}
|
|
|
|
ret = bpf_map_delete_elem(&dst, &tmp);
|
|
return ret && ret != -ENOENT;
|
|
}
|