NAT: Refactoring / Housekeeping (VPP-1415)

Change-Id: Ia3ce24cc94f9b2fb331ad62a4181ddcd41bc78ca
Signed-off-by: Matus Fabian <matfabia@cisco.com>

1 files changed, 332 insertions, 0 deletions

diff --git a/src/plugins/nat/nat_format.c b/src/plugins/nat/nat_format.c
new file mode 100644
index 00000000000..279d6664933
--- /dev/null
+++ b/src/plugins/nat/nat_format.c
@@ -0,0 +1,332 @@
+/*
+ * Copyright (c) 2018 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.
+ */
+/**
+ * @file
+ * @brief NAT formatting
+ */
+
+#include <nat/nat.h>
+#include <nat/nat_det.h>
+
+uword
+unformat_snat_protocol (unformat_input_t * input, va_list * args)
+{
+  u32 *r = va_arg (*args, u32 *);
+
+  if (0);
+#define _(N, i, n, s) else if (unformat (input, s)) *r = SNAT_PROTOCOL_##N;
+  foreach_snat_protocol
+#undef _
+    else
+    return 0;
+  return 1;
+}
+
+u8 *
+format_snat_protocol (u8 * s, va_list * args)
+{
+  u32 i = va_arg (*args, u32);
+  u8 *t = 0;
+
+  switch (i)
+    {
+#define _(N, j, n, str) case SNAT_PROTOCOL_##N: t = (u8 *) str; break;
+      foreach_snat_protocol
+#undef _
+    default:
+      s = format (s, "unknown");
+      return s;
+    }
+  s = format (s, "%s", t);
+  return s;
+}
+
+u8 *
+format_nat_addr_and_port_alloc_alg (u8 * s, va_list * args)
+{
+  u32 i = va_arg (*args, u32);
+  u8 *t = 0;
+
+  switch (i)
+    {
+#define _(v, N, s) case NAT_ADDR_AND_PORT_ALLOC_ALG_##N: t = (u8 *) s; break;
+      foreach_nat_addr_and_port_alloc_alg
+#undef _
+    default:
+      s = format (s, "unknown");
+      return s;
+    }
+  s = format (s, "%s", t);
+  return s;
+}
+
+u8 *
+format_snat_key (u8 * s, va_list * args)
+{
+  snat_session_key_t *key = va_arg (*args, snat_session_key_t *);
+
+  s = format (s, "%U proto %U port %d fib %d",
+	      format_ip4_address, &key->addr,
+	      format_snat_protocol, key->protocol,
+	      clib_net_to_host_u16 (key->port), key->fib_index);
+  return s;
+}
+
+u8 *
+format_static_mapping_key (u8 * s, va_list * args)
+{
+  snat_session_key_t *key = va_arg (*args, snat_session_key_t *);
+
+  s = format (s, "%U proto %U port %d fib %d",
+	      format_ip4_address, &key->addr,
+	      format_snat_protocol, key->protocol, key->port, key->fib_index);
+  return s;
+}
+
+u8 *
+format_snat_session_state (u8 * s, va_list * args)
+{
+  u32 i = va_arg (*args, u32);
+  u8 *t = 0;
+
+  switch (i)
+    {
+#define _(v, N, str) case SNAT_SESSION_##N: t = (u8 *) str; break;
+      foreach_snat_session_state
+#undef _
+    default:
+      t = format (t, "unknown");
+    }
+  s = format (s, "%s", t);
+  return s;
+}
+
+u8 *
+format_snat_session (u8 * s, va_list * args)
+{
+  snat_main_per_thread_data_t *sm =
+    va_arg (*args, snat_main_per_thread_data_t *);
+  snat_session_t *sess = va_arg (*args, snat_session_t *);
+
+  if (snat_is_unk_proto_session (sess))
+    {
+      s = format (s, "  i2o %U proto %u fib %u\n",
+		  format_ip4_address, &sess->in2out.addr,
+		  clib_net_to_host_u16 (sess->in2out.port),
+		  sess->in2out.fib_index);
+      s = format (s, "    o2i %U proto %u fib %u\n",
+		  format_ip4_address, &sess->out2in.addr,
+		  clib_net_to_host_u16 (sess->out2in.port),
+		  sess->out2in.fib_index);
+    }
+  else
+    {
+      s = format (s, "  i2o %U\n", format_snat_key, &sess->in2out);
+      s = format (s, "    o2i %U\n", format_snat_key, &sess->out2in);
+    }
+  if (is_ed_session (sess) || is_fwd_bypass_session (sess))
+    {
+      if (is_twice_nat_session (sess))
+	{
+	  s = format (s, "       external host o2i %U:%d i2o %U:%d\n",
+		      format_ip4_address, &sess->ext_host_addr,
+		      clib_net_to_host_u16 (sess->ext_host_port),
+		      format_ip4_address, &sess->ext_host_nat_addr,
+		      clib_net_to_host_u16 (sess->ext_host_nat_port));
+	}
+      else
+	{
+	  if (sess->ext_host_addr.as_u32)
+	    s = format (s, "       external host %U:%u\n",
+			format_ip4_address, &sess->ext_host_addr,
+			clib_net_to_host_u16 (sess->ext_host_port));
+	}
+    }
+  s = format (s, "       index %llu\n", sess - sm->sessions);
+  s = format (s, "       last heard %.2f\n", sess->last_heard);
+  s = format (s, "       total pkts %d, total bytes %lld\n",
+	      sess->total_pkts, sess->total_bytes);
+  if (snat_is_session_static (sess))
+    s = format (s, "       static translation\n");
+  else
+    s = format (s, "       dynamic translation\n");
+  if (is_fwd_bypass_session (sess))
+    s = format (s, "       forwarding-bypass\n");
+  if (is_lb_session (sess))
+    s = format (s, "       load-balancing\n");
+  if (is_twice_nat_session (sess))
+    s = format (s, "       twice-nat\n");
+
+  return s;
+}
+
+u8 *
+format_snat_user (u8 * s, va_list * args)
+{
+  snat_main_per_thread_data_t *sm =
+    va_arg (*args, snat_main_per_thread_data_t *);
+  snat_user_t *u = va_arg (*args, snat_user_t *);
+  int verbose = va_arg (*args, int);
+  dlist_elt_t *head, *elt;
+  u32 elt_index, head_index;
+  u32 session_index;
+  snat_session_t *sess;
+
+  s = format (s, "%U: %d dynamic translations, %d static translations\n",
+	      format_ip4_address, &u->addr, u->nsessions, u->nstaticsessions);
+
+  if (verbose == 0)
+    return s;
+
+  if (u->nsessions || u->nstaticsessions)
+    {
+      head_index = u->sessions_per_user_list_head_index;
+      head = pool_elt_at_index (sm->list_pool, head_index);
+
+      elt_index = head->next;
+      elt = pool_elt_at_index (sm->list_pool, elt_index);
+      session_index = elt->value;
+
+      while (session_index != ~0)
+	{
+	  sess = pool_elt_at_index (sm->sessions, session_index);
+
+	  s = format (s, "  %U\n", format_snat_session, sm, sess);
+
+	  elt_index = elt->next;
+	  elt = pool_elt_at_index (sm->list_pool, elt_index);
+	  session_index = elt->value;
+	}
+    }
+
+  return s;
+}
+
+u8 *
+format_snat_static_mapping (u8 * s, va_list * args)
+{
+  snat_static_mapping_t *m = va_arg (*args, snat_static_mapping_t *);
+  nat44_lb_addr_port_t *local;
+
+  if (m->addr_only)
+    s = format (s, "local %U external %U vrf %d %s %s",
+		format_ip4_address, &m->local_addr,
+		format_ip4_address, &m->external_addr,
+		m->vrf_id,
+		m->twice_nat == TWICE_NAT ? "twice-nat" :
+		m->twice_nat == TWICE_NAT_SELF ? "self-twice-nat" : "",
+		m->out2in_only ? "out2in-only" : "");
+  else
+    {
+      if (vec_len (m->locals))
+	{
+	  s = format (s, "%U external %U:%d %s %s",
+		      format_snat_protocol, m->proto,
+		      format_ip4_address, &m->external_addr, m->external_port,
+		      m->twice_nat == TWICE_NAT ? "twice-nat" :
+		      m->twice_nat == TWICE_NAT_SELF ? "self-twice-nat" : "",
+		      m->out2in_only ? "out2in-only" : "");
+	  vec_foreach (local, m->locals)
+	    s = format (s, "\n  local %U:%d vrf %d probability %d\%",
+			format_ip4_address, &local->addr, local->port,
+			local->vrf_id, local->probability);
+	}
+      else
+	s = format (s, "%U local %U:%d external %U:%d vrf %d %s %s",
+		    format_snat_protocol, m->proto,
+		    format_ip4_address, &m->local_addr, m->local_port,
+		    format_ip4_address, &m->external_addr, m->external_port,
+		    m->vrf_id,
+		    m->twice_nat == TWICE_NAT ? "twice-nat" :
+		    m->twice_nat == TWICE_NAT_SELF ? "self-twice-nat" : "",
+		    m->out2in_only ? "out2in-only" : "");
+    }
+  return s;
+}
+
+u8 *
+format_snat_static_map_to_resolve (u8 * s, va_list * args)
+{
+  snat_static_map_resolve_t *m = va_arg (*args, snat_static_map_resolve_t *);
+  vnet_main_t *vnm = vnet_get_main ();
+
+  if (m->addr_only)
+    s = format (s, "local %U external %U vrf %d",
+		format_ip4_address, &m->l_addr,
+		format_vnet_sw_if_index_name, vnm, m->sw_if_index, m->vrf_id);
+  else
+    s = format (s, "%U local %U:%d external %U:%d vrf %d",
+		format_snat_protocol, m->proto,
+		format_ip4_address, &m->l_addr, m->l_port,
+		format_vnet_sw_if_index_name, vnm, m->sw_if_index,
+		m->e_port, m->vrf_id);
+
+  return s;
+}
+
+u8 *
+format_det_map_ses (u8 * s, va_list * args)
+{
+  snat_det_map_t *det_map = va_arg (*args, snat_det_map_t *);
+  ip4_address_t in_addr, out_addr;
+  u32 in_offset, out_offset;
+  snat_det_session_t *ses = va_arg (*args, snat_det_session_t *);
+  u32 *i = va_arg (*args, u32 *);
+
+  u32 user_index = *i / SNAT_DET_SES_PER_USER;
+  in_addr.as_u32 =
+    clib_host_to_net_u32 (clib_net_to_host_u32 (det_map->in_addr.as_u32) +
+			  user_index);
+  in_offset =
+    clib_net_to_host_u32 (in_addr.as_u32) -
+    clib_net_to_host_u32 (det_map->in_addr.as_u32);
+  out_offset = in_offset / det_map->sharing_ratio;
+  out_addr.as_u32 =
+    clib_host_to_net_u32 (clib_net_to_host_u32 (det_map->out_addr.as_u32) +
+			  out_offset);
+  s =
+    format (s,
+	    "in %U:%d out %U:%d external host %U:%d state: %U expire: %d\n",
+	    format_ip4_address, &in_addr, clib_net_to_host_u16 (ses->in_port),
+	    format_ip4_address, &out_addr,
+	    clib_net_to_host_u16 (ses->out.out_port), format_ip4_address,
+	    &ses->out.ext_host_addr,
+	    clib_net_to_host_u16 (ses->out.ext_host_port),
+	    format_snat_session_state, ses->state, ses->expire);
+
+  return s;
+}
+
+u8 *
+format_nat44_reass_trace (u8 * s, va_list * args)
+{
+  CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
+  CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
+  nat44_reass_trace_t *t = va_arg (*args, nat44_reass_trace_t *);
+
+  s = format (s, "NAT44_REASS: sw_if_index %d, next index %d, status %s",
+	      t->sw_if_index, t->next_index,
+	      t->cached ? "cached" : "translated");
+
+  return s;
+}
+
+/*
+ * fd.io coding-style-patch-verification: ON
+ *
+ * Local Variables:
+ * eval: (c-set-style "gnu")
+ * End:
+ */