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
get_sockopt(int fd,int name,void * val,socklen_t * len)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
set_sockopt(int fd,int name,void * val,socklen_t len)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
main(void)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