From: Song Liu <songliubraving@fb.com>
To: Martin Lau <kafai@fb.com>
Cc: "bpf@vger.kernel.org" <bpf@vger.kernel.org>,
Alexei Starovoitov <ast@kernel.org>,
Daniel Borkmann <daniel@iogearbox.net>,
Kernel Team <Kernel-team@fb.com>,
"netdev@vger.kernel.org" <netdev@vger.kernel.org>
Subject: Re: [PATCH bpf-next 2/3] bpf: Allow using bpf_sk_storage in FENTRY/FEXIT/RAW_TP
Date: Fri, 6 Nov 2020 22:59:14 +0000 [thread overview]
Message-ID: <8FA16B3B-DC01-4FAB-B5F6-1871C5151D67@fb.com> (raw)
In-Reply-To: <20201106220803.3950648-1-kafai@fb.com>
> On Nov 6, 2020, at 2:08 PM, Martin KaFai Lau <kafai@fb.com> wrote:
>
> This patch enables the FENTRY/FEXIT/RAW_TP tracing program to use
> the bpf_sk_storage_(get|delete) helper, so those tracing programs
> can access the sk's bpf_local_storage and the later selftest
> will show some examples.
>
> The bpf_sk_storage is currently used in bpf-tcp-cc, tc,
> cg sockops...etc which is running either in softirq or
> task context.
>
> This patch adds bpf_sk_storage_get_tracing_proto and
> bpf_sk_storage_delete_tracing_proto. They will check
> in runtime that the helpers can only be called when serving
> softirq or running in a task context. That should enable
> most common tracing use cases on sk.
>
> During the load time, the new tracing_allowed() function
> will ensure the tracing prog using the bpf_sk_storage_(get|delete)
> helper is not tracing any *sk_storage*() function itself.
> The sk is passed as "void *" when calling into bpf_local_storage.
>
> Signed-off-by: Martin KaFai Lau <kafai@fb.com>
> ---
> include/net/bpf_sk_storage.h | 2 +
> kernel/trace/bpf_trace.c | 5 +++
> net/core/bpf_sk_storage.c | 73 ++++++++++++++++++++++++++++++++++++
> 3 files changed, 80 insertions(+)
>
> diff --git a/include/net/bpf_sk_storage.h b/include/net/bpf_sk_storage.h
> index 3c516dd07caf..0e85713f56df 100644
> --- a/include/net/bpf_sk_storage.h
> +++ b/include/net/bpf_sk_storage.h
> @@ -20,6 +20,8 @@ void bpf_sk_storage_free(struct sock *sk);
>
> extern const struct bpf_func_proto bpf_sk_storage_get_proto;
> extern const struct bpf_func_proto bpf_sk_storage_delete_proto;
> +extern const struct bpf_func_proto bpf_sk_storage_get_tracing_proto;
> +extern const struct bpf_func_proto bpf_sk_storage_delete_tracing_proto;
>
> struct bpf_local_storage_elem;
> struct bpf_sk_storage_diag;
> diff --git a/kernel/trace/bpf_trace.c b/kernel/trace/bpf_trace.c
> index e4515b0f62a8..cfce60ad1cb5 100644
> --- a/kernel/trace/bpf_trace.c
> +++ b/kernel/trace/bpf_trace.c
> @@ -16,6 +16,7 @@
> #include <linux/syscalls.h>
> #include <linux/error-injection.h>
> #include <linux/btf_ids.h>
> +#include <net/bpf_sk_storage.h>
>
> #include <uapi/linux/bpf.h>
> #include <uapi/linux/btf.h>
> @@ -1735,6 +1736,10 @@ tracing_prog_func_proto(enum bpf_func_id func_id, const struct bpf_prog *prog)
> return &bpf_skc_to_tcp_request_sock_proto;
> case BPF_FUNC_skc_to_udp6_sock:
> return &bpf_skc_to_udp6_sock_proto;
> + case BPF_FUNC_sk_storage_get:
> + return &bpf_sk_storage_get_tracing_proto;
> + case BPF_FUNC_sk_storage_delete:
> + return &bpf_sk_storage_delete_tracing_proto;
> #endif
> case BPF_FUNC_seq_printf:
> return prog->expected_attach_type == BPF_TRACE_ITER ?
> diff --git a/net/core/bpf_sk_storage.c b/net/core/bpf_sk_storage.c
> index 001eac65e40f..1a41c917e08d 100644
> --- a/net/core/bpf_sk_storage.c
> +++ b/net/core/bpf_sk_storage.c
> @@ -6,6 +6,7 @@
> #include <linux/types.h>
> #include <linux/spinlock.h>
> #include <linux/bpf.h>
> +#include <linux/btf.h>
> #include <linux/btf_ids.h>
> #include <linux/bpf_local_storage.h>
> #include <net/bpf_sk_storage.h>
> @@ -378,6 +379,78 @@ const struct bpf_func_proto bpf_sk_storage_delete_proto = {
> .arg2_type = ARG_PTR_TO_BTF_ID_SOCK_COMMON,
> };
>
> +static bool bpf_sk_storage_tracing_allowed(const struct bpf_prog *prog)
> +{
> + const struct btf *btf_vmlinux;
> + const struct btf_type *t;
> + const char *tname;
> + u32 btf_id;
> +
> + if (prog->aux->dst_prog)
> + return false;
> +
> + /* Ensure the tracing program is not tracing
> + * any *sk_storage*() function and also
> + * use the bpf_sk_storage_(get|delete) helper.
> + */
> + switch (prog->expected_attach_type) {
> + case BPF_TRACE_RAW_TP:
> + /* bpf_sk_storage has no trace point */
> + return true;
> + case BPF_TRACE_FENTRY:
> + case BPF_TRACE_FEXIT:
> + btf_vmlinux = bpf_get_btf_vmlinux();
> + btf_id = prog->aux->attach_btf_id;
> + t = btf_type_by_id(btf_vmlinux, btf_id);
What happens to fentry/fexit attach to other BPF programs? I guess
we should check for t == NULL?
Thanks,
Song
> + tname = btf_name_by_offset(btf_vmlinux, t->name_off);
> + return !strstr(tname, "sk_storage");
> + default:
> + return false;
> + }
> +
> + return false;
> +}
[...]
next prev parent reply other threads:[~2020-11-06 23:00 UTC|newest]
Thread overview: 18+ messages / expand[flat|nested] mbox.gz Atom feed top
2020-11-06 22:07 [PATCH bpf-next 0/3] bpf: Enable bpf_sk_storage for FENTRY/FEXIT/RAW_TP Martin KaFai Lau
2020-11-06 22:07 ` [PATCH bpf-next 1/3] bpf: Folding omem_charge() into sk_storage_charge() Martin KaFai Lau
2020-11-06 22:44 ` Song Liu
2020-11-06 22:08 ` [PATCH bpf-next 2/3] bpf: Allow using bpf_sk_storage in FENTRY/FEXIT/RAW_TP Martin KaFai Lau
2020-11-06 22:59 ` Song Liu [this message]
2020-11-06 23:18 ` Martin KaFai Lau
2020-11-07 0:20 ` Song Liu
2020-11-07 1:14 ` Andrii Nakryiko
2020-11-07 1:52 ` Martin KaFai Lau
2020-11-09 18:09 ` Andrii Nakryiko
2020-11-09 20:32 ` John Fastabend
2020-11-10 22:01 ` KP Singh
2020-11-10 23:43 ` Martin KaFai Lau
2020-11-10 23:53 ` Andrii Nakryiko
2020-11-11 0:07 ` Martin KaFai Lau
2020-11-11 0:17 ` Andrii Nakryiko
2020-11-11 0:20 ` Martin KaFai Lau
2020-11-06 22:08 ` [PATCH bpf-next 3/3] bpf: selftest: Use " Martin KaFai Lau
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=8FA16B3B-DC01-4FAB-B5F6-1871C5151D67@fb.com \
--to=songliubraving@fb.com \
--cc=Kernel-team@fb.com \
--cc=ast@kernel.org \
--cc=bpf@vger.kernel.org \
--cc=daniel@iogearbox.net \
--cc=kafai@fb.com \
--cc=netdev@vger.kernel.org \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for NNTP newsgroup(s).