IGMP improvements
[vpp.git] / src / plugins / igmp / igmp_cli.c
diff --git a/src/plugins/igmp/igmp_cli.c b/src/plugins/igmp/igmp_cli.c
new file mode 100644 (file)
index 0000000..5f09589
--- /dev/null
@@ -0,0 +1,263 @@
+/*
+ *------------------------------------------------------------------
+ * 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 <stdint.h>
+#include <sys/ioctl.h>
+#include <inttypes.h>
+
+#include <vlib/vlib.h>
+#include <vlib/unix/unix.h>
+#include <vnet/ip/ip.h>
+#include <vnet/fib/fib_entry.h>
+#include <vnet/fib/fib_table.h>
+#include <vnet/mfib/mfib_table.h>
+
+#include <igmp/igmp.h>
+
+static clib_error_t *
+igmp_clear_interface_command_fn (vlib_main_t * vm, unformat_input_t * input,
+                                vlib_cli_command_t * cmd)
+{
+  unformat_input_t _line_input, *line_input = &_line_input;
+  clib_error_t *error = NULL;
+  vnet_main_t *vnm = vnet_get_main ();
+  u32 sw_if_index;
+
+  igmp_config_t *config;
+
+  if (!unformat_user (input, unformat_line_input, line_input))
+    {
+      error =
+       clib_error_return (0, "'help clear igmp' or 'clear igmp ?' for help");
+      return error;
+    }
+
+  while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
+    {
+      if (unformat
+         (line_input, "int %U", unformat_vnet_sw_interface, vnm,
+          &sw_if_index));
+      else
+       {
+         error =
+           clib_error_return (0, "unknown input '%U'", format_unformat_error,
+                              line_input);
+         goto done;
+       }
+    }
+
+  config = igmp_config_lookup (sw_if_index);
+  if (config)
+    igmp_clear_config (config);
+
+done:
+  unformat_free (line_input);
+  return error;
+}
+
+/* *INDENT-OFF* */
+VLIB_CLI_COMMAND (igmp_clear_interface_command, static) = {
+  .path = "clear igmp",
+  .short_help = "clear igmp int <interface>",
+  .function = igmp_clear_interface_command_fn,
+};
+/* *INDENT-ON* */
+
+static clib_error_t *
+igmp_listen_command_fn (vlib_main_t * vm, unformat_input_t * input,
+                       vlib_cli_command_t * cmd)
+{
+  unformat_input_t _line_input, *line_input = &_line_input;
+  clib_error_t *error = NULL;
+  u8 enable = 1;
+  ip46_address_t saddr, gaddr;
+  vnet_main_t *vnm = vnet_get_main ();
+  u32 sw_if_index;
+  int rv;
+
+  if (!unformat_user (input, unformat_line_input, line_input))
+    {
+      error =
+       clib_error_return (0,
+                          "'help igmp listen' or 'igmp listen ?' for help");
+      return error;
+    }
+
+  while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
+    {
+      if (unformat (line_input, "enable"))
+       enable = 1;
+      else if (unformat (line_input, "disable"))
+       enable = 0;
+      else
+       if (unformat
+           (line_input, "int %U", unformat_vnet_sw_interface, vnm,
+            &sw_if_index));
+      else
+       if (unformat (line_input, "saddr %U", unformat_ip46_address, &saddr));
+      else
+       if (unformat (line_input, "gaddr %U", unformat_ip46_address, &gaddr));
+      else
+       {
+         error =
+           clib_error_return (0, "unknown input '%U'", format_unformat_error,
+                              line_input);
+         goto done;
+       }
+    }
+
+  if ((vnet_sw_interface_get_flags (vnm, sw_if_index)
+       && VNET_SW_INTERFACE_FLAG_ADMIN_UP) == 0)
+    {
+      error = clib_error_return (0, "Interface is down");
+      goto done;
+    }
+
+  rv = igmp_listen (vm, enable, sw_if_index, &saddr, &gaddr);
+
+  if (rv == -1)
+    {
+      if (enable)
+       error =
+         clib_error_return (0, "This igmp configuration already exists");
+      else
+       error =
+         clib_error_return (0, "This igmp configuration does not nexist");
+    }
+  else if (rv == -2)
+    error =
+      clib_error_return (0,
+                        "Failed to add configuration, interface is in router mode");
+
+done:
+  unformat_free (line_input);
+  return error;
+}
+
+/* *INDENT-OFF* */
+VLIB_CLI_COMMAND (igmp_listen_command, static) = {
+  .path = "igmp listen",
+  .short_help = "igmp listen [<enable|disable>] "
+                "int <interface> saddr <ip4-address> gaddr <ip4-address>",
+  .function = igmp_listen_command_fn,
+};
+/* *INDENT-ON* */
+
+static clib_error_t *
+igmp_show_command_fn (vlib_main_t * vm, unformat_input_t * input,
+                     vlib_cli_command_t * cmd)
+{
+  clib_error_t *error = NULL;
+  igmp_main_t *im = &igmp_main;
+  vnet_main_t *vnm = vnet_get_main ();
+  igmp_config_t *config;
+  igmp_group_t *group;
+  igmp_src_t *src;
+
+  /* *INDENT-OFF* */
+  pool_foreach (config, im->configs,
+    ({
+      vlib_cli_output (vm, "interface: %U", format_vnet_sw_if_index_name,
+                     vnm, config->sw_if_index);
+
+      FOR_EACH_GROUP (group, config,
+        ({
+          vlib_cli_output (vm, "\t%U", format_igmp_key, group->key);
+          FOR_EACH_SRC (src, group, IGMP_FILTER_MODE_INCLUDE,
+          ({
+              vlib_cli_output (vm, "\t\t%U", format_igmp_key, src->key);
+            }));
+        }));
+    }));
+  /* *INDENT-ON* */
+
+  return error;
+}
+
+/* *INDENT-OFF* */
+VLIB_CLI_COMMAND (igmp_show_command, static) = {
+  .path = "show igmp config",
+  .short_help = "show igmp config",
+  .function = igmp_show_command_fn,
+};
+/* *INDENT-ON* */
+
+static clib_error_t *
+igmp_show_timers_command_fn (vlib_main_t * vm,
+                            unformat_input_t * input,
+                            vlib_cli_command_t * cmd)
+{
+#define _(n,f) vlib_cli_output (vm, "%s: %d", #f, igmp_timer_type_get(n));
+  foreach_igmp_timer_type
+#undef _
+    return (NULL);
+}
+
+/* *INDENT-OFF* */
+VLIB_CLI_COMMAND (igmp_show_timers_command, static) = {
+  .path = "show igmp timers",
+  .short_help = "show igmp timers",
+  .function = igmp_show_timers_command_fn,
+};
+/* *INDENT-ON* */
+
+static clib_error_t *
+test_igmp_command_fn (vlib_main_t * vm,
+                     unformat_input_t * input, vlib_cli_command_t * cmd)
+{
+  clib_error_t *error = NULL;
+  u32 value;
+
+  while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
+    {
+      if (unformat (input, "query %d", &value))
+       igmp_timer_type_set (IGMP_TIMER_QUERY, value);
+      else if (unformat (input, "src %d", &value))
+       igmp_timer_type_set (IGMP_TIMER_SRC, value);
+      else if (unformat (input, "leave %d", &value))
+       igmp_timer_type_set (IGMP_TIMER_LEAVE, value);
+      else
+       error = clib_error_return (0, "query or src timers only");
+    }
+
+  return error;
+}
+
+/* *INDENT-OFF* */
+VLIB_CLI_COMMAND (test_igmp_command, static) = {
+  .path = "test igmp timers",
+  .short_help = "Change the default values for IGMP timers - only sensible during unit tests",
+  .function = test_igmp_command_fn,
+};
+/* *INDENT-ON* */
+
+
+clib_error_t *
+igmp_cli_init (vlib_main_t * vm)
+{
+  return 0;
+}
+
+VLIB_INIT_FUNCTION (igmp_cli_init);
+
+/*
+ * fd.io coding-style-patch-verification: ON
+ *
+ * Local Variables:
+ * eval: (c-set-style "gnu")
+ * End:
+ */