Merge pull request #2934 from LabNConsulting/working/master/rfapi_topotest_support

add some testing support for use in rfapi topotest

1 files changed, 59 insertions, 1 deletions

diff --git a/bgpd/rfp-example/librfp/rfp_example.c b/bgpd/rfp-example/librfp/rfp_example.c
index 2751e7a44..75e57a029 100644
--- a/bgpd/rfp-example/librfp/rfp_example.c
+++ b/bgpd/rfp-example/librfp/rfp_example.c
@@ -55,6 +55,54 @@ DEFUN (rfp_example_config_value,
 	return CMD_SUCCESS;
 }
 
+DEFUN (rfp_holddown_factor,
+       rfp_holddown_factor_cmd,
+       "rfp holddown-factor (0-4294967295)",
+       RFP_SHOW_STR
+       "Set Hold-Down Factor as a percentage of registration lifetime.\n"
+       "Percentage of registration lifetime\n")
+{
+	struct rfp_instance_t *rfi;
+	uint32_t value = 0;
+
+	value = strtoul((argv[--argc]->arg), NULL, 10);
+	rfi = rfapi_get_rfp_start_val(VTY_GET_CONTEXT(bgp)); /* BGP_NODE */
+	if (!rfi) {
+		vty_out(vty, "VNC not configured\n");
+		return CMD_WARNING;
+	}
+	rfi->rfapi_config.holddown_factor = value;
+	rfapi_rfp_set_configuration(rfi, &rfi->rfapi_config);
+	return CMD_SUCCESS;
+}
+
+
+DEFUN (rfp_full_table_download,
+       rfp_full_table_download_cmd,
+       "rfp full-table-download <on|off>",
+       RFP_SHOW_STR
+       "RFP full table download support (default=on)\n"
+       "Enable RFP full table download\n"
+       "Disable RFP full table download\n")
+{
+	struct rfp_instance_t *rfi;
+	rfapi_rfp_download_type old;
+
+	rfi = rfapi_get_rfp_start_val(VTY_GET_CONTEXT(bgp)); /* BGP_NODE */
+	if (!rfi) {
+		vty_out(vty, "VNC not configured\n");
+		return CMD_WARNING;
+	}
+	old = rfi->rfapi_config.download_type;
+	if (argv[--argc]->arg[1] == 'n' || argv[argc]->arg[1] == 'N')
+		rfi->rfapi_config.download_type = RFAPI_RFP_DOWNLOAD_FULL;
+	else
+		rfi->rfapi_config.download_type = RFAPI_RFP_DOWNLOAD_PARTIAL;
+	if (old != rfi->rfapi_config.download_type)
+		rfapi_rfp_set_configuration(rfi, &rfi->rfapi_config);
+	return CMD_SUCCESS;
+}
+
 static void rfp_vty_install()
 {
 	static int installed = 0;
@@ -63,6 +111,8 @@ static void rfp_vty_install()
 	installed = 1;
 	/* example of new cli command */
 	install_element(BGP_NODE, &rfp_example_config_value_cmd);
+	install_element(BGP_NODE, &rfp_holddown_factor_cmd);
+	install_element(BGP_NODE, &rfp_full_table_download_cmd);
 }
 
 /***********************************************************************
@@ -196,7 +246,15 @@ static int rfp_cfg_write_cb(struct vty *vty, void *rfp_start_val)
 		vty_out(vty, "\n");
 		write++;
 	}
-
+	if (rfi->rfapi_config.holddown_factor != 0) {
+		vty_out(vty, " rfp holddown-factor %u\n",
+			rfi->rfapi_config.holddown_factor);
+		write++;
+	}
+	if (rfi->rfapi_config.download_type != RFAPI_RFP_DOWNLOAD_FULL) {
+		vty_out(vty, " rfp full-table-download off\n");
+		write++;
+	}
 	return write;
 }