1 /* SPDX-License-Identifier: GPL-2.0 */ 2 /* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */ 3 4 #ifndef _CRYPTO_COMMON_H 5 #define _CRYPTO_COMMON_H 6 7 #include "errno.h" 8 #include <stdbool.h> 9 10 struct bpf_crypto_ctx *bpf_crypto_ctx_create(const struct bpf_crypto_params *params, 11 u32 params__sz, int *err) __ksym; 12 struct bpf_crypto_ctx *bpf_crypto_ctx_acquire(struct bpf_crypto_ctx *ctx) __ksym; 13 void bpf_crypto_ctx_release(struct bpf_crypto_ctx *ctx) __ksym; 14 int bpf_crypto_encrypt(struct bpf_crypto_ctx *ctx, const struct bpf_dynptr *src, 15 const struct bpf_dynptr *dst, const struct bpf_dynptr *iv) __ksym; 16 int bpf_crypto_decrypt(struct bpf_crypto_ctx *ctx, const struct bpf_dynptr *src, 17 const struct bpf_dynptr *dst, const struct bpf_dynptr *iv) __ksym; 18 19 struct __crypto_ctx_value { 20 struct bpf_crypto_ctx __kptr * ctx; 21 }; 22 23 struct array_map { 24 __uint(type, BPF_MAP_TYPE_ARRAY); 25 __type(key, int); 26 __type(value, struct __crypto_ctx_value); 27 __uint(max_entries, 1); 28 } __crypto_ctx_map SEC(".maps"); 29 30 static inline struct __crypto_ctx_value *crypto_ctx_value_lookup(void) 31 { 32 u32 key = 0; 33 34 return bpf_map_lookup_elem(&__crypto_ctx_map, &key); 35 } 36 37 static inline int crypto_ctx_insert(struct bpf_crypto_ctx *ctx) 38 { 39 struct __crypto_ctx_value local, *v; 40 struct bpf_crypto_ctx *old; 41 u32 key = 0; 42 int err; 43 44 local.ctx = NULL; 45 err = bpf_map_update_elem(&__crypto_ctx_map, &key, &local, 0); 46 if (err) { 47 bpf_crypto_ctx_release(ctx); 48 return err; 49 } 50 51 v = bpf_map_lookup_elem(&__crypto_ctx_map, &key); 52 if (!v) { 53 bpf_crypto_ctx_release(ctx); 54 return -ENOENT; 55 } 56 57 old = bpf_kptr_xchg(&v->ctx, ctx); 58 if (old) { 59 bpf_crypto_ctx_release(old); 60 return -EEXIST; 61 } 62 63 return 0; 64 } 65 66 #endif /* _CRYPTO_COMMON_H */ 67