1 // SPDX-License-Identifier: GPL-2.0-only 2 /* Unstable XFRM state BPF helpers. 3 * 4 * Note that it is allowed to break compatibility for these functions since the 5 * interface they are exposed through to BPF programs is explicitly unstable. 6 */ 7 8 #include <linux/bpf.h> 9 #include <linux/btf.h> 10 #include <linux/btf_ids.h> 11 #include <net/xdp.h> 12 #include <net/xfrm.h> 13 14 /* bpf_xfrm_state_opts - Options for XFRM state lookup helpers 15 * 16 * Members: 17 * @error - Out parameter, set for any errors encountered 18 * Values: 19 * -EINVAL - netns_id is less than -1 20 * -EINVAL - opts__sz isn't BPF_XFRM_STATE_OPTS_SZ 21 * -ENONET - No network namespace found for netns_id 22 * -ENOENT - No xfrm_state found 23 * @netns_id - Specify the network namespace for lookup 24 * Values: 25 * BPF_F_CURRENT_NETNS (-1) 26 * Use namespace associated with ctx 27 * [0, S32_MAX] 28 * Network Namespace ID 29 * @mark - XFRM mark to match on 30 * @daddr - Destination address to match on 31 * @spi - Security parameter index to match on 32 * @proto - IP protocol to match on (eg. IPPROTO_ESP) 33 * @family - Protocol family to match on (AF_INET/AF_INET6) 34 */ 35 struct bpf_xfrm_state_opts { 36 s32 error; 37 s32 netns_id; 38 u32 mark; 39 xfrm_address_t daddr; 40 __be32 spi; 41 u8 proto; 42 u16 family; 43 }; 44 45 enum { 46 BPF_XFRM_STATE_OPTS_SZ = sizeof(struct bpf_xfrm_state_opts), 47 }; 48 49 __bpf_kfunc_start_defs(); 50 51 /* bpf_xdp_get_xfrm_state - Get XFRM state 52 * 53 * A `struct xfrm_state *`, if found, must be released with a corresponding 54 * bpf_xdp_xfrm_state_release. 55 * 56 * Parameters: 57 * @ctx - Pointer to ctx (xdp_md) in XDP program 58 * Cannot be NULL 59 * @opts - Options for lookup (documented above) 60 * Cannot be NULL 61 * @opts__sz - Length of the bpf_xfrm_state_opts structure 62 * Must be BPF_XFRM_STATE_OPTS_SZ 63 */ 64 __bpf_kfunc struct xfrm_state * bpf_xdp_get_xfrm_state(struct xdp_md * ctx,struct bpf_xfrm_state_opts * opts,u32 opts__sz)65 bpf_xdp_get_xfrm_state(struct xdp_md *ctx, struct bpf_xfrm_state_opts *opts, u32 opts__sz) 66 { 67 struct xdp_buff *xdp = (struct xdp_buff *)ctx; 68 struct net *net = dev_net(xdp->rxq->dev); 69 struct xfrm_state *x; 70 71 if (!opts || opts__sz < sizeof(opts->error)) 72 return NULL; 73 74 if (opts__sz != BPF_XFRM_STATE_OPTS_SZ) { 75 opts->error = -EINVAL; 76 return NULL; 77 } 78 79 if (unlikely(opts->netns_id < BPF_F_CURRENT_NETNS)) { 80 opts->error = -EINVAL; 81 return NULL; 82 } 83 84 if (opts->netns_id >= 0) { 85 net = get_net_ns_by_id(net, opts->netns_id); 86 if (unlikely(!net)) { 87 opts->error = -ENONET; 88 return NULL; 89 } 90 } 91 92 x = xfrm_state_lookup(net, opts->mark, &opts->daddr, opts->spi, 93 opts->proto, opts->family); 94 95 if (opts->netns_id >= 0) 96 put_net(net); 97 if (!x) 98 opts->error = -ENOENT; 99 100 return x; 101 } 102 103 /* bpf_xdp_xfrm_state_release - Release acquired xfrm_state object 104 * 105 * This must be invoked for referenced PTR_TO_BTF_ID, and the verifier rejects 106 * the program if any references remain in the program in all of the explored 107 * states. 108 * 109 * Parameters: 110 * @x - Pointer to referenced xfrm_state object, obtained using 111 * bpf_xdp_get_xfrm_state. 112 */ bpf_xdp_xfrm_state_release(struct xfrm_state * x)113 __bpf_kfunc void bpf_xdp_xfrm_state_release(struct xfrm_state *x) 114 { 115 xfrm_state_put(x); 116 } 117 118 __bpf_kfunc_end_defs(); 119 120 BTF_KFUNCS_START(xfrm_state_kfunc_set) 121 BTF_ID_FLAGS(func, bpf_xdp_get_xfrm_state, KF_RET_NULL | KF_ACQUIRE) 122 BTF_ID_FLAGS(func, bpf_xdp_xfrm_state_release, KF_RELEASE) 123 BTF_KFUNCS_END(xfrm_state_kfunc_set) 124 125 static const struct btf_kfunc_id_set xfrm_state_xdp_kfunc_set = { 126 .owner = THIS_MODULE, 127 .set = &xfrm_state_kfunc_set, 128 }; 129 register_xfrm_state_bpf(void)130 int __init register_xfrm_state_bpf(void) 131 { 132 return register_btf_kfunc_id_set(BPF_PROG_TYPE_XDP, 133 &xfrm_state_xdp_kfunc_set); 134 } 135