virtif_user.c revision 1.4 1 1.4 alnsn /* $NetBSD: virtif_user.c,v 1.4 2018/12/12 00:48:44 alnsn Exp $ */
2 1.1 pooka
3 1.1 pooka /*
4 1.1 pooka * Copyright (c) 2013 Antti Kantee. All Rights Reserved.
5 1.1 pooka *
6 1.1 pooka * Redistribution and use in source and binary forms, with or without
7 1.1 pooka * modification, are permitted provided that the following conditions
8 1.1 pooka * are met:
9 1.1 pooka * 1. Redistributions of source code must retain the above copyright
10 1.1 pooka * notice, this list of conditions and the following disclaimer.
11 1.1 pooka * 2. Redistributions in binary form must reproduce the above copyright
12 1.1 pooka * notice, this list of conditions and the following disclaimer in the
13 1.1 pooka * documentation and/or other materials provided with the distribution.
14 1.1 pooka *
15 1.1 pooka * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
16 1.1 pooka * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 1.1 pooka * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 1.1 pooka * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 1.1 pooka * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 1.1 pooka * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 1.1 pooka * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 1.1 pooka * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 1.1 pooka * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 1.1 pooka * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 1.1 pooka * SUCH DAMAGE.
26 1.1 pooka */
27 1.1 pooka
28 1.4 alnsn #include <sys/cdefs.h>
29 1.4 alnsn __KERNEL_RCSID(0, "$NetBSD: virtif_user.c,v 1.4 2018/12/12 00:48:44 alnsn Exp $");
30 1.4 alnsn
31 1.1 pooka #ifndef _KERNEL
32 1.1 pooka #include <sys/types.h>
33 1.1 pooka #include <sys/ioctl.h>
34 1.1 pooka #include <sys/uio.h>
35 1.1 pooka
36 1.1 pooka #include <assert.h>
37 1.1 pooka #include <errno.h>
38 1.1 pooka #include <fcntl.h>
39 1.1 pooka #include <inttypes.h>
40 1.1 pooka #include <poll.h>
41 1.1 pooka #include <pthread.h>
42 1.1 pooka #include <stdio.h>
43 1.1 pooka #include <stdlib.h>
44 1.1 pooka #include <string.h>
45 1.1 pooka #include <unistd.h>
46 1.1 pooka
47 1.1 pooka #ifdef __linux__
48 1.1 pooka #include <net/if.h>
49 1.1 pooka #include <linux/if_tun.h>
50 1.1 pooka #endif
51 1.1 pooka
52 1.1 pooka #include <rump/rumpuser_component.h>
53 1.1 pooka
54 1.1 pooka #include "if_virt.h"
55 1.1 pooka #include "virtif_user.h"
56 1.1 pooka
57 1.3 pooka #if VIFHYPER_REVISION != 20140313
58 1.1 pooka #error VIFHYPER_REVISION mismatch
59 1.1 pooka #endif
60 1.1 pooka
61 1.1 pooka struct virtif_user {
62 1.1 pooka struct virtif_sc *viu_virtifsc;
63 1.1 pooka int viu_devnum;
64 1.1 pooka
65 1.1 pooka int viu_fd;
66 1.1 pooka int viu_pipe[2];
67 1.1 pooka pthread_t viu_rcvthr;
68 1.1 pooka
69 1.1 pooka int viu_dying;
70 1.1 pooka
71 1.1 pooka char viu_rcvbuf[9018]; /* jumbo frame max len */
72 1.1 pooka };
73 1.1 pooka
74 1.1 pooka static int
75 1.1 pooka opentapdev(int devnum)
76 1.1 pooka {
77 1.1 pooka int fd = -1;
78 1.1 pooka
79 1.1 pooka #if defined(__NetBSD__) || defined(__DragonFly__)
80 1.1 pooka char tapdev[64];
81 1.1 pooka
82 1.1 pooka snprintf(tapdev, sizeof(tapdev), "/dev/tap%d", devnum);
83 1.1 pooka fd = open(tapdev, O_RDWR);
84 1.1 pooka if (fd == -1) {
85 1.1 pooka fprintf(stderr, "rumpcomp_virtif_create: can't open %s: "
86 1.1 pooka "%s\n", tapdev, strerror(errno));
87 1.1 pooka }
88 1.1 pooka
89 1.1 pooka #elif defined(__linux__)
90 1.1 pooka struct ifreq ifr;
91 1.1 pooka char devname[16];
92 1.1 pooka
93 1.1 pooka fd = open("/dev/net/tun", O_RDWR);
94 1.1 pooka if (fd == -1) {
95 1.1 pooka fprintf(stderr, "rumpcomp_virtif_create: can't open %s: "
96 1.1 pooka "%s\n", "/dev/net/tun", strerror(errno));
97 1.1 pooka return -1;
98 1.1 pooka }
99 1.1 pooka
100 1.1 pooka snprintf(devname, sizeof(devname), "tun%d", devnum);
101 1.1 pooka memset(&ifr, 0, sizeof(ifr));
102 1.1 pooka ifr.ifr_flags = IFF_TAP | IFF_NO_PI;
103 1.1 pooka strncpy(ifr.ifr_name, devname, sizeof(ifr.ifr_name)-1);
104 1.1 pooka
105 1.1 pooka if (ioctl(fd, TUNSETIFF, &ifr) == -1) {
106 1.1 pooka close(fd);
107 1.1 pooka fd = -1;
108 1.1 pooka }
109 1.1 pooka
110 1.1 pooka #else
111 1.1 pooka fprintf(stderr, "virtif not supported on this platform\n");
112 1.1 pooka #endif
113 1.1 pooka
114 1.1 pooka return fd;
115 1.1 pooka }
116 1.1 pooka
117 1.1 pooka static void
118 1.1 pooka closetapdev(struct virtif_user *viu)
119 1.1 pooka {
120 1.1 pooka
121 1.1 pooka close(viu->viu_fd);
122 1.1 pooka }
123 1.1 pooka
124 1.1 pooka static void *
125 1.1 pooka rcvthread(void *aaargh)
126 1.1 pooka {
127 1.1 pooka struct virtif_user *viu = aaargh;
128 1.1 pooka struct pollfd pfd[2];
129 1.1 pooka struct iovec iov;
130 1.1 pooka ssize_t nn = 0;
131 1.1 pooka int prv;
132 1.1 pooka
133 1.1 pooka rumpuser_component_kthread();
134 1.1 pooka
135 1.1 pooka pfd[0].fd = viu->viu_fd;
136 1.1 pooka pfd[0].events = POLLIN;
137 1.1 pooka pfd[1].fd = viu->viu_pipe[0];
138 1.1 pooka pfd[1].events = POLLIN;
139 1.1 pooka
140 1.1 pooka while (!viu->viu_dying) {
141 1.1 pooka prv = poll(pfd, 2, -1);
142 1.1 pooka if (prv == 0)
143 1.1 pooka continue;
144 1.1 pooka if (prv == -1) {
145 1.1 pooka /* XXX */
146 1.1 pooka fprintf(stderr, "virt%d: poll error: %d\n",
147 1.1 pooka viu->viu_devnum, errno);
148 1.1 pooka sleep(1);
149 1.1 pooka continue;
150 1.1 pooka }
151 1.1 pooka if (pfd[1].revents & POLLIN)
152 1.1 pooka continue;
153 1.1 pooka
154 1.1 pooka nn = read(viu->viu_fd,
155 1.1 pooka viu->viu_rcvbuf, sizeof(viu->viu_rcvbuf));
156 1.1 pooka if (nn == -1 && errno == EAGAIN)
157 1.1 pooka continue;
158 1.1 pooka
159 1.1 pooka if (nn < 1) {
160 1.1 pooka /* XXX */
161 1.1 pooka fprintf(stderr, "virt%d: receive failed\n",
162 1.1 pooka viu->viu_devnum);
163 1.1 pooka sleep(1);
164 1.1 pooka continue;
165 1.1 pooka }
166 1.1 pooka iov.iov_base = viu->viu_rcvbuf;
167 1.1 pooka iov.iov_len = nn;
168 1.1 pooka
169 1.1 pooka rumpuser_component_schedule(NULL);
170 1.1 pooka VIF_DELIVERPKT(viu->viu_virtifsc, &iov, 1);
171 1.1 pooka rumpuser_component_unschedule();
172 1.1 pooka }
173 1.1 pooka
174 1.1 pooka assert(viu->viu_dying);
175 1.1 pooka
176 1.1 pooka rumpuser_component_kthread_release();
177 1.1 pooka return NULL;
178 1.1 pooka }
179 1.1 pooka
180 1.1 pooka int
181 1.1 pooka VIFHYPER_CREATE(const char *devstr, struct virtif_sc *vif_sc, uint8_t *enaddr,
182 1.1 pooka struct virtif_user **viup)
183 1.1 pooka {
184 1.1 pooka struct virtif_user *viu = NULL;
185 1.1 pooka void *cookie;
186 1.1 pooka int devnum;
187 1.1 pooka int rv;
188 1.1 pooka
189 1.1 pooka cookie = rumpuser_component_unschedule();
190 1.1 pooka
191 1.1 pooka /*
192 1.1 pooka * Since this interface doesn't do LINKSTR, we know devstr to be
193 1.1 pooka * well-formatted.
194 1.1 pooka */
195 1.1 pooka devnum = atoi(devstr);
196 1.1 pooka
197 1.1 pooka viu = calloc(1, sizeof(*viu));
198 1.1 pooka if (viu == NULL) {
199 1.1 pooka rv = errno;
200 1.1 pooka goto oerr1;
201 1.1 pooka }
202 1.1 pooka viu->viu_virtifsc = vif_sc;
203 1.1 pooka
204 1.1 pooka viu->viu_fd = opentapdev(devnum);
205 1.1 pooka if (viu->viu_fd == -1) {
206 1.1 pooka rv = errno;
207 1.1 pooka goto oerr2;
208 1.1 pooka }
209 1.1 pooka viu->viu_devnum = devnum;
210 1.1 pooka
211 1.1 pooka if (pipe(viu->viu_pipe) == -1) {
212 1.1 pooka rv = errno;
213 1.1 pooka goto oerr3;
214 1.1 pooka }
215 1.1 pooka
216 1.1 pooka if ((rv = pthread_create(&viu->viu_rcvthr, NULL, rcvthread, viu)) != 0)
217 1.1 pooka goto oerr4;
218 1.1 pooka
219 1.1 pooka rumpuser_component_schedule(cookie);
220 1.1 pooka *viup = viu;
221 1.1 pooka return 0;
222 1.1 pooka
223 1.1 pooka oerr4:
224 1.1 pooka close(viu->viu_pipe[0]);
225 1.1 pooka close(viu->viu_pipe[1]);
226 1.1 pooka oerr3:
227 1.1 pooka closetapdev(viu);
228 1.1 pooka oerr2:
229 1.1 pooka free(viu);
230 1.1 pooka oerr1:
231 1.1 pooka rumpuser_component_schedule(cookie);
232 1.1 pooka return rumpuser_component_errtrans(rv);
233 1.1 pooka }
234 1.1 pooka
235 1.1 pooka void
236 1.1 pooka VIFHYPER_SEND(struct virtif_user *viu,
237 1.1 pooka struct iovec *iov, size_t iovlen)
238 1.1 pooka {
239 1.1 pooka void *cookie = rumpuser_component_unschedule();
240 1.1 pooka ssize_t idontcare __attribute__((__unused__));
241 1.1 pooka
242 1.1 pooka /*
243 1.1 pooka * no need to check for return value; packets may be dropped
244 1.1 pooka *
245 1.1 pooka * ... sorry, I spoke too soon. We need to check it because
246 1.1 pooka * apparently gcc reinvented const poisoning and it's very
247 1.1 pooka * hard to say "thanks, I know I'm not using the result,
248 1.1 pooka * but please STFU and let's get on with something useful".
249 1.1 pooka * So let's trick gcc into letting us share the compiler
250 1.1 pooka * experience.
251 1.1 pooka */
252 1.1 pooka idontcare = writev(viu->viu_fd, iov, iovlen);
253 1.1 pooka
254 1.1 pooka rumpuser_component_schedule(cookie);
255 1.1 pooka }
256 1.1 pooka
257 1.2 pooka int
258 1.1 pooka VIFHYPER_DYING(struct virtif_user *viu)
259 1.1 pooka {
260 1.1 pooka void *cookie = rumpuser_component_unschedule();
261 1.1 pooka
262 1.1 pooka viu->viu_dying = 1;
263 1.1 pooka if (write(viu->viu_pipe[1],
264 1.1 pooka &viu->viu_dying, sizeof(viu->viu_dying)) == -1) {
265 1.2 pooka /*
266 1.2 pooka * this is here mostly to avoid a compiler warning
267 1.2 pooka * about ignoring the return value of write()
268 1.2 pooka */
269 1.1 pooka fprintf(stderr, "%s: failed to signal thread\n",
270 1.1 pooka VIF_STRING(VIFHYPER_DYING));
271 1.1 pooka }
272 1.1 pooka
273 1.1 pooka rumpuser_component_schedule(cookie);
274 1.2 pooka
275 1.2 pooka return 0;
276 1.1 pooka }
277 1.1 pooka
278 1.1 pooka void
279 1.1 pooka VIFHYPER_DESTROY(struct virtif_user *viu)
280 1.1 pooka {
281 1.1 pooka void *cookie = rumpuser_component_unschedule();
282 1.1 pooka
283 1.1 pooka pthread_join(viu->viu_rcvthr, NULL);
284 1.1 pooka closetapdev(viu);
285 1.1 pooka close(viu->viu_pipe[0]);
286 1.1 pooka close(viu->viu_pipe[1]);
287 1.1 pooka free(viu);
288 1.1 pooka
289 1.1 pooka rumpuser_component_schedule(cookie);
290 1.1 pooka }
291 1.1 pooka #endif
292