1 files changed, 396 insertions, 0 deletions
diff --git a/src/vnet/bier/bier_disp_table.c b/src/vnet/bier/bier_disp_table.c
new file mode 100644
index 00000000000..5caf86d3705
--- /dev/null
+++ b/src/vnet/bier/bier_disp_table.c
@@ -0,0 +1,396 @@
+/*
+ * Copyright (c) 2017 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 <vnet/bier/bier_disp_table.h>
+#include <vnet/bier/bier_disp_entry.h>
+
+/**
+ * memory pool for disposition tables
+ */
+bier_disp_table_t *bier_disp_table_pool;
+
+/**
+ * Hash table to map client table IDs to VPP index
+ */
+static uword *bier_disp_table_id_to_index;
+
+static index_t
+bier_disp_table_get_index (const bier_disp_table_t *bdt)
+{
+    return (bdt - bier_disp_table_pool);
+}
+
+static void
+bier_disp_table_lock_i (bier_disp_table_t *bdt)
+{
+    bdt->bdt_locks++;
+}
+
+index_t
+bier_disp_table_find(u32 table_id)
+{
+    uword *p;
+
+    p = hash_get(bier_disp_table_id_to_index, table_id);
+
+    if (NULL != p)
+    {
+        return (p[0]);
+    }
+
+    return (INDEX_INVALID);
+}
+
+index_t
+bier_disp_table_add_or_lock (u32 table_id)
+{
+    bier_disp_table_t *bdt;
+    index_t bdti;
+
+    bdti = bier_disp_table_find(table_id);
+
+    if (INDEX_INVALID == bdti)
+    {
+        pool_get_aligned(bier_disp_table_pool, bdt,
+                         CLIB_CACHE_LINE_BYTES);
+
+        bdt->bdt_table_id = table_id;
+        bdt->bdt_locks = 0;
+
+        hash_set(bier_disp_table_id_to_index, table_id,
+                 bier_disp_table_get_index(bdt));
+
+        /**
+         * Set the result for each entry in the DB to be invalid
+         */
+        memset(bdt->bdt_db, 0xff, sizeof(bdt->bdt_db));
+    }
+    else
+    {
+        bdt = pool_elt_at_index(bier_disp_table_pool, bdti);
+    }
+
+    bier_disp_table_lock_i(bdt);
+
+    return (bier_disp_table_get_index(bdt));
+}
+
+void
+bier_disp_table_unlock_w_table_id (u32 table_id)
+{
+    index_t bdti;
+
+    bdti = bier_disp_table_find(table_id);
+
+    if (INDEX_INVALID != bdti)
+    {
+        bier_disp_table_unlock(bdti);
+    }
+}
+
+void
+bier_disp_table_unlock (index_t bdti)
+{
+    bier_disp_table_t *bdt;
+
+    bdt = bier_disp_table_get(bdti);
+
+    bdt->bdt_locks--;
+
+    if (0 == bdt->bdt_locks)
+    {
+        u32 ii;
+
+        for (ii = 0; ii < BIER_BP_MAX; ii++)
+        {
+            bier_disp_entry_unlock(bdt->bdt_db[ii]);
+        }
+        hash_unset(bier_disp_table_id_to_index, bdt->bdt_table_id);
+        pool_put(bier_disp_table_pool, bdt);
+    }
+}
+
+void
+bier_disp_table_lock (index_t bdti)
+{
+    bier_disp_table_lock_i(bier_disp_table_get(bdti));
+}
+
+void
+bier_disp_table_contribute_forwarding (index_t bdti,
+                                       dpo_id_t *dpo)
+{
+    dpo_set(dpo,
+            DPO_BIER_DISP_TABLE,
+            DPO_PROTO_BIER,
+            bdti);
+}
+
+
+u8*
+format_bier_disp_table (u8* s, va_list *ap)
+{
+    index_t bdti = va_arg(*ap, index_t);
+    u32 indent = va_arg(*ap, u32);
+    bier_show_flags_t flags = va_arg(*ap, bier_show_flags_t);
+    bier_disp_table_t *bdt;
+
+    bdt = bier_disp_table_get(bdti);
+
+    s = format(s, "bier-disp-table:[%d]; table-id:%d locks:%d",
+               bdti, bdt->bdt_table_id, bdt->bdt_locks);
+
+    if (flags & BIER_SHOW_DETAIL)
+    {
+        u32 ii;
+
+        for (ii = 0; ii < BIER_BP_MAX; ii++)
+        {
+            if (INDEX_INVALID != bdt->bdt_db[ii])
+            {
+                u16 src = ii;
+                s = format(s, "\n%Usrc:%d", format_white_space, indent,
+                           clib_host_to_net_u16(src));
+                s = format(s, "\n%U%U", format_white_space, indent+2,
+                           format_bier_disp_entry, bdt->bdt_db[ii],
+                           indent+4, BIER_SHOW_BRIEF);
+            }
+        }
+    }
+    return (s);
+}
+
+static u8*
+format_bier_disp_table_dpo (u8* s, va_list *ap)
+{
+    index_t bdti = va_arg(*ap, index_t);
+    u32 indent = va_arg(*ap, u32);
+
+    return (format(s, "%U",
+                   format_bier_disp_table, bdti, indent,
+                   BIER_SHOW_BRIEF));
+}
+
+static void
+bier_disp_table_entry_insert (index_t bdti,
+                              bier_bp_t src,
+                              index_t bdei)
+{
+    bier_disp_table_t *bdt;
+
+    bdt = bier_disp_table_get(bdti);
+    bdt->bdt_db[clib_host_to_net_u16(src)] = bdei;
+}
+
+static void
+bier_disp_table_entry_remove (index_t bdti,
+                              bier_bp_t src)
+{
+    bier_disp_table_t *bdt;
+
+    bdt = bier_disp_table_get(bdti);
+    bdt->bdt_db[clib_host_to_net_u16(src)] = INDEX_INVALID;
+}
+
+static index_t
+bier_disp_table_lookup_hton(index_t bdti,
+                            bier_bp_t src)
+{
+    return (bier_disp_table_lookup(bdti, clib_host_to_net_u16(src)));
+}
+
+void
+bier_disp_table_entry_path_add (u32 table_id,
+                                bier_bp_t src,
+                                bier_hdr_proto_id_t payload_proto,
+                                const fib_route_path_t *rpaths)
+{
+    index_t bdti, bdei;
+
+    bdti = bier_disp_table_find(table_id);
+
+    if (INDEX_INVALID == bdti)
+    {
+        return;
+    }
+
+    bdei = bier_disp_table_lookup_hton(bdti, src);
+
+    if (INDEX_INVALID == bdei)
+    {
+        bdei = bier_disp_entry_add_or_lock();
+        bier_disp_table_entry_insert(bdti, src, bdei);
+    }
+
+    bier_disp_entry_path_add(bdei, payload_proto, rpaths);
+}
+
+void
+bier_disp_table_entry_path_remove (u32 table_id,
+                                   bier_bp_t src,
+                                   bier_hdr_proto_id_t payload_proto,
+                                   const fib_route_path_t *rpath)
+{
+    index_t bdti, bdei;
+
+    bdti = bier_disp_table_find(table_id);
+
+    if (INDEX_INVALID == bdti)
+    {
+        return;
+    }
+
+    bdei = bier_disp_table_lookup_hton(bdti, src);
+
+    if (INDEX_INVALID != bdei)
+    {
+        int remove;
+
+        remove = bier_disp_entry_path_remove(bdei, payload_proto, rpath);
+
+        if (remove)
+        {
+            bier_disp_table_entry_remove(bdti, src);
+            bier_disp_entry_unlock(bdei);
+        }
+    }
+}
+
+void
+bier_disp_table_walk (u32 table_id,
+                      bier_disp_table_walk_fn_t fn,
+                      void *ctx)
+{
+    const bier_disp_table_t *bdt;
+    const bier_disp_entry_t *bde;
+    index_t bdti;
+    u32 ii;
+
+    bdti = bier_disp_table_find(table_id);
+
+    if (INDEX_INVALID != bdti)
+    {
+        bdt = bier_disp_table_get(bdti);
+
+        for (ii = 0; ii < BIER_BP_MAX; ii++)
+        {
+            if (INDEX_INVALID != bdt->bdt_db[ii])
+            {
+                u16 src = ii;
+
+                bde = bier_disp_entry_get(bdt->bdt_db[ii]);
+
+                fn(bdt, bde, clib_host_to_net_u16(src), ctx);
+            }
+        }
+    }
+}
+
+static void
+bier_disp_table_dpo_lock (dpo_id_t *dpo)
+{
+    bier_disp_table_lock(dpo->dpoi_index);
+}
+
+static void
+bier_disp_table_dpo_unlock (dpo_id_t *dpo)
+{
+    bier_disp_table_unlock(dpo->dpoi_index);
+}
+
+static void
+bier_disp_table_dpo_mem_show (void)
+{
+    fib_show_memory_usage("BIER disposition table",
+                          pool_elts(bier_disp_table_pool),
+                          pool_len(bier_disp_table_pool),
+                          sizeof(bier_disp_table_t));
+}
+
+const static dpo_vft_t bier_disp_table_dpo_vft = {
+    .dv_lock = bier_disp_table_dpo_lock,
+    .dv_unlock = bier_disp_table_dpo_unlock,
+    .dv_mem_show = bier_disp_table_dpo_mem_show,
+    .dv_format = format_bier_disp_table_dpo,
+};
+
+const static char *const bier_disp_table_bier_nodes[] =
+{
+    "bier-disp-lookup"
+};
+
+const static char * const * const bier_disp_table_nodes[DPO_PROTO_NUM] =
+{
+    [DPO_PROTO_BIER] = bier_disp_table_bier_nodes,
+};
+
+clib_error_t *
+bier_disp_table_module_init (vlib_main_t *vm)
+{
+    dpo_register(DPO_BIER_DISP_TABLE,
+                 &bier_disp_table_dpo_vft,
+                 bier_disp_table_nodes);
+
+    bier_disp_table_id_to_index = hash_create(0, sizeof(index_t));
+
+    return (NULL);
+}
+
+VLIB_INIT_FUNCTION (bier_disp_table_module_init);
+
+static clib_error_t *
+show_bier_disp_table (vlib_main_t * vm,
+                      unformat_input_t * input,
+                      vlib_cli_command_t * cmd)
+{
+    bier_disp_table_t *bdt;
+    index_t bdti;
+
+    bdti = INDEX_INVALID;
+
+    while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) {
+        if (unformat (input, "%d", &bdti))
+            ;
+        else if (unformat (input, "%d", &bdti))
+            ;
+        else
+        {
+            break;
+        }
+    }
+
+    if (INDEX_INVALID == bdti)
+    {
+        pool_foreach(bdt, bier_disp_table_pool,
+        ({
+            vlib_cli_output(vm, "%U", format_bier_disp_table,
+                            bier_disp_table_get_index(bdt),
+                            1,
+                            BIER_SHOW_BRIEF);
+        }));
+    }
+    else
+    {
+        vlib_cli_output(vm, "%U", format_bier_disp_table, bdti, 1,
+                        BIER_SHOW_DETAIL);
+    }
+    return (NULL);
+}
+
+VLIB_CLI_COMMAND (show_bier_disp_table_node, static) = {
+    .path = "show bier disp table",
+    .short_help = "show bier disp table [index]",
+    .function = show_bier_disp_table,
+};