1 files changed, 159 insertions, 0 deletions

/*
  Copyright (c) 2012 Cisco and/or its affiliates.

  * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
*/

#include <vppinfra/slist.h>

/*
 * skip-list implementation
 *
 * Good news / bad news. As balanced binary tree schemes go,
 * this one seems pretty fast and is reasonably simple. There's a very
 * limited amount that can be done to mitigate sdram read latency.
 *
 * Each active clib_slist_elt_t is on from 1 to N lists. Each active element
 * is always on the "level-0" list. Since most elements are *only* on
 * level 0, we keep the level 0 (and level 1) in the element. For those
 * elements on more than two lists, we switch to a vector. Hence, the
 * "n" union in slib_slist_elt_t. 
 * 
 * The low-order bit of elt->n.next0[0] is 1 for inlined next indices,
 * 0 for vector indices (since the allocator always aligns to at least
 * a 4-byte boundary). We can only represent 2e9 items, but since the
 * practical performance limit is O(1e7), it doesn't matter.
 *
 * We create a "head" element which (by construction) is always
 * lexically lighter than any other element. This makes a large number
 * of irritating special cases go away.
 *
 * User code is in charge of comparing a supplied key with
 * the key component of a user pool element. The user tells this code
 * to add or delete (opaque key, 32-bit integer) pairs to the skip-list.
 * 
 * The algorithm adds new elements to one or more lists.
 * For levels greater than zero, the probability of a new element landing on
 * a list is branching_factor**N. Branching_factor = 0.2 seems to work
 * OK, yielding about 50 compares per search at O(1e7) items.
 */

clib_error_t *
clib_slist_init (clib_slist_t *sp, f64 branching_factor, 
                 clib_slist_key_compare_function_t compare,
                 format_function_t format_user_element)
{
  clib_slist_elt_t *head;
  memset (sp, 0, sizeof (sp[0]));
  sp->branching_factor = branching_factor;
  sp->format_user_element = format_user_element;
  sp->compare = compare;
  sp->seed = 0xdeaddabe;
  pool_get (sp->elts, head);
  vec_add1 (head->n.nexts, (u32)~0);
  head->user_pool_index = (u32)~0;
  vec_validate (sp->path, 1);
  vec_validate (sp->occupancy, 0);

  return 0;
}

/*
 * slist_search_internal
 */
static inline clib_slist_search_result_t
slist_search_internal (clib_slist_t *sp, void *key, int need_full_path)
{
  int level, comp_result;
  clib_slist_elt_t *search_elt, *head_elt;

  sp->ncompares = 0;
  /* 
   * index 0 is the magic listhead element which is 
   * lexically lighter than / to the left of every element
   */
  search_elt = head_elt =  pool_elt_at_index (sp->elts, 0);

  /* 
   * Initial negotiating position, only the head_elt is
   * lighter than the supplied key
   */
  memset (sp->path, 0, vec_len(head_elt->n.nexts) * sizeof (u32));

  /* Walk the fastest lane first */
  level = vec_len (head_elt->n.nexts) - 1;
  _vec_len (sp->path) = level + 1;

  while (1)
    {
      u32 next_index_this_level;
      clib_slist_elt_t *prefetch_elt;

      /* 
       * Prefetching the next element at this level makes a measurable
       * difference, but doesn't fix the dependent read stall problem
       */
      prefetch_elt = sp->elts + 
        clib_slist_get_next_at_level (search_elt, level);

      CLIB_PREFETCH(prefetch_elt, CLIB_CACHE_LINE_BYTES, READ);

      /* Compare the key with the current element */
      comp_result = (search_elt == head_elt) ? 1 :
        sp->compare (key, search_elt->user_pool_index);

      sp->ncompares++;
      /* key "lighter" than this element */
      if (comp_result < 0) 
        {
          /* 
           * Back up to previous item on this list 
           * and search the next finer-grained list
           * starting there.
           */
          search_elt = pool_elt_at_index (sp->elts, sp->path [level]);
        next_list:
          if (level > 0)
            {
              level--;
              continue;
            }
          else
            {
              return CLIB_SLIST_NO_MATCH;
            }
        } 
      /* Match */
      if (comp_result == 0) 
        {
          /* 
           * If we're trying to delete an element, we need to
           * track down all of the elements which point at it.
           * Otherwise, don't bother with it
           */
          if (need_full_path && level > 0)
            {
              search_elt = pool_elt_at_index (sp->elts, sp->path [level]);
              level--;
              continue;
            }
          level = vec_len(head_elt->n.nexts);
          sp->path[level] = search_elt - sp->elts;
          _vec_len (sp->path) = level + 1;
          return CLIB_SLIST_MATCH;
        }
      /* 
       * comp_result positive, key is to the right of
       * this element
       */ 
      sp->path[level] = search_elt - sp->elts;

      /* Out of list at this level? */
      next_index_this_level = clib_slist_get_next_at_level (search_elt, level);
      if (next_index_this_level == (u32)~0) 
        goto next_list;

      /* No, try the next element */
      search_elt = pool_elt_at_index (sp->elts, next_index_this_level);
    }
  return 0; /* notreached */
}

