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.
213 lines
6.4 KiB
213 lines
6.4 KiB
/* |
|
* Check decoding of getsockopt and setsockopt for SOL_NETLINK level. |
|
* |
|
* Copyright (c) 2017 Dmitry V. Levin <ldv@altlinux.org> |
|
* 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 "netlink.h" |
|
#include <stdio.h> |
|
|
|
#ifndef SOL_NETLINK |
|
# define SOL_NETLINK 270 |
|
#endif |
|
|
|
static int rc; |
|
static const char *errstr; |
|
|
|
static int |
|
get_sockopt(int fd, int name, void *val, socklen_t *len) |
|
{ |
|
rc = getsockopt(fd, SOL_NETLINK, name, val, len); |
|
errstr = sprintrc(rc); |
|
return rc; |
|
} |
|
|
|
static int |
|
set_sockopt(int fd, int name, void *val, socklen_t len) |
|
{ |
|
rc = setsockopt(fd, SOL_NETLINK, name, val, len); |
|
errstr = sprintrc(rc); |
|
return rc; |
|
} |
|
|
|
int |
|
main(void) |
|
{ |
|
static const struct { |
|
int val; |
|
const char *str; |
|
} names[] = { |
|
#ifdef NETLINK_ADD_MEMBERSHIP |
|
{ ARG_STR(NETLINK_ADD_MEMBERSHIP) }, |
|
#endif |
|
#ifdef NETLINK_DROP_MEMBERSHIP |
|
{ ARG_STR(NETLINK_DROP_MEMBERSHIP) }, |
|
#endif |
|
#ifdef NETLINK_PKTINFO |
|
{ ARG_STR(NETLINK_PKTINFO) }, |
|
#endif |
|
#ifdef NETLINK_BROADCAST_ERROR |
|
{ ARG_STR(NETLINK_BROADCAST_ERROR) }, |
|
#endif |
|
#ifdef NETLINK_NO_ENOBUFS |
|
{ ARG_STR(NETLINK_NO_ENOBUFS) }, |
|
#endif |
|
#ifdef NETLINK_RX_RING |
|
{ ARG_STR(NETLINK_RX_RING) }, |
|
#endif |
|
#ifdef NETLINK_TX_RING |
|
{ ARG_STR(NETLINK_TX_RING) }, |
|
#endif |
|
#ifdef NETLINK_LISTEN_ALL_NSID |
|
{ ARG_STR(NETLINK_LISTEN_ALL_NSID) }, |
|
#endif |
|
#ifdef NETLINK_LIST_MEMBERSHIPS |
|
{ ARG_STR(NETLINK_LIST_MEMBERSHIPS) }, |
|
#endif |
|
#ifdef NETLINK_CAP_ACK |
|
{ ARG_STR(NETLINK_CAP_ACK) }, |
|
#endif |
|
#ifdef NETLINK_EXT_ACK |
|
{ ARG_STR(NETLINK_EXT_ACK) }, |
|
#endif |
|
}; |
|
|
|
TAIL_ALLOC_OBJECT_CONST_PTR(int, val); |
|
TAIL_ALLOC_OBJECT_CONST_PTR(socklen_t, len); |
|
void *const efault = val + 1; |
|
int fd = socket(AF_NETLINK, SOCK_RAW, 0); |
|
if (fd < 0) |
|
perror_msg_and_skip("socket AF_NETLINK SOCK_RAW"); |
|
unsigned int i; |
|
|
|
for (i = 0; i < ARRAY_SIZE(names); ++i) { |
|
/* getsockopt */ |
|
|
|
/* classic */ |
|
*len = sizeof(*val); |
|
get_sockopt(fd, names[i].val, val, len); |
|
printf("getsockopt(%d, SOL_NETLINK, %s, ", fd, names[i].str); |
|
if (rc) |
|
printf("%p", val); |
|
else |
|
printf("[%d]", *val); |
|
printf(", [%d]) = %s\n", *len, errstr); |
|
|
|
/* optlen larger than necessary - shortened */ |
|
*len = sizeof(*val) + 1; |
|
get_sockopt(fd, names[i].val, val, len); |
|
printf("getsockopt(%d, SOL_NETLINK, %s, ", fd, names[i].str); |
|
if (rc) |
|
printf("%p", val); |
|
else |
|
printf("[%d]", *val); |
|
printf(", [%d", (int) sizeof(*val) + 1); |
|
if ((int) sizeof(*val) + 1 != *len) |
|
printf("->%d", *len); |
|
printf("]) = %s\n", errstr); |
|
|
|
/* zero optlen - print returned optlen */ |
|
*len = 0; |
|
get_sockopt(fd, names[i].val, NULL, len); |
|
printf("getsockopt(%d, SOL_NETLINK, %s, NULL, [0", |
|
fd, names[i].str); |
|
if (*len) |
|
printf("->%d", *len); |
|
printf("]) = %s\n", errstr); |
|
|
|
#ifdef NETLINK_LIST_MEMBERSHIPS |
|
if (names[i].val != NETLINK_LIST_MEMBERSHIPS) { |
|
#endif |
|
/* optlen shorter than necessary - print address */ |
|
*len = sizeof(*val) - 1; |
|
get_sockopt(fd, names[i].val, val, len); |
|
printf("getsockopt(%d, SOL_NETLINK, %s, %p, [%d", |
|
fd, names[i].str, val, (int) sizeof(*val) - 1); |
|
if ((int) sizeof(*val) - 1 != *len) |
|
printf("->%d", *len); |
|
printf("]) = %s\n", errstr); |
|
#ifdef NETLINK_LIST_MEMBERSHIPS |
|
} else { |
|
/* optlen shorter than required for the first element */ |
|
*len = sizeof(*val) - 1; |
|
get_sockopt(fd, names[i].val, efault, len); |
|
printf("getsockopt(%d, SOL_NETLINK, %s, ", |
|
fd, names[i].str); |
|
if (rc) |
|
printf("%p", efault); |
|
else |
|
printf("[]"); |
|
printf(", [%d", (int) sizeof(*val) - 1); |
|
if ((int) sizeof(*val) - 1 != *len) |
|
printf("->%d", *len); |
|
printf("]) = %s\n", errstr); |
|
} |
|
#endif |
|
|
|
/* optval EFAULT - print address */ |
|
*len = sizeof(*val); |
|
get_sockopt(fd, names[i].val, efault, len); |
|
printf("getsockopt(%d, SOL_NETLINK, %s, %p, [%d]) = %s\n", |
|
fd, names[i].str, efault, *len, errstr); |
|
|
|
/* optlen EFAULT - print address */ |
|
get_sockopt(fd, names[i].val, val, len + 1); |
|
printf("getsockopt(%d, SOL_NETLINK, %s, %p, %p) = %s\n", |
|
fd, names[i].str, val, len + 1, errstr); |
|
|
|
/* setsockopt */ |
|
|
|
/* classic */ |
|
*val = 0xdefaced; |
|
set_sockopt(fd, names[i].val, val, sizeof(*val)); |
|
printf("setsockopt(%d, SOL_NETLINK, %s, [%d], %d) = %s\n", |
|
fd, names[i].str, *val, (int) sizeof(*val), errstr); |
|
|
|
/* optlen larger than necessary - shortened */ |
|
set_sockopt(fd, names[i].val, val, sizeof(*val) + 1); |
|
printf("setsockopt(%d, SOL_NETLINK, %s, [%d], %d) = %s\n", |
|
fd, names[i].str, *val, (int) sizeof(*val) + 1, errstr); |
|
|
|
/* optlen < 0 - print address */ |
|
set_sockopt(fd, names[i].val, val, -1U); |
|
printf("setsockopt(%d, SOL_NETLINK, %s, %p, -1) = %s\n", |
|
fd, names[i].str, val, errstr); |
|
|
|
/* optlen smaller than necessary - print address */ |
|
set_sockopt(fd, names[i].val, val, sizeof(*val) - 1); |
|
printf("setsockopt(%d, SOL_NETLINK, %s, %p, %d) = %s\n", |
|
fd, names[i].str, val, (int) sizeof(*val) - 1, errstr); |
|
|
|
/* optval EFAULT - print address */ |
|
set_sockopt(fd, names[i].val, efault, sizeof(*val)); |
|
printf("setsockopt(%d, SOL_NETLINK, %s, %p, %d) = %s\n", |
|
fd, names[i].str, efault, (int) sizeof(*val), errstr); |
|
} |
|
|
|
puts("+++ exited with 0 +++"); |
|
return 0; |
|
}
|
|
|