src/plugins/ioam/analyse/ioam_summary_export.c

   1 /*
   2  * Copyright (c) 2017 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 #include <vlib/vlib.h>
  17 #include <vnet/ip/ip6_packet.h>
  18 #include <ioam/analyse/ioam_summary_export.h>
  19 #include <ioam/analyse/ip6/ip6_ioam_analyse.h>
  20
  21 u8 *
  22 ioam_template_rewrite (flow_report_main_t * frm, flow_report_t * fr,
  23                        ip4_address_t * collector_address,
  24                        ip4_address_t * src_address, u16 collector_port)
  25 {
  26   ip4_header_t *ip;
  27   udp_header_t *udp;
  28   ipfix_message_header_t *h;
  29   ipfix_set_header_t *s;
  30   ipfix_template_header_t *t;
  31   ipfix_field_specifier_t *f;
  32   ipfix_field_specifier_t *first_field;
  33   u8 *rewrite = 0;
  34   ip4_ipfix_template_packet_t *tp;
  35   u32 field_count = 0;
  36   u32 field_index = 0;
  37   flow_report_stream_t *stream;
  38
  39   stream = &frm->streams[fr->stream_index];
  40
  41   /* Determine field count */
  42 #define _(field,mask,item,length)                                   \
  43     {                                                               \
  44   field_count++;                                                    \
  45   fr->fields_to_send = clib_bitmap_set (fr->fields_to_send,         \
  46                                         field_index, 1);            \
  47     }                                                               \
  48     field_index++;
  49
  50   foreach_ioam_ipfix_field;
  51 #undef _
  52
  53   /* Add Src address, dest address, src port, dest port
  54    * path map,  number of paths manually */
  55   field_count += 6;
  56
  57   /* allocate rewrite space */
  58   vec_validate_aligned (rewrite,
  59                         sizeof (ip4_ipfix_template_packet_t)
  60                         + field_count * sizeof (ipfix_field_specifier_t) - 1,
  61                         CLIB_CACHE_LINE_BYTES);
  62
  63   tp = (ip4_ipfix_template_packet_t *) rewrite;
  64   ip = (ip4_header_t *) & tp->ip4;
  65   udp = (udp_header_t *) (ip + 1);
  66   h = (ipfix_message_header_t *) (udp + 1);
  67   s = (ipfix_set_header_t *) (h + 1);
  68   t = (ipfix_template_header_t *) (s + 1);
  69   first_field = f = (ipfix_field_specifier_t *) (t + 1);
  70
  71   ip->ip_version_and_header_length = 0x45;
  72   ip->ttl = 254;
  73   ip->protocol = IP_PROTOCOL_UDP;
  74   ip->src_address.as_u32 = src_address->as_u32;
  75   ip->dst_address.as_u32 = collector_address->as_u32;
  76   udp->src_port = clib_host_to_net_u16 (collector_port);
  77   udp->dst_port = clib_host_to_net_u16 (UDP_DST_PORT_ipfix);
  78   udp->length = clib_host_to_net_u16 (vec_len (rewrite) - sizeof (*ip));
  79
  80   h->domain_id = clib_host_to_net_u32 (stream->domain_id);      //fr->domain_id);
  81
  82   /* Add Src address, dest address, src port, dest port
  83    * path map,  number of paths manually */
  84   f->e_id_length = ipfix_e_id_length (0 /* enterprise */ ,
  85                                       sourceIPv6Address,
  86                                       sizeof (ip6_address_t));
  87   f++;
  88
  89   f->e_id_length = ipfix_e_id_length (0 /* enterprise */ ,
  90                                       destinationIPv6Address,
  91                                       sizeof (ip6_address_t));
  92   f++;
  93
  94   f->e_id_length = ipfix_e_id_length (0 /* enterprise */ ,
  95                                       sourceTransportPort, 2);
  96   f++;
  97
  98   f->e_id_length = ipfix_e_id_length (0 /* enterprise */ ,
  99                                       destinationTransportPort, 2);
 100   f++;
 101
 102 #define _(field,mask,item,length)                               \
 103     {                                                           \
 104   f->e_id_length = ipfix_e_id_length (0 /* enterprise */,       \
 105     item, length);                                              \
 106     f++;                                                        \
 107     }
 108   foreach_ioam_ipfix_field;
 109 #undef _
 110
 111   f->e_id_length = ipfix_e_id_length (0 /* enterprise */ ,
 112                                       ioamNumberOfPaths, 2);
 113   f++;
 114
 115   /* Add ioamPathMap manually */
 116   f->e_id_length = ipfix_e_id_length (0 /* enterprise */ ,
 117                                       ioamPathMap, (1 * sizeof (ioam_path)));
 118   f++;
 119
 120   /* Back to the template packet... */
 121   ip = (ip4_header_t *) & tp->ip4;
 122   udp = (udp_header_t *) (ip + 1);
 123
 124   ASSERT (f - first_field);
 125   /* Field count in this template */
 126   t->id_count = ipfix_id_count (IOAM_FLOW_TEMPLATE_ID, f - first_field);
 127
 128   /* set length in octets */
 129   s->set_id_length =
 130     ipfix_set_id_length (2 /* set_id */ , (u8 *) f - (u8 *) s);
 131
 132   /* message length in octets */
 133   h->version_length = version_length ((u8 *) f - (u8 *) h);
 134
 135   ip->length = clib_host_to_net_u16 ((u8 *) f - (u8 *) ip);
 136   ip->checksum = ip4_header_checksum (ip);
 137
 138   return rewrite;
 139 }
 140
 141 u16
 142 ioam_analyse_add_ipfix_record (flow_report_t * fr,
 143                                ioam_analyser_data_t * record,
 144                                vlib_buffer_t * b0, u16 offset,
 145                                ip6_address_t * src, ip6_address_t * dst,
 146                                u16 src_port, u16 dst_port)
 147 {
 148   while (__sync_lock_test_and_set (record->writer_lock, 1))
 149     ;
 150
 151   int field_index = 0;
 152   u16 tmp;
 153   int i, j;
 154   u16 num_paths = 0;
 155   u16 num_paths_offset;
 156
 157
 158   /* Add IPv6 source address manually */
 159   memcpy (b0->data + offset, &src->as_u64[0], sizeof (u64));
 160   offset += sizeof (u64);
 161   memcpy (b0->data + offset, &src->as_u64[1], sizeof (u64));
 162   offset += sizeof (u64);
 163
 164   /* Add IPv6 destination address manually */
 165   memcpy (b0->data + offset, &dst->as_u64[0], sizeof (u64));
 166   offset += sizeof (u64);
 167   memcpy (b0->data + offset, &dst->as_u64[1], sizeof (u64));
 168   offset += sizeof (u64);
 169
 170   /* Add source port manually */
 171   tmp = clib_host_to_net_u16 (src_port);
 172   memcpy (b0->data + offset, &tmp, sizeof (u16));
 173   offset += sizeof (u16);
 174
 175   /* Add dest port manually */
 176   tmp = clib_host_to_net_u16 (dst_port);
 177   memcpy (b0->data + offset, &tmp, sizeof (u16));
 178   offset += sizeof (u16);
 179
 180 #define _(field,mask,item,length)                            \
 181     if (clib_bitmap_get (fr->fields_to_send, field_index))   \
 182     {                                                        \
 183       /* Expect only 4 bytes */               \
 184       u32 tmp;                                             \
 185       tmp = clib_host_to_net_u32((u32)record->field - (u32)record->chached_data_list->field);\
 186       memcpy (b0->data + offset, &tmp, length);       \
 187       offset += length;                                 \
 188     }
 189   field_index++;
 190   foreach_ioam_ipfix_field;
 191 #undef _
 192
 193   /* Store num_paths_offset here and update later */
 194   num_paths_offset = offset;
 195   offset += sizeof (u16);
 196
 197   /* Add ioamPathMap manually */
 198   for (i = 0; i < IOAM_MAX_PATHS_PER_FLOW; i++)
 199     {
 200       ioam_analyse_trace_record *trace = record->trace_data.path_data + i;
 201       ioam_analyse_trace_record *trace_cached =
 202         record->chached_data_list->trace_data.path_data + i;
 203       ioam_path *path = (ioam_path *) (b0->data + offset);
 204
 205       if (!trace->is_free)
 206         {
 207           num_paths++;
 208
 209           path->num_nodes = trace->num_nodes;
 210
 211           path->trace_type = trace->trace_type;
 212           if (0 < (trace->pkt_counter - trace_cached->pkt_counter))
 213             {
 214               u64 new_sum = trace->mean_delay * record->seqno_data.rx_packets;
 215               u64 old_sum =
 216                 trace_cached->mean_delay *
 217                 record->chached_data_list->seqno_data.rx_packets;
 218               path->mean_delay =
 219                 (u32) ((new_sum - old_sum) / (trace->pkt_counter -
 220                                               trace_cached->pkt_counter));
 221               path->mean_delay = clib_host_to_net_u32 (path->mean_delay);
 222             }
 223           else
 224             path->mean_delay = 0;
 225
 226           path->bytes_counter =
 227             trace->bytes_counter - trace_cached->bytes_counter;
 228           path->bytes_counter = clib_host_to_net_u32 (path->bytes_counter);
 229
 230           path->pkt_counter = trace->pkt_counter - trace_cached->pkt_counter;
 231           path->pkt_counter = clib_host_to_net_u32 (path->pkt_counter);
 232
 233           for (j = 0; j < IOAM_TRACE_MAX_NODES; j++)
 234             {
 235               path->path[j].node_id =
 236                 clib_host_to_net_u32 (trace->path[j].node_id);
 237               path->path[j].ingress_if =
 238                 clib_host_to_net_u16 (trace->path[j].ingress_if);
 239               path->path[j].egress_if =
 240                 clib_host_to_net_u16 (trace->path[j].egress_if);
 241             }
 242
 243           offset += sizeof (ioam_path);
 244         }
 245     }
 246
 247   num_paths = clib_host_to_net_u16 (num_paths);
 248   memcpy (b0->data + num_paths_offset, &num_paths, sizeof (u16));
 249
 250   /* Update cache */
 251   *(record->chached_data_list) = *record;
 252   record->chached_data_list->chached_data_list = NULL;
 253
 254   *(record->writer_lock) = 0;
 255   return offset;
 256 }
 257
 258 vlib_frame_t *
 259 ioam_send_flows (flow_report_main_t * frm, flow_report_t * fr,
 260                  vlib_frame_t * f, u32 * to_next, u32 node_index)
 261 {
 262   vlib_buffer_t *b0 = NULL;
 263   u32 next_offset = 0;
 264   u32 bi0 = ~0;
 265   int i;
 266   ip4_ipfix_template_packet_t *tp;
 267   ipfix_message_header_t *h;
 268   ipfix_set_header_t *s = NULL;
 269   ip4_header_t *ip;
 270   udp_header_t *udp;
 271   u32 records_this_buffer;
 272   u16 new_l0, old_l0;
 273   ip_csum_t sum0;
 274   vlib_main_t *vm = frm->vlib_main;
 275   ip6_address_t temp;
 276   ioam_analyser_data_t *record = NULL;
 277   flow_report_stream_t *stream;
 278   ioam_analyser_data_t *aggregated_data;
 279   u16 data_len;
 280
 281   stream = &frm->streams[fr->stream_index];
 282
 283   memset (&temp, 0, sizeof (ip6_address_t));
 284
 285   aggregated_data = ioam_analyser_main.aggregated_data;
 286   data_len = vec_len (aggregated_data);
 287
 288   vec_foreach_index (i, aggregated_data)
 289   {
 290     u8 flush = 0;
 291     record = aggregated_data + i;
 292
 293     /* Flush if last entry */
 294     if (i == (data_len - 1))
 295       flush = 1;
 296
 297     if (!record->is_free)
 298       {
 299
 300         if (PREDICT_FALSE (b0 == NULL))
 301           {
 302             if (vlib_buffer_alloc (vm, &bi0, 1) != 1)
 303               break;
 304
 305             b0 = vlib_get_buffer (vm, bi0);
 306             memcpy (b0->data, fr->rewrite, vec_len (fr->rewrite));
 307             b0->current_data = 0;
 308             b0->current_length = vec_len (fr->rewrite);
 309             b0->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID;
 310             vnet_buffer (b0)->sw_if_index[VLIB_RX] = 0;
 311             vnet_buffer (b0)->sw_if_index[VLIB_TX] = ~0;
 312
 313             tp = vlib_buffer_get_current (b0);
 314             ip = &tp->ip4;
 315             h = &tp->ipfix.h;
 316             s = &tp->ipfix.s;
 317
 318             /* FIXUP: message header export_time */
 319             h->export_time = clib_host_to_net_u32 (((u32) time (NULL)));
 320
 321             /* FIXUP: message header sequence_number */
 322             h->sequence_number = stream->sequence_number++;
 323             h->sequence_number = clib_host_to_net_u32 (h->sequence_number);
 324             next_offset = (u32) (((u8 *) (s + 1)) - (u8 *) tp);
 325             records_this_buffer = 0;
 326           }
 327
 328         next_offset = ioam_analyse_add_ipfix_record (fr, record,
 329                                                      b0, next_offset,
 330                                                      &temp, &temp, 0, 0);
 331         records_this_buffer++;
 332
 333         /* Flush data if packet len is about to reach path mtu */
 334         if (next_offset > (frm->path_mtu - 250))
 335           flush = 1;
 336       }
 337
 338     if (PREDICT_FALSE (flush && b0))
 339       {
 340         s->set_id_length = ipfix_set_id_length (IOAM_FLOW_TEMPLATE_ID,
 341                                                 next_offset - (sizeof (*ip) +
 342                                                                sizeof (*udp) +
 343                                                                sizeof (*h)));
 344         b0->current_length = next_offset;
 345         b0->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID;
 346         tp = vlib_buffer_get_current (b0);
 347         ip = (ip4_header_t *) & tp->ip4;
 348         udp = (udp_header_t *) (ip + 1);
 349
 350         sum0 = ip->checksum;
 351         old_l0 = ip->length;
 352         new_l0 = clib_host_to_net_u16 ((u16) next_offset);
 353         sum0 = ip_csum_update (sum0, old_l0, new_l0, ip4_header_t,
 354                                length /* changed member */ );
 355
 356         ip->checksum = ip_csum_fold (sum0);
 357         ip->length = new_l0;
 358         udp->length =
 359           clib_host_to_net_u16 (b0->current_length - sizeof (*ip));
 360
 361         if (frm->udp_checksum)
 362           {
 363             /* RFC 7011 section 10.3.2. */
 364             udp->checksum = ip4_tcp_udp_compute_checksum (vm, b0, ip);
 365             if (udp->checksum == 0)
 366               udp->checksum = 0xffff;
 367           }
 368
 369         to_next[0] = bi0;
 370         f->n_vectors++;
 371         to_next++;
 372
 373         if (f->n_vectors == VLIB_FRAME_SIZE)
 374           {
 375             vlib_put_frame_to_node (vm, node_index, f);
 376             f = vlib_get_frame_to_node (vm, node_index);
 377             f->n_vectors = 0;
 378             to_next = vlib_frame_vector_args (f);
 379           }
 380         b0 = 0;
 381         bi0 = ~0;
 382       }
 383   }
 384
 385   return f;
 386 }
 387
 388 clib_error_t *
 389 ioam_flow_create (u8 del)
 390 {
 391   vnet_flow_report_add_del_args_t args;
 392   int rv;
 393   u32 domain_id = 0;
 394   flow_report_main_t *frm = &flow_report_main;
 395
 396   memset (&args, 0, sizeof (args));
 397   args.rewrite_callback = ioam_template_rewrite;
 398   args.flow_data_callback = ioam_send_flows;
 399   del ? (args.is_add = 0) : (args.is_add = 1);
 400   args.domain_id = domain_id;
 401
 402   rv = vnet_flow_report_add_del (frm, &args);
 403
 404   switch (rv)
 405     {
 406     case 0:
 407       break;
 408     case VNET_API_ERROR_NO_SUCH_ENTRY:
 409       return clib_error_return (0, "registration not found...");
 410     default:
 411       return clib_error_return (0, "vnet_flow_report_add_del returned %d",
 412                                 rv);
 413     }
 414
 415   return 0;
 416 }
 417
 418 clib_error_t *
 419 ioam_flow_report_init (vlib_main_t * vm)
 420 {
 421   clib_error_t *error;
 422
 423   if ((error = vlib_call_init_function (vm, flow_report_init)))
 424     return error;
 425
 426   return 0;
 427 }
 428
 429 VLIB_INIT_FUNCTION (ioam_flow_report_init);
 430
 431 /*
 432  * fd.io coding-style-patch-verification: ON
 433  *
 434  * Local Variables:
 435  * eval: (c-set-style "gnu")
 436  * End:
 437  */