src/svm/fifo_segment.c

   1 /*
   2  * Copyright (c) 2016-2019 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
  16 #include <svm/fifo_segment.h>
  17
  18 static inline fifo_segment_slice_t *
  19 fsh_slice_get (fifo_segment_header_t * fsh, u32 slice_index)
  20 {
  21   return &fsh->slices[slice_index];
  22 }
  23
  24 static char *fifo_segment_mem_status_strings[] = {
  25 #define _(sym,str) str,
  26   foreach_segment_mem_status
  27 #undef _
  28 };
  29
  30 /**
  31  * Fifo segment free space
  32  *
  33  * Queries the underlying memory manager, dlmalloc, for free space. Since this
  34  * ends up walking the internal data structures, it should not be called
  35  * indiscriminately.
  36  *
  37  * @param fs            fifo segment
  38  * @return              number of free bytes
  39  */
  40 static uword
  41 fsh_free_space (fifo_segment_header_t * fsh)
  42 {
  43   return clib_mem_get_heap_free_space (fsh->ssvm_sh->heap);
  44 }
  45
  46 static inline void
  47 fsh_free_bytes_sub (fifo_segment_header_t * fsh, int size)
  48 {
  49   clib_atomic_fetch_sub_rel (&fsh->n_free_bytes, size);
  50 }
  51
  52 static inline uword
  53 fsh_n_free_bytes (fifo_segment_header_t * fsh)
  54 {
  55   uword n_free = clib_atomic_load_relax_n (&fsh->n_free_bytes);
  56   return n_free > fsh->n_reserved_bytes ? n_free - fsh->n_reserved_bytes : 0;
  57 }
  58
  59 static inline void
  60 fsh_update_free_bytes (fifo_segment_header_t * fsh)
  61 {
  62   clib_atomic_store_rel_n (&fsh->n_free_bytes, fsh_free_space (fsh));
  63 }
  64
  65 static inline void
  66 fsh_cached_bytes_add (fifo_segment_header_t * fsh, int size)
  67 {
  68   clib_atomic_fetch_add_rel (&fsh->n_cached_bytes, size);
  69 }
  70
  71 static inline void
  72 fsh_cached_bytes_sub (fifo_segment_header_t * fsh, int size)
  73 {
  74   clib_atomic_fetch_sub_rel (&fsh->n_cached_bytes, size);
  75 }
  76
  77 static inline uword
  78 fsh_n_cached_bytes (fifo_segment_header_t * fsh)
  79 {
  80   uword n_cached = clib_atomic_load_relax_n (&fsh->n_cached_bytes);
  81   return n_cached;
  82 }
  83
  84 static inline void
  85 fsh_active_fifos_update (fifo_segment_header_t * fsh, int inc)
  86 {
  87   clib_atomic_fetch_add_rel (&fsh->n_active_fifos, inc);
  88 }
  89
  90 static inline u32
  91 fsh_n_active_fifos (fifo_segment_header_t * fsh)
  92 {
  93   return clib_atomic_load_relax_n (&fsh->n_active_fifos);
  94 }
  95
  96 static inline uword
  97 fsh_virtual_mem (fifo_segment_header_t * fsh)
  98 {
  99   fifo_segment_slice_t *fss;
 100   uword total_vm = 0;
 101   int i;
 102
 103   for (i = 0; i < fsh->n_slices; i++)
 104     {
 105       fss = fsh_slice_get (fsh, i);
 106       total_vm += clib_atomic_load_relax_n (&fss->virtual_mem);
 107     }
 108   return total_vm;
 109 }
 110
 111 void
 112 fsh_virtual_mem_update (fifo_segment_header_t * fsh, u32 slice_index,
 113                         int n_bytes)
 114 {
 115   fifo_segment_slice_t *fss = fsh_slice_get (fsh, slice_index);
 116   fss->virtual_mem += n_bytes;
 117 }
 118
 119 static void
 120 fsh_check_mem (fifo_segment_header_t * fsh)
 121 {
 122   uword thresh;
 123
 124   if (fsh->flags & FIFO_SEGMENT_F_MEM_LIMIT)
 125     return;
 126
 127   thresh = clib_max (0.01 * fsh->ssvm_sh->ssvm_size,
 128                      2 * fsh->n_reserved_bytes);
 129   if (fsh->n_free_bytes > thresh)
 130     return;
 131
 132   fsh->flags |= FIFO_SEGMENT_F_MEM_LIMIT;
 133   fsh_update_free_bytes (fsh);
 134 }
 135
 136 /**
 137  * Initialize fifo segment shared header
 138  */
 139 int
 140 fifo_segment_init (fifo_segment_t * fs)
 141 {
 142   fifo_segment_header_t *fsh;
 143   fifo_segment_slice_t *fss;
 144   ssvm_shared_header_t *sh;
 145   u32 max_chunk_sz;
 146   uword max_fifo;
 147   void *oldheap;
 148   int i;
 149
 150   sh = fs->ssvm.sh;
 151   oldheap = ssvm_push_heap (sh);
 152
 153   /*
 154    * Manually align the fifo segment header to sizeof(uword) = 8 bytes.
 155    * Long story made short: the "process-private" fifo segment
 156    * is allocated from the main heap, not mmapped. dlmalloc
 157    * only guarantees 4-byte alignment, and on aarch64
 158    * the fsh can end up 4-byte but not 8-byte aligned.
 159    * That eventually causes the atomic op in fifo_segment_update_free_bytes
 160    * to backfire.
 161    */
 162   fsh = clib_mem_alloc_aligned (sizeof (*fsh), sizeof (uword));
 163   clib_memset (fsh, 0, sizeof (*fsh));
 164   fs->h = sh->opaque[0] = fsh;
 165   fs->n_slices = clib_max (fs->n_slices, 1);
 166
 167   fsh->ssvm_sh = fs->ssvm.sh;
 168   fsh->n_slices = fs->n_slices;
 169   max_fifo = clib_min ((fsh_free_space (fsh) - 4096) / 2,
 170                        FIFO_SEGMENT_MAX_FIFO_SIZE);
 171   fsh->max_log2_chunk_size = max_log2 (max_fifo);
 172
 173   fsh->slices = clib_mem_alloc (sizeof (*fss) * fs->n_slices);
 174   clib_memset (fsh->slices, 0, sizeof (*fss) * fs->n_slices);
 175   max_chunk_sz = fsh->max_log2_chunk_size - FIFO_SEGMENT_MIN_LOG2_FIFO_SIZE;
 176
 177   for (i = 0; i < fs->n_slices; i++)
 178     {
 179       fss = fsh_slice_get (fsh, i);
 180       vec_validate_init_empty (fss->free_chunks, max_chunk_sz, 0);
 181       vec_validate_init_empty (fss->num_chunks, max_chunk_sz, 0);
 182       clib_spinlock_init (&fss->chunk_lock);
 183     }
 184
 185   ssvm_pop_heap (oldheap);
 186
 187   fsh->n_free_bytes = fsh_free_space (fsh);
 188   fsh->n_cached_bytes = 0;
 189   fsh->n_reserved_bytes = clib_min (0.01 * fsh->n_free_bytes, 256 << 10);
 190   sh->ready = 1;
 191   return (0);
 192 }
 193
 194 /**
 195  * Create a fifo segment and initialize as master
 196  */
 197 int
 198 fifo_segment_create (fifo_segment_main_t * sm, fifo_segment_create_args_t * a)
 199 {
 200   fifo_segment_t *fs;
 201   uword baseva;
 202   int rv;
 203
 204   /* Allocate a fresh segment */
 205   pool_get_zero (sm->segments, fs);
 206
 207   baseva = a->segment_type == SSVM_SEGMENT_PRIVATE ? ~0ULL : sm->next_baseva;
 208   fs->ssvm.ssvm_size = a->segment_size;
 209   fs->ssvm.is_server = 1;
 210   fs->ssvm.my_pid = getpid ();
 211   fs->ssvm.name = format (0, "%s%c", a->segment_name, 0);
 212   fs->ssvm.requested_va = baseva;
 213
 214   if ((rv = ssvm_server_init (&fs->ssvm, a->segment_type)))
 215     {
 216       pool_put (sm->segments, fs);
 217       return (rv);
 218     }
 219
 220   /* Note: requested_va updated due to seg base addr randomization */
 221   sm->next_baseva = fs->ssvm.sh->ssvm_va + fs->ssvm.ssvm_size;
 222
 223   fifo_segment_init (fs);
 224   vec_add1 (a->new_segment_indices, fs - sm->segments);
 225   return (0);
 226 }
 227
 228 /**
 229  * Attach as slave to a fifo segment
 230  */
 231 int
 232 fifo_segment_attach (fifo_segment_main_t * sm, fifo_segment_create_args_t * a)
 233 {
 234   fifo_segment_t *fs;
 235   int rv;
 236
 237   pool_get_zero (sm->segments, fs);
 238
 239   fs->ssvm.ssvm_size = a->segment_size;
 240   fs->ssvm.my_pid = getpid ();
 241   fs->ssvm.name = format (0, "%s%c", a->segment_name, 0);
 242   fs->ssvm.requested_va = sm->next_baseva;
 243   if (a->segment_type == SSVM_SEGMENT_MEMFD)
 244     fs->ssvm.fd = a->memfd_fd;
 245   else
 246     fs->ssvm.attach_timeout = sm->timeout_in_seconds;
 247
 248   if ((rv = ssvm_client_init (&fs->ssvm, a->segment_type)))
 249     {
 250       _vec_len (fs) = vec_len (fs) - 1;
 251       return (rv);
 252     }
 253
 254   /* Fish the segment header */
 255   fs->h = fs->ssvm.sh->opaque[0];
 256
 257   vec_add1 (a->new_segment_indices, fs - sm->segments);
 258   return (0);
 259 }
 260
 261 void
 262 fifo_segment_delete (fifo_segment_main_t * sm, fifo_segment_t * s)
 263 {
 264   ssvm_delete (&s->ssvm);
 265   clib_memset (s, 0xfe, sizeof (*s));
 266   pool_put (sm->segments, s);
 267 }
 268
 269 u32
 270 fifo_segment_index (fifo_segment_main_t * sm, fifo_segment_t * s)
 271 {
 272   return s - sm->segments;
 273 }
 274
 275 fifo_segment_t *
 276 fifo_segment_get_segment (fifo_segment_main_t * sm, u32 segment_index)
 277 {
 278   return pool_elt_at_index (sm->segments, segment_index);
 279 }
 280
 281 void
 282 fifo_segment_info (fifo_segment_t * seg, char **address, size_t * size)
 283 {
 284   *address = (char *) seg->ssvm.sh->ssvm_va;
 285   *size = seg->ssvm.ssvm_size;
 286 }
 287
 288 void
 289 fifo_segment_main_init (fifo_segment_main_t * sm, u64 baseva,
 290                         u32 timeout_in_seconds)
 291 {
 292   sm->next_baseva = baseva;
 293   sm->timeout_in_seconds = timeout_in_seconds;
 294 }
 295
 296 static inline u32
 297 fs_freelist_for_size (u32 size)
 298 {
 299   if (PREDICT_FALSE (size < FIFO_SEGMENT_MIN_FIFO_SIZE))
 300     return 0;
 301   return max_log2 (size) - FIFO_SEGMENT_MIN_LOG2_FIFO_SIZE;
 302 }
 303
 304 static inline u32
 305 fs_freelist_index_to_size (u32 fl_index)
 306 {
 307   return 1 << (fl_index + FIFO_SEGMENT_MIN_LOG2_FIFO_SIZE);
 308 }
 309
 310 static inline int
 311 fs_chunk_size_is_valid (fifo_segment_header_t * fsh, u32 size)
 312 {
 313   /*
 314    * 4K minimum. It's not likely that anything good will happen
 315    * with a smaller FIFO.
 316    */
 317   return size >= FIFO_SEGMENT_MIN_FIFO_SIZE
 318     && size <= (1ULL << fsh->max_log2_chunk_size);
 319 }
 320
 321 static svm_fifo_t *
 322 fs_try_alloc_fifo_freelist (fifo_segment_slice_t * fss, u32 fl_index)
 323 {
 324   svm_fifo_chunk_t *c;
 325   svm_fifo_t *f;
 326
 327   f = fss->free_fifos;
 328   c = fss->free_chunks[fl_index];
 329
 330   if (!f || !c)
 331     return 0;
 332
 333   fss->free_fifos = f->next;
 334   fss->free_chunks[fl_index] = c->next;
 335   c->next = 0;
 336   c->start_byte = 0;
 337   memset (f, 0, sizeof (*f));
 338   f->start_chunk = c;
 339   f->end_chunk = c;
 340
 341   fss->n_fl_chunk_bytes -= fs_freelist_index_to_size (fl_index);
 342   return f;
 343 }
 344
 345 svm_fifo_chunk_t *
 346 fs_try_alloc_multi_chunk (fifo_segment_header_t * fsh,
 347                           fifo_segment_slice_t * fss, u32 data_bytes)
 348 {
 349   u32 fl_index, fl_size, n_alloc = 0, req_bytes = data_bytes;
 350   svm_fifo_chunk_t *c, *first = 0, *next;
 351
 352   fl_index = fs_freelist_for_size (req_bytes);
 353   if (fl_index > 0)
 354     fl_index -= 1;
 355
 356   fl_size = fs_freelist_index_to_size (fl_index);
 357
 358   while (req_bytes)
 359     {
 360       c = fss->free_chunks[fl_index];
 361       if (c)
 362         {
 363           fss->free_chunks[fl_index] = c->next;
 364           c->next = first;
 365           first = c;
 366           n_alloc += fl_size;
 367           req_bytes -= clib_min (fl_size, req_bytes);
 368         }
 369       else
 370         {
 371           /* Failed to allocate with smaller chunks */
 372           if (fl_index == 0)
 373             {
 374               /* free all chunks if any allocated */
 375               c = first;
 376               while (c)
 377                 {
 378                   fl_index = fs_freelist_for_size (c->length);
 379                   fl_size = fs_freelist_index_to_size (fl_index);
 380                   next = c->next;
 381                   c->next = fss->free_chunks[fl_index];
 382                   fss->free_chunks[fl_index] = c;
 383                   fss->n_fl_chunk_bytes += fl_size;
 384                   c = next;
 385                 }
 386               n_alloc = 0;
 387               first = 0;
 388               fl_index = fs_freelist_for_size (data_bytes);
 389               if (fss->free_chunks[fl_index + 1])
 390                 {
 391                   fl_index += 1;
 392                   fl_size = fs_freelist_index_to_size (fl_index);
 393                   continue;
 394                 }
 395
 396               return 0;
 397             }
 398           fl_index -= 1;
 399           fl_size = fl_size >> 1;
 400         }
 401     }
 402
 403   fss->n_fl_chunk_bytes -= n_alloc;
 404   fsh_cached_bytes_sub (fsh, n_alloc);
 405   return first;
 406 }
 407
 408 static svm_fifo_t *
 409 fs_try_alloc_fifo_freelist_multi_chunk (fifo_segment_header_t * fsh,
 410                                         fifo_segment_slice_t * fss,
 411                                         u32 data_bytes)
 412 {
 413   svm_fifo_chunk_t *c, *first = 0, *last = 0, *next;
 414   u32 fl_index, fl_size, n_alloc = 0;
 415   svm_fifo_t *f;
 416
 417   f = fss->free_fifos;
 418   if (!f)
 419     {
 420       if (PREDICT_FALSE (fsh_n_free_bytes (fsh) < sizeof (svm_fifo_t)))
 421         return 0;
 422
 423       void *oldheap = ssvm_push_heap (fsh->ssvm_sh);
 424       f = clib_mem_alloc_aligned_or_null (sizeof (*f), CLIB_CACHE_LINE_BYTES);
 425       ssvm_pop_heap (oldheap);
 426       if (!f)
 427         return 0;
 428       memset (f, 0, sizeof (*f));
 429       fsh_free_bytes_sub (fsh, sizeof (*f));
 430     }
 431   else
 432     {
 433       fss->free_fifos = f->next;
 434     }
 435
 436   fl_index = fs_freelist_for_size (data_bytes);
 437   if (fl_index > 0)
 438     fl_index -= 1;
 439
 440   fl_size = fs_freelist_index_to_size (fl_index);
 441
 442   while (data_bytes)
 443     {
 444       c = fss->free_chunks[fl_index];
 445       if (c)
 446         {
 447           fss->free_chunks[fl_index] = c->next;
 448           if (!last)
 449             last = c;
 450           c->next = first;
 451           first = c;
 452           n_alloc += fl_size;
 453           data_bytes -= clib_min (fl_size, data_bytes);
 454         }
 455       else
 456         {
 457           /* Failed to allocate with smaller chunks */
 458           if (fl_index == 0)
 459             {
 460               /* free all chunks if any allocated */
 461               c = first;
 462               while (c)
 463                 {
 464                   fl_index = fs_freelist_for_size (c->length);
 465                   fl_size = fs_freelist_index_to_size (fl_index);
 466                   next = c->next;
 467                   c->next = fss->free_chunks[fl_index];
 468                   fss->free_chunks[fl_index] = c;
 469                   fss->n_fl_chunk_bytes += fl_size;
 470                   n_alloc -= fl_size;
 471                   data_bytes += fl_size;
 472                   c = next;
 473                 }
 474               first = last = 0;
 475               fl_index = fs_freelist_for_size (data_bytes);
 476               if (fss->free_chunks[fl_index + 1])
 477                 {
 478                   fl_index += 1;
 479                   fl_size = fs_freelist_index_to_size (fl_index);
 480                   continue;
 481                 }
 482
 483               f->next = fss->free_fifos;
 484               fss->free_fifos = f;
 485               return 0;
 486             }
 487           fl_index -= 1;
 488           fl_size = fl_size >> 1;
 489         }
 490     }
 491
 492   f->start_chunk = first;
 493   f->end_chunk = last;
 494   fss->n_fl_chunk_bytes -= n_alloc;
 495   fsh_cached_bytes_sub (fsh, n_alloc);
 496   return f;
 497 }
 498
 499 static int
 500 fsh_try_alloc_chunk_batch (fifo_segment_header_t * fsh,
 501                            fifo_segment_slice_t * fss,
 502                            u32 fl_index, u32 batch_size)
 503 {
 504   u32 rounded_data_size;
 505   svm_fifo_chunk_t *c;
 506   void *oldheap;
 507   uword size;
 508   u8 *cmem;
 509   int i;
 510
 511   rounded_data_size = fs_freelist_index_to_size (fl_index);
 512   size = (uword) (sizeof (*c) + rounded_data_size) * batch_size;
 513
 514   oldheap = ssvm_push_heap (fsh->ssvm_sh);
 515   cmem = clib_mem_alloc_aligned_at_offset (size, CLIB_CACHE_LINE_BYTES,
 516                                            0 /* align_offset */ ,
 517                                            0 /* os_out_of_memory */ );
 518   ssvm_pop_heap (oldheap);
 519
 520   /* Out of space.. */
 521   if (cmem == 0)
 522     return -1;
 523
 524   /* Carve fifo + chunk space */
 525   for (i = 0; i < batch_size; i++)
 526     {
 527       c = (svm_fifo_chunk_t *) cmem;
 528       c->start_byte = 0;
 529       c->length = rounded_data_size;
 530       c->enq_rb_index = RBTREE_TNIL_INDEX;
 531       c->deq_rb_index = RBTREE_TNIL_INDEX;
 532       c->next = fss->free_chunks[fl_index];
 533       fss->free_chunks[fl_index] = c;
 534       cmem += sizeof (*c) + rounded_data_size;
 535     }
 536
 537   fss->num_chunks[fl_index] += batch_size;
 538   fss->n_fl_chunk_bytes += batch_size * rounded_data_size;
 539   fsh_cached_bytes_add (fsh, batch_size * rounded_data_size);
 540   fsh_free_bytes_sub (fsh, size);
 541
 542   return 0;
 543 }
 544
 545 static int
 546 fs_try_alloc_fifo_batch (fifo_segment_header_t * fsh,
 547                          fifo_segment_slice_t * fss,
 548                          u32 fl_index, u32 batch_size)
 549 {
 550   u32 hdrs, rounded_data_size;
 551   svm_fifo_chunk_t *c;
 552   svm_fifo_t *f;
 553   void *oldheap;
 554   uword size;
 555   u8 *fmem;
 556   int i;
 557
 558   rounded_data_size = fs_freelist_index_to_size (fl_index);
 559   hdrs = sizeof (*f) + sizeof (*c);
 560   size = (uword) (hdrs + rounded_data_size) * batch_size;
 561
 562   oldheap = ssvm_push_heap (fsh->ssvm_sh);
 563   fmem = clib_mem_alloc_aligned_at_offset (size, CLIB_CACHE_LINE_BYTES,
 564                                            0 /* align_offset */ ,
 565                                            0 /* os_out_of_memory */ );
 566   ssvm_pop_heap (oldheap);
 567
 568   /* Out of space.. */
 569   if (fmem == 0)
 570     return -1;
 571
 572   /* Carve fifo hdr space */
 573   for (i = 0; i < batch_size; i++)
 574     {
 575       f = (svm_fifo_t *) fmem;
 576       memset (f, 0, sizeof (*f));
 577       f->next = fss->free_fifos;
 578       fss->free_fifos = f;
 579       fmem += sizeof (*f);
 580     }
 581
 582   /* Carve chunk space */
 583   for (i = 0; i < batch_size; i++)
 584     {
 585       c = (svm_fifo_chunk_t *) fmem;
 586       c->start_byte = 0;
 587       c->length = rounded_data_size;
 588       c->enq_rb_index = RBTREE_TNIL_INDEX;
 589       c->deq_rb_index = RBTREE_TNIL_INDEX;
 590       c->next = fss->free_chunks[fl_index];
 591       fss->free_chunks[fl_index] = c;
 592       fmem += sizeof (svm_fifo_chunk_t) + rounded_data_size;
 593     }
 594
 595   fss->num_chunks[fl_index] += batch_size;
 596   fss->n_fl_chunk_bytes += batch_size * rounded_data_size;
 597   fsh_cached_bytes_add (fsh, batch_size * rounded_data_size);
 598   fsh_free_bytes_sub (fsh, size);
 599
 600   return 0;
 601 }
 602
 603 /**
 604  * Try to allocate new fifo
 605  *
 606  * Tries the following steps in order:
 607  * - grab fifo and chunk from freelists
 608  * - batch fifo and chunk allocation
 609  * - single fifo allocation
 610  * - grab multiple fifo chunks from freelists
 611  */
 612 static svm_fifo_t *
 613 fs_try_alloc_fifo (fifo_segment_header_t * fsh, fifo_segment_slice_t * fss,
 614                    u32 data_bytes)
 615 {
 616   u32 fifo_sz, fl_index;
 617   svm_fifo_t *f = 0;
 618   uword n_free_bytes;
 619   u32 min_size;
 620
 621   min_size = clib_max ((fsh->pct_first_alloc * data_bytes) / 100, 4096);
 622   fl_index = fs_freelist_for_size (min_size);
 623
 624   if (fl_index >= vec_len (fss->free_chunks))
 625     return 0;
 626
 627   clib_spinlock_lock (&fss->chunk_lock);
 628
 629   if (fss->free_fifos && fss->free_chunks[fl_index])
 630     {
 631       f = fs_try_alloc_fifo_freelist (fss, fl_index);
 632       if (f)
 633         {
 634           fsh_cached_bytes_sub (fsh, fs_freelist_index_to_size (fl_index));
 635           goto done;
 636         }
 637     }
 638
 639   fifo_sz = sizeof (svm_fifo_t) + sizeof (svm_fifo_chunk_t);
 640   fifo_sz += 1 << max_log2 (min_size);
 641   n_free_bytes = fsh_n_free_bytes (fsh);
 642
 643   if (fifo_sz * FIFO_SEGMENT_ALLOC_BATCH_SIZE < n_free_bytes)
 644     {
 645       if (!fs_try_alloc_fifo_batch (fsh, fss, fl_index,
 646                                     FIFO_SEGMENT_ALLOC_BATCH_SIZE))
 647         {
 648           f = fs_try_alloc_fifo_freelist (fss, fl_index);
 649           if (f)
 650             {
 651               fsh_cached_bytes_sub (fsh,
 652                                     fs_freelist_index_to_size (fl_index));
 653               goto done;
 654             }
 655         }
 656       else
 657         {
 658           fsh_check_mem (fsh);
 659           n_free_bytes = fsh_n_free_bytes (fsh);
 660         }
 661     }
 662   if (fifo_sz <= n_free_bytes)
 663     {
 664       void *oldheap = ssvm_push_heap (fsh->ssvm_sh);
 665       f = svm_fifo_alloc (min_size);
 666       ssvm_pop_heap (oldheap);
 667       if (f)
 668         {
 669           clib_atomic_fetch_add_rel (&fss->num_chunks[fl_index], 1);
 670           fsh_free_bytes_sub (fsh, fifo_sz);
 671           goto done;
 672         }
 673       fsh_check_mem (fsh);
 674     }
 675   /* All failed, try to allocate min of data bytes and fifo sz */
 676   fifo_sz = clib_min (fifo_sz, data_bytes);
 677   if (fifo_sz <= fss->n_fl_chunk_bytes)
 678     f = fs_try_alloc_fifo_freelist_multi_chunk (fsh, fss, fifo_sz);
 679
 680 done:
 681   clib_spinlock_unlock (&fss->chunk_lock);
 682
 683   if (f)
 684     {
 685       f->size = data_bytes;
 686       f->fs_hdr = fsh;
 687     }
 688   return f;
 689 }
 690
 691 svm_fifo_chunk_t *
 692 fsh_alloc_chunk (fifo_segment_header_t * fsh, u32 slice_index, u32 chunk_size)
 693 {
 694   fifo_segment_slice_t *fss;
 695   svm_fifo_chunk_t *c;
 696   int fl_index;
 697
 698   fl_index = fs_freelist_for_size (chunk_size);
 699   fss = fsh_slice_get (fsh, slice_index);
 700
 701   clib_spinlock_lock (&fss->chunk_lock);
 702
 703   ASSERT (vec_len (fss->free_chunks) > fl_index);
 704   c = fss->free_chunks[fl_index];
 705
 706   if (c)
 707     {
 708       fss->free_chunks[fl_index] = c->next;
 709       c->next = 0;
 710       fss->n_fl_chunk_bytes -= fs_freelist_index_to_size (fl_index);
 711       fsh_cached_bytes_sub (fsh, fs_freelist_index_to_size (fl_index));
 712     }
 713   else
 714     {
 715       void *oldheap;
 716       uword n_free;
 717       u32 batch;
 718
 719       chunk_size = fs_freelist_index_to_size (fl_index);
 720       n_free = fsh_n_free_bytes (fsh);
 721
 722       if (chunk_size <= n_free)
 723         {
 724           oldheap = ssvm_push_heap (fsh->ssvm_sh);
 725           c = svm_fifo_chunk_alloc (chunk_size);
 726           ssvm_pop_heap (oldheap);
 727
 728           if (c)
 729             {
 730               clib_atomic_fetch_add_rel (&fss->num_chunks[fl_index], 1);
 731               fsh_free_bytes_sub (fsh, chunk_size + sizeof (*c));
 732               goto done;
 733             }
 734
 735           fsh_check_mem (fsh);
 736           n_free = fsh_n_free_bytes (fsh);
 737         }
 738       if (chunk_size <= fss->n_fl_chunk_bytes)
 739         {
 740           c = fs_try_alloc_multi_chunk (fsh, fss, chunk_size);
 741           if (c)
 742             goto done;
 743           batch = n_free / FIFO_SEGMENT_MIN_FIFO_SIZE;
 744           if (!batch || fsh_try_alloc_chunk_batch (fsh, fss, 0, batch))
 745             {
 746               fsh_check_mem (fsh);
 747               goto done;
 748             }
 749         }
 750       if (chunk_size <= fss->n_fl_chunk_bytes + n_free)
 751         {
 752           u32 min_size = FIFO_SEGMENT_MIN_FIFO_SIZE;
 753
 754           batch = (chunk_size - fss->n_fl_chunk_bytes) / min_size;
 755           batch = clib_min (batch + 1, n_free / min_size);
 756           if (fsh_try_alloc_chunk_batch (fsh, fss, 0, batch))
 757             {
 758               fsh_check_mem (fsh);
 759               goto done;
 760             }
 761           c = fs_try_alloc_multi_chunk (fsh, fss, chunk_size);
 762         }
 763     }
 764
 765 done:
 766
 767   clib_spinlock_unlock (&fss->chunk_lock);
 768
 769   return c;
 770 }
 771
 772 static void
 773 fsh_slice_collect_chunks (fifo_segment_header_t * fsh,
 774                           fifo_segment_slice_t * fss, svm_fifo_chunk_t * c)
 775 {
 776   svm_fifo_chunk_t *next;
 777   int fl_index;
 778   u32 n_collect = 0;
 779
 780   clib_spinlock_lock (&fss->chunk_lock);
 781
 782   while (c)
 783     {
 784       CLIB_MEM_UNPOISON (c, sizeof (*c));
 785       next = c->next;
 786       fl_index = fs_freelist_for_size (c->length);
 787       c->next = fss->free_chunks[fl_index];
 788       c->enq_rb_index = RBTREE_TNIL_INDEX;
 789       c->deq_rb_index = RBTREE_TNIL_INDEX;
 790       fss->free_chunks[fl_index] = c;
 791       n_collect += fs_freelist_index_to_size (fl_index);
 792       c = next;
 793     }
 794
 795   fss->n_fl_chunk_bytes += n_collect;
 796   fsh_cached_bytes_add (fsh, n_collect);
 797
 798   clib_spinlock_unlock (&fss->chunk_lock);
 799 }
 800
 801 void
 802 fsh_collect_chunks (fifo_segment_header_t * fsh, u32 slice_index,
 803                     svm_fifo_chunk_t * c)
 804 {
 805   fifo_segment_slice_t *fss;
 806   fss = fsh_slice_get (fsh, slice_index);
 807   fsh_slice_collect_chunks (fsh, fss, c);
 808 }
 809
 810 static inline void
 811 fss_fifo_add_active_list (fifo_segment_slice_t * fss, svm_fifo_t * f)
 812 {
 813   if (fss->fifos)
 814     {
 815       fss->fifos->prev = f;
 816       f->next = fss->fifos;
 817     }
 818   fss->fifos = f;
 819 }
 820
 821 static inline void
 822 fss_fifo_del_active_list (fifo_segment_slice_t * fss, svm_fifo_t * f)
 823 {
 824   if (f->flags & SVM_FIFO_F_LL_TRACKED)
 825     {
 826       if (f->prev)
 827         f->prev->next = f->next;
 828       else
 829         fss->fifos = f->next;
 830       if (f->next)
 831         f->next->prev = f->prev;
 832     }
 833 }
 834
 835 /**
 836  * Allocate fifo in fifo segment
 837  */
 838 svm_fifo_t *
 839 fifo_segment_alloc_fifo_w_slice (fifo_segment_t * fs, u32 slice_index,
 840                                  u32 data_bytes, fifo_segment_ftype_t ftype)
 841 {
 842   fifo_segment_header_t *fsh = fs->h;
 843   fifo_segment_slice_t *fss;
 844   svm_fifo_t *f = 0;
 845
 846   ASSERT (slice_index < fs->n_slices);
 847
 848   if (PREDICT_FALSE (data_bytes > 1 << fsh->max_log2_chunk_size))
 849     return 0;
 850
 851   fss = fsh_slice_get (fsh, slice_index);
 852   f = fs_try_alloc_fifo (fsh, fss, data_bytes);
 853   if (!f)
 854     goto done;
 855
 856   f->slice_index = slice_index;
 857
 858   svm_fifo_init (f, data_bytes);
 859
 860   /* If rx fifo type add to active fifos list. When cleaning up segment,
 861    * we need a list of active sessions that should be disconnected. Since
 862    * both rx and tx fifos keep pointers to the session, it's enough to track
 863    * only one. */
 864   if (ftype == FIFO_SEGMENT_RX_FIFO)
 865     {
 866       fss_fifo_add_active_list (fss, f);
 867       f->flags |= SVM_FIFO_F_LL_TRACKED;
 868     }
 869
 870   fsh_active_fifos_update (fsh, 1);
 871   fss->virtual_mem += svm_fifo_size (f);
 872
 873 done:
 874   return (f);
 875 }
 876
 877 /**
 878  * Free fifo allocated in fifo segment
 879  */
 880 void
 881 fifo_segment_free_fifo (fifo_segment_t * fs, svm_fifo_t * f)
 882 {
 883   fifo_segment_header_t *fsh = fs->h;
 884   fifo_segment_slice_t *fss;
 885
 886   ASSERT (f->refcnt > 0);
 887
 888   if (--f->refcnt > 0)
 889     return;
 890
 891   fss = fsh_slice_get (fsh, f->slice_index);
 892
 893   /* Remove from active list. Only rx fifos are tracked */
 894   if (f->flags & SVM_FIFO_F_LL_TRACKED)
 895     {
 896       fss_fifo_del_active_list (fss, f);
 897       f->flags &= ~SVM_FIFO_F_LL_TRACKED;
 898     }
 899
 900   /* Free fifo chunks */
 901   fsh_slice_collect_chunks (fsh, fss, f->start_chunk);
 902
 903   f->start_chunk = f->end_chunk = 0;
 904   f->head_chunk = f->tail_chunk = f->ooo_enq = f->ooo_deq = 0;
 905
 906   /* not allocated on segment heap */
 907   svm_fifo_free_chunk_lookup (f);
 908   svm_fifo_free_ooo_data (f);
 909
 910   if (CLIB_DEBUG)
 911     {
 912       f->master_session_index = ~0;
 913       f->master_thread_index = ~0;
 914     }
 915
 916   fss->virtual_mem -= svm_fifo_size (f);
 917
 918   /* Add to free list */
 919   f->next = fss->free_fifos;
 920   f->prev = 0;
 921   fss->free_fifos = f;
 922
 923   fsh_active_fifos_update (fsh, -1);
 924 }
 925
 926 void
 927 fifo_segment_detach_fifo (fifo_segment_t * fs, svm_fifo_t * f)
 928 {
 929   fifo_segment_slice_t *fss;
 930   svm_fifo_chunk_t *c;
 931   u32 fl_index;
 932
 933   ASSERT (f->refcnt == 1);
 934
 935   fss = fsh_slice_get (fs->h, f->slice_index);
 936   fss->virtual_mem -= svm_fifo_size (f);
 937   if (f->flags & SVM_FIFO_F_LL_TRACKED)
 938     fss_fifo_del_active_list (fss, f);
 939
 940   c = f->start_chunk;
 941   while (c)
 942     {
 943       fl_index = fs_freelist_for_size (c->length);
 944       clib_atomic_fetch_sub_rel (&fss->num_chunks[fl_index], 1);
 945       c = c->next;
 946     }
 947 }
 948
 949 void
 950 fifo_segment_attach_fifo (fifo_segment_t * fs, svm_fifo_t * f,
 951                           u32 slice_index)
 952 {
 953   fifo_segment_slice_t *fss;
 954   svm_fifo_chunk_t *c;
 955   u32 fl_index;
 956
 957   f->slice_index = slice_index;
 958   fss = fsh_slice_get (fs->h, f->slice_index);
 959   fss->virtual_mem += svm_fifo_size (f);
 960   if (f->flags & SVM_FIFO_F_LL_TRACKED)
 961     fss_fifo_add_active_list (fss, f);
 962
 963   c = f->start_chunk;
 964   while (c)
 965     {
 966       fl_index = fs_freelist_for_size (c->length);
 967       clib_atomic_fetch_add_rel (&fss->num_chunks[fl_index], 1);
 968       c = c->next;
 969     }
 970 }
 971
 972 int
 973 fifo_segment_prealloc_fifo_hdrs (fifo_segment_t * fs, u32 slice_index,
 974                                  u32 batch_size)
 975 {
 976   fifo_segment_header_t *fsh = fs->h;
 977   fifo_segment_slice_t *fss;
 978   svm_fifo_t *f;
 979   void *oldheap;
 980   uword size;
 981   u8 *fmem;
 982   int i;
 983
 984   fss = fsh_slice_get (fsh, slice_index);
 985   size = (uword) (sizeof (*f)) * batch_size;
 986
 987   oldheap = ssvm_push_heap (fsh->ssvm_sh);
 988   fmem = clib_mem_alloc_aligned_at_offset (size, CLIB_CACHE_LINE_BYTES,
 989                                            0 /* align_offset */ ,
 990                                            0 /* os_out_of_memory */ );
 991   ssvm_pop_heap (oldheap);
 992
 993   /* Out of space.. */
 994   if (fmem == 0)
 995     return -1;
 996
 997   /* Carve fifo + chunk space */
 998   for (i = 0; i < batch_size; i++)
 999     {
1000       f = (svm_fifo_t *) fmem;
1001       memset (f, 0, sizeof (*f));
1002       f->next = fss->free_fifos;
1003       fss->free_fifos = f;
1004       fmem += sizeof (*f);
1005     }
1006
1007   fsh_free_bytes_sub (fsh, size);
1008
1009   return 0;
1010 }
1011
1012 int
1013 fifo_segment_prealloc_fifo_chunks (fifo_segment_t * fs, u32 slice_index,
1014                                    u32 chunk_size, u32 batch_size)
1015 {
1016   fifo_segment_header_t *fsh = fs->h;
1017   u32 rounded_data_size, fl_index;
1018   fifo_segment_slice_t *fss;
1019   svm_fifo_chunk_t *c;
1020   void *oldheap;
1021   uword size;
1022   u8 *cmem;
1023   int i;
1024
1025   if (!fs_chunk_size_is_valid (fsh, chunk_size))
1026     {
1027       clib_warning ("chunk size out of range %d", chunk_size);
1028       return -1;
1029     }
1030
1031   fl_index = fs_freelist_for_size (chunk_size);
1032   rounded_data_size = fs_freelist_index_to_size (fl_index);
1033   size = (uword) (sizeof (*c) + rounded_data_size) * batch_size;
1034
1035   oldheap = ssvm_push_heap (fsh->ssvm_sh);
1036   cmem = clib_mem_alloc_aligned_at_offset (size, CLIB_CACHE_LINE_BYTES,
1037                                            0 /* align_offset */ ,
1038                                            0 /* os_out_of_memory */ );
1039   ssvm_pop_heap (oldheap);
1040
1041   /* Out of space.. */
1042   if (cmem == 0)
1043     return -1;
1044
1045   fss = fsh_slice_get (fsh, slice_index);
1046
1047   /* Carve fifo + chunk space */
1048   for (i = 0; i < batch_size; i++)
1049     {
1050       c = (svm_fifo_chunk_t *) cmem;
1051       c->start_byte = 0;
1052       c->length = rounded_data_size;
1053       c->next = fss->free_chunks[fl_index];
1054       fss->free_chunks[fl_index] = c;
1055       cmem += sizeof (*c) + rounded_data_size;
1056       fsh_cached_bytes_add (fsh, rounded_data_size);
1057     }
1058
1059   fss->num_chunks[fl_index] += batch_size;
1060   fss->n_fl_chunk_bytes += batch_size * rounded_data_size;
1061   fsh_free_bytes_sub (fsh, size);
1062
1063   return 0;
1064 }
1065
1066 /**
1067  * Pre-allocates fifo pairs in fifo segment
1068  */
1069 void
1070 fifo_segment_preallocate_fifo_pairs (fifo_segment_t * fs,
1071                                      u32 rx_fifo_size, u32 tx_fifo_size,
1072                                      u32 * n_fifo_pairs)
1073 {
1074   u32 rx_rounded_data_size, tx_rounded_data_size, pair_size, pairs_to_alloc;
1075   u32 hdrs, pairs_per_slice, alloc_now;
1076   fifo_segment_header_t *fsh = fs->h;
1077   int rx_fl_index, tx_fl_index, i;
1078   fifo_segment_slice_t *fss;
1079   uword space_available;
1080
1081   /* Parameter check */
1082   if (rx_fifo_size == 0 || tx_fifo_size == 0 || *n_fifo_pairs == 0)
1083     return;
1084
1085   if (!fs_chunk_size_is_valid (fsh, rx_fifo_size))
1086     {
1087       clib_warning ("rx fifo_size out of range %d", rx_fifo_size);
1088       return;
1089     }
1090
1091   if (!fs_chunk_size_is_valid (fsh, tx_fifo_size))
1092     {
1093       clib_warning ("tx fifo_size out of range %d", tx_fifo_size);
1094       return;
1095     }
1096
1097   rx_rounded_data_size = (1 << (max_log2 (rx_fifo_size)));
1098   rx_fl_index = fs_freelist_for_size (rx_fifo_size);
1099   tx_rounded_data_size = (1 << (max_log2 (tx_fifo_size)));
1100   tx_fl_index = fs_freelist_for_size (tx_fifo_size);
1101
1102   hdrs = sizeof (svm_fifo_t) + sizeof (svm_fifo_chunk_t);
1103
1104   /* Calculate space requirements */
1105   pair_size = 2 * hdrs + rx_rounded_data_size + tx_rounded_data_size;
1106   space_available = fsh_free_space (fsh);
1107   pairs_to_alloc = space_available / pair_size;
1108   pairs_to_alloc = clib_min (pairs_to_alloc, *n_fifo_pairs);
1109   pairs_per_slice = pairs_to_alloc / fs->n_slices;
1110   pairs_per_slice += pairs_to_alloc % fs->n_slices ? 1 : 0;
1111
1112   if (!pairs_per_slice)
1113     return;
1114
1115   for (i = 0; i < fs->n_slices; i++)
1116     {
1117       fss = fsh_slice_get (fsh, i);
1118       alloc_now = clib_min (pairs_per_slice, *n_fifo_pairs);
1119       if (fs_try_alloc_fifo_batch (fsh, fss, rx_fl_index, alloc_now))
1120         clib_warning ("rx prealloc failed: pairs %u", alloc_now);
1121       if (fs_try_alloc_fifo_batch (fsh, fss, tx_fl_index, alloc_now))
1122         clib_warning ("tx prealloc failed: pairs %u", alloc_now);
1123
1124       /* Account for the pairs allocated */
1125       *n_fifo_pairs -= alloc_now;
1126     }
1127 }
1128
1129 /**
1130  * Get number of active fifos
1131  */
1132 u32
1133 fifo_segment_num_fifos (fifo_segment_t * fs)
1134 {
1135   return fsh_n_active_fifos (fs->h);
1136 }
1137
1138 static u32
1139 fs_slice_num_free_fifos (fifo_segment_slice_t * fss)
1140 {
1141   svm_fifo_t *f;
1142   u32 count = 0;
1143
1144   f = fss->free_fifos;
1145   if (f == 0)
1146     return 0;
1147
1148   while (f)
1149     {
1150       f = f->next;
1151       count++;
1152     }
1153   return count;
1154 }
1155
1156 u32
1157 fifo_segment_num_free_fifos (fifo_segment_t * fs)
1158 {
1159   fifo_segment_header_t *fsh = fs->h;
1160   fifo_segment_slice_t *fss;
1161   int slice_index;
1162   u32 count = 0;
1163
1164   for (slice_index = 0; slice_index < fs->n_slices; slice_index++)
1165     {
1166       fss = fsh_slice_get (fsh, slice_index);
1167       count += fs_slice_num_free_fifos (fss);
1168     }
1169   return count;
1170 }
1171
1172 static u32
1173 fs_slice_num_free_chunks (fifo_segment_slice_t * fss, u32 size)
1174 {
1175   u32 count = 0, rounded_size, fl_index;
1176   svm_fifo_chunk_t *c;
1177   int i;
1178
1179   /* Count all free chunks? */
1180   if (size == ~0)
1181     {
1182       for (i = 0; i < vec_len (fss->free_chunks); i++)
1183         {
1184           c = fss->free_chunks[i];
1185           if (c == 0)
1186             continue;
1187
1188           while (c)
1189             {
1190               c = c->next;
1191               count++;
1192             }
1193         }
1194       return count;
1195     }
1196
1197   rounded_size = (1 << (max_log2 (size)));
1198   fl_index = fs_freelist_for_size (rounded_size);
1199
1200   if (fl_index >= vec_len (fss->free_chunks))
1201     return 0;
1202
1203   c = fss->free_chunks[fl_index];
1204   if (c == 0)
1205     return 0;
1206
1207   while (c)
1208     {
1209       c = c->next;
1210       count++;
1211     }
1212   return count;
1213 }
1214
1215 u32
1216 fifo_segment_num_free_chunks (fifo_segment_t * fs, u32 size)
1217 {
1218   fifo_segment_header_t *fsh = fs->h;
1219   fifo_segment_slice_t *fss;
1220   int slice_index;
1221   u32 count = 0;
1222
1223   for (slice_index = 0; slice_index < fs->n_slices; slice_index++)
1224     {
1225       fss = fsh_slice_get (fsh, slice_index);
1226       count += fs_slice_num_free_chunks (fss, size);
1227     }
1228   return count;
1229 }
1230
1231 void
1232 fifo_segment_update_free_bytes (fifo_segment_t * fs)
1233 {
1234   fsh_update_free_bytes (fs->h);
1235 }
1236
1237 uword
1238 fifo_segment_size (fifo_segment_t * fs)
1239 {
1240   return fs->ssvm.ssvm_size;
1241 }
1242
1243 u8
1244 fsh_has_reached_mem_limit (fifo_segment_header_t * fsh)
1245 {
1246   return (fsh->flags & FIFO_SEGMENT_F_MEM_LIMIT) ? 1 : 0;
1247 }
1248
1249 void
1250 fsh_reset_mem_limit (fifo_segment_header_t * fsh)
1251 {
1252   fsh->flags &= ~FIFO_SEGMENT_F_MEM_LIMIT;
1253 }
1254
1255 uword
1256 fifo_segment_free_bytes (fifo_segment_t * fs)
1257 {
1258   return fsh_n_free_bytes (fs->h);
1259 }
1260
1261 uword
1262 fifo_segment_cached_bytes (fifo_segment_t * fs)
1263 {
1264   return fsh_n_cached_bytes (fs->h);
1265 }
1266
1267 uword
1268 fifo_segment_available_bytes (fifo_segment_t * fs)
1269 {
1270   return fsh_n_free_bytes (fs->h) + fsh_n_cached_bytes (fs->h);
1271 }
1272
1273 uword
1274 fifo_segment_fl_chunk_bytes (fifo_segment_t * fs)
1275 {
1276   fifo_segment_header_t *fsh = fs->h;
1277   fifo_segment_slice_t *fss;
1278   uword n_bytes = 0;
1279   int slice_index;
1280
1281   for (slice_index = 0; slice_index < fs->n_slices; slice_index++)
1282     {
1283       fss = fsh_slice_get (fsh, slice_index);
1284       n_bytes += fss->n_fl_chunk_bytes;
1285     }
1286
1287   return n_bytes;
1288 }
1289
1290 u8
1291 fifo_segment_has_fifos (fifo_segment_t * fs)
1292 {
1293   return (fsh_n_active_fifos (fs->h) != 0);
1294 }
1295
1296 svm_fifo_t *
1297 fifo_segment_get_slice_fifo_list (fifo_segment_t * fs, u32 slice_index)
1298 {
1299   fifo_segment_header_t *fsh = fs->h;
1300   fifo_segment_slice_t *fss;
1301
1302   fss = fsh_slice_get (fsh, slice_index);
1303   return fss->fifos;
1304 }
1305
1306 u8
1307 fifo_segment_get_mem_usage (fifo_segment_t * fs)
1308 {
1309   uword size, in_use;
1310
1311   size = fifo_segment_size (fs);
1312   in_use =
1313     size - fifo_segment_free_bytes (fs) - fifo_segment_cached_bytes (fs);
1314   return (in_use * 100) / size;
1315 }
1316
1317 fifo_segment_mem_status_t
1318 fifo_segment_determine_status (fifo_segment_header_t * fsh, u8 usage)
1319 {
1320   if (!fsh->high_watermark || !fsh->low_watermark)
1321     return MEMORY_PRESSURE_NO_PRESSURE;
1322
1323   /* once the no-memory is detected, the status continues
1324    * until memory usage gets below the high watermark
1325    */
1326   if (fsh_has_reached_mem_limit (fsh))
1327     {
1328       if (usage >= fsh->high_watermark)
1329         return MEMORY_PRESSURE_NO_MEMORY;
1330       else
1331         fsh_reset_mem_limit (fsh);
1332     }
1333
1334   if (usage >= fsh->high_watermark)
1335     return MEMORY_PRESSURE_HIGH_PRESSURE;
1336
1337   else if (usage >= fsh->low_watermark)
1338     return MEMORY_PRESSURE_LOW_PRESSURE;
1339
1340   return MEMORY_PRESSURE_NO_PRESSURE;
1341 }
1342
1343 fifo_segment_mem_status_t
1344 fifo_segment_get_mem_status (fifo_segment_t * fs)
1345 {
1346   fifo_segment_header_t *fsh = fs->h;
1347   u8 usage = fifo_segment_get_mem_usage (fs);
1348
1349   return fifo_segment_determine_status (fsh, usage);
1350 }
1351
1352 u8 *
1353 format_fifo_segment_type (u8 * s, va_list * args)
1354 {
1355   fifo_segment_t *sp;
1356   sp = va_arg (*args, fifo_segment_t *);
1357   ssvm_segment_type_t st = ssvm_type (&sp->ssvm);
1358
1359   if (st == SSVM_SEGMENT_PRIVATE)
1360     s = format (s, "%s", "private");
1361   else if (st == SSVM_SEGMENT_MEMFD)
1362     s = format (s, "%s", "memfd");
1363   else if (st == SSVM_SEGMENT_SHM)
1364     s = format (s, "%s", "shm");
1365   else
1366     s = format (s, "%s", "unknown");
1367   return s;
1368 }
1369
1370 /**
1371  * Segment format function
1372  */
1373 u8 *
1374 format_fifo_segment (u8 * s, va_list * args)
1375 {
1376   u32 count, indent, active_fifos, free_fifos;
1377   fifo_segment_t *fs = va_arg (*args, fifo_segment_t *);
1378   int verbose __attribute__ ((unused)) = va_arg (*args, int);
1379   uword est_chunk_bytes, est_free_seg_bytes, free_chunks;
1380   uword chunk_bytes = 0, free_seg_bytes, chunk_size;
1381   uword tracked_cached_bytes;
1382   uword fifo_hdr = 0, reserved;
1383   fifo_segment_header_t *fsh;
1384   fifo_segment_slice_t *fss;
1385   svm_fifo_chunk_t *c;
1386   u32 slice_index;
1387   char *address;
1388   size_t size;
1389   int i;
1390   uword allocated, in_use, virt;
1391   f64 usage;
1392   fifo_segment_mem_status_t mem_st;
1393
1394   indent = format_get_indent (s) + 2;
1395
1396   if (fs == 0)
1397     {
1398       s = format (s, "%-20s%10s%15s%15s%15s%15s", "Name", "Type",
1399                   "HeapSize (M)", "ActiveFifos", "FreeFifos", "Address");
1400       return s;
1401     }
1402
1403   fifo_segment_info (fs, &address, &size);
1404   active_fifos = fifo_segment_num_fifos (fs);
1405   free_fifos = fifo_segment_num_free_fifos (fs);
1406
1407   s = format (s, "%-20v%10U%15llu%15u%15u%15llx", ssvm_name (&fs->ssvm),
1408               format_fifo_segment_type, fs, size >> 20ULL, active_fifos,
1409               free_fifos, address);
1410
1411   if (!verbose)
1412     return s;
1413
1414   fsh = fs->h;
1415
1416   free_chunks = fifo_segment_num_free_chunks (fs, ~0);
1417   if (free_chunks)
1418     s =
1419       format (s, "\n\n%UFree/Allocated chunks by size:\n", format_white_space,
1420               indent + 2);
1421   else
1422     s = format (s, "\n");
1423
1424   for (slice_index = 0; slice_index < fs->n_slices; slice_index++)
1425     {
1426       fss = fsh_slice_get (fsh, slice_index);
1427       for (i = 0; i < vec_len (fss->free_chunks); i++)
1428         {
1429           c = fss->free_chunks[i];
1430           if (c == 0 && fss->num_chunks[i] == 0)
1431             continue;
1432           count = 0;
1433           while (c)
1434             {
1435               c = c->next;
1436               count++;
1437             }
1438
1439           chunk_size = fs_freelist_index_to_size (i);
1440           s = format (s, "%U%-5u kB: %u/%u\n", format_white_space, indent + 2,
1441                       chunk_size >> 10, count, fss->num_chunks[i]);
1442
1443           chunk_bytes += count * chunk_size;
1444         }
1445     }
1446
1447   fifo_hdr = free_fifos * sizeof (svm_fifo_t);
1448   est_chunk_bytes = fifo_segment_fl_chunk_bytes (fs);
1449   est_free_seg_bytes = fifo_segment_free_bytes (fs);
1450   fifo_segment_update_free_bytes (fs);
1451   free_seg_bytes = fifo_segment_free_bytes (fs);
1452   tracked_cached_bytes = fifo_segment_cached_bytes (fs);
1453   allocated = fifo_segment_size (fs);
1454   in_use = fifo_segment_size (fs) - est_free_seg_bytes - tracked_cached_bytes;
1455   usage = (100.0 * in_use) / allocated;
1456   mem_st = fifo_segment_get_mem_status (fs);
1457   virt = fsh_virtual_mem (fsh);
1458   reserved = fsh->n_reserved_bytes;
1459
1460   s = format (s, "\n%Useg free bytes: %U (%lu) estimated: %U (%lu) reserved:"
1461               " %U (%lu)\n", format_white_space, indent + 2,
1462               format_memory_size, free_seg_bytes, free_seg_bytes,
1463               format_memory_size, est_free_seg_bytes, est_free_seg_bytes,
1464               format_memory_size, reserved, reserved);
1465   s = format (s, "%Uchunk free bytes: %U (%lu) estimated: %U (%lu) tracked:"
1466               " %U (%lu)\n", format_white_space, indent + 2,
1467               format_memory_size, chunk_bytes, chunk_bytes,
1468               format_memory_size, est_chunk_bytes, est_chunk_bytes,
1469               format_memory_size, tracked_cached_bytes, tracked_cached_bytes);
1470   s = format (s, "%Ufifo active: %u hdr free bytes: %U (%u) \n",
1471               format_white_space, indent + 2, fsh->n_active_fifos,
1472               format_memory_size, fifo_hdr, fifo_hdr);
1473   s = format (s, "%Usegment usage: %.2f%% (%U / %U) virt: %U status: %s\n",
1474               format_white_space, indent + 2, usage, format_memory_size,
1475               in_use, format_memory_size, allocated, format_memory_size, virt,
1476               fifo_segment_mem_status_strings[mem_st]);
1477   s = format (s, "\n");
1478
1479   return s;
1480 }
1481
1482 /*
1483  * fd.io coding-style-patch-verification: ON
1484  *
1485  * Local Variables:
1486  * eval: (c-set-style "gnu")
1487  * End:
1488  */