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


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

// 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";