1 files changed, 143 insertions, 0 deletions
diff --git a/src/vnet/mfib/mfib_entry_delegate.c b/src/vnet/mfib/mfib_entry_delegate.c
new file mode 100644
index 00000000000..95915cb3807
--- /dev/null
+++ b/src/vnet/mfib/mfib_entry_delegate.c
@@ -0,0 +1,143 @@
+/*
+ * Copyright (c) 2018 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/mfib/mfib_entry_delegate.h>
+#include <vnet/mfib/mfib_entry.h>
+
+static mfib_entry_delegate_t *
+mfib_entry_delegate_find_i (const mfib_entry_t *mfib_entry,
+                            mfib_entry_delegate_type_t type,
+                            u32 *index)
+{
+    mfib_entry_delegate_t *delegate;
+    int ii;
+
+    ii = 0;
+    vec_foreach(delegate, mfib_entry->fe_delegates)
+    {
+        if (delegate->mfd_type == type)
+        {
+            if (NULL != index)
+                *index = ii;
+
+            return (delegate);
+        }
+        else
+        {
+            ii++;
+        }
+    }
+
+    return (NULL);
+}
+
+mfib_entry_delegate_t *
+mfib_entry_delegate_get (const mfib_entry_t *mfib_entry,
+                         mfib_entry_delegate_type_t type)
+{
+    return (mfib_entry_delegate_find_i(mfib_entry, type, NULL));
+}
+
+void
+mfib_entry_delegate_remove (mfib_entry_t *mfib_entry,
+                            mfib_entry_delegate_type_t type)
+{
+    mfib_entry_delegate_t *fed;
+    u32 index = ~0;
+
+    fed = mfib_entry_delegate_find_i(mfib_entry, type, &index);
+
+    ASSERT(NULL != fed);
+
+    vec_del1(mfib_entry->fe_delegates, index);
+}
+
+static int
+mfib_entry_delegate_cmp_for_sort (void * v1,
+                                  void * v2)
+{
+    mfib_entry_delegate_t *delegate1 = v1, *delegate2 = v2;
+
+    return (delegate1->mfd_type - delegate2->mfd_type);
+}
+
+static void
+mfib_entry_delegate_init (mfib_entry_t *mfib_entry,
+                          mfib_entry_delegate_type_t type)
+
+{
+    mfib_entry_delegate_t delegate = {
+        .mfd_entry_index = mfib_entry_get_index(mfib_entry),
+        .mfd_type = type,
+    };
+
+    vec_add1(mfib_entry->fe_delegates, delegate);
+    vec_sort_with_function(mfib_entry->fe_delegates,
+                           mfib_entry_delegate_cmp_for_sort);
+}
+
+mfib_entry_delegate_t *
+mfib_entry_delegate_find_or_add (mfib_entry_t *mfib_entry,
+                                 mfib_entry_delegate_type_t fdt)
+{
+    mfib_entry_delegate_t *delegate;
+
+    delegate = mfib_entry_delegate_get(mfib_entry, fdt);
+
+    if (NULL == delegate)
+    {
+        mfib_entry_delegate_init(mfib_entry, fdt);
+    }
+
+    return (mfib_entry_delegate_get(mfib_entry, fdt));
+}
+
+/**
+ * typedef for printing a delegate
+ */
+typedef u8 * (*mfib_entry_delegate_format_t)(const mfib_entry_delegate_t *fed,
+                                             u8 *s);
+
+/**
+ * Print a delegate that represents cover tracking
+ */
+static u8 *
+mfib_entry_delegate_fmt_covered (const mfib_entry_delegate_t *fed,
+                                 u8 *s)
+{
+    s = format(s, "covered:[");
+    s = fib_node_children_format(fed->mfd_list, s);
+    s = format(s, "]");
+
+    return (s);
+}
+
+/**
+ * A delegate type to formatter map
+ */
+static mfib_entry_delegate_format_t fed_formatters[] =
+{
+    [MFIB_ENTRY_DELEGATE_COVERED] = mfib_entry_delegate_fmt_covered,
+};
+
+u8 *
+format_mfib_entry_deletegate (u8 * s, va_list * args)
+{
+    mfib_entry_delegate_t *fed;
+
+    fed = va_arg (*args, mfib_entry_delegate_t *);
+
+    return (fed_formatters[fed->mfd_type](fed, s));
+}