From: Donald Lee <dlqs@gmx.com>
Date: Wed, 13 Oct 2021 15:08:31 +0000 (+0800)
Subject: zebra: Add vty cmd zebra on-rib-process script SCRIPT
X-Git-Tag: base_8.2~287^2~1
X-Git-Url: https://git.puffer.fish/?a=commitdiff_plain;h=310dd2b362b77b470540eedf4cc162298bdd586d;p=matthieu%2Ffrr.git

zebra: Add vty cmd zebra on-rib-process script SCRIPT

Signed-off-by: Donald Lee <dlqs@gmx.com>
---

diff --git a/zebra/zebra_vty.c b/zebra/zebra_vty.c
index c6b3eb48a0..76c1014212 100644
--- a/zebra/zebra_vty.c
+++ b/zebra/zebra_vty.c
@@ -60,6 +60,7 @@
 #include "zebra/zebra_nb.h"
 #include "zebra/kernel_netlink.h"
 #include "zebra/table_manager.h"
+#include "zebra/zebra_script.h"
 
 extern int allow_delete;
 
@@ -4317,6 +4318,30 @@ DEFUN(ip_table_range, ip_table_range_cmd,
 	return table_manager_range(vty, true, zvrf, argv[3]->arg, argv[4]->arg);
 }
 
+#ifdef HAVE_SCRIPTING
+
+DEFUN(zebra_on_rib_process_script, zebra_on_rib_process_script_cmd,
+      "zebra on-rib-process script SCRIPT",
+      ZEBRA_STR
+      "on_rib_process_dplane_results hook call\n"
+      "Set a script\n"
+      "Script name (same as filename in /etc/frr/scripts/, without .lua)\n")
+{
+
+	if (frrscript_names_set_script_name(ZEBRA_ON_RIB_PROCESS_HOOK_CALL,
+					    argv[3]->arg)
+	    == 0) {
+		vty_out(vty, "Successfully added script %s for hook call %s\n",
+			argv[3]->arg, ZEBRA_ON_RIB_PROCESS_HOOK_CALL);
+	} else {
+		vty_out(vty, "Failed to add script %s for hook call %s\n",
+			argv[3]->arg, ZEBRA_ON_RIB_PROCESS_HOOK_CALL);
+	}
+	return CMD_SUCCESS;
+}
+
+#endif /* HAVE_SCRIPTING */
+
 /* IP node for static routes. */
 static int zebra_ip_config(struct vty *vty);
 static struct cmd_node ip_node = {
@@ -4473,5 +4498,9 @@ void zebra_vty_init(void)
 	install_element(CONFIG_NODE, &no_zebra_kernel_netlink_batch_tx_buf_cmd);
 #endif /* HAVE_NETLINK */
 
+#ifdef HAVE_SCRIPTING
+	install_element(CONFIG_NODE, &zebra_on_rib_process_script_cmd);
+#endif /* HAVE_SCRIPTING */
+
 	install_element(VIEW_NODE, &zebra_show_routing_tables_summary_cmd);
 }