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