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