tls: cleanup engine hs cb and improve ctx formatting
[vpp.git] / src / vnet / session / session_rules_table.c
index 34bd6a3..acb575b 100644 (file)
 #include <vnet/session/mma_template.c>
 #include <vnet/session/mma_40.h>
 #include <vnet/session/mma_template.c>
-#include <vnet/session/session_rules_table.h>
 #include <vnet/session/transport.h>
+#include <vnet/session/session.h>
+#include <vnet/session/session_table.h>
+#include <vnet/session/session_rules_table.h>
+#include <vnet/session/session_sdl.h>
+
+VLIB_REGISTER_LOG_CLASS (session_rt_log, static) = { .class_name = "session",
+                                                    .subclass_name = "rt" };
+
+#define log_debug(fmt, ...)                                                   \
+  vlib_log_debug (session_rt_log.class, "%s: " fmt, __func__, __VA_ARGS__)
+#define log_warn(fmt, ...)                                                    \
+  vlib_log_warn (session_rt_log.class, fmt, __VA_ARGS__)
+#define log_err(fmt, ...) vlib_log_err (session_rt_log.class, fmt, __VA_ARGS__)
+
+static session_rules_table_group_t *srt_instances;
+const session_rt_engine_vft_t *session_rt_engine_vft;
 
 u32
 session_rule_tag_key_index (u32 ri, u8 is_ip4)
@@ -75,6 +90,7 @@ session_rules_table_del_tag (session_rules_table_t * srt, u8 * tag, u8 is_ip4)
   ASSERT (rt);
   hash_unset_mem (srt->rules_by_tag, tag);
   hash_unset (srt->tags_by_rules, rti_key);
+  vec_free (rt->tag);
   pool_put (srt->rule_tags, rt);
 }
 
@@ -117,6 +133,18 @@ fib_pref_normalize (fib_prefix_t * pref)
     ip6_address_normalize (&pref->fp_addr.ip6, pref->fp_len);
 }
 
+u8 *
+format_session_rule_tag (u8 *s, va_list *args)
+{
+  static u8 *null_tag = 0;
+  u8 *tag = va_arg (*args, u8 *);
+
+  if (!null_tag)
+    null_tag = format (0, "none");
+  s = format (s, "%v", (tag != 0) ? tag : null_tag);
+  return s;
+}
+
 u8 *
 format_session_rule4 (u8 * s, va_list * args)
 {
@@ -124,7 +152,7 @@ format_session_rule4 (u8 * s, va_list * args)
   mma_rule_16_t *sr = va_arg (*args, mma_rule_16_t *);
   session_mask_or_match_4_t *mask, *match;
   mma_rules_table_16_t *srt4;
-  u8 *tag = 0, *null_tag = format (0, "none");
+  u8 *tag = 0;
   u32 ri;
   int i;
 
@@ -134,20 +162,18 @@ format_session_rule4 (u8 * s, va_list * args)
   match = (session_mask_or_match_4_t *) & sr->match;
   mask = (session_mask_or_match_4_t *) & sr->mask;
 
-  s = format (s, "[%d] rule: %U/%d %d %U/%d %d action: %d tag: %v", ri,
+  s = format (s, "[%d] rule: %U/%d %d %U/%d %d action: %d tag: %U", ri,
              format_ip4_address, &match->lcl_ip,
-             ip4_mask_to_preflen (&mask->lcl_ip),
-             clib_net_to_host_u16 (match->lcl_port), format_ip4_address,
-             &match->rmt_ip, ip4_mask_to_preflen (&mask->rmt_ip),
-             clib_net_to_host_u16 (match->rmt_port), sr->action_index,
-             tag ? tag : null_tag);
+             ip4_mask_to_preflen (&mask->lcl_ip), match->lcl_port,
+             format_ip4_address, &match->rmt_ip,
+             ip4_mask_to_preflen (&mask->rmt_ip), match->rmt_port,
+             sr->action_index, format_session_rule_tag, tag);
   if (vec_len (sr->next_indices))
     {
       s = format (s, "\n    children: ");
       for (i = 0; i < vec_len (sr->next_indices); i++)
        s = format (s, "%d ", sr->next_indices[i]);
     }
-  vec_free (null_tag);
   return s;
 }
 
