SCTP: calculate RTO / RTT and RTTVAR as per RFC
[vpp.git] / src / vnet / sctp / sctp_output.c
1 /*
2  * Copyright (c) 2017 SUSE LLC.
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 #include <vnet/sctp/sctp.h>
16 #include <vnet/sctp/sctp_debug.h>
17 #include <vppinfra/random.h>
18 #include <openssl/hmac.h>
19
20 vlib_node_registration_t sctp4_output_node;
21 vlib_node_registration_t sctp6_output_node;
22
23 typedef enum _sctp_output_next
24 {
25   SCTP_OUTPUT_NEXT_DROP,
26   SCTP_OUTPUT_NEXT_IP_LOOKUP,
27   SCTP_OUTPUT_N_NEXT
28 } sctp_output_next_t;
29
30 #define foreach_sctp4_output_next               \
31   _ (DROP, "error-drop")                        \
32   _ (IP_LOOKUP, "ip4-lookup")
33
34 #define foreach_sctp6_output_next               \
35   _ (DROP, "error-drop")                        \
36   _ (IP_LOOKUP, "ip6-lookup")
37
38 static char *sctp_error_strings[] = {
39 #define sctp_error(n,s) s,
40 #include <vnet/sctp/sctp_error.def>
41 #undef sctp_error
42 };
43
44 typedef struct
45 {
46   sctp_header_t sctp_header;
47   sctp_connection_t sctp_connection;
48 } sctp_tx_trace_t;
49
50 /**
51  * Flush tx frame populated by retransmits and timer pops
52  */
53 void
54 sctp_flush_frame_to_output (vlib_main_t * vm, u8 thread_index, u8 is_ip4)
55 {
56   if (sctp_main.tx_frames[!is_ip4][thread_index])
57     {
58       u32 next_index;
59       next_index = is_ip4 ? sctp4_output_node.index : sctp6_output_node.index;
60       vlib_put_frame_to_node (vm, next_index,
61                               sctp_main.tx_frames[!is_ip4][thread_index]);
62       sctp_main.tx_frames[!is_ip4][thread_index] = 0;
63     }
64 }
65
66 /**
67  * Flush ip lookup tx frames populated by timer pops
68  */
69 always_inline void
70 sctp_flush_frame_to_ip_lookup (vlib_main_t * vm, u8 thread_index, u8 is_ip4)
71 {
72   if (sctp_main.ip_lookup_tx_frames[!is_ip4][thread_index])
73     {
74       u32 next_index;
75       next_index = is_ip4 ? ip4_lookup_node.index : ip6_lookup_node.index;
76       vlib_put_frame_to_node (vm, next_index,
77                               sctp_main.ip_lookup_tx_frames[!is_ip4]
78                               [thread_index]);
79       sctp_main.ip_lookup_tx_frames[!is_ip4][thread_index] = 0;
80     }
81 }
82
83 /**
84  * Flush v4 and v6 sctp and ip-lookup tx frames for thread index
85  */
86 void
87 sctp_flush_frames_to_output (u8 thread_index)
88 {
89   vlib_main_t *vm = vlib_get_main ();
90   sctp_flush_frame_to_output (vm, thread_index, 1);
91   sctp_flush_frame_to_output (vm, thread_index, 0);
92   sctp_flush_frame_to_ip_lookup (vm, thread_index, 1);
93   sctp_flush_frame_to_ip_lookup (vm, thread_index, 0);
94 }
95
96 u32
97 ip4_sctp_compute_checksum (vlib_main_t * vm, vlib_buffer_t * p0,
98                            ip4_header_t * ip0)
99 {
100   ip_csum_t checksum;
101   u32 ip_header_length, payload_length_host_byte_order;
102   u32 n_this_buffer, n_bytes_left, n_ip_bytes_this_buffer;
103   void *data_this_buffer;
104
105   /* Initialize checksum with ip header. */
106   ip_header_length = ip4_header_bytes (ip0);
107   payload_length_host_byte_order =
108     clib_net_to_host_u16 (ip0->length) - ip_header_length;
109   checksum =
110     clib_host_to_net_u32 (payload_length_host_byte_order +
111                           (ip0->protocol << 16));
112
113   if (BITS (uword) == 32)
114     {
115       checksum =
116         ip_csum_with_carry (checksum,
117                             clib_mem_unaligned (&ip0->src_address, u32));
118       checksum =
119         ip_csum_with_carry (checksum,
120                             clib_mem_unaligned (&ip0->dst_address, u32));
121     }
122   else
123     checksum =
124       ip_csum_with_carry (checksum,
125                           clib_mem_unaligned (&ip0->src_address, u64));
126
127   n_bytes_left = n_this_buffer = payload_length_host_byte_order;
128   data_this_buffer = (void *) ip0 + ip_header_length;
129   n_ip_bytes_this_buffer =
130     p0->current_length - (((u8 *) ip0 - p0->data) - p0->current_data);
131   if (n_this_buffer + ip_header_length > n_ip_bytes_this_buffer)
132     {
133       n_this_buffer = n_ip_bytes_this_buffer > ip_header_length ?
134         n_ip_bytes_this_buffer - ip_header_length : 0;
135     }
136   while (1)
137     {
138       checksum =
139         ip_incremental_checksum (checksum, data_this_buffer, n_this_buffer);
140       n_bytes_left -= n_this_buffer;
141       if (n_bytes_left == 0)
142         break;
143
144       ASSERT (p0->flags & VLIB_BUFFER_NEXT_PRESENT);
145       p0 = vlib_get_buffer (vm, p0->next_buffer);
146       data_this_buffer = vlib_buffer_get_current (p0);
147       n_this_buffer = p0->current_length;
148     }
149
150   return checksum;
151 }
152
153 u32
154 ip6_sctp_compute_checksum (vlib_main_t * vm, vlib_buffer_t * p0,
155                            ip6_header_t * ip0, int *bogus_lengthp)
156 {
157   ip_csum_t checksum;
158   u16 payload_length_host_byte_order;
159   u32 i, n_this_buffer, n_bytes_left;
160   u32 headers_size = sizeof (ip0[0]);
161   void *data_this_buffer;
162
163   ASSERT (bogus_lengthp);
164   *bogus_lengthp = 0;
165
166   /* Initialize checksum with ip header. */
167   checksum = ip0->payload_length + clib_host_to_net_u16 (ip0->protocol);
168   payload_length_host_byte_order = clib_net_to_host_u16 (ip0->payload_length);
169   data_this_buffer = (void *) (ip0 + 1);
170
171   for (i = 0; i < ARRAY_LEN (ip0->src_address.as_uword); i++)
172     {
173       checksum = ip_csum_with_carry (checksum,
174                                      clib_mem_unaligned (&ip0->
175                                                          src_address.as_uword
176                                                          [i], uword));
177       checksum =
178         ip_csum_with_carry (checksum,
179                             clib_mem_unaligned (&ip0->dst_address.as_uword[i],
180                                                 uword));
181     }
182
183   /* some icmp packets may come with a "router alert" hop-by-hop extension header (e.g., mldv2 packets)
184    * or UDP-Ping packets */
185   if (PREDICT_FALSE (ip0->protocol == IP_PROTOCOL_IP6_HOP_BY_HOP_OPTIONS))
186     {
187       u32 skip_bytes;
188       ip6_hop_by_hop_ext_t *ext_hdr =
189         (ip6_hop_by_hop_ext_t *) data_this_buffer;
190
191       /* validate really icmp6 next */
192       ASSERT ((ext_hdr->next_hdr == IP_PROTOCOL_SCTP));
193
194       skip_bytes = 8 * (1 + ext_hdr->n_data_u64s);
195       data_this_buffer = (void *) ((u8 *) data_this_buffer + skip_bytes);
196
197       payload_length_host_byte_order -= skip_bytes;
198       headers_size += skip_bytes;
199     }
200
201   n_bytes_left = n_this_buffer = payload_length_host_byte_order;
202   if (p0 && n_this_buffer + headers_size > p0->current_length)
203     n_this_buffer =
204       p0->current_length >
205       headers_size ? p0->current_length - headers_size : 0;
206   while (1)
207     {
208       checksum =
209         ip_incremental_checksum (checksum, data_this_buffer, n_this_buffer);
210       n_bytes_left -= n_this_buffer;
211       if (n_bytes_left == 0)
212         break;
213
214       if (!(p0->flags & VLIB_BUFFER_NEXT_PRESENT))
215         {
216           *bogus_lengthp = 1;
217           return 0xfefe;
218         }
219       p0 = vlib_get_buffer (vm, p0->next_buffer);
220       data_this_buffer = vlib_buffer_get_current (p0);
221       n_this_buffer = p0->current_length;
222     }
223
224   return checksum;
225 }
226
227 void
228 sctp_push_ip_hdr (sctp_main_t * tm, sctp_sub_connection_t * sctp_sub_conn,
229                   vlib_buffer_t * b)
230 {
231   sctp_header_t *th = vlib_buffer_get_current (b);
232   vlib_main_t *vm = vlib_get_main ();
233   if (sctp_sub_conn->c_is_ip4)
234     {
235       ip4_header_t *ih;
236       ih = vlib_buffer_push_ip4 (vm, b, &sctp_sub_conn->c_lcl_ip4,
237                                  &sctp_sub_conn->c_rmt_ip4, IP_PROTOCOL_SCTP,
238                                  1);
239       th->checksum = ip4_sctp_compute_checksum (vm, b, ih);
240     }
241   else
242     {
243       ip6_header_t *ih;
244       int bogus = ~0;
245
246       ih = vlib_buffer_push_ip6 (vm, b, &sctp_sub_conn->c_lcl_ip6,
247                                  &sctp_sub_conn->c_rmt_ip6, IP_PROTOCOL_SCTP);
248       th->checksum = ip6_sctp_compute_checksum (vm, b, ih, &bogus);
249       ASSERT (!bogus);
250     }
251 }
252
253 always_inline void *
254 sctp_reuse_buffer (vlib_main_t * vm, vlib_buffer_t * b)
255 {
256   if (b->flags & VLIB_BUFFER_NEXT_PRESENT)
257     vlib_buffer_free_one (vm, b->next_buffer);
258   /* Zero all flags but free list index and trace flag */
259   b->flags &= VLIB_BUFFER_NEXT_PRESENT - 1;
260   b->current_data = 0;
261   b->current_length = 0;
262   b->total_length_not_including_first_buffer = 0;
263   vnet_buffer (b)->sctp.flags = 0;
264
265   /* Leave enough space for headers */
266   return vlib_buffer_make_headroom (b, MAX_HDRS_LEN);
267 }
268
269 always_inline void *
270 sctp_init_buffer (vlib_main_t * vm, vlib_buffer_t * b)
271 {
272   ASSERT ((b->flags & VLIB_BUFFER_NEXT_PRESENT) == 0);
273   b->flags &= VLIB_BUFFER_FREE_LIST_INDEX_MASK;
274   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
275   b->total_length_not_including_first_buffer = 0;
276   vnet_buffer (b)->sctp.flags = 0;
277   VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b);
278   /* Leave enough space for headers */
279   return vlib_buffer_make_headroom (b, MAX_HDRS_LEN);
280 }
281
282 always_inline int
283 sctp_alloc_tx_buffers (sctp_main_t * tm, u8 thread_index, u32 n_free_buffers)
284 {
285   vlib_main_t *vm = vlib_get_main ();
286   u32 current_length = vec_len (tm->tx_buffers[thread_index]);
287   u32 n_allocated;
288
289   vec_validate (tm->tx_buffers[thread_index],
290                 current_length + n_free_buffers - 1);
291   n_allocated =
292     vlib_buffer_alloc (vm, &tm->tx_buffers[thread_index][current_length],
293                        n_free_buffers);
294   _vec_len (tm->tx_buffers[thread_index]) = current_length + n_allocated;
295   /* buffer shortage, report failure */
296   if (vec_len (tm->tx_buffers[thread_index]) == 0)
297     {
298       clib_warning ("out of buffers");
299       return -1;
300     }
301   return 0;
302 }
303
304 always_inline int
305 sctp_get_free_buffer_index (sctp_main_t * tm, u32 * bidx)
306 {
307   u32 *my_tx_buffers;
308   u32 thread_index = vlib_get_thread_index ();
309   if (PREDICT_FALSE (vec_len (tm->tx_buffers[thread_index]) == 0))
310     {
311       if (sctp_alloc_tx_buffers (tm, thread_index, VLIB_FRAME_SIZE))
312         return -1;
313     }
314   my_tx_buffers = tm->tx_buffers[thread_index];
315   *bidx = my_tx_buffers[vec_len (my_tx_buffers) - 1];
316   _vec_len (my_tx_buffers) -= 1;
317   return 0;
318 }
319
320 always_inline void
321 sctp_enqueue_to_output_i (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
322                           u8 is_ip4, u8 flush)
323 {
324   sctp_main_t *tm = vnet_get_sctp_main ();
325   u32 thread_index = vlib_get_thread_index ();
326   u32 *to_next, next_index;
327   vlib_frame_t *f;
328
329   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
330   b->error = 0;
331
332   /* Decide where to send the packet */
333   next_index = is_ip4 ? sctp4_output_node.index : sctp6_output_node.index;
334   sctp_trajectory_add_start (b, 2);
335
336   /* Get frame to v4/6 output node */
337   f = tm->tx_frames[!is_ip4][thread_index];
338   if (!f)
339     {
340       f = vlib_get_frame_to_node (vm, next_index);
341       ASSERT (f);
342       tm->tx_frames[!is_ip4][thread_index] = f;
343     }
344   to_next = vlib_frame_vector_args (f);
345   to_next[f->n_vectors] = bi;
346   f->n_vectors += 1;
347   if (flush || f->n_vectors == VLIB_FRAME_SIZE)
348     {
349       vlib_put_frame_to_node (vm, next_index, f);
350       tm->tx_frames[!is_ip4][thread_index] = 0;
351     }
352 }
353
354 always_inline void
355 sctp_enqueue_to_output_now (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
356                             u8 is_ip4)
357 {
358   sctp_enqueue_to_output_i (vm, b, bi, is_ip4, 1);
359 }
360
361 always_inline void
362 sctp_enqueue_to_ip_lookup_i (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
363                              u8 is_ip4, u8 flush)
364 {
365   sctp_main_t *tm = vnet_get_sctp_main ();
366   u32 thread_index = vlib_get_thread_index ();
367   u32 *to_next, next_index;
368   vlib_frame_t *f;
369
370   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
371   b->error = 0;
372
373   /* Default FIB for now */
374   vnet_buffer (b)->sw_if_index[VLIB_TX] = 0;
375
376   /* Send to IP lookup */
377   next_index = is_ip4 ? ip4_lookup_node.index : ip6_lookup_node.index;
378   if (VLIB_BUFFER_TRACE_TRAJECTORY > 0)
379     {
380       b->pre_data[0] = 2;
381       b->pre_data[1] = next_index;
382     }
383
384   f = tm->ip_lookup_tx_frames[!is_ip4][thread_index];
385   if (!f)
386     {
387       f = vlib_get_frame_to_node (vm, next_index);
388       ASSERT (f);
389       tm->ip_lookup_tx_frames[!is_ip4][thread_index] = f;
390     }
391
392   to_next = vlib_frame_vector_args (f);
393   to_next[f->n_vectors] = bi;
394   f->n_vectors += 1;
395   if (flush || f->n_vectors == VLIB_FRAME_SIZE)
396     {
397       vlib_put_frame_to_node (vm, next_index, f);
398       tm->ip_lookup_tx_frames[!is_ip4][thread_index] = 0;
399     }
400 }
401
402 always_inline void
403 sctp_enqueue_to_ip_lookup (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
404                            u8 is_ip4)
405 {
406   sctp_enqueue_to_ip_lookup_i (vm, b, bi, is_ip4, 0);
407 }
408
409 always_inline void
410 sctp_enqueue_to_ip_lookup_now (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
411                                u8 is_ip4)
412 {
413   sctp_enqueue_to_ip_lookup_i (vm, b, bi, is_ip4, 1);
414 }
415
416 /**
417  * Convert buffer to INIT
418  */
419 void
420 sctp_prepare_init_chunk (sctp_connection_t * sctp_conn, vlib_buffer_t * b)
421 {
422   u32 random_seed = random_default_seed ();
423   u16 alloc_bytes = sizeof (sctp_init_chunk_t);
424   sctp_sub_connection_t *sub_conn =
425     &sctp_conn->sub_conn[sctp_pick_conn_idx_on_chunk (INIT)];
426
427   sctp_ipv4_addr_param_t *ip4_param = 0;
428   sctp_ipv6_addr_param_t *ip6_param = 0;
429
430   if (sub_conn->c_is_ip4)
431     alloc_bytes += sizeof (sctp_ipv4_addr_param_t);
432   else
433     alloc_bytes += sizeof (sctp_ipv6_addr_param_t);
434
435   /* As per RFC 4960 the chunk_length value does NOT contemplate
436    * the size of the first header (see sctp_header_t) and any padding
437    */
438   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
439
440   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
441
442   sctp_init_chunk_t *init_chunk = vlib_buffer_push_uninit (b, alloc_bytes);
443
444   u16 pointer_offset = sizeof (init_chunk);
445   if (sub_conn->c_is_ip4)
446     {
447       ip4_param = (sctp_ipv4_addr_param_t *) init_chunk + pointer_offset;
448       ip4_param->address.as_u32 = sub_conn->c_lcl_ip.ip4.as_u32;
449
450       pointer_offset += sizeof (sctp_ipv4_addr_param_t);
451     }
452   else
453     {
454       ip6_param = (sctp_ipv6_addr_param_t *) init_chunk + pointer_offset;
455       ip6_param->address.as_u64[0] = sub_conn->c_lcl_ip.ip6.as_u64[0];
456       ip6_param->address.as_u64[1] = sub_conn->c_lcl_ip.ip6.as_u64[1];
457
458       pointer_offset += sizeof (sctp_ipv6_addr_param_t);
459     }
460
461   init_chunk->sctp_hdr.src_port = sub_conn->c_lcl_port; /* No need of host_to_net conversion, already in net-byte order */
462   init_chunk->sctp_hdr.dst_port = sub_conn->c_rmt_port; /* No need of host_to_net conversion, already in net-byte order */
463   init_chunk->sctp_hdr.checksum = 0;
464   /* The sender of an INIT must set the VERIFICATION_TAG to 0 as per RFC 4960 Section 8.5.1 */
465   init_chunk->sctp_hdr.verification_tag = 0x0;
466
467   vnet_sctp_set_chunk_type (&init_chunk->chunk_hdr, INIT);
468   vnet_sctp_set_chunk_length (&init_chunk->chunk_hdr, chunk_len);
469   vnet_sctp_common_hdr_params_host_to_net (&init_chunk->chunk_hdr);
470
471   init_chunk->a_rwnd = clib_host_to_net_u32 (DEFAULT_A_RWND);
472   init_chunk->initiate_tag = clib_host_to_net_u32 (random_u32 (&random_seed));
473   init_chunk->inboud_streams_count =
474     clib_host_to_net_u16 (INBOUND_STREAMS_COUNT);
475   init_chunk->outbound_streams_count =
476     clib_host_to_net_u16 (OUTBOUND_STREAMS_COUNT);
477
478   init_chunk->initial_tsn =
479     clib_host_to_net_u32 (sctp_conn->local_initial_tsn);
480   SCTP_CONN_TRACKING_DBG ("sctp_conn->local_initial_tsn = %u",
481                           sctp_conn->local_initial_tsn);
482
483   sctp_conn->local_tag = init_chunk->initiate_tag;
484
485   vnet_buffer (b)->sctp.connection_index = sub_conn->c_c_index;
486
487   SCTP_DBG_STATE_MACHINE ("CONN_INDEX = %u, CURR_CONN_STATE = %u (%s), "
488                           "CHUNK_TYPE = %s, "
489                           "SRC_PORT = %u, DST_PORT = %u",
490                           sub_conn->connection.c_index,
491                           sctp_conn->state,
492                           sctp_state_to_string (sctp_conn->state),
493                           sctp_chunk_to_string (INIT),
494                           init_chunk->sctp_hdr.src_port,
495                           init_chunk->sctp_hdr.dst_port);
496 }
497
498 void
499 sctp_compute_mac (sctp_connection_t * sctp_conn,
500                   sctp_state_cookie_param_t * state_cookie)
501 {
502 #if OPENSSL_VERSION_NUMBER >= 0x10100000L
503   HMAC_CTX *ctx;
504 #else
505   HMAC_CTX ctx;
506   const EVP_MD *md = EVP_sha1 ();
507 #endif
508   unsigned int len = 0;
509
510 #if OPENSSL_VERSION_NUMBER >= 0x10100000L
511   ctx = HMAC_CTX_new ();
512   HMAC_Init_ex (&ctx, &state_cookie->creation_time,
513                 sizeof (state_cookie->creation_time), md, NULL);
514   HMAC_Update (ctx, (const unsigned char *) &sctp_conn, sizeof (sctp_conn));
515   HMAC_Final (ctx, state_cookie->mac, &len);
516 #else
517   HMAC_CTX_init (&ctx);
518   HMAC_Init_ex (&ctx, &state_cookie->creation_time,
519                 sizeof (state_cookie->creation_time), md, NULL);
520
521   HMAC_Update (&ctx, (const unsigned char *) &sctp_conn, sizeof (sctp_conn));
522   HMAC_Final (&ctx, state_cookie->mac, &len);
523   HMAC_CTX_cleanup (&ctx);
524 #endif
525
526   ENDIANESS_SWAP (state_cookie->mac);
527 }
528
529 void
530 sctp_prepare_cookie_ack_chunk (sctp_connection_t * sctp_conn,
531                                vlib_buffer_t * b)
532 {
533   vlib_main_t *vm = vlib_get_main ();
534   u8 idx = sctp_pick_conn_idx_on_chunk (COOKIE_ACK);
535
536   sctp_reuse_buffer (vm, b);
537
538   u16 alloc_bytes = sizeof (sctp_cookie_ack_chunk_t);
539
540   /* As per RFC 4960 the chunk_length value does NOT contemplate
541    * the size of the first header (see sctp_header_t) and any padding
542    */
543   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
544
545   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
546
547   sctp_cookie_ack_chunk_t *cookie_ack_chunk =
548     vlib_buffer_push_uninit (b, alloc_bytes);
549
550   cookie_ack_chunk->sctp_hdr.checksum = 0;
551   cookie_ack_chunk->sctp_hdr.src_port =
552     sctp_conn->sub_conn[idx].connection.lcl_port;
553   cookie_ack_chunk->sctp_hdr.dst_port =
554     sctp_conn->sub_conn[idx].connection.rmt_port;
555   cookie_ack_chunk->sctp_hdr.verification_tag = sctp_conn->remote_tag;
556   vnet_sctp_set_chunk_type (&cookie_ack_chunk->chunk_hdr, COOKIE_ACK);
557   vnet_sctp_set_chunk_length (&cookie_ack_chunk->chunk_hdr, chunk_len);
558
559   /* Measure RTT with this */
560   sctp_conn->sub_conn[idx].rtt_ts = sctp_time_now ();
561
562   vnet_buffer (b)->sctp.connection_index =
563     sctp_conn->sub_conn[idx].connection.c_index;
564 }
565
566 void
567 sctp_prepare_cookie_echo_chunk (sctp_connection_t * sctp_conn,
568                                 vlib_buffer_t * b,
569                                 sctp_state_cookie_param_t * sc)
570 {
571   vlib_main_t *vm = vlib_get_main ();
572   u8 idx = sctp_pick_conn_idx_on_chunk (COOKIE_ECHO);
573
574   sctp_reuse_buffer (vm, b);
575
576   /* The minimum size of the message is given by the sctp_init_ack_chunk_t */
577   u16 alloc_bytes = sizeof (sctp_cookie_echo_chunk_t);
578   /* As per RFC 4960 the chunk_length value does NOT contemplate
579    * the size of the first header (see sctp_header_t) and any padding
580    */
581   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
582   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
583   sctp_cookie_echo_chunk_t *cookie_echo_chunk =
584     vlib_buffer_push_uninit (b, alloc_bytes);
585   cookie_echo_chunk->sctp_hdr.checksum = 0;
586   cookie_echo_chunk->sctp_hdr.src_port =
587     sctp_conn->sub_conn[idx].connection.lcl_port;
588   cookie_echo_chunk->sctp_hdr.dst_port =
589     sctp_conn->sub_conn[idx].connection.rmt_port;
590   cookie_echo_chunk->sctp_hdr.verification_tag = sctp_conn->remote_tag;
591   vnet_sctp_set_chunk_type (&cookie_echo_chunk->chunk_hdr, COOKIE_ECHO);
592   vnet_sctp_set_chunk_length (&cookie_echo_chunk->chunk_hdr, chunk_len);
593   clib_memcpy (&(cookie_echo_chunk->cookie), sc,
594                sizeof (sctp_state_cookie_param_t));
595
596   /* Measure RTT with this */
597   sctp_conn->sub_conn[idx].rtt_ts = sctp_time_now ();
598
599   vnet_buffer (b)->sctp.connection_index =
600     sctp_conn->sub_conn[idx].connection.c_index;
601 }
602
603 /**
604  * Convert buffer to INIT-ACK
605  */
606 void
607 sctp_prepare_initack_chunk (sctp_connection_t * sctp_conn, vlib_buffer_t * b,
608                             ip4_address_t * ip4_addr,
609                             ip6_address_t * ip6_addr)
610 {
611   vlib_main_t *vm = vlib_get_main ();
612   sctp_ipv4_addr_param_t *ip4_param = 0;
613   sctp_ipv6_addr_param_t *ip6_param = 0;
614   u8 idx = sctp_pick_conn_idx_on_chunk (INIT_ACK);
615   u32 random_seed = random_default_seed ();
616
617   sctp_reuse_buffer (vm, b);
618
619   /* The minimum size of the message is given by the sctp_init_ack_chunk_t */
620   u16 alloc_bytes =
621     sizeof (sctp_init_ack_chunk_t) + sizeof (sctp_state_cookie_param_t);
622
623   if (PREDICT_TRUE (ip4_addr != NULL))
624     {
625       /* Create room for variable-length fields in the INIT_ACK chunk */
626       alloc_bytes += SCTP_IPV4_ADDRESS_TYPE_LENGTH;
627     }
628   if (PREDICT_TRUE (ip6_addr != NULL))
629     {
630       /* Create room for variable-length fields in the INIT_ACK chunk */
631       alloc_bytes += SCTP_IPV6_ADDRESS_TYPE_LENGTH;
632     }
633
634   if (sctp_conn->sub_conn[idx].connection.is_ip4)
635     alloc_bytes += sizeof (sctp_ipv4_addr_param_t);
636   else
637     alloc_bytes += sizeof (sctp_ipv6_addr_param_t);
638
639   /* As per RFC 4960 the chunk_length value does NOT contemplate
640    * the size of the first header (see sctp_header_t) and any padding
641    */
642   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
643
644   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
645
646   sctp_init_ack_chunk_t *init_ack_chunk =
647     vlib_buffer_push_uninit (b, alloc_bytes);
648
649   u16 pointer_offset = sizeof (sctp_init_ack_chunk_t);
650
651   /* Create State Cookie parameter */
652   sctp_state_cookie_param_t *state_cookie_param =
653     (sctp_state_cookie_param_t *) ((char *) init_ack_chunk + pointer_offset);
654
655   state_cookie_param->param_hdr.type =
656     clib_host_to_net_u16 (SCTP_STATE_COOKIE_TYPE);
657   state_cookie_param->param_hdr.length =
658     clib_host_to_net_u16 (sizeof (sctp_state_cookie_param_t));
659   state_cookie_param->creation_time = clib_host_to_net_u32 (sctp_time_now ());
660   state_cookie_param->cookie_lifespan =
661     clib_host_to_net_u32 (SCTP_VALID_COOKIE_LIFE);
662
663   sctp_compute_mac (sctp_conn, state_cookie_param);
664
665   pointer_offset += sizeof (sctp_state_cookie_param_t);
666
667   if (PREDICT_TRUE (ip4_addr != NULL))
668     {
669       sctp_ipv4_addr_param_t *ipv4_addr =
670         (sctp_ipv4_addr_param_t *) init_ack_chunk + pointer_offset;
671
672       ipv4_addr->param_hdr.type =
673         clib_host_to_net_u16 (SCTP_IPV4_ADDRESS_TYPE);
674       ipv4_addr->param_hdr.length =
675         clib_host_to_net_u16 (SCTP_IPV4_ADDRESS_TYPE_LENGTH);
676       ipv4_addr->address.as_u32 = ip4_addr->as_u32;
677
678       pointer_offset += SCTP_IPV4_ADDRESS_TYPE_LENGTH;
679     }
680   if (PREDICT_TRUE (ip6_addr != NULL))
681     {
682       sctp_ipv6_addr_param_t *ipv6_addr =
683         (sctp_ipv6_addr_param_t *) init_ack_chunk +
684         sizeof (sctp_init_chunk_t) + pointer_offset;
685
686       ipv6_addr->param_hdr.type =
687         clib_host_to_net_u16 (SCTP_IPV6_ADDRESS_TYPE);
688       ipv6_addr->param_hdr.length =
689         clib_host_to_net_u16 (SCTP_IPV6_ADDRESS_TYPE_LENGTH);
690       ipv6_addr->address.as_u64[0] = ip6_addr->as_u64[0];
691       ipv6_addr->address.as_u64[1] = ip6_addr->as_u64[1];
692
693       pointer_offset += SCTP_IPV6_ADDRESS_TYPE_LENGTH;
694     }
695
696   if (sctp_conn->sub_conn[idx].connection.is_ip4)
697     {
698       ip4_param = (sctp_ipv4_addr_param_t *) init_ack_chunk + pointer_offset;
699       ip4_param->address.as_u32 =
700         sctp_conn->sub_conn[idx].connection.lcl_ip.ip4.as_u32;
701
702       pointer_offset += sizeof (sctp_ipv4_addr_param_t);
703     }
704   else
705     {
706       ip6_param = (sctp_ipv6_addr_param_t *) init_ack_chunk + pointer_offset;
707       ip6_param->address.as_u64[0] =
708         sctp_conn->sub_conn[idx].connection.lcl_ip.ip6.as_u64[0];
709       ip6_param->address.as_u64[1] =
710         sctp_conn->sub_conn[idx].connection.lcl_ip.ip6.as_u64[1];
711
712       pointer_offset += sizeof (sctp_ipv6_addr_param_t);
713     }
714
715   /* src_port & dst_port are already in network byte-order */
716   init_ack_chunk->sctp_hdr.checksum = 0;
717   init_ack_chunk->sctp_hdr.src_port =
718     sctp_conn->sub_conn[idx].connection.lcl_port;
719   init_ack_chunk->sctp_hdr.dst_port =
720     sctp_conn->sub_conn[idx].connection.rmt_port;
721   /* the sctp_conn->verification_tag is already in network byte-order (being a copy of the init_tag coming with the INIT chunk) */
722   init_ack_chunk->sctp_hdr.verification_tag = sctp_conn->remote_tag;
723   init_ack_chunk->initial_tsn =
724     clib_host_to_net_u32 (sctp_conn->local_initial_tsn);
725   SCTP_CONN_TRACKING_DBG ("init_ack_chunk->initial_tsn = %u",
726                           init_ack_chunk->initial_tsn);
727
728   vnet_sctp_set_chunk_type (&init_ack_chunk->chunk_hdr, INIT_ACK);
729   vnet_sctp_set_chunk_length (&init_ack_chunk->chunk_hdr, chunk_len);
730
731   init_ack_chunk->initiate_tag =
732     clib_host_to_net_u32 (random_u32 (&random_seed));
733
734   init_ack_chunk->a_rwnd = clib_host_to_net_u32 (DEFAULT_A_RWND);
735   init_ack_chunk->inboud_streams_count =
736     clib_host_to_net_u16 (INBOUND_STREAMS_COUNT);
737   init_ack_chunk->outbound_streams_count =
738     clib_host_to_net_u16 (OUTBOUND_STREAMS_COUNT);
739
740   sctp_conn->local_tag = init_ack_chunk->initiate_tag;
741
742   /* Measure RTT with this */
743   sctp_conn->sub_conn[idx].rtt_ts = sctp_time_now ();
744
745   vnet_buffer (b)->sctp.connection_index =
746     sctp_conn->sub_conn[idx].connection.c_index;
747 }
748
749 /**
750  * Convert buffer to SHUTDOWN
751  */
752 void
753 sctp_prepare_shutdown_chunk (sctp_connection_t * sctp_conn, vlib_buffer_t * b)
754 {
755   vlib_main_t *vm = vlib_get_main ();
756   u8 idx = sctp_pick_conn_idx_on_chunk (SHUTDOWN);
757   u16 alloc_bytes = sizeof (sctp_shutdown_association_chunk_t);
758
759   b = sctp_reuse_buffer (vm, b);
760
761   /* As per RFC 4960 the chunk_length value does NOT contemplate
762    * the size of the first header (see sctp_header_t) and any padding
763    */
764   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
765
766   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
767
768   sctp_shutdown_association_chunk_t *shutdown_chunk =
769     vlib_buffer_push_uninit (b, alloc_bytes);
770
771   shutdown_chunk->sctp_hdr.checksum = 0;
772   /* No need of host_to_net conversion, already in net-byte order */
773   shutdown_chunk->sctp_hdr.src_port =
774     sctp_conn->sub_conn[idx].connection.lcl_port;
775   shutdown_chunk->sctp_hdr.dst_port =
776     sctp_conn->sub_conn[idx].connection.rmt_port;
777   shutdown_chunk->sctp_hdr.verification_tag = sctp_conn->remote_tag;
778   vnet_sctp_set_chunk_type (&shutdown_chunk->chunk_hdr, SHUTDOWN);
779   vnet_sctp_set_chunk_length (&shutdown_chunk->chunk_hdr, chunk_len);
780
781   shutdown_chunk->cumulative_tsn_ack = sctp_conn->last_rcvd_tsn;
782
783   vnet_buffer (b)->sctp.connection_index =
784     sctp_conn->sub_conn[idx].connection.c_index;
785 }
786
787 /*
788  * Send SHUTDOWN
789  */
790 void
791 sctp_send_shutdown (sctp_connection_t * sctp_conn)
792 {
793   vlib_buffer_t *b;
794   u32 bi;
795   sctp_main_t *tm = vnet_get_sctp_main ();
796   vlib_main_t *vm = vlib_get_main ();
797
798   if (sctp_check_outstanding_data_chunks (sctp_conn) > 0)
799     return;
800
801   if (PREDICT_FALSE (sctp_get_free_buffer_index (tm, &bi)))
802     return;
803
804   b = vlib_get_buffer (vm, bi);
805   sctp_init_buffer (vm, b);
806   sctp_prepare_shutdown_chunk (sctp_conn, b);
807
808   u8 idx = sctp_pick_conn_idx_on_chunk (SHUTDOWN);
809   sctp_push_ip_hdr (tm, &sctp_conn->sub_conn[idx], b);
810   sctp_enqueue_to_output_now (vm, b, bi,
811                               sctp_conn->sub_conn[idx].connection.is_ip4);
812
813   /* Measure RTT with this */
814   sctp_conn->sub_conn[idx].rtt_ts = sctp_time_now ();
815 }
816
817 /**
818  * Convert buffer to SHUTDOWN_ACK
819  */
820 void
821 sctp_prepare_shutdown_ack_chunk (sctp_connection_t * sctp_conn,
822                                  vlib_buffer_t * b)
823 {
824   u8 idx = sctp_pick_conn_idx_on_chunk (SHUTDOWN_ACK);
825   u16 alloc_bytes = sizeof (sctp_shutdown_association_chunk_t);
826   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
827
828   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
829
830   sctp_shutdown_ack_chunk_t *shutdown_ack_chunk =
831     vlib_buffer_push_uninit (b, alloc_bytes);
832
833   shutdown_ack_chunk->sctp_hdr.checksum = 0;
834   /* No need of host_to_net conversion, already in net-byte order */
835   shutdown_ack_chunk->sctp_hdr.src_port =
836     sctp_conn->sub_conn[idx].connection.lcl_port;
837   shutdown_ack_chunk->sctp_hdr.dst_port =
838     sctp_conn->sub_conn[idx].connection.rmt_port;
839   shutdown_ack_chunk->sctp_hdr.verification_tag = sctp_conn->remote_tag;
840
841   vnet_sctp_set_chunk_type (&shutdown_ack_chunk->chunk_hdr, SHUTDOWN_ACK);
842   vnet_sctp_set_chunk_length (&shutdown_ack_chunk->chunk_hdr, chunk_len);
843
844   vnet_buffer (b)->sctp.connection_index =
845     sctp_conn->sub_conn[idx].connection.c_index;
846 }
847
848 /*
849  * Send SHUTDOWN_ACK
850  */
851 void
852 sctp_send_shutdown_ack (sctp_connection_t * sctp_conn)
853 {
854   vlib_buffer_t *b;
855   u32 bi;
856   sctp_main_t *tm = vnet_get_sctp_main ();
857   vlib_main_t *vm = vlib_get_main ();
858
859   if (sctp_check_outstanding_data_chunks (sctp_conn) > 0)
860     return;
861
862   if (PREDICT_FALSE (sctp_get_free_buffer_index (tm, &bi)))
863     return;
864
865   b = vlib_get_buffer (vm, bi);
866   sctp_init_buffer (vm, b);
867   sctp_prepare_shutdown_ack_chunk (sctp_conn, b);
868
869   u8 idx = sctp_pick_conn_idx_on_chunk (SHUTDOWN_ACK);
870   sctp_push_ip_hdr (tm, &sctp_conn->sub_conn[idx], b);
871   sctp_enqueue_to_ip_lookup (vm, b, bi,
872                              sctp_conn->sub_conn[idx].connection.is_ip4);
873
874   /* Measure RTT with this */
875   sctp_conn->sub_conn[idx].rtt_ts = sctp_time_now ();
876
877   /* Start the SCTP_TIMER_T2_SHUTDOWN timer */
878   sctp_timer_set (sctp_conn, idx, SCTP_TIMER_T2_SHUTDOWN,
879                   sctp_conn->sub_conn[idx].RTO);
880   sctp_conn->state = SCTP_STATE_SHUTDOWN_ACK_SENT;
881 }
882
883 /**
884  * Convert buffer to SACK
885  */
886 void
887 sctp_prepare_sack_chunk (sctp_connection_t * sctp_conn, vlib_buffer_t * b)
888 {
889   vlib_main_t *vm = vlib_get_main ();
890   u8 idx = sctp_pick_conn_idx_on_chunk (SACK);
891
892   sctp_reuse_buffer (vm, b);
893
894   u16 alloc_bytes = sizeof (sctp_selective_ack_chunk_t);
895
896   /* As per RFC 4960 the chunk_length value does NOT contemplate
897    * the size of the first header (see sctp_header_t) and any padding
898    */
899   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
900
901   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
902
903   sctp_selective_ack_chunk_t *sack = vlib_buffer_push_uninit (b, alloc_bytes);
904
905   sack->sctp_hdr.checksum = 0;
906   sack->sctp_hdr.src_port = sctp_conn->sub_conn[idx].connection.lcl_port;
907   sack->sctp_hdr.dst_port = sctp_conn->sub_conn[idx].connection.rmt_port;
908   sack->sctp_hdr.verification_tag = sctp_conn->remote_tag;
909   vnet_sctp_set_chunk_type (&sack->chunk_hdr, SACK);
910   vnet_sctp_set_chunk_length (&sack->chunk_hdr, chunk_len);
911
912   sack->cumulative_tsn_ack = sctp_conn->next_tsn_expected;
913
914   sctp_conn->ack_state = 0;
915
916   vnet_buffer (b)->sctp.connection_index =
917     sctp_conn->sub_conn[idx].connection.c_index;
918 }
919
920 /**
921  * Convert buffer to HEARTBEAT
922  */
923 void
924 sctp_prepare_heartbeat_chunk (sctp_connection_t * sctp_conn,
925                               vlib_buffer_t * b)
926 {
927   vlib_main_t *vm = vlib_get_main ();
928   u8 idx = sctp_pick_conn_idx_on_chunk (HEARTBEAT);
929   u16 alloc_bytes = sizeof (sctp_hb_req_chunk_t);
930
931   b = sctp_reuse_buffer (vm, b);
932
933   /* As per RFC 4960 the chunk_length value does NOT contemplate
934    * the size of the first header (see sctp_header_t) and any padding
935    */
936   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
937
938   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
939
940   sctp_hb_req_chunk_t *hb_req = vlib_buffer_push_uninit (b, alloc_bytes);
941
942   hb_req->sctp_hdr.checksum = 0;
943   /* No need of host_to_net conversion, already in net-byte order */
944   hb_req->sctp_hdr.src_port = sctp_conn->sub_conn[idx].connection.lcl_port;
945   hb_req->sctp_hdr.dst_port = sctp_conn->sub_conn[idx].connection.rmt_port;
946   hb_req->sctp_hdr.verification_tag = sctp_conn->remote_tag;
947   hb_req->hb_info.param_hdr.type = clib_host_to_net_u16 (1);
948   hb_req->hb_info.param_hdr.length =
949     clib_host_to_net_u16 (sizeof (hb_req->hb_info.hb_info));
950
951   vnet_sctp_set_chunk_type (&hb_req->chunk_hdr, HEARTBEAT);
952   vnet_sctp_set_chunk_length (&hb_req->chunk_hdr, chunk_len);
953
954   vnet_buffer (b)->sctp.connection_index =
955     sctp_conn->sub_conn[idx].connection.c_index;
956 }
957
958 void
959 sctp_send_heartbeat (sctp_connection_t * sctp_conn)
960 {
961   vlib_buffer_t *b;
962   u32 bi;
963   sctp_main_t *tm = vnet_get_sctp_main ();
964   vlib_main_t *vm = vlib_get_main ();
965
966   if (PREDICT_FALSE (sctp_get_free_buffer_index (tm, &bi)))
967     return;
968
969   b = vlib_get_buffer (vm, bi);
970   sctp_init_buffer (vm, b);
971   sctp_prepare_heartbeat_chunk (sctp_conn, b);
972
973   u8 idx = sctp_pick_conn_idx_on_state (SCTP_STATE_ESTABLISHED);
974   sctp_push_ip_hdr (tm, &sctp_conn->sub_conn[idx], b);
975   sctp_enqueue_to_ip_lookup (vm, b, bi,
976                              sctp_conn->sub_conn[idx].connection.is_ip4);
977 }
978
979 /**
980  * Convert buffer to SHUTDOWN_COMPLETE
981  */
982 void
983 sctp_prepare_shutdown_complete_chunk (sctp_connection_t * sctp_conn,
984                                       vlib_buffer_t * b)
985 {
986   u8 idx = sctp_pick_conn_idx_on_chunk (SHUTDOWN_COMPLETE);
987   u16 alloc_bytes = sizeof (sctp_shutdown_association_chunk_t);
988   alloc_bytes += vnet_sctp_calculate_padding (alloc_bytes);
989
990   u16 chunk_len = alloc_bytes - sizeof (sctp_header_t);
991
992   sctp_shutdown_complete_chunk_t *shutdown_complete =
993     vlib_buffer_push_uninit (b, alloc_bytes);
994
995   shutdown_complete->sctp_hdr.checksum = 0;
996   /* No need of host_to_net conversion, already in net-byte order */
997   shutdown_complete->sctp_hdr.src_port =
998     sctp_conn->sub_conn[idx].connection.lcl_port;
999   shutdown_complete->sctp_hdr.dst_port =
1000     sctp_conn->sub_conn[idx].connection.rmt_port;
1001   shutdown_complete->sctp_hdr.verification_tag = sctp_conn->remote_tag;
1002
1003   vnet_sctp_set_chunk_type (&shutdown_complete->chunk_hdr, SHUTDOWN_COMPLETE);
1004   vnet_sctp_set_chunk_length (&shutdown_complete->chunk_hdr, chunk_len);
1005
1006   vnet_buffer (b)->sctp.connection_index =
1007     sctp_conn->sub_conn[idx].connection.c_index;
1008 }
1009
1010 void
1011 sctp_send_shutdown_complete (sctp_connection_t * sctp_conn)
1012 {
1013   vlib_buffer_t *b;
1014   u32 bi;
1015   sctp_main_t *tm = vnet_get_sctp_main ();
1016   vlib_main_t *vm = vlib_get_main ();
1017
1018   if (PREDICT_FALSE (sctp_get_free_buffer_index (tm, &bi)))
1019     return;
1020
1021   b = vlib_get_buffer (vm, bi);
1022   sctp_init_buffer (vm, b);
1023   sctp_prepare_shutdown_complete_chunk (sctp_conn, b);
1024
1025   u8 idx = sctp_pick_conn_idx_on_chunk (SHUTDOWN_COMPLETE);
1026   sctp_push_ip_hdr (tm, &sctp_conn->sub_conn[idx], b);
1027   sctp_enqueue_to_ip_lookup (vm, b, bi,
1028                              sctp_conn->sub_conn[idx].connection.is_ip4);
1029
1030   sctp_conn->state = SCTP_STATE_CLOSED;
1031 }
1032
1033
1034 /*
1035  *  Send INIT
1036  */
1037 void
1038 sctp_send_init (sctp_connection_t * sctp_conn)
1039 {
1040   vlib_buffer_t *b;
1041   u32 bi;
1042   sctp_main_t *tm = vnet_get_sctp_main ();
1043   vlib_main_t *vm = vlib_get_main ();
1044
1045   if (PREDICT_FALSE (sctp_get_free_buffer_index (tm, &bi)))
1046     return;
1047
1048   b = vlib_get_buffer (vm, bi);
1049   u8 idx = sctp_pick_conn_idx_on_chunk (INIT);
1050
1051   sctp_init_buffer (vm, b);
1052   sctp_prepare_init_chunk (sctp_conn, b);
1053
1054   sctp_push_ip_hdr (tm, &sctp_conn->sub_conn[idx], b);
1055   sctp_enqueue_to_ip_lookup_now (vm, b, bi,
1056                                  sctp_conn->sub_conn[idx].c_is_ip4);
1057
1058   /* Measure RTT with this */
1059   sctp_conn->sub_conn[idx].rtt_ts = sctp_time_now ();
1060
1061   /* Start the T1_INIT timer */
1062   sctp_timer_set (sctp_conn, idx, SCTP_TIMER_T1_INIT,
1063                   sctp_conn->sub_conn[idx].RTO);
1064
1065   /* Change state to COOKIE_WAIT */
1066   sctp_conn->state = SCTP_STATE_COOKIE_WAIT;
1067 }
1068
1069 /**
1070  * Push SCTP header and update connection variables
1071  */
1072 static void
1073 sctp_push_hdr_i (sctp_connection_t * sctp_conn, u8 idx, vlib_buffer_t * b,
1074                  sctp_state_t next_state)
1075 {
1076   u16 data_len =
1077     b->current_length + b->total_length_not_including_first_buffer;
1078   ASSERT (!b->total_length_not_including_first_buffer
1079           || (b->flags & VLIB_BUFFER_NEXT_PRESENT));
1080
1081   SCTP_ADV_DBG_OUTPUT ("b->current_length = %u, "
1082                        "b->current_data = %p "
1083                        "data_len = %u",
1084                        b->current_length, b->current_data, data_len);
1085
1086   u16 bytes_to_add = sizeof (sctp_payload_data_chunk_t);
1087   u16 chunk_length = data_len + bytes_to_add - sizeof (sctp_header_t);
1088
1089   bytes_to_add += vnet_sctp_calculate_padding (bytes_to_add + data_len);
1090
1091   sctp_payload_data_chunk_t *data_chunk =
1092     vlib_buffer_push_uninit (b, bytes_to_add);
1093
1094   data_chunk->sctp_hdr.checksum = 0;
1095   data_chunk->sctp_hdr.src_port =
1096     sctp_conn->sub_conn[idx].connection.lcl_port;
1097   data_chunk->sctp_hdr.dst_port =
1098     sctp_conn->sub_conn[idx].connection.rmt_port;
1099   data_chunk->sctp_hdr.verification_tag = sctp_conn->remote_tag;
1100
1101   data_chunk->tsn = clib_host_to_net_u32 (sctp_conn->next_tsn);
1102   data_chunk->stream_id = clib_host_to_net_u16 (0);
1103   data_chunk->stream_seq = clib_host_to_net_u16 (0);
1104
1105   vnet_sctp_set_chunk_type (&data_chunk->chunk_hdr, DATA);
1106   vnet_sctp_set_chunk_length (&data_chunk->chunk_hdr, chunk_length);
1107
1108   vnet_sctp_set_bbit (&data_chunk->chunk_hdr);
1109   vnet_sctp_set_ebit (&data_chunk->chunk_hdr);
1110
1111   SCTP_ADV_DBG_OUTPUT ("POINTER_WITH_DATA = %p, DATA_OFFSET = %u",
1112                        b->data, b->current_data);
1113
1114   sctp_conn->next_tsn += data_len;
1115
1116   vnet_buffer (b)->sctp.connection_index =
1117     sctp_conn->sub_conn[idx].connection.c_index;
1118 }
1119
1120 u32
1121 sctp_push_header (transport_connection_t * trans_conn, vlib_buffer_t * b)
1122 {
1123   sctp_connection_t *sctp_conn =
1124     sctp_get_connection_from_transport (trans_conn);
1125
1126   u8 idx = sctp_pick_conn_idx_on_chunk (DATA);
1127
1128   sctp_push_hdr_i (sctp_conn, idx, b, SCTP_STATE_ESTABLISHED);
1129
1130   if (sctp_conn->sub_conn[idx].RTO_pending == 0)
1131     {
1132       sctp_conn->sub_conn[idx].RTO_pending = 1;
1133       sctp_conn->sub_conn[idx].rtt_ts = sctp_time_now ();
1134     }
1135
1136   sctp_trajectory_add_start (b0, 3);
1137
1138   return 0;
1139
1140 }
1141
1142 always_inline uword
1143 sctp46_output_inline (vlib_main_t * vm,
1144                       vlib_node_runtime_t * node,
1145                       vlib_frame_t * from_frame, int is_ip4)
1146 {
1147   u32 n_left_from, next_index, *from, *to_next;
1148   u32 my_thread_index = vm->thread_index;
1149
1150   from = vlib_frame_vector_args (from_frame);
1151   n_left_from = from_frame->n_vectors;
1152   next_index = node->cached_next_index;
1153   sctp_set_time_now (my_thread_index);
1154
1155   while (n_left_from > 0)
1156     {
1157       u32 n_left_to_next;
1158
1159       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
1160
1161       while (n_left_from > 0 && n_left_to_next > 0)
1162         {
1163           u32 bi0;
1164           vlib_buffer_t *b0;
1165           sctp_header_t *sctp_hdr = 0;
1166           sctp_connection_t *sctp_conn;
1167           sctp_tx_trace_t *t0;
1168           sctp_header_t *th0 = 0;
1169           u32 error0 = SCTP_ERROR_PKTS_SENT, next0 =
1170             SCTP_OUTPUT_NEXT_IP_LOOKUP;
1171
1172 #if SCTP_DEBUG_STATE_MACHINE
1173           u16 packet_length = 0;
1174 #endif
1175
1176           bi0 = from[0];
1177           to_next[0] = bi0;
1178           from += 1;
1179           to_next += 1;
1180           n_left_from -= 1;
1181           n_left_to_next -= 1;
1182
1183           b0 = vlib_get_buffer (vm, bi0);
1184           sctp_conn =
1185             sctp_connection_get (vnet_buffer (b0)->sctp.connection_index,
1186                                  my_thread_index);
1187
1188           if (PREDICT_FALSE (sctp_conn == 0))
1189             {
1190               error0 = SCTP_ERROR_INVALID_CONNECTION;
1191               next0 = SCTP_OUTPUT_NEXT_DROP;
1192               goto done;
1193             }
1194
1195           u8 idx = sctp_pick_conn_idx_on_state (sctp_conn->state);
1196
1197           th0 = vlib_buffer_get_current (b0);
1198
1199           if (is_ip4)
1200             {
1201               ip4_header_t *iph4 = vlib_buffer_push_ip4 (vm,
1202                                                          b0,
1203                                                          &sctp_conn->sub_conn
1204                                                          [idx].connection.
1205                                                          lcl_ip.ip4,
1206                                                          &sctp_conn->
1207                                                          sub_conn
1208                                                          [idx].connection.
1209                                                          rmt_ip.ip4,
1210                                                          IP_PROTOCOL_SCTP, 1);
1211
1212               u32 checksum = ip4_sctp_compute_checksum (vm, b0, iph4);
1213
1214               sctp_hdr = ip4_next_header (iph4);
1215               sctp_hdr->checksum = checksum;
1216
1217               vnet_buffer (b0)->l4_hdr_offset = (u8 *) th0 - b0->data;
1218
1219 #if SCTP_DEBUG_STATE_MACHINE
1220               packet_length = clib_net_to_host_u16 (iph4->length);
1221 #endif
1222             }
1223           else
1224             {
1225               ip6_header_t *iph6 = vlib_buffer_push_ip6 (vm,
1226                                                          b0,
1227                                                          &sctp_conn->sub_conn
1228                                                          [idx].
1229                                                          connection.lcl_ip.
1230                                                          ip6,
1231                                                          &sctp_conn->sub_conn
1232                                                          [idx].
1233                                                          connection.rmt_ip.
1234                                                          ip6,
1235                                                          IP_PROTOCOL_SCTP);
1236
1237               int bogus = ~0;
1238               u32 checksum = ip6_sctp_compute_checksum (vm, b0, iph6, &bogus);
1239               ASSERT (!bogus);
1240
1241               sctp_hdr = ip6_next_header (iph6);
1242               sctp_hdr->checksum = checksum;
1243
1244               vnet_buffer (b0)->l3_hdr_offset = (u8 *) iph6 - b0->data;
1245               vnet_buffer (b0)->l4_hdr_offset = (u8 *) th0 - b0->data;
1246
1247 #if SCTP_DEBUG_STATE_MACHINE
1248               packet_length = clib_net_to_host_u16 (iph6->payload_length);
1249 #endif
1250             }
1251
1252           u8 is_valid =
1253             (sctp_conn->sub_conn[idx].connection.lcl_port ==
1254              sctp_hdr->src_port
1255              || sctp_conn->sub_conn[idx].connection.lcl_port ==
1256              sctp_hdr->dst_port)
1257             && (sctp_conn->sub_conn[idx].connection.rmt_port ==
1258                 sctp_hdr->dst_port
1259                 || sctp_conn->sub_conn[idx].connection.rmt_port ==
1260                 sctp_hdr->src_port);
1261
1262           sctp_full_hdr_t *full_hdr = (sctp_full_hdr_t *) sctp_hdr;
1263           u8 chunk_type = vnet_sctp_get_chunk_type (&full_hdr->common_hdr);
1264
1265           if (!is_valid)
1266             {
1267               SCTP_DBG_STATE_MACHINE ("BUFFER IS INCORRECT: conn_index = %u, "
1268                                       "packet_length = %u, "
1269                                       "chunk_type = %u [%s], "
1270                                       "connection.lcl_port = %u, sctp_hdr->src_port = %u, "
1271                                       "connection.rmt_port = %u, sctp_hdr->dst_port = %u",
1272                                       sctp_conn->sub_conn
1273                                       [idx].connection.c_index, packet_length,
1274                                       chunk_type,
1275                                       sctp_chunk_to_string (chunk_type),
1276                                       sctp_conn->sub_conn[idx].
1277                                       connection.lcl_port, sctp_hdr->src_port,
1278                                       sctp_conn->sub_conn[idx].
1279                                       connection.rmt_port,
1280                                       sctp_hdr->dst_port);
1281
1282               error0 = SCTP_ERROR_UNKOWN_CHUNK;
1283               next0 = SCTP_OUTPUT_NEXT_DROP;
1284               goto done;
1285             }
1286
1287           SCTP_DBG_STATE_MACHINE
1288             ("CONN_INDEX = %u, CURR_CONN_STATE = %u (%s), "
1289              "CHUNK_TYPE = %s, " "SRC_PORT = %u, DST_PORT = %u",
1290              sctp_conn->sub_conn[idx].connection.c_index,
1291              sctp_conn->state, sctp_state_to_string (sctp_conn->state),
1292              sctp_chunk_to_string (chunk_type), full_hdr->hdr.src_port,
1293              full_hdr->hdr.dst_port);
1294
1295           if (chunk_type == DATA)
1296             SCTP_ADV_DBG_OUTPUT ("PACKET_LENGTH = %u", packet_length);
1297
1298           /* Let's make sure the state-machine does not send anything crazy */
1299           switch (sctp_conn->state)
1300             {
1301             case SCTP_STATE_CLOSED:
1302               {
1303                 if (chunk_type != INIT && chunk_type != INIT_ACK)
1304                   {
1305                     SCTP_DBG_STATE_MACHINE
1306                       ("Sending the wrong chunk (%s) based on state-machine status (%s)",
1307                        sctp_chunk_to_string (chunk_type),
1308                        sctp_state_to_string (sctp_conn->state));
1309
1310                     error0 = SCTP_ERROR_UNKOWN_CHUNK;
1311                     next0 = SCTP_OUTPUT_NEXT_DROP;
1312                     goto done;
1313                   }
1314                 break;
1315               }
1316             case SCTP_STATE_ESTABLISHED:
1317               if (chunk_type != DATA && chunk_type != HEARTBEAT &&
1318                   chunk_type != HEARTBEAT_ACK && chunk_type != SACK &&
1319                   chunk_type != COOKIE_ACK && chunk_type != SHUTDOWN)
1320                 {
1321                   SCTP_DBG_STATE_MACHINE
1322                     ("Sending the wrong chunk (%s) based on state-machine status (%s)",
1323                      sctp_chunk_to_string (chunk_type),
1324                      sctp_state_to_string (sctp_conn->state));
1325
1326                   error0 = SCTP_ERROR_UNKOWN_CHUNK;
1327                   next0 = SCTP_OUTPUT_NEXT_DROP;
1328                   goto done;
1329                 }
1330               break;
1331             case SCTP_STATE_COOKIE_WAIT:
1332               if (chunk_type != COOKIE_ECHO)
1333                 {
1334                   SCTP_DBG_STATE_MACHINE
1335                     ("Sending the wrong chunk (%s) based on state-machine status (%s)",
1336                      sctp_chunk_to_string (chunk_type),
1337                      sctp_state_to_string (sctp_conn->state));
1338
1339                   error0 = SCTP_ERROR_UNKOWN_CHUNK;
1340                   next0 = SCTP_OUTPUT_NEXT_DROP;
1341                   goto done;
1342                 }
1343               /* Change state */
1344               sctp_conn->state = SCTP_STATE_COOKIE_ECHOED;
1345               break;
1346             default:
1347               SCTP_DBG_STATE_MACHINE
1348                 ("Sending chunk (%s) based on state-machine status (%s)",
1349                  sctp_chunk_to_string (chunk_type),
1350                  sctp_state_to_string (sctp_conn->state));
1351               break;
1352             }
1353
1354           if (chunk_type == SHUTDOWN)
1355             {
1356               /* Start the SCTP_TIMER_T2_SHUTDOWN timer */
1357               sctp_timer_set (sctp_conn, idx, SCTP_TIMER_T2_SHUTDOWN,
1358                               sctp_conn->sub_conn[idx].RTO);
1359               sctp_conn->state = SCTP_STATE_SHUTDOWN_SENT;
1360             }
1361
1362           if (chunk_type == DATA)
1363             {
1364               sctp_timer_update (sctp_conn, idx, SCTP_TIMER_T3_RXTX,
1365                                  sctp_conn->sub_conn[idx].RTO);
1366             }
1367
1368           vnet_buffer (b0)->sw_if_index[VLIB_RX] = 0;
1369           vnet_buffer (b0)->sw_if_index[VLIB_TX] = ~0;
1370
1371           b0->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
1372
1373           SCTP_DBG_STATE_MACHINE ("CONNECTION_INDEX = %u, "
1374                                   "NEW_STATE = %s, "
1375                                   "CHUNK_SENT = %s",
1376                                   sctp_conn->sub_conn[idx].connection.c_index,
1377                                   sctp_state_to_string (sctp_conn->state),
1378                                   sctp_chunk_to_string (chunk_type));
1379
1380           vnet_sctp_common_hdr_params_host_to_net (&full_hdr->common_hdr);
1381
1382         done:
1383           b0->error = node->errors[error0];
1384           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
1385             {
1386               t0 = vlib_add_trace (vm, node, b0, sizeof (*t0));
1387               if (th0)
1388                 {
1389                   clib_memcpy (&t0->sctp_header, th0,
1390                                sizeof (t0->sctp_header));
1391                 }
1392               else
1393                 {
1394                   memset (&t0->sctp_header, 0, sizeof (t0->sctp_header));
1395                 }
1396               clib_memcpy (&t0->sctp_connection, sctp_conn,
1397                            sizeof (t0->sctp_connection));
1398             }
1399
1400           vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
1401                                            n_left_to_next, bi0, next0);
1402         }
1403
1404       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
1405     }
1406
1407   return from_frame->n_vectors;
1408 }
1409
1410 static uword
1411 sctp4_output (vlib_main_t * vm, vlib_node_runtime_t * node,
1412               vlib_frame_t * from_frame)
1413 {
1414   return sctp46_output_inline (vm, node, from_frame, 1 /* is_ip4 */ );
1415 }
1416
1417 static uword
1418 sctp6_output (vlib_main_t * vm, vlib_node_runtime_t * node,
1419               vlib_frame_t * from_frame)
1420 {
1421   return sctp46_output_inline (vm, node, from_frame, 0 /* is_ip4 */ );
1422 }
1423
1424 /* *INDENT-OFF* */
1425 VLIB_REGISTER_NODE (sctp4_output_node) =
1426 {
1427   .function = sctp4_output,.name = "sctp4-output",
1428     /* Takes a vector of packets. */
1429     .vector_size = sizeof (u32),
1430     .n_errors = SCTP_N_ERROR,
1431     .error_strings = sctp_error_strings,
1432     .n_next_nodes = SCTP_OUTPUT_N_NEXT,
1433     .next_nodes = {
1434 #define _(s,n) [SCTP_OUTPUT_NEXT_##s] = n,
1435     foreach_sctp4_output_next
1436 #undef _
1437     },
1438     .format_buffer = format_sctp_header,
1439     .format_trace = format_sctp_tx_trace,
1440 };
1441 /* *INDENT-ON* */
1442
1443 VLIB_NODE_FUNCTION_MULTIARCH (sctp4_output_node, sctp4_output);
1444
1445 /* *INDENT-OFF* */
1446 VLIB_REGISTER_NODE (sctp6_output_node) =
1447 {
1448   .function = sctp6_output,
1449   .name = "sctp6-output",
1450     /* Takes a vector of packets. */
1451   .vector_size = sizeof (u32),
1452   .n_errors = SCTP_N_ERROR,
1453   .error_strings = sctp_error_strings,
1454   .n_next_nodes = SCTP_OUTPUT_N_NEXT,
1455   .next_nodes = {
1456 #define _(s,n) [SCTP_OUTPUT_NEXT_##s] = n,
1457     foreach_sctp6_output_next
1458 #undef _
1459   },
1460   .format_buffer = format_sctp_header,
1461   .format_trace = format_sctp_tx_trace,
1462 };
1463 /* *INDENT-ON* */
1464
1465 VLIB_NODE_FUNCTION_MULTIARCH (sctp6_output_node, sctp6_output);
1466
1467 /*
1468  * fd.io coding-style-patch-verification: ON
1469  *
1470  * Local Variables:
1471  * eval: (c-set-style "gnu")
1472  * End:
1473  */