From 7005bcd451920ef5ec01295f17620ff155620b7a Mon Sep 17 00:00:00 2001 From: JingPiao Chen Date: Mon, 5 Jun 2017 22:05:07 +0800 Subject: [PATCH] tests: add create_nl_socket function to libtests * tests/create_nl_socket.c: New file. * tests/tests.h (create_nl_socket): New prototype. * tests/Makefile.am (libtests_a_SOURCES): Add create_nl_socket.c. --- tests/Makefile.am | 1 + tests/create_nl_socket.c | 50 ++++++++++++++++++++++++++++++++++++++++ tests/tests.h | 3 +++ 3 files changed, 54 insertions(+) create mode 100644 tests/create_nl_socket.c diff --git a/tests/Makefile.am b/tests/Makefile.am index 3f5b7659..0490ae58 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -42,6 +42,7 @@ AM_CPPFLAGS = $(ARCH_MFLAGS) \ AM_LDFLAGS = $(ARCH_MFLAGS) libtests_a_SOURCES = \ + create_nl_socket.c \ errno2name.c \ error_msg.c \ fill_memory.c \ diff --git a/tests/create_nl_socket.c b/tests/create_nl_socket.c new file mode 100644 index 00000000..c43669f6 --- /dev/null +++ b/tests/create_nl_socket.c @@ -0,0 +1,50 @@ +/* + * Copyright (c) 2015-2016 Dmitry V. Levin + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. The name of the author may not be used to endorse or promote products + * derived from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR + * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. + * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, + * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT + * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF + * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "tests.h" +#include +#include +#include + +int +create_nl_socket(int proto) +{ + struct sockaddr_nl addr; + socklen_t len = sizeof(addr); + int fd; + + memset(&addr, 0, sizeof(addr)); + addr.nl_family = AF_NETLINK; + + if ((fd = socket(AF_NETLINK, SOCK_RAW, proto)) == -1) + perror_msg_and_skip("socket AF_NETLINK"); + + if (bind(fd, (struct sockaddr *) &addr, len)) + perror_msg_and_skip("bind"); + + return fd; +} diff --git a/tests/tests.h b/tests/tests.h index dcc10576..fd7f194a 100644 --- a/tests/tests.h +++ b/tests/tests.h @@ -167,6 +167,9 @@ struct timespec; int recv_mmsg(int, struct mmsghdr *, unsigned int, unsigned int, struct timespec *); int send_mmsg(int, struct mmsghdr *, unsigned int, unsigned int); +/* Create a netlink socket. */ +int create_nl_socket(int); + /* Create a pipe with maximized descriptor numbers. */ void pipe_maxfd(int pipefd[2]); -- 2.40.0