1 /* 2 * Check decoding of getsockopt and setsockopt for SOL_NETLINK level. 3 * 4 * Copyright (c) 2017 Dmitry V. Levin <ldv@altlinux.org> 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. The name of the author may not be used to endorse or promote products 16 * derived from this software without specific prior written permission. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 */ 29 30 #include "tests.h" 31 #include "netlink.h" 32 #include <stdio.h> 33 34 #ifndef SOL_NETLINK 35 # define SOL_NETLINK 270 36 #endif 37 38 static int rc; 39 static const char *errstr; 40 41 static int 42 get_sockopt(int fd, int name, void *val, socklen_t *len) 43 { 44 rc = getsockopt(fd, SOL_NETLINK, name, val, len); 45 errstr = sprintrc(rc); 46 return rc; 47 } 48 49 static int 50 set_sockopt(int fd, int name, void *val, socklen_t len) 51 { 52 rc = setsockopt(fd, SOL_NETLINK, name, val, len); 53 errstr = sprintrc(rc); 54 return rc; 55 } 56 57 int 58 main(void) 59 { 60 static const struct { 61 int val; 62 const char *str; 63 } names[] = { 64 #ifdef NETLINK_ADD_MEMBERSHIP 65 { ARG_STR(NETLINK_ADD_MEMBERSHIP) }, 66 #endif 67 #ifdef NETLINK_DROP_MEMBERSHIP 68 { ARG_STR(NETLINK_DROP_MEMBERSHIP) }, 69 #endif 70 #ifdef NETLINK_PKTINFO 71 { ARG_STR(NETLINK_PKTINFO) }, 72 #endif 73 #ifdef NETLINK_BROADCAST_ERROR 74 { ARG_STR(NETLINK_BROADCAST_ERROR) }, 75 #endif 76 #ifdef NETLINK_NO_ENOBUFS 77 { ARG_STR(NETLINK_NO_ENOBUFS) }, 78 #endif 79 #ifdef NETLINK_RX_RING 80 { ARG_STR(NETLINK_RX_RING) }, 81 #endif 82 #ifdef NETLINK_TX_RING 83 { ARG_STR(NETLINK_TX_RING) }, 84 #endif 85 #ifdef NETLINK_LISTEN_ALL_NSID 86 { ARG_STR(NETLINK_LISTEN_ALL_NSID) }, 87 #endif 88 #ifdef NETLINK_LIST_MEMBERSHIPS 89 { ARG_STR(NETLINK_LIST_MEMBERSHIPS) }, 90 #endif 91 #ifdef NETLINK_CAP_ACK 92 { ARG_STR(NETLINK_CAP_ACK) }, 93 #endif 94 #ifdef NETLINK_EXT_ACK 95 { ARG_STR(NETLINK_EXT_ACK) }, 96 #endif 97 }; 98 99 TAIL_ALLOC_OBJECT_CONST_PTR(int, val); 100 TAIL_ALLOC_OBJECT_CONST_PTR(socklen_t, len); 101 void *const efault = val + 1; 102 int fd = socket(AF_NETLINK, SOCK_RAW, 0); 103 if (fd < 0) 104 perror_msg_and_skip("socket AF_NETLINK SOCK_RAW"); 105 unsigned int i; 106 107 for (i = 0; i < ARRAY_SIZE(names); ++i) { 108 /* getsockopt */ 109 110 /* classic */ 111 *len = sizeof(*val); 112 get_sockopt(fd, names[i].val, val, len); 113 printf("getsockopt(%d, SOL_NETLINK, %s, ", fd, names[i].str); 114 if (rc) 115 printf("%p", val); 116 else 117 printf("[%d]", *val); 118 printf(", [%d]) = %s\n", *len, errstr); 119 120 /* optlen larger than necessary - shortened */ 121 *len = sizeof(*val) + 1; 122 get_sockopt(fd, names[i].val, val, len); 123 printf("getsockopt(%d, SOL_NETLINK, %s, ", fd, names[i].str); 124 if (rc) 125 printf("%p", val); 126 else 127 printf("[%d]", *val); 128 printf(", [%d", (int) sizeof(*val) + 1); 129 if ((int) sizeof(*val) + 1 != *len) 130 printf("->%d", *len); 131 printf("]) = %s\n", errstr); 132 133 /* zero optlen - print returned optlen */ 134 *len = 0; 135 get_sockopt(fd, names[i].val, NULL, len); 136 printf("getsockopt(%d, SOL_NETLINK, %s, NULL, [0", 137 fd, names[i].str); 138 if (*len) 139 printf("->%d", *len); 140 printf("]) = %s\n", errstr); 141 142 #ifdef NETLINK_LIST_MEMBERSHIPS 143 if (names[i].val != NETLINK_LIST_MEMBERSHIPS) { 144 #endif 145 /* optlen shorter than necessary - print address */ 146 *len = sizeof(*val) - 1; 147 get_sockopt(fd, names[i].val, val, len); 148 printf("getsockopt(%d, SOL_NETLINK, %s, %p, [%d", 149 fd, names[i].str, val, (int) sizeof(*val) - 1); 150 if ((int) sizeof(*val) - 1 != *len) 151 printf("->%d", *len); 152 printf("]) = %s\n", errstr); 153 #ifdef NETLINK_LIST_MEMBERSHIPS 154 } else { 155 /* optlen shorter than required for the first element */ 156 *len = sizeof(*val) - 1; 157 get_sockopt(fd, names[i].val, efault, len); 158 printf("getsockopt(%d, SOL_NETLINK, %s, ", 159 fd, names[i].str); 160 if (rc) 161 printf("%p", efault); 162 else 163 printf("[]"); 164 printf(", [%d", (int) sizeof(*val) - 1); 165 if ((int) sizeof(*val) - 1 != *len) 166 printf("->%d", *len); 167 printf("]) = %s\n", errstr); 168 } 169 #endif 170 171 /* optval EFAULT - print address */ 172 *len = sizeof(*val); 173 get_sockopt(fd, names[i].val, efault, len); 174 printf("getsockopt(%d, SOL_NETLINK, %s, %p, [%d]) = %s\n", 175 fd, names[i].str, efault, *len, errstr); 176 177 /* optlen EFAULT - print address */ 178 get_sockopt(fd, names[i].val, val, len + 1); 179 printf("getsockopt(%d, SOL_NETLINK, %s, %p, %p) = %s\n", 180 fd, names[i].str, val, len + 1, errstr); 181 182 /* setsockopt */ 183 184 /* classic */ 185 *val = 0xdefaced; 186 set_sockopt(fd, names[i].val, val, sizeof(*val)); 187 printf("setsockopt(%d, SOL_NETLINK, %s, [%d], %d) = %s\n", 188 fd, names[i].str, *val, (int) sizeof(*val), errstr); 189 190 /* optlen larger than necessary - shortened */ 191 set_sockopt(fd, names[i].val, val, sizeof(*val) + 1); 192 printf("setsockopt(%d, SOL_NETLINK, %s, [%d], %d) = %s\n", 193 fd, names[i].str, *val, (int) sizeof(*val) + 1, errstr); 194 195 /* optlen < 0 - print address */ 196 set_sockopt(fd, names[i].val, val, -1U); 197 printf("setsockopt(%d, SOL_NETLINK, %s, %p, -1) = %s\n", 198 fd, names[i].str, val, errstr); 199 200 /* optlen smaller than necessary - print address */ 201 set_sockopt(fd, names[i].val, val, sizeof(*val) - 1); 202 printf("setsockopt(%d, SOL_NETLINK, %s, %p, %d) = %s\n", 203 fd, names[i].str, val, (int) sizeof(*val) - 1, errstr); 204 205 /* optval EFAULT - print address */ 206 set_sockopt(fd, names[i].val, efault, sizeof(*val)); 207 printf("setsockopt(%d, SOL_NETLINK, %s, %p, %d) = %s\n", 208 fd, names[i].str, efault, (int) sizeof(*val), errstr); 209 } 210 211 puts("+++ exited with 0 +++"); 212 return 0; 213 } 214