udp session vcl: add udp iperf test

Type: improvement

Signed-off-by: Florin Coras <fcoras@cisco.com>
Change-Id: Ib4bc2ce781887a84055a4d5cdb7f453fc7d52c79
diff --git a/src/vnet/session/session_node.c b/src/vnet/session/session_node.c
index 66300bc..7d727b7 100644
--- a/src/vnet/session/session_node.c
+++ b/src/vnet/session/session_node.c
@@ -50,6 +50,8 @@
   clib_memset (a, 0, sizeof (*a));
   a->sep.is_ip4 = mp->is_ip4;
   clib_memcpy_fast (&a->sep.ip, &mp->ip, sizeof (mp->ip));
+  if (mp->is_ip4)
+    ip46_address_mask_ip4 (&a->sep.ip);
   a->sep.port = mp->port;
   a->sep.fib_index = mp->vrf;
   a->sep.sw_if_index = ENDPOINT_INVALID_INDEX;
@@ -113,6 +115,11 @@
   a->sep.transport_proto = mp->proto;
   a->sep.peer.fib_index = mp->vrf;
   clib_memcpy_fast (&a->sep.peer.ip, &mp->lcl_ip, sizeof (mp->lcl_ip));
+  if (mp->is_ip4)
+    {
+      ip46_address_mask_ip4 (&a->sep.ip);
+      ip46_address_mask_ip4 (&a->sep.peer.ip);
+    }
   a->sep.peer.port = mp->lcl_port;
   a->sep.peer.sw_if_index = ENDPOINT_INVALID_INDEX;
   a->sep_ext.parent_handle = mp->parent_handle;
diff --git a/src/vnet/udp/udp.c b/src/vnet/udp/udp.c
index 375e95b..ba1821a 100644
--- a/src/vnet/udp/udp.c
+++ b/src/vnet/udp/udp.c
@@ -132,13 +132,21 @@
   pool_put (udp_main.connections[thread_index], uc);
 }
 
+static void
+udp_connection_cleanup (udp_connection_t * uc)
+{
+  transport_endpoint_cleanup (TRANSPORT_PROTO_UDP, &uc->c_lcl_ip,
+			      uc->c_lcl_port);
+  udp_connection_unregister_port (clib_net_to_host_u16 (uc->c_lcl_port),
+				  uc->c_is_ip4);
+  udp_connection_free (uc);
+}
+
 void
 udp_connection_delete (udp_connection_t * uc)
 {
-  udp_connection_unregister_port (clib_net_to_host_u16 (uc->c_lcl_port),
-				  uc->c_is_ip4);
   session_transport_delete_notify (&uc->connection);
-  udp_connection_free (uc);
+  udp_connection_cleanup (uc);
 }
 
 u32
@@ -149,12 +157,14 @@
   transport_endpoint_cfg_t *lcl_ext;
   udp_connection_t *listener;
   udp_dst_port_info_t *pi;
+  u16 lcl_port_ho;
   void *iface_ip;
 
-  pi = udp_get_dst_port_info (um, clib_net_to_host_u16 (lcl->port),
-			      lcl->is_ip4);
+  lcl_port_ho = clib_net_to_host_u16 (lcl->port);
+  pi = udp_get_dst_port_info (um, lcl_port_ho, lcl->is_ip4);
 
-  if (pi && !pi->n_connections)
+  if (pi && !pi->n_connections
+      && udp_is_valid_dst_port (lcl_port_ho, lcl->is_ip4))
     {
       clib_warning ("port already used");
       return -1;
@@ -186,8 +196,7 @@
     listener->c_flags |= TRANSPORT_CONNECTION_F_CLESS;
   clib_spinlock_init (&listener->rx_lock);
 
-  udp_connection_register_port (vm, clib_net_to_host_u16 (lcl->port),
-				lcl->is_ip4);
+  udp_connection_register_port (vm, lcl_port_ho, lcl->is_ip4);
   return listener->c_c_index;
 }
 
@@ -276,7 +285,7 @@
   udp_connection_t *uc;
   uc = udp_connection_get (connection_index, thread_index);
   if (uc)
-    udp_connection_free (uc);
+    udp_connection_cleanup (uc);
 }
 
 u8 *