Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 | // SPDX-License-Identifier: GPL-2.0 /* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */ #include <vmlinux.h> #include <bpf/bpf_tracing.h> #include <bpf/bpf_helpers.h> #include <bpf/bpf_core_read.h> #include "../bpf_experimental.h" #include "../bpf_testmod/bpf_testmod_kfunc.h" struct node_data { long key; long data; struct bpf_rb_node node; }; struct plain_local { long key; long data; }; struct map_value { struct prog_test_ref_kfunc *not_kptr; struct prog_test_ref_kfunc __kptr *val; struct node_data __kptr *node; struct plain_local __kptr *plain; }; /* This is necessary so that LLVM generates BTF for node_data struct * If it's not included, a fwd reference for node_data will be generated but * no struct. Example BTF of "node" field in map_value when not included: * * [10] PTR '(anon)' type_id=35 * [34] FWD 'node_data' fwd_kind=struct * [35] TYPE_TAG 'kptr_ref' type_id=34 * * (with no node_data struct defined) * Had to do the same w/ bpf_kfunc_call_test_release below */ struct node_data *just_here_because_btf_bug; struct { __uint(type, BPF_MAP_TYPE_ARRAY); __type(key, int); __type(value, struct map_value); __uint(max_entries, 2); } some_nodes SEC(".maps"); static int create_and_stash(int idx, int val) { struct map_value *mapval; struct node_data *res; mapval = bpf_map_lookup_elem(&some_nodes, &idx); if (!mapval) return 1; res = bpf_obj_new(typeof(*res)); if (!res) return 1; res->key = val; res = bpf_kptr_xchg(&mapval->node, res); if (res) bpf_obj_drop(res); return 0; } SEC("tc") long stash_rb_nodes(void *ctx) { return create_and_stash(0, 41) ?: create_and_stash(1, 42); } SEC("tc") long stash_plain(void *ctx) { struct map_value *mapval; struct plain_local *res; int idx = 0; mapval = bpf_map_lookup_elem(&some_nodes, &idx); if (!mapval) return 1; res = bpf_obj_new(typeof(*res)); if (!res) return 1; res->key = 41; res = bpf_kptr_xchg(&mapval->plain, res); if (res) bpf_obj_drop(res); return 0; } SEC("tc") long unstash_rb_node(void *ctx) { struct map_value *mapval; struct node_data *res; long retval; int key = 1; mapval = bpf_map_lookup_elem(&some_nodes, &key); if (!mapval) return 1; res = bpf_kptr_xchg(&mapval->node, NULL); if (res) { retval = res->key; bpf_obj_drop(res); return retval; } return 1; } SEC("tc") long stash_test_ref_kfunc(void *ctx) { struct prog_test_ref_kfunc *res; struct map_value *mapval; int key = 0; mapval = bpf_map_lookup_elem(&some_nodes, &key); if (!mapval) return 1; res = bpf_kptr_xchg(&mapval->val, NULL); if (res) bpf_kfunc_call_test_release(res); return 0; } char _license[] SEC("license") = "GPL"; |