lib/librte_sched/rte_red.c

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2010-2013 Intel Corporation. All rights reserved.
   5  *   All rights reserved.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following conditions
   9  *   are met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above copyright
  14  *       notice, this list of conditions and the following disclaimer in
  15  *       the documentation and/or other materials provided with the
  16  *       distribution.
  17  *     * Neither the name of Intel Corporation nor the names of its
  18  *       contributors may be used to endorse or promote products derived
  19  *       from this software without specific prior written permission.
  20  *
  21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  *
  33  */
  34
  35 #include <math.h>
  36 #include "rte_red.h"
  37 #include <rte_random.h>
  38
  39 #ifdef __INTEL_COMPILER
  40 #pragma warning(disable:2259) /* conversion may lose significant bits */
  41 #endif
  42
  43 #define DIM(x) (sizeof(x)/sizeof(x[0]))
  44
  45 static int rte_red_init_done = 0;     /**< Flag to indicate that global initialisation is done */
  46 uint32_t rte_red_rand_val = 0;        /**< Random value cache */
  47 uint32_t rte_red_rand_seed = 0;       /**< Seed for random number generation */
  48
  49 /**
  50  * table[i] = log2(1-Wq) * Scale * -1
  51  *       Wq = 1/(2^i)
  52  */
  53 uint16_t rte_red_log2_1_minus_Wq[RTE_RED_WQ_LOG2_NUM];
  54
  55 /**
  56  * table[i] = 2^(i/16) * Scale
  57  */
  58 uint16_t rte_red_pow2_frac_inv[16];
  59
  60 /**
  61  * @brief Initialize tables used to compute average
  62  *        queue size when queue is empty.
  63  */
  64 static void
  65 __rte_red_init_tables(void)
  66 {
  67         uint32_t i = 0;
  68         double scale = 0.0;
  69         double table_size = 0.0;
  70
  71         scale = (double)(1 << RTE_RED_SCALING);
  72         table_size = (double)(DIM(rte_red_pow2_frac_inv));
  73
  74         for (i = 0; i < DIM(rte_red_pow2_frac_inv); i++) {
  75                 double m = (double)i;
  76
  77                 rte_red_pow2_frac_inv[i] = (uint16_t) round(scale / pow(2, m / table_size));
  78         }
  79
  80         scale = 1024.0;
  81
  82         RTE_RED_ASSERT(RTE_RED_WQ_LOG2_NUM == DIM(rte_red_log2_1_minus_Wq));
  83
  84         for (i = RTE_RED_WQ_LOG2_MIN; i <= RTE_RED_WQ_LOG2_MAX; i++) {
  85                 double n = (double)i;
  86                 double Wq = pow(2, -n);
  87                 uint32_t index = i - RTE_RED_WQ_LOG2_MIN;
  88
  89                 rte_red_log2_1_minus_Wq[index] = (uint16_t) round(-1.0 * scale * log2(1.0 - Wq));
  90                 /**
  91                 * Table entry of zero, corresponds to a Wq of zero
  92                 * which is not valid (avg would remain constant no
  93                 * matter how long the queue is empty). So we have
  94                 * to check for zero and round up to one.
  95                 */
  96                 if (rte_red_log2_1_minus_Wq[index] == 0) {
  97                         rte_red_log2_1_minus_Wq[index] = 1;
  98                 }
  99         }
 100 }
 101
 102 int
 103 rte_red_rt_data_init(struct rte_red *red)
 104 {
 105         if (red == NULL)
 106                 return -1;
 107
 108         red->avg = 0;
 109         red->count = 0;
 110         red->q_time = 0;
 111         return 0;
 112 }
 113
 114 int
 115 rte_red_config_init(struct rte_red_config *red_cfg,
 116         const uint16_t wq_log2,
 117         const uint16_t min_th,
 118         const uint16_t max_th,
 119         const uint16_t maxp_inv)
 120 {
 121         if (red_cfg == NULL) {
 122                 return -1;
 123         }
 124         if (max_th > RTE_RED_MAX_TH_MAX) {
 125                 return -2;
 126         }
 127         if (min_th >= max_th) {
 128                 return -3;
 129         }
 130         if (wq_log2 > RTE_RED_WQ_LOG2_MAX) {
 131                 return -4;
 132         }
 133         if (wq_log2 < RTE_RED_WQ_LOG2_MIN) {
 134                 return -5;
 135         }
 136         if (maxp_inv < RTE_RED_MAXP_INV_MIN) {
 137                 return -6;
 138         }
 139         if (maxp_inv > RTE_RED_MAXP_INV_MAX) {
 140                 return -7;
 141         }
 142
 143         /**
 144          *  Initialize the RED module if not already done
 145          */
 146         if (!rte_red_init_done) {
 147                 rte_red_rand_seed = rte_rand();
 148                 rte_red_rand_val = rte_fast_rand();
 149                 __rte_red_init_tables();
 150                 rte_red_init_done = 1;
 151         }
 152
 153         red_cfg->min_th = ((uint32_t) min_th) << (wq_log2 + RTE_RED_SCALING);
 154         red_cfg->max_th = ((uint32_t) max_th) << (wq_log2 + RTE_RED_SCALING);
 155         red_cfg->pa_const = (2 * (max_th - min_th) * maxp_inv) << RTE_RED_SCALING;
 156         red_cfg->maxp_inv = maxp_inv;
 157         red_cfg->wq_log2 = wq_log2;
 158
 159         return 0;
 160 }