Quelle  crypto_common.h   Sprache: C

/* SPDX-License-Identifier: GPL-2.0 */
/* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */

#ifndef _CRYPTO_COMMON_H
#define _CRYPTO_COMMON_H

#include "errno.h"
#include <stdbool.h>

struct bpf_crypto_ctx *bpf_crypto_ctx_create(const struct bpf_crypto_params *params,
          u32 params__sz, int *err) __ksym;
struct bpf_crypto_ctx *bpf_crypto_ctx_acquire(struct bpf_crypto_ctx *ctx) __ksym;
void bpf_crypto_ctx_release(struct bpf_crypto_ctx *ctx) __ksym;
int bpf_crypto_encrypt(struct bpf_crypto_ctx *ctx, const struct bpf_dynptr *src,
         const struct bpf_dynptr *dst, const struct bpf_dynptr *iv) __ksym;
int bpf_crypto_decrypt(struct bpf_crypto_ctx *ctx, const struct bpf_dynptr *src,
         const struct bpf_dynptr *dst, const struct bpf_dynptr *iv) __ksym;

struct __crypto_ctx_value {
 struct bpf_crypto_ctx __kptr * ctx;
};

struct array_map {
 __uint(type, BPF_MAP_TYPE_ARRAY);
 __type(key, int);
 __type(value, struct __crypto_ctx_value);
 __uint(max_entries, 1);
} __crypto_ctx_map SEC(".maps");

static inline struct __crypto_ctx_value *crypto_ctx_value_lookup(void)
{
 u32 key = 0;

 return bpf_map_lookup_elem(&__crypto_ctx_map, &key);
}

static inline int crypto_ctx_insert(struct bpf_crypto_ctx *ctx)
{
 struct __crypto_ctx_value local, *v;
 struct bpf_crypto_ctx *old;
 u32 key = 0;
 int err;

 local.ctx = NULL;
 err = bpf_map_update_elem(&__crypto_ctx_map, &key, &local, 0);
 if (err) {
  bpf_crypto_ctx_release(ctx);
  return err;
 }

 v = bpf_map_lookup_elem(&__crypto_ctx_map, &key);
 if (!v) {
  bpf_crypto_ctx_release(ctx);
  return -ENOENT;
 }

 old = bpf_kptr_xchg(&v->ctx, ctx);
 if (old) {
  bpf_crypto_ctx_release(old);
  return -EEXIST;
 }

 return 0;
}

#endif /* _CRYPTO_COMMON_H */