svm/ssvm.c

   1 /*
   2  * Copyright (c) 2015 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15 #include "ssvm.h"
  16
  17 int
  18 ssvm_master_init (ssvm_private_t * ssvm, u32 master_index)
  19 {
  20   int ssvm_fd;
  21   u8 *ssvm_filename;
  22   u8 junk = 0;
  23   int flags;
  24   ssvm_shared_header_t *sh;
  25   u64 ticks = clib_cpu_time_now ();
  26   u64 randomize_baseva;
  27   void *oldheap;
  28
  29   if (ssvm->ssvm_size == 0)
  30     return SSVM_API_ERROR_NO_SIZE;
  31
  32   ssvm_filename = format (0, "/dev/shm/%s%c", ssvm->name, 0);
  33
  34   unlink ((char *) ssvm_filename);
  35
  36   vec_free (ssvm_filename);
  37
  38   ssvm_fd = shm_open ((char *) ssvm->name, O_RDWR | O_CREAT | O_EXCL, 0777);
  39
  40   if (ssvm_fd < 0)
  41     {
  42       clib_unix_warning ("create segment '%s'", ssvm->name);
  43       return SSVM_API_ERROR_CREATE_FAILURE;
  44     }
  45
  46   lseek (ssvm_fd, ssvm->ssvm_size, SEEK_SET);
  47   if (write (ssvm_fd, &junk, 1) != 1)
  48     {
  49       clib_unix_warning ("set ssvm size");
  50       close (ssvm_fd);
  51       return SSVM_API_ERROR_SET_SIZE;
  52     }
  53
  54   flags = MAP_SHARED;
  55   if (ssvm->requested_va)
  56     flags |= MAP_FIXED;
  57
  58   randomize_baseva = (ticks & 15) * MMAP_PAGESIZE;
  59
  60   if (ssvm->requested_va)
  61     ssvm->requested_va += randomize_baseva;
  62
  63   sh = ssvm->sh =
  64     (ssvm_shared_header_t *) mmap ((void *) ssvm->requested_va,
  65                                    ssvm->ssvm_size, PROT_READ | PROT_WRITE,
  66                                    flags, ssvm_fd, 0);
  67
  68   if (ssvm->sh == MAP_FAILED)
  69     {
  70       clib_unix_warning ("mmap");
  71       close (ssvm_fd);
  72       return SSVM_API_ERROR_MMAP;
  73     }
  74
  75   close (ssvm_fd);
  76
  77   ssvm->my_pid = getpid ();
  78   sh->master_pid = ssvm->my_pid;
  79   sh->ssvm_size = ssvm->ssvm_size;
  80   sh->heap = mheap_alloc_with_flags
  81     (((u8 *) sh) + MMAP_PAGESIZE, ssvm->ssvm_size - MMAP_PAGESIZE,
  82      MHEAP_FLAG_DISABLE_VM | MHEAP_FLAG_THREAD_SAFE);
  83
  84   sh->ssvm_va = pointer_to_uword (sh);
  85   sh->master_index = master_index;
  86
  87   oldheap = ssvm_push_heap (sh);
  88   sh->name = format (0, "%s%c", ssvm->name, 0);
  89   ssvm_pop_heap (oldheap);
  90
  91   ssvm->i_am_master = 1;
  92
  93   /* The application has to set set sh->ready... */
  94   return 0;
  95 }
  96
  97 int
  98 ssvm_slave_init (ssvm_private_t * ssvm, int timeout_in_seconds)
  99 {
 100   struct stat stat;
 101   int ssvm_fd = -1;
 102   ssvm_shared_header_t *sh;
 103
 104   ssvm->i_am_master = 0;
 105
 106   while (timeout_in_seconds-- > 0)
 107     {
 108       if (ssvm_fd < 0)
 109         ssvm_fd = shm_open ((char *) ssvm->name, O_RDWR, 0777);
 110       if (ssvm_fd < 0)
 111         {
 112           sleep (1);
 113           continue;
 114         }
 115       if (fstat (ssvm_fd, &stat) < 0)
 116         {
 117           sleep (1);
 118           continue;
 119         }
 120
 121       if (stat.st_size > 0)
 122         goto map_it;
 123     }
 124   clib_warning ("slave timeout");
 125   return SSVM_API_ERROR_SLAVE_TIMEOUT;
 126
 127 map_it:
 128   sh = (void *) mmap (0, MMAP_PAGESIZE, PROT_READ | PROT_WRITE, MAP_SHARED,
 129                       ssvm_fd, 0);
 130   if (sh == MAP_FAILED)
 131     {
 132       clib_unix_warning ("slave research mmap");
 133       close (ssvm_fd);
 134       return SSVM_API_ERROR_MMAP;
 135     }
 136
 137   while (timeout_in_seconds-- > 0)
 138     {
 139       if (sh->ready)
 140         goto re_map_it;
 141     }
 142   close (ssvm_fd);
 143   munmap (sh, MMAP_PAGESIZE);
 144   clib_warning ("slave timeout 2");
 145   return SSVM_API_ERROR_SLAVE_TIMEOUT;
 146
 147 re_map_it:
 148   ssvm->requested_va = (u64) sh->ssvm_va;
 149   ssvm->ssvm_size = sh->ssvm_size;
 150   munmap (sh, MMAP_PAGESIZE);
 151
 152   sh = ssvm->sh = (void *) mmap ((void *) ssvm->requested_va, ssvm->ssvm_size,
 153                                  PROT_READ | PROT_WRITE,
 154                                  MAP_SHARED | MAP_FIXED, ssvm_fd, 0);
 155
 156   if (sh == MAP_FAILED)
 157     {
 158       clib_unix_warning ("slave final mmap");
 159       close (ssvm_fd);
 160       return SSVM_API_ERROR_MMAP;
 161     }
 162   sh->slave_pid = getpid ();
 163   return 0;
 164 }
 165
 166 /*
 167  * fd.io coding-style-patch-verification: ON
 168  *
 169  * Local Variables:
 170  * eval: (c-set-style "gnu")
 171  * End:
 172  */