xref: /openbmc/qemu/tests/vhost-user-bridge.c (revision 37b0b24e)
1 /*
2  * Vhost User Bridge
3  *
4  * Copyright (c) 2015 Red Hat, Inc.
5  *
6  * Authors:
7  *  Victor Kaplansky <victork@redhat.com>
8  *
9  * This work is licensed under the terms of the GNU GPL, version 2 or
10  * later.  See the COPYING file in the top-level directory.
11  */
12 
13 /*
14  * TODO:
15  *     - main should get parameters from the command line.
16  *     - implement all request handlers. Still not implemented:
17  *          vubr_get_queue_num_exec()
18  *          vubr_send_rarp_exec()
19  *     - test for broken requests and virtqueue.
20  *     - implement features defined by Virtio 1.0 spec.
21  *     - support mergeable buffers and indirect descriptors.
22  *     - implement clean shutdown.
23  *     - implement non-blocking writes to UDP backend.
24  *     - implement polling strategy.
25  *     - implement clean starting/stopping of vq processing
26  *     - implement clean starting/stopping of used and buffers
27  *       dirty page logging.
28  */
29 
30 #define _FILE_OFFSET_BITS 64
31 
32 #include "qemu/osdep.h"
33 #include "qemu/atomic.h"
34 #include "qemu/ctype.h"
35 #include "qemu/iov.h"
36 #include "standard-headers/linux/virtio_net.h"
37 #include "libvhost-user.h"
38 
39 #define VHOST_USER_BRIDGE_DEBUG 1
40 
41 #define DPRINT(...) \
42     do { \
43         if (VHOST_USER_BRIDGE_DEBUG) { \
44             printf(__VA_ARGS__); \
45         } \
46     } while (0)
47 
48 enum {
49     VHOST_USER_BRIDGE_MAX_QUEUES = 8,
50 };
51 
52 typedef void (*CallbackFunc)(int sock, void *ctx);
53 
54 typedef struct Event {
55     void *ctx;
56     CallbackFunc callback;
57 } Event;
58 
59 typedef struct Dispatcher {
60     int max_sock;
61     fd_set fdset;
62     Event events[FD_SETSIZE];
63 } Dispatcher;
64 
65 typedef struct VubrDev {
66     VuDev vudev;
67     Dispatcher dispatcher;
68     int backend_udp_sock;
69     struct sockaddr_in backend_udp_dest;
70     int hdrlen;
71     int sock;
72     int ready;
73     int quit;
74     struct {
75         int fd;
76         void *addr;
77         pthread_t thread;
78     } notifier;
79 } VubrDev;
80 
81 static void
vubr_die(const char * s)82 vubr_die(const char *s)
83 {
84     perror(s);
85     exit(1);
86 }
87 
88 static int
dispatcher_init(Dispatcher * dispr)89 dispatcher_init(Dispatcher *dispr)
90 {
91     FD_ZERO(&dispr->fdset);
92     dispr->max_sock = -1;
93     return 0;
94 }
95 
96 static int
dispatcher_add(Dispatcher * dispr,int sock,void * ctx,CallbackFunc cb)97 dispatcher_add(Dispatcher *dispr, int sock, void *ctx, CallbackFunc cb)
98 {
99     if (sock >= FD_SETSIZE) {
100         fprintf(stderr,
101                 "Error: Failed to add new event. sock %d should be less than %d\n",
102                 sock, FD_SETSIZE);
103         return -1;
104     }
105 
106     dispr->events[sock].ctx = ctx;
107     dispr->events[sock].callback = cb;
108 
109     FD_SET(sock, &dispr->fdset);
110     if (sock > dispr->max_sock) {
111         dispr->max_sock = sock;
112     }
113     DPRINT("Added sock %d for watching. max_sock: %d\n",
114            sock, dispr->max_sock);
115     return 0;
116 }
117 
118 static int
dispatcher_remove(Dispatcher * dispr,int sock)119 dispatcher_remove(Dispatcher *dispr, int sock)
120 {
121     if (sock >= FD_SETSIZE) {
122         fprintf(stderr,
123                 "Error: Failed to remove event. sock %d should be less than %d\n",
124                 sock, FD_SETSIZE);
125         return -1;
126     }
127 
128     FD_CLR(sock, &dispr->fdset);
129     DPRINT("Sock %d removed from dispatcher watch.\n", sock);
130     return 0;
131 }
132 
133 /* timeout in us */
134 static int
dispatcher_wait(Dispatcher * dispr,uint32_t timeout)135 dispatcher_wait(Dispatcher *dispr, uint32_t timeout)
136 {
137     struct timeval tv;
138     tv.tv_sec = timeout / 1000000;
139     tv.tv_usec = timeout % 1000000;
140 
141     fd_set fdset = dispr->fdset;
142 
143     /* wait until some of sockets become readable. */
144     int rc = select(dispr->max_sock + 1, &fdset, 0, 0, &tv);
145 
146     if (rc == -1) {
147         vubr_die("select");
148     }
149 
150     /* Timeout */
151     if (rc == 0) {
152         return 0;
153     }
154 
155     /* Now call callback for every ready socket. */
156 
157     int sock;
158     for (sock = 0; sock < dispr->max_sock + 1; sock++) {
159         /* The callback on a socket can remove other sockets from the
160          * dispatcher, thus we have to check that the socket is
161          * still not removed from dispatcher's list
162          */
163         if (FD_ISSET(sock, &fdset) && FD_ISSET(sock, &dispr->fdset)) {
164             Event *e = &dispr->events[sock];
165             e->callback(sock, e->ctx);
166         }
167     }
168 
169     return 0;
170 }
171 
172 static void
vubr_handle_tx(VuDev * dev,int qidx)173 vubr_handle_tx(VuDev *dev, int qidx)
174 {
175     VuVirtq *vq = vu_get_queue(dev, qidx);
176     VubrDev *vubr = container_of(dev, VubrDev, vudev);
177     int hdrlen = vubr->hdrlen;
178     VuVirtqElement *elem = NULL;
179 
180     assert(qidx % 2);
181 
182     for (;;) {
183         ssize_t ret;
184         unsigned int out_num;
185         struct iovec sg[VIRTQUEUE_MAX_SIZE], *out_sg;
186 
187         elem = vu_queue_pop(dev, vq, sizeof(VuVirtqElement));
188         if (!elem) {
189             break;
190         }
191 
192         out_num = elem->out_num;
193         out_sg = elem->out_sg;
194         if (out_num < 1) {
195             fprintf(stderr, "virtio-net header not in first element\n");
196             break;
197         }
198         if (VHOST_USER_BRIDGE_DEBUG) {
199             iov_hexdump(out_sg, out_num, stderr, "TX:", 1024);
200         }
201 
202         if (hdrlen) {
203             unsigned sg_num = iov_copy(sg, ARRAY_SIZE(sg),
204                                        out_sg, out_num,
205                                        hdrlen, -1);
206             out_num = sg_num;
207             out_sg = sg;
208         }
209 
210         struct msghdr msg = {
211             .msg_name = (struct sockaddr *) &vubr->backend_udp_dest,
212             .msg_namelen = sizeof(struct sockaddr_in),
213             .msg_iov = out_sg,
214             .msg_iovlen = out_num,
215         };
216         do {
217             ret = sendmsg(vubr->backend_udp_sock, &msg, 0);
218         } while (ret == -1 && (errno == EAGAIN || errno == EINTR));
219 
220         if (ret == -1) {
221             vubr_die("sendmsg()");
222         }
223 
224         vu_queue_push(dev, vq, elem, 0);
225         vu_queue_notify(dev, vq);
226 
227         free(elem);
228         elem = NULL;
229     }
230 
231     free(elem);
232 }
233 
234 
235 /* this function reverse the effect of iov_discard_front() it must be
236  * called with 'front' being the original struct iovec and 'bytes'
237  * being the number of bytes you shaved off
238  */
239 static void
iov_restore_front(struct iovec * front,struct iovec * iov,size_t bytes)240 iov_restore_front(struct iovec *front, struct iovec *iov, size_t bytes)
241 {
242     struct iovec *cur;
243 
244     for (cur = front; cur != iov; cur++) {
245         assert(bytes >= cur->iov_len);
246         bytes -= cur->iov_len;
247     }
248 
249     cur->iov_base -= bytes;
250     cur->iov_len += bytes;
251 }
252 
253 static void
iov_truncate(struct iovec * iov,unsigned iovc,size_t bytes)254 iov_truncate(struct iovec *iov, unsigned iovc, size_t bytes)
255 {
256     unsigned i;
257 
258     for (i = 0; i < iovc; i++, iov++) {
259         if (bytes < iov->iov_len) {
260             iov->iov_len = bytes;
261             return;
262         }
263 
264         bytes -= iov->iov_len;
265     }
266 
267     assert(!"couldn't truncate iov");
268 }
269 
270 static void
vubr_backend_recv_cb(int sock,void * ctx)271 vubr_backend_recv_cb(int sock, void *ctx)
272 {
273     VubrDev *vubr = (VubrDev *) ctx;
274     VuDev *dev = &vubr->vudev;
275     VuVirtq *vq = vu_get_queue(dev, 0);
276     VuVirtqElement *elem = NULL;
277     struct iovec mhdr_sg[VIRTQUEUE_MAX_SIZE];
278     struct virtio_net_hdr_mrg_rxbuf mhdr;
279     unsigned mhdr_cnt = 0;
280     int hdrlen = vubr->hdrlen;
281     int i = 0;
282     struct virtio_net_hdr hdr = {
283         .flags = 0,
284         .gso_type = VIRTIO_NET_HDR_GSO_NONE
285     };
286 
287     DPRINT("\n\n   ***   IN UDP RECEIVE CALLBACK    ***\n\n");
288     DPRINT("    hdrlen = %d\n", hdrlen);
289 
290     if (!vu_queue_enabled(dev, vq) ||
291         !vu_queue_started(dev, vq) ||
292         !vu_queue_avail_bytes(dev, vq, hdrlen, 0)) {
293         DPRINT("Got UDP packet, but no available descriptors on RX virtq.\n");
294         return;
295     }
296 
297     while (1) {
298         struct iovec *sg;
299         ssize_t ret, total = 0;
300         unsigned int num;
301 
302         elem = vu_queue_pop(dev, vq, sizeof(VuVirtqElement));
303         if (!elem) {
304             break;
305         }
306 
307         if (elem->in_num < 1) {
308             fprintf(stderr, "virtio-net contains no in buffers\n");
309             break;
310         }
311 
312         sg = elem->in_sg;
313         num = elem->in_num;
314         if (i == 0) {
315             if (hdrlen == 12) {
316                 mhdr_cnt = iov_copy(mhdr_sg, ARRAY_SIZE(mhdr_sg),
317                                     sg, elem->in_num,
318                                     offsetof(typeof(mhdr), num_buffers),
319                                     sizeof(mhdr.num_buffers));
320             }
321             iov_from_buf(sg, elem->in_num, 0, &hdr, sizeof hdr);
322             total += hdrlen;
323             ret = iov_discard_front(&sg, &num, hdrlen);
324             assert(ret == hdrlen);
325         }
326 
327         struct msghdr msg = {
328             .msg_name = (struct sockaddr *) &vubr->backend_udp_dest,
329             .msg_namelen = sizeof(struct sockaddr_in),
330             .msg_iov = sg,
331             .msg_iovlen = num,
332             .msg_flags = MSG_DONTWAIT,
333         };
334         ret = RETRY_ON_EINTR(recvmsg(vubr->backend_udp_sock, &msg, 0));
335 
336         if (i == 0) {
337             iov_restore_front(elem->in_sg, sg, hdrlen);
338         }
339 
340         if (ret == -1) {
341             if (errno == EWOULDBLOCK) {
342                 vu_queue_rewind(dev, vq, 1);
343                 break;
344             }
345 
346             vubr_die("recvmsg()");
347         }
348 
349         total += ret;
350         iov_truncate(elem->in_sg, elem->in_num, total);
351         vu_queue_fill(dev, vq, elem, total, i++);
352 
353         free(elem);
354         elem = NULL;
355 
356         break;        /* could loop if DONTWAIT worked? */
357     }
358 
359     if (mhdr_cnt) {
360         mhdr.num_buffers = i;
361         iov_from_buf(mhdr_sg, mhdr_cnt,
362                      0,
363                      &mhdr.num_buffers, sizeof mhdr.num_buffers);
364     }
365 
366     vu_queue_flush(dev, vq, i);
367     vu_queue_notify(dev, vq);
368 
369     free(elem);
370 }
371 
372 static void
vubr_receive_cb(int sock,void * ctx)373 vubr_receive_cb(int sock, void *ctx)
374 {
375     VubrDev *vubr = (VubrDev *)ctx;
376 
377     if (!vu_dispatch(&vubr->vudev)) {
378         fprintf(stderr, "Error while dispatching\n");
379     }
380 }
381 
382 typedef struct WatchData {
383     VuDev *dev;
384     vu_watch_cb cb;
385     void *data;
386 } WatchData;
387 
388 static void
watch_cb(int sock,void * ctx)389 watch_cb(int sock, void *ctx)
390 {
391     struct WatchData *wd = ctx;
392 
393     wd->cb(wd->dev, VU_WATCH_IN, wd->data);
394 }
395 
396 static void
vubr_set_watch(VuDev * dev,int fd,int condition,vu_watch_cb cb,void * data)397 vubr_set_watch(VuDev *dev, int fd, int condition,
398                vu_watch_cb cb, void *data)
399 {
400     VubrDev *vubr = container_of(dev, VubrDev, vudev);
401     static WatchData watches[FD_SETSIZE];
402     struct WatchData *wd = &watches[fd];
403 
404     wd->cb = cb;
405     wd->data = data;
406     wd->dev = dev;
407     dispatcher_add(&vubr->dispatcher, fd, wd, watch_cb);
408 }
409 
410 static void
vubr_remove_watch(VuDev * dev,int fd)411 vubr_remove_watch(VuDev *dev, int fd)
412 {
413     VubrDev *vubr = container_of(dev, VubrDev, vudev);
414 
415     dispatcher_remove(&vubr->dispatcher, fd);
416 }
417 
418 static int
vubr_send_rarp_exec(VuDev * dev,VhostUserMsg * vmsg)419 vubr_send_rarp_exec(VuDev *dev, VhostUserMsg *vmsg)
420 {
421     DPRINT("Function %s() not implemented yet.\n", __func__);
422     return 0;
423 }
424 
425 static int
vubr_process_msg(VuDev * dev,VhostUserMsg * vmsg,int * do_reply)426 vubr_process_msg(VuDev *dev, VhostUserMsg *vmsg, int *do_reply)
427 {
428     switch (vmsg->request) {
429     case VHOST_USER_SEND_RARP:
430         *do_reply = vubr_send_rarp_exec(dev, vmsg);
431         return 1;
432     default:
433         /* let the library handle the rest */
434         return 0;
435     }
436 
437     return 0;
438 }
439 
440 static void
vubr_set_features(VuDev * dev,uint64_t features)441 vubr_set_features(VuDev *dev, uint64_t features)
442 {
443     VubrDev *vubr = container_of(dev, VubrDev, vudev);
444 
445     if ((features & (1ULL << VIRTIO_F_VERSION_1)) ||
446         (features & (1ULL << VIRTIO_NET_F_MRG_RXBUF))) {
447         vubr->hdrlen = 12;
448     } else {
449         vubr->hdrlen = 10;
450     }
451 }
452 
453 static uint64_t
vubr_get_features(VuDev * dev)454 vubr_get_features(VuDev *dev)
455 {
456     return 1ULL << VIRTIO_NET_F_GUEST_ANNOUNCE |
457         1ULL << VIRTIO_NET_F_MRG_RXBUF |
458         1ULL << VIRTIO_F_VERSION_1;
459 }
460 
461 static void
vubr_queue_set_started(VuDev * dev,int qidx,bool started)462 vubr_queue_set_started(VuDev *dev, int qidx, bool started)
463 {
464     VubrDev *vubr = container_of(dev, VubrDev, vudev);
465     VuVirtq *vq = vu_get_queue(dev, qidx);
466 
467     if (started && vubr->notifier.fd >= 0) {
468         vu_set_queue_host_notifier(dev, vq, vubr->notifier.fd,
469                                    qemu_real_host_page_size(),
470                                    qidx * qemu_real_host_page_size());
471     }
472 
473     if (qidx % 2 == 1) {
474         vu_set_queue_handler(dev, vq, started ? vubr_handle_tx : NULL);
475     }
476 }
477 
478 static void
vubr_panic(VuDev * dev,const char * msg)479 vubr_panic(VuDev *dev, const char *msg)
480 {
481     VubrDev *vubr = container_of(dev, VubrDev, vudev);
482 
483     fprintf(stderr, "PANIC: %s\n", msg);
484 
485     dispatcher_remove(&vubr->dispatcher, dev->sock);
486     vubr->quit = 1;
487 }
488 
489 static bool
vubr_queue_is_processed_in_order(VuDev * dev,int qidx)490 vubr_queue_is_processed_in_order(VuDev *dev, int qidx)
491 {
492     return true;
493 }
494 
495 static const VuDevIface vuiface = {
496     .get_features = vubr_get_features,
497     .set_features = vubr_set_features,
498     .process_msg = vubr_process_msg,
499     .queue_set_started = vubr_queue_set_started,
500     .queue_is_processed_in_order = vubr_queue_is_processed_in_order,
501 };
502 
503 static void
vubr_accept_cb(int sock,void * ctx)504 vubr_accept_cb(int sock, void *ctx)
505 {
506     VubrDev *dev = (VubrDev *)ctx;
507     int conn_fd;
508     struct sockaddr_un un;
509     socklen_t len = sizeof(un);
510 
511     conn_fd = accept(sock, (struct sockaddr *) &un, &len);
512     if (conn_fd == -1) {
513         vubr_die("accept()");
514     }
515     DPRINT("Got connection from remote peer on sock %d\n", conn_fd);
516 
517     if (!vu_init(&dev->vudev,
518                  VHOST_USER_BRIDGE_MAX_QUEUES,
519                  conn_fd,
520                  vubr_panic,
521                  NULL,
522                  vubr_set_watch,
523                  vubr_remove_watch,
524                  &vuiface)) {
525         fprintf(stderr, "Failed to initialize libvhost-user\n");
526         exit(1);
527     }
528 
529     dispatcher_add(&dev->dispatcher, conn_fd, ctx, vubr_receive_cb);
530     dispatcher_remove(&dev->dispatcher, sock);
531 }
532 
533 static VubrDev *
vubr_new(const char * path,bool client)534 vubr_new(const char *path, bool client)
535 {
536     VubrDev *dev = (VubrDev *) calloc(1, sizeof(VubrDev));
537     struct sockaddr_un un;
538     CallbackFunc cb;
539     size_t len;
540 
541     if (strlen(path) >= sizeof(un.sun_path)) {
542         fprintf(stderr, "unix domain socket path '%s' is too long\n", path);
543         exit(1);
544     }
545 
546     /* Get a UNIX socket. */
547     dev->sock = socket(AF_UNIX, SOCK_STREAM, 0);
548     if (dev->sock == -1) {
549         vubr_die("socket");
550     }
551 
552     dev->notifier.fd = -1;
553 
554     un.sun_family = AF_UNIX;
555     strcpy(un.sun_path, path);
556     len = sizeof(un.sun_family) + strlen(path);
557 
558     if (!client) {
559         unlink(path);
560 
561         if (bind(dev->sock, (struct sockaddr *) &un, len) == -1) {
562             vubr_die("bind");
563         }
564 
565         if (listen(dev->sock, 1) == -1) {
566             vubr_die("listen");
567         }
568         cb = vubr_accept_cb;
569 
570         DPRINT("Waiting for connections on UNIX socket %s ...\n", path);
571     } else {
572         if (connect(dev->sock, (struct sockaddr *)&un, len) == -1) {
573             vubr_die("connect");
574         }
575 
576         if (!vu_init(&dev->vudev,
577                      VHOST_USER_BRIDGE_MAX_QUEUES,
578                      dev->sock,
579                      vubr_panic,
580                      NULL,
581                      vubr_set_watch,
582                      vubr_remove_watch,
583                      &vuiface)) {
584             fprintf(stderr, "Failed to initialize libvhost-user\n");
585             exit(1);
586         }
587 
588         cb = vubr_receive_cb;
589     }
590 
591     dispatcher_init(&dev->dispatcher);
592 
593     dispatcher_add(&dev->dispatcher, dev->sock, (void *)dev, cb);
594 
595     return dev;
596 }
597 
notifier_thread(void * arg)598 static void *notifier_thread(void *arg)
599 {
600     VuDev *dev = (VuDev *)arg;
601     VubrDev *vubr = container_of(dev, VubrDev, vudev);
602     int pagesize = qemu_real_host_page_size();
603     int qidx;
604 
605     while (true) {
606         for (qidx = 0; qidx < VHOST_USER_BRIDGE_MAX_QUEUES; qidx++) {
607             uint16_t *n = vubr->notifier.addr + pagesize * qidx;
608 
609             if (*n == qidx) {
610                 *n = 0xffff;
611                 /* We won't miss notifications if we reset
612                  * the memory first. */
613                 smp_mb();
614 
615                 DPRINT("Got a notification for queue%d via host notifier.\n",
616                        qidx);
617 
618                 if (qidx % 2 == 1) {
619                     vubr_handle_tx(dev, qidx);
620                 }
621             }
622             usleep(1000);
623         }
624     }
625 
626     return NULL;
627 }
628 
629 static void
vubr_host_notifier_setup(VubrDev * dev)630 vubr_host_notifier_setup(VubrDev *dev)
631 {
632     pthread_t thread;
633     size_t length;
634     void *addr;
635     int fd;
636 
637     length = qemu_real_host_page_size() * VHOST_USER_BRIDGE_MAX_QUEUES;
638 
639     fd = g_file_open_tmp("vubr-XXXXXX", NULL, NULL);
640     if (fd < 0) {
641         vubr_die("mkstemp()");
642     }
643 
644     if (posix_fallocate(fd, 0, length) != 0) {
645         vubr_die("posix_fallocate()");
646     }
647 
648     addr = mmap(NULL, length, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
649     if (addr == MAP_FAILED) {
650         vubr_die("mmap()");
651     }
652 
653     memset(addr, 0xff, length);
654 
655     if (pthread_create(&thread, NULL, notifier_thread, &dev->vudev) != 0) {
656         vubr_die("pthread_create()");
657     }
658 
659     dev->notifier.fd = fd;
660     dev->notifier.addr = addr;
661     dev->notifier.thread = thread;
662 }
663 
664 static void
vubr_set_host(struct sockaddr_in * saddr,const char * host)665 vubr_set_host(struct sockaddr_in *saddr, const char *host)
666 {
667     if (qemu_isdigit(host[0])) {
668         if (!inet_aton(host, &saddr->sin_addr)) {
669             fprintf(stderr, "inet_aton() failed.\n");
670             exit(1);
671         }
672     } else {
673         struct hostent *he = gethostbyname(host);
674 
675         if (!he) {
676             fprintf(stderr, "gethostbyname() failed.\n");
677             exit(1);
678         }
679         saddr->sin_addr = *(struct in_addr *)he->h_addr;
680     }
681 }
682 
683 static void
vubr_backend_udp_setup(VubrDev * dev,const char * local_host,const char * local_port,const char * remote_host,const char * remote_port)684 vubr_backend_udp_setup(VubrDev *dev,
685                        const char *local_host,
686                        const char *local_port,
687                        const char *remote_host,
688                        const char *remote_port)
689 {
690     int sock;
691     const char *r;
692 
693     int lport, rport;
694 
695     lport = strtol(local_port, (char **)&r, 0);
696     if (r == local_port) {
697         fprintf(stderr, "lport parsing failed.\n");
698         exit(1);
699     }
700 
701     rport = strtol(remote_port, (char **)&r, 0);
702     if (r == remote_port) {
703         fprintf(stderr, "rport parsing failed.\n");
704         exit(1);
705     }
706 
707     struct sockaddr_in si_local = {
708         .sin_family = AF_INET,
709         .sin_port = htons(lport),
710     };
711 
712     vubr_set_host(&si_local, local_host);
713 
714     /* setup destination for sends */
715     dev->backend_udp_dest = (struct sockaddr_in) {
716         .sin_family = AF_INET,
717         .sin_port = htons(rport),
718     };
719     vubr_set_host(&dev->backend_udp_dest, remote_host);
720 
721     sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
722     if (sock == -1) {
723         vubr_die("socket");
724     }
725 
726     if (bind(sock, (struct sockaddr *)&si_local, sizeof(si_local)) == -1) {
727         vubr_die("bind");
728     }
729 
730     dev->backend_udp_sock = sock;
731     dispatcher_add(&dev->dispatcher, sock, dev, vubr_backend_recv_cb);
732     DPRINT("Waiting for data from udp backend on %s:%d...\n",
733            local_host, lport);
734 }
735 
736 static void
vubr_run(VubrDev * dev)737 vubr_run(VubrDev *dev)
738 {
739     while (!dev->quit) {
740         /* timeout 200ms */
741         dispatcher_wait(&dev->dispatcher, 200000);
742         /* Here one can try polling strategy. */
743     }
744 }
745 
746 static int
vubr_parse_host_port(const char ** host,const char ** port,const char * buf)747 vubr_parse_host_port(const char **host, const char **port, const char *buf)
748 {
749     char *p = strchr(buf, ':');
750 
751     if (!p) {
752         return -1;
753     }
754     *p = '\0';
755     *host = strdup(buf);
756     *port = strdup(p + 1);
757     return 0;
758 }
759 
760 #define DEFAULT_UD_SOCKET "/tmp/vubr.sock"
761 #define DEFAULT_LHOST "127.0.0.1"
762 #define DEFAULT_LPORT "4444"
763 #define DEFAULT_RHOST "127.0.0.1"
764 #define DEFAULT_RPORT "5555"
765 
766 static const char *ud_socket_path = DEFAULT_UD_SOCKET;
767 static const char *lhost = DEFAULT_LHOST;
768 static const char *lport = DEFAULT_LPORT;
769 static const char *rhost = DEFAULT_RHOST;
770 static const char *rport = DEFAULT_RPORT;
771 
772 int
main(int argc,char * argv[])773 main(int argc, char *argv[])
774 {
775     VubrDev *dev;
776     int opt;
777     bool client = false;
778     bool host_notifier = false;
779 
780     while ((opt = getopt(argc, argv, "l:r:u:cH")) != -1) {
781 
782         switch (opt) {
783         case 'l':
784             if (vubr_parse_host_port(&lhost, &lport, optarg) < 0) {
785                 goto out;
786             }
787             break;
788         case 'r':
789             if (vubr_parse_host_port(&rhost, &rport, optarg) < 0) {
790                 goto out;
791             }
792             break;
793         case 'u':
794             ud_socket_path = strdup(optarg);
795             break;
796         case 'c':
797             client = true;
798             break;
799         case 'H':
800             host_notifier = true;
801             break;
802         default:
803             goto out;
804         }
805     }
806 
807     DPRINT("ud socket: %s (%s)\n", ud_socket_path,
808            client ? "client" : "server");
809     DPRINT("local:     %s:%s\n", lhost, lport);
810     DPRINT("remote:    %s:%s\n", rhost, rport);
811 
812     dev = vubr_new(ud_socket_path, client);
813     if (!dev) {
814         return 1;
815     }
816 
817     if (host_notifier) {
818         vubr_host_notifier_setup(dev);
819     }
820 
821     vubr_backend_udp_setup(dev, lhost, lport, rhost, rport);
822     vubr_run(dev);
823 
824     vu_deinit(&dev->vudev);
825 
826     return 0;
827 
828 out:
829     fprintf(stderr, "Usage: %s ", argv[0]);
830     fprintf(stderr, "[-c] [-H] [-u ud_socket_path] [-l lhost:lport] [-r rhost:rport]\n");
831     fprintf(stderr, "\t-u path to unix domain socket. default: %s\n",
832             DEFAULT_UD_SOCKET);
833     fprintf(stderr, "\t-l local host and port. default: %s:%s\n",
834             DEFAULT_LHOST, DEFAULT_LPORT);
835     fprintf(stderr, "\t-r remote host and port. default: %s:%s\n",
836             DEFAULT_RHOST, DEFAULT_RPORT);
837     fprintf(stderr, "\t-c client mode\n");
838     fprintf(stderr, "\t-H use host notifier\n");
839 
840     return 1;
841 }
842