/* * Copyright (c) 2016 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 <vlib/vlib.h> #include <vnet/dpo/drop_dpo.h> #include <vnet/mfib/mfib_table.h> #include <vnet/mfib/ip4_mfib.h> #include <vnet/mfib/ip6_mfib.h> #include <vnet/mfib/mfib_entry.h> #include <vnet/mfib/mfib_signal.h> mfib_table_t * mfib_table_get (fib_node_index_t index, fib_protocol_t proto) { switch (proto) { case FIB_PROTOCOL_IP4: return (pool_elt_at_index(ip4_main.mfibs, index)); case FIB_PROTOCOL_IP6: return (pool_elt_at_index(ip6_main.mfibs, index)); case FIB_PROTOCOL_MPLS: break; } ASSERT(0); return (NULL); } static inline fib_node_index_t mfib_table_lookup_i (const mfib_table_t *mfib_table, const mfib_prefix_t *prefix) { switch (prefix->fp_proto) { case FIB_PROTOCOL_IP4: return (ip4_mfib_table_lookup(&mfib_table->v4, &prefix->fp_src_addr.ip4, &prefix->fp_grp_addr.ip4, prefix->fp_len)); case FIB_PROTOCOL_IP6: return (ip6_mfib_table_lookup(&mfib_table->v6, &prefix->fp_src_addr.ip6, &prefix->fp_grp_addr.ip6, prefix->fp_len)); case FIB_PROTOCOL_MPLS: break; } return (FIB_NODE_INDEX_INVALID); } fib_node_index_t mfib_table_lookup (u32 fib_index, const mfib_prefix_t *prefix) { return (mfib_table_lookup_i(mfib_table_get(fib_index, prefix->fp_proto), prefix)); } static inline fib_node_index_t mfib_table_lookup_exact_match_i (const mfib_table_t *mfib_table, const mfib_prefix_t *prefix) { switch (prefix->fp_proto) { case FIB_PROTOCOL_IP4: return (ip4_mfib_table_lookup_exact_match(&mfib_table->v4, &prefix->fp_grp_addr.ip4, &prefix->fp_src_addr.ip4, prefix->fp_len)); case FIB_PROTOCOL_IP6: return (ip6_mfib_table_lookup_exact_match(&mfib_table->v6, &prefix->fp_grp_addr.ip6, &prefix->fp_src_addr.ip6, prefix->fp_len)); case FIB_PROTOCOL_MPLS: break; } return (FIB_NODE_INDEX_INVALID); } fib_node_index_t mfib_table_lookup_exact_match (u32 fib_index, const mfib_prefix_t *prefix) { return (mfib_table_lookup_exact_match_i(mfib_table_get(fib_index, prefix->fp_proto), prefix)); } static void mfib_table_entry_remove (mfib_table_t *mfib_table, const mfib_prefix_t *prefix, fib_node_index_t fib_entry_index) { vlib_smp_unsafe_warning(); mfib_table->mft_total_route_counts--; switch (prefix->fp_proto) { case FIB_PROTOCOL_IP4: ip4_mfib_table_entry_remove(&mfib_table->v4, &prefix->fp_grp_addr.ip4, &prefix->fp_src_addr.ip4, prefix->fp_len); break; case FIB_PROTOCOL_IP6: ip6_mfib_table_entry_remove(&mfib_table->v6, &prefix->fp_grp_addr.ip6, &prefix->fp_src_addr.ip6, prefix->fp_len); break; case FIB_PROTOCOL_MPLS: ASSERT(0); break; } mfib_entry_unlock(fib_entry_index); } static void mfib_table_entry_insert (mfib_table_t *mfib_table, const mfib_prefix_t *prefix, fib_node_index_t mfib_entry_index) { vlib_smp_unsafe_warning(); mfib_entry_lock(mfib_entry_index); mfib_table->mft_total_route_counts++; switch (prefix->fp_proto) { case FIB_PROTOCOL_IP4: ip4_mfib_table_entry_insert(&mfib_table->v4, &prefix->fp_grp_addr.ip4, &prefix->fp_src_addr.ip4, prefix->fp_len, mfib_entry_index); break; case FIB_PROTOCOL_IP6: ip6_mfib_table_entry_insert(&mfib_table->v6, &prefix->fp_grp_addr.ip6, &prefix->fp_src_addr.ip6, prefix->fp_len, mfib_entry_index); break; case FIB_PROTOCOL_MPLS: break; } } fib_node_index_t mfib_table_entry_update (u32 fib_index, const mfib_prefix_t *prefix, mfib_source_t source, fib_rpf_id_t rpf_id, mfib_entry_flags_t entry_flags) { fib_node_index_t mfib_entry_index; mfib_table_t *mfib_table; mfib_table = mfib_table_get(fib_index, prefix->fp_proto); mfib_entry_index = mfib_table_lookup_exact_match_i(mfib_table, prefix); if (FIB_NODE_INDEX_INVALID == mfib_entry_index) { if (MFIB_ENTRY_FLAG_NONE != entry_flags) { /* * update to a non-existing entry with non-zero flags */ mfib_entry_index = mfib_entry_create(fib_index, source, prefix, rpf_id, entry_flags); mfib_table_entry_insert(mfib_table, prefix, mfib_entry_index); } /* * else * the entry doesn't exist and the request is to set no flags * the result would be an entry that doesn't exist - so do nothing */ } else { mfib_entry_lock(mfib_entry_index); if (mfib_entry_update(mfib_entry_index, source, entry_flags, rpf_id, INDEX_INVALID)) { /* * this update means we can now remove the entry. */ mfib_table_entry_remove(mfib_table, prefix, mfib_entry_index); } mfib_entry_unlock(mfib_entry_index); } return (mfib_entry_index); } fib_node_index_t mfib_table_entry_path_update (u32 fib_index, const mfib_prefix_t *prefix, mfib_source_t source, const fib_route_path_t *rpath, mfib_itf_flags_t itf_flags) { fib_node_index_t mfib_entry_index; mfib_table_t *mfib_table; mfib_table = mfib_table_get(fib_index, prefix->fp_proto); mfib_entry_index = mfib_table_lookup_exact_match_i(mfib_table, prefix); if (FIB_NODE_INDEX_INVALID == mfib_entry_index) { mfib_entry_index = mfib_entry_create(fib_index, source, prefix, MFIB_RPF_ID_NONE, MFIB_ENTRY_FLAG_NONE); mfib_table_entry_insert(mfib_table, prefix, mfib_entry_index); } mfib_entry_path_update(mfib_entry_index, source, rpath, itf_flags); return (mfib_entry_index); } void mfib_table_entry_path_remove (u32 fib_index, const mfib_prefix_t *prefix, mfib_source_t source, const fib_route_path_t *rpath) { fib_node_index_t mfib_entry_index; mfib_table_t *mfib_table; mfib_table = mfib_table_get(fib_index, prefix->fp_proto); mfib_entry_index = mfib_table_lookup_exact_match_i(mfib_table, prefix); if (FIB_NODE_INDEX_INVALID == mfib_entry_index) { /* * removing an etry that does not exist. i'll allow it. */ } else { int no_more_sources; /* * don't nobody go nowhere */ mfib_entry_lock(mfib_entry_index); no_more_sources = mfib_entry_path_remove(mfib_entry_index, source, rpath); if (no_more_sources) { /* * last source gone. remove from the table */ mfib_table_entry_remove(mfib_table, prefix, mfib_entry_index); } mfib_entry_unlock(mfib_entry_index); } } fib_node_index_t mfib_table_entry_special_add (u32 fib_index, const mfib_prefix_t *prefix, mfib_source_t source, mfib_entry_flags_t entry_flags, index_t rep_dpo) { fib_node_index_t mfib_entry_index; mfib_table_t *mfib_table; mfib_table = mfib_table_get(fib_index, prefix->fp_proto); mfib_entry_index = mfib_table_lookup_exact_match_i(mfib_table, prefix); if (FIB_NODE_INDEX_INVALID == mfib_entry_index) { mfib_entry_index = mfib_entry_create(fib_index, source, prefix, MFIB_RPF_ID_NONE, MFIB_ENTRY_FLAG_NONE); mfib_table_entry_insert(mfib_table, prefix, mfib_entry_index); } mfib_entry_update(mfib_entry_index, source, (MFIB_ENTRY_FLAG_EXCLUSIVE | entry_flags), MFIB_RPF_ID_NONE, rep_dpo); return (mfib_entry_index); } static void mfib_table_entry_delete_i (u32 fib_index, fib_node_index_t mfib_entry_index, const mfib_prefix_t *prefix, mfib_source_t source) { mfib_table_t *mfib_table; mfib_table = mfib_table_get(fib_index, prefix->fp_proto); /* * don't nobody go nowhere */ mfib_entry_lock(mfib_entry_index); if (mfib_entry_delete(mfib_entry_index, source)) { /* * last source gone. remove from the table */ mfib_table_entry_remove(mfib_table, prefix, mfib_entry_index); } /* * else * still has sources, leave it be. */ mfib_entry_unlock(mfib_entry_index); } void mfib_table_entry_delete (u32 fib_index, const mfib_prefix_t *prefix, mfib_source_t source) { fib_node_index_t mfib_entry_index; mfib_entry_index = mfib_table_lookup_exact_match(fib_index, prefix); if (FIB_NODE_INDEX_INVALID == mfib_entry_index) { /* * removing an etry that does not exist. * i'll allow it, but i won't like it. */ clib_warning("%U not in FIB", format_mfib_prefix, prefix); } else { mfib_table_entry_delete_i(fib_index, mfib_entry_index, prefix, source); } } void mfib_table_entry_delete_index (fib_node_index_t mfib_entry_index, mfib_source_t source) { mfib_prefix_t prefix; mfib_entry_get_prefix(mfib_entry_index, &prefix); mfib_table_entry_delete_i(mfib_entry_get_fib_index(mfib_entry_index), mfib_entry_index, &prefix, source); } u32 mfib_table_get_index_for_sw_if_index (fib_protocol_t proto, u32 sw_if_index) { switch (proto) { case FIB_PROTOCOL_IP4: return (ip4_mfib_table_get_index_for_sw_if_index(sw_if_index)); case FIB_PROTOCOL_IP6: return (ip6_mfib_table_get_index_for_sw_if_index(sw_if_index)); case FIB_PROTOCOL_MPLS: ASSERT(0); break; } return (~0); } u32 mfib_table_find (fib_protocol_t proto, u32 table_id) { switch (proto) { case FIB_PROTOCOL_IP4: return (ip4_mfib_index_from_table_id(table_id)); case FIB_PROTOCOL_IP6: return (ip6_mfib_index_from_table_id(table_id)); case FIB_PROTOCOL_MPLS: ASSERT(0); break; } return (~0); } static u32 mfib_table_find_or_create_and_lock_i (fib_protocol_t proto, u32 table_id, mfib_source_t src, const u8 *name) { mfib_table_t *mfib_table; fib_node_index_t fi; switch (proto) { case FIB_PROTOCOL_IP4: fi = ip4_mfib_table_find_or_create_and_lock(table_id, src); break; case FIB_PROTOCOL_IP6: fi = ip6_mfib_table_find_or_create_and_lock(table_id, src); break; case FIB_PROTOCOL_MPLS: default: return (~0); } mfib_table = mfib_table_get(fi, proto); if (NULL == mfib_table->mft_desc) { if (name && name[0]) { mfib_table->mft_desc = format(NULL, "%s", name); } else { mfib_table->mft_desc = format(NULL, "%U-VRF:%d", format_fib_protocol, proto, table_id); } } return (fi); } u32 mfib_table_find_or_create_and_lock (fib_protocol_t proto, u32 table_id, mfib_source_t src) { return (mfib_table_find_or_create_and_lock_i(proto, table_id, src, NULL)); } u32 mfib_table_find_or_create_and_lock_w_name (fib_protocol_t proto, u32 table_id, mfib_source_t src, const u8 *name) { return (mfib_table_find_or_create_and_lock_i(proto, table_id, src, name)); } /** * @brief Table flush context. Store the indicies of matching FIB entries * that need to be removed. */ typedef struct mfib_table_flush_ctx_t_ { /** * The list of entries to flush */ fib_node_index_t *mftf_entries; /** * The source we are flushing */ mfib_source_t mftf_source; } mfib_table_flush_ctx_t; static int mfib_table_flush_cb (fib_node_index_t mfib_entry_index, void *arg) { mfib_table_flush_ctx_t *ctx = arg; if (mfib_entry_is_sourced(mfib_entry_index, ctx->mftf_source)) { vec_add1(ctx->mftf_entries, mfib_entry_index); } return (1); } void mfib_table_flush (u32 mfib_index, fib_protocol_t proto, mfib_source_t source) { fib_node_index_t *mfib_entry_index; mfib_table_flush_ctx_t ctx = { .mftf_entries = NULL, .mftf_source = source, }; mfib_table_walk(mfib_index, proto, mfib_table_flush_cb, &ctx); vec_foreach(mfib_entry_index, ctx.mftf_entries) { mfib_table_entry_delete_index(*mfib_entry_index, source); } vec_free(ctx.mftf_entries); } static void mfib_table_destroy (mfib_table_t *mfib_table) { vec_free(mfib_table->mft_desc); switch (mfib_table->mft_proto) { case FIB_PROTOCOL_IP4: ip4_mfib_table_destroy(&mfib_table->v4); break; case FIB_PROTOCOL_IP6: ip6_mfib_table_destroy(&mfib_table->v6); break; case FIB_PROTOCOL_MPLS: ASSERT(0); break; } } void mfib_table_unlock (u32 fib_index, fib_protocol_t proto, mfib_source_t source) { mfib_table_t *mfib_table; mfib_table = mfib_table_get(fib_index, proto); mfib_table->mft_locks[source]--; mfib_table->mft_locks[MFIB_TABLE_TOTAL_LOCKS]--; if (0 == mfib_table->mft_locks[source]) { /* * The source no longer needs the table. flush any routes * from it just in case */ mfib_table_flush(fib_index, proto, source); } if (0 == mfib_table->mft_locks[MFIB_TABLE_TOTAL_LOCKS]) { /* * no more locak from any source - kill it */ mfib_table_destroy(mfib_table); } } void mfib_table_lock (u32 fib_index, fib_protocol_t proto, mfib_source_t source) { mfib_table_t *mfib_table; mfib_table = mfib_table_get(fib_index, proto); mfib_table->mft_locks[source]++; mfib_table->mft_locks[MFIB_TABLE_TOTAL_LOCKS]++; } void mfib_table_walk (u32 fib_index, fib_protocol_t proto, mfib_table_walk_fn_t fn, void *ctx) { switch (proto) { case FIB_PROTOCOL_IP4: ip4_mfib_table_walk(ip4_mfib_get(fib_index), fn, ctx); break; case FIB_PROTOCOL_IP6: ip6_mfib_table_walk(ip6_mfib_get(fib_index), fn, ctx); break; case FIB_PROTOCOL_MPLS: break; } } u8* format_mfib_table_name (u8* s, va_list *ap) { fib_node_index_t fib_index = va_arg(*ap, fib_node_index_t); fib_protocol_t proto = va_arg(*ap, int); // int promotion mfib_table_t *mfib_table; mfib_table = mfib_table_get(fib_index, proto); s = format(s, "%v", mfib_table->mft_desc); return (s); } u8 * format_mfib_table_memory (u8 *s, va_list *args) { s = format(s, "%U", format_ip4_mfib_table_memory); s = format(s, "%U", format_ip6_mfib_table_memory); return (s); } static clib_error_t * mfib_module_init (vlib_main_t * vm) { clib_error_t * error; mfib_entry_module_init(); mfib_signal_module_init(); if ((error = vlib_call_init_function (vm, fib_module_init))) return (error); if ((error = vlib_call_init_function (vm, rn_module_init))) return (error); return (error); } VLIB_INIT_FUNCTION(mfib_module_init);