11f87dcf1SSong Liu // SPDX-License-Identifier: GPL-2.0
21f87dcf1SSong Liu /* Copyright (c) 2021 Facebook */
31f87dcf1SSong Liu
41f87dcf1SSong Liu #include "vmlinux.h"
51f87dcf1SSong Liu #include <bpf/bpf_helpers.h>
61f87dcf1SSong Liu #include <bpf/bpf_tracing.h>
71f87dcf1SSong Liu
81f87dcf1SSong Liu char _license[] SEC("license") = "GPL";
91f87dcf1SSong Liu
101f87dcf1SSong Liu struct {
111f87dcf1SSong Liu __uint(type, BPF_MAP_TYPE_TASK_STORAGE);
121f87dcf1SSong Liu __uint(map_flags, BPF_F_NO_PREALLOC);
131f87dcf1SSong Liu __type(key, int);
141f87dcf1SSong Liu __type(value, __u64);
151f87dcf1SSong Liu } task_storage SEC(".maps");
161f87dcf1SSong Liu
17*eb814cf1SDelyan Kratunov int run_count = 0;
181f87dcf1SSong Liu int valid_ptr_count = 0;
191f87dcf1SSong Liu int null_ptr_count = 0;
201f87dcf1SSong Liu
211f87dcf1SSong Liu SEC("fentry/exit_creds")
BPF_PROG(trace_exit_creds,struct task_struct * task)221f87dcf1SSong Liu int BPF_PROG(trace_exit_creds, struct task_struct *task)
231f87dcf1SSong Liu {
241f87dcf1SSong Liu __u64 *ptr;
251f87dcf1SSong Liu
261f87dcf1SSong Liu ptr = bpf_task_storage_get(&task_storage, task, 0,
271f87dcf1SSong Liu BPF_LOCAL_STORAGE_GET_F_CREATE);
281f87dcf1SSong Liu if (ptr)
291f87dcf1SSong Liu __sync_fetch_and_add(&valid_ptr_count, 1);
301f87dcf1SSong Liu else
311f87dcf1SSong Liu __sync_fetch_and_add(&null_ptr_count, 1);
32*eb814cf1SDelyan Kratunov
33*eb814cf1SDelyan Kratunov __sync_fetch_and_add(&run_count, 1);
341f87dcf1SSong Liu return 0;
351f87dcf1SSong Liu }
36