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:
+ */