f4543c837db0137c0aed5f967e4cbf1e69155af0
[vpp.git] / src / plugins / memif / memif.c
1 /*
2  *------------------------------------------------------------------
3  * Copyright (c) 2017 Cisco and/or its affiliates.
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at:
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  *------------------------------------------------------------------
16  */
17
18
19 #define _GNU_SOURCE
20 #include <stdint.h>
21 #include <net/if.h>
22 #include <sys/types.h>
23 #include <fcntl.h>
24 #include <sys/ioctl.h>
25 #include <sys/socket.h>
26 #include <sys/un.h>
27 #include <sys/uio.h>
28 #include <sys/mman.h>
29 #include <sys/prctl.h>
30 #include <sys/eventfd.h>
31 #include <inttypes.h>
32 #include <limits.h>
33
34 #include <vlib/vlib.h>
35 #include <vlib/unix/unix.h>
36 #include <vnet/plugin/plugin.h>
37 #include <vnet/ethernet/ethernet.h>
38 #include <vnet/interface/rx_queue_funcs.h>
39 #include <vnet/interface/tx_queue_funcs.h>
40 #include <vpp/app/version.h>
41 #include <memif/memif.h>
42 #include <memif/private.h>
43
44 memif_main_t memif_main;
45
46 static u32
47 memif_eth_flag_change (vnet_main_t * vnm, vnet_hw_interface_t * hi, u32 flags)
48 {
49   /* nothing for now */
50   return 0;
51 }
52
53 static clib_error_t *
54 memif_eth_set_max_frame_size (vnet_main_t *vnm, vnet_hw_interface_t *hi,
55                               u32 flags)
56 {
57   /* nothing for now */
58   return 0;
59 }
60
61 static void
62 memif_queue_intfd_close (memif_queue_t * mq)
63 {
64   if (mq->int_clib_file_index != ~0)
65     {
66       memif_file_del_by_index (mq->int_clib_file_index);
67       mq->int_clib_file_index = ~0;
68       mq->int_fd = -1;
69     }
70   else if (mq->int_fd > -1)
71     {
72       close (mq->int_fd);
73       mq->int_fd = -1;
74     }
75 }
76
77 static void
78 memif_disconnect_free_zc_queue_buffer (memif_queue_t * mq, u8 is_rx)
79 {
80   vlib_main_t *vm = vlib_get_main ();
81   u16 ring_size, n_slots, mask, start;
82
83   ring_size = 1 << mq->log2_ring_size;
84   mask = ring_size - 1;
85   n_slots = mq->ring->head - mq->last_tail;
86   start = mq->last_tail & mask;
87   if (is_rx)
88     vlib_buffer_free_from_ring (vm, mq->buffers, start, ring_size, n_slots);
89   else
90     vlib_buffer_free_from_ring_no_next (vm, mq->buffers, start, ring_size,
91                                         n_slots);
92   vec_free (mq->buffers);
93 }
94
95 void
96 memif_disconnect (memif_if_t * mif, clib_error_t * err)
97 {
98   memif_main_t *mm = &memif_main;
99   vnet_main_t *vnm = vnet_get_main ();
100   memif_region_t *mr;
101   memif_queue_t *mq;
102   int i;
103
104   if (mif == 0)
105     return;
106
107   memif_log_debug (mif, "disconnect %u (%v)", mif->dev_instance,
108                    err ? err->what : 0);
109
110   if (err)
111     {
112       clib_error_t *e = 0;
113       mif->local_disc_string = vec_dup (err->what);
114       if (mif->sock && clib_socket_is_connected (mif->sock))
115         e = memif_msg_send_disconnect (mif, err);
116       clib_error_free (e);
117     }
118
119   /* set interface down */
120   mif->flags &= ~(MEMIF_IF_FLAG_CONNECTED | MEMIF_IF_FLAG_CONNECTING);
121   if (mif->hw_if_index != ~0)
122     vnet_hw_interface_set_flags (vnm, mif->hw_if_index, 0);
123
124   /* close connection socket */
125   if (mif->sock && mif->sock->fd)
126     {
127       memif_socket_file_t *msf = vec_elt_at_index (mm->socket_files,
128                                                    mif->socket_file_index);
129       hash_unset (msf->dev_instance_by_fd, mif->sock->fd);
130       memif_socket_close (&mif->sock);
131     }
132   else if (mif->sock)
133     {
134       clib_error_t *err;
135       err = clib_socket_close (mif->sock);
136       if (err)
137         {
138           memif_log_err (mif, "%U", format_clib_error, err);
139           clib_error_free (err);
140         }
141       clib_mem_free (mif->sock);
142     }
143
144   /* *INDENT-OFF* */
145   vec_foreach_index (i, mif->rx_queues)
146     {
147       mq = vec_elt_at_index (mif->rx_queues, i);
148       if (mq->ring)
149         {
150           if (mif->flags & MEMIF_IF_FLAG_ZERO_COPY)
151           {
152             memif_disconnect_free_zc_queue_buffer(mq, 1);
153           }
154           mq->ring = 0;
155         }
156     }
157   vnet_hw_if_unregister_all_rx_queues (vnm, mif->hw_if_index);
158
159   /* *INDENT-OFF* */
160   vec_foreach_index (i, mif->tx_queues)
161   {
162     mq = vec_elt_at_index (mif->tx_queues, i);
163     if (mq->ring)
164     {
165       if (mif->flags & MEMIF_IF_FLAG_ZERO_COPY)
166       {
167         memif_disconnect_free_zc_queue_buffer(mq, 0);
168       }
169       clib_spinlock_free (&mq->lockp);
170     }
171     mq->ring = 0;
172   }
173   vnet_hw_if_unregister_all_tx_queues (vnm, mif->hw_if_index);
174   vnet_hw_if_update_runtime_data (vnm, mif->hw_if_index);
175
176   /* free tx and rx queues */
177   vec_foreach (mq, mif->rx_queues)
178     memif_queue_intfd_close (mq);
179   vec_free (mif->rx_queues);
180
181   vec_foreach (mq, mif->tx_queues)
182     memif_queue_intfd_close (mq);
183   vec_free (mif->tx_queues);
184
185   /* free memory regions */
186   vec_foreach (mr, mif->regions)
187     {
188       int rv;
189       if (mr->is_external)
190         continue;
191       if ((rv = munmap (mr->shm, mr->region_size)))
192         memif_log_err (mif, "munmap failed, rv = %d", rv);
193       if (mr->fd > -1)
194         close (mr->fd);
195     }
196   /* *INDENT-ON* */
197   vec_free (mif->regions);
198   vec_free (mif->remote_name);
199   vec_free (mif->remote_if_name);
200   clib_fifo_free (mif->msg_queue);
201 }
202
203 static clib_error_t *
204 memif_int_fd_write_ready (clib_file_t * uf)
205 {
206   memif_main_t *mm = &memif_main;
207   u16 qid = uf->private_data & 0xFFFF;
208   memif_if_t *mif = vec_elt_at_index (mm->interfaces, uf->private_data >> 16);
209
210   memif_log_warn (mif, "unexpected EPOLLOUT on RX for queue %u", qid);
211   return 0;
212 }
213
214 static clib_error_t *
215 memif_int_fd_read_ready (clib_file_t * uf)
216 {
217   memif_main_t *mm = &memif_main;
218   vnet_main_t *vnm = vnet_get_main ();
219   u16 qid = uf->private_data & 0xFFFF;
220   memif_if_t *mif = vec_elt_at_index (mm->interfaces, uf->private_data >> 16);
221   memif_queue_t *mq = vec_elt_at_index (mif->rx_queues, qid);
222   u64 b;
223   ssize_t size;
224
225   size = read (uf->file_descriptor, &b, sizeof (b));
226   if (size < 0)
227     {
228       memif_log_debug (mif, "Failed to read from socket");
229       return 0;
230     }
231
232   vnet_hw_if_rx_queue_set_int_pending (vnm, mq->queue_index);
233   mq->int_count++;
234
235   return 0;
236 }
237
238
239 clib_error_t *
240 memif_connect (memif_if_t * mif)
241 {
242   memif_main_t *mm = &memif_main;
243   vlib_main_t *vm = vlib_get_main ();
244   vnet_main_t *vnm = vnet_get_main ();
245   clib_file_t template = { 0 };
246   memif_region_t *mr;
247   int i, j;
248   u32 n_txqs = 0, n_threads = vlib_get_n_threads ();
249   clib_error_t *err = NULL;
250   u8 max_log2_ring_sz = 0;
251   int with_barrier = 0;
252
253   memif_log_debug (mif, "connect %u", mif->dev_instance);
254
255   vec_free (mif->local_disc_string);
256   vec_free (mif->remote_disc_string);
257
258   /* *INDENT-OFF* */
259   vec_foreach (mr, mif->regions)
260     {
261       if (mr->shm)
262         continue;
263
264       if (mr->fd < 0)
265         {
266           err = clib_error_return (0, "no memory region fd");
267           goto error;
268         }
269
270       if ((mr->shm = mmap (NULL, mr->region_size, PROT_READ | PROT_WRITE,
271                            MAP_SHARED, mr->fd, 0)) == MAP_FAILED)
272         {
273           err = clib_error_return_unix (0, "mmap");
274           goto error;
275         }
276     }
277   /* *INDENT-ON* */
278
279   template.read_function = memif_int_fd_read_ready;
280   template.write_function = memif_int_fd_write_ready;
281
282   with_barrier = 1;
283   if (vlib_worker_thread_barrier_held ())
284     with_barrier = 0;
285
286   if (with_barrier)
287     vlib_worker_thread_barrier_sync (vm);
288
289   /* *INDENT-OFF* */
290   vec_foreach_index (i, mif->tx_queues)
291     {
292       memif_queue_t *mq = vec_elt_at_index (mif->tx_queues, i);
293       max_log2_ring_sz = clib_max (max_log2_ring_sz, mq->log2_ring_size);
294
295       mq->ring = mif->regions[mq->region].shm + mq->offset;
296       if (mq->ring->cookie != MEMIF_COOKIE)
297         {
298           err = clib_error_return (0, "wrong cookie on tx ring %u", i);
299           goto error;
300         }
301       mq->queue_index =
302         vnet_hw_if_register_tx_queue (vnm, mif->hw_if_index, i);
303       clib_spinlock_init (&mq->lockp);
304     }
305
306   if (vec_len (mif->tx_queues) > 0)
307     {
308       n_txqs = vec_len (mif->tx_queues);
309       for (j = 0; j < n_threads; j++)
310         {
311           u32 qi = mif->tx_queues[j % n_txqs].queue_index;
312           vnet_hw_if_tx_queue_assign_thread (vnm, qi, j);
313         }
314     }
315
316   vec_foreach_index (i, mif->rx_queues)
317     {
318       memif_queue_t *mq = vec_elt_at_index (mif->rx_queues, i);
319       u32 ti;
320       u32 qi;
321       int rv;
322
323       max_log2_ring_sz = clib_max (max_log2_ring_sz, mq->log2_ring_size);
324
325       mq->ring = mif->regions[mq->region].shm + mq->offset;
326       if (mq->ring->cookie != MEMIF_COOKIE)
327         {
328           err = clib_error_return (0, "wrong cookie on tx ring %u", i);
329           goto error;
330         }
331       qi = vnet_hw_if_register_rx_queue (vnm, mif->hw_if_index, i,
332                                          VNET_HW_IF_RXQ_THREAD_ANY);
333       mq->queue_index = qi;
334       if (mq->int_fd > -1)
335         {
336           template.file_descriptor = mq->int_fd;
337           template.private_data = (mif->dev_instance << 16) | (i & 0xFFFF);
338           template.description = format (0, "%U rx %u int",
339                                          format_memif_device_name,
340                                          mif->dev_instance, i);
341           memif_file_add (&mq->int_clib_file_index, &template);
342           vnet_hw_if_set_rx_queue_file_index (vnm, qi,
343                                               mq->int_clib_file_index);
344         }
345       ti = vnet_hw_if_get_rx_queue_thread_index (vnm, qi);
346       mq->buffer_pool_index = vlib_buffer_pool_get_default_for_numa (
347         vm, vlib_get_main_by_index (ti)->numa_node);
348       rv = vnet_hw_if_set_rx_queue_mode (vnm, qi, VNET_HW_IF_RX_MODE_DEFAULT);
349       vnet_hw_if_update_runtime_data (vnm, mif->hw_if_index);
350
351       if (rv)
352         memif_log_err
353           (mif, "Warning: unable to set rx mode for interface %d queue %d: "
354            "rc=%d", mif->hw_if_index, i, rv);
355       else
356         {
357           vnet_hw_if_rx_mode rxmode = vnet_hw_if_get_rx_queue_mode (vnm, qi);
358
359           if (rxmode == VNET_HW_IF_RX_MODE_POLLING)
360             mq->ring->flags |= MEMIF_RING_FLAG_MASK_INT;
361           else
362             vnet_hw_if_rx_queue_set_int_pending (vnm, qi);
363         }
364     }
365   /* *INDENT-ON* */
366
367   if (1 << max_log2_ring_sz > vec_len (mm->per_thread_data[0].desc_data))
368     {
369       memif_per_thread_data_t *ptd;
370
371       vec_foreach (ptd, mm->per_thread_data)
372         {
373           vec_validate_aligned (ptd->desc_data, pow2_mask (max_log2_ring_sz),
374                                 CLIB_CACHE_LINE_BYTES);
375           vec_validate_aligned (ptd->desc_len, pow2_mask (max_log2_ring_sz),
376                                 CLIB_CACHE_LINE_BYTES);
377           vec_validate_aligned (ptd->desc_status, pow2_mask (max_log2_ring_sz),
378                                 CLIB_CACHE_LINE_BYTES);
379         }
380     }
381   if (with_barrier)
382     vlib_worker_thread_barrier_release (vm);
383
384   mif->flags &= ~MEMIF_IF_FLAG_CONNECTING;
385   mif->flags |= MEMIF_IF_FLAG_CONNECTED;
386
387   vnet_hw_interface_set_flags (vnm, mif->hw_if_index,
388                                VNET_HW_INTERFACE_FLAG_LINK_UP);
389   return 0;
390
391 error:
392   if (with_barrier)
393     vlib_worker_thread_barrier_release (vm);
394   memif_log_err (mif, "%U", format_clib_error, err);
395   return err;
396 }
397
398 static_always_inline memif_ring_t *
399 memif_get_ring (memif_if_t * mif, memif_ring_type_t type, u16 ring_num)
400 {
401   if (vec_len (mif->regions) == 0)
402     return NULL;
403   void *p = mif->regions[0].shm;
404   int ring_size =
405     sizeof (memif_ring_t) +
406     sizeof (memif_desc_t) * (1 << mif->run.log2_ring_size);
407   p += (ring_num + type * mif->run.num_s2m_rings) * ring_size;
408
409   return (memif_ring_t *) p;
410 }
411
412 clib_error_t *
413 memif_init_regions_and_queues (memif_if_t * mif)
414 {
415   vlib_main_t *vm = vlib_get_main ();
416   memif_socket_file_t *msf;
417   memif_ring_t *ring = NULL;
418   int fd, i, j;
419   u64 buffer_offset;
420   memif_region_t *r;
421   clib_error_t *err;
422
423   ASSERT (vec_len (mif->regions) == 0);
424   vec_add2_aligned (mif->regions, r, 1, CLIB_CACHE_LINE_BYTES);
425
426   buffer_offset = (mif->run.num_s2m_rings + mif->run.num_m2s_rings) *
427     (sizeof (memif_ring_t) +
428      sizeof (memif_desc_t) * (1 << mif->run.log2_ring_size));
429
430   r->region_size = buffer_offset;
431
432   if ((mif->flags & MEMIF_IF_FLAG_ZERO_COPY) == 0)
433     r->region_size += mif->run.buffer_size * (1 << mif->run.log2_ring_size) *
434       (mif->run.num_s2m_rings + mif->run.num_m2s_rings);
435
436   if ((fd = clib_mem_vm_create_fd (CLIB_MEM_PAGE_SZ_DEFAULT, "%U region 0",
437                                    format_memif_device_name,
438                                    mif->dev_instance)) == -1)
439     {
440       err = clib_mem_get_last_error ();
441       goto error;
442     }
443
444   if ((ftruncate (fd, r->region_size)) == -1)
445     {
446       err = clib_error_return_unix (0, "ftruncate");
447       goto error;
448     }
449
450   msf = pool_elt_at_index (memif_main.socket_files, mif->socket_file_index);
451   r->shm = clib_mem_vm_map_shared (0, r->region_size, fd, 0, "memif%lu/%lu:0",
452                                    msf->socket_id, mif->id);
453
454   if (r->shm == CLIB_MEM_VM_MAP_FAILED)
455     {
456       err = clib_error_return_unix (0, "memif shared region map failed");
457       goto error;
458     }
459
460   r->fd = fd;
461
462   if (mif->flags & MEMIF_IF_FLAG_ZERO_COPY)
463     {
464       vlib_buffer_pool_t *bp;
465       /* *INDENT-OFF* */
466       vec_foreach (bp, vm->buffer_main->buffer_pools)
467         {
468           vlib_physmem_map_t *pm;
469           pm = vlib_physmem_get_map (vm, bp->physmem_map_index);
470           vec_add2_aligned (mif->regions, r, 1, CLIB_CACHE_LINE_BYTES);
471           r->fd = pm->fd;
472           r->region_size = pm->n_pages << pm->log2_page_size;
473           r->shm = pm->base;
474           r->is_external = 1;
475         }
476       /* *INDENT-ON* */
477     }
478
479   for (i = 0; i < mif->run.num_s2m_rings; i++)
480     {
481       ring = memif_get_ring (mif, MEMIF_RING_S2M, i);
482       ring->head = ring->tail = 0;
483       ring->cookie = MEMIF_COOKIE;
484
485       if (mif->flags & MEMIF_IF_FLAG_ZERO_COPY)
486         continue;
487
488       for (j = 0; j < (1 << mif->run.log2_ring_size); j++)
489         {
490           u16 slot = i * (1 << mif->run.log2_ring_size) + j;
491           ring->desc[j].region = 0;
492           ring->desc[j].offset =
493             buffer_offset + (u32) (slot * mif->run.buffer_size);
494           ring->desc[j].length = mif->run.buffer_size;
495         }
496     }
497   for (i = 0; i < mif->run.num_m2s_rings; i++)
498     {
499       ring = memif_get_ring (mif, MEMIF_RING_M2S, i);
500       ring->head = ring->tail = 0;
501       ring->cookie = MEMIF_COOKIE;
502
503       if (mif->flags & MEMIF_IF_FLAG_ZERO_COPY)
504         continue;
505
506       for (j = 0; j < (1 << mif->run.log2_ring_size); j++)
507         {
508           u16 slot =
509             (i + mif->run.num_s2m_rings) * (1 << mif->run.log2_ring_size) + j;
510           ring->desc[j].region = 0;
511           ring->desc[j].offset =
512             buffer_offset + (u32) (slot * mif->run.buffer_size);
513           ring->desc[j].length = mif->run.buffer_size;
514         }
515     }
516
517   ASSERT (mif->tx_queues == 0);
518   vec_validate_aligned (mif->tx_queues, mif->run.num_s2m_rings - 1,
519                         CLIB_CACHE_LINE_BYTES);
520
521   /* *INDENT-OFF* */
522   vec_foreach_index (i, mif->tx_queues)
523     {
524       memif_queue_t *mq = vec_elt_at_index (mif->tx_queues, i);
525       if ((mq->int_fd = eventfd (0, EFD_NONBLOCK)) < 0)
526         {
527           err = clib_error_return_unix (0, "eventfd[tx queue %u]", i);
528           goto error;
529         }
530
531       mq->int_clib_file_index = ~0;
532       mq->ring = memif_get_ring (mif, MEMIF_RING_S2M, i);
533       mq->log2_ring_size = mif->cfg.log2_ring_size;
534       mq->region = 0;
535       mq->offset = (void *) mq->ring - (void *) mif->regions[mq->region].shm;
536       mq->last_head = 0;
537       mq->type = MEMIF_RING_S2M;
538       if (mif->flags & MEMIF_IF_FLAG_ZERO_COPY)
539         vec_validate_aligned (mq->buffers, 1 << mq->log2_ring_size,
540                               CLIB_CACHE_LINE_BYTES);
541     }
542   /* *INDENT-ON* */
543
544   ASSERT (mif->rx_queues == 0);
545   vec_validate_aligned (mif->rx_queues, mif->run.num_m2s_rings - 1,
546                         CLIB_CACHE_LINE_BYTES);
547
548   /* *INDENT-OFF* */
549   vec_foreach_index (i, mif->rx_queues)
550     {
551       memif_queue_t *mq = vec_elt_at_index (mif->rx_queues, i);
552       if ((mq->int_fd = eventfd (0, EFD_NONBLOCK)) < 0)
553         {
554           err = clib_error_return_unix (0, "eventfd[rx queue %u]", i);
555           goto error;
556         }
557       mq->int_clib_file_index = ~0;
558       mq->ring = memif_get_ring (mif, MEMIF_RING_M2S, i);
559       mq->log2_ring_size = mif->cfg.log2_ring_size;
560       mq->region = 0;
561       mq->offset = (void *) mq->ring - (void *) mif->regions[mq->region].shm;
562       mq->last_head = 0;
563       mq->type = MEMIF_RING_M2S;
564       if (mif->flags & MEMIF_IF_FLAG_ZERO_COPY)
565         vec_validate_aligned (mq->buffers, 1 << mq->log2_ring_size,
566                               CLIB_CACHE_LINE_BYTES);
567     }
568   /* *INDENT-ON* */
569
570   return 0;
571
572 error:
573   memif_log_err (mif, "%U", format_clib_error, err);
574   return err;
575 }
576
577 static uword
578 memif_process (vlib_main_t * vm, vlib_node_runtime_t * rt, vlib_frame_t * f)
579 {
580   memif_main_t *mm = &memif_main;
581   memif_if_t *mif;
582   clib_socket_t *sock;
583   uword *event_data = 0, event_type;
584   u8 enabled = 0;
585   f64 start_time, last_run_duration = 0, now;
586   clib_error_t *err;
587
588   sock = clib_mem_alloc (sizeof (clib_socket_t));
589   clib_memset (sock, 0, sizeof (clib_socket_t));
590
591   while (1)
592     {
593       if (enabled)
594         vlib_process_wait_for_event_or_clock (vm, (f64) 3 -
595                                               last_run_duration);
596       else
597         vlib_process_wait_for_event (vm);
598
599       event_type = vlib_process_get_events (vm, &event_data);
600       vec_reset_length (event_data);
601
602       switch (event_type)
603         {
604         case ~0:
605           break;
606         case MEMIF_PROCESS_EVENT_START:
607           enabled = 1;
608           break;
609         case MEMIF_PROCESS_EVENT_STOP:
610           enabled = 0;
611           continue;
612         case MEMIF_PROCESS_EVENT_ADMIN_UP_DOWN:
613           break;
614         default:
615           ASSERT (0);
616         }
617
618       last_run_duration = start_time = vlib_time_now (vm);
619       /* *INDENT-OFF* */
620       pool_foreach (mif, mm->interfaces)
621          {
622           memif_socket_file_t * msf = vec_elt_at_index (mm->socket_files, mif->socket_file_index);
623           /* Allow no more than 10us without a pause */
624           now = vlib_time_now (vm);
625           if (now > start_time + 10e-6)
626             {
627               vlib_process_suspend (vm, 100e-6);        /* suspend for 100 us */
628               start_time = vlib_time_now (vm);
629             }
630
631           if ((mif->flags & MEMIF_IF_FLAG_ADMIN_UP) == 0)
632             continue;
633
634           if (mif->flags & MEMIF_IF_FLAG_CONNECTING)
635             continue;
636
637           if (mif->flags & MEMIF_IF_FLAG_CONNECTED)
638             continue;
639
640           if (mif->flags & MEMIF_IF_FLAG_IS_SLAVE)
641             {
642               clib_memset (sock, 0, sizeof(clib_socket_t));
643               sock->config = (char *) msf->filename;
644               sock->flags = CLIB_SOCKET_F_IS_CLIENT | CLIB_SOCKET_F_SEQPACKET |
645                             CLIB_SOCKET_F_BLOCKING;
646
647               if ((err = clib_socket_init (sock)))
648                 {
649                   clib_error_free (err);
650                 }
651               else
652                 {
653                   clib_file_t t = { 0 };
654
655                   t.read_function = memif_slave_conn_fd_read_ready;
656                   t.write_function = memif_slave_conn_fd_write_ready;
657                   t.error_function = memif_slave_conn_fd_error;
658                   t.file_descriptor = sock->fd;
659                   t.private_data = mif->dev_instance;
660                   memif_file_add (&sock->private_data, &t);
661                   t.description = format (0, "%U ctl",
662                                           format_memif_device_name,
663                                           mif->dev_instance);
664                   hash_set (msf->dev_instance_by_fd, sock->fd, mif->dev_instance);
665
666                   mif->flags |= MEMIF_IF_FLAG_CONNECTING;
667                   mif->sock = sock;
668                   sock = clib_mem_alloc (sizeof(clib_socket_t));
669                 }
670             }
671         }
672       /* *INDENT-ON* */
673       last_run_duration = vlib_time_now (vm) - last_run_duration;
674     }
675   return 0;
676 }
677
678 /* *INDENT-OFF* */
679 VLIB_REGISTER_NODE (memif_process_node,static) = {
680   .function = memif_process,
681   .type = VLIB_NODE_TYPE_PROCESS,
682   .name = "memif-process",
683 };
684 /* *INDENT-ON* */
685
686 static int
687 memif_add_socket_file (u32 sock_id, u8 * socket_filename)
688 {
689   memif_main_t *mm = &memif_main;
690   uword *p;
691   memif_socket_file_t *msf;
692
693   p = hash_get (mm->socket_file_index_by_sock_id, sock_id);
694   if (p)
695     {
696       msf = pool_elt_at_index (mm->socket_files, *p);
697       if (strcmp ((char *) msf->filename, (char *) socket_filename) == 0)
698         {
699           /* Silently accept identical "add". */
700           return 0;
701         }
702
703       /* But don't allow a direct add of a different filename. */
704       return VNET_API_ERROR_ENTRY_ALREADY_EXISTS;
705     }
706
707   pool_get (mm->socket_files, msf);
708   clib_memset (msf, 0, sizeof (memif_socket_file_t));
709
710   msf->filename = socket_filename;
711   msf->socket_id = sock_id;
712
713   hash_set (mm->socket_file_index_by_sock_id, sock_id,
714             msf - mm->socket_files);
715
716   return 0;
717 }
718
719 static int
720 memif_delete_socket_file (u32 sock_id)
721 {
722   memif_main_t *mm = &memif_main;
723   uword *p;
724   memif_socket_file_t *msf;
725
726   p = hash_get (mm->socket_file_index_by_sock_id, sock_id);
727   if (!p)
728     {
729       /* Don't delete non-existent entries. */
730       return VNET_API_ERROR_INVALID_ARGUMENT;
731     }
732
733   msf = pool_elt_at_index (mm->socket_files, *p);
734   if (msf->ref_cnt > 0)
735     {
736       return VNET_API_ERROR_UNEXPECTED_INTF_STATE;
737     }
738
739   vec_free (msf->filename);
740   pool_put (mm->socket_files, msf);
741
742   hash_unset (mm->socket_file_index_by_sock_id, sock_id);
743
744   return 0;
745 }
746
747 int
748 memif_socket_filename_add_del (u8 is_add, u32 sock_id, u8 * sock_filename)
749 {
750   char *dir = 0, *tmp;
751   u32 idx = 0;
752
753   /* allow adding socket id 0 */
754   if ((sock_id == 0 && is_add == 0) || sock_id == ~0)
755     {
756       return VNET_API_ERROR_INVALID_ARGUMENT;
757     }
758
759   if (is_add == 0)
760     {
761       return memif_delete_socket_file (sock_id);
762     }
763
764   if (sock_filename == 0 || sock_filename[0] == 0)
765     {
766       return VNET_API_ERROR_INVALID_ARGUMENT;
767     }
768
769   if (sock_filename[0] != '/')
770     {
771       clib_error_t *error;
772
773       /* copy runtime dir path */
774       vec_add (dir, vlib_unix_get_runtime_dir (),
775                strlen (vlib_unix_get_runtime_dir ()));
776       vec_add1 (dir, '/');
777
778       /* if sock_filename contains dirs, add them to path */
779       tmp = strrchr ((char *) sock_filename, '/');
780       if (tmp)
781         {
782           idx = tmp - (char *) sock_filename;
783           vec_add (dir, sock_filename, idx);
784         }
785
786       vec_add1 (dir, '\0');
787       /* create socket dir */
788       error = vlib_unix_recursive_mkdir (dir);
789       if (error)
790         {
791           clib_error_free (error);
792           return VNET_API_ERROR_SYSCALL_ERROR_1;
793         }
794
795       sock_filename = format (0, "%s/%s%c", vlib_unix_get_runtime_dir (),
796                               sock_filename, 0);
797     }
798   else
799     {
800       sock_filename = vec_dup (sock_filename);
801
802       /* check if directory exists */
803       tmp = strrchr ((char *) sock_filename, '/');
804       if (tmp)
805         {
806           idx = tmp - (char *) sock_filename;
807           vec_add (dir, sock_filename, idx);
808           vec_add1 (dir, '\0');
809         }
810
811       /* check dir existance and access rights for effective user/group IDs */
812       if ((dir == NULL)
813           ||
814           (faccessat ( /* ignored */ -1, dir, F_OK | R_OK | W_OK, AT_EACCESS)
815            < 0))
816         {
817           vec_free (dir);
818           return VNET_API_ERROR_INVALID_ARGUMENT;
819         }
820     }
821   vec_free (dir);
822
823   return memif_add_socket_file (sock_id, sock_filename);
824 }
825
826 int
827 memif_delete_if (vlib_main_t * vm, memif_if_t * mif)
828 {
829   vnet_main_t *vnm = vnet_get_main ();
830   memif_main_t *mm = &memif_main;
831   memif_socket_file_t *msf =
832     vec_elt_at_index (mm->socket_files, mif->socket_file_index);
833   clib_error_t *err;
834
835   mif->flags |= MEMIF_IF_FLAG_DELETING;
836   vec_free (mif->local_disc_string);
837   vec_free (mif->remote_disc_string);
838
839   /* bring down the interface */
840   vnet_hw_interface_set_flags (vnm, mif->hw_if_index, 0);
841   vnet_sw_interface_set_flags (vnm, mif->sw_if_index, 0);
842
843   err = clib_error_return (0, "interface deleted");
844   memif_disconnect (mif, err);
845   clib_error_free (err);
846
847   if (mif->hw_if_index != ~0)
848     {
849       /* remove the interface */
850       if (mif->mode == MEMIF_INTERFACE_MODE_IP)
851         vnet_delete_hw_interface (vnm, mif->hw_if_index);
852       else
853         ethernet_delete_interface (vnm, mif->hw_if_index);
854       mif->hw_if_index = ~0;
855     }
856
857   /* free interface data structures */
858   mhash_unset (&msf->dev_instance_by_id, &mif->id, 0);
859
860   /* remove socket file */
861   if (--(msf->ref_cnt) == 0)
862     {
863       if (msf->is_listener)
864         {
865           int i;
866           /* *INDENT-OFF* */
867           vec_foreach_index (i, msf->pending_clients)
868             memif_socket_close (msf->pending_clients + i);
869           /* *INDENT-ON* */
870           memif_socket_close (&msf->sock);
871           vec_free (msf->pending_clients);
872         }
873       mhash_free (&msf->dev_instance_by_id);
874       hash_free (msf->dev_instance_by_fd);
875       if (msf->sock)
876         {
877           err = clib_socket_close (msf->sock);
878           if (err)
879             {
880               memif_log_err (mif, "%U", format_clib_error, err);
881               clib_error_free (err);
882             }
883           clib_mem_free (msf->sock);
884         }
885     }
886
887   vec_free (mif->local_disc_string);
888   clib_memset (mif, 0, sizeof (*mif));
889   pool_put (mm->interfaces, mif);
890
891   if (pool_elts (mm->interfaces) == 0)
892     vlib_process_signal_event (vm, memif_process_node.index,
893                                MEMIF_PROCESS_EVENT_STOP, 0);
894
895   return 0;
896 }
897
898 /* *INDENT-OFF* */
899 VNET_HW_INTERFACE_CLASS (memif_ip_hw_if_class, static) = {
900   .name = "memif-ip",
901   .flags = VNET_HW_INTERFACE_CLASS_FLAG_P2P,
902   .tx_hash_fn_type = VNET_HASH_FN_TYPE_IP,
903 };
904 /* *INDENT-ON* */
905
906 int
907 memif_create_if (vlib_main_t * vm, memif_create_if_args_t * args)
908 {
909   memif_main_t *mm = &memif_main;
910   vlib_thread_main_t *tm = vlib_get_thread_main ();
911   vnet_main_t *vnm = vnet_get_main ();
912   vnet_eth_interface_registration_t eir = {};
913   memif_if_t *mif = 0;
914   vnet_sw_interface_t *sw;
915   clib_error_t *error = 0;
916   int ret = 0;
917   uword *p;
918   memif_socket_file_t *msf = 0;
919   int rv = 0;
920
921   p = hash_get (mm->socket_file_index_by_sock_id, args->socket_id);
922   if (p == 0)
923     {
924       rv = VNET_API_ERROR_INVALID_ARGUMENT;
925       goto done;
926     }
927
928   msf = vec_elt_at_index (mm->socket_files, p[0]);
929
930   /* existing socket file can be either master or slave but cannot be both */
931   if (msf->ref_cnt > 0)
932     {
933       if ((!msf->is_listener != !args->is_master))
934         {
935           rv = VNET_API_ERROR_SUBIF_ALREADY_EXISTS;
936           goto done;
937         }
938
939       p = mhash_get (&msf->dev_instance_by_id, &args->id);
940       if (p)
941         {
942           rv = VNET_API_ERROR_SUBIF_ALREADY_EXISTS;
943           goto done;
944         }
945     }
946
947   /* Create new socket file */
948   if (msf->ref_cnt == 0)
949     {
950       struct stat file_stat;
951
952       /* If we are creating listener make sure file doesn't exist or if it
953        * exists thn delete it if it is old socket file */
954       if (args->is_master && (stat ((char *) msf->filename, &file_stat) == 0))
955         {
956           if (S_ISSOCK (file_stat.st_mode))
957             {
958               unlink ((char *) msf->filename);
959             }
960           else
961             {
962               error = clib_error_return (0, "File exists for %s",
963                                          msf->filename);
964               rv = VNET_API_ERROR_VALUE_EXIST;
965               goto done;
966             }
967         }
968
969       mhash_init (&msf->dev_instance_by_id, sizeof (uword),
970                   sizeof (memif_interface_id_t));
971       msf->dev_instance_by_fd = hash_create (0, sizeof (uword));
972       msf->is_listener = (args->is_master != 0);
973
974       memif_log_debug (0, "initializing socket file %s", msf->filename);
975     }
976
977   if (mm->per_thread_data == 0)
978     {
979       int i;
980
981       vec_validate_aligned (mm->per_thread_data, tm->n_vlib_mains - 1,
982                             CLIB_CACHE_LINE_BYTES);
983
984       for (i = 0; i < tm->n_vlib_mains; i++)
985         {
986           memif_per_thread_data_t *ptd =
987             vec_elt_at_index (mm->per_thread_data, i);
988           vlib_buffer_t *bt = &ptd->buffer_template;
989           clib_memset (bt, 0, sizeof (vlib_buffer_t));
990           bt->flags = VLIB_BUFFER_TOTAL_LENGTH_VALID;
991           bt->total_length_not_including_first_buffer = 0;
992           vnet_buffer (bt)->sw_if_index[VLIB_TX] = (u32) ~ 0;
993
994           vec_validate_aligned (ptd->copy_ops, 0, CLIB_CACHE_LINE_BYTES);
995           vec_reset_length (ptd->copy_ops);
996           vec_validate_aligned (ptd->buffers, 0, CLIB_CACHE_LINE_BYTES);
997           vec_reset_length (ptd->buffers);
998         }
999     }
1000
1001   pool_get (mm->interfaces, mif);
1002   clib_memset (mif, 0, sizeof (*mif));
1003   mif->dev_instance = mif - mm->interfaces;
1004   mif->socket_file_index = msf - mm->socket_files;
1005   mif->id = args->id;
1006   mif->sw_if_index = mif->hw_if_index = mif->per_interface_next_index = ~0;
1007   mif->mode = args->mode;
1008   if (args->secret)
1009     mif->secret = vec_dup (args->secret);
1010
1011   if (mif->mode == MEMIF_INTERFACE_MODE_ETHERNET)
1012     {
1013
1014       if (!args->hw_addr_set)
1015         {
1016           f64 now = vlib_time_now (vm);
1017           u32 rnd;
1018           rnd = (u32) (now * 1e6);
1019           rnd = random_u32 (&rnd);
1020
1021           memcpy (args->hw_addr + 2, &rnd, sizeof (rnd));
1022           args->hw_addr[0] = 2;
1023           args->hw_addr[1] = 0xfe;
1024         }
1025
1026       eir.dev_class_index = memif_device_class.index;
1027       eir.dev_instance = mif->dev_instance;
1028       eir.address = args->hw_addr;
1029       eir.cb.flag_change = memif_eth_flag_change;
1030       eir.cb.set_max_frame_size = memif_eth_set_max_frame_size;
1031       mif->hw_if_index = vnet_eth_register_interface (vnm, &eir);
1032     }
1033   else if (mif->mode == MEMIF_INTERFACE_MODE_IP)
1034     {
1035       mif->hw_if_index =
1036         vnet_register_interface (vnm, memif_device_class.index,
1037                                  mif->dev_instance,
1038                                  memif_ip_hw_if_class.index,
1039                                  mif->dev_instance);
1040     }
1041   else
1042     error = clib_error_return (0, "unsupported interface mode");
1043
1044   if (error)
1045     {
1046       ret = VNET_API_ERROR_SYSCALL_ERROR_2;
1047       goto error;
1048     }
1049
1050   sw = vnet_get_hw_sw_interface (vnm, mif->hw_if_index);
1051   mif->sw_if_index = sw->sw_if_index;
1052
1053   mif->cfg.log2_ring_size = args->log2_ring_size;
1054   mif->cfg.buffer_size = args->buffer_size;
1055   mif->cfg.num_s2m_rings =
1056     args->is_master ? args->rx_queues : args->tx_queues;
1057   mif->cfg.num_m2s_rings =
1058     args->is_master ? args->tx_queues : args->rx_queues;
1059
1060   args->sw_if_index = mif->sw_if_index;
1061
1062   /* If this is new one, start listening */
1063   if (msf->is_listener && msf->ref_cnt == 0)
1064     {
1065       struct stat file_stat;
1066       clib_socket_t *s = clib_mem_alloc (sizeof (clib_socket_t));
1067
1068       ASSERT (msf->sock == 0);
1069       msf->sock = s;
1070
1071       clib_memset (s, 0, sizeof (clib_socket_t));
1072       s->config = (char *) msf->filename;
1073       s->flags = CLIB_SOCKET_F_IS_SERVER |
1074         CLIB_SOCKET_F_ALLOW_GROUP_WRITE |
1075         CLIB_SOCKET_F_SEQPACKET | CLIB_SOCKET_F_PASSCRED;
1076
1077       if ((error = clib_socket_init (s)))
1078         {
1079           ret = VNET_API_ERROR_SYSCALL_ERROR_4;
1080           goto error;
1081         }
1082
1083       if (stat ((char *) msf->filename, &file_stat) == -1)
1084         {
1085           ret = VNET_API_ERROR_SYSCALL_ERROR_8;
1086           goto error;
1087         }
1088
1089       clib_file_t template = { 0 };
1090       template.read_function = memif_conn_fd_accept_ready;
1091       template.file_descriptor = msf->sock->fd;
1092       template.private_data = mif->socket_file_index;
1093       template.description = format (0, "memif listener %s", msf->filename);
1094       memif_file_add (&msf->sock->private_data, &template);
1095     }
1096
1097   msf->ref_cnt++;
1098
1099   if (args->is_master == 0)
1100     {
1101       mif->flags |= MEMIF_IF_FLAG_IS_SLAVE;
1102       if (args->is_zero_copy)
1103         mif->flags |= MEMIF_IF_FLAG_ZERO_COPY;
1104     }
1105
1106   vnet_hw_if_set_caps (vnm, mif->hw_if_index, VNET_HW_IF_CAP_INT_MODE);
1107   vnet_hw_if_set_input_node (vnm, mif->hw_if_index, memif_input_node.index);
1108   mhash_set (&msf->dev_instance_by_id, &mif->id, mif->dev_instance, 0);
1109
1110   if (pool_elts (mm->interfaces) == 1)
1111     {
1112       vlib_process_signal_event (vm, memif_process_node.index,
1113                                  MEMIF_PROCESS_EVENT_START, 0);
1114     }
1115   goto done;
1116
1117 error:
1118   memif_delete_if (vm, mif);
1119   if (error)
1120     {
1121       memif_log_err (mif, "%U", format_clib_error, error);
1122       clib_error_free (error);
1123     }
1124   return ret;
1125
1126 done:
1127   return rv;
1128 }
1129
1130 clib_error_t *
1131 memif_interface_admin_up_down (vnet_main_t *vnm, u32 hw_if_index, u32 flags)
1132 {
1133   memif_main_t *mm = &memif_main;
1134   vnet_hw_interface_t *hw = vnet_get_hw_interface (vnm, hw_if_index);
1135   memif_if_t *mif = pool_elt_at_index (mm->interfaces, hw->dev_instance);
1136   static clib_error_t *error = 0;
1137
1138   if (flags & VNET_SW_INTERFACE_FLAG_ADMIN_UP)
1139     {
1140       if (mif->flags & MEMIF_IF_FLAG_CONNECTED)
1141         {
1142           vnet_hw_interface_set_flags (vnm, mif->hw_if_index,
1143                                        VNET_HW_INTERFACE_FLAG_LINK_UP);
1144         }
1145       mif->flags |= MEMIF_IF_FLAG_ADMIN_UP;
1146     }
1147   else
1148     mif->flags &= ~MEMIF_IF_FLAG_ADMIN_UP;
1149
1150   vlib_process_signal_event (vnm->vlib_main, memif_process_node.index,
1151                              MEMIF_PROCESS_EVENT_ADMIN_UP_DOWN, 0);
1152   return error;
1153 }
1154
1155 static clib_error_t *
1156 memif_init (vlib_main_t * vm)
1157 {
1158   memif_main_t *mm = &memif_main;
1159
1160   clib_memset (mm, 0, sizeof (memif_main_t));
1161
1162   mm->log_class = vlib_log_register_class ("memif_plugin", 0);
1163   memif_log_debug (0, "initialized");
1164
1165   /* initialize binary API */
1166   memif_plugin_api_hookup (vm);
1167
1168   /*
1169    * Pre-stuff socket filename pool with a non-modifieable mapping
1170    * for socket-id 0 to MEMIF_DEFAULT_SOCKET_FILENAME in the
1171    * default run-time directory.
1172    */
1173   memif_socket_filename_add_del (1, 0, (u8 *) MEMIF_DEFAULT_SOCKET_FILENAME);
1174
1175   return 0;
1176 }
1177
1178 VLIB_INIT_FUNCTION (memif_init);
1179
1180 /* *INDENT-OFF* */
1181 VLIB_PLUGIN_REGISTER () = {
1182     .version = VPP_BUILD_VER,
1183     .description = "Packet Memory Interface (memif) -- Experimental",
1184 };
1185 /* *INDENT-ON* */
1186
1187 /*
1188  * fd.io coding-style-patch-verification: ON
1189  *
1190  * Local Variables:
1191  * eval: (c-set-style "gnu")
1192  * End:
1193  */