Adding bpf_check_attach_model function that returns
model for function specified by btf_id. It will be
used in following patches.
Signed-off-by: Jiri Olsa <jolsa@xxxxxxxxxx>
---
include/linux/bpf_verifier.h | 4 ++++
kernel/bpf/verifier.c | 29 +++++++++++++++++++++++++++++
2 files changed, 33 insertions(+)
diff --git a/include/linux/bpf_verifier.h b/include/linux/bpf_verifier.h
index c8a78e830fca..b561c0b08e68 100644
--- a/include/linux/bpf_verifier.h
+++ b/include/linux/bpf_verifier.h
@@ -527,6 +527,10 @@ int bpf_check_attach_target(struct bpf_verifier_log *log,
const struct bpf_prog *tgt_prog,
u32 btf_id,
struct bpf_attach_target_info *tgt_info);
+int bpf_check_attach_model(const struct bpf_prog *prog,
+ const struct bpf_prog *tgt_prog,
+ u32 btf_id,
+ struct btf_func_model *fmodel);
void bpf_free_kfunc_btf_tab(struct bpf_kfunc_btf_tab *tab);
diff --git a/kernel/bpf/verifier.c b/kernel/bpf/verifier.c
index cbbbf47e1832..fac0c3518add 100644
--- a/kernel/bpf/verifier.c
+++ b/kernel/bpf/verifier.c
@@ -13749,6 +13749,35 @@ static int __bpf_check_attach_target(struct bpf_verifier_log *log,
return 0;
}
+int bpf_check_attach_model(const struct bpf_prog *prog,
+ const struct bpf_prog *tgt_prog,
+ u32 btf_id,
+ struct btf_func_model *fmodel)
+{
+ struct attach_target target = { };
+ int ret;
+
+ ret = __bpf_check_attach_target(NULL, prog, tgt_prog, btf_id, &target);
+ if (ret)
+ return ret;
+
+ switch (prog->expected_attach_type) {
+ case BPF_TRACE_RAW_TP:
+ break;
+ case BPF_TRACE_ITER:
+ ret = btf_distill_func_proto(NULL, target.btf, target.t, target.tname, fmodel);
+ break;
+ default:
+ case BPF_MODIFY_RETURN:
+ case BPF_LSM_MAC:
+ case BPF_TRACE_FENTRY:
+ case BPF_TRACE_FEXIT:
+ ret = btf_distill_func_proto(NULL, target.btf, target.t, target.tname, fmodel);
+ }
+
+ return ret;
+}
+
int bpf_check_attach_target(struct bpf_verifier_log *log,
const struct bpf_prog *prog,
const struct bpf_prog *tgt_prog,
--
2.31.1
