Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2016 Cisco and/or its affiliates. |
| 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> |
| 24 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 25 | udp_connection_t * |
| 26 | udp_connection_alloc (u32 thread_index) |
| 27 | { |
| 28 | udp_main_t *um = &udp_main; |
| 29 | udp_connection_t *uc; |
| 30 | u32 will_expand = 0; |
| 31 | pool_get_aligned_will_expand (um->connections[thread_index], will_expand, |
| 32 | CLIB_CACHE_LINE_BYTES); |
| 33 | |
| 34 | if (PREDICT_FALSE (will_expand)) |
| 35 | { |
| 36 | clib_spinlock_lock_if_init (&udp_main.peekers_write_locks |
| 37 | [thread_index]); |
| 38 | pool_get_aligned (udp_main.connections[thread_index], uc, |
| 39 | CLIB_CACHE_LINE_BYTES); |
| 40 | clib_spinlock_unlock_if_init (&udp_main.peekers_write_locks |
| 41 | [thread_index]); |
| 42 | } |
| 43 | else |
| 44 | { |
| 45 | pool_get_aligned (um->connections[thread_index], uc, |
| 46 | CLIB_CACHE_LINE_BYTES); |
| 47 | } |
| 48 | memset (uc, 0, sizeof (*uc)); |
| 49 | uc->c_c_index = uc - um->connections[thread_index]; |
| 50 | uc->c_thread_index = thread_index; |
| 51 | uc->c_proto = TRANSPORT_PROTO_UDP; |
| 52 | return uc; |
| 53 | } |
| 54 | |
| 55 | void |
| 56 | udp_connection_free (udp_connection_t * uc) |
| 57 | { |
| 58 | pool_put (udp_main.connections[uc->c_thread_index], uc); |
| 59 | if (CLIB_DEBUG) |
| 60 | memset (uc, 0xFA, sizeof (*uc)); |
| 61 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 62 | |
| 63 | u32 |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 64 | udp_session_bind (u32 session_index, transport_endpoint_t * lcl) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 65 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 66 | udp_main_t *um = vnet_get_udp_main (); |
| 67 | vlib_main_t *vm = vlib_get_main (); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 68 | udp_connection_t *listener; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 69 | u32 node_index; |
| 70 | void *iface_ip; |
| 71 | udp_dst_port_info_t *pi; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 72 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 73 | pi = udp_get_dst_port_info (um, lcl->port, lcl->is_ip4); |
| 74 | if (pi) |
| 75 | return -1; |
| 76 | |
| 77 | pool_get (um->listener_pool, listener); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 78 | memset (listener, 0, sizeof (udp_connection_t)); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 79 | |
Florin Coras | 0e49568 | 2017-09-19 22:27:18 -0700 | [diff] [blame] | 80 | listener->c_lcl_port = lcl->port; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 81 | listener->c_c_index = listener - um->listener_pool; |
| 82 | |
| 83 | /* If we are provided a sw_if_index, bind using one of its ips */ |
| 84 | if (ip_is_zero (&lcl->ip, 1) && lcl->sw_if_index != ENDPOINT_INVALID_INDEX) |
| 85 | { |
| 86 | if ((iface_ip = ip_interface_get_first_ip (lcl->sw_if_index, |
| 87 | lcl->is_ip4))) |
| 88 | ip_set (&lcl->ip, iface_ip, lcl->is_ip4); |
| 89 | } |
| 90 | ip_copy (&listener->c_lcl_ip, &lcl->ip, lcl->is_ip4); |
| 91 | listener->c_is_ip4 = lcl->is_ip4; |
| 92 | listener->c_proto = TRANSPORT_PROTO_UDP; |
| 93 | listener->c_s_index = session_index; |
| 94 | listener->c_fib_index = lcl->fib_index; |
| 95 | |
| 96 | node_index = lcl->is_ip4 ? udp4_input_node.index : udp6_input_node.index; |
| 97 | udp_register_dst_port (vm, clib_net_to_host_u16 (lcl->port), node_index, |
| 98 | 1 /* is_ipv4 */ ); |
| 99 | return listener->c_c_index; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 100 | } |
| 101 | |
| 102 | u32 |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 103 | udp_session_unbind (u32 listener_index) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 104 | { |
Florin Coras | e69f495 | 2017-03-07 10:06:24 -0800 | [diff] [blame] | 105 | vlib_main_t *vm = vlib_get_main (); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 106 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 107 | udp_connection_t *listener; |
| 108 | listener = udp_listener_get (listener_index); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 109 | udp_unregister_dst_port (vm, clib_net_to_host_u16 (listener->c_lcl_port), |
| 110 | listener->c_is_ip4); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 111 | return 0; |
| 112 | } |
| 113 | |
| 114 | transport_connection_t * |
| 115 | udp_session_get_listener (u32 listener_index) |
| 116 | { |
| 117 | udp_connection_t *us; |
| 118 | |
| 119 | us = udp_listener_get (listener_index); |
| 120 | return &us->connection; |
| 121 | } |
| 122 | |
| 123 | u32 |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 124 | udp_push_header (transport_connection_t * tc, vlib_buffer_t * b) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 125 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 126 | udp_connection_t *uc; |
| 127 | vlib_main_t *vm = vlib_get_main (); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 128 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 129 | uc = udp_get_connection_from_transport (tc); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 130 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 131 | vlib_buffer_push_udp (b, uc->c_lcl_port, uc->c_rmt_port, 1); |
| 132 | if (tc->is_ip4) |
| 133 | vlib_buffer_push_ip4 (vm, b, &uc->c_lcl_ip4, &uc->c_rmt_ip4, |
| 134 | IP_PROTOCOL_UDP, 1); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 135 | else |
| 136 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 137 | ip6_header_t *ih; |
| 138 | ih = vlib_buffer_push_ip6 (vm, b, &uc->c_lcl_ip6, &uc->c_rmt_ip6, |
| 139 | IP_PROTOCOL_UDP); |
| 140 | vnet_buffer (b)->l3_hdr_offset = (u8 *) ih - b->data; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 141 | } |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 142 | vnet_buffer (b)->sw_if_index[VLIB_RX] = 0; |
| 143 | vnet_buffer (b)->sw_if_index[VLIB_TX] = ~0; |
| 144 | b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED; |
| 145 | |
| 146 | return 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 147 | } |
| 148 | |
| 149 | transport_connection_t * |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 150 | udp_session_get (u32 connection_index, u32 thread_index) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 151 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 152 | udp_connection_t *uc; |
| 153 | uc = udp_connection_get (connection_index, thread_index); |
| 154 | if (uc) |
| 155 | return &uc->connection; |
| 156 | return 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 157 | } |
| 158 | |
| 159 | void |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 160 | udp_session_close (u32 connection_index, u32 thread_index) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 161 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 162 | vlib_main_t *vm = vlib_get_main (); |
| 163 | udp_connection_t *uc; |
| 164 | uc = udp_connection_get (connection_index, thread_index); |
| 165 | if (uc) |
| 166 | { |
| 167 | udp_unregister_dst_port (vm, clib_net_to_host_u16 (uc->c_lcl_port), |
| 168 | uc->c_is_ip4); |
| 169 | stream_session_delete_notify (&uc->connection); |
| 170 | udp_connection_free (uc); |
| 171 | } |
| 172 | } |
| 173 | |
| 174 | void |
| 175 | udp_session_cleanup (u32 connection_index, u32 thread_index) |
| 176 | { |
| 177 | udp_connection_t *uc; |
| 178 | uc = udp_connection_get (connection_index, thread_index); |
| 179 | if (uc) |
| 180 | udp_connection_free (uc); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 181 | } |
| 182 | |
| 183 | u8 * |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 184 | format_udp_connection_id (u8 * s, va_list * args) |
| 185 | { |
| 186 | udp_connection_t *uc = va_arg (*args, udp_connection_t *); |
| 187 | if (!uc) |
| 188 | return s; |
| 189 | if (uc->c_is_ip4) |
| 190 | s = format (s, "[#%d][%s] %U:%d->%U:%d", uc->c_thread_index, "U", |
| 191 | format_ip4_address, &uc->c_lcl_ip4, |
| 192 | clib_net_to_host_u16 (uc->c_lcl_port), format_ip4_address, |
| 193 | &uc->c_rmt_ip4, clib_net_to_host_u16 (uc->c_rmt_port)); |
| 194 | else |
| 195 | s = format (s, "[#%d][%s] %U:%d->%U:%d", uc->c_thread_index, "U", |
| 196 | format_ip6_address, &uc->c_lcl_ip6, |
| 197 | clib_net_to_host_u16 (uc->c_lcl_port), format_ip6_address, |
| 198 | &uc->c_rmt_ip6, clib_net_to_host_u16 (uc->c_rmt_port)); |
| 199 | return s; |
| 200 | } |
| 201 | |
| 202 | u8 * |
| 203 | format_udp_connection (u8 * s, va_list * args) |
| 204 | { |
| 205 | udp_connection_t *uc = va_arg (*args, udp_connection_t *); |
| 206 | u32 verbose = va_arg (*args, u32); |
| 207 | if (!uc) |
| 208 | return s; |
| 209 | s = format (s, "%-50U", format_udp_connection_id, uc); |
| 210 | if (verbose) |
| 211 | { |
| 212 | if (verbose == 1) |
| 213 | s = format (s, "%-15s", "-"); |
| 214 | else |
| 215 | s = format (s, "\n"); |
| 216 | } |
| 217 | return s; |
| 218 | } |
| 219 | |
| 220 | u8 * |
| 221 | format_udp_session (u8 * s, va_list * args) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 222 | { |
| 223 | u32 uci = va_arg (*args, u32); |
| 224 | u32 thread_index = va_arg (*args, u32); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 225 | u32 verbose = va_arg (*args, u32); |
| 226 | udp_connection_t *uc; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 227 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 228 | uc = udp_connection_get (uci, thread_index); |
| 229 | return format (s, "%U", format_udp_connection, uc, verbose); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 230 | } |
| 231 | |
| 232 | u8 * |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 233 | format_udp_half_open_session (u8 * s, va_list * args) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 234 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 235 | clib_warning ("BUG"); |
| 236 | return 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 237 | } |
| 238 | |
| 239 | u8 * |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 240 | format_udp_listener_session (u8 * s, va_list * args) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 241 | { |
| 242 | u32 tci = va_arg (*args, u32); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 243 | udp_connection_t *uc = udp_listener_get (tci); |
| 244 | return format (s, "%U", format_udp_connection, uc); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 245 | } |
| 246 | |
| 247 | u16 |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 248 | udp_send_mss (transport_connection_t * t) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 249 | { |
| 250 | /* TODO figure out MTU of output interface */ |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 251 | return 1460; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 252 | } |
| 253 | |
| 254 | u32 |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 255 | udp_send_space (transport_connection_t * t) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 256 | { |
| 257 | /* No constraint on TX window */ |
| 258 | return ~0; |
| 259 | } |
| 260 | |
| 261 | int |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 262 | udp_open_connection (transport_endpoint_t * rmt) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 263 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 264 | udp_main_t *um = vnet_get_udp_main (); |
| 265 | vlib_main_t *vm = vlib_get_main (); |
| 266 | u32 thread_index = vlib_get_thread_index (); |
| 267 | udp_connection_t *uc; |
| 268 | ip46_address_t lcl_addr; |
| 269 | u32 node_index; |
| 270 | u16 lcl_port; |
| 271 | |
| 272 | if (transport_alloc_local_endpoint (TRANSPORT_PROTO_UDP, rmt, &lcl_addr, |
| 273 | &lcl_port)) |
| 274 | return -1; |
| 275 | |
| 276 | while (udp_get_dst_port_info (um, lcl_port, rmt->is_ip4)) |
| 277 | { |
| 278 | lcl_port = transport_alloc_local_port (TRANSPORT_PROTO_UDP, &lcl_addr); |
| 279 | if (lcl_port < 1) |
| 280 | { |
| 281 | clib_warning ("Failed to allocate src port"); |
| 282 | return -1; |
| 283 | } |
| 284 | } |
| 285 | |
| 286 | node_index = rmt->is_ip4 ? udp4_input_node.index : udp6_input_node.index; |
| 287 | udp_register_dst_port (vm, lcl_port, node_index, 1 /* is_ipv4 */ ); |
| 288 | |
| 289 | uc = udp_connection_alloc (thread_index); |
| 290 | ip_copy (&uc->c_rmt_ip, &rmt->ip, rmt->is_ip4); |
| 291 | ip_copy (&uc->c_lcl_ip, &lcl_addr, rmt->is_ip4); |
| 292 | uc->c_rmt_port = rmt->port; |
| 293 | uc->c_lcl_port = clib_host_to_net_u16 (lcl_port); |
| 294 | uc->c_is_ip4 = rmt->is_ip4; |
| 295 | uc->c_proto = TRANSPORT_PROTO_UDP; |
| 296 | uc->c_fib_index = rmt->fib_index; |
| 297 | |
| 298 | return uc->c_c_index; |
| 299 | } |
| 300 | |
| 301 | transport_connection_t * |
| 302 | udp_half_open_session_get_transport (u32 conn_index) |
| 303 | { |
| 304 | udp_connection_t *uc; |
| 305 | uc = udp_connection_get (conn_index, vlib_get_thread_index ()); |
| 306 | return &uc->connection; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 307 | } |
| 308 | |
| 309 | /* *INDENT-OFF* */ |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 310 | const static transport_proto_vft_t udp_proto = { |
| 311 | .bind = udp_session_bind, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 312 | .open = udp_open_connection, |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 313 | .unbind = udp_session_unbind, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 314 | .push_header = udp_push_header, |
| 315 | .get_connection = udp_session_get, |
| 316 | .get_listener = udp_session_get_listener, |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 317 | .get_half_open = udp_half_open_session_get_transport, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 318 | .close = udp_session_close, |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 319 | .cleanup = udp_session_cleanup, |
| 320 | .send_mss = udp_send_mss, |
| 321 | .send_space = udp_send_space, |
| 322 | .format_connection = format_udp_session, |
| 323 | .format_half_open = format_udp_half_open_session, |
Florin Coras | 371ca50 | 2018-02-21 12:07:41 -0800 | [diff] [blame^] | 324 | .format_listener = format_udp_listener_session, |
| 325 | .tx_type = TRANSPORT_TX_DEQUEUE, |
| 326 | .service_type = TRANSPORT_SERVICE_VC, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 327 | }; |
| 328 | /* *INDENT-ON* */ |
| 329 | |
| 330 | static clib_error_t * |
| 331 | udp_init (vlib_main_t * vm) |
| 332 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 333 | udp_main_t *um = vnet_get_udp_main (); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 334 | ip_main_t *im = &ip_main; |
| 335 | vlib_thread_main_t *tm = vlib_get_thread_main (); |
| 336 | u32 num_threads; |
| 337 | clib_error_t *error = 0; |
| 338 | ip_protocol_info_t *pi; |
Florin Coras | 29ca16f | 2017-11-02 18:14:17 -0400 | [diff] [blame] | 339 | int i; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 340 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 341 | if ((error = vlib_call_init_function (vm, ip_main_init))) |
| 342 | return error; |
| 343 | if ((error = vlib_call_init_function (vm, ip4_lookup_init))) |
| 344 | return error; |
| 345 | if ((error = vlib_call_init_function (vm, ip6_lookup_init))) |
| 346 | return error; |
| 347 | |
| 348 | /* |
| 349 | * Registrations |
| 350 | */ |
| 351 | |
| 352 | /* IP registration */ |
| 353 | pi = ip_get_protocol_info (im, IP_PROTOCOL_UDP); |
| 354 | if (pi == 0) |
| 355 | return clib_error_return (0, "UDP protocol info AWOL"); |
| 356 | pi->format_header = format_udp_header; |
| 357 | pi->unformat_pg_edit = unformat_pg_udp_header; |
| 358 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 359 | /* Register as transport with URI */ |
Florin Coras | 561af9b | 2017-12-09 10:19:43 -0800 | [diff] [blame] | 360 | transport_register_protocol (TRANSPORT_PROTO_UDP, &udp_proto, |
| 361 | FIB_PROTOCOL_IP4, ip4_lookup_node.index); |
| 362 | transport_register_protocol (TRANSPORT_PROTO_UDP, &udp_proto, |
| 363 | FIB_PROTOCOL_IP6, ip6_lookup_node.index); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 364 | |
| 365 | /* |
| 366 | * Initialize data structures |
| 367 | */ |
| 368 | |
| 369 | num_threads = 1 /* main thread */ + tm->n_threads; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 370 | vec_validate (um->connections, num_threads - 1); |
| 371 | vec_validate (um->connection_peekers, num_threads - 1); |
| 372 | vec_validate (um->peekers_readers_locks, num_threads - 1); |
| 373 | vec_validate (um->peekers_write_locks, num_threads - 1); |
Florin Coras | 29ca16f | 2017-11-02 18:14:17 -0400 | [diff] [blame] | 374 | |
| 375 | if (num_threads > 1) |
| 376 | for (i = 0; i < num_threads; i++) |
| 377 | { |
| 378 | clib_spinlock_init (&um->peekers_readers_locks[i]); |
| 379 | clib_spinlock_init (&um->peekers_write_locks[i]); |
| 380 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 381 | return error; |
| 382 | } |
| 383 | |
| 384 | VLIB_INIT_FUNCTION (udp_init); |
| 385 | |
| 386 | /* |
| 387 | * fd.io coding-style-patch-verification: ON |
| 388 | * |
| 389 | * Local Variables: |
| 390 | * eval: (c-set-style "gnu") |
| 391 | * End: |
| 392 | */ |