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