Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 1 | /* |
Florin Coras | c5df8c7 | 2019-04-08 07:42:30 -0700 | [diff] [blame] | 2 | * Copyright (c) 2016-2019 Cisco and/or its affiliates. |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 3 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | * you may not use this file except in compliance with the License. |
| 5 | * You may obtain a copy of the License at: |
| 6 | * |
| 7 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | * |
| 9 | * Unless required by applicable law or agreed to in writing, software |
| 10 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | * See the License for the specific language governing permissions and |
| 13 | * limitations under the License. |
| 14 | */ |
| 15 | |
| 16 | /** @file |
| 17 | udp state machine, etc. |
| 18 | */ |
| 19 | |
| 20 | #include <vnet/udp/udp.h> |
| 21 | #include <vnet/session/session.h> |
| 22 | #include <vnet/dpo/load_balance.h> |
| 23 | #include <vnet/fib/ip4_fib.h> |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 24 | #include <vppinfra/sparse_vec.h> |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 25 | |
Filip Tehlar | 2c49ffe | 2019-03-06 07:16:08 -0800 | [diff] [blame] | 26 | udp_main_t udp_main; |
| 27 | |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 28 | static void |
| 29 | udp_connection_register_port (vlib_main_t * vm, u16 lcl_port, u8 is_ip4) |
| 30 | { |
| 31 | udp_main_t *um = &udp_main; |
| 32 | udp_dst_port_info_t *pi; |
| 33 | u16 *n; |
| 34 | |
| 35 | pi = udp_get_dst_port_info (um, lcl_port, is_ip4); |
| 36 | if (!pi) |
| 37 | { |
| 38 | udp_add_dst_port (um, lcl_port, 0, is_ip4); |
| 39 | pi = udp_get_dst_port_info (um, lcl_port, is_ip4); |
| 40 | pi->n_connections = 1; |
| 41 | } |
| 42 | else |
| 43 | { |
| 44 | pi->n_connections += 1; |
| 45 | /* Do not return. The fact that the pi is valid does not mean |
| 46 | * it's up to date */ |
| 47 | } |
| 48 | |
| 49 | pi->node_index = is_ip4 ? udp4_input_node.index : udp6_input_node.index; |
| 50 | pi->next_index = um->local_to_input_edge[is_ip4]; |
| 51 | |
| 52 | /* Setup udp protocol -> next index sparse vector mapping. */ |
| 53 | if (is_ip4) |
| 54 | n = sparse_vec_validate (um->next_by_dst_port4, |
| 55 | clib_host_to_net_u16 (lcl_port)); |
| 56 | else |
| 57 | n = sparse_vec_validate (um->next_by_dst_port6, |
| 58 | clib_host_to_net_u16 (lcl_port)); |
| 59 | |
| 60 | n[0] = pi->next_index; |
| 61 | } |
| 62 | |
| 63 | static void |
| 64 | udp_connection_unregister_port (u16 lcl_port, u8 is_ip4) |
| 65 | { |
| 66 | udp_main_t *um = &udp_main; |
| 67 | udp_dst_port_info_t *pi; |
| 68 | |
| 69 | pi = udp_get_dst_port_info (um, lcl_port, is_ip4); |
| 70 | if (!pi) |
| 71 | return; |
| 72 | |
| 73 | if (!pi->n_connections) |
| 74 | { |
| 75 | clib_warning ("no connections using port %u", lcl_port); |
| 76 | return; |
| 77 | } |
| 78 | |
| 79 | if (!clib_atomic_sub_fetch (&pi->n_connections, 1)) |
| 80 | udp_unregister_dst_port (0, lcl_port, is_ip4); |
| 81 | } |
| 82 | |
| 83 | void |
| 84 | udp_connection_share_port (u16 lcl_port, u8 is_ip4) |
| 85 | { |
| 86 | udp_main_t *um = &udp_main; |
| 87 | udp_dst_port_info_t *pi; |
| 88 | |
| 89 | /* Done without a lock but the operation is atomic. Writers to pi hash |
| 90 | * table and vector should be guarded by a barrier sync */ |
| 91 | pi = udp_get_dst_port_info (um, lcl_port, is_ip4); |
| 92 | clib_atomic_fetch_add_rel (&pi->n_connections, 1); |
| 93 | } |
| 94 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 95 | udp_connection_t * |
| 96 | udp_connection_alloc (u32 thread_index) |
| 97 | { |
| 98 | udp_main_t *um = &udp_main; |
| 99 | udp_connection_t *uc; |
| 100 | u32 will_expand = 0; |
| 101 | pool_get_aligned_will_expand (um->connections[thread_index], will_expand, |
| 102 | CLIB_CACHE_LINE_BYTES); |
| 103 | |
| 104 | if (PREDICT_FALSE (will_expand)) |
| 105 | { |
| 106 | clib_spinlock_lock_if_init (&udp_main.peekers_write_locks |
| 107 | [thread_index]); |
| 108 | pool_get_aligned (udp_main.connections[thread_index], uc, |
| 109 | CLIB_CACHE_LINE_BYTES); |
| 110 | clib_spinlock_unlock_if_init (&udp_main.peekers_write_locks |
| 111 | [thread_index]); |
| 112 | } |
| 113 | else |
| 114 | { |
| 115 | pool_get_aligned (um->connections[thread_index], uc, |
| 116 | CLIB_CACHE_LINE_BYTES); |
| 117 | } |
Dave Barach | b7b9299 | 2018-10-17 10:38:51 -0400 | [diff] [blame] | 118 | clib_memset (uc, 0, sizeof (*uc)); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 119 | uc->c_c_index = uc - um->connections[thread_index]; |
| 120 | uc->c_thread_index = thread_index; |
| 121 | uc->c_proto = TRANSPORT_PROTO_UDP; |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 122 | clib_spinlock_init (&uc->rx_lock); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 123 | return uc; |
| 124 | } |
| 125 | |
| 126 | void |
| 127 | udp_connection_free (udp_connection_t * uc) |
| 128 | { |
Benoît Ganne | d4aeb84 | 2019-07-18 18:38:42 +0200 | [diff] [blame] | 129 | u32 thread_index = uc->c_thread_index; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 130 | if (CLIB_DEBUG) |
Dave Barach | b7b9299 | 2018-10-17 10:38:51 -0400 | [diff] [blame] | 131 | clib_memset (uc, 0xFA, sizeof (*uc)); |
Benoît Ganne | d4aeb84 | 2019-07-18 18:38:42 +0200 | [diff] [blame] | 132 | pool_put (udp_main.connections[thread_index], uc); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 133 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 134 | |
Florin Coras | 57a5a2d | 2020-04-01 23:16:11 +0000 | [diff] [blame] | 135 | static void |
| 136 | udp_connection_cleanup (udp_connection_t * uc) |
| 137 | { |
| 138 | transport_endpoint_cleanup (TRANSPORT_PROTO_UDP, &uc->c_lcl_ip, |
| 139 | uc->c_lcl_port); |
| 140 | udp_connection_unregister_port (clib_net_to_host_u16 (uc->c_lcl_port), |
| 141 | uc->c_is_ip4); |
| 142 | udp_connection_free (uc); |
| 143 | } |
| 144 | |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 145 | void |
| 146 | udp_connection_delete (udp_connection_t * uc) |
| 147 | { |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 148 | session_transport_delete_notify (&uc->connection); |
Florin Coras | 57a5a2d | 2020-04-01 23:16:11 +0000 | [diff] [blame] | 149 | udp_connection_cleanup (uc); |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 150 | } |
| 151 | |
Florin Coras | 57660d9 | 2020-04-04 22:45:34 +0000 | [diff] [blame] | 152 | static u8 |
| 153 | udp_connection_port_used_extern (u16 lcl_port, u8 is_ip4) |
| 154 | { |
| 155 | udp_main_t *um = vnet_get_udp_main (); |
| 156 | udp_dst_port_info_t *pi; |
| 157 | |
| 158 | pi = udp_get_dst_port_info (um, lcl_port, is_ip4); |
| 159 | return (pi && !pi->n_connections |
| 160 | && udp_is_valid_dst_port (lcl_port, is_ip4)); |
| 161 | } |
| 162 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 163 | u32 |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 164 | udp_session_bind (u32 session_index, transport_endpoint_t * lcl) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 165 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 166 | udp_main_t *um = vnet_get_udp_main (); |
| 167 | vlib_main_t *vm = vlib_get_main (); |
Florin Coras | 87b7e3d | 2020-03-27 15:06:07 +0000 | [diff] [blame] | 168 | transport_endpoint_cfg_t *lcl_ext; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 169 | udp_connection_t *listener; |
Florin Coras | 57a5a2d | 2020-04-01 23:16:11 +0000 | [diff] [blame] | 170 | u16 lcl_port_ho; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 171 | void *iface_ip; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 172 | |
Florin Coras | 57a5a2d | 2020-04-01 23:16:11 +0000 | [diff] [blame] | 173 | lcl_port_ho = clib_net_to_host_u16 (lcl->port); |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 174 | |
Florin Coras | 57660d9 | 2020-04-04 22:45:34 +0000 | [diff] [blame] | 175 | if (udp_connection_port_used_extern (lcl_port_ho, lcl->is_ip4)) |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 176 | { |
| 177 | clib_warning ("port already used"); |
Florin Coras | 57660d9 | 2020-04-04 22:45:34 +0000 | [diff] [blame] | 178 | return SESSION_E_PORTINUSE; |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 179 | } |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 180 | |
| 181 | pool_get (um->listener_pool, listener); |
Dave Barach | b7b9299 | 2018-10-17 10:38:51 -0400 | [diff] [blame] | 182 | clib_memset (listener, 0, sizeof (udp_connection_t)); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 183 | |
Florin Coras | 0e49568 | 2017-09-19 22:27:18 -0700 | [diff] [blame] | 184 | listener->c_lcl_port = lcl->port; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 185 | listener->c_c_index = listener - um->listener_pool; |
| 186 | |
| 187 | /* If we are provided a sw_if_index, bind using one of its ips */ |
| 188 | if (ip_is_zero (&lcl->ip, 1) && lcl->sw_if_index != ENDPOINT_INVALID_INDEX) |
| 189 | { |
| 190 | if ((iface_ip = ip_interface_get_first_ip (lcl->sw_if_index, |
| 191 | lcl->is_ip4))) |
| 192 | ip_set (&lcl->ip, iface_ip, lcl->is_ip4); |
| 193 | } |
| 194 | ip_copy (&listener->c_lcl_ip, &lcl->ip, lcl->is_ip4); |
| 195 | listener->c_is_ip4 = lcl->is_ip4; |
| 196 | listener->c_proto = TRANSPORT_PROTO_UDP; |
| 197 | listener->c_s_index = session_index; |
| 198 | listener->c_fib_index = lcl->fib_index; |
Florin Coras | ba78e23 | 2020-04-06 21:28:59 +0000 | [diff] [blame] | 199 | listener->mss = um->default_mtu - sizeof (udp_header_t); |
Florin Coras | 9c4ec6f | 2020-04-01 02:50:13 +0000 | [diff] [blame] | 200 | listener->flags |= UDP_CONN_F_OWNS_PORT | UDP_CONN_F_LISTEN; |
Florin Coras | 87b7e3d | 2020-03-27 15:06:07 +0000 | [diff] [blame] | 201 | lcl_ext = (transport_endpoint_cfg_t *) lcl; |
| 202 | if (lcl_ext->transport_flags & TRANSPORT_CFG_F_CONNECTED) |
| 203 | listener->flags |= UDP_CONN_F_CONNECTED; |
| 204 | else |
| 205 | listener->c_flags |= TRANSPORT_CONNECTION_F_CLESS; |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 206 | clib_spinlock_init (&listener->rx_lock); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 207 | |
Florin Coras | 57a5a2d | 2020-04-01 23:16:11 +0000 | [diff] [blame] | 208 | udp_connection_register_port (vm, lcl_port_ho, lcl->is_ip4); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 209 | return listener->c_c_index; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 210 | } |
| 211 | |
| 212 | u32 |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 213 | udp_session_unbind (u32 listener_index) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 214 | { |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 215 | udp_main_t *um = &udp_main; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 216 | udp_connection_t *listener; |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 217 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 218 | listener = udp_listener_get (listener_index); |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 219 | udp_connection_unregister_port (clib_net_to_host_u16 (listener->c_lcl_port), |
| 220 | listener->c_is_ip4); |
| 221 | pool_put (um->listener_pool, listener); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 222 | return 0; |
| 223 | } |
| 224 | |
| 225 | transport_connection_t * |
| 226 | udp_session_get_listener (u32 listener_index) |
| 227 | { |
| 228 | udp_connection_t *us; |
| 229 | |
| 230 | us = udp_listener_get (listener_index); |
| 231 | return &us->connection; |
| 232 | } |
| 233 | |
| 234 | u32 |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 235 | udp_push_header (transport_connection_t * tc, vlib_buffer_t * b) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 236 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 237 | udp_connection_t *uc; |
| 238 | vlib_main_t *vm = vlib_get_main (); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 239 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 240 | uc = udp_get_connection_from_transport (tc); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 241 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 242 | vlib_buffer_push_udp (b, uc->c_lcl_port, uc->c_rmt_port, 1); |
| 243 | if (tc->is_ip4) |
Florin Coras | ab57edb | 2020-04-07 03:46:07 +0000 | [diff] [blame] | 244 | vlib_buffer_push_ip4_custom (vm, b, &uc->c_lcl_ip4, &uc->c_rmt_ip4, |
| 245 | IP_PROTOCOL_UDP, 1 /* csum offload */ , |
| 246 | 0 /* is_df */ ); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 247 | else |
Florin Coras | ab57edb | 2020-04-07 03:46:07 +0000 | [diff] [blame] | 248 | vlib_buffer_push_ip6 (vm, b, &uc->c_lcl_ip6, &uc->c_rmt_ip6, |
| 249 | IP_PROTOCOL_UDP); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 250 | vnet_buffer (b)->sw_if_index[VLIB_RX] = 0; |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 251 | vnet_buffer (b)->sw_if_index[VLIB_TX] = uc->c_fib_index; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 252 | b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED; |
| 253 | |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 254 | if (PREDICT_FALSE (uc->flags & UDP_CONN_F_CLOSING)) |
| 255 | { |
| 256 | if (!transport_max_tx_dequeue (&uc->connection)) |
| 257 | udp_connection_delete (uc); |
| 258 | } |
| 259 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 260 | return 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 261 | } |
| 262 | |
| 263 | transport_connection_t * |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 264 | udp_session_get (u32 connection_index, u32 thread_index) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 265 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 266 | udp_connection_t *uc; |
| 267 | uc = udp_connection_get (connection_index, thread_index); |
| 268 | if (uc) |
| 269 | return &uc->connection; |
| 270 | return 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 271 | } |
| 272 | |
| 273 | void |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 274 | udp_session_close (u32 connection_index, u32 thread_index) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 275 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 276 | udp_connection_t *uc; |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 277 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 278 | uc = udp_connection_get (connection_index, thread_index); |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 279 | if (!uc) |
| 280 | return; |
| 281 | |
| 282 | if (!transport_max_tx_dequeue (&uc->connection)) |
| 283 | udp_connection_delete (uc); |
| 284 | else |
| 285 | uc->flags |= UDP_CONN_F_CLOSING; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 286 | } |
| 287 | |
| 288 | void |
| 289 | udp_session_cleanup (u32 connection_index, u32 thread_index) |
| 290 | { |
| 291 | udp_connection_t *uc; |
| 292 | uc = udp_connection_get (connection_index, thread_index); |
Florin Coras | d85666f | 2020-04-03 00:58:48 +0000 | [diff] [blame] | 293 | if (!uc) |
| 294 | return; |
| 295 | if (uc->flags & UDP_CONN_F_MIGRATED) |
| 296 | udp_connection_free (uc); |
| 297 | else |
Florin Coras | 57a5a2d | 2020-04-01 23:16:11 +0000 | [diff] [blame] | 298 | udp_connection_cleanup (uc); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 299 | } |
| 300 | |
| 301 | u8 * |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 302 | format_udp_connection_id (u8 * s, va_list * args) |
| 303 | { |
| 304 | udp_connection_t *uc = va_arg (*args, udp_connection_t *); |
| 305 | if (!uc) |
| 306 | return s; |
| 307 | if (uc->c_is_ip4) |
Florin Coras | 9c4ec6f | 2020-04-01 02:50:13 +0000 | [diff] [blame] | 308 | s = format (s, "[%u:%u][%s] %U:%d->%U:%d", uc->c_thread_index, |
| 309 | uc->c_s_index, "U", format_ip4_address, &uc->c_lcl_ip4, |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 310 | clib_net_to_host_u16 (uc->c_lcl_port), format_ip4_address, |
| 311 | &uc->c_rmt_ip4, clib_net_to_host_u16 (uc->c_rmt_port)); |
| 312 | else |
Florin Coras | 9c4ec6f | 2020-04-01 02:50:13 +0000 | [diff] [blame] | 313 | s = format (s, "[%u:%u][%s] %U:%d->%U:%d", uc->c_thread_index, |
| 314 | uc->c_s_index, "U", format_ip6_address, &uc->c_lcl_ip6, |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 315 | clib_net_to_host_u16 (uc->c_lcl_port), format_ip6_address, |
| 316 | &uc->c_rmt_ip6, clib_net_to_host_u16 (uc->c_rmt_port)); |
| 317 | return s; |
| 318 | } |
| 319 | |
Florin Coras | 9c4ec6f | 2020-04-01 02:50:13 +0000 | [diff] [blame] | 320 | const char *udp_connection_flags_str[] = { |
| 321 | #define _(sym, str) str, |
| 322 | foreach_udp_connection_flag |
| 323 | #undef _ |
| 324 | }; |
| 325 | |
| 326 | static u8 * |
| 327 | format_udp_connection_flags (u8 * s, va_list * args) |
| 328 | { |
| 329 | udp_connection_t *uc = va_arg (*args, udp_connection_t *); |
| 330 | int i, last = -1; |
| 331 | |
| 332 | for (i = 0; i < UDP_CONN_N_FLAGS; i++) |
| 333 | if (uc->flags & (1 << i)) |
| 334 | last = i; |
| 335 | for (i = 0; i < last; i++) |
| 336 | { |
| 337 | if (uc->flags & (1 << i)) |
| 338 | s = format (s, "%s, ", udp_connection_flags_str[i]); |
| 339 | } |
| 340 | if (last >= 0) |
| 341 | s = format (s, "%s", udp_connection_flags_str[last]); |
| 342 | return s; |
| 343 | } |
| 344 | |
| 345 | static u8 * |
| 346 | format_udp_vars (u8 * s, va_list * args) |
| 347 | { |
| 348 | udp_connection_t *uc = va_arg (*args, udp_connection_t *); |
| 349 | s = format (s, " index %u flags: %U", uc->c_c_index, |
| 350 | format_udp_connection_flags, uc); |
| 351 | |
| 352 | if (!(uc->flags & UDP_CONN_F_LISTEN)) |
| 353 | s = format (s, "\n"); |
| 354 | return s; |
| 355 | } |
| 356 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 357 | u8 * |
| 358 | format_udp_connection (u8 * s, va_list * args) |
| 359 | { |
| 360 | udp_connection_t *uc = va_arg (*args, udp_connection_t *); |
| 361 | u32 verbose = va_arg (*args, u32); |
| 362 | if (!uc) |
| 363 | return s; |
| 364 | s = format (s, "%-50U", format_udp_connection_id, uc); |
| 365 | if (verbose) |
| 366 | { |
Florin Coras | 9c4ec6f | 2020-04-01 02:50:13 +0000 | [diff] [blame] | 367 | s = format (s, "%-15s", |
| 368 | (uc->flags & UDP_CONN_F_LISTEN) ? "LISTEN" : "OPENED", uc); |
| 369 | if (verbose > 1) |
| 370 | s = format (s, "\n%U", format_udp_vars, uc); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 371 | } |
| 372 | return s; |
| 373 | } |
| 374 | |
| 375 | u8 * |
| 376 | format_udp_session (u8 * s, va_list * args) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 377 | { |
| 378 | u32 uci = va_arg (*args, u32); |
| 379 | u32 thread_index = va_arg (*args, u32); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 380 | u32 verbose = va_arg (*args, u32); |
| 381 | udp_connection_t *uc; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 382 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 383 | uc = udp_connection_get (uci, thread_index); |
| 384 | return format (s, "%U", format_udp_connection, uc, verbose); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 385 | } |
| 386 | |
| 387 | u8 * |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 388 | format_udp_half_open_session (u8 * s, va_list * args) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 389 | { |
Dave Wallace | 3df5682 | 2019-05-22 18:56:57 -0400 | [diff] [blame] | 390 | u32 __clib_unused tci = va_arg (*args, u32); |
Aloys Augustin | a0abbff | 2019-07-12 12:16:16 +0200 | [diff] [blame] | 391 | u32 __clib_unused thread_index = va_arg (*args, u32); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 392 | clib_warning ("BUG"); |
| 393 | return 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 394 | } |
| 395 | |
| 396 | u8 * |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 397 | format_udp_listener_session (u8 * s, va_list * args) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 398 | { |
| 399 | u32 tci = va_arg (*args, u32); |
Aloys Augustin | a0abbff | 2019-07-12 12:16:16 +0200 | [diff] [blame] | 400 | u32 __clib_unused thread_index = va_arg (*args, u32); |
Aloys Augustin | 95dd1fe | 2019-07-12 17:11:04 +0200 | [diff] [blame] | 401 | u32 verbose = va_arg (*args, u32); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 402 | udp_connection_t *uc = udp_listener_get (tci); |
Aloys Augustin | 95dd1fe | 2019-07-12 17:11:04 +0200 | [diff] [blame] | 403 | return format (s, "%U", format_udp_connection, uc, verbose); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 404 | } |
| 405 | |
Florin Coras | 70f879d | 2020-03-13 17:54:42 +0000 | [diff] [blame] | 406 | static int |
| 407 | udp_session_send_params (transport_connection_t * tconn, |
| 408 | transport_send_params_t * sp) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 409 | { |
Florin Coras | ba78e23 | 2020-04-06 21:28:59 +0000 | [diff] [blame] | 410 | udp_connection_t *uc; |
| 411 | |
| 412 | uc = udp_get_connection_from_transport (tconn); |
| 413 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 414 | /* No constraint on TX window */ |
Florin Coras | 70f879d | 2020-03-13 17:54:42 +0000 | [diff] [blame] | 415 | sp->snd_space = ~0; |
| 416 | /* TODO figure out MTU of output interface */ |
Florin Coras | ba78e23 | 2020-04-06 21:28:59 +0000 | [diff] [blame] | 417 | sp->snd_mss = uc->mss; |
Florin Coras | 70f879d | 2020-03-13 17:54:42 +0000 | [diff] [blame] | 418 | sp->tx_offset = 0; |
| 419 | sp->flags = 0; |
| 420 | return 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 421 | } |
| 422 | |
| 423 | int |
Florin Coras | 5665ced | 2018-10-25 18:03:45 -0700 | [diff] [blame] | 424 | udp_open_connection (transport_endpoint_cfg_t * rmt) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 425 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 426 | vlib_main_t *vm = vlib_get_main (); |
Damjan Marion | 067cd62 | 2018-07-11 12:47:43 +0200 | [diff] [blame] | 427 | u32 thread_index = vm->thread_index; |
Florin Coras | ba78e23 | 2020-04-06 21:28:59 +0000 | [diff] [blame] | 428 | udp_main_t *um = &udp_main; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 429 | ip46_address_t lcl_addr; |
Florin Coras | 57660d9 | 2020-04-04 22:45:34 +0000 | [diff] [blame] | 430 | udp_connection_t *uc; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 431 | u16 lcl_port; |
Florin Coras | 00e01d3 | 2019-10-21 16:07:46 -0700 | [diff] [blame] | 432 | int rv; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 433 | |
Florin Coras | 00e01d3 | 2019-10-21 16:07:46 -0700 | [diff] [blame] | 434 | rv = transport_alloc_local_endpoint (TRANSPORT_PROTO_UDP, rmt, &lcl_addr, |
| 435 | &lcl_port); |
| 436 | if (rv) |
Florin Coras | 57660d9 | 2020-04-04 22:45:34 +0000 | [diff] [blame] | 437 | { |
| 438 | if (rv != SESSION_E_PORTINUSE) |
| 439 | return rv; |
| 440 | |
| 441 | if (udp_connection_port_used_extern (lcl_port, rmt->is_ip4)) |
| 442 | return SESSION_E_PORTINUSE; |
| 443 | |
| 444 | /* If port in use, check if 5-tuple is also in use */ |
| 445 | if (session_lookup_connection (rmt->fib_index, &lcl_addr, &rmt->ip, |
| 446 | lcl_port, rmt->port, TRANSPORT_PROTO_UDP, |
| 447 | rmt->is_ip4)) |
| 448 | return SESSION_E_PORTINUSE; |
| 449 | |
| 450 | /* 5-tuple is available so increase lcl endpoint refcount and proceed |
| 451 | * with connection allocation */ |
| 452 | transport_share_local_endpoint (TRANSPORT_PROTO_UDP, &lcl_addr, |
| 453 | lcl_port); |
| 454 | goto conn_alloc; |
| 455 | } |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 456 | |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 457 | if (udp_is_valid_dst_port (lcl_port, rmt->is_ip4)) |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 458 | { |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 459 | /* If specific source port was requested abort */ |
| 460 | if (rmt->peer.port) |
Florin Coras | 00e01d3 | 2019-10-21 16:07:46 -0700 | [diff] [blame] | 461 | return SESSION_E_PORTINUSE; |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 462 | |
| 463 | /* Try to find a port that's not used */ |
| 464 | while (udp_is_valid_dst_port (lcl_port, rmt->is_ip4)) |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 465 | { |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 466 | lcl_port = transport_alloc_local_port (TRANSPORT_PROTO_UDP, |
| 467 | &lcl_addr); |
| 468 | if (lcl_port < 1) |
Florin Coras | 00e01d3 | 2019-10-21 16:07:46 -0700 | [diff] [blame] | 469 | return SESSION_E_PORTINUSE; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 470 | } |
| 471 | } |
| 472 | |
Florin Coras | 57660d9 | 2020-04-04 22:45:34 +0000 | [diff] [blame] | 473 | conn_alloc: |
| 474 | |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 475 | udp_connection_register_port (vm, lcl_port, rmt->is_ip4); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 476 | |
Florin Coras | 40903ac | 2018-06-10 14:41:23 -0700 | [diff] [blame] | 477 | /* We don't poll main thread if we have workers */ |
| 478 | if (vlib_num_workers ()) |
| 479 | thread_index = 1; |
| 480 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 481 | uc = udp_connection_alloc (thread_index); |
| 482 | ip_copy (&uc->c_rmt_ip, &rmt->ip, rmt->is_ip4); |
| 483 | ip_copy (&uc->c_lcl_ip, &lcl_addr, rmt->is_ip4); |
| 484 | uc->c_rmt_port = rmt->port; |
| 485 | uc->c_lcl_port = clib_host_to_net_u16 (lcl_port); |
| 486 | uc->c_is_ip4 = rmt->is_ip4; |
| 487 | uc->c_proto = TRANSPORT_PROTO_UDP; |
| 488 | uc->c_fib_index = rmt->fib_index; |
Florin Coras | ba78e23 | 2020-04-06 21:28:59 +0000 | [diff] [blame] | 489 | uc->mss = rmt->mss ? rmt->mss : (um->default_mtu - sizeof (udp_header_t)); |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 490 | uc->flags |= UDP_CONN_F_OWNS_PORT; |
Florin Coras | 87b7e3d | 2020-03-27 15:06:07 +0000 | [diff] [blame] | 491 | if (rmt->transport_flags & TRANSPORT_CFG_F_CONNECTED) |
| 492 | uc->flags |= UDP_CONN_F_CONNECTED; |
| 493 | else |
| 494 | uc->c_flags |= TRANSPORT_CONNECTION_F_CLESS; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 495 | |
| 496 | return uc->c_c_index; |
| 497 | } |
| 498 | |
| 499 | transport_connection_t * |
Florin Coras | 40903ac | 2018-06-10 14:41:23 -0700 | [diff] [blame] | 500 | udp_session_get_half_open (u32 conn_index) |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 501 | { |
| 502 | udp_connection_t *uc; |
Florin Coras | 40903ac | 2018-06-10 14:41:23 -0700 | [diff] [blame] | 503 | u32 thread_index; |
| 504 | |
| 505 | /* We don't poll main thread if we have workers */ |
| 506 | thread_index = vlib_num_workers ()? 1 : 0; |
| 507 | uc = udp_connection_get (conn_index, thread_index); |
Nathan Skrzypczak | e971bc9 | 2019-06-19 13:42:37 +0200 | [diff] [blame] | 508 | if (!uc) |
| 509 | return 0; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 510 | return &uc->connection; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 511 | } |
| 512 | |
| 513 | /* *INDENT-OFF* */ |
Nathan Skrzypczak | e971bc9 | 2019-06-19 13:42:37 +0200 | [diff] [blame] | 514 | static const transport_proto_vft_t udp_proto = { |
Florin Coras | 1ee7830 | 2019-02-05 15:51:15 -0800 | [diff] [blame] | 515 | .start_listen = udp_session_bind, |
| 516 | .connect = udp_open_connection, |
| 517 | .stop_listen = udp_session_unbind, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 518 | .push_header = udp_push_header, |
| 519 | .get_connection = udp_session_get, |
| 520 | .get_listener = udp_session_get_listener, |
Florin Coras | 40903ac | 2018-06-10 14:41:23 -0700 | [diff] [blame] | 521 | .get_half_open = udp_session_get_half_open, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 522 | .close = udp_session_close, |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 523 | .cleanup = udp_session_cleanup, |
Florin Coras | 70f879d | 2020-03-13 17:54:42 +0000 | [diff] [blame] | 524 | .send_params = udp_session_send_params, |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 525 | .format_connection = format_udp_session, |
| 526 | .format_half_open = format_udp_half_open_session, |
Florin Coras | 371ca50 | 2018-02-21 12:07:41 -0800 | [diff] [blame] | 527 | .format_listener = format_udp_listener_session, |
Nathan Skrzypczak | e971bc9 | 2019-06-19 13:42:37 +0200 | [diff] [blame] | 528 | .transport_options = { |
Florin Coras | 07063b8 | 2020-03-13 04:44:51 +0000 | [diff] [blame] | 529 | .name = "udp", |
| 530 | .short_name = "U", |
Nathan Skrzypczak | e971bc9 | 2019-06-19 13:42:37 +0200 | [diff] [blame] | 531 | .tx_type = TRANSPORT_TX_DGRAM, |
| 532 | .service_type = TRANSPORT_SERVICE_CL, |
| 533 | }, |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 534 | }; |
| 535 | /* *INDENT-ON* */ |
| 536 | |
| 537 | |
| 538 | int |
Florin Coras | 5665ced | 2018-10-25 18:03:45 -0700 | [diff] [blame] | 539 | udpc_connection_open (transport_endpoint_cfg_t * rmt) |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 540 | { |
| 541 | udp_connection_t *uc; |
Aloys Augustin | 8e58499 | 2019-04-24 13:21:43 +0200 | [diff] [blame] | 542 | /* Reproduce the logic of udp_open_connection to find the correct thread */ |
| 543 | u32 thread_index = vlib_num_workers ()? 1 : vlib_get_main ()->thread_index; |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 544 | u32 uc_index; |
| 545 | uc_index = udp_open_connection (rmt); |
Nathan Skrzypczak | 50f4a41 | 2019-07-04 14:20:17 +0200 | [diff] [blame] | 546 | if (uc_index == (u32) ~ 0) |
| 547 | return -1; |
Aloys Augustin | 8e58499 | 2019-04-24 13:21:43 +0200 | [diff] [blame] | 548 | uc = udp_connection_get (uc_index, thread_index); |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 549 | uc->flags |= UDP_CONN_F_CONNECTED; |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 550 | return uc_index; |
| 551 | } |
| 552 | |
| 553 | u32 |
| 554 | udpc_connection_listen (u32 session_index, transport_endpoint_t * lcl) |
| 555 | { |
| 556 | udp_connection_t *listener; |
Nathan Skrzypczak | 50f4a41 | 2019-07-04 14:20:17 +0200 | [diff] [blame] | 557 | u32 li_index; |
| 558 | li_index = udp_session_bind (session_index, lcl); |
| 559 | if (li_index == (u32) ~ 0) |
| 560 | return -1; |
| 561 | listener = udp_listener_get (li_index); |
Florin Coras | c754239 | 2019-07-22 08:08:43 -0700 | [diff] [blame] | 562 | listener->flags |= UDP_CONN_F_CONNECTED; |
Florin Coras | 95cd864 | 2019-12-30 21:53:19 -0800 | [diff] [blame] | 563 | /* Fake udp listener, i.e., make sure session layer adds a udp instead of |
| 564 | * udpc listener to the lookup table */ |
| 565 | ((session_endpoint_cfg_t *) lcl)->transport_proto = TRANSPORT_PROTO_UDP; |
Nathan Skrzypczak | 50f4a41 | 2019-07-04 14:20:17 +0200 | [diff] [blame] | 566 | return li_index; |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 567 | } |
| 568 | |
| 569 | /* *INDENT-OFF* */ |
Nathan Skrzypczak | e971bc9 | 2019-06-19 13:42:37 +0200 | [diff] [blame] | 570 | static const transport_proto_vft_t udpc_proto = { |
Florin Coras | 1ee7830 | 2019-02-05 15:51:15 -0800 | [diff] [blame] | 571 | .start_listen = udpc_connection_listen, |
| 572 | .stop_listen = udp_session_unbind, |
| 573 | .connect = udpc_connection_open, |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 574 | .push_header = udp_push_header, |
| 575 | .get_connection = udp_session_get, |
| 576 | .get_listener = udp_session_get_listener, |
Florin Coras | 40903ac | 2018-06-10 14:41:23 -0700 | [diff] [blame] | 577 | .get_half_open = udp_session_get_half_open, |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 578 | .close = udp_session_close, |
| 579 | .cleanup = udp_session_cleanup, |
Florin Coras | 70f879d | 2020-03-13 17:54:42 +0000 | [diff] [blame] | 580 | .send_params = udp_session_send_params, |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 581 | .format_connection = format_udp_session, |
| 582 | .format_half_open = format_udp_half_open_session, |
| 583 | .format_listener = format_udp_listener_session, |
Nathan Skrzypczak | e971bc9 | 2019-06-19 13:42:37 +0200 | [diff] [blame] | 584 | .transport_options = { |
Florin Coras | 07063b8 | 2020-03-13 04:44:51 +0000 | [diff] [blame] | 585 | .name = "udpc", |
| 586 | .short_name = "U", |
Nathan Skrzypczak | e971bc9 | 2019-06-19 13:42:37 +0200 | [diff] [blame] | 587 | .tx_type = TRANSPORT_TX_DGRAM, |
| 588 | .service_type = TRANSPORT_SERVICE_VC, |
Nathan Skrzypczak | a26349d | 2019-06-26 13:53:08 +0200 | [diff] [blame] | 589 | .half_open_has_fifos = 1 |
Nathan Skrzypczak | e971bc9 | 2019-06-19 13:42:37 +0200 | [diff] [blame] | 590 | }, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 591 | }; |
| 592 | /* *INDENT-ON* */ |
| 593 | |
| 594 | static clib_error_t * |
| 595 | udp_init (vlib_main_t * vm) |
| 596 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 597 | udp_main_t *um = vnet_get_udp_main (); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 598 | ip_main_t *im = &ip_main; |
| 599 | vlib_thread_main_t *tm = vlib_get_thread_main (); |
| 600 | u32 num_threads; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 601 | ip_protocol_info_t *pi; |
Florin Coras | 29ca16f | 2017-11-02 18:14:17 -0400 | [diff] [blame] | 602 | int i; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 603 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 604 | /* |
| 605 | * Registrations |
| 606 | */ |
| 607 | |
| 608 | /* IP registration */ |
| 609 | pi = ip_get_protocol_info (im, IP_PROTOCOL_UDP); |
| 610 | if (pi == 0) |
| 611 | return clib_error_return (0, "UDP protocol info AWOL"); |
| 612 | pi->format_header = format_udp_header; |
| 613 | pi->unformat_pg_edit = unformat_pg_udp_header; |
| 614 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 615 | /* Register as transport with URI */ |
Florin Coras | 561af9b | 2017-12-09 10:19:43 -0800 | [diff] [blame] | 616 | transport_register_protocol (TRANSPORT_PROTO_UDP, &udp_proto, |
| 617 | FIB_PROTOCOL_IP4, ip4_lookup_node.index); |
| 618 | transport_register_protocol (TRANSPORT_PROTO_UDP, &udp_proto, |
| 619 | FIB_PROTOCOL_IP6, ip6_lookup_node.index); |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 620 | transport_register_protocol (TRANSPORT_PROTO_UDPC, &udpc_proto, |
| 621 | FIB_PROTOCOL_IP4, ip4_lookup_node.index); |
| 622 | transport_register_protocol (TRANSPORT_PROTO_UDPC, &udpc_proto, |
| 623 | FIB_PROTOCOL_IP6, ip6_lookup_node.index); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 624 | |
| 625 | /* |
| 626 | * Initialize data structures |
| 627 | */ |
| 628 | |
| 629 | num_threads = 1 /* main thread */ + tm->n_threads; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 630 | vec_validate (um->connections, num_threads - 1); |
| 631 | vec_validate (um->connection_peekers, num_threads - 1); |
| 632 | vec_validate (um->peekers_readers_locks, num_threads - 1); |
| 633 | vec_validate (um->peekers_write_locks, num_threads - 1); |
Florin Coras | 29ca16f | 2017-11-02 18:14:17 -0400 | [diff] [blame] | 634 | |
| 635 | if (num_threads > 1) |
| 636 | for (i = 0; i < num_threads; i++) |
| 637 | { |
| 638 | clib_spinlock_init (&um->peekers_readers_locks[i]); |
| 639 | clib_spinlock_init (&um->peekers_write_locks[i]); |
| 640 | } |
Florin Coras | a039620 | 2020-04-01 00:11:16 +0000 | [diff] [blame] | 641 | |
| 642 | um->local_to_input_edge[UDP_IP4] = |
| 643 | vlib_node_add_next (vm, udp4_local_node.index, udp4_input_node.index); |
| 644 | um->local_to_input_edge[UDP_IP6] = |
| 645 | vlib_node_add_next (vm, udp6_local_node.index, udp6_input_node.index); |
Florin Coras | ba78e23 | 2020-04-06 21:28:59 +0000 | [diff] [blame] | 646 | |
| 647 | um->default_mtu = 1500; |
Dave Barach | f8d5068 | 2019-05-14 18:01:44 -0400 | [diff] [blame] | 648 | return 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 649 | } |
| 650 | |
Dave Barach | f8d5068 | 2019-05-14 18:01:44 -0400 | [diff] [blame] | 651 | /* *INDENT-OFF* */ |
| 652 | VLIB_INIT_FUNCTION (udp_init) = |
| 653 | { |
| 654 | .runs_after = VLIB_INITS("ip_main_init", "ip4_lookup_init", |
| 655 | "ip6_lookup_init"), |
| 656 | }; |
| 657 | /* *INDENT-ON* */ |
| 658 | |
Florin Coras | ba78e23 | 2020-04-06 21:28:59 +0000 | [diff] [blame] | 659 | static clib_error_t * |
| 660 | udp_config_fn (vlib_main_t * vm, unformat_input_t * input) |
| 661 | { |
| 662 | udp_main_t *um = &udp_main; |
| 663 | u32 tmp; |
| 664 | |
| 665 | while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) |
| 666 | { |
| 667 | if (unformat (input, "mtu %u", &tmp)) |
| 668 | um->default_mtu = tmp; |
| 669 | else |
| 670 | return clib_error_return (0, "unknown input `%U'", |
| 671 | format_unformat_error, input); |
| 672 | } |
| 673 | return 0; |
| 674 | } |
| 675 | |
| 676 | VLIB_CONFIG_FUNCTION (udp_config_fn, "udp"); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 677 | |
Pavel Kotucek | e88865d | 2018-11-28 07:42:11 +0100 | [diff] [blame] | 678 | static clib_error_t * |
| 679 | show_udp_punt_fn (vlib_main_t * vm, unformat_input_t * input, |
| 680 | vlib_cli_command_t * cmd_arg) |
| 681 | { |
| 682 | udp_main_t *um = vnet_get_udp_main (); |
| 683 | |
| 684 | clib_error_t *error = NULL; |
| 685 | |
| 686 | if (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) |
| 687 | return clib_error_return (0, "unknown input `%U'", format_unformat_error, |
| 688 | input); |
| 689 | |
| 690 | udp_dst_port_info_t *port_info; |
| 691 | if (um->punt_unknown4) |
| 692 | { |
| 693 | vlib_cli_output (vm, "IPv4 UDP punt: enabled"); |
| 694 | } |
| 695 | else |
| 696 | { |
| 697 | u8 *s = NULL; |
| 698 | vec_foreach (port_info, um->dst_port_infos[UDP_IP4]) |
| 699 | { |
| 700 | if (udp_is_valid_dst_port (port_info->dst_port, 1)) |
| 701 | { |
| 702 | s = format (s, (!s) ? "%d" : ", %d", port_info->dst_port); |
| 703 | } |
| 704 | } |
| 705 | s = format (s, "%c", 0); |
| 706 | vlib_cli_output (vm, "IPV4 UDP ports punt : %s", s); |
| 707 | } |
| 708 | |
| 709 | if (um->punt_unknown6) |
| 710 | { |
| 711 | vlib_cli_output (vm, "IPv6 UDP punt: enabled"); |
| 712 | } |
| 713 | else |
| 714 | { |
| 715 | u8 *s = NULL; |
| 716 | vec_foreach (port_info, um->dst_port_infos[UDP_IP6]) |
| 717 | { |
| 718 | if (udp_is_valid_dst_port (port_info->dst_port, 01)) |
| 719 | { |
| 720 | s = format (s, (!s) ? "%d" : ", %d", port_info->dst_port); |
| 721 | } |
| 722 | } |
| 723 | s = format (s, "%c", 0); |
| 724 | vlib_cli_output (vm, "IPV6 UDP ports punt : %s", s); |
| 725 | } |
| 726 | |
| 727 | return (error); |
| 728 | } |
| 729 | /* *INDENT-OFF* */ |
| 730 | VLIB_CLI_COMMAND (show_tcp_punt_command, static) = |
| 731 | { |
| 732 | .path = "show udp punt", |
| 733 | .short_help = "show udp punt [ipv4|ipv6]", |
| 734 | .function = show_udp_punt_fn, |
| 735 | }; |
| 736 | /* *INDENT-ON* */ |
| 737 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 738 | /* |
| 739 | * fd.io coding-style-patch-verification: ON |
| 740 | * |
| 741 | * Local Variables: |
| 742 | * eval: (c-set-style "gnu") |
| 743 | * End: |
| 744 | */ |