xref: /freebsd/sys/net/netmap_user.h (revision ab40f58ccfe6c07ebefddc72f4661a52fe746353)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (C) 2011-2016 Universita` di Pisa
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  *
11  *   1. Redistributions of source code must retain the above copyright
12  *      notice, this list of conditions and the following disclaimer.
13  *   2. Redistributions in binary form must reproduce the above copyright
14  *      notice, this list of conditions and the following disclaimer in the
15  *      documentation and/or other materials provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 /*
31  * $FreeBSD$
32  *
33  * Functions and macros to manipulate netmap structures and packets
34  * in userspace. See netmap(4) for more information.
35  *
36  * The address of the struct netmap_if, say nifp, is computed from the
37  * value returned from ioctl(.., NIOCREG, ...) and the mmap region:
38  *	ioctl(fd, NIOCREG, &req);
39  *	mem = mmap(0, ... );
40  *	nifp = NETMAP_IF(mem, req.nr_nifp);
41  *		(so simple, we could just do it manually)
42  *
43  * From there:
44  *	struct netmap_ring *NETMAP_TXRING(nifp, index)
45  *	struct netmap_ring *NETMAP_RXRING(nifp, index)
46  *		we can access ring->cur, ring->head, ring->tail, etc.
47  *
48  *	ring->slot[i] gives us the i-th slot (we can access
49  *		directly len, flags, buf_idx)
50  *
51  *	char *buf = NETMAP_BUF(ring, x) returns a pointer to
52  *		the buffer numbered x
53  *
54  * All ring indexes (head, cur, tail) should always move forward.
55  * To compute the next index in a circular ring you can use
56  *	i = nm_ring_next(ring, i);
57  *
58  * To ease porting apps from pcap to netmap we supply a few fuctions
59  * that can be called to open, close, read and write on netmap in a way
60  * similar to libpcap. Note that the read/write function depend on
61  * an ioctl()/select()/poll() being issued to refill rings or push
62  * packets out.
63  *
64  * In order to use these, include #define NETMAP_WITH_LIBS
65  * in the source file that invokes these functions.
66  */
67 
68 #ifndef _NET_NETMAP_USER_H_
69 #define _NET_NETMAP_USER_H_
70 
71 #define NETMAP_DEVICE_NAME "/dev/netmap"
72 
73 #ifdef __CYGWIN__
74 /*
75  * we can compile userspace apps with either cygwin or msvc,
76  * and we use _WIN32 to identify windows specific code
77  */
78 #ifndef _WIN32
79 #define _WIN32
80 #endif	/* _WIN32 */
81 
82 #endif	/* __CYGWIN__ */
83 
84 #ifdef _WIN32
85 #undef NETMAP_DEVICE_NAME
86 #define NETMAP_DEVICE_NAME "/proc/sys/DosDevices/Global/netmap"
87 #include <windows.h>
88 #include <WinDef.h>
89 #include <sys/cygwin.h>
90 #endif /* _WIN32 */
91 
92 #include <stdint.h>
93 #include <sys/socket.h>		/* apple needs sockaddr */
94 #include <net/if.h>		/* IFNAMSIZ */
95 #include <ctype.h>
96 
97 #ifndef likely
98 #define likely(x)	__builtin_expect(!!(x), 1)
99 #define unlikely(x)	__builtin_expect(!!(x), 0)
100 #endif /* likely and unlikely */
101 
102 #include <net/netmap.h>
103 
104 /* helper macro */
105 #define _NETMAP_OFFSET(type, ptr, offset) \
106 	((type)(void *)((char *)(ptr) + (offset)))
107 
108 #define NETMAP_IF(_base, _ofs)	_NETMAP_OFFSET(struct netmap_if *, _base, _ofs)
109 
110 #define NETMAP_TXRING(nifp, index) _NETMAP_OFFSET(struct netmap_ring *, \
111 	nifp, (nifp)->ring_ofs[index] )
112 
113 #define NETMAP_RXRING(nifp, index) _NETMAP_OFFSET(struct netmap_ring *,	\
114 	nifp, (nifp)->ring_ofs[index + (nifp)->ni_tx_rings + 1] )
115 
116 #define NETMAP_BUF(ring, index)				\
117 	((char *)(ring) + (ring)->buf_ofs + ((index)*(ring)->nr_buf_size))
118 
119 #define NETMAP_BUF_IDX(ring, buf)			\
120 	( ((char *)(buf) - ((char *)(ring) + (ring)->buf_ofs) ) / \
121 		(ring)->nr_buf_size )
122 
123 
124 static inline uint32_t
125 nm_ring_next(struct netmap_ring *r, uint32_t i)
126 {
127 	return ( unlikely(i + 1 == r->num_slots) ? 0 : i + 1);
128 }
129 
130 
131 /*
132  * Return 1 if we have pending transmissions in the tx ring.
133  * When everything is complete ring->head = ring->tail + 1 (modulo ring size)
134  */
135 static inline int
136 nm_tx_pending(struct netmap_ring *r)
137 {
138 	return nm_ring_next(r, r->tail) != r->head;
139 }
140 
141 
142 static inline uint32_t
143 nm_ring_space(struct netmap_ring *ring)
144 {
145         int ret = ring->tail - ring->cur;
146         if (ret < 0)
147                 ret += ring->num_slots;
148         return ret;
149 }
150 
151 
152 #ifdef NETMAP_WITH_LIBS
153 /*
154  * Support for simple I/O libraries.
155  * Include other system headers required for compiling this.
156  */
157 
158 #ifndef HAVE_NETMAP_WITH_LIBS
159 #define HAVE_NETMAP_WITH_LIBS
160 
161 #include <stdio.h>
162 #include <sys/time.h>
163 #include <sys/mman.h>
164 #include <string.h>	/* memset */
165 #include <sys/ioctl.h>
166 #include <sys/errno.h>	/* EINVAL */
167 #include <fcntl.h>	/* O_RDWR */
168 #include <unistd.h>	/* close() */
169 #include <signal.h>
170 #include <stdlib.h>
171 
172 #ifndef ND /* debug macros */
173 /* debug support */
174 #define ND(_fmt, ...) do {} while(0)
175 #define D(_fmt, ...)						\
176 	do {							\
177 		struct timeval _t0;				\
178 		gettimeofday(&_t0, NULL);			\
179 		fprintf(stderr, "%03d.%06d %s [%d] " _fmt "\n",	\
180 		    (int)(_t0.tv_sec % 1000), (int)_t0.tv_usec,	\
181 		    __FUNCTION__, __LINE__, ##__VA_ARGS__);	\
182         } while (0)
183 
184 /* Rate limited version of "D", lps indicates how many per second */
185 #define RD(lps, format, ...)                                    \
186     do {                                                        \
187         static int __t0, __cnt;                                 \
188         struct timeval __xxts;                                  \
189         gettimeofday(&__xxts, NULL);                            \
190         if (__t0 != __xxts.tv_sec) {                            \
191             __t0 = __xxts.tv_sec;                               \
192             __cnt = 0;                                          \
193         }                                                       \
194         if (__cnt++ < lps) {                                    \
195             D(format, ##__VA_ARGS__);                           \
196         }                                                       \
197     } while (0)
198 #endif
199 
200 struct nm_pkthdr {	/* first part is the same as pcap_pkthdr */
201 	struct timeval	ts;
202 	uint32_t	caplen;
203 	uint32_t	len;
204 
205 	uint64_t flags;	/* NM_MORE_PKTS etc */
206 #define NM_MORE_PKTS	1
207 	struct nm_desc *d;
208 	struct netmap_slot *slot;
209 	uint8_t *buf;
210 };
211 
212 struct nm_stat {	/* same as pcap_stat	*/
213 	u_int	ps_recv;
214 	u_int	ps_drop;
215 	u_int	ps_ifdrop;
216 #ifdef WIN32 /* XXX or _WIN32 ? */
217 	u_int	bs_capt;
218 #endif /* WIN32 */
219 };
220 
221 #define NM_ERRBUF_SIZE	512
222 
223 struct nm_desc {
224 	struct nm_desc *self; /* point to self if netmap. */
225 	int fd;
226 	void *mem;
227 	uint32_t memsize;
228 	int done_mmap;	/* set if mem is the result of mmap */
229 	struct netmap_if * const nifp;
230 	uint16_t first_tx_ring, last_tx_ring, cur_tx_ring;
231 	uint16_t first_rx_ring, last_rx_ring, cur_rx_ring;
232 	struct nmreq req;	/* also contains the nr_name = ifname */
233 	struct nm_pkthdr hdr;
234 
235 	/*
236 	 * The memory contains netmap_if, rings and then buffers.
237 	 * Given a pointer (e.g. to nm_inject) we can compare with
238 	 * mem/buf_start/buf_end to tell if it is a buffer or
239 	 * some other descriptor in our region.
240 	 * We also store a pointer to some ring as it helps in the
241 	 * translation from buffer indexes to addresses.
242 	 */
243 	struct netmap_ring * const some_ring;
244 	void * const buf_start;
245 	void * const buf_end;
246 	/* parameters from pcap_open_live */
247 	int snaplen;
248 	int promisc;
249 	int to_ms;
250 	char *errbuf;
251 
252 	/* save flags so we can restore them on close */
253 	uint32_t if_flags;
254         uint32_t if_reqcap;
255         uint32_t if_curcap;
256 
257 	struct nm_stat st;
258 	char msg[NM_ERRBUF_SIZE];
259 };
260 
261 /*
262  * when the descriptor is open correctly, d->self == d
263  * Eventually we should also use some magic number.
264  */
265 #define P2NMD(p)		((struct nm_desc *)(p))
266 #define IS_NETMAP_DESC(d)	((d) && P2NMD(d)->self == P2NMD(d))
267 #define NETMAP_FD(d)		(P2NMD(d)->fd)
268 
269 
270 /*
271  * this is a slightly optimized copy routine which rounds
272  * to multiple of 64 bytes and is often faster than dealing
273  * with other odd sizes. We assume there is enough room
274  * in the source and destination buffers.
275  */
276 static inline void
277 nm_pkt_copy(const void *_src, void *_dst, int l)
278 {
279 	const uint64_t *src = (const uint64_t *)_src;
280 	uint64_t *dst = (uint64_t *)_dst;
281 
282 	if (unlikely(l >= 1024 || l % 64)) {
283 		memcpy(dst, src, l);
284 		return;
285 	}
286 	for (; likely(l > 0); l-=64) {
287 		*dst++ = *src++;
288 		*dst++ = *src++;
289 		*dst++ = *src++;
290 		*dst++ = *src++;
291 		*dst++ = *src++;
292 		*dst++ = *src++;
293 		*dst++ = *src++;
294 		*dst++ = *src++;
295 	}
296 }
297 
298 
299 /*
300  * The callback, invoked on each received packet. Same as libpcap
301  */
302 typedef void (*nm_cb_t)(u_char *, const struct nm_pkthdr *, const u_char *d);
303 
304 /*
305  *--- the pcap-like API ---
306  *
307  * nm_open() opens a file descriptor, binds to a port and maps memory.
308  *
309  * ifname	(netmap:foo or vale:foo) is the port name
310  *		a suffix can indicate the follwing:
311  *		^		bind the host (sw) ring pair
312  *		*		bind host and NIC ring pairs
313  *		-NN		bind individual NIC ring pair
314  *		{NN		bind master side of pipe NN
315  *		}NN		bind slave side of pipe NN
316  *		a suffix starting with / and the following flags,
317  *		in any order:
318  *		x		exclusive access
319  *		z		zero copy monitor (both tx and rx)
320  *		t		monitor tx side (copy monitor)
321  *		r		monitor rx side (copy monitor)
322  *		R		bind only RX ring(s)
323  *		T		bind only TX ring(s)
324  *
325  * req		provides the initial values of nmreq before parsing ifname.
326  *		Remember that the ifname parsing will override the ring
327  *		number in nm_ringid, and part of nm_flags;
328  * flags	special functions, normally 0
329  *		indicates which fields of *arg are significant
330  * arg		special functions, normally NULL
331  *		if passed a netmap_desc with mem != NULL,
332  *		use that memory instead of mmap.
333  */
334 
335 static struct nm_desc *nm_open(const char *ifname, const struct nmreq *req,
336 	uint64_t flags, const struct nm_desc *arg);
337 
338 /*
339  * nm_open can import some fields from the parent descriptor.
340  * These flags control which ones.
341  * Also in flags you can specify NETMAP_NO_TX_POLL and NETMAP_DO_RX_POLL,
342  * which set the initial value for these flags.
343  * Note that the 16 low bits of the flags are reserved for data
344  * that may go into the nmreq.
345  */
346 enum {
347 	NM_OPEN_NO_MMAP =	0x040000, /* reuse mmap from parent */
348 	NM_OPEN_IFNAME =	0x080000, /* nr_name, nr_ringid, nr_flags */
349 	NM_OPEN_ARG1 =		0x100000,
350 	NM_OPEN_ARG2 =		0x200000,
351 	NM_OPEN_ARG3 =		0x400000,
352 	NM_OPEN_RING_CFG =	0x800000, /* tx|rx rings|slots */
353 };
354 
355 
356 /*
357  * nm_close()	closes and restores the port to its previous state
358  */
359 
360 static int nm_close(struct nm_desc *);
361 
362 /*
363  * nm_mmap()    do mmap or inherit from parent if the nr_arg2
364  *              (memory block) matches.
365  */
366 
367 static int nm_mmap(struct nm_desc *, const struct nm_desc *);
368 
369 /*
370  * nm_inject() is the same as pcap_inject()
371  * nm_dispatch() is the same as pcap_dispatch()
372  * nm_nextpkt() is the same as pcap_next()
373  */
374 
375 static int nm_inject(struct nm_desc *, const void *, size_t);
376 static int nm_dispatch(struct nm_desc *, int, nm_cb_t, u_char *);
377 static u_char *nm_nextpkt(struct nm_desc *, struct nm_pkthdr *);
378 
379 #ifdef _WIN32
380 
381 intptr_t _get_osfhandle(int); /* defined in io.h in windows */
382 
383 /*
384  * In windows we do not have yet native poll support, so we keep track
385  * of file descriptors associated to netmap ports to emulate poll on
386  * them and fall back on regular poll on other file descriptors.
387  */
388 struct win_netmap_fd_list {
389 	struct win_netmap_fd_list *next;
390 	int win_netmap_fd;
391 	HANDLE win_netmap_handle;
392 };
393 
394 /*
395  * list head containing all the netmap opened fd and their
396  * windows HANDLE counterparts
397  */
398 static struct win_netmap_fd_list *win_netmap_fd_list_head;
399 
400 static void
401 win_insert_fd_record(int fd)
402 {
403 	struct win_netmap_fd_list *curr;
404 
405 	for (curr = win_netmap_fd_list_head; curr; curr = curr->next) {
406 		if (fd == curr->win_netmap_fd) {
407 			return;
408 		}
409 	}
410 	curr = calloc(1, sizeof(*curr));
411 	curr->next = win_netmap_fd_list_head;
412 	curr->win_netmap_fd = fd;
413 	curr->win_netmap_handle = IntToPtr(_get_osfhandle(fd));
414 	win_netmap_fd_list_head = curr;
415 }
416 
417 void
418 win_remove_fd_record(int fd)
419 {
420 	struct win_netmap_fd_list *curr = win_netmap_fd_list_head;
421 	struct win_netmap_fd_list *prev = NULL;
422 	for (; curr ; prev = curr, curr = curr->next) {
423 		if (fd != curr->win_netmap_fd)
424 			continue;
425 		/* found the entry */
426 		if (prev == NULL) { /* we are freeing the first entry */
427 			win_netmap_fd_list_head = curr->next;
428 		} else {
429 			prev->next = curr->next;
430 		}
431 		free(curr);
432 		break;
433 	}
434 }
435 
436 
437 HANDLE
438 win_get_netmap_handle(int fd)
439 {
440 	struct win_netmap_fd_list *curr;
441 
442 	for (curr = win_netmap_fd_list_head; curr; curr = curr->next) {
443 		if (fd == curr->win_netmap_fd) {
444 			return curr->win_netmap_handle;
445 		}
446 	}
447 	return NULL;
448 }
449 
450 /*
451  * we need to wrap ioctl and mmap, at least for the netmap file descriptors
452  */
453 
454 /*
455  * use this function only from netmap_user.h internal functions
456  * same as ioctl, returns 0 on success and -1 on error
457  */
458 static int
459 win_nm_ioctl_internal(HANDLE h, int32_t ctlCode, void *arg)
460 {
461 	DWORD bReturn = 0, szIn, szOut;
462 	BOOL ioctlReturnStatus;
463 	void *inParam = arg, *outParam = arg;
464 
465 	switch (ctlCode) {
466 	case NETMAP_POLL:
467 		szIn = sizeof(POLL_REQUEST_DATA);
468 		szOut = sizeof(POLL_REQUEST_DATA);
469 		break;
470 	case NETMAP_MMAP:
471 		szIn = 0;
472 		szOut = sizeof(void*);
473 		inParam = NULL; /* nothing on input */
474 		break;
475 	case NIOCTXSYNC:
476 	case NIOCRXSYNC:
477 		szIn = 0;
478 		szOut = 0;
479 		break;
480 	case NIOCREGIF:
481 		szIn = sizeof(struct nmreq);
482 		szOut = sizeof(struct nmreq);
483 		break;
484 	case NIOCCONFIG:
485 		D("unsupported NIOCCONFIG!");
486 		return -1;
487 
488 	default: /* a regular ioctl */
489 		D("invalid ioctl %x on netmap fd", ctlCode);
490 		return -1;
491 	}
492 
493 	ioctlReturnStatus = DeviceIoControl(h,
494 				ctlCode, inParam, szIn,
495 				outParam, szOut,
496 				&bReturn, NULL);
497 	// XXX note windows returns 0 on error or async call, 1 on success
498 	// we could call GetLastError() to figure out what happened
499 	return ioctlReturnStatus ? 0 : -1;
500 }
501 
502 /*
503  * this function is what must be called from user-space programs
504  * same as ioctl, returns 0 on success and -1 on error
505  */
506 static int
507 win_nm_ioctl(int fd, int32_t ctlCode, void *arg)
508 {
509 	HANDLE h = win_get_netmap_handle(fd);
510 
511 	if (h == NULL) {
512 		return ioctl(fd, ctlCode, arg);
513 	} else {
514 		return win_nm_ioctl_internal(h, ctlCode, arg);
515 	}
516 }
517 
518 #define ioctl win_nm_ioctl /* from now on, within this file ... */
519 
520 /*
521  * We cannot use the native mmap on windows
522  * The only parameter used is "fd", the other ones are just declared to
523  * make this signature comparable to the FreeBSD/Linux one
524  */
525 static void *
526 win32_mmap_emulated(void *addr, size_t length, int prot, int flags, int fd, int32_t offset)
527 {
528 	HANDLE h = win_get_netmap_handle(fd);
529 
530 	if (h == NULL) {
531 		return mmap(addr, length, prot, flags, fd, offset);
532 	} else {
533 		MEMORY_ENTRY ret;
534 
535 		return win_nm_ioctl_internal(h, NETMAP_MMAP, &ret) ?
536 			NULL : ret.pUsermodeVirtualAddress;
537 	}
538 }
539 
540 #define mmap win32_mmap_emulated
541 
542 #include <sys/poll.h> /* XXX needed to use the structure pollfd */
543 
544 static int
545 win_nm_poll(struct pollfd *fds, int nfds, int timeout)
546 {
547 	HANDLE h;
548 
549 	if (nfds != 1 || fds == NULL || (h = win_get_netmap_handle(fds->fd)) == NULL) {;
550 		return poll(fds, nfds, timeout);
551 	} else {
552 		POLL_REQUEST_DATA prd;
553 
554 		prd.timeout = timeout;
555 		prd.events = fds->events;
556 
557 		win_nm_ioctl_internal(h, NETMAP_POLL, &prd);
558 		if ((prd.revents == POLLERR) || (prd.revents == STATUS_TIMEOUT)) {
559 			return -1;
560 		}
561 		return 1;
562 	}
563 }
564 
565 #define poll win_nm_poll
566 
567 static int
568 win_nm_open(char* pathname, int flags)
569 {
570 
571 	if (strcmp(pathname, NETMAP_DEVICE_NAME) == 0) {
572 		int fd = open(NETMAP_DEVICE_NAME, O_RDWR);
573 		if (fd < 0) {
574 			return -1;
575 		}
576 
577 		win_insert_fd_record(fd);
578 		return fd;
579 	} else {
580 		return open(pathname, flags);
581 	}
582 }
583 
584 #define open win_nm_open
585 
586 static int
587 win_nm_close(int fd)
588 {
589 	if (fd != -1) {
590 		close(fd);
591 		if (win_get_netmap_handle(fd) != NULL) {
592 			win_remove_fd_record(fd);
593 		}
594 	}
595 	return 0;
596 }
597 
598 #define close win_nm_close
599 
600 #endif /* _WIN32 */
601 
602 static int
603 nm_is_identifier(const char *s, const char *e)
604 {
605 	for (; s != e; s++) {
606 		if (!isalnum(*s) && *s != '_') {
607 			return 0;
608 		}
609 	}
610 
611 	return 1;
612 }
613 
614 #define MAXERRMSG 80
615 static int
616 nm_parse(const char *ifname, struct nm_desc *d, char *err)
617 {
618 	int is_vale;
619 	const char *port = NULL;
620 	const char *vpname = NULL;
621 	u_int namelen;
622 	uint32_t nr_ringid = 0, nr_flags;
623 	char errmsg[MAXERRMSG] = "";
624 	long num;
625 	uint16_t nr_arg2 = 0;
626 	enum { P_START, P_RNGSFXOK, P_GETNUM, P_FLAGS, P_FLAGSOK, P_MEMID } p_state;
627 
628 	errno = 0;
629 
630 	is_vale = (ifname[0] == 'v');
631 	if (is_vale) {
632 		port = index(ifname, ':');
633 		if (port == NULL) {
634 			snprintf(errmsg, MAXERRMSG,
635 				 "missing ':' in vale name");
636 			goto fail;
637 		}
638 
639 		if (!nm_is_identifier(ifname + 4, port)) {
640 			snprintf(errmsg, MAXERRMSG, "invalid bridge name");
641 			goto fail;
642 		}
643 
644 		vpname = ++port;
645 	} else {
646 		ifname += 7;
647 		port = ifname;
648 	}
649 
650 	/* scan for a separator */
651 	for (; *port && !index("-*^{}/@", *port); port++)
652 		;
653 
654 	if (is_vale && !nm_is_identifier(vpname, port)) {
655 		snprintf(errmsg, MAXERRMSG, "invalid bridge port name");
656 		goto fail;
657 	}
658 
659 	namelen = port - ifname;
660 	if (namelen >= sizeof(d->req.nr_name)) {
661 		snprintf(errmsg, MAXERRMSG, "name too long");
662 		goto fail;
663 	}
664 	memcpy(d->req.nr_name, ifname, namelen);
665 	d->req.nr_name[namelen] = '\0';
666 
667 	p_state = P_START;
668 	nr_flags = NR_REG_ALL_NIC; /* default for no suffix */
669 	while (*port) {
670 		switch (p_state) {
671 		case P_START:
672 			switch (*port) {
673 			case '^': /* only SW ring */
674 				nr_flags = NR_REG_SW;
675 				p_state = P_RNGSFXOK;
676 				break;
677 			case '*': /* NIC and SW */
678 				nr_flags = NR_REG_NIC_SW;
679 				p_state = P_RNGSFXOK;
680 				break;
681 			case '-': /* one NIC ring pair */
682 				nr_flags = NR_REG_ONE_NIC;
683 				p_state = P_GETNUM;
684 				break;
685 			case '{': /* pipe (master endpoint) */
686 				nr_flags = NR_REG_PIPE_MASTER;
687 				p_state = P_GETNUM;
688 				break;
689 			case '}': /* pipe (slave endoint) */
690 				nr_flags = NR_REG_PIPE_SLAVE;
691 				p_state = P_GETNUM;
692 				break;
693 			case '/': /* start of flags */
694 				p_state = P_FLAGS;
695 				break;
696 			case '@': /* start of memid */
697 				p_state = P_MEMID;
698 				break;
699 			default:
700 				snprintf(errmsg, MAXERRMSG, "unknown modifier: '%c'", *port);
701 				goto fail;
702 			}
703 			port++;
704 			break;
705 		case P_RNGSFXOK:
706 			switch (*port) {
707 			case '/':
708 				p_state = P_FLAGS;
709 				break;
710 			case '@':
711 				p_state = P_MEMID;
712 				break;
713 			default:
714 				snprintf(errmsg, MAXERRMSG, "unexpected character: '%c'", *port);
715 				goto fail;
716 			}
717 			port++;
718 			break;
719 		case P_GETNUM:
720 			num = strtol(port, (char **)&port, 10);
721 			if (num < 0 || num >= NETMAP_RING_MASK) {
722 				snprintf(errmsg, MAXERRMSG, "'%ld' out of range [0, %d)",
723 						num, NETMAP_RING_MASK);
724 				goto fail;
725 			}
726 			nr_ringid = num & NETMAP_RING_MASK;
727 			p_state = P_RNGSFXOK;
728 			break;
729 		case P_FLAGS:
730 		case P_FLAGSOK:
731 			if (*port == '@') {
732 				port++;
733 				p_state = P_MEMID;
734 				break;
735 			}
736 			switch (*port) {
737 			case 'x':
738 				nr_flags |= NR_EXCLUSIVE;
739 				break;
740 			case 'z':
741 				nr_flags |= NR_ZCOPY_MON;
742 				break;
743 			case 't':
744 				nr_flags |= NR_MONITOR_TX;
745 				break;
746 			case 'r':
747 				nr_flags |= NR_MONITOR_RX;
748 				break;
749 			case 'R':
750 				nr_flags |= NR_RX_RINGS_ONLY;
751 				break;
752 			case 'T':
753 				nr_flags |= NR_TX_RINGS_ONLY;
754 				break;
755 			default:
756 				snprintf(errmsg, MAXERRMSG, "unrecognized flag: '%c'", *port);
757 				goto fail;
758 			}
759 			port++;
760 			p_state = P_FLAGSOK;
761 			break;
762 		case P_MEMID:
763 			if (nr_arg2 != 0) {
764 				snprintf(errmsg, MAXERRMSG, "double setting of memid");
765 				goto fail;
766 			}
767 			num = strtol(port, (char **)&port, 10);
768 			if (num <= 0) {
769 				snprintf(errmsg, MAXERRMSG, "invalid memid %ld, must be >0", num);
770 				goto fail;
771 			}
772 			nr_arg2 = num;
773 			p_state = P_RNGSFXOK;
774 			break;
775 		}
776 	}
777 	if (p_state != P_START && p_state != P_RNGSFXOK && p_state != P_FLAGSOK) {
778 		snprintf(errmsg, MAXERRMSG, "unexpected end of port name");
779 		goto fail;
780 	}
781 	ND("flags: %s %s %s %s",
782 			(nr_flags & NR_EXCLUSIVE) ? "EXCLUSIVE" : "",
783 			(nr_flags & NR_ZCOPY_MON) ? "ZCOPY_MON" : "",
784 			(nr_flags & NR_MONITOR_TX) ? "MONITOR_TX" : "",
785 			(nr_flags & NR_MONITOR_RX) ? "MONITOR_RX" : "");
786 
787 	d->req.nr_flags |= nr_flags;
788 	d->req.nr_ringid |= nr_ringid;
789 	d->req.nr_arg2 = nr_arg2;
790 
791 	d->self = d;
792 
793 	return 0;
794 fail:
795 	if (!errno)
796 		errno = EINVAL;
797 	if (err)
798 		strncpy(err, errmsg, MAXERRMSG);
799 	return -1;
800 }
801 
802 /*
803  * Try to open, return descriptor if successful, NULL otherwise.
804  * An invalid netmap name will return errno = 0;
805  * You can pass a pointer to a pre-filled nm_desc to add special
806  * parameters. Flags is used as follows
807  * NM_OPEN_NO_MMAP	use the memory from arg, only XXX avoid mmap
808  *			if the nr_arg2 (memory block) matches.
809  * NM_OPEN_ARG1		use req.nr_arg1 from arg
810  * NM_OPEN_ARG2		use req.nr_arg2 from arg
811  * NM_OPEN_RING_CFG	user ring config from arg
812  */
813 static struct nm_desc *
814 nm_open(const char *ifname, const struct nmreq *req,
815 	uint64_t new_flags, const struct nm_desc *arg)
816 {
817 	struct nm_desc *d = NULL;
818 	const struct nm_desc *parent = arg;
819 	char errmsg[MAXERRMSG] = "";
820 	uint32_t nr_reg;
821 
822 	if (strncmp(ifname, "netmap:", 7) &&
823 			strncmp(ifname, NM_BDG_NAME, strlen(NM_BDG_NAME))) {
824 		errno = 0; /* name not recognised, not an error */
825 		return NULL;
826 	}
827 
828 	d = (struct nm_desc *)calloc(1, sizeof(*d));
829 	if (d == NULL) {
830 		snprintf(errmsg, MAXERRMSG, "nm_desc alloc failure");
831 		errno = ENOMEM;
832 		return NULL;
833 	}
834 	d->self = d;	/* set this early so nm_close() works */
835 	d->fd = open(NETMAP_DEVICE_NAME, O_RDWR);
836 	if (d->fd < 0) {
837 		snprintf(errmsg, MAXERRMSG, "cannot open /dev/netmap: %s", strerror(errno));
838 		goto fail;
839 	}
840 
841 	if (req)
842 		d->req = *req;
843 
844 	if (!(new_flags & NM_OPEN_IFNAME)) {
845 		if (nm_parse(ifname, d, errmsg) < 0)
846 			goto fail;
847 	}
848 
849 	d->req.nr_version = NETMAP_API;
850 	d->req.nr_ringid &= NETMAP_RING_MASK;
851 
852 	/* optionally import info from parent */
853 	if (IS_NETMAP_DESC(parent) && new_flags) {
854 		if (new_flags & NM_OPEN_ARG1)
855 			D("overriding ARG1 %d", parent->req.nr_arg1);
856 		d->req.nr_arg1 = new_flags & NM_OPEN_ARG1 ?
857 			parent->req.nr_arg1 : 4;
858 		if (new_flags & NM_OPEN_ARG2) {
859 			D("overriding ARG2 %d", parent->req.nr_arg2);
860 			d->req.nr_arg2 =  parent->req.nr_arg2;
861 		}
862 		if (new_flags & NM_OPEN_ARG3)
863 			D("overriding ARG3 %d", parent->req.nr_arg3);
864 		d->req.nr_arg3 = new_flags & NM_OPEN_ARG3 ?
865 			parent->req.nr_arg3 : 0;
866 		if (new_flags & NM_OPEN_RING_CFG) {
867 			D("overriding RING_CFG");
868 			d->req.nr_tx_slots = parent->req.nr_tx_slots;
869 			d->req.nr_rx_slots = parent->req.nr_rx_slots;
870 			d->req.nr_tx_rings = parent->req.nr_tx_rings;
871 			d->req.nr_rx_rings = parent->req.nr_rx_rings;
872 		}
873 		if (new_flags & NM_OPEN_IFNAME) {
874 			D("overriding ifname %s ringid 0x%x flags 0x%x",
875 				parent->req.nr_name, parent->req.nr_ringid,
876 				parent->req.nr_flags);
877 			memcpy(d->req.nr_name, parent->req.nr_name,
878 				sizeof(d->req.nr_name));
879 			d->req.nr_ringid = parent->req.nr_ringid;
880 			d->req.nr_flags = parent->req.nr_flags;
881 		}
882 	}
883 	/* add the *XPOLL flags */
884 	d->req.nr_ringid |= new_flags & (NETMAP_NO_TX_POLL | NETMAP_DO_RX_POLL);
885 
886 	if (ioctl(d->fd, NIOCREGIF, &d->req)) {
887 		snprintf(errmsg, MAXERRMSG, "NIOCREGIF failed: %s", strerror(errno));
888 		goto fail;
889 	}
890 
891 	nr_reg = d->req.nr_flags & NR_REG_MASK;
892 
893 	if (nr_reg == NR_REG_SW) { /* host stack */
894 		d->first_tx_ring = d->last_tx_ring = d->req.nr_tx_rings;
895 		d->first_rx_ring = d->last_rx_ring = d->req.nr_rx_rings;
896 	} else if (nr_reg ==  NR_REG_ALL_NIC) { /* only nic */
897 		d->first_tx_ring = 0;
898 		d->first_rx_ring = 0;
899 		d->last_tx_ring = d->req.nr_tx_rings - 1;
900 		d->last_rx_ring = d->req.nr_rx_rings - 1;
901 	} else if (nr_reg ==  NR_REG_NIC_SW) {
902 		d->first_tx_ring = 0;
903 		d->first_rx_ring = 0;
904 		d->last_tx_ring = d->req.nr_tx_rings;
905 		d->last_rx_ring = d->req.nr_rx_rings;
906 	} else if (nr_reg == NR_REG_ONE_NIC) {
907 		/* XXX check validity */
908 		d->first_tx_ring = d->last_tx_ring =
909 		d->first_rx_ring = d->last_rx_ring = d->req.nr_ringid & NETMAP_RING_MASK;
910 	} else { /* pipes */
911 		d->first_tx_ring = d->last_tx_ring = 0;
912 		d->first_rx_ring = d->last_rx_ring = 0;
913 	}
914 
915         /* if parent is defined, do nm_mmap() even if NM_OPEN_NO_MMAP is set */
916 	if ((!(new_flags & NM_OPEN_NO_MMAP) || parent) && nm_mmap(d, parent)) {
917 	        snprintf(errmsg, MAXERRMSG, "mmap failed: %s", strerror(errno));
918 		goto fail;
919 	}
920 
921 
922 #ifdef DEBUG_NETMAP_USER
923     { /* debugging code */
924 	int i;
925 
926 	D("%s tx %d .. %d %d rx %d .. %d %d", ifname,
927 		d->first_tx_ring, d->last_tx_ring, d->req.nr_tx_rings,
928                 d->first_rx_ring, d->last_rx_ring, d->req.nr_rx_rings);
929 	for (i = 0; i <= d->req.nr_tx_rings; i++) {
930 		struct netmap_ring *r = NETMAP_TXRING(d->nifp, i);
931 		D("TX%d %p h %d c %d t %d", i, r, r->head, r->cur, r->tail);
932 	}
933 	for (i = 0; i <= d->req.nr_rx_rings; i++) {
934 		struct netmap_ring *r = NETMAP_RXRING(d->nifp, i);
935 		D("RX%d %p h %d c %d t %d", i, r, r->head, r->cur, r->tail);
936 	}
937     }
938 #endif /* debugging */
939 
940 	d->cur_tx_ring = d->first_tx_ring;
941 	d->cur_rx_ring = d->first_rx_ring;
942 	return d;
943 
944 fail:
945 	nm_close(d);
946 	if (errmsg[0])
947 		D("%s %s", errmsg, ifname);
948 	if (errno == 0)
949 		errno = EINVAL;
950 	return NULL;
951 }
952 
953 
954 static int
955 nm_close(struct nm_desc *d)
956 {
957 	/*
958 	 * ugly trick to avoid unused warnings
959 	 */
960 	static void *__xxzt[] __attribute__ ((unused))  =
961 		{ (void *)nm_open, (void *)nm_inject,
962 		  (void *)nm_dispatch, (void *)nm_nextpkt } ;
963 
964 	if (d == NULL || d->self != d)
965 		return EINVAL;
966 	if (d->done_mmap && d->mem)
967 		munmap(d->mem, d->memsize);
968 	if (d->fd != -1) {
969 		close(d->fd);
970 	}
971 
972 	bzero(d, sizeof(*d));
973 	free(d);
974 	return 0;
975 }
976 
977 
978 static int
979 nm_mmap(struct nm_desc *d, const struct nm_desc *parent)
980 {
981 	//XXX TODO: check if mmap is already done
982 
983 	if (IS_NETMAP_DESC(parent) && parent->mem &&
984 	    parent->req.nr_arg2 == d->req.nr_arg2) {
985 		/* do not mmap, inherit from parent */
986 		D("do not mmap, inherit from parent");
987 		d->memsize = parent->memsize;
988 		d->mem = parent->mem;
989 	} else {
990 		/* XXX TODO: check if memsize is too large (or there is overflow) */
991 		d->memsize = d->req.nr_memsize;
992 		d->mem = mmap(0, d->memsize, PROT_WRITE | PROT_READ, MAP_SHARED,
993 				d->fd, 0);
994 		if (d->mem == MAP_FAILED) {
995 			goto fail;
996 		}
997 		d->done_mmap = 1;
998 	}
999 	{
1000 		struct netmap_if *nifp = NETMAP_IF(d->mem, d->req.nr_offset);
1001 		struct netmap_ring *r = NETMAP_RXRING(nifp, d->first_rx_ring);
1002 		if ((void *)r == (void *)nifp) {
1003 			/* the descriptor is open for TX only */
1004 			r = NETMAP_TXRING(nifp, d->first_tx_ring);
1005 		}
1006 
1007 		*(struct netmap_if **)(uintptr_t)&(d->nifp) = nifp;
1008 		*(struct netmap_ring **)(uintptr_t)&d->some_ring = r;
1009 		*(void **)(uintptr_t)&d->buf_start = NETMAP_BUF(r, 0);
1010 		*(void **)(uintptr_t)&d->buf_end =
1011 			(char *)d->mem + d->memsize;
1012 	}
1013 
1014 	return 0;
1015 
1016 fail:
1017 	return EINVAL;
1018 }
1019 
1020 /*
1021  * Same prototype as pcap_inject(), only need to cast.
1022  */
1023 static int
1024 nm_inject(struct nm_desc *d, const void *buf, size_t size)
1025 {
1026 	u_int c, n = d->last_tx_ring - d->first_tx_ring + 1,
1027 		ri = d->cur_tx_ring;
1028 
1029 	for (c = 0; c < n ; c++, ri++) {
1030 		/* compute current ring to use */
1031 		struct netmap_ring *ring;
1032 		uint32_t i, idx;
1033 
1034 		if (ri > d->last_tx_ring)
1035 			ri = d->first_tx_ring;
1036 		ring = NETMAP_TXRING(d->nifp, ri);
1037 		if (nm_ring_empty(ring)) {
1038 			continue;
1039 		}
1040 		i = ring->cur;
1041 		idx = ring->slot[i].buf_idx;
1042 		ring->slot[i].len = size;
1043 		nm_pkt_copy(buf, NETMAP_BUF(ring, idx), size);
1044 		d->cur_tx_ring = ri;
1045 		ring->head = ring->cur = nm_ring_next(ring, i);
1046 		return size;
1047 	}
1048 	return 0; /* fail */
1049 }
1050 
1051 
1052 /*
1053  * Same prototype as pcap_dispatch(), only need to cast.
1054  */
1055 static int
1056 nm_dispatch(struct nm_desc *d, int cnt, nm_cb_t cb, u_char *arg)
1057 {
1058 	int n = d->last_rx_ring - d->first_rx_ring + 1;
1059 	int c, got = 0, ri = d->cur_rx_ring;
1060 	d->hdr.buf = NULL;
1061 	d->hdr.flags = NM_MORE_PKTS;
1062 	d->hdr.d = d;
1063 
1064 	if (cnt == 0)
1065 		cnt = -1;
1066 	/* cnt == -1 means infinite, but rings have a finite amount
1067 	 * of buffers and the int is large enough that we never wrap,
1068 	 * so we can omit checking for -1
1069 	 */
1070 	for (c=0; c < n && cnt != got; c++, ri++) {
1071 		/* compute current ring to use */
1072 		struct netmap_ring *ring;
1073 
1074 		if (ri > d->last_rx_ring)
1075 			ri = d->first_rx_ring;
1076 		ring = NETMAP_RXRING(d->nifp, ri);
1077 		for ( ; !nm_ring_empty(ring) && cnt != got; got++) {
1078 			u_int idx, i;
1079 			if (d->hdr.buf) { /* from previous round */
1080 				cb(arg, &d->hdr, d->hdr.buf);
1081 			}
1082 			i = ring->cur;
1083 			idx = ring->slot[i].buf_idx;
1084 			/* d->cur_rx_ring doesn't change inside this loop, but
1085 			 * set it here, so it reflects d->hdr.buf's ring */
1086 			d->cur_rx_ring = ri;
1087 			d->hdr.slot = &ring->slot[i];
1088 			d->hdr.buf = (u_char *)NETMAP_BUF(ring, idx);
1089 			// __builtin_prefetch(buf);
1090 			d->hdr.len = d->hdr.caplen = ring->slot[i].len;
1091 			d->hdr.ts = ring->ts;
1092 			ring->head = ring->cur = nm_ring_next(ring, i);
1093 		}
1094 	}
1095 	if (d->hdr.buf) { /* from previous round */
1096 		d->hdr.flags = 0;
1097 		cb(arg, &d->hdr, d->hdr.buf);
1098 	}
1099 	return got;
1100 }
1101 
1102 static u_char *
1103 nm_nextpkt(struct nm_desc *d, struct nm_pkthdr *hdr)
1104 {
1105 	int ri = d->cur_rx_ring;
1106 
1107 	do {
1108 		/* compute current ring to use */
1109 		struct netmap_ring *ring = NETMAP_RXRING(d->nifp, ri);
1110 		if (!nm_ring_empty(ring)) {
1111 			u_int i = ring->cur;
1112 			u_int idx = ring->slot[i].buf_idx;
1113 			u_char *buf = (u_char *)NETMAP_BUF(ring, idx);
1114 
1115 			// __builtin_prefetch(buf);
1116 			hdr->ts = ring->ts;
1117 			hdr->len = hdr->caplen = ring->slot[i].len;
1118 			ring->cur = nm_ring_next(ring, i);
1119 			/* we could postpone advancing head if we want
1120 			 * to hold the buffer. This can be supported in
1121 			 * the future.
1122 			 */
1123 			ring->head = ring->cur;
1124 			d->cur_rx_ring = ri;
1125 			return buf;
1126 		}
1127 		ri++;
1128 		if (ri > d->last_rx_ring)
1129 			ri = d->first_rx_ring;
1130 	} while (ri != d->cur_rx_ring);
1131 	return NULL; /* nothing found */
1132 }
1133 
1134 #endif /* !HAVE_NETMAP_WITH_LIBS */
1135 
1136 #endif /* NETMAP_WITH_LIBS */
1137 
1138 #endif /* _NET_NETMAP_USER_H_ */
1139