1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2021 Facebook */
5 #include <bpf/bpf_helpers.h>
6 #include <bpf/bpf_tracing.h>
12 char _license[] SEC("license") = "GPL";
17 __uint(type, BPF_MAP_TYPE_TASK_STORAGE);
18 __uint(map_flags, BPF_F_NO_PREALLOC);
24 __uint(type, BPF_MAP_TYPE_TASK_STORAGE);
25 __uint(map_flags, BPF_F_NO_PREALLOC);
30 SEC("fentry/bpf_local_storage_lookup")
31 int BPF_PROG(on_lookup)
33 struct task_struct *task = bpf_get_current_task_btf();
35 if (!test_pid || task->pid != test_pid)
38 /* The bpf_task_storage_delete will call
39 * bpf_local_storage_lookup. The prog->active will
42 bpf_task_storage_delete(&map_a, task);
43 bpf_task_storage_delete(&map_b, task);
47 SEC("fentry/bpf_local_storage_update")
48 int BPF_PROG(on_update)
50 struct task_struct *task = bpf_get_current_task_btf();
53 if (!test_pid || task->pid != test_pid)
56 ptr = bpf_task_storage_get(&map_a, task, 0,
57 BPF_LOCAL_STORAGE_GET_F_CREATE);
58 /* ptr will not be NULL when it is called from
59 * the bpf_task_storage_get(&map_b,...F_CREATE) in
60 * the BPF_PROG(on_enter) below. It is because
61 * the value can be found in map_a and the kernel
62 * does not need to acquire any spin_lock.
68 err = bpf_task_storage_delete(&map_a, task);
73 /* This will still fail because map_b is empty and
74 * this BPF_PROG(on_update) has failed to acquire
75 * the percpu busy lock => meaning potential
76 * deadlock is detected and it will fail to create
79 ptr = bpf_task_storage_get(&map_b, task, 0,
80 BPF_LOCAL_STORAGE_GET_F_CREATE);
87 SEC("tp_btf/sys_enter")
88 int BPF_PROG(on_enter, struct pt_regs *regs, long id)
90 struct task_struct *task;
93 task = bpf_get_current_task_btf();
94 if (!test_pid || task->pid != test_pid)
97 ptr = bpf_task_storage_get(&map_a, task, 0,
98 BPF_LOCAL_STORAGE_GET_F_CREATE);
102 ptr = bpf_task_storage_get(&map_b, task, 0,
103 BPF_LOCAL_STORAGE_GET_F_CREATE);