aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/cnat/cnat_scanner.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/plugins/cnat/cnat_scanner.c')
-rw-r--r--src/plugins/cnat/cnat_scanner.c113
1 files changed, 113 insertions, 0 deletions
diff --git a/src/plugins/cnat/cnat_scanner.c b/src/plugins/cnat/cnat_scanner.c
new file mode 100644
index 00000000000..f5af327bffe
--- /dev/null
+++ b/src/plugins/cnat/cnat_scanner.c
@@ -0,0 +1,113 @@
+/*
+ * Copyright (c) 2020 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 <cnat/cnat_session.h>
+#include <cnat/cnat_client.h>
+
+typedef enum cnat_scanner_cmd_t_
+{
+ CNAT_SCANNER_OFF,
+ CNAT_SCANNER_ON,
+} cnat_scanner_cmd_t;
+
+static uword
+cnat_scanner_process (vlib_main_t * vm,
+ vlib_node_runtime_t * rt, vlib_frame_t * f)
+{
+ uword event_type, *event_data = 0;
+ cnat_main_t *cm = &cnat_main;
+ f64 start_time;
+ int enabled = 1, i = 0;
+
+ while (1)
+ {
+ if (enabled)
+ vlib_process_wait_for_event_or_clock (vm, cm->scanner_timeout);
+ else
+ vlib_process_wait_for_event (vm);
+
+ event_type = vlib_process_get_events (vm, &event_data);
+ vec_reset_length (event_data);
+
+ start_time = vlib_time_now (vm);
+
+ switch (event_type)
+ {
+ /* timer expired */
+ case ~0:
+ break;
+ case CNAT_SCANNER_OFF:
+ enabled = 0;
+ break;
+ case CNAT_SCANNER_ON:
+ enabled = 1;
+ break;
+ default:
+ ASSERT (0);
+ }
+
+ cnat_client_throttle_pool_process ();
+ i = cnat_session_scan (vm, start_time, i);
+ }
+ return 0;
+}
+
+/* *INDENT-OFF* */
+VLIB_REGISTER_NODE (cnat_scanner_process_node) = {
+ .function = cnat_scanner_process,
+ .type = VLIB_NODE_TYPE_PROCESS,
+ .name = "cnat-scanner-process",
+};
+/* *INDENT-ON* */
+
+static clib_error_t *
+cnat_scanner_cmd (vlib_main_t * vm,
+ unformat_input_t * input, vlib_cli_command_t * c)
+{
+ cnat_scanner_cmd_t cmd;
+
+ cmd = CNAT_SCANNER_ON;
+
+ while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
+ {
+ if (unformat (input, "on"))
+ cmd = CNAT_SCANNER_ON;
+ else if (unformat (input, "off"))
+ cmd = CNAT_SCANNER_OFF;
+ else
+ return (clib_error_return (0, "unknown input '%U'",
+ format_unformat_error, input));
+ }
+
+ vlib_process_signal_event (vm, cnat_scanner_process_node.index, cmd, 0);
+
+ return (NULL);
+}
+
+/* *INDENT-OFF* */
+VLIB_CLI_COMMAND (cnat_scanner_cmd_node, static) = {
+ .path = "test cnat scanner",
+ .function = cnat_scanner_cmd,
+ .short_help = "test cnat scanner",
+};
+/* *INDENT-ON* */
+
+/*
+ * fd.io coding-style-patch-verification: ON
+ *
+ * Local Variables:
+ * eval: (c-set-style "gnu")
+ * End:
+ */