d749b5b16c5f71f711f2f07469c99c5d7e6b27ed
[dpdk.git] / drivers / net / memif / rte_eth_memif.c
1 /* SPDX-License-Identifier: BSD-3-Clause
2  * Copyright 2018-2019 Cisco Systems, Inc.  All rights reserved.
3  */
4
5 #include <stdlib.h>
6 #include <fcntl.h>
7 #include <unistd.h>
8 #include <sys/types.h>
9 #include <sys/socket.h>
10 #include <sys/un.h>
11 #include <sys/ioctl.h>
12 #include <sys/mman.h>
13 #include <linux/if_ether.h>
14 #include <errno.h>
15 #include <sys/eventfd.h>
16
17 #include <rte_version.h>
18 #include <rte_mbuf.h>
19 #include <rte_ether.h>
20 #include <rte_ethdev_driver.h>
21 #include <rte_ethdev_vdev.h>
22 #include <rte_malloc.h>
23 #include <rte_kvargs.h>
24 #include <rte_bus_vdev.h>
25 #include <rte_string_fns.h>
26 #include <rte_errno.h>
27 #include <rte_memory.h>
28 #include <rte_memzone.h>
29 #include <rte_eal_memconfig.h>
30
31 #include "rte_eth_memif.h"
32 #include "memif_socket.h"
33
34 #define ETH_MEMIF_ID_ARG                "id"
35 #define ETH_MEMIF_ROLE_ARG              "role"
36 #define ETH_MEMIF_PKT_BUFFER_SIZE_ARG   "bsize"
37 #define ETH_MEMIF_RING_SIZE_ARG         "rsize"
38 #define ETH_MEMIF_SOCKET_ARG            "socket"
39 #define ETH_MEMIF_MAC_ARG               "mac"
40 #define ETH_MEMIF_ZC_ARG                "zero-copy"
41 #define ETH_MEMIF_SECRET_ARG            "secret"
42
43 static const char * const valid_arguments[] = {
44         ETH_MEMIF_ID_ARG,
45         ETH_MEMIF_ROLE_ARG,
46         ETH_MEMIF_PKT_BUFFER_SIZE_ARG,
47         ETH_MEMIF_RING_SIZE_ARG,
48         ETH_MEMIF_SOCKET_ARG,
49         ETH_MEMIF_MAC_ARG,
50         ETH_MEMIF_ZC_ARG,
51         ETH_MEMIF_SECRET_ARG,
52         NULL
53 };
54
55 static const struct rte_eth_link pmd_link = {
56         .link_speed = ETH_SPEED_NUM_10G,
57         .link_duplex = ETH_LINK_FULL_DUPLEX,
58         .link_status = ETH_LINK_DOWN,
59         .link_autoneg = ETH_LINK_AUTONEG
60 };
61
62 #define MEMIF_MP_SEND_REGION            "memif_mp_send_region"
63
64
65 static int memif_region_init_zc(const struct rte_memseg_list *msl,
66                                 const struct rte_memseg *ms, void *arg);
67
68 const char *
69 memif_version(void)
70 {
71         return ("memif-" RTE_STR(MEMIF_VERSION_MAJOR) "." RTE_STR(MEMIF_VERSION_MINOR));
72 }
73
74 /* Message header to synchronize regions */
75 struct mp_region_msg {
76         char port_name[RTE_DEV_NAME_MAX_LEN];
77         memif_region_index_t idx;
78         memif_region_size_t size;
79 };
80
81 static int
82 memif_mp_send_region(const struct rte_mp_msg *msg, const void *peer)
83 {
84         struct rte_eth_dev *dev;
85         struct pmd_process_private *proc_private;
86         const struct mp_region_msg *msg_param = (const struct mp_region_msg *)msg->param;
87         struct rte_mp_msg reply;
88         struct mp_region_msg *reply_param = (struct mp_region_msg *)reply.param;
89         uint16_t port_id;
90         int ret;
91
92         /* Get requested port */
93         ret = rte_eth_dev_get_port_by_name(msg_param->port_name, &port_id);
94         if (ret) {
95                 MIF_LOG(ERR, "Failed to get port id for %s",
96                         msg_param->port_name);
97                 return -1;
98         }
99         dev = &rte_eth_devices[port_id];
100         proc_private = dev->process_private;
101
102         memset(&reply, 0, sizeof(reply));
103         strlcpy(reply.name, msg->name, sizeof(reply.name));
104         reply_param->idx = msg_param->idx;
105         if (proc_private->regions[msg_param->idx] != NULL) {
106                 reply_param->size = proc_private->regions[msg_param->idx]->region_size;
107                 reply.fds[0] = proc_private->regions[msg_param->idx]->fd;
108                 reply.num_fds = 1;
109         }
110         reply.len_param = sizeof(*reply_param);
111         if (rte_mp_reply(&reply, peer) < 0) {
112                 MIF_LOG(ERR, "Failed to reply to an add region request");
113                 return -1;
114         }
115
116         return 0;
117 }
118
119 /*
120  * Request regions
121  * Called by secondary process, when ports link status goes up.
122  */
123 static int
124 memif_mp_request_regions(struct rte_eth_dev *dev)
125 {
126         int ret, i;
127         struct timespec timeout = {.tv_sec = 5, .tv_nsec = 0};
128         struct rte_mp_msg msg, *reply;
129         struct rte_mp_reply replies;
130         struct mp_region_msg *msg_param = (struct mp_region_msg *)msg.param;
131         struct mp_region_msg *reply_param;
132         struct memif_region *r;
133         struct pmd_process_private *proc_private = dev->process_private;
134         struct pmd_internals *pmd = dev->data->dev_private;
135         /* in case of zero-copy slave, only request region 0 */
136         uint16_t max_region_num = (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY) ?
137                                    1 : ETH_MEMIF_MAX_REGION_NUM;
138
139         MIF_LOG(DEBUG, "Requesting memory regions");
140
141         for (i = 0; i < max_region_num; i++) {
142                 /* Prepare the message */
143                 memset(&msg, 0, sizeof(msg));
144                 strlcpy(msg.name, MEMIF_MP_SEND_REGION, sizeof(msg.name));
145                 strlcpy(msg_param->port_name, dev->data->name,
146                         sizeof(msg_param->port_name));
147                 msg_param->idx = i;
148                 msg.len_param = sizeof(*msg_param);
149
150                 /* Send message */
151                 ret = rte_mp_request_sync(&msg, &replies, &timeout);
152                 if (ret < 0 || replies.nb_received != 1) {
153                         MIF_LOG(ERR, "Failed to send mp msg: %d",
154                                 rte_errno);
155                         return -1;
156                 }
157
158                 reply = &replies.msgs[0];
159                 reply_param = (struct mp_region_msg *)reply->param;
160
161                 if (reply_param->size > 0) {
162                         r = rte_zmalloc("region", sizeof(struct memif_region), 0);
163                         if (r == NULL) {
164                                 MIF_LOG(ERR, "Failed to alloc memif region.");
165                                 free(reply);
166                                 return -ENOMEM;
167                         }
168                         r->region_size = reply_param->size;
169                         if (reply->num_fds < 1) {
170                                 MIF_LOG(ERR, "Missing file descriptor.");
171                                 free(reply);
172                                 return -1;
173                         }
174                         r->fd = reply->fds[0];
175                         r->addr = NULL;
176
177                         proc_private->regions[reply_param->idx] = r;
178                         proc_private->regions_num++;
179                 }
180                 free(reply);
181         }
182
183         if (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY) {
184                 ret = rte_memseg_walk(memif_region_init_zc, (void *)proc_private);
185                 if (ret < 0)
186                         return ret;
187         }
188
189         return memif_connect(dev);
190 }
191
192 static int
193 memif_dev_info(struct rte_eth_dev *dev __rte_unused, struct rte_eth_dev_info *dev_info)
194 {
195         dev_info->max_mac_addrs = 1;
196         dev_info->max_rx_pktlen = (uint32_t)ETH_FRAME_LEN;
197         dev_info->max_rx_queues = ETH_MEMIF_MAX_NUM_Q_PAIRS;
198         dev_info->max_tx_queues = ETH_MEMIF_MAX_NUM_Q_PAIRS;
199         dev_info->min_rx_bufsize = 0;
200
201         return 0;
202 }
203
204 static memif_ring_t *
205 memif_get_ring(struct pmd_internals *pmd, struct pmd_process_private *proc_private,
206                memif_ring_type_t type, uint16_t ring_num)
207 {
208         /* rings only in region 0 */
209         void *p = proc_private->regions[0]->addr;
210         int ring_size = sizeof(memif_ring_t) + sizeof(memif_desc_t) *
211             (1 << pmd->run.log2_ring_size);
212
213         p = (uint8_t *)p + (ring_num + type * pmd->run.num_s2m_rings) * ring_size;
214
215         return (memif_ring_t *)p;
216 }
217
218 static memif_region_offset_t
219 memif_get_ring_offset(struct rte_eth_dev *dev, struct memif_queue *mq,
220                       memif_ring_type_t type, uint16_t num)
221 {
222         struct pmd_internals *pmd = dev->data->dev_private;
223         struct pmd_process_private *proc_private = dev->process_private;
224
225         return ((uint8_t *)memif_get_ring(pmd, proc_private, type, num) -
226                 (uint8_t *)proc_private->regions[mq->region]->addr);
227 }
228
229 static memif_ring_t *
230 memif_get_ring_from_queue(struct pmd_process_private *proc_private,
231                           struct memif_queue *mq)
232 {
233         struct memif_region *r;
234
235         r = proc_private->regions[mq->region];
236         if (r == NULL)
237                 return NULL;
238
239         return (memif_ring_t *)((uint8_t *)r->addr + mq->ring_offset);
240 }
241
242 static void *
243 memif_get_buffer(struct pmd_process_private *proc_private, memif_desc_t *d)
244 {
245         return ((uint8_t *)proc_private->regions[d->region]->addr + d->offset);
246 }
247
248 /* Free mbufs received by master */
249 static void
250 memif_free_stored_mbufs(struct pmd_process_private *proc_private, struct memif_queue *mq)
251 {
252         uint16_t mask = (1 << mq->log2_ring_size) - 1;
253         memif_ring_t *ring = memif_get_ring_from_queue(proc_private, mq);
254
255         /* FIXME: improve performance */
256         /* The ring->tail acts as a guard variable between Tx and Rx
257          * threads, so using load-acquire pairs with store-release
258          * to synchronize it between threads.
259          */
260         while (mq->last_tail != __atomic_load_n(&ring->tail,
261                                                 __ATOMIC_ACQUIRE)) {
262                 RTE_MBUF_PREFETCH_TO_FREE(mq->buffers[(mq->last_tail + 1) & mask]);
263                 /* Decrement refcnt and free mbuf. (current segment) */
264                 rte_mbuf_refcnt_update(mq->buffers[mq->last_tail & mask], -1);
265                 rte_pktmbuf_free_seg(mq->buffers[mq->last_tail & mask]);
266                 mq->last_tail++;
267         }
268 }
269
270 static int
271 memif_pktmbuf_chain(struct rte_mbuf *head, struct rte_mbuf *cur_tail,
272                     struct rte_mbuf *tail)
273 {
274         /* Check for number-of-segments-overflow */
275         if (unlikely(head->nb_segs + tail->nb_segs > RTE_MBUF_MAX_NB_SEGS))
276                 return -EOVERFLOW;
277
278         /* Chain 'tail' onto the old tail */
279         cur_tail->next = tail;
280
281         /* accumulate number of segments and total length. */
282         head->nb_segs = (uint16_t)(head->nb_segs + tail->nb_segs);
283
284         tail->pkt_len = tail->data_len;
285         head->pkt_len += tail->pkt_len;
286
287         return 0;
288 }
289
290 static uint16_t
291 eth_memif_rx(void *queue, struct rte_mbuf **bufs, uint16_t nb_pkts)
292 {
293         struct memif_queue *mq = queue;
294         struct pmd_internals *pmd = rte_eth_devices[mq->in_port].data->dev_private;
295         struct pmd_process_private *proc_private =
296                 rte_eth_devices[mq->in_port].process_private;
297         memif_ring_t *ring = memif_get_ring_from_queue(proc_private, mq);
298         uint16_t cur_slot, last_slot, n_slots, ring_size, mask, s0;
299         uint16_t n_rx_pkts = 0;
300         uint16_t mbuf_size = rte_pktmbuf_data_room_size(mq->mempool) -
301                 RTE_PKTMBUF_HEADROOM;
302         uint16_t src_len, src_off, dst_len, dst_off, cp_len;
303         memif_ring_type_t type = mq->type;
304         memif_desc_t *d0;
305         struct rte_mbuf *mbuf, *mbuf_head, *mbuf_tail;
306         uint64_t b;
307         ssize_t size __rte_unused;
308         uint16_t head;
309         int ret;
310         struct rte_eth_link link;
311
312         if (unlikely((pmd->flags & ETH_MEMIF_FLAG_CONNECTED) == 0))
313                 return 0;
314         if (unlikely(ring == NULL)) {
315                 /* Secondary process will attempt to request regions. */
316                 ret = rte_eth_link_get(mq->in_port, &link);
317                 if (ret < 0)
318                         MIF_LOG(ERR, "Failed to get port %u link info: %s",
319                                 mq->in_port, rte_strerror(-ret));
320                 return 0;
321         }
322
323         /* consume interrupt */
324         if ((ring->flags & MEMIF_RING_FLAG_MASK_INT) == 0)
325                 size = read(mq->intr_handle.fd, &b, sizeof(b));
326
327         ring_size = 1 << mq->log2_ring_size;
328         mask = ring_size - 1;
329
330         if (type == MEMIF_RING_S2M) {
331                 cur_slot = mq->last_head;
332                 last_slot = __atomic_load_n(&ring->head, __ATOMIC_ACQUIRE);
333         } else {
334                 cur_slot = mq->last_tail;
335                 last_slot = __atomic_load_n(&ring->tail, __ATOMIC_ACQUIRE);
336         }
337
338         if (cur_slot == last_slot)
339                 goto refill;
340         n_slots = last_slot - cur_slot;
341
342         while (n_slots && n_rx_pkts < nb_pkts) {
343                 mbuf_head = rte_pktmbuf_alloc(mq->mempool);
344                 if (unlikely(mbuf_head == NULL))
345                         goto no_free_bufs;
346                 mbuf = mbuf_head;
347                 mbuf->port = mq->in_port;
348
349 next_slot:
350                 s0 = cur_slot & mask;
351                 d0 = &ring->desc[s0];
352
353                 src_len = d0->length;
354                 dst_off = 0;
355                 src_off = 0;
356
357                 do {
358                         dst_len = mbuf_size - dst_off;
359                         if (dst_len == 0) {
360                                 dst_off = 0;
361                                 dst_len = mbuf_size;
362
363                                 /* store pointer to tail */
364                                 mbuf_tail = mbuf;
365                                 mbuf = rte_pktmbuf_alloc(mq->mempool);
366                                 if (unlikely(mbuf == NULL))
367                                         goto no_free_bufs;
368                                 mbuf->port = mq->in_port;
369                                 ret = memif_pktmbuf_chain(mbuf_head, mbuf_tail, mbuf);
370                                 if (unlikely(ret < 0)) {
371                                         MIF_LOG(ERR, "number-of-segments-overflow");
372                                         rte_pktmbuf_free(mbuf);
373                                         goto no_free_bufs;
374                                 }
375                         }
376                         cp_len = RTE_MIN(dst_len, src_len);
377
378                         rte_pktmbuf_data_len(mbuf) += cp_len;
379                         rte_pktmbuf_pkt_len(mbuf) = rte_pktmbuf_data_len(mbuf);
380                         if (mbuf != mbuf_head)
381                                 rte_pktmbuf_pkt_len(mbuf_head) += cp_len;
382
383                         memcpy(rte_pktmbuf_mtod_offset(mbuf, void *, dst_off),
384                                (uint8_t *)memif_get_buffer(proc_private, d0) + src_off,
385                                cp_len);
386
387                         src_off += cp_len;
388                         dst_off += cp_len;
389                         src_len -= cp_len;
390                 } while (src_len);
391
392                 cur_slot++;
393                 n_slots--;
394
395                 if (d0->flags & MEMIF_DESC_FLAG_NEXT)
396                         goto next_slot;
397
398                 mq->n_bytes += rte_pktmbuf_pkt_len(mbuf_head);
399                 *bufs++ = mbuf_head;
400                 n_rx_pkts++;
401         }
402
403 no_free_bufs:
404         if (type == MEMIF_RING_S2M) {
405                 __atomic_store_n(&ring->tail, cur_slot, __ATOMIC_RELEASE);
406                 mq->last_head = cur_slot;
407         } else {
408                 mq->last_tail = cur_slot;
409         }
410
411 refill:
412         if (type == MEMIF_RING_M2S) {
413                 head = __atomic_load_n(&ring->head, __ATOMIC_ACQUIRE);
414                 n_slots = ring_size - head + mq->last_tail;
415
416                 while (n_slots--) {
417                         s0 = head++ & mask;
418                         d0 = &ring->desc[s0];
419                         d0->length = pmd->run.pkt_buffer_size;
420                 }
421                 __atomic_store_n(&ring->head, head, __ATOMIC_RELEASE);
422         }
423
424         mq->n_pkts += n_rx_pkts;
425         return n_rx_pkts;
426 }
427
428 static uint16_t
429 eth_memif_rx_zc(void *queue, struct rte_mbuf **bufs, uint16_t nb_pkts)
430 {
431         struct memif_queue *mq = queue;
432         struct pmd_internals *pmd = rte_eth_devices[mq->in_port].data->dev_private;
433         struct pmd_process_private *proc_private =
434                 rte_eth_devices[mq->in_port].process_private;
435         memif_ring_t *ring = memif_get_ring_from_queue(proc_private, mq);
436         uint16_t cur_slot, last_slot, n_slots, ring_size, mask, s0, head;
437         uint16_t n_rx_pkts = 0;
438         memif_desc_t *d0;
439         struct rte_mbuf *mbuf, *mbuf_tail;
440         struct rte_mbuf *mbuf_head = NULL;
441         int ret;
442         struct rte_eth_link link;
443
444         if (unlikely((pmd->flags & ETH_MEMIF_FLAG_CONNECTED) == 0))
445                 return 0;
446         if (unlikely(ring == NULL)) {
447                 /* Secondary process will attempt to request regions. */
448                 rte_eth_link_get(mq->in_port, &link);
449                 return 0;
450         }
451
452         /* consume interrupt */
453         if ((ring->flags & MEMIF_RING_FLAG_MASK_INT) == 0) {
454                 uint64_t b;
455                 ssize_t size __rte_unused;
456                 size = read(mq->intr_handle.fd, &b, sizeof(b));
457         }
458
459         ring_size = 1 << mq->log2_ring_size;
460         mask = ring_size - 1;
461
462         cur_slot = mq->last_tail;
463         /* The ring->tail acts as a guard variable between Tx and Rx
464          * threads, so using load-acquire pairs with store-release
465          * to synchronize it between threads.
466          */
467         last_slot = __atomic_load_n(&ring->tail, __ATOMIC_ACQUIRE);
468         if (cur_slot == last_slot)
469                 goto refill;
470         n_slots = last_slot - cur_slot;
471
472         while (n_slots && n_rx_pkts < nb_pkts) {
473                 s0 = cur_slot & mask;
474
475                 d0 = &ring->desc[s0];
476                 mbuf_head = mq->buffers[s0];
477                 mbuf = mbuf_head;
478
479 next_slot:
480                 /* prefetch next descriptor */
481                 if (n_rx_pkts + 1 < nb_pkts)
482                         rte_prefetch0(&ring->desc[(cur_slot + 1) & mask]);
483
484                 mbuf->port = mq->in_port;
485                 rte_pktmbuf_data_len(mbuf) = d0->length;
486                 rte_pktmbuf_pkt_len(mbuf) = rte_pktmbuf_data_len(mbuf);
487
488                 mq->n_bytes += rte_pktmbuf_data_len(mbuf);
489
490                 cur_slot++;
491                 n_slots--;
492                 if (d0->flags & MEMIF_DESC_FLAG_NEXT) {
493                         s0 = cur_slot & mask;
494                         d0 = &ring->desc[s0];
495                         mbuf_tail = mbuf;
496                         mbuf = mq->buffers[s0];
497                         ret = memif_pktmbuf_chain(mbuf_head, mbuf_tail, mbuf);
498                         if (unlikely(ret < 0)) {
499                                 MIF_LOG(ERR, "number-of-segments-overflow");
500                                 goto refill;
501                         }
502                         goto next_slot;
503                 }
504
505                 *bufs++ = mbuf_head;
506                 n_rx_pkts++;
507         }
508
509         mq->last_tail = cur_slot;
510
511 /* Supply master with new buffers */
512 refill:
513         /* The ring->head acts as a guard variable between Tx and Rx
514          * threads, so using load-acquire pairs with store-release
515          * to synchronize it between threads.
516          */
517         head = __atomic_load_n(&ring->head, __ATOMIC_ACQUIRE);
518         n_slots = ring_size - head + mq->last_tail;
519
520         if (n_slots < 32)
521                 goto no_free_mbufs;
522
523         ret = rte_pktmbuf_alloc_bulk(mq->mempool, &mq->buffers[head & mask], n_slots);
524         if (unlikely(ret < 0))
525                 goto no_free_mbufs;
526
527         while (n_slots--) {
528                 s0 = head++ & mask;
529                 if (n_slots > 0)
530                         rte_prefetch0(mq->buffers[head & mask]);
531                 d0 = &ring->desc[s0];
532                 /* store buffer header */
533                 mbuf = mq->buffers[s0];
534                 /* populate descriptor */
535                 d0->length = rte_pktmbuf_data_room_size(mq->mempool) -
536                                 RTE_PKTMBUF_HEADROOM;
537                 d0->region = 1;
538                 d0->offset = rte_pktmbuf_mtod(mbuf, uint8_t *) -
539                         (uint8_t *)proc_private->regions[d0->region]->addr;
540         }
541 no_free_mbufs:
542         __atomic_store_n(&ring->head, head, __ATOMIC_RELEASE);
543
544         mq->n_pkts += n_rx_pkts;
545
546         return n_rx_pkts;
547 }
548
549 static uint16_t
550 eth_memif_tx(void *queue, struct rte_mbuf **bufs, uint16_t nb_pkts)
551 {
552         struct memif_queue *mq = queue;
553         struct pmd_internals *pmd = rte_eth_devices[mq->in_port].data->dev_private;
554         struct pmd_process_private *proc_private =
555                 rte_eth_devices[mq->in_port].process_private;
556         memif_ring_t *ring = memif_get_ring_from_queue(proc_private, mq);
557         uint16_t slot, saved_slot, n_free, ring_size, mask, n_tx_pkts = 0;
558         uint16_t src_len, src_off, dst_len, dst_off, cp_len;
559         memif_ring_type_t type = mq->type;
560         memif_desc_t *d0;
561         struct rte_mbuf *mbuf;
562         struct rte_mbuf *mbuf_head;
563         uint64_t a;
564         ssize_t size;
565         struct rte_eth_link link;
566
567         if (unlikely((pmd->flags & ETH_MEMIF_FLAG_CONNECTED) == 0))
568                 return 0;
569         if (unlikely(ring == NULL)) {
570                 int ret;
571
572                 /* Secondary process will attempt to request regions. */
573                 ret = rte_eth_link_get(mq->in_port, &link);
574                 if (ret < 0)
575                         MIF_LOG(ERR, "Failed to get port %u link info: %s",
576                                 mq->in_port, rte_strerror(-ret));
577                 return 0;
578         }
579
580         ring_size = 1 << mq->log2_ring_size;
581         mask = ring_size - 1;
582
583         if (type == MEMIF_RING_S2M) {
584                 slot = __atomic_load_n(&ring->head, __ATOMIC_ACQUIRE);
585                 n_free = ring_size - slot +
586                                 __atomic_load_n(&ring->tail, __ATOMIC_ACQUIRE);
587         } else {
588                 slot = __atomic_load_n(&ring->tail, __ATOMIC_ACQUIRE);
589                 n_free = __atomic_load_n(&ring->head, __ATOMIC_ACQUIRE) - slot;
590         }
591
592         while (n_tx_pkts < nb_pkts && n_free) {
593                 mbuf_head = *bufs++;
594                 mbuf = mbuf_head;
595
596                 saved_slot = slot;
597                 d0 = &ring->desc[slot & mask];
598                 dst_off = 0;
599                 dst_len = (type == MEMIF_RING_S2M) ?
600                         pmd->run.pkt_buffer_size : d0->length;
601
602 next_in_chain:
603                 src_off = 0;
604                 src_len = rte_pktmbuf_data_len(mbuf);
605
606                 while (src_len) {
607                         if (dst_len == 0) {
608                                 if (n_free) {
609                                         slot++;
610                                         n_free--;
611                                         d0->flags |= MEMIF_DESC_FLAG_NEXT;
612                                         d0 = &ring->desc[slot & mask];
613                                         dst_off = 0;
614                                         dst_len = (type == MEMIF_RING_S2M) ?
615                                             pmd->run.pkt_buffer_size : d0->length;
616                                         d0->flags = 0;
617                                 } else {
618                                         slot = saved_slot;
619                                         goto no_free_slots;
620                                 }
621                         }
622                         cp_len = RTE_MIN(dst_len, src_len);
623
624                         memcpy((uint8_t *)memif_get_buffer(proc_private, d0) + dst_off,
625                                rte_pktmbuf_mtod_offset(mbuf, void *, src_off),
626                                cp_len);
627
628                         mq->n_bytes += cp_len;
629                         src_off += cp_len;
630                         dst_off += cp_len;
631                         src_len -= cp_len;
632                         dst_len -= cp_len;
633
634                         d0->length = dst_off;
635                 }
636
637                 if (rte_pktmbuf_is_contiguous(mbuf) == 0) {
638                         mbuf = mbuf->next;
639                         goto next_in_chain;
640                 }
641
642                 n_tx_pkts++;
643                 slot++;
644                 n_free--;
645                 rte_pktmbuf_free(mbuf_head);
646         }
647
648 no_free_slots:
649         if (type == MEMIF_RING_S2M)
650                 __atomic_store_n(&ring->head, slot, __ATOMIC_RELEASE);
651         else
652                 __atomic_store_n(&ring->tail, slot, __ATOMIC_RELEASE);
653
654         if ((ring->flags & MEMIF_RING_FLAG_MASK_INT) == 0) {
655                 a = 1;
656                 size = write(mq->intr_handle.fd, &a, sizeof(a));
657                 if (unlikely(size < 0)) {
658                         MIF_LOG(WARNING,
659                                 "Failed to send interrupt. %s", strerror(errno));
660                 }
661         }
662
663         mq->n_pkts += n_tx_pkts;
664         return n_tx_pkts;
665 }
666
667
668 static int
669 memif_tx_one_zc(struct pmd_process_private *proc_private, struct memif_queue *mq,
670                 memif_ring_t *ring, struct rte_mbuf *mbuf, const uint16_t mask,
671                 uint16_t slot, uint16_t n_free)
672 {
673         memif_desc_t *d0;
674         int used_slots = 1;
675
676 next_in_chain:
677         /* store pointer to mbuf to free it later */
678         mq->buffers[slot & mask] = mbuf;
679         /* Increment refcnt to make sure the buffer is not freed before master
680          * receives it. (current segment)
681          */
682         rte_mbuf_refcnt_update(mbuf, 1);
683         /* populate descriptor */
684         d0 = &ring->desc[slot & mask];
685         d0->length = rte_pktmbuf_data_len(mbuf);
686         /* FIXME: get region index */
687         d0->region = 1;
688         d0->offset = rte_pktmbuf_mtod(mbuf, uint8_t *) -
689                 (uint8_t *)proc_private->regions[d0->region]->addr;
690         d0->flags = 0;
691
692         /* check if buffer is chained */
693         if (rte_pktmbuf_is_contiguous(mbuf) == 0) {
694                 if (n_free < 2)
695                         return 0;
696                 /* mark buffer as chained */
697                 d0->flags |= MEMIF_DESC_FLAG_NEXT;
698                 /* advance mbuf */
699                 mbuf = mbuf->next;
700                 /* update counters */
701                 used_slots++;
702                 slot++;
703                 n_free--;
704                 goto next_in_chain;
705         }
706         return used_slots;
707 }
708
709 static uint16_t
710 eth_memif_tx_zc(void *queue, struct rte_mbuf **bufs, uint16_t nb_pkts)
711 {
712         struct memif_queue *mq = queue;
713         struct pmd_internals *pmd = rte_eth_devices[mq->in_port].data->dev_private;
714         struct pmd_process_private *proc_private =
715                 rte_eth_devices[mq->in_port].process_private;
716         memif_ring_t *ring = memif_get_ring_from_queue(proc_private, mq);
717         uint16_t slot, n_free, ring_size, mask, n_tx_pkts = 0;
718         memif_ring_type_t type = mq->type;
719         struct rte_eth_link link;
720
721         if (unlikely((pmd->flags & ETH_MEMIF_FLAG_CONNECTED) == 0))
722                 return 0;
723         if (unlikely(ring == NULL)) {
724                 /* Secondary process will attempt to request regions. */
725                 rte_eth_link_get(mq->in_port, &link);
726                 return 0;
727         }
728
729         ring_size = 1 << mq->log2_ring_size;
730         mask = ring_size - 1;
731
732         /* free mbufs received by master */
733         memif_free_stored_mbufs(proc_private, mq);
734
735         /* ring type always MEMIF_RING_S2M */
736         /* The ring->head acts as a guard variable between Tx and Rx
737          * threads, so using load-acquire pairs with store-release
738          * to synchronize it between threads.
739          */
740         slot = __atomic_load_n(&ring->head, __ATOMIC_ACQUIRE);
741         n_free = ring_size - slot + mq->last_tail;
742
743         int used_slots;
744
745         while (n_free && (n_tx_pkts < nb_pkts)) {
746                 while ((n_free > 4) && ((nb_pkts - n_tx_pkts) > 4)) {
747                         if ((nb_pkts - n_tx_pkts) > 8) {
748                                 rte_prefetch0(*bufs + 4);
749                                 rte_prefetch0(*bufs + 5);
750                                 rte_prefetch0(*bufs + 6);
751                                 rte_prefetch0(*bufs + 7);
752                         }
753                         used_slots = memif_tx_one_zc(proc_private, mq, ring, *bufs++,
754                                 mask, slot, n_free);
755                         if (unlikely(used_slots < 1))
756                                 goto no_free_slots;
757                         n_tx_pkts++;
758                         slot += used_slots;
759                         n_free -= used_slots;
760
761                         used_slots = memif_tx_one_zc(proc_private, mq, ring, *bufs++,
762                                 mask, slot, n_free);
763                         if (unlikely(used_slots < 1))
764                                 goto no_free_slots;
765                         n_tx_pkts++;
766                         slot += used_slots;
767                         n_free -= used_slots;
768
769                         used_slots = memif_tx_one_zc(proc_private, mq, ring, *bufs++,
770                                 mask, slot, n_free);
771                         if (unlikely(used_slots < 1))
772                                 goto no_free_slots;
773                         n_tx_pkts++;
774                         slot += used_slots;
775                         n_free -= used_slots;
776
777                         used_slots = memif_tx_one_zc(proc_private, mq, ring, *bufs++,
778                                 mask, slot, n_free);
779                         if (unlikely(used_slots < 1))
780                                 goto no_free_slots;
781                         n_tx_pkts++;
782                         slot += used_slots;
783                         n_free -= used_slots;
784                 }
785                 used_slots = memif_tx_one_zc(proc_private, mq, ring, *bufs++,
786                         mask, slot, n_free);
787                 if (unlikely(used_slots < 1))
788                         goto no_free_slots;
789                 n_tx_pkts++;
790                 slot += used_slots;
791                 n_free -= used_slots;
792         }
793
794 no_free_slots:
795         /* update ring pointers */
796         if (type == MEMIF_RING_S2M)
797                 __atomic_store_n(&ring->head, slot, __ATOMIC_RELEASE);
798         else
799                 __atomic_store_n(&ring->tail, slot, __ATOMIC_RELEASE);
800
801         /* Send interrupt, if enabled. */
802         if ((ring->flags & MEMIF_RING_FLAG_MASK_INT) == 0) {
803                 uint64_t a = 1;
804                 ssize_t size = write(mq->intr_handle.fd, &a, sizeof(a));
805                 if (unlikely(size < 0)) {
806                         MIF_LOG(WARNING,
807                                 "Failed to send interrupt. %s", strerror(errno));
808                 }
809         }
810
811         /* increment queue counters */
812         mq->n_pkts += n_tx_pkts;
813
814         return n_tx_pkts;
815 }
816
817 void
818 memif_free_regions(struct rte_eth_dev *dev)
819 {
820         struct pmd_process_private *proc_private = dev->process_private;
821         struct pmd_internals *pmd = dev->data->dev_private;
822         int i;
823         struct memif_region *r;
824
825         /* regions are allocated contiguously, so it's
826          * enough to loop until 'proc_private->regions_num'
827          */
828         for (i = 0; i < proc_private->regions_num; i++) {
829                 r = proc_private->regions[i];
830                 if (r != NULL) {
831                         /* This is memzone */
832                         if (i > 0 && (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY)) {
833                                 r->addr = NULL;
834                                 if (r->fd > 0)
835                                         close(r->fd);
836                         }
837                         if (r->addr != NULL) {
838                                 munmap(r->addr, r->region_size);
839                                 if (r->fd > 0) {
840                                         close(r->fd);
841                                         r->fd = -1;
842                                 }
843                         }
844                         rte_free(r);
845                         proc_private->regions[i] = NULL;
846                 }
847         }
848         proc_private->regions_num = 0;
849 }
850
851 static int
852 memif_region_init_zc(const struct rte_memseg_list *msl, const struct rte_memseg *ms,
853                      void *arg)
854 {
855         struct pmd_process_private *proc_private = (struct pmd_process_private *)arg;
856         struct memif_region *r;
857
858         if (proc_private->regions_num < 1) {
859                 MIF_LOG(ERR, "Missing descriptor region");
860                 return -1;
861         }
862
863         r = proc_private->regions[proc_private->regions_num - 1];
864
865         if (r->addr != msl->base_va)
866                 r = proc_private->regions[++proc_private->regions_num - 1];
867
868         if (r == NULL) {
869                 r = rte_zmalloc("region", sizeof(struct memif_region), 0);
870                 if (r == NULL) {
871                         MIF_LOG(ERR, "Failed to alloc memif region.");
872                         return -ENOMEM;
873                 }
874
875                 r->addr = msl->base_va;
876                 r->region_size = ms->len;
877                 r->fd = rte_memseg_get_fd(ms);
878                 if (r->fd < 0)
879                         return -1;
880                 r->pkt_buffer_offset = 0;
881
882                 proc_private->regions[proc_private->regions_num - 1] = r;
883         } else {
884                 r->region_size += ms->len;
885         }
886
887         return 0;
888 }
889
890 static int
891 memif_region_init_shm(struct rte_eth_dev *dev, uint8_t has_buffers)
892 {
893         struct pmd_internals *pmd = dev->data->dev_private;
894         struct pmd_process_private *proc_private = dev->process_private;
895         char shm_name[ETH_MEMIF_SHM_NAME_SIZE];
896         int ret = 0;
897         struct memif_region *r;
898
899         if (proc_private->regions_num >= ETH_MEMIF_MAX_REGION_NUM) {
900                 MIF_LOG(ERR, "Too many regions.");
901                 return -1;
902         }
903
904         r = rte_zmalloc("region", sizeof(struct memif_region), 0);
905         if (r == NULL) {
906                 MIF_LOG(ERR, "Failed to alloc memif region.");
907                 return -ENOMEM;
908         }
909
910         /* calculate buffer offset */
911         r->pkt_buffer_offset = (pmd->run.num_s2m_rings + pmd->run.num_m2s_rings) *
912             (sizeof(memif_ring_t) + sizeof(memif_desc_t) *
913             (1 << pmd->run.log2_ring_size));
914
915         r->region_size = r->pkt_buffer_offset;
916         /* if region has buffers, add buffers size to region_size */
917         if (has_buffers == 1)
918                 r->region_size += (uint32_t)(pmd->run.pkt_buffer_size *
919                         (1 << pmd->run.log2_ring_size) *
920                         (pmd->run.num_s2m_rings +
921                          pmd->run.num_m2s_rings));
922
923         memset(shm_name, 0, sizeof(char) * ETH_MEMIF_SHM_NAME_SIZE);
924         snprintf(shm_name, ETH_MEMIF_SHM_NAME_SIZE, "memif_region_%d",
925                  proc_private->regions_num);
926
927         r->fd = memfd_create(shm_name, MFD_ALLOW_SEALING);
928         if (r->fd < 0) {
929                 MIF_LOG(ERR, "Failed to create shm file: %s.", strerror(errno));
930                 ret = -1;
931                 goto error;
932         }
933
934         ret = fcntl(r->fd, F_ADD_SEALS, F_SEAL_SHRINK);
935         if (ret < 0) {
936                 MIF_LOG(ERR, "Failed to add seals to shm file: %s.", strerror(errno));
937                 goto error;
938         }
939
940         ret = ftruncate(r->fd, r->region_size);
941         if (ret < 0) {
942                 MIF_LOG(ERR, "Failed to truncate shm file: %s.", strerror(errno));
943                 goto error;
944         }
945
946         r->addr = mmap(NULL, r->region_size, PROT_READ |
947                        PROT_WRITE, MAP_SHARED, r->fd, 0);
948         if (r->addr == MAP_FAILED) {
949                 MIF_LOG(ERR, "Failed to mmap shm region: %s.", strerror(ret));
950                 ret = -1;
951                 goto error;
952         }
953
954         proc_private->regions[proc_private->regions_num] = r;
955         proc_private->regions_num++;
956
957         return ret;
958
959 error:
960         if (r->fd > 0)
961                 close(r->fd);
962         r->fd = -1;
963
964         return ret;
965 }
966
967 static int
968 memif_regions_init(struct rte_eth_dev *dev)
969 {
970         struct pmd_internals *pmd = dev->data->dev_private;
971         int ret;
972
973         /*
974          * Zero-copy exposes dpdk memory.
975          * Each memseg list will be represented by memif region.
976          * Zero-copy regions indexing: memseg list idx + 1,
977          * as we already have region 0 reserved for descriptors.
978          */
979         if (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY) {
980                 /* create region idx 0 containing descriptors */
981                 ret = memif_region_init_shm(dev, 0);
982                 if (ret < 0)
983                         return ret;
984                 ret = rte_memseg_walk(memif_region_init_zc, (void *)dev->process_private);
985                 if (ret < 0)
986                         return ret;
987         } else {
988                 /* create one memory region contaning rings and buffers */
989                 ret = memif_region_init_shm(dev, /* has buffers */ 1);
990                 if (ret < 0)
991                         return ret;
992         }
993
994         return 0;
995 }
996
997 static void
998 memif_init_rings(struct rte_eth_dev *dev)
999 {
1000         struct pmd_internals *pmd = dev->data->dev_private;
1001         struct pmd_process_private *proc_private = dev->process_private;
1002         memif_ring_t *ring;
1003         int i, j;
1004         uint16_t slot;
1005
1006         for (i = 0; i < pmd->run.num_s2m_rings; i++) {
1007                 ring = memif_get_ring(pmd, proc_private, MEMIF_RING_S2M, i);
1008                 __atomic_store_n(&ring->head, 0, __ATOMIC_RELAXED);
1009                 __atomic_store_n(&ring->tail, 0, __ATOMIC_RELAXED);
1010                 ring->cookie = MEMIF_COOKIE;
1011                 ring->flags = 0;
1012
1013                 if (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY)
1014                         continue;
1015
1016                 for (j = 0; j < (1 << pmd->run.log2_ring_size); j++) {
1017                         slot = i * (1 << pmd->run.log2_ring_size) + j;
1018                         ring->desc[j].region = 0;
1019                         ring->desc[j].offset =
1020                                 proc_private->regions[0]->pkt_buffer_offset +
1021                                 (uint32_t)(slot * pmd->run.pkt_buffer_size);
1022                         ring->desc[j].length = pmd->run.pkt_buffer_size;
1023                 }
1024         }
1025
1026         for (i = 0; i < pmd->run.num_m2s_rings; i++) {
1027                 ring = memif_get_ring(pmd, proc_private, MEMIF_RING_M2S, i);
1028                 __atomic_store_n(&ring->head, 0, __ATOMIC_RELAXED);
1029                 __atomic_store_n(&ring->tail, 0, __ATOMIC_RELAXED);
1030                 ring->cookie = MEMIF_COOKIE;
1031                 ring->flags = 0;
1032
1033                 if (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY)
1034                         continue;
1035
1036                 for (j = 0; j < (1 << pmd->run.log2_ring_size); j++) {
1037                         slot = (i + pmd->run.num_s2m_rings) *
1038                             (1 << pmd->run.log2_ring_size) + j;
1039                         ring->desc[j].region = 0;
1040                         ring->desc[j].offset =
1041                                 proc_private->regions[0]->pkt_buffer_offset +
1042                                 (uint32_t)(slot * pmd->run.pkt_buffer_size);
1043                         ring->desc[j].length = pmd->run.pkt_buffer_size;
1044                 }
1045         }
1046 }
1047
1048 /* called only by slave */
1049 static int
1050 memif_init_queues(struct rte_eth_dev *dev)
1051 {
1052         struct pmd_internals *pmd = dev->data->dev_private;
1053         struct memif_queue *mq;
1054         int i;
1055
1056         for (i = 0; i < pmd->run.num_s2m_rings; i++) {
1057                 mq = dev->data->tx_queues[i];
1058                 mq->log2_ring_size = pmd->run.log2_ring_size;
1059                 /* queues located only in region 0 */
1060                 mq->region = 0;
1061                 mq->ring_offset = memif_get_ring_offset(dev, mq, MEMIF_RING_S2M, i);
1062                 mq->last_head = 0;
1063                 mq->last_tail = 0;
1064                 mq->intr_handle.fd = eventfd(0, EFD_NONBLOCK);
1065                 if (mq->intr_handle.fd < 0) {
1066                         MIF_LOG(WARNING,
1067                                 "Failed to create eventfd for tx queue %d: %s.", i,
1068                                 strerror(errno));
1069                 }
1070                 mq->buffers = NULL;
1071                 if (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY) {
1072                         mq->buffers = rte_zmalloc("bufs", sizeof(struct rte_mbuf *) *
1073                                                   (1 << mq->log2_ring_size), 0);
1074                         if (mq->buffers == NULL)
1075                                 return -ENOMEM;
1076                 }
1077         }
1078
1079         for (i = 0; i < pmd->run.num_m2s_rings; i++) {
1080                 mq = dev->data->rx_queues[i];
1081                 mq->log2_ring_size = pmd->run.log2_ring_size;
1082                 /* queues located only in region 0 */
1083                 mq->region = 0;
1084                 mq->ring_offset = memif_get_ring_offset(dev, mq, MEMIF_RING_M2S, i);
1085                 mq->last_head = 0;
1086                 mq->last_tail = 0;
1087                 mq->intr_handle.fd = eventfd(0, EFD_NONBLOCK);
1088                 if (mq->intr_handle.fd < 0) {
1089                         MIF_LOG(WARNING,
1090                                 "Failed to create eventfd for rx queue %d: %s.", i,
1091                                 strerror(errno));
1092                 }
1093                 mq->buffers = NULL;
1094                 if (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY) {
1095                         mq->buffers = rte_zmalloc("bufs", sizeof(struct rte_mbuf *) *
1096                                                   (1 << mq->log2_ring_size), 0);
1097                         if (mq->buffers == NULL)
1098                                 return -ENOMEM;
1099                 }
1100         }
1101         return 0;
1102 }
1103
1104 int
1105 memif_init_regions_and_queues(struct rte_eth_dev *dev)
1106 {
1107         int ret;
1108
1109         ret = memif_regions_init(dev);
1110         if (ret < 0)
1111                 return ret;
1112
1113         memif_init_rings(dev);
1114
1115         ret = memif_init_queues(dev);
1116         if (ret < 0)
1117                 return ret;
1118
1119         return 0;
1120 }
1121
1122 int
1123 memif_connect(struct rte_eth_dev *dev)
1124 {
1125         struct pmd_internals *pmd = dev->data->dev_private;
1126         struct pmd_process_private *proc_private = dev->process_private;
1127         struct memif_region *mr;
1128         struct memif_queue *mq;
1129         memif_ring_t *ring;
1130         int i;
1131
1132         for (i = 0; i < proc_private->regions_num; i++) {
1133                 mr = proc_private->regions[i];
1134                 if (mr != NULL) {
1135                         if (mr->addr == NULL) {
1136                                 if (mr->fd < 0)
1137                                         return -1;
1138                                 mr->addr = mmap(NULL, mr->region_size,
1139                                                 PROT_READ | PROT_WRITE,
1140                                                 MAP_SHARED, mr->fd, 0);
1141                                 if (mr->addr == MAP_FAILED) {
1142                                         MIF_LOG(ERR, "mmap failed: %s\n",
1143                                                 strerror(errno));
1144                                         return -1;
1145                                 }
1146                         }
1147                         if (i > 0 && (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY)) {
1148                                 /* close memseg file */
1149                                 close(mr->fd);
1150                                 mr->fd = -1;
1151                         }
1152                 }
1153         }
1154
1155         if (rte_eal_process_type() == RTE_PROC_PRIMARY) {
1156                 for (i = 0; i < pmd->run.num_s2m_rings; i++) {
1157                         mq = (pmd->role == MEMIF_ROLE_SLAVE) ?
1158                             dev->data->tx_queues[i] : dev->data->rx_queues[i];
1159                         ring = memif_get_ring_from_queue(proc_private, mq);
1160                         if (ring == NULL || ring->cookie != MEMIF_COOKIE) {
1161                                 MIF_LOG(ERR, "Wrong ring");
1162                                 return -1;
1163                         }
1164                         __atomic_store_n(&ring->head, 0, __ATOMIC_RELAXED);
1165                         __atomic_store_n(&ring->tail, 0, __ATOMIC_RELAXED);
1166                         mq->last_head = 0;
1167                         mq->last_tail = 0;
1168                         /* enable polling mode */
1169                         if (pmd->role == MEMIF_ROLE_MASTER)
1170                                 ring->flags = MEMIF_RING_FLAG_MASK_INT;
1171                 }
1172                 for (i = 0; i < pmd->run.num_m2s_rings; i++) {
1173                         mq = (pmd->role == MEMIF_ROLE_SLAVE) ?
1174                             dev->data->rx_queues[i] : dev->data->tx_queues[i];
1175                         ring = memif_get_ring_from_queue(proc_private, mq);
1176                         if (ring == NULL || ring->cookie != MEMIF_COOKIE) {
1177                                 MIF_LOG(ERR, "Wrong ring");
1178                                 return -1;
1179                         }
1180                         __atomic_store_n(&ring->head, 0, __ATOMIC_RELAXED);
1181                         __atomic_store_n(&ring->tail, 0, __ATOMIC_RELAXED);
1182                         mq->last_head = 0;
1183                         mq->last_tail = 0;
1184                         /* enable polling mode */
1185                         if (pmd->role == MEMIF_ROLE_SLAVE)
1186                                 ring->flags = MEMIF_RING_FLAG_MASK_INT;
1187                 }
1188
1189                 pmd->flags &= ~ETH_MEMIF_FLAG_CONNECTING;
1190                 pmd->flags |= ETH_MEMIF_FLAG_CONNECTED;
1191                 dev->data->dev_link.link_status = ETH_LINK_UP;
1192         }
1193         MIF_LOG(INFO, "Connected.");
1194         return 0;
1195 }
1196
1197 static int
1198 memif_dev_start(struct rte_eth_dev *dev)
1199 {
1200         struct pmd_internals *pmd = dev->data->dev_private;
1201         int ret = 0;
1202
1203         switch (pmd->role) {
1204         case MEMIF_ROLE_SLAVE:
1205                 ret = memif_connect_slave(dev);
1206                 break;
1207         case MEMIF_ROLE_MASTER:
1208                 ret = memif_connect_master(dev);
1209                 break;
1210         default:
1211                 MIF_LOG(ERR, "Unknown role: %d.", pmd->role);
1212                 ret = -1;
1213                 break;
1214         }
1215
1216         return ret;
1217 }
1218
1219 static int
1220 memif_dev_close(struct rte_eth_dev *dev)
1221 {
1222         struct pmd_internals *pmd = dev->data->dev_private;
1223         int i;
1224
1225         if (rte_eal_process_type() == RTE_PROC_PRIMARY) {
1226                 memif_msg_enq_disconnect(pmd->cc, "Device closed", 0);
1227                 memif_disconnect(dev);
1228
1229                 for (i = 0; i < dev->data->nb_rx_queues; i++)
1230                         (*dev->dev_ops->rx_queue_release)(dev->data->rx_queues[i]);
1231                 for (i = 0; i < dev->data->nb_tx_queues; i++)
1232                         (*dev->dev_ops->tx_queue_release)(dev->data->tx_queues[i]);
1233
1234                 memif_socket_remove_device(dev);
1235         } else {
1236                 memif_disconnect(dev);
1237         }
1238
1239         rte_free(dev->process_private);
1240
1241         return 0;
1242 }
1243
1244 static int
1245 memif_dev_configure(struct rte_eth_dev *dev)
1246 {
1247         struct pmd_internals *pmd = dev->data->dev_private;
1248
1249         /*
1250          * SLAVE - TXQ
1251          * MASTER - RXQ
1252          */
1253         pmd->cfg.num_s2m_rings = (pmd->role == MEMIF_ROLE_SLAVE) ?
1254                                   dev->data->nb_tx_queues : dev->data->nb_rx_queues;
1255
1256         /*
1257          * SLAVE - RXQ
1258          * MASTER - TXQ
1259          */
1260         pmd->cfg.num_m2s_rings = (pmd->role == MEMIF_ROLE_SLAVE) ?
1261                                   dev->data->nb_rx_queues : dev->data->nb_tx_queues;
1262
1263         return 0;
1264 }
1265
1266 static int
1267 memif_tx_queue_setup(struct rte_eth_dev *dev,
1268                      uint16_t qid,
1269                      uint16_t nb_tx_desc __rte_unused,
1270                      unsigned int socket_id __rte_unused,
1271                      const struct rte_eth_txconf *tx_conf __rte_unused)
1272 {
1273         struct pmd_internals *pmd = dev->data->dev_private;
1274         struct memif_queue *mq;
1275
1276         mq = rte_zmalloc("tx-queue", sizeof(struct memif_queue), 0);
1277         if (mq == NULL) {
1278                 MIF_LOG(ERR, "Failed to allocate tx queue id: %u", qid);
1279                 return -ENOMEM;
1280         }
1281
1282         mq->type =
1283             (pmd->role == MEMIF_ROLE_SLAVE) ? MEMIF_RING_S2M : MEMIF_RING_M2S;
1284         mq->n_pkts = 0;
1285         mq->n_bytes = 0;
1286         mq->intr_handle.fd = -1;
1287         mq->intr_handle.type = RTE_INTR_HANDLE_EXT;
1288         mq->in_port = dev->data->port_id;
1289         dev->data->tx_queues[qid] = mq;
1290
1291         return 0;
1292 }
1293
1294 static int
1295 memif_rx_queue_setup(struct rte_eth_dev *dev,
1296                      uint16_t qid,
1297                      uint16_t nb_rx_desc __rte_unused,
1298                      unsigned int socket_id __rte_unused,
1299                      const struct rte_eth_rxconf *rx_conf __rte_unused,
1300                      struct rte_mempool *mb_pool)
1301 {
1302         struct pmd_internals *pmd = dev->data->dev_private;
1303         struct memif_queue *mq;
1304
1305         mq = rte_zmalloc("rx-queue", sizeof(struct memif_queue), 0);
1306         if (mq == NULL) {
1307                 MIF_LOG(ERR, "Failed to allocate rx queue id: %u", qid);
1308                 return -ENOMEM;
1309         }
1310
1311         mq->type = (pmd->role == MEMIF_ROLE_SLAVE) ? MEMIF_RING_M2S : MEMIF_RING_S2M;
1312         mq->n_pkts = 0;
1313         mq->n_bytes = 0;
1314         mq->intr_handle.fd = -1;
1315         mq->intr_handle.type = RTE_INTR_HANDLE_EXT;
1316         mq->mempool = mb_pool;
1317         mq->in_port = dev->data->port_id;
1318         dev->data->rx_queues[qid] = mq;
1319
1320         return 0;
1321 }
1322
1323 static void
1324 memif_queue_release(void *queue)
1325 {
1326         struct memif_queue *mq = (struct memif_queue *)queue;
1327
1328         if (!mq)
1329                 return;
1330
1331         rte_free(mq);
1332 }
1333
1334 static int
1335 memif_link_update(struct rte_eth_dev *dev,
1336                   int wait_to_complete __rte_unused)
1337 {
1338         struct pmd_process_private *proc_private;
1339
1340         if (rte_eal_process_type() == RTE_PROC_SECONDARY) {
1341                 proc_private = dev->process_private;
1342                 if (dev->data->dev_link.link_status == ETH_LINK_UP &&
1343                                 proc_private->regions_num == 0) {
1344                         memif_mp_request_regions(dev);
1345                 } else if (dev->data->dev_link.link_status == ETH_LINK_DOWN &&
1346                                 proc_private->regions_num > 0) {
1347                         memif_free_regions(dev);
1348                 }
1349         }
1350         return 0;
1351 }
1352
1353 static int
1354 memif_stats_get(struct rte_eth_dev *dev, struct rte_eth_stats *stats)
1355 {
1356         struct pmd_internals *pmd = dev->data->dev_private;
1357         struct memif_queue *mq;
1358         int i;
1359         uint8_t tmp, nq;
1360
1361         stats->ipackets = 0;
1362         stats->ibytes = 0;
1363         stats->opackets = 0;
1364         stats->obytes = 0;
1365
1366         tmp = (pmd->role == MEMIF_ROLE_SLAVE) ? pmd->run.num_s2m_rings :
1367             pmd->run.num_m2s_rings;
1368         nq = (tmp < RTE_ETHDEV_QUEUE_STAT_CNTRS) ? tmp :
1369             RTE_ETHDEV_QUEUE_STAT_CNTRS;
1370
1371         /* RX stats */
1372         for (i = 0; i < nq; i++) {
1373                 mq = dev->data->rx_queues[i];
1374                 stats->q_ipackets[i] = mq->n_pkts;
1375                 stats->q_ibytes[i] = mq->n_bytes;
1376                 stats->ipackets += mq->n_pkts;
1377                 stats->ibytes += mq->n_bytes;
1378         }
1379
1380         tmp = (pmd->role == MEMIF_ROLE_SLAVE) ? pmd->run.num_m2s_rings :
1381             pmd->run.num_s2m_rings;
1382         nq = (tmp < RTE_ETHDEV_QUEUE_STAT_CNTRS) ? tmp :
1383             RTE_ETHDEV_QUEUE_STAT_CNTRS;
1384
1385         /* TX stats */
1386         for (i = 0; i < nq; i++) {
1387                 mq = dev->data->tx_queues[i];
1388                 stats->q_opackets[i] = mq->n_pkts;
1389                 stats->q_obytes[i] = mq->n_bytes;
1390                 stats->opackets += mq->n_pkts;
1391                 stats->obytes += mq->n_bytes;
1392         }
1393         return 0;
1394 }
1395
1396 static int
1397 memif_stats_reset(struct rte_eth_dev *dev)
1398 {
1399         struct pmd_internals *pmd = dev->data->dev_private;
1400         int i;
1401         struct memif_queue *mq;
1402
1403         for (i = 0; i < pmd->run.num_s2m_rings; i++) {
1404                 mq = (pmd->role == MEMIF_ROLE_SLAVE) ? dev->data->tx_queues[i] :
1405                     dev->data->rx_queues[i];
1406                 mq->n_pkts = 0;
1407                 mq->n_bytes = 0;
1408         }
1409         for (i = 0; i < pmd->run.num_m2s_rings; i++) {
1410                 mq = (pmd->role == MEMIF_ROLE_SLAVE) ? dev->data->rx_queues[i] :
1411                     dev->data->tx_queues[i];
1412                 mq->n_pkts = 0;
1413                 mq->n_bytes = 0;
1414         }
1415
1416         return 0;
1417 }
1418
1419 static int
1420 memif_rx_queue_intr_enable(struct rte_eth_dev *dev __rte_unused,
1421                            uint16_t qid __rte_unused)
1422 {
1423         MIF_LOG(WARNING, "Interrupt mode not supported.");
1424
1425         return -1;
1426 }
1427
1428 static int
1429 memif_rx_queue_intr_disable(struct rte_eth_dev *dev, uint16_t qid __rte_unused)
1430 {
1431         struct pmd_internals *pmd __rte_unused = dev->data->dev_private;
1432
1433         return 0;
1434 }
1435
1436 static const struct eth_dev_ops ops = {
1437         .dev_start = memif_dev_start,
1438         .dev_close = memif_dev_close,
1439         .dev_infos_get = memif_dev_info,
1440         .dev_configure = memif_dev_configure,
1441         .tx_queue_setup = memif_tx_queue_setup,
1442         .rx_queue_setup = memif_rx_queue_setup,
1443         .rx_queue_release = memif_queue_release,
1444         .tx_queue_release = memif_queue_release,
1445         .rx_queue_intr_enable = memif_rx_queue_intr_enable,
1446         .rx_queue_intr_disable = memif_rx_queue_intr_disable,
1447         .link_update = memif_link_update,
1448         .stats_get = memif_stats_get,
1449         .stats_reset = memif_stats_reset,
1450 };
1451
1452 static int
1453 memif_create(struct rte_vdev_device *vdev, enum memif_role_t role,
1454              memif_interface_id_t id, uint32_t flags,
1455              const char *socket_filename,
1456              memif_log2_ring_size_t log2_ring_size,
1457              uint16_t pkt_buffer_size, const char *secret,
1458              struct rte_ether_addr *ether_addr)
1459 {
1460         int ret = 0;
1461         struct rte_eth_dev *eth_dev;
1462         struct rte_eth_dev_data *data;
1463         struct pmd_internals *pmd;
1464         struct pmd_process_private *process_private;
1465         const unsigned int numa_node = vdev->device.numa_node;
1466         const char *name = rte_vdev_device_name(vdev);
1467
1468         eth_dev = rte_eth_vdev_allocate(vdev, sizeof(*pmd));
1469         if (eth_dev == NULL) {
1470                 MIF_LOG(ERR, "%s: Unable to allocate device struct.", name);
1471                 return -1;
1472         }
1473
1474         process_private = (struct pmd_process_private *)
1475                 rte_zmalloc(name, sizeof(struct pmd_process_private),
1476                             RTE_CACHE_LINE_SIZE);
1477
1478         if (process_private == NULL) {
1479                 MIF_LOG(ERR, "Failed to alloc memory for process private");
1480                 return -1;
1481         }
1482         eth_dev->process_private = process_private;
1483
1484         pmd = eth_dev->data->dev_private;
1485         memset(pmd, 0, sizeof(*pmd));
1486
1487         pmd->id = id;
1488         pmd->flags = flags;
1489         pmd->flags |= ETH_MEMIF_FLAG_DISABLED;
1490         pmd->role = role;
1491         /* Zero-copy flag irelevant to master. */
1492         if (pmd->role == MEMIF_ROLE_MASTER)
1493                 pmd->flags &= ~ETH_MEMIF_FLAG_ZERO_COPY;
1494
1495         ret = memif_socket_init(eth_dev, socket_filename);
1496         if (ret < 0)
1497                 return ret;
1498
1499         memset(pmd->secret, 0, sizeof(char) * ETH_MEMIF_SECRET_SIZE);
1500         if (secret != NULL)
1501                 strlcpy(pmd->secret, secret, sizeof(pmd->secret));
1502
1503         pmd->cfg.log2_ring_size = log2_ring_size;
1504         /* set in .dev_configure() */
1505         pmd->cfg.num_s2m_rings = 0;
1506         pmd->cfg.num_m2s_rings = 0;
1507
1508         pmd->cfg.pkt_buffer_size = pkt_buffer_size;
1509         rte_spinlock_init(&pmd->cc_lock);
1510
1511         data = eth_dev->data;
1512         data->dev_private = pmd;
1513         data->numa_node = numa_node;
1514         data->dev_link = pmd_link;
1515         data->mac_addrs = ether_addr;
1516         data->promiscuous = 1;
1517
1518         eth_dev->dev_ops = &ops;
1519         eth_dev->device = &vdev->device;
1520         if (pmd->flags & ETH_MEMIF_FLAG_ZERO_COPY) {
1521                 eth_dev->rx_pkt_burst = eth_memif_rx_zc;
1522                 eth_dev->tx_pkt_burst = eth_memif_tx_zc;
1523         } else {
1524                 eth_dev->rx_pkt_burst = eth_memif_rx;
1525                 eth_dev->tx_pkt_burst = eth_memif_tx;
1526         }
1527
1528         rte_eth_dev_probing_finish(eth_dev);
1529
1530         return 0;
1531 }
1532
1533 static int
1534 memif_set_role(const char *key __rte_unused, const char *value,
1535                void *extra_args)
1536 {
1537         enum memif_role_t *role = (enum memif_role_t *)extra_args;
1538
1539         if (strstr(value, "master") != NULL) {
1540                 *role = MEMIF_ROLE_MASTER;
1541         } else if (strstr(value, "slave") != NULL) {
1542                 *role = MEMIF_ROLE_SLAVE;
1543         } else {
1544                 MIF_LOG(ERR, "Unknown role: %s.", value);
1545                 return -EINVAL;
1546         }
1547         return 0;
1548 }
1549
1550 static int
1551 memif_set_zc(const char *key __rte_unused, const char *value, void *extra_args)
1552 {
1553         uint32_t *flags = (uint32_t *)extra_args;
1554
1555         if (strstr(value, "yes") != NULL) {
1556                 if (!rte_mcfg_get_single_file_segments()) {
1557                         MIF_LOG(ERR, "Zero-copy doesn't support multi-file segments.");
1558                         return -ENOTSUP;
1559                 }
1560                 *flags |= ETH_MEMIF_FLAG_ZERO_COPY;
1561         } else if (strstr(value, "no") != NULL) {
1562                 *flags &= ~ETH_MEMIF_FLAG_ZERO_COPY;
1563         } else {
1564                 MIF_LOG(ERR, "Failed to parse zero-copy param: %s.", value);
1565                 return -EINVAL;
1566         }
1567         return 0;
1568 }
1569
1570 static int
1571 memif_set_id(const char *key __rte_unused, const char *value, void *extra_args)
1572 {
1573         memif_interface_id_t *id = (memif_interface_id_t *)extra_args;
1574
1575         /* even if parsing fails, 0 is a valid id */
1576         *id = strtoul(value, NULL, 10);
1577         return 0;
1578 }
1579
1580 static int
1581 memif_set_bs(const char *key __rte_unused, const char *value, void *extra_args)
1582 {
1583         unsigned long tmp;
1584         uint16_t *pkt_buffer_size = (uint16_t *)extra_args;
1585
1586         tmp = strtoul(value, NULL, 10);
1587         if (tmp == 0 || tmp > 0xFFFF) {
1588                 MIF_LOG(ERR, "Invalid buffer size: %s.", value);
1589                 return -EINVAL;
1590         }
1591         *pkt_buffer_size = tmp;
1592         return 0;
1593 }
1594
1595 static int
1596 memif_set_rs(const char *key __rte_unused, const char *value, void *extra_args)
1597 {
1598         unsigned long tmp;
1599         memif_log2_ring_size_t *log2_ring_size =
1600             (memif_log2_ring_size_t *)extra_args;
1601
1602         tmp = strtoul(value, NULL, 10);
1603         if (tmp == 0 || tmp > ETH_MEMIF_MAX_LOG2_RING_SIZE) {
1604                 MIF_LOG(ERR, "Invalid ring size: %s (max %u).",
1605                         value, ETH_MEMIF_MAX_LOG2_RING_SIZE);
1606                 return -EINVAL;
1607         }
1608         *log2_ring_size = tmp;
1609         return 0;
1610 }
1611
1612 /* check if directory exists and if we have permission to read/write */
1613 static int
1614 memif_check_socket_filename(const char *filename)
1615 {
1616         char *dir = NULL, *tmp;
1617         uint32_t idx;
1618         int ret = 0;
1619
1620         if (strlen(filename) >= MEMIF_SOCKET_UN_SIZE) {
1621                 MIF_LOG(ERR, "Unix socket address too long (max 108).");
1622                 return -1;
1623         }
1624
1625         tmp = strrchr(filename, '/');
1626         if (tmp != NULL) {
1627                 idx = tmp - filename;
1628                 dir = rte_zmalloc("memif_tmp", sizeof(char) * (idx + 1), 0);
1629                 if (dir == NULL) {
1630                         MIF_LOG(ERR, "Failed to allocate memory.");
1631                         return -1;
1632                 }
1633                 strlcpy(dir, filename, sizeof(char) * (idx + 1));
1634         }
1635
1636         if (dir == NULL || (faccessat(-1, dir, F_OK | R_OK |
1637                                         W_OK, AT_EACCESS) < 0)) {
1638                 MIF_LOG(ERR, "Invalid socket directory.");
1639                 ret = -EINVAL;
1640         }
1641
1642         if (dir != NULL)
1643                 rte_free(dir);
1644
1645         return ret;
1646 }
1647
1648 static int
1649 memif_set_socket_filename(const char *key __rte_unused, const char *value,
1650                           void *extra_args)
1651 {
1652         const char **socket_filename = (const char **)extra_args;
1653
1654         *socket_filename = value;
1655         return memif_check_socket_filename(*socket_filename);
1656 }
1657
1658 static int
1659 memif_set_mac(const char *key __rte_unused, const char *value, void *extra_args)
1660 {
1661         struct rte_ether_addr *ether_addr = (struct rte_ether_addr *)extra_args;
1662
1663         if (rte_ether_unformat_addr(value, ether_addr) < 0)
1664                 MIF_LOG(WARNING, "Failed to parse mac '%s'.", value);
1665         return 0;
1666 }
1667
1668 static int
1669 memif_set_secret(const char *key __rte_unused, const char *value, void *extra_args)
1670 {
1671         const char **secret = (const char **)extra_args;
1672
1673         *secret = value;
1674         return 0;
1675 }
1676
1677 static int
1678 rte_pmd_memif_probe(struct rte_vdev_device *vdev)
1679 {
1680         RTE_BUILD_BUG_ON(sizeof(memif_msg_t) != 128);
1681         RTE_BUILD_BUG_ON(sizeof(memif_desc_t) != 16);
1682         int ret = 0;
1683         struct rte_kvargs *kvlist;
1684         const char *name = rte_vdev_device_name(vdev);
1685         enum memif_role_t role = MEMIF_ROLE_SLAVE;
1686         memif_interface_id_t id = 0;
1687         uint16_t pkt_buffer_size = ETH_MEMIF_DEFAULT_PKT_BUFFER_SIZE;
1688         memif_log2_ring_size_t log2_ring_size = ETH_MEMIF_DEFAULT_RING_SIZE;
1689         const char *socket_filename = ETH_MEMIF_DEFAULT_SOCKET_FILENAME;
1690         uint32_t flags = 0;
1691         const char *secret = NULL;
1692         struct rte_ether_addr *ether_addr = rte_zmalloc("",
1693                 sizeof(struct rte_ether_addr), 0);
1694         struct rte_eth_dev *eth_dev;
1695
1696         rte_eth_random_addr(ether_addr->addr_bytes);
1697
1698         MIF_LOG(INFO, "Initialize MEMIF: %s.", name);
1699
1700         if (rte_eal_process_type() == RTE_PROC_SECONDARY) {
1701                 eth_dev = rte_eth_dev_attach_secondary(name);
1702                 if (!eth_dev) {
1703                         MIF_LOG(ERR, "Failed to probe %s", name);
1704                         return -1;
1705                 }
1706
1707                 eth_dev->dev_ops = &ops;
1708                 eth_dev->device = &vdev->device;
1709                 eth_dev->rx_pkt_burst = eth_memif_rx;
1710                 eth_dev->tx_pkt_burst = eth_memif_tx;
1711
1712                 if (!rte_eal_primary_proc_alive(NULL)) {
1713                         MIF_LOG(ERR, "Primary process is missing");
1714                         return -1;
1715                 }
1716
1717                 eth_dev->process_private = (struct pmd_process_private *)
1718                         rte_zmalloc(name,
1719                                 sizeof(struct pmd_process_private),
1720                                 RTE_CACHE_LINE_SIZE);
1721                 if (eth_dev->process_private == NULL) {
1722                         MIF_LOG(ERR,
1723                                 "Failed to alloc memory for process private");
1724                         return -1;
1725                 }
1726
1727                 rte_eth_dev_probing_finish(eth_dev);
1728
1729                 return 0;
1730         }
1731
1732         ret = rte_mp_action_register(MEMIF_MP_SEND_REGION, memif_mp_send_region);
1733         /*
1734          * Primary process can continue probing, but secondary process won't
1735          * be able to get memory regions information
1736          */
1737         if (ret < 0 && rte_errno != EEXIST)
1738                 MIF_LOG(WARNING, "Failed to register mp action callback: %s",
1739                         strerror(rte_errno));
1740
1741         kvlist = rte_kvargs_parse(rte_vdev_device_args(vdev), valid_arguments);
1742
1743         /* parse parameters */
1744         if (kvlist != NULL) {
1745                 ret = rte_kvargs_process(kvlist, ETH_MEMIF_ROLE_ARG,
1746                                          &memif_set_role, &role);
1747                 if (ret < 0)
1748                         goto exit;
1749                 ret = rte_kvargs_process(kvlist, ETH_MEMIF_ID_ARG,
1750                                          &memif_set_id, &id);
1751                 if (ret < 0)
1752                         goto exit;
1753                 ret = rte_kvargs_process(kvlist, ETH_MEMIF_PKT_BUFFER_SIZE_ARG,
1754                                          &memif_set_bs, &pkt_buffer_size);
1755                 if (ret < 0)
1756                         goto exit;
1757                 ret = rte_kvargs_process(kvlist, ETH_MEMIF_RING_SIZE_ARG,
1758                                          &memif_set_rs, &log2_ring_size);
1759                 if (ret < 0)
1760                         goto exit;
1761                 ret = rte_kvargs_process(kvlist, ETH_MEMIF_SOCKET_ARG,
1762                                          &memif_set_socket_filename,
1763                                          (void *)(&socket_filename));
1764                 if (ret < 0)
1765                         goto exit;
1766                 ret = rte_kvargs_process(kvlist, ETH_MEMIF_MAC_ARG,
1767                                          &memif_set_mac, ether_addr);
1768                 if (ret < 0)
1769                         goto exit;
1770                 ret = rte_kvargs_process(kvlist, ETH_MEMIF_ZC_ARG,
1771                                          &memif_set_zc, &flags);
1772                 if (ret < 0)
1773                         goto exit;
1774                 ret = rte_kvargs_process(kvlist, ETH_MEMIF_SECRET_ARG,
1775                                          &memif_set_secret, (void *)(&secret));
1776                 if (ret < 0)
1777                         goto exit;
1778         }
1779
1780         /* create interface */
1781         ret = memif_create(vdev, role, id, flags, socket_filename,
1782                            log2_ring_size, pkt_buffer_size, secret, ether_addr);
1783
1784 exit:
1785         if (kvlist != NULL)
1786                 rte_kvargs_free(kvlist);
1787         return ret;
1788 }
1789
1790 static int
1791 rte_pmd_memif_remove(struct rte_vdev_device *vdev)
1792 {
1793         struct rte_eth_dev *eth_dev;
1794
1795         eth_dev = rte_eth_dev_allocated(rte_vdev_device_name(vdev));
1796         if (eth_dev == NULL)
1797                 return 0;
1798
1799         rte_eth_dev_close(eth_dev->data->port_id);
1800
1801         return 0;
1802 }
1803
1804 static struct rte_vdev_driver pmd_memif_drv = {
1805         .probe = rte_pmd_memif_probe,
1806         .remove = rte_pmd_memif_remove,
1807 };
1808
1809 RTE_PMD_REGISTER_VDEV(net_memif, pmd_memif_drv);
1810
1811 RTE_PMD_REGISTER_PARAM_STRING(net_memif,
1812                               ETH_MEMIF_ID_ARG "=<int>"
1813                               ETH_MEMIF_ROLE_ARG "=master|slave"
1814                               ETH_MEMIF_PKT_BUFFER_SIZE_ARG "=<int>"
1815                               ETH_MEMIF_RING_SIZE_ARG "=<int>"
1816                               ETH_MEMIF_SOCKET_ARG "=<string>"
1817                               ETH_MEMIF_MAC_ARG "=xx:xx:xx:xx:xx:xx"
1818                               ETH_MEMIF_ZC_ARG "=yes|no"
1819                               ETH_MEMIF_SECRET_ARG "=<string>");
1820
1821 RTE_LOG_REGISTER(memif_logtype, pmd.net.memif, NOTICE);