-
Notifications
You must be signed in to change notification settings - Fork 98
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
selftests: bpf: crypto skcipher algo selftests
Add simple tc hook selftests to show the way to work with new crypto BPF API. Some weird structre and map are added to setup program to make verifier happy about dynptr initialization from memory. Simple AES-ECB algo is used to demonstrate encryption and decryption of fixed size buffers. Signed-off-by: Vadim Fedorenko <vadfed@meta.com>
- Loading branch information
1 parent
ac21267
commit 0d58f7a
Showing
4 changed files
with
413 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,148 @@ | ||
// SPDX-License-Identifier: GPL-2.0 | ||
/* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */ | ||
|
||
#include <sys/types.h> | ||
#include <sys/socket.h> | ||
#include <net/if.h> | ||
#include <linux/in6.h> | ||
|
||
#include "test_progs.h" | ||
#include "network_helpers.h" | ||
#include "crypto_sanity.skel.h" | ||
|
||
#define NS_TEST "crypto_sanity_ns" | ||
#define IPV6_IFACE_ADDR "face::1" | ||
#define UDP_TEST_PORT 7777 | ||
static const char plain_text[] = "stringtoencrypt0"; | ||
static const char crypted_data[] = "\x5B\x59\x39\xEA\xD9\x7A\x2D\xAD\xA7\xE0\x43" \ | ||
"\x37\x8A\x77\x17\xB2"; | ||
|
||
void test_crypto_sanity(void) | ||
{ | ||
LIBBPF_OPTS(bpf_tc_hook, qdisc_hook, .attach_point = BPF_TC_EGRESS); | ||
LIBBPF_OPTS(bpf_tc_opts, tc_attach_enc); | ||
LIBBPF_OPTS(bpf_tc_opts, tc_attach_dec); | ||
LIBBPF_OPTS(bpf_test_run_opts, opts, | ||
.repeat = 1, | ||
); | ||
struct nstoken *nstoken = NULL; | ||
struct crypto_sanity *skel; | ||
struct sockaddr_in6 addr; | ||
int sockfd, err, pfd; | ||
socklen_t addrlen; | ||
|
||
skel = crypto_sanity__open(); | ||
if (!ASSERT_OK_PTR(skel, "skel open")) | ||
return; | ||
|
||
bpf_program__set_autoload(skel->progs.crypto_accuire, true); | ||
|
||
err = crypto_sanity__load(skel); | ||
if (!ASSERT_ERR(err, "crypto_accuire unexpected load success")) | ||
goto fail; | ||
|
||
crypto_sanity__destroy(skel); | ||
|
||
skel = crypto_sanity__open(); | ||
if (!ASSERT_OK_PTR(skel, "skel open")) | ||
return; | ||
|
||
bpf_program__set_autoload(skel->progs.crypto_accuire, false); | ||
|
||
SYS(fail, "ip netns add %s", NS_TEST); | ||
SYS(fail, "ip -net %s -6 addr add %s/128 dev lo nodad", NS_TEST, IPV6_IFACE_ADDR); | ||
SYS(fail, "ip -net %s link set dev lo up", NS_TEST); | ||
|
||
err = crypto_sanity__load(skel); | ||
if (!ASSERT_OK(err, "crypto_sanity__load")) | ||
goto fail; | ||
|
||
nstoken = open_netns(NS_TEST); | ||
if (!ASSERT_OK_PTR(nstoken, "open_netns")) | ||
goto fail; | ||
|
||
qdisc_hook.ifindex = if_nametoindex("lo"); | ||
if (!ASSERT_GT(qdisc_hook.ifindex, 0, "if_nametoindex lo")) | ||
goto fail; | ||
|
||
err = crypto_sanity__attach(skel); | ||
if (!ASSERT_OK(err, "crypto_sanity__attach")) | ||
goto fail; | ||
|
||
pfd = bpf_program__fd(skel->progs.crypto_release); | ||
if (!ASSERT_GT(pfd, 0, "crypto_release fd")) | ||
goto fail; | ||
|
||
err = bpf_prog_test_run_opts(pfd, &opts); | ||
if (!ASSERT_OK(err, "crypto_release") || | ||
!ASSERT_OK(opts.retval, "crypto_release retval")) | ||
goto fail; | ||
|
||
pfd = bpf_program__fd(skel->progs.skb_crypto_setup); | ||
if (!ASSERT_GT(pfd, 0, "skb_crypto_setup fd")) | ||
goto fail; | ||
|
||
err = bpf_prog_test_run_opts(pfd, &opts); | ||
if (!ASSERT_OK(err, "skb_crypto_setup") || | ||
!ASSERT_OK(opts.retval, "skb_crypto_setup retval")) | ||
goto fail; | ||
|
||
if (!ASSERT_OK(skel->bss->status, "skb_crypto_setup status")) | ||
goto fail; | ||
|
||
err = bpf_tc_hook_create(&qdisc_hook); | ||
if (!ASSERT_OK(err, "create qdisc hook")) | ||
goto fail; | ||
|
||
addrlen = sizeof(addr); | ||
err = make_sockaddr(AF_INET6, IPV6_IFACE_ADDR, UDP_TEST_PORT, | ||
(void *)&addr, &addrlen); | ||
if (!ASSERT_OK(err, "make_sockaddr")) | ||
goto fail; | ||
|
||
tc_attach_dec.prog_fd = bpf_program__fd(skel->progs.decrypt_sanity); | ||
err = bpf_tc_attach(&qdisc_hook, &tc_attach_dec); | ||
if (!ASSERT_OK(err, "attach decrypt filter")) | ||
goto fail; | ||
|
||
sockfd = socket(AF_INET6, SOCK_DGRAM, 0); | ||
if (!ASSERT_NEQ(sockfd, -1, "decrypt socket")) | ||
goto fail; | ||
err = sendto(sockfd, crypted_data, 16, 0, (void *)&addr, addrlen); | ||
close(sockfd); | ||
if (!ASSERT_EQ(err, 16, "decrypt send")) | ||
goto fail; | ||
|
||
bpf_tc_detach(&qdisc_hook, &tc_attach_dec); | ||
if (!ASSERT_OK(skel->bss->status, "decrypt status")) | ||
goto fail; | ||
if (!ASSERT_STRNEQ(skel->bss->dst, plain_text, sizeof(plain_text), "decrypt")) | ||
goto fail; | ||
|
||
tc_attach_enc.prog_fd = bpf_program__fd(skel->progs.encrypt_sanity); | ||
err = bpf_tc_attach(&qdisc_hook, &tc_attach_enc); | ||
if (!ASSERT_OK(err, "attach encrypt filter")) | ||
goto fail; | ||
|
||
sockfd = socket(AF_INET6, SOCK_DGRAM, 0); | ||
if (!ASSERT_NEQ(sockfd, -1, "encrypt socket")) | ||
goto fail; | ||
err = sendto(sockfd, plain_text, 16, 0, (void *)&addr, addrlen); | ||
close(sockfd); | ||
if (!ASSERT_EQ(err, 16, "encrypt send")) | ||
goto fail; | ||
|
||
bpf_tc_detach(&qdisc_hook, &tc_attach_enc); | ||
if (!ASSERT_OK(skel->bss->status, "encrypt status")) | ||
goto fail; | ||
if (!ASSERT_STRNEQ(skel->bss->dst, crypted_data, sizeof(crypted_data), "encrypt")) | ||
goto fail; | ||
|
||
fail: | ||
if (nstoken) { | ||
bpf_tc_hook_destroy(&qdisc_hook); | ||
close_netns(nstoken); | ||
} | ||
SYS_NOFAIL("ip netns del " NS_TEST " &> /dev/null"); | ||
crypto_sanity__destroy(skel); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,69 @@ | ||
/* SPDX-License-Identifier: GPL-2.0 */ | ||
/* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */ | ||
|
||
#ifndef _CRYPTO_COMMON_H | ||
#define _CRYPTO_COMMON_H | ||
|
||
#include "errno.h" | ||
#include <stdbool.h> | ||
|
||
struct bpf_crypto_lskcipher_ctx *bpf_crypto_lskcipher_ctx_create(const char *algo__str, | ||
const struct bpf_dynptr *key, | ||
int *err) __ksym; | ||
struct bpf_crypto_lskcipher_ctx *bpf_crypto_lskcipher_ctx_acquire(struct bpf_crypto_lskcipher_ctx *ctx) __ksym; | ||
void bpf_crypto_lskcipher_ctx_release(struct bpf_crypto_lskcipher_ctx *ctx) __ksym; | ||
int bpf_crypto_lskcipher_encrypt(struct bpf_crypto_lskcipher_ctx *ctx, | ||
const struct bpf_dynptr *src, struct bpf_dynptr *dst, | ||
struct bpf_dynptr *iv) __ksym; | ||
int bpf_crypto_lskcipher_decrypt(struct bpf_crypto_lskcipher_ctx *ctx, | ||
const struct bpf_dynptr *src, struct bpf_dynptr *dst, | ||
struct bpf_dynptr *iv) __ksym; | ||
|
||
struct __crypto_lskcipher_ctx_value { | ||
struct bpf_crypto_lskcipher_ctx __kptr * ctx; | ||
}; | ||
|
||
struct array_map { | ||
__uint(type, BPF_MAP_TYPE_ARRAY); | ||
__type(key, int); | ||
__type(value, struct __crypto_lskcipher_ctx_value); | ||
__uint(max_entries, 1); | ||
} __crypto_lskcipher_ctx_map SEC(".maps"); | ||
|
||
static inline struct __crypto_lskcipher_ctx_value *crypto_lskcipher_ctx_value_lookup(void) | ||
{ | ||
u32 key = 0; | ||
|
||
return bpf_map_lookup_elem(&__crypto_lskcipher_ctx_map, &key); | ||
} | ||
|
||
static inline int crypto_lskcipher_ctx_insert(struct bpf_crypto_lskcipher_ctx *ctx) | ||
{ | ||
struct __crypto_lskcipher_ctx_value local, *v; | ||
struct bpf_crypto_lskcipher_ctx *old; | ||
u32 key = 0; | ||
int err; | ||
|
||
local.ctx = NULL; | ||
err = bpf_map_update_elem(&__crypto_lskcipher_ctx_map, &key, &local, 0); | ||
if (err) { | ||
bpf_crypto_lskcipher_ctx_release(ctx); | ||
return err; | ||
} | ||
|
||
v = bpf_map_lookup_elem(&__crypto_lskcipher_ctx_map, &key); | ||
if (!v) { | ||
bpf_crypto_lskcipher_ctx_release(ctx); | ||
return -ENOENT; | ||
} | ||
|
||
old = bpf_kptr_xchg(&v->ctx, ctx); | ||
if (old) { | ||
bpf_crypto_lskcipher_ctx_release(old); | ||
return -EEXIST; | ||
} | ||
|
||
return 0; | ||
} | ||
|
||
#endif /* _CRYPTO_COMMON_H */ |
Oops, something went wrong.