FIB table add/delete API

[vpp.git] / src / vnet / fib / fib_table.c

diff --git a/src/vnet/fib/fib_table.c b/src/vnet/fib/fib_table.c
index 0938ce9..75d1562 100644 (file)
--- a/src/vnet/fib/fib_table.c
+++ b/src/vnet/fib/fib_table.c
@@ -505,7 +505,7 @@ fib_table_entry_path_add (u32 fib_index,
                          const fib_prefix_t *prefix,
                          fib_source_t source,
                          fib_entry_flag_t flags,
-                         fib_protocol_t next_hop_proto,
+                         dpo_proto_t next_hop_proto,
                          const ip46_address_t *next_hop,
                          u32 next_hop_sw_if_index,
                          u32 next_hop_fib_index,
@@ -608,11 +608,19 @@ fib_table_entry_path_remove2 (u32 fib_index,
        fib_entry_src_flag_t src_flag;
         int was_sourced;
 
-       /*
+        /*
+         * if it's not sourced, then there's nowt to remove
+         */
+        was_sourced = fib_entry_is_sourced(fib_entry_index, source);
+        if (!was_sourced)
+        {
+            return;
+        }
+
+        /*
         * don't nobody go nowhere
         */
        fib_entry_lock(fib_entry_index);
-        was_sourced = fib_entry_is_sourced(fib_entry_index, source);
 
         for (ii = 0; ii < vec_len(rpath); ii++)
         {
@@ -656,7 +664,7 @@ void
 fib_table_entry_path_remove (u32 fib_index,
                             const fib_prefix_t *prefix,
                             fib_source_t source,
-                            fib_protocol_t next_hop_proto,
+                            dpo_proto_t next_hop_proto,
                             const ip46_address_t *next_hop,
                             u32 next_hop_sw_if_index,
                             u32 next_hop_fib_index,
@@ -747,7 +755,7 @@ fib_table_entry_update_one_path (u32 fib_index,
                                 const fib_prefix_t *prefix,
                                 fib_source_t source,
                                 fib_entry_flag_t flags,
-                                fib_protocol_t next_hop_proto,
+                                dpo_proto_t next_hop_proto,
                                 const ip46_address_t *next_hop,
                                 u32 next_hop_sw_if_index,
                                 u32 next_hop_fib_index,
@@ -936,19 +944,69 @@ fib_table_get_index_for_sw_if_index (fib_protocol_t proto,
 flow_hash_config_t
 fib_table_get_flow_hash_config (u32 fib_index,
                                fib_protocol_t proto)
+{
+    fib_table_t *fib;
+
+    fib = fib_table_get(fib_index, proto);
+
+    return (fib->ft_flow_hash_config);
+}
+flow_hash_config_t
+fib_table_get_default_flow_hash_config (fib_protocol_t proto)
 {
     switch (proto)
     {
     case FIB_PROTOCOL_IP4:
-       return (ip4_fib_table_get_flow_hash_config(fib_index));
     case FIB_PROTOCOL_IP6:
-       return (ip6_fib_table_get_flow_hash_config(fib_index));
+       return (IP_FLOW_HASH_DEFAULT);
+
     case FIB_PROTOCOL_MPLS:
-       return (mpls_fib_table_get_flow_hash_config(fib_index));
+       return (MPLS_FLOW_HASH_DEFAULT);
     }
-    return (0);
+
+    ASSERT(0);
+    return (IP_FLOW_HASH_DEFAULT);
 }
 
+/**
+ * @brief Table set flow hash config context.
+ */
+typedef struct fib_table_set_flow_hash_config_ctx_t_
+{
+    /**
+     * the flow hash config to set
+     */
+    flow_hash_config_t hash_config;
+} fib_table_set_flow_hash_config_ctx_t;
+
+static int
+fib_table_set_flow_hash_config_cb (fib_node_index_t fib_entry_index,
+                                   void *arg)
+{
+    fib_table_set_flow_hash_config_ctx_t *ctx = arg;
+
+    fib_entry_set_flow_hash_config(fib_entry_index, ctx->hash_config);
+
+    return (1);
+}
+
+void
+fib_table_set_flow_hash_config (u32 fib_index,
+                                fib_protocol_t proto,
+                                flow_hash_config_t hash_config)
+{
+    fib_table_set_flow_hash_config_ctx_t ctx = {
+        .hash_config = hash_config,
+    };
+    fib_table_t *fib;
+
+    fib = fib_table_get(fib_index, proto);
+    fib->ft_flow_hash_config = hash_config;
+
+    fib_table_walk(fib_index, proto,
+                   fib_table_set_flow_hash_config_cb,
+                   &ctx);
+}
 
 u32
 fib_table_get_table_id_for_sw_if_index (fib_protocol_t proto,
@@ -981,7 +1039,8 @@ fib_table_find (fib_protocol_t proto,
 
 u32
 fib_table_find_or_create_and_lock (fib_protocol_t proto,
-                                  u32 table_id)
+                                  u32 table_id,
+                                   fib_source_t src)
 {
     fib_table_t *fib_table;
     fib_node_index_t fi;
@@ -989,13 +1048,13 @@ fib_table_find_or_create_and_lock (fib_protocol_t proto,
     switch (proto)
     {
     case FIB_PROTOCOL_IP4:
-       fi = ip4_fib_table_find_or_create_and_lock(table_id);
+       fi = ip4_fib_table_find_or_create_and_lock(table_id, src);
         break;
     case FIB_PROTOCOL_IP6:
-       fi = ip6_fib_table_find_or_create_and_lock(table_id);
+       fi = ip6_fib_table_find_or_create_and_lock(table_id, src);
         break;
     case FIB_PROTOCOL_MPLS:
-       fi = mpls_fib_table_find_or_create_and_lock(table_id);
+       fi = mpls_fib_table_find_or_create_and_lock(table_id, src);
         break;
     default:
         return (~0);        
@@ -1012,6 +1071,7 @@ fib_table_find_or_create_and_lock (fib_protocol_t proto,
 
 u32
 fib_table_create_and_lock (fib_protocol_t proto,
+                           fib_source_t src,
                            const char *const fmt,
                            ...)
 {
@@ -1024,13 +1084,13 @@ fib_table_create_and_lock (fib_protocol_t proto,
     switch (proto)
     {
     case FIB_PROTOCOL_IP4:
-       fi = ip4_fib_table_create_and_lock();
+       fi = ip4_fib_table_create_and_lock(src);
         break;
     case FIB_PROTOCOL_IP6:
-       fi = ip6_fib_table_create_and_lock();
+       fi = ip6_fib_table_create_and_lock(src);
         break;
      case FIB_PROTOCOL_MPLS:
-       fi = mpls_fib_table_create_and_lock();
+       fi = mpls_fib_table_create_and_lock(src);
         break;
    default:
         return (~0);        
@@ -1085,26 +1145,43 @@ fib_table_walk (u32 fib_index,
 
 void
 fib_table_unlock (u32 fib_index,
-                 fib_protocol_t proto)
+                 fib_protocol_t proto,
+                  fib_source_t source)
 {
     fib_table_t *fib_table;
 
     fib_table = fib_table_get(fib_index, proto);
-    fib_table->ft_locks--;
+    fib_table->ft_locks[source]--;
+    fib_table->ft_locks[FIB_TABLE_TOTAL_LOCKS]--;
 
-    if (0 == fib_table->ft_locks)
+    if (0 == fib_table->ft_locks[source])
     {
+        /*
+         * The source no longer needs the table. flush any routes
+         * from it just in case
+         */
+        fib_table_flush(fib_index, proto, source);
+    }
+
+    if (0 == fib_table->ft_locks[FIB_TABLE_TOTAL_LOCKS])
+    {
+        /*
+         * no more locak from any source - kill it
+         */
        fib_table_destroy(fib_table);
     }
 }
+
 void
 fib_table_lock (u32 fib_index,
-               fib_protocol_t proto)
+               fib_protocol_t proto,
+                fib_source_t source)
 {
     fib_table_t *fib_table;
 
     fib_table = fib_table_get(fib_index, proto);
-    fib_table->ft_locks++;
+    fib_table->ft_locks[source]++;
+    fib_table->ft_locks[FIB_TABLE_TOTAL_LOCKS]++;
 }
 
 u32