You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
149 lines
3.4 KiB
149 lines
3.4 KiB
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
/*
|
|
* Copyright 2019 Google LLC
|
|
*/
|
|
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
|
|
#define TST_NO_DEFAULT_MAIN
|
|
#include "tst_test.h"
|
|
#include "tst_af_alg.h"
|
|
#include "lapi/socket.h"
|
|
|
|
int tst_alg_create(void)
|
|
{
|
|
TEST(socket(AF_ALG, SOCK_SEQPACKET, 0));
|
|
if (TST_RET >= 0)
|
|
return TST_RET;
|
|
if (TST_ERR == EAFNOSUPPORT)
|
|
tst_brk(TCONF, "kernel doesn't support AF_ALG");
|
|
tst_brk(TBROK | TTERRNO, "unexpected error creating AF_ALG socket");
|
|
return -1;
|
|
}
|
|
|
|
void tst_alg_bind_addr(int algfd, const struct sockaddr_alg *addr)
|
|
{
|
|
TEST(bind(algfd, (const struct sockaddr *)addr, sizeof(*addr)));
|
|
if (TST_RET == 0)
|
|
return;
|
|
if (TST_ERR == ENOENT) {
|
|
tst_brk(TCONF, "kernel doesn't support %s algorithm '%s'",
|
|
addr->salg_type, addr->salg_name);
|
|
}
|
|
tst_brk(TBROK | TTERRNO,
|
|
"unexpected error binding AF_ALG socket to %s algorithm '%s'",
|
|
addr->salg_type, addr->salg_name);
|
|
}
|
|
|
|
static void init_sockaddr_alg(struct sockaddr_alg *addr,
|
|
const char *algtype, const char *algname)
|
|
{
|
|
memset(addr, 0, sizeof(*addr));
|
|
|
|
addr->salg_family = AF_ALG;
|
|
|
|
strncpy((char *)addr->salg_type, algtype, sizeof(addr->salg_type));
|
|
if (addr->salg_type[sizeof(addr->salg_type) - 1] != '\0')
|
|
tst_brk(TBROK, "algorithm type too long: '%s'", algtype);
|
|
|
|
strncpy((char *)addr->salg_name, algname, sizeof(addr->salg_name));
|
|
if (addr->salg_name[sizeof(addr->salg_name) - 1] != '\0')
|
|
tst_brk(TBROK, "algorithm name too long: '%s'", algname);
|
|
}
|
|
|
|
void tst_alg_bind(int algfd, const char *algtype, const char *algname)
|
|
{
|
|
struct sockaddr_alg addr;
|
|
|
|
init_sockaddr_alg(&addr, algtype, algname);
|
|
|
|
tst_alg_bind_addr(algfd, &addr);
|
|
}
|
|
|
|
bool tst_have_alg(const char *algtype, const char *algname)
|
|
{
|
|
int algfd;
|
|
struct sockaddr_alg addr;
|
|
bool have_alg = true;
|
|
|
|
algfd = tst_alg_create();
|
|
|
|
init_sockaddr_alg(&addr, algtype, algname);
|
|
|
|
TEST(bind(algfd, (const struct sockaddr *)&addr, sizeof(addr)));
|
|
if (TST_RET != 0) {
|
|
if (TST_ERR != ENOENT) {
|
|
tst_brk(TBROK | TTERRNO,
|
|
"unexpected error binding AF_ALG socket to %s algorithm '%s'",
|
|
algtype, algname);
|
|
}
|
|
have_alg = false;
|
|
}
|
|
|
|
close(algfd);
|
|
return have_alg;
|
|
}
|
|
|
|
void tst_require_alg(const char *algtype, const char *algname)
|
|
{
|
|
int algfd = tst_alg_create();
|
|
|
|
tst_alg_bind(algfd, algtype, algname);
|
|
|
|
close(algfd);
|
|
}
|
|
|
|
void tst_alg_setkey(int algfd, const uint8_t *key, unsigned int keylen)
|
|
{
|
|
uint8_t *keybuf = NULL;
|
|
unsigned int i;
|
|
|
|
if (key == NULL) {
|
|
/* generate a random key */
|
|
keybuf = SAFE_MALLOC(keylen);
|
|
for (i = 0; i < keylen; i++)
|
|
keybuf[i] = rand();
|
|
key = keybuf;
|
|
}
|
|
TEST(setsockopt(algfd, SOL_ALG, ALG_SET_KEY, key, keylen));
|
|
if (TST_RET != 0) {
|
|
tst_brk(TBROK | TTERRNO,
|
|
"unexpected error setting key (len=%u)", keylen);
|
|
}
|
|
free(keybuf);
|
|
}
|
|
|
|
int tst_alg_accept(int algfd)
|
|
{
|
|
TEST(accept(algfd, NULL, NULL));
|
|
if (TST_RET < 0) {
|
|
tst_brk(TBROK | TTERRNO,
|
|
"unexpected error accept()ing AF_ALG request socket");
|
|
}
|
|
return TST_RET;
|
|
}
|
|
|
|
int tst_alg_setup(const char *algtype, const char *algname,
|
|
const uint8_t *key, unsigned int keylen)
|
|
{
|
|
int algfd = tst_alg_create();
|
|
|
|
tst_alg_bind(algfd, algtype, algname);
|
|
|
|
if (keylen != 0)
|
|
tst_alg_setkey(algfd, key, keylen);
|
|
|
|
return algfd;
|
|
}
|
|
|
|
int tst_alg_setup_reqfd(const char *algtype, const char *algname,
|
|
const uint8_t *key, unsigned int keylen)
|
|
{
|
|
int algfd = tst_alg_setup(algtype, algname, key, keylen);
|
|
int reqfd = tst_alg_accept(algfd);
|
|
|
|
close(algfd);
|
|
return reqfd;
|
|
}
|