linux/tools/testing/selftests/bpf/progs/crypto_basic.c

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

#include "vmlinux.h"
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>
#include "bpf_misc.h"
#include "bpf_kfuncs.h"
#include "crypto_common.h"

int status;
SEC("syscall")
int crypto_release(void *ctx)
{
	struct bpf_crypto_params params = {
		.type = "skcipher",
		.algo = "ecb(aes)",
		.key_len = 16,
	};

	struct bpf_crypto_ctx *cctx;
	int err = 0;

	status = 0;

	cctx = bpf_crypto_ctx_create(&params, sizeof(params), &err);

	if (!cctx) {
		status = err;
		return 0;
	}

	bpf_crypto_ctx_release(cctx);

	return 0;
}

SEC("syscall")
__failure __msg("Unreleased reference")
int crypto_acquire(void *ctx)
{
	struct bpf_crypto_params params = {
		.type = "skcipher",
		.algo = "ecb(aes)",
		.key_len = 16,
	};
	struct bpf_crypto_ctx *cctx;
	int err = 0;

	status = 0;

	cctx = bpf_crypto_ctx_create(&params, sizeof(params), &err);

	if (!cctx) {
		status = err;
		return 0;
	}

	cctx = bpf_crypto_ctx_acquire(cctx);
	if (!cctx)
		return -EINVAL;

	bpf_crypto_ctx_release(cctx);

	return 0;
}

char __license[] SEC("license") = "GPL";