lib/librte_member/rte_member.c

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2017 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 #include <string.h>
  35
  36 #include <rte_eal.h>
  37 #include <rte_eal_memconfig.h>
  38 #include <rte_memory.h>
  39 #include <rte_memzone.h>
  40 #include <rte_malloc.h>
  41 #include <rte_errno.h>
  42
  43 #include "rte_member.h"
  44 #include "rte_member_ht.h"
  45 #include "rte_member_vbf.h"
  46
  47 int librte_member_logtype;
  48
  49 TAILQ_HEAD(rte_member_list, rte_tailq_entry);
  50 static struct rte_tailq_elem rte_member_tailq = {
  51         .name = "RTE_MEMBER",
  52 };
  53 EAL_REGISTER_TAILQ(rte_member_tailq)
  54
  55 struct rte_member_setsum *
  56 rte_member_find_existing(const char *name)
  57 {
  58         struct rte_member_setsum *setsum = NULL;
  59         struct rte_tailq_entry *te;
  60         struct rte_member_list *member_list;
  61
  62         member_list = RTE_TAILQ_CAST(rte_member_tailq.head, rte_member_list);
  63
  64         rte_rwlock_read_lock(RTE_EAL_TAILQ_RWLOCK);
  65         TAILQ_FOREACH(te, member_list, next) {
  66                 setsum = (struct rte_member_setsum *) te->data;
  67                 if (strncmp(name, setsum->name, RTE_MEMBER_NAMESIZE) == 0)
  68                         break;
  69         }
  70         rte_rwlock_read_unlock(RTE_EAL_TAILQ_RWLOCK);
  71
  72         if (te == NULL) {
  73                 rte_errno = ENOENT;
  74                 return NULL;
  75         }
  76         return setsum;
  77 }
  78
  79 void
  80 rte_member_free(struct rte_member_setsum *setsum)
  81 {
  82         struct rte_member_list *member_list;
  83         struct rte_tailq_entry *te;
  84
  85         if (setsum == NULL)
  86                 return;
  87         member_list = RTE_TAILQ_CAST(rte_member_tailq.head, rte_member_list);
  88         rte_rwlock_write_lock(RTE_EAL_TAILQ_RWLOCK);
  89         TAILQ_FOREACH(te, member_list, next) {
  90                 if (te->data == (void *)setsum)
  91                         break;
  92         }
  93         if (te == NULL) {
  94                 rte_rwlock_write_unlock(RTE_EAL_TAILQ_RWLOCK);
  95                 return;
  96         }
  97         TAILQ_REMOVE(member_list, te, next);
  98         rte_rwlock_write_unlock(RTE_EAL_TAILQ_RWLOCK);
  99
 100         switch (setsum->type) {
 101         case RTE_MEMBER_TYPE_HT:
 102                 rte_member_free_ht(setsum);
 103                 break;
 104         case RTE_MEMBER_TYPE_VBF:
 105                 rte_member_free_vbf(setsum);
 106                 break;
 107         default:
 108                 break;
 109         }
 110         rte_free(setsum);
 111         rte_free(te);
 112 }
 113
 114 struct rte_member_setsum *
 115 rte_member_create(const struct rte_member_parameters *params)
 116 {
 117         struct rte_tailq_entry *te;
 118         struct rte_member_list *member_list;
 119         struct rte_member_setsum *setsum;
 120         int ret;
 121
 122         if (params == NULL) {
 123                 rte_errno = EINVAL;
 124                 return NULL;
 125         }
 126
 127         if (params->key_len == 0 ||
 128                         params->prim_hash_seed == params->sec_hash_seed) {
 129                 rte_errno = EINVAL;
 130                 RTE_MEMBER_LOG(ERR, "Create setsummary with "
 131                                         "invalid parameters\n");
 132                 return NULL;
 133         }
 134
 135         member_list = RTE_TAILQ_CAST(rte_member_tailq.head, rte_member_list);
 136
 137         rte_rwlock_write_lock(RTE_EAL_TAILQ_RWLOCK);
 138
 139         TAILQ_FOREACH(te, member_list, next) {
 140                 setsum = (struct rte_member_setsum *) te->data;
 141                 if (strncmp(params->name, setsum->name,
 142                                 RTE_MEMBER_NAMESIZE) == 0)
 143                         break;
 144         }
 145         setsum = NULL;
 146         if (te != NULL) {
 147                 rte_errno = EEXIST;
 148                 te = NULL;
 149                 goto error_unlock_exit;
 150         }
 151         te = rte_zmalloc("MEMBER_TAILQ_ENTRY", sizeof(*te), 0);
 152         if (te == NULL) {
 153                 RTE_MEMBER_LOG(ERR, "tailq entry allocation failed\n");
 154                 goto error_unlock_exit;
 155         }
 156
 157         /* Create a new setsum structure */
 158         setsum = (struct rte_member_setsum *) rte_zmalloc_socket(params->name,
 159                         sizeof(struct rte_member_setsum), RTE_CACHE_LINE_SIZE,
 160                         params->socket_id);
 161         if (setsum == NULL) {
 162                 RTE_MEMBER_LOG(ERR, "Create setsummary failed\n");
 163                 goto error_unlock_exit;
 164         }
 165         snprintf(setsum->name, sizeof(setsum->name), "%s", params->name);
 166         setsum->type = params->type;
 167         setsum->socket_id = params->socket_id;
 168         setsum->key_len = params->key_len;
 169         setsum->num_set = params->num_set;
 170         setsum->prim_hash_seed = params->prim_hash_seed;
 171         setsum->sec_hash_seed = params->sec_hash_seed;
 172
 173         switch (setsum->type) {
 174         case RTE_MEMBER_TYPE_HT:
 175                 ret = rte_member_create_ht(setsum, params);
 176                 break;
 177         case RTE_MEMBER_TYPE_VBF:
 178                 ret = rte_member_create_vbf(setsum, params);
 179                 break;
 180         default:
 181                 goto error_unlock_exit;
 182         }
 183         if (ret < 0)
 184                 goto error_unlock_exit;
 185
 186         RTE_MEMBER_LOG(DEBUG, "Creating a setsummary table with "
 187                         "mode %u\n", setsum->type);
 188
 189         te->data = (void *)setsum;
 190         TAILQ_INSERT_TAIL(member_list, te, next);
 191         rte_rwlock_write_unlock(RTE_EAL_TAILQ_RWLOCK);
 192         return setsum;
 193
 194 error_unlock_exit:
 195         rte_rwlock_write_unlock(RTE_EAL_TAILQ_RWLOCK);
 196         rte_member_free(setsum);
 197         return NULL;
 198 }
 199
 200 int
 201 rte_member_add(const struct rte_member_setsum *setsum, const void *key,
 202                         member_set_t set_id)
 203 {
 204         if (setsum == NULL || key == NULL)
 205                 return -EINVAL;
 206
 207         switch (setsum->type) {
 208         case RTE_MEMBER_TYPE_HT:
 209                 return rte_member_add_ht(setsum, key, set_id);
 210         case RTE_MEMBER_TYPE_VBF:
 211                 return rte_member_add_vbf(setsum, key, set_id);
 212         default:
 213                 return -EINVAL;
 214         }
 215 }
 216
 217 int
 218 rte_member_lookup(const struct rte_member_setsum *setsum, const void *key,
 219                         member_set_t *set_id)
 220 {
 221         if (setsum == NULL || key == NULL || set_id == NULL)
 222                 return -EINVAL;
 223
 224         switch (setsum->type) {
 225         case RTE_MEMBER_TYPE_HT:
 226                 return rte_member_lookup_ht(setsum, key, set_id);
 227         case RTE_MEMBER_TYPE_VBF:
 228                 return rte_member_lookup_vbf(setsum, key, set_id);
 229         default:
 230                 return -EINVAL;
 231         }
 232 }
 233
 234 int
 235 rte_member_lookup_bulk(const struct rte_member_setsum *setsum,
 236                                 const void **keys, uint32_t num_keys,
 237                                 member_set_t *set_ids)
 238 {
 239         if (setsum == NULL || keys == NULL || set_ids == NULL)
 240                 return -EINVAL;
 241
 242         switch (setsum->type) {
 243         case RTE_MEMBER_TYPE_HT:
 244                 return rte_member_lookup_bulk_ht(setsum, keys, num_keys,
 245                                 set_ids);
 246         case RTE_MEMBER_TYPE_VBF:
 247                 return rte_member_lookup_bulk_vbf(setsum, keys, num_keys,
 248                                 set_ids);
 249         default:
 250                 return -EINVAL;
 251         }
 252 }
 253
 254 int
 255 rte_member_lookup_multi(const struct rte_member_setsum *setsum, const void *key,
 256                                 uint32_t match_per_key, member_set_t *set_id)
 257 {
 258         if (setsum == NULL || key == NULL || set_id == NULL)
 259                 return -EINVAL;
 260
 261         switch (setsum->type) {
 262         case RTE_MEMBER_TYPE_HT:
 263                 return rte_member_lookup_multi_ht(setsum, key, match_per_key,
 264                                 set_id);
 265         case RTE_MEMBER_TYPE_VBF:
 266                 return rte_member_lookup_multi_vbf(setsum, key, match_per_key,
 267                                 set_id);
 268         default:
 269                 return -EINVAL;
 270         }
 271 }
 272
 273 int
 274 rte_member_lookup_multi_bulk(const struct rte_member_setsum *setsum,
 275                         const void **keys, uint32_t num_keys,
 276                         uint32_t max_match_per_key, uint32_t *match_count,
 277                         member_set_t *set_ids)
 278 {
 279         if (setsum == NULL || keys == NULL || set_ids == NULL ||
 280                         match_count == NULL)
 281                 return -EINVAL;
 282
 283         switch (setsum->type) {
 284         case RTE_MEMBER_TYPE_HT:
 285                 return rte_member_lookup_multi_bulk_ht(setsum, keys, num_keys,
 286                                 max_match_per_key, match_count, set_ids);
 287         case RTE_MEMBER_TYPE_VBF:
 288                 return rte_member_lookup_multi_bulk_vbf(setsum, keys, num_keys,
 289                                 max_match_per_key, match_count, set_ids);
 290         default:
 291                 return -EINVAL;
 292         }
 293 }
 294
 295 int
 296 rte_member_delete(const struct rte_member_setsum *setsum, const void *key,
 297                         member_set_t set_id)
 298 {
 299         if (setsum == NULL || key == NULL)
 300                 return -EINVAL;
 301
 302         switch (setsum->type) {
 303         case RTE_MEMBER_TYPE_HT:
 304                 return rte_member_delete_ht(setsum, key, set_id);
 305         /* current vBF implementation does not support delete function */
 306         case RTE_MEMBER_TYPE_VBF:
 307         default:
 308                 return -EINVAL;
 309         }
 310 }
 311
 312 void
 313 rte_member_reset(const struct rte_member_setsum *setsum)
 314 {
 315         if (setsum == NULL)
 316                 return;
 317         switch (setsum->type) {
 318         case RTE_MEMBER_TYPE_HT:
 319                 rte_member_reset_ht(setsum);
 320                 return;
 321         case RTE_MEMBER_TYPE_VBF:
 322                 rte_member_reset_vbf(setsum);
 323                 return;
 324         default:
 325                 return;
 326         }
 327 }
 328
 329 RTE_INIT(librte_member_init_log);
 330
 331 static void
 332 librte_member_init_log(void)
 333 {
 334         librte_member_logtype = rte_log_register("librte.member");
 335         if (librte_member_logtype >= 0)
 336                 rte_log_set_level(librte_member_logtype, RTE_LOG_DEBUG);
 337 }