tools/testing/selftests/bpf/progs/nested_trust_success.c

   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 }