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:
7 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 udp state machine, etc.
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>
25 udp_uri_main_t udp_uri_main;
28 udp_session_bind_ip4 (u32 session_index, transport_endpoint_t * lcl)
30 udp_uri_main_t *um = vnet_get_udp_main ();
31 udp_connection_t *listener;
33 pool_get (um->udp_listeners, listener);
34 memset (listener, 0, sizeof (udp_connection_t));
35 listener->c_lcl_port = clib_host_to_net_u16 (lcl->port);
36 listener->c_lcl_ip4.as_u32 = lcl->ip.ip4.as_u32;
37 listener->c_transport_proto = TRANSPORT_PROTO_UDP;
38 udp_register_dst_port (um->vlib_main, lcl->port, udp4_uri_input_node.index,
44 udp_session_bind_ip6 (u32 session_index, transport_endpoint_t * lcl)
46 udp_uri_main_t *um = vnet_get_udp_main ();
47 udp_connection_t *listener;
49 pool_get (um->udp_listeners, listener);
50 listener->c_lcl_port = clib_host_to_net_u16 (lcl->port);
51 clib_memcpy (&listener->c_lcl_ip6, &lcl->ip.ip6, sizeof (ip6_address_t));
52 listener->c_transport_proto = TRANSPORT_PROTO_UDP;
53 udp_register_dst_port (um->vlib_main, lcl->port,
54 udp4_uri_input_node.index, 0 /* is_ipv4 */ );
59 udp_session_unbind_ip4 (u32 listener_index)
61 vlib_main_t *vm = vlib_get_main ();
62 udp_connection_t *listener;
63 listener = udp_listener_get (listener_index);
65 /* deregister the udp_local mapping */
66 udp_unregister_dst_port (vm, listener->c_lcl_port, 1 /* is_ipv4 */ );
71 udp_session_unbind_ip6 (u32 listener_index)
73 vlib_main_t *vm = vlib_get_main ();
74 udp_connection_t *listener;
76 listener = udp_listener_get (listener_index);
78 /* deregister the udp_local mapping */
79 udp_unregister_dst_port (vm, listener->c_lcl_port, 0 /* is_ipv4 */ );
83 transport_connection_t *
84 udp_session_get_listener (u32 listener_index)
88 us = udp_listener_get (listener_index);
89 return &us->connection;
93 udp_push_header (transport_connection_t * tconn, vlib_buffer_t * b)
99 us = (udp_connection_t *) tconn;
105 data = vlib_buffer_get_current (b);
106 udp = (udp_header_t *) (data - sizeof (udp_header_t));
107 ip = (ip4_header_t *) ((u8 *) udp - sizeof (ip4_header_t));
109 /* Build packet header, swap rx key src + dst fields */
110 ip->src_address.as_u32 = us->c_lcl_ip4.as_u32;
111 ip->dst_address.as_u32 = us->c_rmt_ip4.as_u32;
112 ip->ip_version_and_header_length = 0x45;
114 ip->protocol = IP_PROTOCOL_UDP;
115 ip->length = clib_host_to_net_u16 (b->current_length + sizeof (*udp));
116 ip->checksum = ip4_header_checksum (ip);
118 udp->src_port = us->c_lcl_port;
119 udp->dst_port = us->c_rmt_port;
120 udp->length = clib_host_to_net_u16 (b->current_length);
123 b->current_length = sizeof (*ip) + sizeof (*udp);
124 return SESSION_QUEUE_NEXT_IP4_LOOKUP;
128 vlib_main_t *vm = vlib_get_main ();
133 data = vlib_buffer_get_current (b);
134 udp = (udp_header_t *) (data - sizeof (udp_header_t));
135 ip = (ip6_header_t *) ((u8 *) udp - sizeof (ip6_header_t));
137 /* Build packet header, swap rx key src + dst fields */
138 clib_memcpy (&ip->src_address, &us->c_lcl_ip6, sizeof (ip6_address_t));
139 clib_memcpy (&ip->dst_address, &us->c_rmt_ip6, sizeof (ip6_address_t));
141 ip->ip_version_traffic_class_and_flow_label =
142 clib_host_to_net_u32 (0x6 << 28);
144 ip->hop_limit = 0xff;
145 ip->protocol = IP_PROTOCOL_UDP;
147 payload_length = vlib_buffer_length_in_chain (vm, b);
148 payload_length -= sizeof (*ip);
150 ip->payload_length = clib_host_to_net_u16 (payload_length);
152 udp->checksum = ip6_tcp_udp_icmp_compute_checksum (vm, b, ip, &bogus);
155 udp->src_port = us->c_lcl_port;
156 udp->dst_port = us->c_rmt_port;
157 udp->length = clib_host_to_net_u16 (b->current_length);
160 b->current_length = sizeof (*ip) + sizeof (*udp);
162 return SESSION_QUEUE_NEXT_IP6_LOOKUP;
166 transport_connection_t *
167 udp_session_get (u32 connection_index, u32 my_thread_index)
169 udp_uri_main_t *um = vnet_get_udp_main ();
171 udp_connection_t *us;
173 pool_elt_at_index (um->udp_sessions[my_thread_index], connection_index);
174 return &us->connection;
178 udp_session_close (u32 connection_index, u32 my_thread_index)
180 udp_uri_main_t *um = vnet_get_udp_main ();
181 pool_put_index (um->udp_sessions[my_thread_index], connection_index);
185 format_udp_session_ip4 (u8 * s, va_list * args)
187 u32 uci = va_arg (*args, u32);
188 u32 thread_index = va_arg (*args, u32);
189 udp_connection_t *u4;
191 u4 = udp_connection_get (uci, thread_index);
193 s = format (s, "[%s] %U:%d->%U:%d", "udp", format_ip4_address,
194 &u4->c_lcl_ip4, clib_net_to_host_u16 (u4->c_lcl_port),
195 format_ip4_address, &u4->c_rmt_ip4,
196 clib_net_to_host_u16 (u4->c_rmt_port));
201 format_udp_session_ip6 (u8 * s, va_list * args)
203 u32 uci = va_arg (*args, u32);
204 u32 thread_index = va_arg (*args, u32);
205 udp_connection_t *tc = udp_connection_get (uci, thread_index);
206 s = format (s, "[%s] %U:%d->%U:%d", "udp", format_ip6_address,
207 &tc->c_lcl_ip6, clib_net_to_host_u16 (tc->c_lcl_port),
208 format_ip6_address, &tc->c_rmt_ip6,
209 clib_net_to_host_u16 (tc->c_rmt_port));
214 format_udp_listener_session_ip4 (u8 * s, va_list * args)
216 u32 tci = va_arg (*args, u32);
217 udp_connection_t *tc = udp_listener_get (tci);
218 s = format (s, "[%s] %U:%d->%U:%d", "udp", format_ip4_address,
219 &tc->c_lcl_ip4, clib_net_to_host_u16 (tc->c_lcl_port),
220 format_ip4_address, &tc->c_rmt_ip4,
221 clib_net_to_host_u16 (tc->c_rmt_port));
226 format_udp_listener_session_ip6 (u8 * s, va_list * args)
228 u32 tci = va_arg (*args, u32);
229 udp_connection_t *tc = udp_listener_get (tci);
230 s = format (s, "[%s] %U:%d->%U:%d", "udp", format_ip6_address,
231 &tc->c_lcl_ip6, clib_net_to_host_u16 (tc->c_lcl_port),
232 format_ip6_address, &tc->c_rmt_ip6,
233 clib_net_to_host_u16 (tc->c_rmt_port));
238 udp_send_mss_uri (transport_connection_t * t)
240 /* TODO figure out MTU of output interface */
245 udp_send_space_uri (transport_connection_t * t)
247 /* No constraint on TX window */
252 udp_open_connection (transport_endpoint_t * tep)
254 clib_warning ("Not implemented");
259 const static transport_proto_vft_t udp4_proto = {
260 .bind = udp_session_bind_ip4,
261 .open = udp_open_connection,
262 .unbind = udp_session_unbind_ip4,
263 .push_header = udp_push_header,
264 .get_connection = udp_session_get,
265 .get_listener = udp_session_get_listener,
266 .close = udp_session_close,
267 .send_mss = udp_send_mss_uri,
268 .send_space = udp_send_space_uri,
269 .format_connection = format_udp_session_ip4,
270 .format_listener = format_udp_listener_session_ip4
273 const static transport_proto_vft_t udp6_proto = {
274 .bind = udp_session_bind_ip6,
275 .open = udp_open_connection,
276 .unbind = udp_session_unbind_ip6,
277 .push_header = udp_push_header,
278 .get_connection = udp_session_get,
279 .get_listener = udp_session_get_listener,
280 .close = udp_session_close,
281 .send_mss = udp_send_mss_uri,
282 .send_space = udp_send_space_uri,
283 .format_connection = format_udp_session_ip6,
284 .format_listener = format_udp_listener_session_ip6
288 static clib_error_t *
289 udp_init (vlib_main_t * vm)
291 udp_uri_main_t *um = vnet_get_udp_main ();
292 ip_main_t *im = &ip_main;
293 vlib_thread_main_t *tm = vlib_get_thread_main ();
295 clib_error_t *error = 0;
296 ip_protocol_info_t *pi;
299 um->vnet_main = vnet_get_main ();
301 if ((error = vlib_call_init_function (vm, ip_main_init)))
303 if ((error = vlib_call_init_function (vm, ip4_lookup_init)))
305 if ((error = vlib_call_init_function (vm, ip6_lookup_init)))
312 /* IP registration */
313 pi = ip_get_protocol_info (im, IP_PROTOCOL_UDP);
315 return clib_error_return (0, "UDP protocol info AWOL");
316 pi->format_header = format_udp_header;
317 pi->unformat_pg_edit = unformat_pg_udp_header;
320 /* Register as transport with URI */
321 session_register_transport (TRANSPORT_PROTO_UDP, 1, &udp4_proto);
322 session_register_transport (TRANSPORT_PROTO_UDP, 0, &udp6_proto);
325 * Initialize data structures
328 num_threads = 1 /* main thread */ + tm->n_threads;
329 vec_validate (um->udp_sessions, num_threads - 1);
334 VLIB_INIT_FUNCTION (udp_init);
337 * fd.io coding-style-patch-verification: ON
340 * eval: (c-set-style "gnu")