@@ -158,7 +184,7 @@ format_session_rule6 (u8 * s, va_list * args)
   mma_rule_40_t *sr = va_arg (*args, mma_rule_40_t *);
   session_mask_or_match_6_t *mask, *match;
   mma_rules_table_40_t *srt6;
-  u8 *tag = 0, *null_tag = format (0, "none");
+  u8 *tag = 0;
   u32 ri;
   int i;
 
@@ -168,20 +194,18 @@ format_session_rule6 (u8 * s, va_list * args)
   match = (session_mask_or_match_6_t *) & sr->match;
   mask = (session_mask_or_match_6_t *) & sr->mask;
 
-  s = format (s, "[%d] rule: %U/%d %d %U/%d %d action: %d tag: %v", ri,
+  s = format (s, "[%d] rule: %U/%d %d %U/%d %d action: %d tag: %U", ri,
              format_ip6_address, &match->lcl_ip,
-             ip6_mask_to_preflen (&mask->lcl_ip),
-             clib_net_to_host_u16 (match->lcl_port), format_ip6_address,
-             &match->rmt_ip, ip6_mask_to_preflen (&mask->rmt_ip),
-             clib_net_to_host_u16 (match->rmt_port), sr->action_index,
-             tag ? tag : null_tag);
+             ip6_mask_to_preflen (&mask->lcl_ip), match->lcl_port,
+             format_ip6_address, &match->rmt_ip,
+             ip6_mask_to_preflen (&mask->rmt_ip), match->rmt_port,
+             sr->action_index, format_session_rule_tag, tag);
   if (vec_len (sr->next_indices))
     {
       s = format (s, "\n    children: ");
       for (i = 0; i < vec_len (sr->next_indices); i++)
        s = format (s, "%d ", sr->next_indices[i]);
     }
-  vec_free (null_tag);
   return s;
 }
 
@@ -312,11 +336,10 @@ session_rules_table_alloc_rule_40 (mma_rules_table_40_t * srt,
   return rule;
 }
 
