Message ID | 20220811001654.1316689-3-kuifeng@fb.com (mailing list archive) |
---|---|
State | Superseded |
Delegated to: | BPF |
Headers | show |
Series | Parameterize task iterators. | expand |
On 8/10/22 5:16 PM, Kui-Feng Lee wrote: > Add new fields to bpf_link_info that users can query it through > bpf_obj_get_info_by_fd(). > > Signed-off-by: Kui-Feng Lee <kuifeng@fb.com> > --- > include/uapi/linux/bpf.h | 4 ++++ > kernel/bpf/task_iter.c | 18 ++++++++++++++++++ > tools/include/uapi/linux/bpf.h | 4 ++++ > 3 files changed, 26 insertions(+) > > diff --git a/include/uapi/linux/bpf.h b/include/uapi/linux/bpf.h > index 6328aca0cf5c..627a16981c90 100644 > --- a/include/uapi/linux/bpf.h > +++ b/include/uapi/linux/bpf.h > @@ -6143,6 +6143,10 @@ struct bpf_link_info { > struct { > __u32 map_id; > } map; > + struct { > + __u32 tid; > + __u32 tgid; pid/tgid or tid/pid? > + } task; Please use another union outside of struct { ... } map. Please see cgroup_iter patch for details: https://lore.kernel.org/bpf/20220812202802.3774257-2-haoluo@google.com/ > }; > } iter; > struct { > diff --git a/kernel/bpf/task_iter.c b/kernel/bpf/task_iter.c > index f2e21efe075d..d392b46c6d19 100644 > --- a/kernel/bpf/task_iter.c > +++ b/kernel/bpf/task_iter.c > @@ -606,6 +606,21 @@ static const struct bpf_iter_seq_info task_seq_info = { > .seq_priv_size = sizeof(struct bpf_iter_seq_task_info), > }; > > +static int bpf_iter_fill_link_info(const struct bpf_iter_aux_info *aux, struct bpf_link_info *info) > +{ > + switch (aux->task.type) { > + case BPF_TASK_ITER_TID: > + info->iter.task.tid = aux->task.tid; > + break; > + case BPF_TASK_ITER_TGID: > + info->iter.task.tgid = aux->task.tgid; > + break; > + default: > + break; > + } > + return 0; > +} > + > static struct bpf_iter_reg task_reg_info = { > .target = "task", > .attach_target = bpf_iter_attach_task, > @@ -616,6 +631,7 @@ static struct bpf_iter_reg task_reg_info = { > PTR_TO_BTF_ID_OR_NULL }, > }, > .seq_info = &task_seq_info, > + .fill_link_info = bpf_iter_fill_link_info, How about show_fdinfo? > }; > > static const struct bpf_iter_seq_info task_file_seq_info = { > @@ -637,6 +653,7 @@ static struct bpf_iter_reg task_file_reg_info = { > PTR_TO_BTF_ID_OR_NULL }, > }, > .seq_info = &task_file_seq_info, > + .fill_link_info = bpf_iter_fill_link_info, > }; > > static const struct bpf_iter_seq_info task_vma_seq_info = { > @@ -658,6 +675,7 @@ static struct bpf_iter_reg task_vma_reg_info = { > PTR_TO_BTF_ID_OR_NULL }, > }, > .seq_info = &task_vma_seq_info, > + .fill_link_info = bpf_iter_fill_link_info, > }; > [...]
diff --git a/include/uapi/linux/bpf.h b/include/uapi/linux/bpf.h index 6328aca0cf5c..627a16981c90 100644 --- a/include/uapi/linux/bpf.h +++ b/include/uapi/linux/bpf.h @@ -6143,6 +6143,10 @@ struct bpf_link_info { struct { __u32 map_id; } map; + struct { + __u32 tid; + __u32 tgid; + } task; }; } iter; struct { diff --git a/kernel/bpf/task_iter.c b/kernel/bpf/task_iter.c index f2e21efe075d..d392b46c6d19 100644 --- a/kernel/bpf/task_iter.c +++ b/kernel/bpf/task_iter.c @@ -606,6 +606,21 @@ static const struct bpf_iter_seq_info task_seq_info = { .seq_priv_size = sizeof(struct bpf_iter_seq_task_info), }; +static int bpf_iter_fill_link_info(const struct bpf_iter_aux_info *aux, struct bpf_link_info *info) +{ + switch (aux->task.type) { + case BPF_TASK_ITER_TID: + info->iter.task.tid = aux->task.tid; + break; + case BPF_TASK_ITER_TGID: + info->iter.task.tgid = aux->task.tgid; + break; + default: + break; + } + return 0; +} + static struct bpf_iter_reg task_reg_info = { .target = "task", .attach_target = bpf_iter_attach_task, @@ -616,6 +631,7 @@ static struct bpf_iter_reg task_reg_info = { PTR_TO_BTF_ID_OR_NULL }, }, .seq_info = &task_seq_info, + .fill_link_info = bpf_iter_fill_link_info, }; static const struct bpf_iter_seq_info task_file_seq_info = { @@ -637,6 +653,7 @@ static struct bpf_iter_reg task_file_reg_info = { PTR_TO_BTF_ID_OR_NULL }, }, .seq_info = &task_file_seq_info, + .fill_link_info = bpf_iter_fill_link_info, }; static const struct bpf_iter_seq_info task_vma_seq_info = { @@ -658,6 +675,7 @@ static struct bpf_iter_reg task_vma_reg_info = { PTR_TO_BTF_ID_OR_NULL }, }, .seq_info = &task_vma_seq_info, + .fill_link_info = bpf_iter_fill_link_info, }; BPF_CALL_5(bpf_find_vma, struct task_struct *, task, u64, start, diff --git a/tools/include/uapi/linux/bpf.h b/tools/include/uapi/linux/bpf.h index 6328aca0cf5c..627a16981c90 100644 --- a/tools/include/uapi/linux/bpf.h +++ b/tools/include/uapi/linux/bpf.h @@ -6143,6 +6143,10 @@ struct bpf_link_info { struct { __u32 map_id; } map; + struct { + __u32 tid; + __u32 tgid; + } task; }; } iter; struct {
Add new fields to bpf_link_info that users can query it through bpf_obj_get_info_by_fd(). Signed-off-by: Kui-Feng Lee <kuifeng@fb.com> --- include/uapi/linux/bpf.h | 4 ++++ kernel/bpf/task_iter.c | 18 ++++++++++++++++++ tools/include/uapi/linux/bpf.h | 4 ++++ 3 files changed, 26 insertions(+)