GBP V2

update the GBP plugin to implement the full NAT feature set of opflex agent

Change-Id: Ic06a039c889445ed0b9087fa1f292634192b0f8d
Signed-off-by: Neale Ranns <neale.ranns@cisco.com>

1 files changed, 263 insertions, 0 deletions

diff --git a/src/plugins/gbp/gbp_endpoint_group.c b/src/plugins/gbp/gbp_endpoint_group.c
new file mode 100644
index 00000000000..ed312d305a1
--- /dev/null
+++ b/src/plugins/gbp/gbp_endpoint_group.c
@@ -0,0 +1,263 @@
+/*
+ * gbp.h : Group Based Policy
+ *
+ * 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 <plugins/gbp/gbp_endpoint_group.h>
+#include <plugins/gbp/gbp_endpoint.h>
+
+#include <vnet/dpo/dvr_dpo.h>
+#include <vnet/fib/fib_table.h>
+
+/**
+ * Pool of GBP endpoint_groups
+ */
+gbp_endpoint_group_t *gbp_endpoint_group_pool;
+
+/**
+ * DB of endpoint_groups
+ */
+gbp_endpoint_group_db_t gbp_endpoint_group_db;
+
+gbp_endpoint_group_t *
+gbp_endpoint_group_find (epg_id_t epg_id)
+{
+  uword *p;
+
+  p = hash_get (gbp_endpoint_group_db.gepg_hash, epg_id);
+
+  if (NULL != p)
+    return (pool_elt_at_index (gbp_endpoint_group_pool, p[0]));
+
+  return (NULL);
+}
+
+int
+gbp_endpoint_group_add (epg_id_t epg_id,
+			u32 bd_id,
+			u32 ip4_table_id,
+			u32 ip6_table_id, u32 uplink_sw_if_index)
+{
+  gbp_endpoint_group_t *gepg;
+
+  gepg = gbp_endpoint_group_find (epg_id);
+
+  if (NULL == gepg)
+    {
+      fib_protocol_t fproto;
+
+      pool_get (gbp_endpoint_group_pool, gepg);
+      memset (gepg, 0, sizeof (*gepg));
+
+      gepg->gepg_id = epg_id;
+      gepg->gepg_bd = bd_id;
+      gepg->gepg_rd[FIB_PROTOCOL_IP4] = ip4_table_id;
+      gepg->gepg_rd[FIB_PROTOCOL_IP6] = ip6_table_id;
+      gepg->gepg_uplink_sw_if_index = uplink_sw_if_index;
+
+      /*
+       * an egress DVR dpo for internal subnets to use when sending
+       * on the uplink interface
+       */
+      FOR_EACH_FIB_IP_PROTOCOL (fproto)
+      {
+	gepg->gepg_fib_index[fproto] =
+	  fib_table_find_or_create_and_lock (fproto,
+					     gepg->gepg_rd[fproto],
+					     FIB_SOURCE_PLUGIN_HI);
+
+	if (~0 == gepg->gepg_fib_index[fproto])
+	  {
+	    return (VNET_API_ERROR_NO_SUCH_FIB);
+	  }
+
+	dvr_dpo_add_or_lock (uplink_sw_if_index,
+			     fib_proto_to_dpo (fproto),
+			     &gepg->gepg_dpo[fproto]);
+      }
+
+      /*
+       * packets direct from the uplink have had policy applied
+       */
+      l2input_intf_bitmap_enable (gepg->gepg_uplink_sw_if_index,
+				  L2INPUT_FEAT_GBP_NULL_CLASSIFY, 1);
+
+      hash_set (gbp_endpoint_group_db.gepg_hash,
+		gepg->gepg_id, gepg - gbp_endpoint_group_pool);
+
+    }
+
+  return (0);
+}
+
+void
+gbp_endpoint_group_delete (epg_id_t epg_id)
+{
+  gbp_endpoint_group_t *gepg;
+  uword *p;
+
+  p = hash_get (gbp_endpoint_group_db.gepg_hash, epg_id);
+
+  if (NULL != p)
+    {
+      fib_protocol_t fproto;
+
+      gepg = pool_elt_at_index (gbp_endpoint_group_pool, p[0]);
+
+      l2input_intf_bitmap_enable (gepg->gepg_uplink_sw_if_index,
+				  L2INPUT_FEAT_GBP_NULL_CLASSIFY, 0);
+
+      FOR_EACH_FIB_IP_PROTOCOL (fproto)
+      {
+	dpo_reset (&gepg->gepg_dpo[fproto]);
+	fib_table_unlock (gepg->gepg_fib_index[fproto],
+			  fproto, FIB_SOURCE_PLUGIN_HI);
+      }
+
+      hash_unset (gbp_endpoint_group_db.gepg_hash, epg_id);
+
+      pool_put (gbp_endpoint_group_pool, gepg);
+    }
+}
+
+void
+gbp_endpoint_group_walk (gbp_endpoint_group_cb_t cb, void *ctx)
+{
+  gbp_endpoint_group_t *gbpe;
+
+  /* *INDENT-OFF* */
+  pool_foreach(gbpe, gbp_endpoint_group_pool,
+  {
+    if (!cb(gbpe, ctx))
+      break;
+  });
+  /* *INDENT-ON* */
+}
+
+static clib_error_t *
+gbp_endpoint_group_cli (vlib_main_t * vm,
+			unformat_input_t * input, vlib_cli_command_t * cmd)
+{
+  vnet_main_t *vnm = vnet_get_main ();
+  epg_id_t epg_id = EPG_INVALID;
+  u32 uplink_sw_if_index = ~0;
+  u32 bd_id = ~0;
+  u32 rd_id = ~0;
+  u8 add = 1;
+
+  while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
+    {
+      if (unformat (input, "%U", unformat_vnet_sw_interface,
+		    vnm, &uplink_sw_if_index))
+	;
+      else if (unformat (input, "add"))
+	add = 1;
+      else if (unformat (input, "del"))
+	add = 0;
+      else if (unformat (input, "epg %d", &epg_id))
+	;
+      else if (unformat (input, "bd %d", &bd_id))
+	;
+      else if (unformat (input, "rd %d", &rd_id))
+	;
+      else
+	break;
+    }
+
+  if (EPG_INVALID == epg_id)
+    return clib_error_return (0, "EPG-ID must be specified");
+
+  if (add)
+    {
+      if (~0 == uplink_sw_if_index)
+	return clib_error_return (0, "interface must be specified");
+      if (~0 == bd_id)
+	return clib_error_return (0, "Bridge-domain must be specified");
+      if (~0 == rd_id)
+	return clib_error_return (0, "route-domain must be specified");
+
+      gbp_endpoint_group_add (epg_id, bd_id, rd_id, rd_id,
+			      uplink_sw_if_index);
+    }
+  else
+    gbp_endpoint_group_delete (epg_id);
+
+  return (NULL);
+}
+
+/*?
+ * Configure a GBP Endpoint Group
+ *
+ * @cliexpar
+ * @cliexstart{set gbp endpoint-group [del] epg <ID> bd <ID> <interface>}
+ * @cliexend
+ ?*/
+/* *INDENT-OFF* */
+VLIB_CLI_COMMAND (gbp_endpoint_group_cli_node, static) = {
+  .path = "gbp endpoint-group",
+  .short_help = "gbp endpoint-group [del] epg <ID> bd <ID> <interface>",
+  .function = gbp_endpoint_group_cli,
+};
+
+static int
+gbp_endpoint_group_show_one (gbp_endpoint_group_t *gepg, void *ctx)
+{
+  vnet_main_t *vnm = vnet_get_main ();
+  vlib_main_t *vm;
+
+  vm = ctx;
+  vlib_cli_output (vm, "  %d, bd:%d, ip4:%d ip6:%d uplink:%U",
+                   gepg->gepg_id,
+                   gepg->gepg_bd,
+                   gepg->gepg_rd[FIB_PROTOCOL_IP4],
+                   gepg->gepg_rd[FIB_PROTOCOL_IP6],
+		   format_vnet_sw_if_index_name, vnm, gepg->gepg_uplink_sw_if_index);
+
+  return (1);
+}
+
+static clib_error_t *
+gbp_endpoint_group_show (vlib_main_t * vm,
+		   unformat_input_t * input, vlib_cli_command_t * cmd)
+{
+  vlib_cli_output (vm, "Endpoint-Groups:");
+  gbp_endpoint_group_walk (gbp_endpoint_group_show_one, vm);
+
+  return (NULL);
+}
+
+
+/*?
+ * Show Group Based Policy Endpoint_Groups and derived information
+ *
+ * @cliexpar
+ * @cliexstart{show gbp endpoint_group}
+ * @cliexend
+ ?*/
+/* *INDENT-OFF* */
+VLIB_CLI_COMMAND (gbp_endpoint_group_show_node, static) = {
+  .path = "show gbp endpoint-group",
+  .short_help = "show gbp endpoint-group\n",
+  .function = gbp_endpoint_group_show,
+};
+/* *INDENT-ON* */
+
+/*
+ * fd.io coding-style-patch-verification: ON
+ *
+ * Local Variables:
+ * eval: (c-set-style "gnu")
+ * End:
+ */