xref: /dpdk/drivers/net/af_packet/rte_eth_af_packet.c (revision 200bc52e5aa0d72e70464c9cd22b55cf536ed13c)
1 /* SPDX-License-Identifier: BSD-3-Clause
2  * Copyright(c) 2014 John W. Linville <linville@tuxdriver.com>
3  * Originally based upon librte_pmd_pcap code:
4  * Copyright(c) 2010-2015 Intel Corporation.
5  * Copyright(c) 2014 6WIND S.A.
6  * All rights reserved.
7  */
8 
9 #include <rte_string_fns.h>
10 #include <rte_mbuf.h>
11 #include <rte_ethdev_driver.h>
12 #include <rte_ethdev_vdev.h>
13 #include <rte_malloc.h>
14 #include <rte_kvargs.h>
15 #include <rte_bus_vdev.h>
16 
17 #include <linux/if_ether.h>
18 #include <linux/if_packet.h>
19 #include <arpa/inet.h>
20 #include <net/if.h>
21 #include <sys/types.h>
22 #include <sys/socket.h>
23 #include <sys/ioctl.h>
24 #include <sys/mman.h>
25 #include <unistd.h>
26 #include <poll.h>
27 
28 #define ETH_AF_PACKET_IFACE_ARG		"iface"
29 #define ETH_AF_PACKET_NUM_Q_ARG		"qpairs"
30 #define ETH_AF_PACKET_BLOCKSIZE_ARG	"blocksz"
31 #define ETH_AF_PACKET_FRAMESIZE_ARG	"framesz"
32 #define ETH_AF_PACKET_FRAMECOUNT_ARG	"framecnt"
33 #define ETH_AF_PACKET_QDISC_BYPASS_ARG	"qdisc_bypass"
34 
35 #define DFLT_BLOCK_SIZE		(1 << 12)
36 #define DFLT_FRAME_SIZE		(1 << 11)
37 #define DFLT_FRAME_COUNT	(1 << 9)
38 
39 #define RTE_PMD_AF_PACKET_MAX_RINGS 16
40 
41 struct pkt_rx_queue {
42 	int sockfd;
43 
44 	struct iovec *rd;
45 	uint8_t *map;
46 	unsigned int framecount;
47 	unsigned int framenum;
48 
49 	struct rte_mempool *mb_pool;
50 	uint16_t in_port;
51 
52 	volatile unsigned long rx_pkts;
53 	volatile unsigned long err_pkts;
54 	volatile unsigned long rx_bytes;
55 };
56 
57 struct pkt_tx_queue {
58 	int sockfd;
59 	unsigned int frame_data_size;
60 
61 	struct iovec *rd;
62 	uint8_t *map;
63 	unsigned int framecount;
64 	unsigned int framenum;
65 
66 	volatile unsigned long tx_pkts;
67 	volatile unsigned long err_pkts;
68 	volatile unsigned long tx_bytes;
69 };
70 
71 struct pmd_internals {
72 	unsigned nb_queues;
73 
74 	int if_index;
75 	char *if_name;
76 	struct rte_ether_addr eth_addr;
77 
78 	struct tpacket_req req;
79 
80 	struct pkt_rx_queue rx_queue[RTE_PMD_AF_PACKET_MAX_RINGS];
81 	struct pkt_tx_queue tx_queue[RTE_PMD_AF_PACKET_MAX_RINGS];
82 };
83 
84 static const char *valid_arguments[] = {
85 	ETH_AF_PACKET_IFACE_ARG,
86 	ETH_AF_PACKET_NUM_Q_ARG,
87 	ETH_AF_PACKET_BLOCKSIZE_ARG,
88 	ETH_AF_PACKET_FRAMESIZE_ARG,
89 	ETH_AF_PACKET_FRAMECOUNT_ARG,
90 	ETH_AF_PACKET_QDISC_BYPASS_ARG,
91 	NULL
92 };
93 
94 static struct rte_eth_link pmd_link = {
95 	.link_speed = ETH_SPEED_NUM_10G,
96 	.link_duplex = ETH_LINK_FULL_DUPLEX,
97 	.link_status = ETH_LINK_DOWN,
98 	.link_autoneg = ETH_LINK_FIXED,
99 };
100 
101 static int af_packet_logtype;
102 
103 #define PMD_LOG(level, fmt, args...) \
104 	rte_log(RTE_LOG_ ## level, af_packet_logtype, \
105 		"%s(): " fmt "\n", __func__, ##args)
106 
107 static uint16_t
108 eth_af_packet_rx(void *queue, struct rte_mbuf **bufs, uint16_t nb_pkts)
109 {
110 	unsigned i;
111 	struct tpacket2_hdr *ppd;
112 	struct rte_mbuf *mbuf;
113 	uint8_t *pbuf;
114 	struct pkt_rx_queue *pkt_q = queue;
115 	uint16_t num_rx = 0;
116 	unsigned long num_rx_bytes = 0;
117 	unsigned int framecount, framenum;
118 
119 	if (unlikely(nb_pkts == 0))
120 		return 0;
121 
122 	/*
123 	 * Reads the given number of packets from the AF_PACKET socket one by
124 	 * one and copies the packet data into a newly allocated mbuf.
125 	 */
126 	framecount = pkt_q->framecount;
127 	framenum = pkt_q->framenum;
128 	for (i = 0; i < nb_pkts; i++) {
129 		/* point at the next incoming frame */
130 		ppd = (struct tpacket2_hdr *) pkt_q->rd[framenum].iov_base;
131 		if ((ppd->tp_status & TP_STATUS_USER) == 0)
132 			break;
133 
134 		/* allocate the next mbuf */
135 		mbuf = rte_pktmbuf_alloc(pkt_q->mb_pool);
136 		if (unlikely(mbuf == NULL))
137 			break;
138 
139 		/* packet will fit in the mbuf, go ahead and receive it */
140 		rte_pktmbuf_pkt_len(mbuf) = rte_pktmbuf_data_len(mbuf) = ppd->tp_snaplen;
141 		pbuf = (uint8_t *) ppd + ppd->tp_mac;
142 		memcpy(rte_pktmbuf_mtod(mbuf, void *), pbuf, rte_pktmbuf_data_len(mbuf));
143 
144 		/* check for vlan info */
145 		if (ppd->tp_status & TP_STATUS_VLAN_VALID) {
146 			mbuf->vlan_tci = ppd->tp_vlan_tci;
147 			mbuf->ol_flags |= (PKT_RX_VLAN | PKT_RX_VLAN_STRIPPED);
148 		}
149 
150 		/* release incoming frame and advance ring buffer */
151 		ppd->tp_status = TP_STATUS_KERNEL;
152 		if (++framenum >= framecount)
153 			framenum = 0;
154 		mbuf->port = pkt_q->in_port;
155 
156 		/* account for the receive frame */
157 		bufs[i] = mbuf;
158 		num_rx++;
159 		num_rx_bytes += mbuf->pkt_len;
160 	}
161 	pkt_q->framenum = framenum;
162 	pkt_q->rx_pkts += num_rx;
163 	pkt_q->rx_bytes += num_rx_bytes;
164 	return num_rx;
165 }
166 
167 /*
168  * Callback to handle sending packets through a real NIC.
169  */
170 static uint16_t
171 eth_af_packet_tx(void *queue, struct rte_mbuf **bufs, uint16_t nb_pkts)
172 {
173 	struct tpacket2_hdr *ppd;
174 	struct rte_mbuf *mbuf;
175 	uint8_t *pbuf;
176 	unsigned int framecount, framenum;
177 	struct pollfd pfd;
178 	struct pkt_tx_queue *pkt_q = queue;
179 	uint16_t num_tx = 0;
180 	unsigned long num_tx_bytes = 0;
181 	int i;
182 
183 	if (unlikely(nb_pkts == 0))
184 		return 0;
185 
186 	memset(&pfd, 0, sizeof(pfd));
187 	pfd.fd = pkt_q->sockfd;
188 	pfd.events = POLLOUT;
189 	pfd.revents = 0;
190 
191 	framecount = pkt_q->framecount;
192 	framenum = pkt_q->framenum;
193 	ppd = (struct tpacket2_hdr *) pkt_q->rd[framenum].iov_base;
194 	for (i = 0; i < nb_pkts; i++) {
195 		mbuf = *bufs++;
196 
197 		/* drop oversized packets */
198 		if (mbuf->pkt_len > pkt_q->frame_data_size) {
199 			rte_pktmbuf_free(mbuf);
200 			continue;
201 		}
202 
203 		/* insert vlan info if necessary */
204 		if (mbuf->ol_flags & PKT_TX_VLAN_PKT) {
205 			if (rte_vlan_insert(&mbuf)) {
206 				rte_pktmbuf_free(mbuf);
207 				continue;
208 			}
209 		}
210 
211 		/* point at the next incoming frame */
212 		if ((ppd->tp_status != TP_STATUS_AVAILABLE) &&
213 		    (poll(&pfd, 1, -1) < 0))
214 			break;
215 
216 		/* copy the tx frame data */
217 		pbuf = (uint8_t *) ppd + TPACKET2_HDRLEN -
218 			sizeof(struct sockaddr_ll);
219 
220 		struct rte_mbuf *tmp_mbuf = mbuf;
221 		while (tmp_mbuf) {
222 			uint16_t data_len = rte_pktmbuf_data_len(tmp_mbuf);
223 			memcpy(pbuf, rte_pktmbuf_mtod(tmp_mbuf, void*), data_len);
224 			pbuf += data_len;
225 			tmp_mbuf = tmp_mbuf->next;
226 		}
227 
228 		ppd->tp_len = mbuf->pkt_len;
229 		ppd->tp_snaplen = mbuf->pkt_len;
230 
231 		/* release incoming frame and advance ring buffer */
232 		ppd->tp_status = TP_STATUS_SEND_REQUEST;
233 		if (++framenum >= framecount)
234 			framenum = 0;
235 		ppd = (struct tpacket2_hdr *) pkt_q->rd[framenum].iov_base;
236 
237 		num_tx++;
238 		num_tx_bytes += mbuf->pkt_len;
239 		rte_pktmbuf_free(mbuf);
240 	}
241 
242 	/* kick-off transmits */
243 	if (sendto(pkt_q->sockfd, NULL, 0, MSG_DONTWAIT, NULL, 0) == -1) {
244 		/* error sending -- no packets transmitted */
245 		num_tx = 0;
246 		num_tx_bytes = 0;
247 	}
248 
249 	pkt_q->framenum = framenum;
250 	pkt_q->tx_pkts += num_tx;
251 	pkt_q->err_pkts += i - num_tx;
252 	pkt_q->tx_bytes += num_tx_bytes;
253 	return i;
254 }
255 
256 static int
257 eth_dev_start(struct rte_eth_dev *dev)
258 {
259 	dev->data->dev_link.link_status = ETH_LINK_UP;
260 	return 0;
261 }
262 
263 /*
264  * This function gets called when the current port gets stopped.
265  */
266 static void
267 eth_dev_stop(struct rte_eth_dev *dev)
268 {
269 	unsigned i;
270 	int sockfd;
271 	struct pmd_internals *internals = dev->data->dev_private;
272 
273 	for (i = 0; i < internals->nb_queues; i++) {
274 		sockfd = internals->rx_queue[i].sockfd;
275 		if (sockfd != -1)
276 			close(sockfd);
277 
278 		/* Prevent use after free in case tx fd == rx fd */
279 		if (sockfd != internals->tx_queue[i].sockfd) {
280 			sockfd = internals->tx_queue[i].sockfd;
281 			if (sockfd != -1)
282 				close(sockfd);
283 		}
284 
285 		internals->rx_queue[i].sockfd = -1;
286 		internals->tx_queue[i].sockfd = -1;
287 	}
288 
289 	dev->data->dev_link.link_status = ETH_LINK_DOWN;
290 }
291 
292 static int
293 eth_dev_configure(struct rte_eth_dev *dev __rte_unused)
294 {
295 	return 0;
296 }
297 
298 static void
299 eth_dev_info(struct rte_eth_dev *dev, struct rte_eth_dev_info *dev_info)
300 {
301 	struct pmd_internals *internals = dev->data->dev_private;
302 
303 	dev_info->if_index = internals->if_index;
304 	dev_info->max_mac_addrs = 1;
305 	dev_info->max_rx_pktlen = (uint32_t)ETH_FRAME_LEN;
306 	dev_info->max_rx_queues = (uint16_t)internals->nb_queues;
307 	dev_info->max_tx_queues = (uint16_t)internals->nb_queues;
308 	dev_info->min_rx_bufsize = 0;
309 }
310 
311 static int
312 eth_stats_get(struct rte_eth_dev *dev, struct rte_eth_stats *igb_stats)
313 {
314 	unsigned i, imax;
315 	unsigned long rx_total = 0, tx_total = 0, tx_err_total = 0;
316 	unsigned long rx_bytes_total = 0, tx_bytes_total = 0;
317 	const struct pmd_internals *internal = dev->data->dev_private;
318 
319 	imax = (internal->nb_queues < RTE_ETHDEV_QUEUE_STAT_CNTRS ?
320 	        internal->nb_queues : RTE_ETHDEV_QUEUE_STAT_CNTRS);
321 	for (i = 0; i < imax; i++) {
322 		igb_stats->q_ipackets[i] = internal->rx_queue[i].rx_pkts;
323 		igb_stats->q_ibytes[i] = internal->rx_queue[i].rx_bytes;
324 		rx_total += igb_stats->q_ipackets[i];
325 		rx_bytes_total += igb_stats->q_ibytes[i];
326 	}
327 
328 	imax = (internal->nb_queues < RTE_ETHDEV_QUEUE_STAT_CNTRS ?
329 	        internal->nb_queues : RTE_ETHDEV_QUEUE_STAT_CNTRS);
330 	for (i = 0; i < imax; i++) {
331 		igb_stats->q_opackets[i] = internal->tx_queue[i].tx_pkts;
332 		igb_stats->q_errors[i] = internal->tx_queue[i].err_pkts;
333 		igb_stats->q_obytes[i] = internal->tx_queue[i].tx_bytes;
334 		tx_total += igb_stats->q_opackets[i];
335 		tx_err_total += igb_stats->q_errors[i];
336 		tx_bytes_total += igb_stats->q_obytes[i];
337 	}
338 
339 	igb_stats->ipackets = rx_total;
340 	igb_stats->ibytes = rx_bytes_total;
341 	igb_stats->opackets = tx_total;
342 	igb_stats->oerrors = tx_err_total;
343 	igb_stats->obytes = tx_bytes_total;
344 	return 0;
345 }
346 
347 static void
348 eth_stats_reset(struct rte_eth_dev *dev)
349 {
350 	unsigned i;
351 	struct pmd_internals *internal = dev->data->dev_private;
352 
353 	for (i = 0; i < internal->nb_queues; i++) {
354 		internal->rx_queue[i].rx_pkts = 0;
355 		internal->rx_queue[i].rx_bytes = 0;
356 	}
357 
358 	for (i = 0; i < internal->nb_queues; i++) {
359 		internal->tx_queue[i].tx_pkts = 0;
360 		internal->tx_queue[i].err_pkts = 0;
361 		internal->tx_queue[i].tx_bytes = 0;
362 	}
363 }
364 
365 static void
366 eth_dev_close(struct rte_eth_dev *dev __rte_unused)
367 {
368 }
369 
370 static void
371 eth_queue_release(void *q __rte_unused)
372 {
373 }
374 
375 static int
376 eth_link_update(struct rte_eth_dev *dev __rte_unused,
377                 int wait_to_complete __rte_unused)
378 {
379 	return 0;
380 }
381 
382 static int
383 eth_rx_queue_setup(struct rte_eth_dev *dev,
384                    uint16_t rx_queue_id,
385                    uint16_t nb_rx_desc __rte_unused,
386                    unsigned int socket_id __rte_unused,
387                    const struct rte_eth_rxconf *rx_conf __rte_unused,
388                    struct rte_mempool *mb_pool)
389 {
390 	struct pmd_internals *internals = dev->data->dev_private;
391 	struct pkt_rx_queue *pkt_q = &internals->rx_queue[rx_queue_id];
392 	unsigned int buf_size, data_size;
393 
394 	pkt_q->mb_pool = mb_pool;
395 
396 	/* Now get the space available for data in the mbuf */
397 	buf_size = rte_pktmbuf_data_room_size(pkt_q->mb_pool) -
398 		RTE_PKTMBUF_HEADROOM;
399 	data_size = internals->req.tp_frame_size;
400 	data_size -= TPACKET2_HDRLEN - sizeof(struct sockaddr_ll);
401 
402 	if (data_size > buf_size) {
403 		PMD_LOG(ERR,
404 			"%s: %d bytes will not fit in mbuf (%d bytes)",
405 			dev->device->name, data_size, buf_size);
406 		return -ENOMEM;
407 	}
408 
409 	dev->data->rx_queues[rx_queue_id] = pkt_q;
410 	pkt_q->in_port = dev->data->port_id;
411 
412 	return 0;
413 }
414 
415 static int
416 eth_tx_queue_setup(struct rte_eth_dev *dev,
417                    uint16_t tx_queue_id,
418                    uint16_t nb_tx_desc __rte_unused,
419                    unsigned int socket_id __rte_unused,
420                    const struct rte_eth_txconf *tx_conf __rte_unused)
421 {
422 
423 	struct pmd_internals *internals = dev->data->dev_private;
424 
425 	dev->data->tx_queues[tx_queue_id] = &internals->tx_queue[tx_queue_id];
426 	return 0;
427 }
428 
429 static int
430 eth_dev_mtu_set(struct rte_eth_dev *dev, uint16_t mtu)
431 {
432 	struct pmd_internals *internals = dev->data->dev_private;
433 	struct ifreq ifr = { .ifr_mtu = mtu };
434 	int ret;
435 	int s;
436 	unsigned int data_size = internals->req.tp_frame_size -
437 				 TPACKET2_HDRLEN;
438 
439 	if (mtu > data_size)
440 		return -EINVAL;
441 
442 	s = socket(PF_INET, SOCK_DGRAM, 0);
443 	if (s < 0)
444 		return -EINVAL;
445 
446 	strlcpy(ifr.ifr_name, internals->if_name, IFNAMSIZ);
447 	ret = ioctl(s, SIOCSIFMTU, &ifr);
448 	close(s);
449 
450 	if (ret < 0)
451 		return -EINVAL;
452 
453 	return 0;
454 }
455 
456 static void
457 eth_dev_change_flags(char *if_name, uint32_t flags, uint32_t mask)
458 {
459 	struct ifreq ifr;
460 	int s;
461 
462 	s = socket(PF_INET, SOCK_DGRAM, 0);
463 	if (s < 0)
464 		return;
465 
466 	strlcpy(ifr.ifr_name, if_name, IFNAMSIZ);
467 	if (ioctl(s, SIOCGIFFLAGS, &ifr) < 0)
468 		goto out;
469 	ifr.ifr_flags &= mask;
470 	ifr.ifr_flags |= flags;
471 	if (ioctl(s, SIOCSIFFLAGS, &ifr) < 0)
472 		goto out;
473 out:
474 	close(s);
475 }
476 
477 static void
478 eth_dev_promiscuous_enable(struct rte_eth_dev *dev)
479 {
480 	struct pmd_internals *internals = dev->data->dev_private;
481 
482 	eth_dev_change_flags(internals->if_name, IFF_PROMISC, ~0);
483 }
484 
485 static void
486 eth_dev_promiscuous_disable(struct rte_eth_dev *dev)
487 {
488 	struct pmd_internals *internals = dev->data->dev_private;
489 
490 	eth_dev_change_flags(internals->if_name, 0, ~IFF_PROMISC);
491 }
492 
493 static const struct eth_dev_ops ops = {
494 	.dev_start = eth_dev_start,
495 	.dev_stop = eth_dev_stop,
496 	.dev_close = eth_dev_close,
497 	.dev_configure = eth_dev_configure,
498 	.dev_infos_get = eth_dev_info,
499 	.mtu_set = eth_dev_mtu_set,
500 	.promiscuous_enable = eth_dev_promiscuous_enable,
501 	.promiscuous_disable = eth_dev_promiscuous_disable,
502 	.rx_queue_setup = eth_rx_queue_setup,
503 	.tx_queue_setup = eth_tx_queue_setup,
504 	.rx_queue_release = eth_queue_release,
505 	.tx_queue_release = eth_queue_release,
506 	.link_update = eth_link_update,
507 	.stats_get = eth_stats_get,
508 	.stats_reset = eth_stats_reset,
509 };
510 
511 /*
512  * Opens an AF_PACKET socket
513  */
514 static int
515 open_packet_iface(const char *key __rte_unused,
516                   const char *value __rte_unused,
517                   void *extra_args)
518 {
519 	int *sockfd = extra_args;
520 
521 	/* Open an AF_PACKET socket... */
522 	*sockfd = socket(AF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
523 	if (*sockfd == -1) {
524 		PMD_LOG(ERR, "Could not open AF_PACKET socket");
525 		return -1;
526 	}
527 
528 	return 0;
529 }
530 
531 static struct rte_vdev_driver pmd_af_packet_drv;
532 
533 static int
534 rte_pmd_init_internals(struct rte_vdev_device *dev,
535                        const int sockfd,
536                        const unsigned nb_queues,
537                        unsigned int blocksize,
538                        unsigned int blockcnt,
539                        unsigned int framesize,
540                        unsigned int framecnt,
541 		       unsigned int qdisc_bypass,
542                        struct pmd_internals **internals,
543                        struct rte_eth_dev **eth_dev,
544                        struct rte_kvargs *kvlist)
545 {
546 	const char *name = rte_vdev_device_name(dev);
547 	const unsigned int numa_node = dev->device.numa_node;
548 	struct rte_eth_dev_data *data = NULL;
549 	struct rte_kvargs_pair *pair = NULL;
550 	struct ifreq ifr;
551 	size_t ifnamelen;
552 	unsigned k_idx;
553 	struct sockaddr_ll sockaddr;
554 	struct tpacket_req *req;
555 	struct pkt_rx_queue *rx_queue;
556 	struct pkt_tx_queue *tx_queue;
557 	int rc, tpver, discard;
558 	int qsockfd = -1;
559 	unsigned int i, q, rdsize;
560 #if defined(PACKET_FANOUT)
561 	int fanout_arg;
562 #endif
563 
564 	for (k_idx = 0; k_idx < kvlist->count; k_idx++) {
565 		pair = &kvlist->pairs[k_idx];
566 		if (strstr(pair->key, ETH_AF_PACKET_IFACE_ARG) != NULL)
567 			break;
568 	}
569 	if (pair == NULL) {
570 		PMD_LOG(ERR,
571 			"%s: no interface specified for AF_PACKET ethdev",
572 		        name);
573 		return -1;
574 	}
575 
576 	PMD_LOG(INFO,
577 		"%s: creating AF_PACKET-backed ethdev on numa socket %u",
578 		name, numa_node);
579 
580 	*internals = rte_zmalloc_socket(name, sizeof(**internals),
581 	                                0, numa_node);
582 	if (*internals == NULL)
583 		return -1;
584 
585 	for (q = 0; q < nb_queues; q++) {
586 		(*internals)->rx_queue[q].map = MAP_FAILED;
587 		(*internals)->tx_queue[q].map = MAP_FAILED;
588 	}
589 
590 	req = &((*internals)->req);
591 
592 	req->tp_block_size = blocksize;
593 	req->tp_block_nr = blockcnt;
594 	req->tp_frame_size = framesize;
595 	req->tp_frame_nr = framecnt;
596 
597 	ifnamelen = strlen(pair->value);
598 	if (ifnamelen < sizeof(ifr.ifr_name)) {
599 		memcpy(ifr.ifr_name, pair->value, ifnamelen);
600 		ifr.ifr_name[ifnamelen] = '\0';
601 	} else {
602 		PMD_LOG(ERR,
603 			"%s: I/F name too long (%s)",
604 			name, pair->value);
605 		return -1;
606 	}
607 	if (ioctl(sockfd, SIOCGIFINDEX, &ifr) == -1) {
608 		PMD_LOG(ERR,
609 			"%s: ioctl failed (SIOCGIFINDEX)",
610 		        name);
611 		return -1;
612 	}
613 	(*internals)->if_name = strdup(pair->value);
614 	if ((*internals)->if_name == NULL)
615 		return -1;
616 	(*internals)->if_index = ifr.ifr_ifindex;
617 
618 	if (ioctl(sockfd, SIOCGIFHWADDR, &ifr) == -1) {
619 		PMD_LOG(ERR,
620 			"%s: ioctl failed (SIOCGIFHWADDR)",
621 		        name);
622 		return -1;
623 	}
624 	memcpy(&(*internals)->eth_addr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
625 
626 	memset(&sockaddr, 0, sizeof(sockaddr));
627 	sockaddr.sll_family = AF_PACKET;
628 	sockaddr.sll_protocol = htons(ETH_P_ALL);
629 	sockaddr.sll_ifindex = (*internals)->if_index;
630 
631 #if defined(PACKET_FANOUT)
632 	fanout_arg = (getpid() ^ (*internals)->if_index) & 0xffff;
633 	fanout_arg |= (PACKET_FANOUT_HASH | PACKET_FANOUT_FLAG_DEFRAG) << 16;
634 #if defined(PACKET_FANOUT_FLAG_ROLLOVER)
635 	fanout_arg |= PACKET_FANOUT_FLAG_ROLLOVER << 16;
636 #endif
637 #endif
638 
639 	for (q = 0; q < nb_queues; q++) {
640 		/* Open an AF_PACKET socket for this queue... */
641 		qsockfd = socket(AF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
642 		if (qsockfd == -1) {
643 			PMD_LOG(ERR,
644 				"%s: could not open AF_PACKET socket",
645 			        name);
646 			return -1;
647 		}
648 
649 		tpver = TPACKET_V2;
650 		rc = setsockopt(qsockfd, SOL_PACKET, PACKET_VERSION,
651 				&tpver, sizeof(tpver));
652 		if (rc == -1) {
653 			PMD_LOG(ERR,
654 				"%s: could not set PACKET_VERSION on AF_PACKET socket for %s",
655 				name, pair->value);
656 			goto error;
657 		}
658 
659 		discard = 1;
660 		rc = setsockopt(qsockfd, SOL_PACKET, PACKET_LOSS,
661 				&discard, sizeof(discard));
662 		if (rc == -1) {
663 			PMD_LOG(ERR,
664 				"%s: could not set PACKET_LOSS on AF_PACKET socket for %s",
665 				name, pair->value);
666 			goto error;
667 		}
668 
669 #if defined(PACKET_QDISC_BYPASS)
670 		rc = setsockopt(qsockfd, SOL_PACKET, PACKET_QDISC_BYPASS,
671 				&qdisc_bypass, sizeof(qdisc_bypass));
672 		if (rc == -1) {
673 			PMD_LOG(ERR,
674 				"%s: could not set PACKET_QDISC_BYPASS on AF_PACKET socket for %s",
675 				name, pair->value);
676 			goto error;
677 		}
678 #else
679 		RTE_SET_USED(qdisc_bypass);
680 #endif
681 
682 		rc = setsockopt(qsockfd, SOL_PACKET, PACKET_RX_RING, req, sizeof(*req));
683 		if (rc == -1) {
684 			PMD_LOG(ERR,
685 				"%s: could not set PACKET_RX_RING on AF_PACKET socket for %s",
686 				name, pair->value);
687 			goto error;
688 		}
689 
690 		rc = setsockopt(qsockfd, SOL_PACKET, PACKET_TX_RING, req, sizeof(*req));
691 		if (rc == -1) {
692 			PMD_LOG(ERR,
693 				"%s: could not set PACKET_TX_RING on AF_PACKET "
694 				"socket for %s", name, pair->value);
695 			goto error;
696 		}
697 
698 		rx_queue = &((*internals)->rx_queue[q]);
699 		rx_queue->framecount = req->tp_frame_nr;
700 
701 		rx_queue->map = mmap(NULL, 2 * req->tp_block_size * req->tp_block_nr,
702 				    PROT_READ | PROT_WRITE, MAP_SHARED | MAP_LOCKED,
703 				    qsockfd, 0);
704 		if (rx_queue->map == MAP_FAILED) {
705 			PMD_LOG(ERR,
706 				"%s: call to mmap failed on AF_PACKET socket for %s",
707 				name, pair->value);
708 			goto error;
709 		}
710 
711 		/* rdsize is same for both Tx and Rx */
712 		rdsize = req->tp_frame_nr * sizeof(*(rx_queue->rd));
713 
714 		rx_queue->rd = rte_zmalloc_socket(name, rdsize, 0, numa_node);
715 		if (rx_queue->rd == NULL)
716 			goto error;
717 		for (i = 0; i < req->tp_frame_nr; ++i) {
718 			rx_queue->rd[i].iov_base = rx_queue->map + (i * framesize);
719 			rx_queue->rd[i].iov_len = req->tp_frame_size;
720 		}
721 		rx_queue->sockfd = qsockfd;
722 
723 		tx_queue = &((*internals)->tx_queue[q]);
724 		tx_queue->framecount = req->tp_frame_nr;
725 		tx_queue->frame_data_size = req->tp_frame_size;
726 		tx_queue->frame_data_size -= TPACKET2_HDRLEN -
727 			sizeof(struct sockaddr_ll);
728 
729 		tx_queue->map = rx_queue->map + req->tp_block_size * req->tp_block_nr;
730 
731 		tx_queue->rd = rte_zmalloc_socket(name, rdsize, 0, numa_node);
732 		if (tx_queue->rd == NULL)
733 			goto error;
734 		for (i = 0; i < req->tp_frame_nr; ++i) {
735 			tx_queue->rd[i].iov_base = tx_queue->map + (i * framesize);
736 			tx_queue->rd[i].iov_len = req->tp_frame_size;
737 		}
738 		tx_queue->sockfd = qsockfd;
739 
740 		rc = bind(qsockfd, (const struct sockaddr*)&sockaddr, sizeof(sockaddr));
741 		if (rc == -1) {
742 			PMD_LOG(ERR,
743 				"%s: could not bind AF_PACKET socket to %s",
744 			        name, pair->value);
745 			goto error;
746 		}
747 
748 #if defined(PACKET_FANOUT)
749 		rc = setsockopt(qsockfd, SOL_PACKET, PACKET_FANOUT,
750 				&fanout_arg, sizeof(fanout_arg));
751 		if (rc == -1) {
752 			PMD_LOG(ERR,
753 				"%s: could not set PACKET_FANOUT on AF_PACKET socket "
754 				"for %s", name, pair->value);
755 			goto error;
756 		}
757 #endif
758 	}
759 
760 	/* reserve an ethdev entry */
761 	*eth_dev = rte_eth_vdev_allocate(dev, 0);
762 	if (*eth_dev == NULL)
763 		goto error;
764 
765 	/*
766 	 * now put it all together
767 	 * - store queue data in internals,
768 	 * - store numa_node in eth_dev
769 	 * - point eth_dev_data to internals
770 	 * - and point eth_dev structure to new eth_dev_data structure
771 	 */
772 
773 	(*internals)->nb_queues = nb_queues;
774 
775 	data = (*eth_dev)->data;
776 	data->dev_private = *internals;
777 	data->nb_rx_queues = (uint16_t)nb_queues;
778 	data->nb_tx_queues = (uint16_t)nb_queues;
779 	data->dev_link = pmd_link;
780 	data->mac_addrs = &(*internals)->eth_addr;
781 
782 	(*eth_dev)->dev_ops = &ops;
783 
784 	return 0;
785 
786 error:
787 	if (qsockfd != -1)
788 		close(qsockfd);
789 	for (q = 0; q < nb_queues; q++) {
790 		munmap((*internals)->rx_queue[q].map,
791 		       2 * req->tp_block_size * req->tp_block_nr);
792 
793 		rte_free((*internals)->rx_queue[q].rd);
794 		rte_free((*internals)->tx_queue[q].rd);
795 		if (((*internals)->rx_queue[q].sockfd != 0) &&
796 			((*internals)->rx_queue[q].sockfd != qsockfd))
797 			close((*internals)->rx_queue[q].sockfd);
798 	}
799 	free((*internals)->if_name);
800 	rte_free(*internals);
801 	return -1;
802 }
803 
804 static int
805 rte_eth_from_packet(struct rte_vdev_device *dev,
806                     int const *sockfd,
807                     struct rte_kvargs *kvlist)
808 {
809 	const char *name = rte_vdev_device_name(dev);
810 	struct pmd_internals *internals = NULL;
811 	struct rte_eth_dev *eth_dev = NULL;
812 	struct rte_kvargs_pair *pair = NULL;
813 	unsigned k_idx;
814 	unsigned int blockcount;
815 	unsigned int blocksize = DFLT_BLOCK_SIZE;
816 	unsigned int framesize = DFLT_FRAME_SIZE;
817 	unsigned int framecount = DFLT_FRAME_COUNT;
818 	unsigned int qpairs = 1;
819 	unsigned int qdisc_bypass = 1;
820 
821 	/* do some parameter checking */
822 	if (*sockfd < 0)
823 		return -1;
824 
825 	/*
826 	 * Walk arguments for configurable settings
827 	 */
828 	for (k_idx = 0; k_idx < kvlist->count; k_idx++) {
829 		pair = &kvlist->pairs[k_idx];
830 		if (strstr(pair->key, ETH_AF_PACKET_NUM_Q_ARG) != NULL) {
831 			qpairs = atoi(pair->value);
832 			if (qpairs < 1 ||
833 			    qpairs > RTE_PMD_AF_PACKET_MAX_RINGS) {
834 				PMD_LOG(ERR,
835 					"%s: invalid qpairs value",
836 				        name);
837 				return -1;
838 			}
839 			continue;
840 		}
841 		if (strstr(pair->key, ETH_AF_PACKET_BLOCKSIZE_ARG) != NULL) {
842 			blocksize = atoi(pair->value);
843 			if (!blocksize) {
844 				PMD_LOG(ERR,
845 					"%s: invalid blocksize value",
846 				        name);
847 				return -1;
848 			}
849 			continue;
850 		}
851 		if (strstr(pair->key, ETH_AF_PACKET_FRAMESIZE_ARG) != NULL) {
852 			framesize = atoi(pair->value);
853 			if (!framesize) {
854 				PMD_LOG(ERR,
855 					"%s: invalid framesize value",
856 				        name);
857 				return -1;
858 			}
859 			continue;
860 		}
861 		if (strstr(pair->key, ETH_AF_PACKET_FRAMECOUNT_ARG) != NULL) {
862 			framecount = atoi(pair->value);
863 			if (!framecount) {
864 				PMD_LOG(ERR,
865 					"%s: invalid framecount value",
866 				        name);
867 				return -1;
868 			}
869 			continue;
870 		}
871 		if (strstr(pair->key, ETH_AF_PACKET_QDISC_BYPASS_ARG) != NULL) {
872 			qdisc_bypass = atoi(pair->value);
873 			if (qdisc_bypass > 1) {
874 				PMD_LOG(ERR,
875 					"%s: invalid bypass value",
876 					name);
877 				return -1;
878 			}
879 			continue;
880 		}
881 	}
882 
883 	if (framesize > blocksize) {
884 		PMD_LOG(ERR,
885 			"%s: AF_PACKET MMAP frame size exceeds block size!",
886 		        name);
887 		return -1;
888 	}
889 
890 	blockcount = framecount / (blocksize / framesize);
891 	if (!blockcount) {
892 		PMD_LOG(ERR,
893 			"%s: invalid AF_PACKET MMAP parameters", name);
894 		return -1;
895 	}
896 
897 	PMD_LOG(INFO, "%s: AF_PACKET MMAP parameters:", name);
898 	PMD_LOG(INFO, "%s:\tblock size %d", name, blocksize);
899 	PMD_LOG(INFO, "%s:\tblock count %d", name, blockcount);
900 	PMD_LOG(INFO, "%s:\tframe size %d", name, framesize);
901 	PMD_LOG(INFO, "%s:\tframe count %d", name, framecount);
902 
903 	if (rte_pmd_init_internals(dev, *sockfd, qpairs,
904 				   blocksize, blockcount,
905 				   framesize, framecount,
906 				   qdisc_bypass,
907 				   &internals, &eth_dev,
908 				   kvlist) < 0)
909 		return -1;
910 
911 	eth_dev->rx_pkt_burst = eth_af_packet_rx;
912 	eth_dev->tx_pkt_burst = eth_af_packet_tx;
913 
914 	rte_eth_dev_probing_finish(eth_dev);
915 	return 0;
916 }
917 
918 static int
919 rte_pmd_af_packet_probe(struct rte_vdev_device *dev)
920 {
921 	int ret = 0;
922 	struct rte_kvargs *kvlist;
923 	int sockfd = -1;
924 	struct rte_eth_dev *eth_dev;
925 	const char *name = rte_vdev_device_name(dev);
926 
927 	PMD_LOG(INFO, "Initializing pmd_af_packet for %s", name);
928 
929 	if (rte_eal_process_type() == RTE_PROC_SECONDARY) {
930 		eth_dev = rte_eth_dev_attach_secondary(name);
931 		if (!eth_dev) {
932 			PMD_LOG(ERR, "Failed to probe %s", name);
933 			return -1;
934 		}
935 		/* TODO: request info from primary to set up Rx and Tx */
936 		eth_dev->dev_ops = &ops;
937 		eth_dev->device = &dev->device;
938 		rte_eth_dev_probing_finish(eth_dev);
939 		return 0;
940 	}
941 
942 	kvlist = rte_kvargs_parse(rte_vdev_device_args(dev), valid_arguments);
943 	if (kvlist == NULL) {
944 		ret = -1;
945 		goto exit;
946 	}
947 
948 	/*
949 	 * If iface argument is passed we open the NICs and use them for
950 	 * reading / writing
951 	 */
952 	if (rte_kvargs_count(kvlist, ETH_AF_PACKET_IFACE_ARG) == 1) {
953 
954 		ret = rte_kvargs_process(kvlist, ETH_AF_PACKET_IFACE_ARG,
955 		                         &open_packet_iface, &sockfd);
956 		if (ret < 0)
957 			goto exit;
958 	}
959 
960 	if (dev->device.numa_node == SOCKET_ID_ANY)
961 		dev->device.numa_node = rte_socket_id();
962 
963 	ret = rte_eth_from_packet(dev, &sockfd, kvlist);
964 	close(sockfd); /* no longer needed */
965 
966 exit:
967 	rte_kvargs_free(kvlist);
968 	return ret;
969 }
970 
971 static int
972 rte_pmd_af_packet_remove(struct rte_vdev_device *dev)
973 {
974 	struct rte_eth_dev *eth_dev = NULL;
975 	struct pmd_internals *internals;
976 	unsigned q;
977 
978 	PMD_LOG(INFO, "Closing AF_PACKET ethdev on numa socket %u",
979 		rte_socket_id());
980 
981 	if (dev == NULL)
982 		return -1;
983 
984 	/* find the ethdev entry */
985 	eth_dev = rte_eth_dev_allocated(rte_vdev_device_name(dev));
986 	if (eth_dev == NULL)
987 		return -1;
988 
989 	/* mac_addrs must not be freed alone because part of dev_private */
990 	eth_dev->data->mac_addrs = NULL;
991 
992 	if (rte_eal_process_type() != RTE_PROC_PRIMARY)
993 		return rte_eth_dev_release_port(eth_dev);
994 
995 	internals = eth_dev->data->dev_private;
996 	for (q = 0; q < internals->nb_queues; q++) {
997 		rte_free(internals->rx_queue[q].rd);
998 		rte_free(internals->tx_queue[q].rd);
999 	}
1000 	free(internals->if_name);
1001 
1002 	rte_eth_dev_release_port(eth_dev);
1003 
1004 	return 0;
1005 }
1006 
1007 static struct rte_vdev_driver pmd_af_packet_drv = {
1008 	.probe = rte_pmd_af_packet_probe,
1009 	.remove = rte_pmd_af_packet_remove,
1010 };
1011 
1012 RTE_PMD_REGISTER_VDEV(net_af_packet, pmd_af_packet_drv);
1013 RTE_PMD_REGISTER_ALIAS(net_af_packet, eth_af_packet);
1014 RTE_PMD_REGISTER_PARAM_STRING(net_af_packet,
1015 	"iface=<string> "
1016 	"qpairs=<int> "
1017 	"blocksz=<int> "
1018 	"framesz=<int> "
1019 	"framecnt=<int> "
1020 	"qdisc_bypass=<0|1>");
1021 
1022 RTE_INIT(af_packet_init_log)
1023 {
1024 	af_packet_logtype = rte_log_register("pmd.net.packet");
1025 	if (af_packet_logtype >= 0)
1026 		rte_log_set_level(af_packet_logtype, RTE_LOG_NOTICE);
1027 }
1028