misc: minimize dependencies on udp.h

[vpp.git] / src / plugins / cnat / cnat_types.c

diff --git a/src/plugins/cnat/cnat_types.c b/src/plugins/cnat/cnat_types.c
index 1f2287e..c15c2f6 100644 (file)
--- a/src/plugins/cnat/cnat_types.c
+++ b/src/plugins/cnat/cnat_types.c
@@ -26,17 +26,75 @@ char *cnat_error_strings[] = {
 #undef cnat_error
 };
 
+u8
+cnat_resolve_addr (u32 sw_if_index, ip_address_family_t af,
+                  ip_address_t * addr)
+{
+  /* Tries to resolve IP from sw_if_index
+   * returns 1 if we need to schedule DHCP */
+  if (INDEX_INVALID == sw_if_index)
+    return 0;
+  if (af == AF_IP6)
+    {
+      ip6_address_t *ip6 = 0;
+      ip6 = ip6_interface_first_address (&ip6_main, sw_if_index);
+      if (ip6)
+       {
+         ip_address_set (addr, ip6, AF_IP6);
+         return 0;
+       }
+      else
+       return 1;
+    }
+  else
+    {
+      ip4_address_t *ip4 = 0;
+      ip4 = ip4_interface_first_address (&ip4_main, sw_if_index, 0);
+      if (ip4)
+       {
+         ip_address_set (addr, ip4, AF_IP4);
+         return 0;
+       }
+      else
+       return 1;
+    }
+}
+
+u8
+cnat_resolve_ep (cnat_endpoint_t * ep)
+{
+  int rv;
+  rv = cnat_resolve_addr (ep->ce_sw_if_index, ep->ce_ip.version, &ep->ce_ip);
+  if (0 == rv)
+    ep->ce_flags |= CNAT_EP_FLAG_RESOLVED;
+  return rv;
+}
+
 uword
 unformat_cnat_ep (unformat_input_t * input, va_list * args)
 {
   cnat_endpoint_t *a = va_arg (*args, cnat_endpoint_t *);
+  vnet_main_t *vnm = vnet_get_main ();
   int port = 0;
 
   clib_memset (a, 0, sizeof (*a));
+  a->ce_sw_if_index = INDEX_INVALID;
   if (unformat (input, "%U %d", unformat_ip_address, &a->ce_ip, &port))
     ;
   else if (unformat_user (input, unformat_ip_address, &a->ce_ip))
     ;
+  else if (unformat (input, "%U v6 %d", unformat_vnet_sw_interface,
+                    vnm, &a->ce_sw_if_index, &port))
+    a->ce_ip.version = AF_IP6;
+  else if (unformat (input, "%U v6", unformat_vnet_sw_interface,
+                    vnm, &a->ce_sw_if_index))
+    a->ce_ip.version = AF_IP6;
+  else if (unformat (input, "%U %d", unformat_vnet_sw_interface,
+                    vnm, &a->ce_sw_if_index, &port))
+    a->ce_ip.version = AF_IP4;
+  else if (unformat_user (input, unformat_vnet_sw_interface,
+                         vnm, &a->ce_sw_if_index))
+    a->ce_ip.version = AF_IP4;
   else if (unformat (input, "%d", &port))
     ;
   else
@@ -65,9 +123,21 @@ u8 *
 format_cnat_endpoint (u8 * s, va_list * args)
 {
   cnat_endpoint_t *cep = va_arg (*args, cnat_endpoint_t *);
-
-  s = format (s, "%U;%d", format_ip_address, &cep->ce_ip, cep->ce_port);
-
+  vnet_main_t *vnm = vnet_get_main ();
+  if (INDEX_INVALID == cep->ce_sw_if_index)
+    s = format (s, "%U;%d", format_ip_address, &cep->ce_ip, cep->ce_port);
+  else
+    {
+      if (cep->ce_flags & CNAT_EP_FLAG_RESOLVED)
+       s = format (s, "%U (%U);%d", format_vnet_sw_if_index_name, vnm,
+                   cep->ce_sw_if_index, format_ip_address, &cep->ce_ip,
+                   cep->ce_port);
+      else
+       s =
+         format (s, "%U (%U);%d", format_vnet_sw_if_index_name, vnm,
+                 cep->ce_sw_if_index, format_ip_address_family,
+                 cep->ce_ip.version, cep->ce_port);
+    }
   return (s);
 }
 
@@ -80,14 +150,30 @@ cnat_types_init (vlib_main_t * vm)
                                         CNAT_FIB_SOURCE_PRIORITY,
                                         FIB_SOURCE_BH_SIMPLE);
 
+
   clib_rwlock_init (&cnat_main.ts_lock);
-  clib_spinlock_init (&cnat_main.src_ports_lock);
-  clib_bitmap_validate (cnat_main.src_ports, UINT16_MAX);
   throttle_init (&cnat_throttle, n_vlib_mains, 1e-3);
 
   return (NULL);
 }
 
+void
+cnat_enable_disable_scanner (cnat_scanner_cmd_t event_type)
+{
+  vlib_main_t *vm = vlib_get_main ();
+  vlib_process_signal_event (vm, cnat_main.scanner_node_index, event_type, 0);
+}
+
+void
+cnat_lazy_init ()
+{
+  cnat_main_t *cm = &cnat_main;
+  if (cm->lazy_init_done)
+    return;
+  cnat_enable_disable_scanner (cm->default_scanner_state);
+  cm->lazy_init_done = 1;
+}
+
 static clib_error_t *
 cnat_config (vlib_main_t * vm, unformat_input_t * input)
 {
@@ -102,6 +188,8 @@ cnat_config (vlib_main_t * vm, unformat_input_t * input)
   cm->scanner_timeout = CNAT_DEFAULT_SCANNER_TIMEOUT;
   cm->session_max_age = CNAT_DEFAULT_SESSION_MAX_AGE;
   cm->tcp_max_age = CNAT_DEFAULT_TCP_MAX_AGE;
+  cm->default_scanner_state = CNAT_SCANNER_ON;
+  cm->lazy_init_done = 0;
 
   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
     {
@@ -125,6 +213,10 @@ cnat_config (vlib_main_t * vm, unformat_input_t * input)
       else if (unformat (input, "session-cleanup-timeout %f",
                         &cm->scanner_timeout))
        ;
+      else if (unformat (input, "scanner off"))
+       cm->default_scanner_state = CNAT_SCANNER_OFF;
+      else if (unformat (input, "scanner on"))
+       cm->default_scanner_state = CNAT_SCANNER_ON;
       else if (unformat (input, "session-max-age %u", &cm->session_max_age))
        ;
       else if (unformat (input, "tcp-max-age %u", &cm->tcp_max_age))
@@ -137,6 +229,12 @@ cnat_config (vlib_main_t * vm, unformat_input_t * input)
   return 0;
 }
 
+cnat_main_t *
+cnat_get_main ()
+{
+  return &cnat_main;
+}
+
 VLIB_EARLY_CONFIG_FUNCTION (cnat_config, "cnat");
 VLIB_INIT_FUNCTION (cnat_types_init);