u32 clib_slist_search (clib_slist_t *sp, void *key, u32 *ncompares)
{
  clib_slist_search_result_t rv;

  rv = slist_search_internal (sp, key, 0 /* dont need full path */);
  if (rv == CLIB_SLIST_MATCH)
    {
      clib_slist_elt_t *elt;
      elt = pool_elt_at_index (sp->elts, sp->path[vec_len(sp->path)-1]);
      if (ncompares)
        *ncompares = sp->ncompares;
      return elt->user_pool_index;
    }
  return (u32)~0;
}

void clib_slist_add (clib_slist_t *sp, void *key, u32 user_pool_index)
{
  clib_slist_elt_t *new_elt;
  clib_slist_search_result_t search_result;
  int level;

  search_result = slist_search_internal (sp, key, 
                                         0 /* don't need full path */);

  /* Special case: key exists, just replace user_pool_index */
  if (PREDICT_FALSE(search_result == CLIB_SLIST_MATCH))
    {
      clib_slist_elt_t *elt;
      elt = pool_elt_at_index (sp->elts, sp->path[0]);
      elt->user_pool_index = user_pool_index;
      return;
    }
    
  pool_get (sp->elts, new_elt);
  new_elt->n.nexts = 0;
  new_elt->user_pool_index = user_pool_index;

  /* sp->path lists elements to the left of key, by level */
  for (level = 0; level < vec_len(sp->path); level++)
    {
      clib_slist_elt_t *prev_elt_this_level;
      u32 prev_elt_next_index_this_level;

      /* Add to list at the current level */
      prev_elt_this_level = pool_elt_at_index (sp->elts, sp->path[level]);
      prev_elt_next_index_this_level = clib_slist_get_next_at_level 
        (prev_elt_this_level, level);
                                                                 
      clib_slist_set_next_at_level (new_elt, prev_elt_next_index_this_level, 
                                    level);

      clib_slist_set_next_at_level (prev_elt_this_level, new_elt - sp->elts,
                                    level);
      sp->occupancy[level]++;
      
      /* Randomly add to the next-higher level */
      if (random_f64 (&sp->seed) > sp->branching_factor)
        break;
    }
    {
    /* Time to add a new ply? */
      clib_slist_elt_t *head_elt = pool_elt_at_index (sp->elts, 0);
      int top_level = vec_len(head_elt->n.nexts) - 1;
      if (((f64)sp->occupancy[top_level]) * sp->branching_factor > 1.0)
        {
          vec_add1 (sp->occupancy, 0);
          vec_add1 (head_elt->n.nexts, (u32)~0);
          /* full match case returns n+1 items */
          vec_validate (sp->path, vec_len(head_elt->n.nexts));
        }
    }
}

clib_slist_search_result_t
clib_slist_del (clib_slist_t *sp, void *key)
{
  clib_slist_search_result_t search_result;
  clib_slist_elt_t *del_elt;
  int level;
  
  search_result = slist_search_internal (sp, key, 1 /* need full path */);

  if (PREDICT_FALSE(search_result == CLIB_SLIST_NO_MATCH))
    return search_result;

  del_elt = pool_elt_at_index (sp->elts, sp->path[vec_len(sp->path)-1]);
  ASSERT(vec_len(sp->path) > 1);
  
  for (level = 0; level < vec_len (sp->path)-1; level++)
    {
      clib_slist_elt_t *path_elt;
      u32 path_elt_next_index;
      
      path_elt = pool_elt_at_index (sp->elts, sp->path[level]);
      path_elt_next_index = clib_slist_get_next_at_level (path_elt, level);
      
      /* Splice the item out of the list if it's adjacent to the victim */
      if (path_elt_next_index == del_elt - sp->elts)
        {
          sp->occupancy[level]--;
          path_elt_next_index = clib_slist_get_next_at_level (del_elt, level);
          clib_slist_set_next_at_level (path_elt, path_elt_next_index, level);
        }
    }

  /* If this element is on more than two lists it has a vector of nexts */
  if (! (del_elt->n.next0[0] & 1))
    vec_free (del_elt->n.nexts);
  pool_put (sp->elts, del_elt);
  return CLIB_SLIST_MATCH;
}

u8 * format_slist (u8 * s, va_list *args)
{
  clib_slist_t *sl = va_arg (*args, clib_slist_t *);
  int verbose = va_arg (*args, int);
  int i;
  clib_slist_elt_t *head_elt, *elt;

  s = format (s, "slist 0x%x, %u items, branching_factor %.2f\n", sl, 
              sl->occupancy ? sl->occupancy[0] : 0, sl->branching_factor);
      
  if (pool_elts(sl->elts) == 0)
    return s;
          
  head_elt = pool_elt_at_index (sl->elts, 0);
  
  for (i = 0; i < vec_len (head_elt->n.nexts); i++)
    {
      s = format (s, "level %d: %d elts\n", i, sl->occupancy[i]);
      
      if (verbose && head_elt->n.nexts[i] != (u32)~0)
        {
          elt = pool_elt_at_index (sl->elts, head_elt->n.nexts[i]);
          while (elt)
            {
              u32 next_index;
              s = format (s, "%U(%d) ", sl->format_user_element, 
                          elt->user_pool_index, elt - sl->elts);
              next_index = clib_slist_get_next_at_level (elt, i);
              ASSERT(next_index != 0x7fffffff);
              if (next_index == (u32)~0)
                break;
              else
                elt = pool_elt_at_index (sl->elts, next_index);
            }
        }
      s = format (s, "\n");
    }
  return s;
}