~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

TOMOYO Linux Cross Reference
Linux/tools/testing/selftests/net/ipv6_flowlabel.c

Version: ~ [ linux-5.19-rc8 ] ~ [ linux-5.18.14 ] ~ [ linux-5.17.15 ] ~ [ linux-5.16.20 ] ~ [ linux-5.15.57 ] ~ [ linux-5.14.21 ] ~ [ linux-5.13.19 ] ~ [ linux-5.12.19 ] ~ [ linux-5.11.22 ] ~ [ linux-5.10.133 ] ~ [ linux-5.9.16 ] ~ [ linux-5.8.18 ] ~ [ linux-5.7.19 ] ~ [ linux-5.6.19 ] ~ [ linux-5.5.19 ] ~ [ linux-5.4.207 ] ~ [ linux-5.3.18 ] ~ [ linux-5.2.21 ] ~ [ linux-5.1.21 ] ~ [ linux-5.0.21 ] ~ [ linux-4.20.17 ] ~ [ linux-4.19.253 ] ~ [ linux-4.18.20 ] ~ [ linux-4.17.19 ] ~ [ linux-4.16.18 ] ~ [ linux-4.15.18 ] ~ [ linux-4.14.289 ] ~ [ linux-4.13.16 ] ~ [ linux-4.12.14 ] ~ [ linux-4.11.12 ] ~ [ linux-4.10.17 ] ~ [ linux-4.9.324 ] ~ [ linux-4.8.17 ] ~ [ linux-4.7.10 ] ~ [ linux-4.6.7 ] ~ [ linux-4.5.7 ] ~ [ linux-4.4.302 ] ~ [ linux-4.3.6 ] ~ [ linux-4.2.8 ] ~ [ linux-4.1.52 ] ~ [ linux-4.0.9 ] ~ [ linux-3.10.108 ] ~ [ linux-2.6.32.71 ] ~ [ linux-2.6.0 ] ~ [ linux-2.4.37.11 ] ~ [ unix-v6-master ] ~ [ ccs-tools-1.8.9 ] ~ [ policy-sample ] ~
Architecture: ~ [ i386 ] ~ [ alpha ] ~ [ m68k ] ~ [ mips ] ~ [ ppc ] ~ [ sparc ] ~ [ sparc64 ] ~

  1 // SPDX-License-Identifier: GPL-2.0
  2 /* Test IPV6_FLOWINFO cmsg on send and recv */
  3 
  4 #define _GNU_SOURCE
  5 
  6 #include <arpa/inet.h>
  7 #include <asm/byteorder.h>
  8 #include <error.h>
  9 #include <errno.h>
 10 #include <fcntl.h>
 11 #include <limits.h>
 12 #include <linux/in6.h>
 13 #include <stdbool.h>
 14 #include <stdio.h>
 15 #include <stdint.h>
 16 #include <stdlib.h>
 17 #include <string.h>
 18 #include <sys/socket.h>
 19 #include <sys/stat.h>
 20 #include <sys/time.h>
 21 #include <sys/types.h>
 22 #include <unistd.h>
 23 
 24 /* uapi/glibc weirdness may leave this undefined */
 25 #ifndef IPV6_FLOWINFO
 26 #define IPV6_FLOWINFO 11
 27 #endif
 28 
 29 #ifndef IPV6_FLOWLABEL_MGR
 30 #define IPV6_FLOWLABEL_MGR 32
 31 #endif
 32 
 33 #define FLOWLABEL_WILDCARD      ((uint32_t) -1)
 34 
 35 static const char cfg_data[]    = "a";
 36 static uint32_t cfg_label       = 1;
 37 
 38 static void do_send(int fd, bool with_flowlabel, uint32_t flowlabel)
 39 {
 40         char control[CMSG_SPACE(sizeof(flowlabel))] = {0};
 41         struct msghdr msg = {0};
 42         struct iovec iov = {0};
 43         int ret;
 44 
 45         iov.iov_base = (char *)cfg_data;
 46         iov.iov_len = sizeof(cfg_data);
 47 
 48         msg.msg_iov = &iov;
 49         msg.msg_iovlen = 1;
 50 
 51         if (with_flowlabel) {
 52                 struct cmsghdr *cm;
 53 
 54                 cm = (void *)control;
 55                 cm->cmsg_len = CMSG_LEN(sizeof(flowlabel));
 56                 cm->cmsg_level = SOL_IPV6;
 57                 cm->cmsg_type = IPV6_FLOWINFO;
 58                 *(uint32_t *)CMSG_DATA(cm) = htonl(flowlabel);
 59 
 60                 msg.msg_control = control;
 61                 msg.msg_controllen = sizeof(control);
 62         }
 63 
 64         ret = sendmsg(fd, &msg, 0);
 65         if (ret == -1)
 66                 error(1, errno, "send");
 67 
 68         if (with_flowlabel)
 69                 fprintf(stderr, "sent with label %u\n", flowlabel);
 70         else
 71                 fprintf(stderr, "sent without label\n");
 72 }
 73 
 74 static void do_recv(int fd, bool with_flowlabel, uint32_t expect)
 75 {
 76         char control[CMSG_SPACE(sizeof(expect))];
 77         char data[sizeof(cfg_data)];
 78         struct msghdr msg = {0};
 79         struct iovec iov = {0};
 80         struct cmsghdr *cm;
 81         uint32_t flowlabel;
 82         int ret;
 83 
 84         iov.iov_base = data;
 85         iov.iov_len = sizeof(data);
 86 
 87         msg.msg_iov = &iov;
 88         msg.msg_iovlen = 1;
 89 
 90         memset(control, 0, sizeof(control));
 91         msg.msg_control = control;
 92         msg.msg_controllen = sizeof(control);
 93 
 94         ret = recvmsg(fd, &msg, 0);
 95         if (ret == -1)
 96                 error(1, errno, "recv");
 97         if (msg.msg_flags & (MSG_TRUNC | MSG_CTRUNC))
 98                 error(1, 0, "recv: truncated");
 99         if (ret != sizeof(cfg_data))
100                 error(1, 0, "recv: length mismatch");
101         if (memcmp(data, cfg_data, sizeof(data)))
102                 error(1, 0, "recv: data mismatch");
103 
104         cm = CMSG_FIRSTHDR(&msg);
105         if (with_flowlabel) {
106                 if (!cm)
107                         error(1, 0, "recv: missing cmsg");
108                 if (CMSG_NXTHDR(&msg, cm))
109                         error(1, 0, "recv: too many cmsg");
110                 if (cm->cmsg_level != SOL_IPV6 ||
111                     cm->cmsg_type != IPV6_FLOWINFO)
112                         error(1, 0, "recv: unexpected cmsg level or type");
113 
114                 flowlabel = ntohl(*(uint32_t *)CMSG_DATA(cm));
115                 fprintf(stderr, "recv with label %u\n", flowlabel);
116 
117                 if (expect != FLOWLABEL_WILDCARD && expect != flowlabel)
118                         fprintf(stderr, "recv: incorrect flowlabel %u != %u\n",
119                                         flowlabel, expect);
120 
121         } else {
122                 fprintf(stderr, "recv without label\n");
123         }
124 }
125 
126 static bool get_autoflowlabel_enabled(void)
127 {
128         int fd, ret;
129         char val;
130 
131         fd = open("/proc/sys/net/ipv6/auto_flowlabels", O_RDONLY);
132         if (fd == -1)
133                 error(1, errno, "open sysctl");
134 
135         ret = read(fd, &val, 1);
136         if (ret == -1)
137                 error(1, errno, "read sysctl");
138         if (ret == 0)
139                 error(1, 0, "read sysctl: 0");
140 
141         if (close(fd))
142                 error(1, errno, "close sysctl");
143 
144         return val == '1';
145 }
146 
147 static void flowlabel_get(int fd, uint32_t label, uint8_t share, uint16_t flags)
148 {
149         struct in6_flowlabel_req req = {
150                 .flr_action = IPV6_FL_A_GET,
151                 .flr_label = htonl(label),
152                 .flr_flags = flags,
153                 .flr_share = share,
154         };
155 
156         /* do not pass IPV6_ADDR_ANY or IPV6_ADDR_MAPPED */
157         req.flr_dst.s6_addr[0] = 0xfd;
158         req.flr_dst.s6_addr[15] = 0x1;
159 
160         if (setsockopt(fd, SOL_IPV6, IPV6_FLOWLABEL_MGR, &req, sizeof(req)))
161                 error(1, errno, "setsockopt flowlabel get");
162 }
163 
164 static void parse_opts(int argc, char **argv)
165 {
166         int c;
167 
168         while ((c = getopt(argc, argv, "l:")) != -1) {
169                 switch (c) {
170                 case 'l':
171                         cfg_label = strtoul(optarg, NULL, 0);
172                         break;
173                 default:
174                         error(1, 0, "%s: parse error", argv[0]);
175                 }
176         }
177 }
178 
179 int main(int argc, char **argv)
180 {
181         struct sockaddr_in6 addr = {
182                 .sin6_family = AF_INET6,
183                 .sin6_port = htons(8000),
184                 .sin6_addr = IN6ADDR_LOOPBACK_INIT,
185         };
186         const int one = 1;
187         int fdt, fdr;
188 
189         parse_opts(argc, argv);
190 
191         fdt = socket(PF_INET6, SOCK_DGRAM, 0);
192         if (fdt == -1)
193                 error(1, errno, "socket t");
194 
195         fdr = socket(PF_INET6, SOCK_DGRAM, 0);
196         if (fdr == -1)
197                 error(1, errno, "socket r");
198 
199         if (connect(fdt, (void *)&addr, sizeof(addr)))
200                 error(1, errno, "connect");
201         if (bind(fdr, (void *)&addr, sizeof(addr)))
202                 error(1, errno, "bind");
203 
204         flowlabel_get(fdt, cfg_label, IPV6_FL_S_EXCL, IPV6_FL_F_CREATE);
205 
206         if (setsockopt(fdr, SOL_IPV6, IPV6_FLOWINFO, &one, sizeof(one)))
207                 error(1, errno, "setsockopt flowinfo");
208 
209         if (get_autoflowlabel_enabled()) {
210                 fprintf(stderr, "send no label: recv auto flowlabel\n");
211                 do_send(fdt, false, 0);
212                 do_recv(fdr, true, FLOWLABEL_WILDCARD);
213         } else {
214                 fprintf(stderr, "send no label: recv no label (auto off)\n");
215                 do_send(fdt, false, 0);
216                 do_recv(fdr, false, 0);
217         }
218 
219         fprintf(stderr, "send label\n");
220         do_send(fdt, true, cfg_label);
221         do_recv(fdr, true, cfg_label);
222 
223         if (close(fdr))
224                 error(1, errno, "close r");
225         if (close(fdt))
226                 error(1, errno, "close t");
227 
228         return 0;
229 }
230 

~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

kernel.org | git.kernel.org | LWN.net | Project Home | Wiki (Japanese) | Wiki (English) | SVN repository | Mail admin

Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.

osdn.jp