-u32
-session_rules_table_lookup_rule4 (session_rules_table_t * srt,
-                                 ip4_address_t * lcl_ip,
-                                 ip4_address_t * rmt_ip, u16 lcl_port,
-                                 u16 rmt_port)
+static u32
+session_rules_table_lookup_rule4 (session_rules_table_t *srt,
+                                 ip4_address_t *lcl_ip, ip4_address_t *rmt_ip,
+                                 u16 lcl_port, u16 rmt_port)
 {
   mma_rules_table_16_t *srt4 = &srt->session_rules_tables_16;
   session_mask_or_match_4_t key = {
@@ -325,16 +348,20 @@ session_rules_table_lookup_rule4 (session_rules_table_t * srt,
     .lcl_port = lcl_port,
     .rmt_port = rmt_port,
   };
+
+  if (srt4->rules == 0)
+    return SESSION_TABLE_INVALID_INDEX;
   return mma_rules_table_lookup_rule_16 (srt4,
                                         (mma_mask_or_match_16_t *) & key,
                                         srt4->root_index);
 }
 
 u32
-session_rules_table_lookup4 (session_rules_table_t * srt,
-                            ip4_address_t * lcl_ip, ip4_address_t * rmt_ip,
-                            u16 lcl_port, u16 rmt_port)
+session_rules_table_lookup4_ (u32 srtg_handle, u32 proto,
+                             ip4_address_t *lcl_ip, ip4_address_t *rmt_ip,
+                             u16 lcl_port, u16 rmt_port)
 {
+  session_rules_table_t *srt = srtg_handle_to_srt (srtg_handle, proto);
   mma_rules_table_16_t *srt4 = &srt->session_rules_tables_16;
   session_mask_or_match_4_t key = {
     .lcl_ip.as_u32 = lcl_ip->as_u32,
@@ -342,21 +369,36 @@ session_rules_table_lookup4 (session_rules_table_t * srt,
     .lcl_port = lcl_port,
     .rmt_port = rmt_port,
   };
+
+  if (srt4->rules == 0)
+    return SESSION_TABLE_INVALID_INDEX;
   return mma_rules_table_lookup_16 (srt4, (mma_mask_or_match_16_t *) & key,
                                    srt4->root_index);
 }
 
-u32
-session_rules_table_lookup_rule6 (session_rules_table_t * srt,
-                                 ip6_address_t * lcl_ip,
-                                 ip6_address_t * rmt_ip, u16 lcl_port,
-                                 u16 rmt_port)
+session_rules_table_t *
+srtg_handle_to_srt (u32 srtg_handle, u32 proto)
+{
+  session_rules_table_group_t *srtg =
+    pool_elt_at_index (srt_instances, srtg_handle);
+  session_rules_table_t *srt = &srtg->session_rules[proto];
+
+  return srt;
+}
+
+static u32
+session_rules_table_lookup_rule6 (session_rules_table_t *srt,
+                                 ip6_address_t *lcl_ip, ip6_address_t *rmt_ip,
+                                 u16 lcl_port, u16 rmt_port)
 {
   mma_rules_table_40_t *srt6 = &srt->session_rules_tables_40;
   session_mask_or_match_6_t key = {
     .lcl_port = lcl_port,
     .rmt_port = rmt_port,
   };
+
+  if (srt6->rules == 0)
+    return SESSION_TABLE_INVALID_INDEX;
   clib_memcpy_fast (&key.lcl_ip, lcl_ip, sizeof (*lcl_ip));
   clib_memcpy_fast (&key.rmt_ip, rmt_ip, sizeof (*rmt_ip));
   return mma_rules_table_lookup_rule_40 (srt6,
@@ -365,15 +407,19 @@ session_rules_table_lookup_rule6 (session_rules_table_t * srt,
 }
 
 u32
-session_rules_table_lookup6 (session_rules_table_t * srt,
-                            ip6_address_t * lcl_ip, ip6_address_t * rmt_ip,
-                            u16 lcl_port, u16 rmt_port)
+session_rules_table_lookup6_ (u32 srtg_handle, u32 proto,
+                             ip6_address_t *lcl_ip, ip6_address_t *rmt_ip,
+                             u16 lcl_port, u16 rmt_port)
 {
+  session_rules_table_t *srt = srtg_handle_to_srt (srtg_handle, proto);
   mma_rules_table_40_t *srt6 = &srt->session_rules_tables_40;
   session_mask_or_match_6_t key = {
     .lcl_port = lcl_port,
     .rmt_port = rmt_port,
   };
+
+  if (srt6->rules == 0)
+    return SESSION_TABLE_INVALID_INDEX;
   clib_memcpy_fast (&key.lcl_ip, lcl_ip, sizeof (*lcl_ip));
   clib_memcpy_fast (&key.rmt_ip, rmt_ip, sizeof (*rmt_ip));
   return mma_rules_table_lookup_40 (srt6, (mma_mask_or_match_40_t *) & key,
@@ -386,19 +432,20 @@ session_rules_table_lookup6 (session_rules_table_t * srt,
  * @param srt table where rule should be added
  * @param args rule arguments
  *
- * @return 0 if success, clib_error_t error otherwise
+ * @return 0 if success, session_error_t error otherwise
  */
-int
-session_rules_table_add_del (session_rules_table_t * srt,
-                            session_rule_table_add_del_args_t * args)
+session_error_t
+session_rules_table_add_del_ (u32 srtg_handle, u32 proto,
+                             session_rule_table_add_del_args_t *args)
 {
+  session_rules_table_t *srt = srtg_handle_to_srt (srtg_handle, proto);
   u8 fib_proto = args->rmt.fp_proto, *rt;
   u32 ri_from_tag, ri;
   int rv;
 
   ri_from_tag = session_rules_table_rule_for_tag (srt, args->tag);
   if (args->is_add && ri_from_tag != SESSION_RULES_TABLE_INVALID_INDEX)
-    return VNET_API_ERROR_INVALID_VALUE;
+    return SESSION_E_INVALID;
 
   if (fib_proto == FIB_PROTOCOL_IP4)
     {
@@ -509,45 +556,95 @@ session_rules_table_add_del (session_rules_table_t * srt,
        }
     }
   else
-    return VNET_API_ERROR_INVALID_VALUE_2;
+    return SESSION_E_INVALID;
   return 0;
 }
 
 void
-session_rules_table_init (session_rules_table_t * srt)
+session_rules_table_free_ (session_table_t *st, u8 fib_proto)
+{
+  session_rules_table_group_t *srtg =
+    pool_elt_at_index (srt_instances, st->srtg_handle);
+  session_rules_table_t *srt;
+
+  vec_foreach (srt, srtg->session_rules)
+    {
+      mma_rules_table_free_16 (&srt->session_rules_tables_16);
+      mma_rules_table_free_40 (&srt->session_rules_tables_40);
+
+      hash_free (srt->tags_by_rules);
+      hash_free (srt->rules_by_tag);
+    }
+  srtg_instance_free (st);
+}
+
+void
+srtg_instance_free (session_table_t *st)
+{
+  session_rules_table_group_t *srtg =
+    pool_elt_at_index (srt_instances, st->srtg_handle);
+
+  vec_free (srtg->session_rules);
+  pool_put (srt_instances, srtg);
+  st->srtg_handle = SESSION_SRTG_HANDLE_INVALID;
+}
+
+session_rules_table_group_t *
+srtg_instance_alloc (session_table_t *st, u32 n_proto)
+{
+  session_rules_table_group_t *srtg;
+
+  pool_get (srt_instances, srtg);
+  vec_validate (srtg->session_rules, n_proto);
+  st->srtg_handle = srtg - srt_instances;
+  return (srtg);
+}
+
+void
+session_rules_table_init_ (session_table_t *st, u8 fib_proto)
 {
   mma_rules_table_16_t *srt4;
   mma_rules_table_40_t *srt6;
   mma_rule_16_t *rule4;
   mma_rule_40_t *rule6;
   fib_prefix_t null_prefix;
+  session_rules_table_t *srt;
+  session_rules_table_group_t *srtg;
 
-  clib_memset (&null_prefix, 0, sizeof (null_prefix));
+  srtg = srtg_instance_alloc (st, TRANSPORT_N_PROTOS - 1);
 
-  srt4 = &srt->session_rules_tables_16;
-  rule4 = session_rules_table_alloc_rule_16 (srt4, &null_prefix, 0,
-                                            &null_prefix, 0);
-  rule4->action_index = SESSION_RULES_TABLE_INVALID_INDEX;
-  srt4->root_index = mma_rules_table_rule_index_16 (srt4, rule4);
-  srt4->rule_cmp_fn = rule_cmp_16;
+  clib_memset (&null_prefix, 0, sizeof (null_prefix));
+  vec_foreach (srt, srtg->session_rules)
+    {
+      srt4 = &srt->session_rules_tables_16;
 
-  srt6 = &srt->session_rules_tables_40;
-  rule6 = session_rules_table_alloc_rule_40 (srt6, &null_prefix, 0,
-                                            &null_prefix, 0);
-  rule6->action_index = SESSION_RULES_TABLE_INVALID_INDEX;
-  srt6->root_index = mma_rules_table_rule_index_40 (srt6, rule6);
-  srt6->rule_cmp_fn = rule_cmp_40;
+      ASSERT (srt4->rules == 0);
+      rule4 = session_rules_table_alloc_rule_16 (srt4, &null_prefix, 0,
+                                                &null_prefix, 0);
+      rule4->action_index = SESSION_RULES_TABLE_INVALID_INDEX;
+      srt4->root_index = mma_rules_table_rule_index_16 (srt4, rule4);
+      srt4->rule_cmp_fn = rule_cmp_16;
 
-  srt->rules_by_tag = hash_create_vec (0, sizeof (u8), sizeof (uword));
-  srt->tags_by_rules = hash_create (0, sizeof (uword));
+      srt6 = &srt->session_rules_tables_40;
+      ASSERT (srt6->rules == 0);
+      rule6 = session_rules_table_alloc_rule_40 (srt6, &null_prefix, 0,
+                                                &null_prefix, 0);
+      rule6->action_index = SESSION_RULES_TABLE_INVALID_INDEX;
+      srt6->root_index = mma_rules_table_rule_index_40 (srt6, rule6);
+      srt6->rule_cmp_fn = rule_cmp_40;
+
+      srt->rules_by_tag = hash_create_vec (0, sizeof (u8), sizeof (uword));
+      srt->tags_by_rules = hash_create (0, sizeof (uword));
+    }
 }
 
 void
-session_rules_table_show_rule (vlib_main_t * vm, session_rules_table_t * srt,
-                              ip46_address_t * lcl_ip, u16 lcl_port,
-                              ip46_address_t * rmt_ip, u16 rmt_port,
-                              u8 is_ip4)
+session_rules_table_show_rule_ (vlib_main_t *vm, u32 srtg_handle, u32 proto,
+                               ip46_address_t *lcl_ip, u16 lcl_port,
+                               ip46_address_t *rmt_ip, u16 rmt_port,
+                               u8 is_ip4)
 {
+  session_rules_table_t *srt = srtg_handle_to_srt (srtg_handle, proto);
   mma_rules_table_16_t *srt4;
   mma_rules_table_40_t *srt6;
   mma_rule_16_t *sr4;
@@ -588,9 +685,10 @@ session_rules_table_show_rule (vlib_main_t * vm, session_rules_table_t * srt,
 }
 
 void
-session_rules_table_cli_dump (vlib_main_t * vm, session_rules_table_t * srt,
-                             u8 fib_proto)
+session_rules_table_cli_dump_ (vlib_main_t *vm, u32 srtg_handle, u32 proto,
+                              u8 fib_proto)
 {
+  session_rules_table_t *srt = srtg_handle_to_srt (srtg_handle, proto);
   if (fib_proto == FIB_PROTOCOL_IP4)
     {
       mma_rules_table_16_t *srt4;
@@ -598,11 +696,9 @@ session_rules_table_cli_dump (vlib_main_t * vm, session_rules_table_t * srt,
       srt4 = &srt->session_rules_tables_16;
       vlib_cli_output (vm, "IP4 rules");
 
-      /* *INDENT-OFF* */
       pool_foreach (sr4, srt4->rules)  {
        vlib_cli_output (vm, "%U", format_session_rule4, srt, sr4);
       }
-      /* *INDENT-ON* */
 
     }
   else if (fib_proto == FIB_PROTOCOL_IP6)
@@ -612,15 +708,115 @@ session_rules_table_cli_dump (vlib_main_t * vm, session_rules_table_t * srt,
       srt6 = &srt->session_rules_tables_40;
       vlib_cli_output (vm, "IP6 rules");
 
-      /* *INDENT-OFF* */
       pool_foreach (sr6, srt6->rules)  {
         vlib_cli_output (vm, "%U", format_session_rule6, srt, sr6);
       }
-      /* *INDENT-ON* */
 
     }
 }
 
+static const session_rt_engine_vft_t session_rules_table_vft = {
+  .backend_engine = RT_BACKEND_ENGINE_RULE_TABLE,
+  .table_lookup4 = session_rules_table_lookup4_,
+  .table_lookup6 = session_rules_table_lookup6_,
+  .table_cli_dump = session_rules_table_cli_dump_,
+  .table_show_rule = session_rules_table_show_rule_,
+  .table_add_del = session_rules_table_add_del_,
+  .table_init = session_rules_table_init_,
+  .table_free = session_rules_table_free_,
+};
+
+static void
+session_rules_table_app_namespace_walk_cb (app_namespace_t *app_ns, void *ctx)
+{
+  u32 fib_index, table_index;
+  session_table_t *st;
+
+  log_debug ("disable app_ns %s", app_ns->ns_id);
+  st = session_table_get (app_ns->local_table_index);
+  if (st)
+    session_rules_table_free (st, FIB_PROTOCOL_MAX);
+
+  fib_index = app_namespace_get_fib_index (app_ns, FIB_PROTOCOL_IP4);
+  table_index = session_lookup_get_index_for_fib (FIB_PROTOCOL_IP4, fib_index);
+  st = session_table_get (table_index);
+  if (st)
+    session_rules_table_free (st, FIB_PROTOCOL_IP4);
+
+  fib_index = app_namespace_get_fib_index (app_ns, FIB_PROTOCOL_IP6);
+  table_index = session_lookup_get_index_for_fib (FIB_PROTOCOL_IP6, fib_index);
+  st = session_table_get (table_index);
+  if (st)
+    session_rules_table_free (st, FIB_PROTOCOL_IP6);
+}
+
+clib_error_t *
+session_rules_table_enable_disable (int enable)
+{
+  clib_error_t *error;
+
+  if (enable)
+    error = session_rule_table_register_engine (&session_rules_table_vft);
+  else
+    {
+      app_namespace_walk (session_rules_table_app_namespace_walk_cb, 0);
+      error = session_rule_table_deregister_engine (&session_rules_table_vft);
+    }
+
+  return error;
+}
+
+clib_error_t *
+session_rt_backend_enable_disable (session_rt_engine_type_t rt_engine_type)
+{
+  session_main_t *smm = &session_main;
+  clib_error_t *error = 0;
+
+  if (rt_engine_type < RT_BACKEND_ENGINE_DISABLE ||
+      rt_engine_type > RT_BACKEND_ENGINE_SDL)
+    return clib_error_return (0, "invalid rt-backend %d", rt_engine_type);
+
+  if (rt_engine_type == RT_BACKEND_ENGINE_SDL)
+    error = session_sdl_enable_disable (1);
+  else if (rt_engine_type == RT_BACKEND_ENGINE_RULE_TABLE)
+    error = session_rules_table_enable_disable (1);
+  else if (rt_engine_type == RT_BACKEND_ENGINE_DISABLE)
+    {
+      if (session_sdl_is_enabled ())
+      error = session_sdl_enable_disable (0);
+      else if (session_rule_table_is_enabled ())
+      error = session_rules_table_enable_disable (0);
+    }
+
+  if (!error)
+    smm->rt_engine_type = rt_engine_type;
+  return error;
+}
+
+clib_error_t *
+session_rule_table_register_engine (const session_rt_engine_vft_t *vft)
+{
+  if (session_rt_engine_vft == vft)
+    return 0;
+  if (session_rt_engine_vft)
+    return clib_error_return (0, "session rule engine is already registered");
+
+  session_rt_engine_vft = vft;
+  return 0;
+}
+
+clib_error_t *
+session_rule_table_deregister_engine (const session_rt_engine_vft_t *vft)
+{
+  if (session_rt_engine_vft == vft)
+    session_rt_engine_vft = 0;
+  else
+    return clib_error_return (
+      0, "session rule engine is not registered to this engine");
+
+  return 0;
+}
+
 /*
  * fd.io coding-style-patch-verification: ON
  *