1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */ 3 4 #include <vmlinux.h> 5 #include <bpf/bpf_tracing.h> 6 #include <bpf/bpf_helpers.h> 7 #include "bpf_misc.h" 8 9 #include "nested_trust_common.h" 10 11 char _license[] SEC("license") = "GPL"; 12 13 struct { 14 __uint(type, BPF_MAP_TYPE_SK_STORAGE); 15 __uint(map_flags, BPF_F_NO_PREALLOC); 16 __type(key, int); 17 __type(value, u64); 18 } sk_storage_map SEC(".maps"); 19 20 SEC("tp_btf/task_newtask") 21 __success 22 int BPF_PROG(test_read_cpumask, struct task_struct *task, u64 clone_flags) 23 { 24 bpf_cpumask_test_cpu(0, task->cpus_ptr); 25 return 0; 26 } 27 28 SEC("tp_btf/tcp_probe") 29 __success 30 int BPF_PROG(test_skb_field, struct sock *sk, struct sk_buff *skb) 31 { 32 bpf_sk_storage_get(&sk_storage_map, skb->sk, 0, 0); 33 return 0; 34 } 35 36 SEC("tp_btf/task_newtask") 37 __success 38 int BPF_PROG(test_nested_offset, struct task_struct *task, u64 clone_flags) 39 { 40 bpf_cpumask_first_zero(&task->cpus_mask); 41 return 0; 42 } 43