2196 lines
52 KiB
C
2196 lines
52 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/*
|
|
* ipsec.c - Check xfrm on veth inside a net-ns.
|
|
* Copyright (c) 2018 Dmitry Safonov
|
|
*/
|
|
|
|
#define _GNU_SOURCE
|
|
|
|
#include <arpa/inet.h>
|
|
#include <asm/types.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <limits.h>
|
|
#include <linux/limits.h>
|
|
#include <linux/netlink.h>
|
|
#include <linux/random.h>
|
|
#include <linux/rtnetlink.h>
|
|
#include <linux/veth.h>
|
|
#include <linux/xfrm.h>
|
|
#include <netinet/in.h>
|
|
#include <net/if.h>
|
|
#include <sched.h>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/mman.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/syscall.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
|
|
#include "../kselftest.h"
|
|
|
|
#define printk(fmt, ...) \
|
|
ksft_print_msg("%d[%u] " fmt "\n", getpid(), __LINE__, ##__VA_ARGS__)
|
|
|
|
#define pr_err(fmt, ...) printk(fmt ": %m", ##__VA_ARGS__)
|
|
|
|
#define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
|
|
#define BUILD_BUG_ON(condition) ((void)sizeof(char[1 - 2*!!(condition)]))
|
|
|
|
#define IPV4_STR_SZ 16 /* xxx.xxx.xxx.xxx is longest + \0 */
|
|
#define MAX_PAYLOAD 2048
|
|
#define XFRM_ALGO_KEY_BUF_SIZE 512
|
|
#define MAX_PROCESSES (1 << 14) /* /16 mask divided by /30 subnets */
|
|
#define INADDR_A ((in_addr_t) 0x0a000000) /* 10.0.0.0 */
|
|
#define INADDR_B ((in_addr_t) 0xc0a80000) /* 192.168.0.0 */
|
|
|
|
/* /30 mask for one veth connection */
|
|
#define PREFIX_LEN 30
|
|
#define child_ip(nr) (4*nr + 1)
|
|
#define grchild_ip(nr) (4*nr + 2)
|
|
|
|
#define VETH_FMT "ktst-%d"
|
|
#define VETH_LEN 12
|
|
|
|
static int nsfd_parent = -1;
|
|
static int nsfd_childa = -1;
|
|
static int nsfd_childb = -1;
|
|
static long page_size;
|
|
|
|
/*
|
|
* ksft_cnt is static in kselftest, so isn't shared with children.
|
|
* We have to send a test result back to parent and count there.
|
|
* results_fd is a pipe with test feedback from children.
|
|
*/
|
|
static int results_fd[2];
|
|
|
|
const unsigned int ping_delay_nsec = 50 * 1000 * 1000;
|
|
const unsigned int ping_timeout = 300;
|
|
const unsigned int ping_count = 100;
|
|
const unsigned int ping_success = 80;
|
|
|
|
static void randomize_buffer(void *buf, size_t buflen)
|
|
{
|
|
int *p = (int *)buf;
|
|
size_t words = buflen / sizeof(int);
|
|
size_t leftover = buflen % sizeof(int);
|
|
|
|
if (!buflen)
|
|
return;
|
|
|
|
while (words--)
|
|
*p++ = rand();
|
|
|
|
if (leftover) {
|
|
int tmp = rand();
|
|
|
|
memcpy(buf + buflen - leftover, &tmp, leftover);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
static int unshare_open(void)
|
|
{
|
|
const char *netns_path = "/proc/self/ns/net";
|
|
int fd;
|
|
|
|
if (unshare(CLONE_NEWNET) != 0) {
|
|
pr_err("unshare()");
|
|
return -1;
|
|
}
|
|
|
|
fd = open(netns_path, O_RDONLY);
|
|
if (fd <= 0) {
|
|
pr_err("open(%s)", netns_path);
|
|
return -1;
|
|
}
|
|
|
|
return fd;
|
|
}
|
|
|
|
static int switch_ns(int fd)
|
|
{
|
|
if (setns(fd, CLONE_NEWNET)) {
|
|
pr_err("setns()");
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Running the test inside a new parent net namespace to bother less
|
|
* about cleanup on error-path.
|
|
*/
|
|
static int init_namespaces(void)
|
|
{
|
|
nsfd_parent = unshare_open();
|
|
if (nsfd_parent <= 0)
|
|
return -1;
|
|
|
|
nsfd_childa = unshare_open();
|
|
if (nsfd_childa <= 0)
|
|
return -1;
|
|
|
|
if (switch_ns(nsfd_parent))
|
|
return -1;
|
|
|
|
nsfd_childb = unshare_open();
|
|
if (nsfd_childb <= 0)
|
|
return -1;
|
|
|
|
if (switch_ns(nsfd_parent))
|
|
return -1;
|
|
return 0;
|
|
}
|
|
|
|
static int netlink_sock(int *sock, uint32_t *seq_nr, int proto)
|
|
{
|
|
if (*sock > 0) {
|
|
seq_nr++;
|
|
return 0;
|
|
}
|
|
|
|
*sock = socket(AF_NETLINK, SOCK_RAW | SOCK_CLOEXEC, proto);
|
|
if (*sock <= 0) {
|
|
pr_err("socket(AF_NETLINK)");
|
|
return -1;
|
|
}
|
|
|
|
randomize_buffer(seq_nr, sizeof(*seq_nr));
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline struct rtattr *rtattr_hdr(struct nlmsghdr *nh)
|
|
{
|
|
return (struct rtattr *)((char *)(nh) + RTA_ALIGN((nh)->nlmsg_len));
|
|
}
|
|
|
|
static int rtattr_pack(struct nlmsghdr *nh, size_t req_sz,
|
|
unsigned short rta_type, const void *payload, size_t size)
|
|
{
|
|
/* NLMSG_ALIGNTO == RTA_ALIGNTO, nlmsg_len already aligned */
|
|
struct rtattr *attr = rtattr_hdr(nh);
|
|
size_t nl_size = RTA_ALIGN(nh->nlmsg_len) + RTA_LENGTH(size);
|
|
|
|
if (req_sz < nl_size) {
|
|
printk("req buf is too small: %zu < %zu", req_sz, nl_size);
|
|
return -1;
|
|
}
|
|
nh->nlmsg_len = nl_size;
|
|
|
|
attr->rta_len = RTA_LENGTH(size);
|
|
attr->rta_type = rta_type;
|
|
memcpy(RTA_DATA(attr), payload, size);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static struct rtattr *_rtattr_begin(struct nlmsghdr *nh, size_t req_sz,
|
|
unsigned short rta_type, const void *payload, size_t size)
|
|
{
|
|
struct rtattr *ret = rtattr_hdr(nh);
|
|
|
|
if (rtattr_pack(nh, req_sz, rta_type, payload, size))
|
|
return 0;
|
|
|
|
return ret;
|
|
}
|
|
|
|
static inline struct rtattr *rtattr_begin(struct nlmsghdr *nh, size_t req_sz,
|
|
unsigned short rta_type)
|
|
{
|
|
return _rtattr_begin(nh, req_sz, rta_type, 0, 0);
|
|
}
|
|
|
|
static inline void rtattr_end(struct nlmsghdr *nh, struct rtattr *attr)
|
|
{
|
|
char *nlmsg_end = (char *)nh + nh->nlmsg_len;
|
|
|
|
attr->rta_len = nlmsg_end - (char *)attr;
|
|
}
|
|
|
|
static int veth_pack_peerb(struct nlmsghdr *nh, size_t req_sz,
|
|
const char *peer, int ns)
|
|
{
|
|
struct ifinfomsg pi;
|
|
struct rtattr *peer_attr;
|
|
|
|
memset(&pi, 0, sizeof(pi));
|
|
pi.ifi_family = AF_UNSPEC;
|
|
pi.ifi_change = 0xFFFFFFFF;
|
|
|
|
peer_attr = _rtattr_begin(nh, req_sz, VETH_INFO_PEER, &pi, sizeof(pi));
|
|
if (!peer_attr)
|
|
return -1;
|
|
|
|
if (rtattr_pack(nh, req_sz, IFLA_IFNAME, peer, strlen(peer)))
|
|
return -1;
|
|
|
|
if (rtattr_pack(nh, req_sz, IFLA_NET_NS_FD, &ns, sizeof(ns)))
|
|
return -1;
|
|
|
|
rtattr_end(nh, peer_attr);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int netlink_check_answer(int sock)
|
|
{
|
|
struct nlmsgerror {
|
|
struct nlmsghdr hdr;
|
|
int error;
|
|
struct nlmsghdr orig_msg;
|
|
} answer;
|
|
|
|
if (recv(sock, &answer, sizeof(answer), 0) < 0) {
|
|
pr_err("recv()");
|
|
return -1;
|
|
} else if (answer.hdr.nlmsg_type != NLMSG_ERROR) {
|
|
printk("expected NLMSG_ERROR, got %d", (int)answer.hdr.nlmsg_type);
|
|
return -1;
|
|
} else if (answer.error) {
|
|
printk("NLMSG_ERROR: %d: %s",
|
|
answer.error, strerror(-answer.error));
|
|
return answer.error;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int veth_add(int sock, uint32_t seq, const char *peera, int ns_a,
|
|
const char *peerb, int ns_b)
|
|
{
|
|
uint16_t flags = NLM_F_REQUEST | NLM_F_ACK | NLM_F_EXCL | NLM_F_CREATE;
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct ifinfomsg info;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
const char veth_type[] = "veth";
|
|
struct rtattr *link_info, *info_data;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.info));
|
|
req.nh.nlmsg_type = RTM_NEWLINK;
|
|
req.nh.nlmsg_flags = flags;
|
|
req.nh.nlmsg_seq = seq;
|
|
req.info.ifi_family = AF_UNSPEC;
|
|
req.info.ifi_change = 0xFFFFFFFF;
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), IFLA_IFNAME, peera, strlen(peera)))
|
|
return -1;
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), IFLA_NET_NS_FD, &ns_a, sizeof(ns_a)))
|
|
return -1;
|
|
|
|
link_info = rtattr_begin(&req.nh, sizeof(req), IFLA_LINKINFO);
|
|
if (!link_info)
|
|
return -1;
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), IFLA_INFO_KIND, veth_type, sizeof(veth_type)))
|
|
return -1;
|
|
|
|
info_data = rtattr_begin(&req.nh, sizeof(req), IFLA_INFO_DATA);
|
|
if (!info_data)
|
|
return -1;
|
|
|
|
if (veth_pack_peerb(&req.nh, sizeof(req), peerb, ns_b))
|
|
return -1;
|
|
|
|
rtattr_end(&req.nh, info_data);
|
|
rtattr_end(&req.nh, link_info);
|
|
|
|
if (send(sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
return netlink_check_answer(sock);
|
|
}
|
|
|
|
static int ip4_addr_set(int sock, uint32_t seq, const char *intf,
|
|
struct in_addr addr, uint8_t prefix)
|
|
{
|
|
uint16_t flags = NLM_F_REQUEST | NLM_F_ACK | NLM_F_EXCL | NLM_F_CREATE;
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct ifaddrmsg info;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.info));
|
|
req.nh.nlmsg_type = RTM_NEWADDR;
|
|
req.nh.nlmsg_flags = flags;
|
|
req.nh.nlmsg_seq = seq;
|
|
req.info.ifa_family = AF_INET;
|
|
req.info.ifa_prefixlen = prefix;
|
|
req.info.ifa_index = if_nametoindex(intf);
|
|
|
|
#ifdef DEBUG
|
|
{
|
|
char addr_str[IPV4_STR_SZ] = {};
|
|
|
|
strncpy(addr_str, inet_ntoa(addr), IPV4_STR_SZ - 1);
|
|
|
|
printk("ip addr set %s", addr_str);
|
|
}
|
|
#endif
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), IFA_LOCAL, &addr, sizeof(addr)))
|
|
return -1;
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), IFA_ADDRESS, &addr, sizeof(addr)))
|
|
return -1;
|
|
|
|
if (send(sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
return netlink_check_answer(sock);
|
|
}
|
|
|
|
static int link_set_up(int sock, uint32_t seq, const char *intf)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct ifinfomsg info;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.info));
|
|
req.nh.nlmsg_type = RTM_NEWLINK;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
req.nh.nlmsg_seq = seq;
|
|
req.info.ifi_family = AF_UNSPEC;
|
|
req.info.ifi_change = 0xFFFFFFFF;
|
|
req.info.ifi_index = if_nametoindex(intf);
|
|
req.info.ifi_flags = IFF_UP;
|
|
req.info.ifi_change = IFF_UP;
|
|
|
|
if (send(sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
return netlink_check_answer(sock);
|
|
}
|
|
|
|
static int ip4_route_set(int sock, uint32_t seq, const char *intf,
|
|
struct in_addr src, struct in_addr dst)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct rtmsg rt;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
unsigned int index = if_nametoindex(intf);
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.rt));
|
|
req.nh.nlmsg_type = RTM_NEWROUTE;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK | NLM_F_CREATE;
|
|
req.nh.nlmsg_seq = seq;
|
|
req.rt.rtm_family = AF_INET;
|
|
req.rt.rtm_dst_len = 32;
|
|
req.rt.rtm_table = RT_TABLE_MAIN;
|
|
req.rt.rtm_protocol = RTPROT_BOOT;
|
|
req.rt.rtm_scope = RT_SCOPE_LINK;
|
|
req.rt.rtm_type = RTN_UNICAST;
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), RTA_DST, &dst, sizeof(dst)))
|
|
return -1;
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), RTA_PREFSRC, &src, sizeof(src)))
|
|
return -1;
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), RTA_OIF, &index, sizeof(index)))
|
|
return -1;
|
|
|
|
if (send(sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
|
|
return netlink_check_answer(sock);
|
|
}
|
|
|
|
static int tunnel_set_route(int route_sock, uint32_t *route_seq, char *veth,
|
|
struct in_addr tunsrc, struct in_addr tundst)
|
|
{
|
|
if (ip4_addr_set(route_sock, (*route_seq)++, "lo",
|
|
tunsrc, PREFIX_LEN)) {
|
|
printk("Failed to set ipv4 addr");
|
|
return -1;
|
|
}
|
|
|
|
if (ip4_route_set(route_sock, (*route_seq)++, veth, tunsrc, tundst)) {
|
|
printk("Failed to set ipv4 route");
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int init_child(int nsfd, char *veth, unsigned int src, unsigned int dst)
|
|
{
|
|
struct in_addr intsrc = inet_makeaddr(INADDR_B, src);
|
|
struct in_addr tunsrc = inet_makeaddr(INADDR_A, src);
|
|
struct in_addr tundst = inet_makeaddr(INADDR_A, dst);
|
|
int route_sock = -1, ret = -1;
|
|
uint32_t route_seq;
|
|
|
|
if (switch_ns(nsfd))
|
|
return -1;
|
|
|
|
if (netlink_sock(&route_sock, &route_seq, NETLINK_ROUTE)) {
|
|
printk("Failed to open netlink route socket in child");
|
|
return -1;
|
|
}
|
|
|
|
if (ip4_addr_set(route_sock, route_seq++, veth, intsrc, PREFIX_LEN)) {
|
|
printk("Failed to set ipv4 addr");
|
|
goto err;
|
|
}
|
|
|
|
if (link_set_up(route_sock, route_seq++, veth)) {
|
|
printk("Failed to bring up %s", veth);
|
|
goto err;
|
|
}
|
|
|
|
if (tunnel_set_route(route_sock, &route_seq, veth, tunsrc, tundst)) {
|
|
printk("Failed to add tunnel route on %s", veth);
|
|
goto err;
|
|
}
|
|
ret = 0;
|
|
|
|
err:
|
|
close(route_sock);
|
|
return ret;
|
|
}
|
|
|
|
#define ALGO_LEN 64
|
|
enum desc_type {
|
|
CREATE_TUNNEL = 0,
|
|
ALLOCATE_SPI,
|
|
MONITOR_ACQUIRE,
|
|
EXPIRE_STATE,
|
|
EXPIRE_POLICY,
|
|
};
|
|
const char *desc_name[] = {
|
|
"create tunnel",
|
|
"alloc spi",
|
|
"monitor acquire",
|
|
"expire state",
|
|
"expire policy"
|
|
};
|
|
struct xfrm_desc {
|
|
enum desc_type type;
|
|
uint8_t proto;
|
|
char a_algo[ALGO_LEN];
|
|
char e_algo[ALGO_LEN];
|
|
char c_algo[ALGO_LEN];
|
|
char ae_algo[ALGO_LEN];
|
|
unsigned int icv_len;
|
|
/* unsigned key_len; */
|
|
};
|
|
|
|
enum msg_type {
|
|
MSG_ACK = 0,
|
|
MSG_EXIT,
|
|
MSG_PING,
|
|
MSG_XFRM_PREPARE,
|
|
MSG_XFRM_ADD,
|
|
MSG_XFRM_DEL,
|
|
MSG_XFRM_CLEANUP,
|
|
};
|
|
|
|
struct test_desc {
|
|
enum msg_type type;
|
|
union {
|
|
struct {
|
|
in_addr_t reply_ip;
|
|
unsigned int port;
|
|
} ping;
|
|
struct xfrm_desc xfrm_desc;
|
|
} body;
|
|
};
|
|
|
|
struct test_result {
|
|
struct xfrm_desc desc;
|
|
unsigned int res;
|
|
};
|
|
|
|
static void write_test_result(unsigned int res, struct xfrm_desc *d)
|
|
{
|
|
struct test_result tr = {};
|
|
ssize_t ret;
|
|
|
|
tr.desc = *d;
|
|
tr.res = res;
|
|
|
|
ret = write(results_fd[1], &tr, sizeof(tr));
|
|
if (ret != sizeof(tr))
|
|
pr_err("Failed to write the result in pipe %zd", ret);
|
|
}
|
|
|
|
static void write_msg(int fd, struct test_desc *msg, bool exit_of_fail)
|
|
{
|
|
ssize_t bytes = write(fd, msg, sizeof(*msg));
|
|
|
|
/* Make sure that write/read is atomic to a pipe */
|
|
BUILD_BUG_ON(sizeof(struct test_desc) > PIPE_BUF);
|
|
|
|
if (bytes < 0) {
|
|
pr_err("write()");
|
|
if (exit_of_fail)
|
|
exit(KSFT_FAIL);
|
|
}
|
|
if (bytes != sizeof(*msg)) {
|
|
pr_err("sent part of the message %zd/%zu", bytes, sizeof(*msg));
|
|
if (exit_of_fail)
|
|
exit(KSFT_FAIL);
|
|
}
|
|
}
|
|
|
|
static void read_msg(int fd, struct test_desc *msg, bool exit_of_fail)
|
|
{
|
|
ssize_t bytes = read(fd, msg, sizeof(*msg));
|
|
|
|
if (bytes < 0) {
|
|
pr_err("read()");
|
|
if (exit_of_fail)
|
|
exit(KSFT_FAIL);
|
|
}
|
|
if (bytes != sizeof(*msg)) {
|
|
pr_err("got incomplete message %zd/%zu", bytes, sizeof(*msg));
|
|
if (exit_of_fail)
|
|
exit(KSFT_FAIL);
|
|
}
|
|
}
|
|
|
|
static int udp_ping_init(struct in_addr listen_ip, unsigned int u_timeout,
|
|
unsigned int *server_port, int sock[2])
|
|
{
|
|
struct sockaddr_in server;
|
|
struct timeval t = { .tv_sec = 0, .tv_usec = u_timeout };
|
|
socklen_t s_len = sizeof(server);
|
|
|
|
sock[0] = socket(AF_INET, SOCK_DGRAM, 0);
|
|
if (sock[0] < 0) {
|
|
pr_err("socket()");
|
|
return -1;
|
|
}
|
|
|
|
server.sin_family = AF_INET;
|
|
server.sin_port = 0;
|
|
memcpy(&server.sin_addr.s_addr, &listen_ip, sizeof(struct in_addr));
|
|
|
|
if (bind(sock[0], (struct sockaddr *)&server, s_len)) {
|
|
pr_err("bind()");
|
|
goto err_close_server;
|
|
}
|
|
|
|
if (getsockname(sock[0], (struct sockaddr *)&server, &s_len)) {
|
|
pr_err("getsockname()");
|
|
goto err_close_server;
|
|
}
|
|
|
|
*server_port = ntohs(server.sin_port);
|
|
|
|
if (setsockopt(sock[0], SOL_SOCKET, SO_RCVTIMEO, (const char *)&t, sizeof t)) {
|
|
pr_err("setsockopt()");
|
|
goto err_close_server;
|
|
}
|
|
|
|
sock[1] = socket(AF_INET, SOCK_DGRAM, 0);
|
|
if (sock[1] < 0) {
|
|
pr_err("socket()");
|
|
goto err_close_server;
|
|
}
|
|
|
|
return 0;
|
|
|
|
err_close_server:
|
|
close(sock[0]);
|
|
return -1;
|
|
}
|
|
|
|
static int udp_ping_send(int sock[2], in_addr_t dest_ip, unsigned int port,
|
|
char *buf, size_t buf_len)
|
|
{
|
|
struct sockaddr_in server;
|
|
const struct sockaddr *dest_addr = (struct sockaddr *)&server;
|
|
char *sock_buf[buf_len];
|
|
ssize_t r_bytes, s_bytes;
|
|
|
|
server.sin_family = AF_INET;
|
|
server.sin_port = htons(port);
|
|
server.sin_addr.s_addr = dest_ip;
|
|
|
|
s_bytes = sendto(sock[1], buf, buf_len, 0, dest_addr, sizeof(server));
|
|
if (s_bytes < 0) {
|
|
pr_err("sendto()");
|
|
return -1;
|
|
} else if (s_bytes != buf_len) {
|
|
printk("send part of the message: %zd/%zu", s_bytes, sizeof(server));
|
|
return -1;
|
|
}
|
|
|
|
r_bytes = recv(sock[0], sock_buf, buf_len, 0);
|
|
if (r_bytes < 0) {
|
|
if (errno != EAGAIN)
|
|
pr_err("recv()");
|
|
return -1;
|
|
} else if (r_bytes == 0) { /* EOF */
|
|
printk("EOF on reply to ping");
|
|
return -1;
|
|
} else if (r_bytes != buf_len || memcmp(buf, sock_buf, buf_len)) {
|
|
printk("ping reply packet is corrupted %zd/%zu", r_bytes, buf_len);
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int udp_ping_reply(int sock[2], in_addr_t dest_ip, unsigned int port,
|
|
char *buf, size_t buf_len)
|
|
{
|
|
struct sockaddr_in server;
|
|
const struct sockaddr *dest_addr = (struct sockaddr *)&server;
|
|
char *sock_buf[buf_len];
|
|
ssize_t r_bytes, s_bytes;
|
|
|
|
server.sin_family = AF_INET;
|
|
server.sin_port = htons(port);
|
|
server.sin_addr.s_addr = dest_ip;
|
|
|
|
r_bytes = recv(sock[0], sock_buf, buf_len, 0);
|
|
if (r_bytes < 0) {
|
|
if (errno != EAGAIN)
|
|
pr_err("recv()");
|
|
return -1;
|
|
}
|
|
if (r_bytes == 0) { /* EOF */
|
|
printk("EOF on reply to ping");
|
|
return -1;
|
|
}
|
|
if (r_bytes != buf_len || memcmp(buf, sock_buf, buf_len)) {
|
|
printk("ping reply packet is corrupted %zd/%zu", r_bytes, buf_len);
|
|
return -1;
|
|
}
|
|
|
|
s_bytes = sendto(sock[1], buf, buf_len, 0, dest_addr, sizeof(server));
|
|
if (s_bytes < 0) {
|
|
pr_err("sendto()");
|
|
return -1;
|
|
} else if (s_bytes != buf_len) {
|
|
printk("send part of the message: %zd/%zu", s_bytes, sizeof(server));
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
typedef int (*ping_f)(int sock[2], in_addr_t dest_ip, unsigned int port,
|
|
char *buf, size_t buf_len);
|
|
static int do_ping(int cmd_fd, char *buf, size_t buf_len, struct in_addr from,
|
|
bool init_side, int d_port, in_addr_t to, ping_f func)
|
|
{
|
|
struct test_desc msg;
|
|
unsigned int s_port, i, ping_succeeded = 0;
|
|
int ping_sock[2];
|
|
char to_str[IPV4_STR_SZ] = {}, from_str[IPV4_STR_SZ] = {};
|
|
|
|
if (udp_ping_init(from, ping_timeout, &s_port, ping_sock)) {
|
|
printk("Failed to init ping");
|
|
return -1;
|
|
}
|
|
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.type = MSG_PING;
|
|
msg.body.ping.port = s_port;
|
|
memcpy(&msg.body.ping.reply_ip, &from, sizeof(from));
|
|
|
|
write_msg(cmd_fd, &msg, 0);
|
|
if (init_side) {
|
|
/* The other end sends ip to ping */
|
|
read_msg(cmd_fd, &msg, 0);
|
|
if (msg.type != MSG_PING)
|
|
return -1;
|
|
to = msg.body.ping.reply_ip;
|
|
d_port = msg.body.ping.port;
|
|
}
|
|
|
|
for (i = 0; i < ping_count ; i++) {
|
|
struct timespec sleep_time = {
|
|
.tv_sec = 0,
|
|
.tv_nsec = ping_delay_nsec,
|
|
};
|
|
|
|
ping_succeeded += !func(ping_sock, to, d_port, buf, page_size);
|
|
nanosleep(&sleep_time, 0);
|
|
}
|
|
|
|
close(ping_sock[0]);
|
|
close(ping_sock[1]);
|
|
|
|
strncpy(to_str, inet_ntoa(*(struct in_addr *)&to), IPV4_STR_SZ - 1);
|
|
strncpy(from_str, inet_ntoa(from), IPV4_STR_SZ - 1);
|
|
|
|
if (ping_succeeded < ping_success) {
|
|
printk("ping (%s) %s->%s failed %u/%u times",
|
|
init_side ? "send" : "reply", from_str, to_str,
|
|
ping_count - ping_succeeded, ping_count);
|
|
return -1;
|
|
}
|
|
|
|
#ifdef DEBUG
|
|
printk("ping (%s) %s->%s succeeded %u/%u times",
|
|
init_side ? "send" : "reply", from_str, to_str,
|
|
ping_succeeded, ping_count);
|
|
#endif
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int xfrm_fill_key(char *name, char *buf,
|
|
size_t buf_len, unsigned int *key_len)
|
|
{
|
|
/* TODO: use set/map instead */
|
|
if (strncmp(name, "digest_null", ALGO_LEN) == 0)
|
|
*key_len = 0;
|
|
else if (strncmp(name, "ecb(cipher_null)", ALGO_LEN) == 0)
|
|
*key_len = 0;
|
|
else if (strncmp(name, "cbc(des)", ALGO_LEN) == 0)
|
|
*key_len = 64;
|
|
else if (strncmp(name, "hmac(md5)", ALGO_LEN) == 0)
|
|
*key_len = 128;
|
|
else if (strncmp(name, "cmac(aes)", ALGO_LEN) == 0)
|
|
*key_len = 128;
|
|
else if (strncmp(name, "xcbc(aes)", ALGO_LEN) == 0)
|
|
*key_len = 128;
|
|
else if (strncmp(name, "cbc(cast5)", ALGO_LEN) == 0)
|
|
*key_len = 128;
|
|
else if (strncmp(name, "cbc(serpent)", ALGO_LEN) == 0)
|
|
*key_len = 128;
|
|
else if (strncmp(name, "hmac(sha1)", ALGO_LEN) == 0)
|
|
*key_len = 160;
|
|
else if (strncmp(name, "hmac(rmd160)", ALGO_LEN) == 0)
|
|
*key_len = 160;
|
|
else if (strncmp(name, "cbc(des3_ede)", ALGO_LEN) == 0)
|
|
*key_len = 192;
|
|
else if (strncmp(name, "hmac(sha256)", ALGO_LEN) == 0)
|
|
*key_len = 256;
|
|
else if (strncmp(name, "cbc(aes)", ALGO_LEN) == 0)
|
|
*key_len = 256;
|
|
else if (strncmp(name, "cbc(camellia)", ALGO_LEN) == 0)
|
|
*key_len = 256;
|
|
else if (strncmp(name, "cbc(twofish)", ALGO_LEN) == 0)
|
|
*key_len = 256;
|
|
else if (strncmp(name, "rfc3686(ctr(aes))", ALGO_LEN) == 0)
|
|
*key_len = 288;
|
|
else if (strncmp(name, "hmac(sha384)", ALGO_LEN) == 0)
|
|
*key_len = 384;
|
|
else if (strncmp(name, "cbc(blowfish)", ALGO_LEN) == 0)
|
|
*key_len = 448;
|
|
else if (strncmp(name, "hmac(sha512)", ALGO_LEN) == 0)
|
|
*key_len = 512;
|
|
else if (strncmp(name, "rfc4106(gcm(aes))-128", ALGO_LEN) == 0)
|
|
*key_len = 160;
|
|
else if (strncmp(name, "rfc4543(gcm(aes))-128", ALGO_LEN) == 0)
|
|
*key_len = 160;
|
|
else if (strncmp(name, "rfc4309(ccm(aes))-128", ALGO_LEN) == 0)
|
|
*key_len = 152;
|
|
else if (strncmp(name, "rfc4106(gcm(aes))-192", ALGO_LEN) == 0)
|
|
*key_len = 224;
|
|
else if (strncmp(name, "rfc4543(gcm(aes))-192", ALGO_LEN) == 0)
|
|
*key_len = 224;
|
|
else if (strncmp(name, "rfc4309(ccm(aes))-192", ALGO_LEN) == 0)
|
|
*key_len = 216;
|
|
else if (strncmp(name, "rfc4106(gcm(aes))-256", ALGO_LEN) == 0)
|
|
*key_len = 288;
|
|
else if (strncmp(name, "rfc4543(gcm(aes))-256", ALGO_LEN) == 0)
|
|
*key_len = 288;
|
|
else if (strncmp(name, "rfc4309(ccm(aes))-256", ALGO_LEN) == 0)
|
|
*key_len = 280;
|
|
else if (strncmp(name, "rfc7539(chacha20,poly1305)-128", ALGO_LEN) == 0)
|
|
*key_len = 0;
|
|
|
|
if (*key_len > buf_len) {
|
|
printk("Can't pack a key - too big for buffer");
|
|
return -1;
|
|
}
|
|
|
|
randomize_buffer(buf, *key_len);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int xfrm_state_pack_algo(struct nlmsghdr *nh, size_t req_sz,
|
|
struct xfrm_desc *desc)
|
|
{
|
|
struct {
|
|
union {
|
|
struct xfrm_algo alg;
|
|
struct xfrm_algo_aead aead;
|
|
struct xfrm_algo_auth auth;
|
|
} u;
|
|
char buf[XFRM_ALGO_KEY_BUF_SIZE];
|
|
} alg = {};
|
|
size_t alen, elen, clen, aelen;
|
|
unsigned short type;
|
|
|
|
alen = strlen(desc->a_algo);
|
|
elen = strlen(desc->e_algo);
|
|
clen = strlen(desc->c_algo);
|
|
aelen = strlen(desc->ae_algo);
|
|
|
|
/* Verify desc */
|
|
switch (desc->proto) {
|
|
case IPPROTO_AH:
|
|
if (!alen || elen || clen || aelen) {
|
|
printk("BUG: buggy ah desc");
|
|
return -1;
|
|
}
|
|
strncpy(alg.u.alg.alg_name, desc->a_algo, ALGO_LEN - 1);
|
|
if (xfrm_fill_key(desc->a_algo, alg.u.alg.alg_key,
|
|
sizeof(alg.buf), &alg.u.alg.alg_key_len))
|
|
return -1;
|
|
type = XFRMA_ALG_AUTH;
|
|
break;
|
|
case IPPROTO_COMP:
|
|
if (!clen || elen || alen || aelen) {
|
|
printk("BUG: buggy comp desc");
|
|
return -1;
|
|
}
|
|
strncpy(alg.u.alg.alg_name, desc->c_algo, ALGO_LEN - 1);
|
|
if (xfrm_fill_key(desc->c_algo, alg.u.alg.alg_key,
|
|
sizeof(alg.buf), &alg.u.alg.alg_key_len))
|
|
return -1;
|
|
type = XFRMA_ALG_COMP;
|
|
break;
|
|
case IPPROTO_ESP:
|
|
if (!((alen && elen) ^ aelen) || clen) {
|
|
printk("BUG: buggy esp desc");
|
|
return -1;
|
|
}
|
|
if (aelen) {
|
|
alg.u.aead.alg_icv_len = desc->icv_len;
|
|
strncpy(alg.u.aead.alg_name, desc->ae_algo, ALGO_LEN - 1);
|
|
if (xfrm_fill_key(desc->ae_algo, alg.u.aead.alg_key,
|
|
sizeof(alg.buf), &alg.u.aead.alg_key_len))
|
|
return -1;
|
|
type = XFRMA_ALG_AEAD;
|
|
} else {
|
|
|
|
strncpy(alg.u.alg.alg_name, desc->e_algo, ALGO_LEN - 1);
|
|
type = XFRMA_ALG_CRYPT;
|
|
if (xfrm_fill_key(desc->e_algo, alg.u.alg.alg_key,
|
|
sizeof(alg.buf), &alg.u.alg.alg_key_len))
|
|
return -1;
|
|
if (rtattr_pack(nh, req_sz, type, &alg, sizeof(alg)))
|
|
return -1;
|
|
|
|
strncpy(alg.u.alg.alg_name, desc->a_algo, ALGO_LEN);
|
|
type = XFRMA_ALG_AUTH;
|
|
if (xfrm_fill_key(desc->a_algo, alg.u.alg.alg_key,
|
|
sizeof(alg.buf), &alg.u.alg.alg_key_len))
|
|
return -1;
|
|
}
|
|
break;
|
|
default:
|
|
printk("BUG: unknown proto in desc");
|
|
return -1;
|
|
}
|
|
|
|
if (rtattr_pack(nh, req_sz, type, &alg, sizeof(alg)))
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline uint32_t gen_spi(struct in_addr src)
|
|
{
|
|
return htonl(inet_lnaof(src));
|
|
}
|
|
|
|
static int xfrm_state_add(int xfrm_sock, uint32_t seq, uint32_t spi,
|
|
struct in_addr src, struct in_addr dst,
|
|
struct xfrm_desc *desc)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct xfrm_usersa_info info;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.info));
|
|
req.nh.nlmsg_type = XFRM_MSG_NEWSA;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
req.nh.nlmsg_seq = seq;
|
|
|
|
/* Fill selector. */
|
|
memcpy(&req.info.sel.daddr, &dst, sizeof(dst));
|
|
memcpy(&req.info.sel.saddr, &src, sizeof(src));
|
|
req.info.sel.family = AF_INET;
|
|
req.info.sel.prefixlen_d = PREFIX_LEN;
|
|
req.info.sel.prefixlen_s = PREFIX_LEN;
|
|
|
|
/* Fill id */
|
|
memcpy(&req.info.id.daddr, &dst, sizeof(dst));
|
|
/* Note: zero-spi cannot be deleted */
|
|
req.info.id.spi = spi;
|
|
req.info.id.proto = desc->proto;
|
|
|
|
memcpy(&req.info.saddr, &src, sizeof(src));
|
|
|
|
/* Fill lifteme_cfg */
|
|
req.info.lft.soft_byte_limit = XFRM_INF;
|
|
req.info.lft.hard_byte_limit = XFRM_INF;
|
|
req.info.lft.soft_packet_limit = XFRM_INF;
|
|
req.info.lft.hard_packet_limit = XFRM_INF;
|
|
|
|
req.info.family = AF_INET;
|
|
req.info.mode = XFRM_MODE_TUNNEL;
|
|
|
|
if (xfrm_state_pack_algo(&req.nh, sizeof(req), desc))
|
|
return -1;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
|
|
return netlink_check_answer(xfrm_sock);
|
|
}
|
|
|
|
static bool xfrm_usersa_found(struct xfrm_usersa_info *info, uint32_t spi,
|
|
struct in_addr src, struct in_addr dst,
|
|
struct xfrm_desc *desc)
|
|
{
|
|
if (memcmp(&info->sel.daddr, &dst, sizeof(dst)))
|
|
return false;
|
|
|
|
if (memcmp(&info->sel.saddr, &src, sizeof(src)))
|
|
return false;
|
|
|
|
if (info->sel.family != AF_INET ||
|
|
info->sel.prefixlen_d != PREFIX_LEN ||
|
|
info->sel.prefixlen_s != PREFIX_LEN)
|
|
return false;
|
|
|
|
if (info->id.spi != spi || info->id.proto != desc->proto)
|
|
return false;
|
|
|
|
if (memcmp(&info->id.daddr, &dst, sizeof(dst)))
|
|
return false;
|
|
|
|
if (memcmp(&info->saddr, &src, sizeof(src)))
|
|
return false;
|
|
|
|
if (info->lft.soft_byte_limit != XFRM_INF ||
|
|
info->lft.hard_byte_limit != XFRM_INF ||
|
|
info->lft.soft_packet_limit != XFRM_INF ||
|
|
info->lft.hard_packet_limit != XFRM_INF)
|
|
return false;
|
|
|
|
if (info->family != AF_INET || info->mode != XFRM_MODE_TUNNEL)
|
|
return false;
|
|
|
|
/* XXX: check xfrm algo, see xfrm_state_pack_algo(). */
|
|
|
|
return true;
|
|
}
|
|
|
|
static int xfrm_state_check(int xfrm_sock, uint32_t seq, uint32_t spi,
|
|
struct in_addr src, struct in_addr dst,
|
|
struct xfrm_desc *desc)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
union {
|
|
struct xfrm_usersa_info info;
|
|
int error;
|
|
};
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} answer;
|
|
struct xfrm_address_filter filter = {};
|
|
bool found = false;
|
|
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(0);
|
|
req.nh.nlmsg_type = XFRM_MSG_GETSA;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
|
|
req.nh.nlmsg_seq = seq;
|
|
|
|
/*
|
|
* Add dump filter by source address as there may be other tunnels
|
|
* in this netns (if tests run in parallel).
|
|
*/
|
|
filter.family = AF_INET;
|
|
filter.splen = 0x1f; /* 0xffffffff mask see addr_match() */
|
|
memcpy(&filter.saddr, &src, sizeof(src));
|
|
if (rtattr_pack(&req.nh, sizeof(req), XFRMA_ADDRESS_FILTER,
|
|
&filter, sizeof(filter)))
|
|
return -1;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
|
|
while (1) {
|
|
if (recv(xfrm_sock, &answer, sizeof(answer), 0) < 0) {
|
|
pr_err("recv()");
|
|
return -1;
|
|
}
|
|
if (answer.nh.nlmsg_type == NLMSG_ERROR) {
|
|
printk("NLMSG_ERROR: %d: %s",
|
|
answer.error, strerror(-answer.error));
|
|
return -1;
|
|
} else if (answer.nh.nlmsg_type == NLMSG_DONE) {
|
|
if (found)
|
|
return 0;
|
|
printk("didn't find allocated xfrm state in dump");
|
|
return -1;
|
|
} else if (answer.nh.nlmsg_type == XFRM_MSG_NEWSA) {
|
|
if (xfrm_usersa_found(&answer.info, spi, src, dst, desc))
|
|
found = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
static int xfrm_set(int xfrm_sock, uint32_t *seq,
|
|
struct in_addr src, struct in_addr dst,
|
|
struct in_addr tunsrc, struct in_addr tundst,
|
|
struct xfrm_desc *desc)
|
|
{
|
|
int err;
|
|
|
|
err = xfrm_state_add(xfrm_sock, (*seq)++, gen_spi(src), src, dst, desc);
|
|
if (err) {
|
|
printk("Failed to add xfrm state");
|
|
return -1;
|
|
}
|
|
|
|
err = xfrm_state_add(xfrm_sock, (*seq)++, gen_spi(src), dst, src, desc);
|
|
if (err) {
|
|
printk("Failed to add xfrm state");
|
|
return -1;
|
|
}
|
|
|
|
/* Check dumps for XFRM_MSG_GETSA */
|
|
err = xfrm_state_check(xfrm_sock, (*seq)++, gen_spi(src), src, dst, desc);
|
|
err |= xfrm_state_check(xfrm_sock, (*seq)++, gen_spi(src), dst, src, desc);
|
|
if (err) {
|
|
printk("Failed to check xfrm state");
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int xfrm_policy_add(int xfrm_sock, uint32_t seq, uint32_t spi,
|
|
struct in_addr src, struct in_addr dst, uint8_t dir,
|
|
struct in_addr tunsrc, struct in_addr tundst, uint8_t proto)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct xfrm_userpolicy_info info;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
struct xfrm_user_tmpl tmpl;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
memset(&tmpl, 0, sizeof(tmpl));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.info));
|
|
req.nh.nlmsg_type = XFRM_MSG_NEWPOLICY;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
req.nh.nlmsg_seq = seq;
|
|
|
|
/* Fill selector. */
|
|
memcpy(&req.info.sel.daddr, &dst, sizeof(tundst));
|
|
memcpy(&req.info.sel.saddr, &src, sizeof(tunsrc));
|
|
req.info.sel.family = AF_INET;
|
|
req.info.sel.prefixlen_d = PREFIX_LEN;
|
|
req.info.sel.prefixlen_s = PREFIX_LEN;
|
|
|
|
/* Fill lifteme_cfg */
|
|
req.info.lft.soft_byte_limit = XFRM_INF;
|
|
req.info.lft.hard_byte_limit = XFRM_INF;
|
|
req.info.lft.soft_packet_limit = XFRM_INF;
|
|
req.info.lft.hard_packet_limit = XFRM_INF;
|
|
|
|
req.info.dir = dir;
|
|
|
|
/* Fill tmpl */
|
|
memcpy(&tmpl.id.daddr, &dst, sizeof(dst));
|
|
/* Note: zero-spi cannot be deleted */
|
|
tmpl.id.spi = spi;
|
|
tmpl.id.proto = proto;
|
|
tmpl.family = AF_INET;
|
|
memcpy(&tmpl.saddr, &src, sizeof(src));
|
|
tmpl.mode = XFRM_MODE_TUNNEL;
|
|
tmpl.aalgos = (~(uint32_t)0);
|
|
tmpl.ealgos = (~(uint32_t)0);
|
|
tmpl.calgos = (~(uint32_t)0);
|
|
|
|
if (rtattr_pack(&req.nh, sizeof(req), XFRMA_TMPL, &tmpl, sizeof(tmpl)))
|
|
return -1;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
|
|
return netlink_check_answer(xfrm_sock);
|
|
}
|
|
|
|
static int xfrm_prepare(int xfrm_sock, uint32_t *seq,
|
|
struct in_addr src, struct in_addr dst,
|
|
struct in_addr tunsrc, struct in_addr tundst, uint8_t proto)
|
|
{
|
|
if (xfrm_policy_add(xfrm_sock, (*seq)++, gen_spi(src), src, dst,
|
|
XFRM_POLICY_OUT, tunsrc, tundst, proto)) {
|
|
printk("Failed to add xfrm policy");
|
|
return -1;
|
|
}
|
|
|
|
if (xfrm_policy_add(xfrm_sock, (*seq)++, gen_spi(src), dst, src,
|
|
XFRM_POLICY_IN, tunsrc, tundst, proto)) {
|
|
printk("Failed to add xfrm policy");
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int xfrm_policy_del(int xfrm_sock, uint32_t seq,
|
|
struct in_addr src, struct in_addr dst, uint8_t dir,
|
|
struct in_addr tunsrc, struct in_addr tundst)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct xfrm_userpolicy_id id;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.id));
|
|
req.nh.nlmsg_type = XFRM_MSG_DELPOLICY;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
req.nh.nlmsg_seq = seq;
|
|
|
|
/* Fill id */
|
|
memcpy(&req.id.sel.daddr, &dst, sizeof(tundst));
|
|
memcpy(&req.id.sel.saddr, &src, sizeof(tunsrc));
|
|
req.id.sel.family = AF_INET;
|
|
req.id.sel.prefixlen_d = PREFIX_LEN;
|
|
req.id.sel.prefixlen_s = PREFIX_LEN;
|
|
req.id.dir = dir;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
|
|
return netlink_check_answer(xfrm_sock);
|
|
}
|
|
|
|
static int xfrm_cleanup(int xfrm_sock, uint32_t *seq,
|
|
struct in_addr src, struct in_addr dst,
|
|
struct in_addr tunsrc, struct in_addr tundst)
|
|
{
|
|
if (xfrm_policy_del(xfrm_sock, (*seq)++, src, dst,
|
|
XFRM_POLICY_OUT, tunsrc, tundst)) {
|
|
printk("Failed to add xfrm policy");
|
|
return -1;
|
|
}
|
|
|
|
if (xfrm_policy_del(xfrm_sock, (*seq)++, dst, src,
|
|
XFRM_POLICY_IN, tunsrc, tundst)) {
|
|
printk("Failed to add xfrm policy");
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int xfrm_state_del(int xfrm_sock, uint32_t seq, uint32_t spi,
|
|
struct in_addr src, struct in_addr dst, uint8_t proto)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct xfrm_usersa_id id;
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
xfrm_address_t saddr = {};
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.id));
|
|
req.nh.nlmsg_type = XFRM_MSG_DELSA;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
req.nh.nlmsg_seq = seq;
|
|
|
|
memcpy(&req.id.daddr, &dst, sizeof(dst));
|
|
req.id.family = AF_INET;
|
|
req.id.proto = proto;
|
|
/* Note: zero-spi cannot be deleted */
|
|
req.id.spi = spi;
|
|
|
|
memcpy(&saddr, &src, sizeof(src));
|
|
if (rtattr_pack(&req.nh, sizeof(req), XFRMA_SRCADDR, &saddr, sizeof(saddr)))
|
|
return -1;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return -1;
|
|
}
|
|
|
|
return netlink_check_answer(xfrm_sock);
|
|
}
|
|
|
|
static int xfrm_delete(int xfrm_sock, uint32_t *seq,
|
|
struct in_addr src, struct in_addr dst,
|
|
struct in_addr tunsrc, struct in_addr tundst, uint8_t proto)
|
|
{
|
|
if (xfrm_state_del(xfrm_sock, (*seq)++, gen_spi(src), src, dst, proto)) {
|
|
printk("Failed to remove xfrm state");
|
|
return -1;
|
|
}
|
|
|
|
if (xfrm_state_del(xfrm_sock, (*seq)++, gen_spi(src), dst, src, proto)) {
|
|
printk("Failed to remove xfrm state");
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int xfrm_state_allocspi(int xfrm_sock, uint32_t *seq,
|
|
uint32_t spi, uint8_t proto)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct xfrm_userspi_info spi;
|
|
} req;
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
union {
|
|
struct xfrm_usersa_info info;
|
|
int error;
|
|
};
|
|
} answer;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.spi));
|
|
req.nh.nlmsg_type = XFRM_MSG_ALLOCSPI;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST;
|
|
req.nh.nlmsg_seq = (*seq)++;
|
|
|
|
req.spi.info.family = AF_INET;
|
|
req.spi.min = spi;
|
|
req.spi.max = spi;
|
|
req.spi.info.id.proto = proto;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
return KSFT_FAIL;
|
|
}
|
|
|
|
if (recv(xfrm_sock, &answer, sizeof(answer), 0) < 0) {
|
|
pr_err("recv()");
|
|
return KSFT_FAIL;
|
|
} else if (answer.nh.nlmsg_type == XFRM_MSG_NEWSA) {
|
|
uint32_t new_spi = htonl(answer.info.id.spi);
|
|
|
|
if (new_spi != spi) {
|
|
printk("allocated spi is different from requested: %#x != %#x",
|
|
new_spi, spi);
|
|
return KSFT_FAIL;
|
|
}
|
|
return KSFT_PASS;
|
|
} else if (answer.nh.nlmsg_type != NLMSG_ERROR) {
|
|
printk("expected NLMSG_ERROR, got %d", (int)answer.nh.nlmsg_type);
|
|
return KSFT_FAIL;
|
|
}
|
|
|
|
printk("NLMSG_ERROR: %d: %s", answer.error, strerror(-answer.error));
|
|
return (answer.error) ? KSFT_FAIL : KSFT_PASS;
|
|
}
|
|
|
|
static int netlink_sock_bind(int *sock, uint32_t *seq, int proto, uint32_t groups)
|
|
{
|
|
struct sockaddr_nl snl = {};
|
|
socklen_t addr_len;
|
|
int ret = -1;
|
|
|
|
snl.nl_family = AF_NETLINK;
|
|
snl.nl_groups = groups;
|
|
|
|
if (netlink_sock(sock, seq, proto)) {
|
|
printk("Failed to open xfrm netlink socket");
|
|
return -1;
|
|
}
|
|
|
|
if (bind(*sock, (struct sockaddr *)&snl, sizeof(snl)) < 0) {
|
|
pr_err("bind()");
|
|
goto out_close;
|
|
}
|
|
|
|
addr_len = sizeof(snl);
|
|
if (getsockname(*sock, (struct sockaddr *)&snl, &addr_len) < 0) {
|
|
pr_err("getsockname()");
|
|
goto out_close;
|
|
}
|
|
if (addr_len != sizeof(snl)) {
|
|
printk("Wrong address length %d", addr_len);
|
|
goto out_close;
|
|
}
|
|
if (snl.nl_family != AF_NETLINK) {
|
|
printk("Wrong address family %d", snl.nl_family);
|
|
goto out_close;
|
|
}
|
|
return 0;
|
|
|
|
out_close:
|
|
close(*sock);
|
|
return ret;
|
|
}
|
|
|
|
static int xfrm_monitor_acquire(int xfrm_sock, uint32_t *seq, unsigned int nr)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
union {
|
|
struct xfrm_user_acquire acq;
|
|
int error;
|
|
};
|
|
char attrbuf[MAX_PAYLOAD];
|
|
} req;
|
|
struct xfrm_user_tmpl xfrm_tmpl = {};
|
|
int xfrm_listen = -1, ret = KSFT_FAIL;
|
|
uint32_t seq_listen;
|
|
|
|
if (netlink_sock_bind(&xfrm_listen, &seq_listen, NETLINK_XFRM, XFRMNLGRP_ACQUIRE))
|
|
return KSFT_FAIL;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.acq));
|
|
req.nh.nlmsg_type = XFRM_MSG_ACQUIRE;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
req.nh.nlmsg_seq = (*seq)++;
|
|
|
|
req.acq.policy.sel.family = AF_INET;
|
|
req.acq.aalgos = 0xfeed;
|
|
req.acq.ealgos = 0xbaad;
|
|
req.acq.calgos = 0xbabe;
|
|
|
|
xfrm_tmpl.family = AF_INET;
|
|
xfrm_tmpl.id.proto = IPPROTO_ESP;
|
|
if (rtattr_pack(&req.nh, sizeof(req), XFRMA_TMPL, &xfrm_tmpl, sizeof(xfrm_tmpl)))
|
|
goto out_close;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
goto out_close;
|
|
}
|
|
|
|
if (recv(xfrm_sock, &req, sizeof(req), 0) < 0) {
|
|
pr_err("recv()");
|
|
goto out_close;
|
|
} else if (req.nh.nlmsg_type != NLMSG_ERROR) {
|
|
printk("expected NLMSG_ERROR, got %d", (int)req.nh.nlmsg_type);
|
|
goto out_close;
|
|
}
|
|
|
|
if (req.error) {
|
|
printk("NLMSG_ERROR: %d: %s", req.error, strerror(-req.error));
|
|
ret = req.error;
|
|
goto out_close;
|
|
}
|
|
|
|
if (recv(xfrm_listen, &req, sizeof(req), 0) < 0) {
|
|
pr_err("recv()");
|
|
goto out_close;
|
|
}
|
|
|
|
if (req.acq.aalgos != 0xfeed || req.acq.ealgos != 0xbaad
|
|
|| req.acq.calgos != 0xbabe) {
|
|
printk("xfrm_user_acquire has changed %x %x %x",
|
|
req.acq.aalgos, req.acq.ealgos, req.acq.calgos);
|
|
goto out_close;
|
|
}
|
|
|
|
ret = KSFT_PASS;
|
|
out_close:
|
|
close(xfrm_listen);
|
|
return ret;
|
|
}
|
|
|
|
static int xfrm_expire_state(int xfrm_sock, uint32_t *seq,
|
|
unsigned int nr, struct xfrm_desc *desc)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
union {
|
|
struct xfrm_user_expire expire;
|
|
int error;
|
|
};
|
|
} req;
|
|
struct in_addr src, dst;
|
|
int xfrm_listen = -1, ret = KSFT_FAIL;
|
|
uint32_t seq_listen;
|
|
|
|
src = inet_makeaddr(INADDR_B, child_ip(nr));
|
|
dst = inet_makeaddr(INADDR_B, grchild_ip(nr));
|
|
|
|
if (xfrm_state_add(xfrm_sock, (*seq)++, gen_spi(src), src, dst, desc)) {
|
|
printk("Failed to add xfrm state");
|
|
return KSFT_FAIL;
|
|
}
|
|
|
|
if (netlink_sock_bind(&xfrm_listen, &seq_listen, NETLINK_XFRM, XFRMNLGRP_EXPIRE))
|
|
return KSFT_FAIL;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.expire));
|
|
req.nh.nlmsg_type = XFRM_MSG_EXPIRE;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
req.nh.nlmsg_seq = (*seq)++;
|
|
|
|
memcpy(&req.expire.state.id.daddr, &dst, sizeof(dst));
|
|
req.expire.state.id.spi = gen_spi(src);
|
|
req.expire.state.id.proto = desc->proto;
|
|
req.expire.state.family = AF_INET;
|
|
req.expire.hard = 0xff;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
goto out_close;
|
|
}
|
|
|
|
if (recv(xfrm_sock, &req, sizeof(req), 0) < 0) {
|
|
pr_err("recv()");
|
|
goto out_close;
|
|
} else if (req.nh.nlmsg_type != NLMSG_ERROR) {
|
|
printk("expected NLMSG_ERROR, got %d", (int)req.nh.nlmsg_type);
|
|
goto out_close;
|
|
}
|
|
|
|
if (req.error) {
|
|
printk("NLMSG_ERROR: %d: %s", req.error, strerror(-req.error));
|
|
ret = req.error;
|
|
goto out_close;
|
|
}
|
|
|
|
if (recv(xfrm_listen, &req, sizeof(req), 0) < 0) {
|
|
pr_err("recv()");
|
|
goto out_close;
|
|
}
|
|
|
|
if (req.expire.hard != 0x1) {
|
|
printk("expire.hard is not set: %x", req.expire.hard);
|
|
goto out_close;
|
|
}
|
|
|
|
ret = KSFT_PASS;
|
|
out_close:
|
|
close(xfrm_listen);
|
|
return ret;
|
|
}
|
|
|
|
static int xfrm_expire_policy(int xfrm_sock, uint32_t *seq,
|
|
unsigned int nr, struct xfrm_desc *desc)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
union {
|
|
struct xfrm_user_polexpire expire;
|
|
int error;
|
|
};
|
|
} req;
|
|
struct in_addr src, dst, tunsrc, tundst;
|
|
int xfrm_listen = -1, ret = KSFT_FAIL;
|
|
uint32_t seq_listen;
|
|
|
|
src = inet_makeaddr(INADDR_B, child_ip(nr));
|
|
dst = inet_makeaddr(INADDR_B, grchild_ip(nr));
|
|
tunsrc = inet_makeaddr(INADDR_A, child_ip(nr));
|
|
tundst = inet_makeaddr(INADDR_A, grchild_ip(nr));
|
|
|
|
if (xfrm_policy_add(xfrm_sock, (*seq)++, gen_spi(src), src, dst,
|
|
XFRM_POLICY_OUT, tunsrc, tundst, desc->proto)) {
|
|
printk("Failed to add xfrm policy");
|
|
return KSFT_FAIL;
|
|
}
|
|
|
|
if (netlink_sock_bind(&xfrm_listen, &seq_listen, NETLINK_XFRM, XFRMNLGRP_EXPIRE))
|
|
return KSFT_FAIL;
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(req.expire));
|
|
req.nh.nlmsg_type = XFRM_MSG_POLEXPIRE;
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
|
|
req.nh.nlmsg_seq = (*seq)++;
|
|
|
|
/* Fill selector. */
|
|
memcpy(&req.expire.pol.sel.daddr, &dst, sizeof(tundst));
|
|
memcpy(&req.expire.pol.sel.saddr, &src, sizeof(tunsrc));
|
|
req.expire.pol.sel.family = AF_INET;
|
|
req.expire.pol.sel.prefixlen_d = PREFIX_LEN;
|
|
req.expire.pol.sel.prefixlen_s = PREFIX_LEN;
|
|
req.expire.pol.dir = XFRM_POLICY_OUT;
|
|
req.expire.hard = 0xff;
|
|
|
|
if (send(xfrm_sock, &req, req.nh.nlmsg_len, 0) < 0) {
|
|
pr_err("send()");
|
|
goto out_close;
|
|
}
|
|
|
|
if (recv(xfrm_sock, &req, sizeof(req), 0) < 0) {
|
|
pr_err("recv()");
|
|
goto out_close;
|
|
} else if (req.nh.nlmsg_type != NLMSG_ERROR) {
|
|
printk("expected NLMSG_ERROR, got %d", (int)req.nh.nlmsg_type);
|
|
goto out_close;
|
|
}
|
|
|
|
if (req.error) {
|
|
printk("NLMSG_ERROR: %d: %s", req.error, strerror(-req.error));
|
|
ret = req.error;
|
|
goto out_close;
|
|
}
|
|
|
|
if (recv(xfrm_listen, &req, sizeof(req), 0) < 0) {
|
|
pr_err("recv()");
|
|
goto out_close;
|
|
}
|
|
|
|
if (req.expire.hard != 0x1) {
|
|
printk("expire.hard is not set: %x", req.expire.hard);
|
|
goto out_close;
|
|
}
|
|
|
|
ret = KSFT_PASS;
|
|
out_close:
|
|
close(xfrm_listen);
|
|
return ret;
|
|
}
|
|
|
|
static int child_serv(int xfrm_sock, uint32_t *seq,
|
|
unsigned int nr, int cmd_fd, void *buf, struct xfrm_desc *desc)
|
|
{
|
|
struct in_addr src, dst, tunsrc, tundst;
|
|
struct test_desc msg;
|
|
int ret = KSFT_FAIL;
|
|
|
|
src = inet_makeaddr(INADDR_B, child_ip(nr));
|
|
dst = inet_makeaddr(INADDR_B, grchild_ip(nr));
|
|
tunsrc = inet_makeaddr(INADDR_A, child_ip(nr));
|
|
tundst = inet_makeaddr(INADDR_A, grchild_ip(nr));
|
|
|
|
/* UDP pinging without xfrm */
|
|
if (do_ping(cmd_fd, buf, page_size, src, true, 0, 0, udp_ping_send)) {
|
|
printk("ping failed before setting xfrm");
|
|
return KSFT_FAIL;
|
|
}
|
|
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.type = MSG_XFRM_PREPARE;
|
|
memcpy(&msg.body.xfrm_desc, desc, sizeof(*desc));
|
|
write_msg(cmd_fd, &msg, 1);
|
|
|
|
if (xfrm_prepare(xfrm_sock, seq, src, dst, tunsrc, tundst, desc->proto)) {
|
|
printk("failed to prepare xfrm");
|
|
goto cleanup;
|
|
}
|
|
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.type = MSG_XFRM_ADD;
|
|
memcpy(&msg.body.xfrm_desc, desc, sizeof(*desc));
|
|
write_msg(cmd_fd, &msg, 1);
|
|
if (xfrm_set(xfrm_sock, seq, src, dst, tunsrc, tundst, desc)) {
|
|
printk("failed to set xfrm");
|
|
goto delete;
|
|
}
|
|
|
|
/* UDP pinging with xfrm tunnel */
|
|
if (do_ping(cmd_fd, buf, page_size, tunsrc,
|
|
true, 0, 0, udp_ping_send)) {
|
|
printk("ping failed for xfrm");
|
|
goto delete;
|
|
}
|
|
|
|
ret = KSFT_PASS;
|
|
delete:
|
|
/* xfrm delete */
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.type = MSG_XFRM_DEL;
|
|
memcpy(&msg.body.xfrm_desc, desc, sizeof(*desc));
|
|
write_msg(cmd_fd, &msg, 1);
|
|
|
|
if (xfrm_delete(xfrm_sock, seq, src, dst, tunsrc, tundst, desc->proto)) {
|
|
printk("failed ping to remove xfrm");
|
|
ret = KSFT_FAIL;
|
|
}
|
|
|
|
cleanup:
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.type = MSG_XFRM_CLEANUP;
|
|
memcpy(&msg.body.xfrm_desc, desc, sizeof(*desc));
|
|
write_msg(cmd_fd, &msg, 1);
|
|
if (xfrm_cleanup(xfrm_sock, seq, src, dst, tunsrc, tundst)) {
|
|
printk("failed ping to cleanup xfrm");
|
|
ret = KSFT_FAIL;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static int child_f(unsigned int nr, int test_desc_fd, int cmd_fd, void *buf)
|
|
{
|
|
struct xfrm_desc desc;
|
|
struct test_desc msg;
|
|
int xfrm_sock = -1;
|
|
uint32_t seq;
|
|
|
|
if (switch_ns(nsfd_childa))
|
|
exit(KSFT_FAIL);
|
|
|
|
if (netlink_sock(&xfrm_sock, &seq, NETLINK_XFRM)) {
|
|
printk("Failed to open xfrm netlink socket");
|
|
exit(KSFT_FAIL);
|
|
}
|
|
|
|
/* Check that seq sock is ready, just for sure. */
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.type = MSG_ACK;
|
|
write_msg(cmd_fd, &msg, 1);
|
|
read_msg(cmd_fd, &msg, 1);
|
|
if (msg.type != MSG_ACK) {
|
|
printk("Ack failed");
|
|
exit(KSFT_FAIL);
|
|
}
|
|
|
|
for (;;) {
|
|
ssize_t received = read(test_desc_fd, &desc, sizeof(desc));
|
|
int ret;
|
|
|
|
if (received == 0) /* EOF */
|
|
break;
|
|
|
|
if (received != sizeof(desc)) {
|
|
pr_err("read() returned %zd", received);
|
|
exit(KSFT_FAIL);
|
|
}
|
|
|
|
switch (desc.type) {
|
|
case CREATE_TUNNEL:
|
|
ret = child_serv(xfrm_sock, &seq, nr,
|
|
cmd_fd, buf, &desc);
|
|
break;
|
|
case ALLOCATE_SPI:
|
|
ret = xfrm_state_allocspi(xfrm_sock, &seq,
|
|
-1, desc.proto);
|
|
break;
|
|
case MONITOR_ACQUIRE:
|
|
ret = xfrm_monitor_acquire(xfrm_sock, &seq, nr);
|
|
break;
|
|
case EXPIRE_STATE:
|
|
ret = xfrm_expire_state(xfrm_sock, &seq, nr, &desc);
|
|
break;
|
|
case EXPIRE_POLICY:
|
|
ret = xfrm_expire_policy(xfrm_sock, &seq, nr, &desc);
|
|
break;
|
|
default:
|
|
printk("Unknown desc type %d", desc.type);
|
|
exit(KSFT_FAIL);
|
|
}
|
|
write_test_result(ret, &desc);
|
|
}
|
|
|
|
close(xfrm_sock);
|
|
|
|
msg.type = MSG_EXIT;
|
|
write_msg(cmd_fd, &msg, 1);
|
|
exit(KSFT_PASS);
|
|
}
|
|
|
|
static void grand_child_serv(unsigned int nr, int cmd_fd, void *buf,
|
|
struct test_desc *msg, int xfrm_sock, uint32_t *seq)
|
|
{
|
|
struct in_addr src, dst, tunsrc, tundst;
|
|
bool tun_reply;
|
|
struct xfrm_desc *desc = &msg->body.xfrm_desc;
|
|
|
|
src = inet_makeaddr(INADDR_B, grchild_ip(nr));
|
|
dst = inet_makeaddr(INADDR_B, child_ip(nr));
|
|
tunsrc = inet_makeaddr(INADDR_A, grchild_ip(nr));
|
|
tundst = inet_makeaddr(INADDR_A, child_ip(nr));
|
|
|
|
switch (msg->type) {
|
|
case MSG_EXIT:
|
|
exit(KSFT_PASS);
|
|
case MSG_ACK:
|
|
write_msg(cmd_fd, msg, 1);
|
|
break;
|
|
case MSG_PING:
|
|
tun_reply = memcmp(&dst, &msg->body.ping.reply_ip, sizeof(in_addr_t));
|
|
/* UDP pinging without xfrm */
|
|
if (do_ping(cmd_fd, buf, page_size, tun_reply ? tunsrc : src,
|
|
false, msg->body.ping.port,
|
|
msg->body.ping.reply_ip, udp_ping_reply)) {
|
|
printk("ping failed before setting xfrm");
|
|
}
|
|
break;
|
|
case MSG_XFRM_PREPARE:
|
|
if (xfrm_prepare(xfrm_sock, seq, src, dst, tunsrc, tundst,
|
|
desc->proto)) {
|
|
xfrm_cleanup(xfrm_sock, seq, src, dst, tunsrc, tundst);
|
|
printk("failed to prepare xfrm");
|
|
}
|
|
break;
|
|
case MSG_XFRM_ADD:
|
|
if (xfrm_set(xfrm_sock, seq, src, dst, tunsrc, tundst, desc)) {
|
|
xfrm_cleanup(xfrm_sock, seq, src, dst, tunsrc, tundst);
|
|
printk("failed to set xfrm");
|
|
}
|
|
break;
|
|
case MSG_XFRM_DEL:
|
|
if (xfrm_delete(xfrm_sock, seq, src, dst, tunsrc, tundst,
|
|
desc->proto)) {
|
|
xfrm_cleanup(xfrm_sock, seq, src, dst, tunsrc, tundst);
|
|
printk("failed to remove xfrm");
|
|
}
|
|
break;
|
|
case MSG_XFRM_CLEANUP:
|
|
if (xfrm_cleanup(xfrm_sock, seq, src, dst, tunsrc, tundst)) {
|
|
printk("failed to cleanup xfrm");
|
|
}
|
|
break;
|
|
default:
|
|
printk("got unknown msg type %d", msg->type);
|
|
}
|
|
}
|
|
|
|
static int grand_child_f(unsigned int nr, int cmd_fd, void *buf)
|
|
{
|
|
struct test_desc msg;
|
|
int xfrm_sock = -1;
|
|
uint32_t seq;
|
|
|
|
if (switch_ns(nsfd_childb))
|
|
exit(KSFT_FAIL);
|
|
|
|
if (netlink_sock(&xfrm_sock, &seq, NETLINK_XFRM)) {
|
|
printk("Failed to open xfrm netlink socket");
|
|
exit(KSFT_FAIL);
|
|
}
|
|
|
|
do {
|
|
read_msg(cmd_fd, &msg, 1);
|
|
grand_child_serv(nr, cmd_fd, buf, &msg, xfrm_sock, &seq);
|
|
} while (1);
|
|
|
|
close(xfrm_sock);
|
|
exit(KSFT_FAIL);
|
|
}
|
|
|
|
static int start_child(unsigned int nr, char *veth, int test_desc_fd[2])
|
|
{
|
|
int cmd_sock[2];
|
|
void *data_map;
|
|
pid_t child;
|
|
|
|
if (init_child(nsfd_childa, veth, child_ip(nr), grchild_ip(nr)))
|
|
return -1;
|
|
|
|
if (init_child(nsfd_childb, veth, grchild_ip(nr), child_ip(nr)))
|
|
return -1;
|
|
|
|
child = fork();
|
|
if (child < 0) {
|
|
pr_err("fork()");
|
|
return -1;
|
|
} else if (child) {
|
|
/* in parent - selftest */
|
|
return switch_ns(nsfd_parent);
|
|
}
|
|
|
|
if (close(test_desc_fd[1])) {
|
|
pr_err("close()");
|
|
return -1;
|
|
}
|
|
|
|
/* child */
|
|
data_map = mmap(0, page_size, PROT_READ | PROT_WRITE,
|
|
MAP_SHARED | MAP_ANONYMOUS, -1, 0);
|
|
if (data_map == MAP_FAILED) {
|
|
pr_err("mmap()");
|
|
return -1;
|
|
}
|
|
|
|
randomize_buffer(data_map, page_size);
|
|
|
|
if (socketpair(PF_LOCAL, SOCK_SEQPACKET, 0, cmd_sock)) {
|
|
pr_err("socketpair()");
|
|
return -1;
|
|
}
|
|
|
|
child = fork();
|
|
if (child < 0) {
|
|
pr_err("fork()");
|
|
return -1;
|
|
} else if (child) {
|
|
if (close(cmd_sock[0])) {
|
|
pr_err("close()");
|
|
return -1;
|
|
}
|
|
return child_f(nr, test_desc_fd[0], cmd_sock[1], data_map);
|
|
}
|
|
if (close(cmd_sock[1])) {
|
|
pr_err("close()");
|
|
return -1;
|
|
}
|
|
return grand_child_f(nr, cmd_sock[0], data_map);
|
|
}
|
|
|
|
static void exit_usage(char **argv)
|
|
{
|
|
printk("Usage: %s [nr_process]", argv[0]);
|
|
exit(KSFT_FAIL);
|
|
}
|
|
|
|
static int __write_desc(int test_desc_fd, struct xfrm_desc *desc)
|
|
{
|
|
ssize_t ret;
|
|
|
|
ret = write(test_desc_fd, desc, sizeof(*desc));
|
|
|
|
if (ret == sizeof(*desc))
|
|
return 0;
|
|
|
|
pr_err("Writing test's desc failed %ld", ret);
|
|
|
|
return -1;
|
|
}
|
|
|
|
static int write_desc(int proto, int test_desc_fd,
|
|
char *a, char *e, char *c, char *ae)
|
|
{
|
|
struct xfrm_desc desc = {};
|
|
|
|
desc.type = CREATE_TUNNEL;
|
|
desc.proto = proto;
|
|
|
|
if (a)
|
|
strncpy(desc.a_algo, a, ALGO_LEN - 1);
|
|
if (e)
|
|
strncpy(desc.e_algo, e, ALGO_LEN - 1);
|
|
if (c)
|
|
strncpy(desc.c_algo, c, ALGO_LEN - 1);
|
|
if (ae)
|
|
strncpy(desc.ae_algo, ae, ALGO_LEN - 1);
|
|
|
|
return __write_desc(test_desc_fd, &desc);
|
|
}
|
|
|
|
int proto_list[] = { IPPROTO_AH, IPPROTO_COMP, IPPROTO_ESP };
|
|
char *ah_list[] = {
|
|
"digest_null", "hmac(md5)", "hmac(sha1)", "hmac(sha256)",
|
|
"hmac(sha384)", "hmac(sha512)", "hmac(rmd160)",
|
|
"xcbc(aes)", "cmac(aes)"
|
|
};
|
|
char *comp_list[] = {
|
|
"deflate",
|
|
#if 0
|
|
/* No compression backend realization */
|
|
"lzs", "lzjh"
|
|
#endif
|
|
};
|
|
char *e_list[] = {
|
|
"ecb(cipher_null)", "cbc(des)", "cbc(des3_ede)", "cbc(cast5)",
|
|
"cbc(blowfish)", "cbc(aes)", "cbc(serpent)", "cbc(camellia)",
|
|
"cbc(twofish)", "rfc3686(ctr(aes))"
|
|
};
|
|
char *ae_list[] = {
|
|
#if 0
|
|
/* not implemented */
|
|
"rfc4106(gcm(aes))", "rfc4309(ccm(aes))", "rfc4543(gcm(aes))",
|
|
"rfc7539esp(chacha20,poly1305)"
|
|
#endif
|
|
};
|
|
|
|
const unsigned int proto_plan = ARRAY_SIZE(ah_list) + ARRAY_SIZE(comp_list) \
|
|
+ (ARRAY_SIZE(ah_list) * ARRAY_SIZE(e_list)) \
|
|
+ ARRAY_SIZE(ae_list);
|
|
|
|
static int write_proto_plan(int fd, int proto)
|
|
{
|
|
unsigned int i;
|
|
|
|
switch (proto) {
|
|
case IPPROTO_AH:
|
|
for (i = 0; i < ARRAY_SIZE(ah_list); i++) {
|
|
if (write_desc(proto, fd, ah_list[i], 0, 0, 0))
|
|
return -1;
|
|
}
|
|
break;
|
|
case IPPROTO_COMP:
|
|
for (i = 0; i < ARRAY_SIZE(comp_list); i++) {
|
|
if (write_desc(proto, fd, 0, 0, comp_list[i], 0))
|
|
return -1;
|
|
}
|
|
break;
|
|
case IPPROTO_ESP:
|
|
for (i = 0; i < ARRAY_SIZE(ah_list); i++) {
|
|
int j;
|
|
|
|
for (j = 0; j < ARRAY_SIZE(e_list); j++) {
|
|
if (write_desc(proto, fd, ah_list[i],
|
|
e_list[j], 0, 0))
|
|
return -1;
|
|
}
|
|
}
|
|
for (i = 0; i < ARRAY_SIZE(ae_list); i++) {
|
|
if (write_desc(proto, fd, 0, 0, 0, ae_list[i]))
|
|
return -1;
|
|
}
|
|
break;
|
|
default:
|
|
printk("BUG: Specified unknown proto %d", proto);
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Some structures in xfrm uapi header differ in size between
|
|
* 64-bit and 32-bit ABI:
|
|
*
|
|
* 32-bit UABI | 64-bit UABI
|
|
* -------------------------------------|-------------------------------------
|
|
* sizeof(xfrm_usersa_info) = 220 | sizeof(xfrm_usersa_info) = 224
|
|
* sizeof(xfrm_userpolicy_info) = 164 | sizeof(xfrm_userpolicy_info) = 168
|
|
* sizeof(xfrm_userspi_info) = 228 | sizeof(xfrm_userspi_info) = 232
|
|
* sizeof(xfrm_user_acquire) = 276 | sizeof(xfrm_user_acquire) = 280
|
|
* sizeof(xfrm_user_expire) = 224 | sizeof(xfrm_user_expire) = 232
|
|
* sizeof(xfrm_user_polexpire) = 168 | sizeof(xfrm_user_polexpire) = 176
|
|
*
|
|
* Check the affected by the UABI difference structures.
|
|
*/
|
|
const unsigned int compat_plan = 4;
|
|
static int write_compat_struct_tests(int test_desc_fd)
|
|
{
|
|
struct xfrm_desc desc = {};
|
|
|
|
desc.type = ALLOCATE_SPI;
|
|
desc.proto = IPPROTO_AH;
|
|
strncpy(desc.a_algo, ah_list[0], ALGO_LEN - 1);
|
|
|
|
if (__write_desc(test_desc_fd, &desc))
|
|
return -1;
|
|
|
|
desc.type = MONITOR_ACQUIRE;
|
|
if (__write_desc(test_desc_fd, &desc))
|
|
return -1;
|
|
|
|
desc.type = EXPIRE_STATE;
|
|
if (__write_desc(test_desc_fd, &desc))
|
|
return -1;
|
|
|
|
desc.type = EXPIRE_POLICY;
|
|
if (__write_desc(test_desc_fd, &desc))
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int write_test_plan(int test_desc_fd)
|
|
{
|
|
unsigned int i;
|
|
pid_t child;
|
|
|
|
child = fork();
|
|
if (child < 0) {
|
|
pr_err("fork()");
|
|
return -1;
|
|
}
|
|
if (child) {
|
|
if (close(test_desc_fd))
|
|
printk("close(): %m");
|
|
return 0;
|
|
}
|
|
|
|
if (write_compat_struct_tests(test_desc_fd))
|
|
exit(KSFT_FAIL);
|
|
|
|
for (i = 0; i < ARRAY_SIZE(proto_list); i++) {
|
|
if (write_proto_plan(test_desc_fd, proto_list[i]))
|
|
exit(KSFT_FAIL);
|
|
}
|
|
|
|
exit(KSFT_PASS);
|
|
}
|
|
|
|
static int children_cleanup(void)
|
|
{
|
|
unsigned ret = KSFT_PASS;
|
|
|
|
while (1) {
|
|
int status;
|
|
pid_t p = wait(&status);
|
|
|
|
if ((p < 0) && errno == ECHILD)
|
|
break;
|
|
|
|
if (p < 0) {
|
|
pr_err("wait()");
|
|
return KSFT_FAIL;
|
|
}
|
|
|
|
if (!WIFEXITED(status)) {
|
|
ret = KSFT_FAIL;
|
|
continue;
|
|
}
|
|
|
|
if (WEXITSTATUS(status) == KSFT_FAIL)
|
|
ret = KSFT_FAIL;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
typedef void (*print_res)(const char *, ...);
|
|
|
|
static int check_results(void)
|
|
{
|
|
struct test_result tr = {};
|
|
struct xfrm_desc *d = &tr.desc;
|
|
int ret = KSFT_PASS;
|
|
|
|
while (1) {
|
|
ssize_t received = read(results_fd[0], &tr, sizeof(tr));
|
|
print_res result;
|
|
|
|
if (received == 0) /* EOF */
|
|
break;
|
|
|
|
if (received != sizeof(tr)) {
|
|
pr_err("read() returned %zd", received);
|
|
return KSFT_FAIL;
|
|
}
|
|
|
|
switch (tr.res) {
|
|
case KSFT_PASS:
|
|
result = ksft_test_result_pass;
|
|
break;
|
|
case KSFT_FAIL:
|
|
default:
|
|
result = ksft_test_result_fail;
|
|
ret = KSFT_FAIL;
|
|
}
|
|
|
|
result(" %s: [%u, '%s', '%s', '%s', '%s', %u]\n",
|
|
desc_name[d->type], (unsigned int)d->proto, d->a_algo,
|
|
d->e_algo, d->c_algo, d->ae_algo, d->icv_len);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
unsigned int nr_process = 1;
|
|
int route_sock = -1, ret = KSFT_SKIP;
|
|
int test_desc_fd[2];
|
|
uint32_t route_seq;
|
|
unsigned int i;
|
|
|
|
if (argc > 2)
|
|
exit_usage(argv);
|
|
|
|
if (argc > 1) {
|
|
char *endptr;
|
|
|
|
errno = 0;
|
|
nr_process = strtol(argv[1], &endptr, 10);
|
|
if ((errno == ERANGE && (nr_process == LONG_MAX || nr_process == LONG_MIN))
|
|
|| (errno != 0 && nr_process == 0)
|
|
|| (endptr == argv[1]) || (*endptr != '\0')) {
|
|
printk("Failed to parse [nr_process]");
|
|
exit_usage(argv);
|
|
}
|
|
|
|
if (nr_process > MAX_PROCESSES || !nr_process) {
|
|
printk("nr_process should be between [1; %u]",
|
|
MAX_PROCESSES);
|
|
exit_usage(argv);
|
|
}
|
|
}
|
|
|
|
srand(time(NULL));
|
|
page_size = sysconf(_SC_PAGESIZE);
|
|
if (page_size < 1)
|
|
ksft_exit_skip("sysconf(): %m\n");
|
|
|
|
if (pipe2(test_desc_fd, O_DIRECT) < 0)
|
|
ksft_exit_skip("pipe(): %m\n");
|
|
|
|
if (pipe2(results_fd, O_DIRECT) < 0)
|
|
ksft_exit_skip("pipe(): %m\n");
|
|
|
|
if (init_namespaces())
|
|
ksft_exit_skip("Failed to create namespaces\n");
|
|
|
|
if (netlink_sock(&route_sock, &route_seq, NETLINK_ROUTE))
|
|
ksft_exit_skip("Failed to open netlink route socket\n");
|
|
|
|
for (i = 0; i < nr_process; i++) {
|
|
char veth[VETH_LEN];
|
|
|
|
snprintf(veth, VETH_LEN, VETH_FMT, i);
|
|
|
|
if (veth_add(route_sock, route_seq++, veth, nsfd_childa, veth, nsfd_childb)) {
|
|
close(route_sock);
|
|
ksft_exit_fail_msg("Failed to create veth device");
|
|
}
|
|
|
|
if (start_child(i, veth, test_desc_fd)) {
|
|
close(route_sock);
|
|
ksft_exit_fail_msg("Child %u failed to start", i);
|
|
}
|
|
}
|
|
|
|
if (close(route_sock) || close(test_desc_fd[0]) || close(results_fd[1]))
|
|
ksft_exit_fail_msg("close(): %m");
|
|
|
|
ksft_set_plan(proto_plan + compat_plan);
|
|
|
|
if (write_test_plan(test_desc_fd[1]))
|
|
ksft_exit_fail_msg("Failed to write test plan to pipe");
|
|
|
|
ret = check_results();
|
|
|
|
if (children_cleanup() == KSFT_FAIL)
|
|
exit(KSFT_FAIL);
|
|
|
|
exit(ret);
|
|
}
|