163 lines
3.4 KiB
C
163 lines
3.4 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
|
|
#define _GNU_SOURCE
|
|
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <linux/if.h>
|
|
#include <linux/if_tun.h>
|
|
#include <linux/netlink.h>
|
|
#include <linux/rtnetlink.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/socket.h>
|
|
|
|
#include "../kselftest_harness.h"
|
|
|
|
static int tun_attach(int fd, char *dev)
|
|
{
|
|
struct ifreq ifr;
|
|
|
|
memset(&ifr, 0, sizeof(ifr));
|
|
strcpy(ifr.ifr_name, dev);
|
|
ifr.ifr_flags = IFF_ATTACH_QUEUE;
|
|
|
|
return ioctl(fd, TUNSETQUEUE, (void *) &ifr);
|
|
}
|
|
|
|
static int tun_detach(int fd, char *dev)
|
|
{
|
|
struct ifreq ifr;
|
|
|
|
memset(&ifr, 0, sizeof(ifr));
|
|
strcpy(ifr.ifr_name, dev);
|
|
ifr.ifr_flags = IFF_DETACH_QUEUE;
|
|
|
|
return ioctl(fd, TUNSETQUEUE, (void *) &ifr);
|
|
}
|
|
|
|
static int tun_alloc(char *dev)
|
|
{
|
|
struct ifreq ifr;
|
|
int fd, err;
|
|
|
|
fd = open("/dev/net/tun", O_RDWR);
|
|
if (fd < 0) {
|
|
fprintf(stderr, "can't open tun: %s\n", strerror(errno));
|
|
return fd;
|
|
}
|
|
|
|
memset(&ifr, 0, sizeof(ifr));
|
|
strcpy(ifr.ifr_name, dev);
|
|
ifr.ifr_flags = IFF_TAP | IFF_NAPI | IFF_MULTI_QUEUE;
|
|
|
|
err = ioctl(fd, TUNSETIFF, (void *) &ifr);
|
|
if (err < 0) {
|
|
fprintf(stderr, "can't TUNSETIFF: %s\n", strerror(errno));
|
|
close(fd);
|
|
return err;
|
|
}
|
|
strcpy(dev, ifr.ifr_name);
|
|
return fd;
|
|
}
|
|
|
|
static int tun_delete(char *dev)
|
|
{
|
|
struct {
|
|
struct nlmsghdr nh;
|
|
struct ifinfomsg ifm;
|
|
unsigned char data[64];
|
|
} req;
|
|
struct rtattr *rta;
|
|
int ret, rtnl;
|
|
|
|
rtnl = socket(AF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE);
|
|
if (rtnl < 0) {
|
|
fprintf(stderr, "can't open rtnl: %s\n", strerror(errno));
|
|
return 1;
|
|
}
|
|
|
|
memset(&req, 0, sizeof(req));
|
|
req.nh.nlmsg_len = NLMSG_ALIGN(NLMSG_LENGTH(sizeof(req.ifm)));
|
|
req.nh.nlmsg_flags = NLM_F_REQUEST;
|
|
req.nh.nlmsg_type = RTM_DELLINK;
|
|
|
|
req.ifm.ifi_family = AF_UNSPEC;
|
|
|
|
rta = (struct rtattr *)(((char *)&req) + NLMSG_ALIGN(req.nh.nlmsg_len));
|
|
rta->rta_type = IFLA_IFNAME;
|
|
rta->rta_len = RTA_LENGTH(IFNAMSIZ);
|
|
req.nh.nlmsg_len += rta->rta_len;
|
|
memcpy(RTA_DATA(rta), dev, IFNAMSIZ);
|
|
|
|
ret = send(rtnl, &req, req.nh.nlmsg_len, 0);
|
|
if (ret < 0)
|
|
fprintf(stderr, "can't send: %s\n", strerror(errno));
|
|
ret = (unsigned int)ret != req.nh.nlmsg_len;
|
|
|
|
close(rtnl);
|
|
return ret;
|
|
}
|
|
|
|
FIXTURE(tun)
|
|
{
|
|
char ifname[IFNAMSIZ];
|
|
int fd, fd2;
|
|
};
|
|
|
|
FIXTURE_SETUP(tun)
|
|
{
|
|
memset(self->ifname, 0, sizeof(self->ifname));
|
|
|
|
self->fd = tun_alloc(self->ifname);
|
|
ASSERT_GE(self->fd, 0);
|
|
|
|
self->fd2 = tun_alloc(self->ifname);
|
|
ASSERT_GE(self->fd2, 0);
|
|
}
|
|
|
|
FIXTURE_TEARDOWN(tun)
|
|
{
|
|
if (self->fd >= 0)
|
|
close(self->fd);
|
|
if (self->fd2 >= 0)
|
|
close(self->fd2);
|
|
}
|
|
|
|
TEST_F(tun, delete_detach_close) {
|
|
EXPECT_EQ(tun_delete(self->ifname), 0);
|
|
EXPECT_EQ(tun_detach(self->fd, self->ifname), -1);
|
|
EXPECT_EQ(errno, 22);
|
|
}
|
|
|
|
TEST_F(tun, detach_delete_close) {
|
|
EXPECT_EQ(tun_detach(self->fd, self->ifname), 0);
|
|
EXPECT_EQ(tun_delete(self->ifname), 0);
|
|
}
|
|
|
|
TEST_F(tun, detach_close_delete) {
|
|
EXPECT_EQ(tun_detach(self->fd, self->ifname), 0);
|
|
close(self->fd);
|
|
self->fd = -1;
|
|
EXPECT_EQ(tun_delete(self->ifname), 0);
|
|
}
|
|
|
|
TEST_F(tun, reattach_delete_close) {
|
|
EXPECT_EQ(tun_detach(self->fd, self->ifname), 0);
|
|
EXPECT_EQ(tun_attach(self->fd, self->ifname), 0);
|
|
EXPECT_EQ(tun_delete(self->ifname), 0);
|
|
}
|
|
|
|
TEST_F(tun, reattach_close_delete) {
|
|
EXPECT_EQ(tun_detach(self->fd, self->ifname), 0);
|
|
EXPECT_EQ(tun_attach(self->fd, self->ifname), 0);
|
|
close(self->fd);
|
|
self->fd = -1;
|
|
EXPECT_EQ(tun_delete(self->ifname), 0);
|
|
}
|
|
|
|
TEST_HARNESS_MAIN
|