diff options
Diffstat (limited to 'src/plugins/ioam/udp-ping/udp_ping_util.c')
| -rw-r--r-- | src/plugins/ioam/udp-ping/udp_ping_util.c | 330 |
1 files changed, 330 insertions, 0 deletions
diff --git a/src/plugins/ioam/udp-ping/udp_ping_util.c b/src/plugins/ioam/udp-ping/udp_ping_util.c new file mode 100644 index 00000000..55f48ea4 --- /dev/null +++ b/src/plugins/ioam/udp-ping/udp_ping_util.c @@ -0,0 +1,330 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include <vnet/vnet.h> +#include <vnet/ip/ip.h> +#include <vnet/ip/ip6_hop_by_hop.h> +#include <ioam/encap/ip6_ioam_e2e.h> +#include <ioam/encap/ip6_ioam_trace.h> +#include <ioam/udp-ping/udp_ping_packet.h> +#include <ioam/udp-ping/udp_ping.h> + +#define UDP_PING_REWRITE_LEN 1000 + +u16 +udp_ping_fill_udp_data (udp_ping_t * udp_ping, + u16 src_port, u16 dst_port, u8 msg_type, u16 ctx) +{ + /* Populate udp ping header */ + udp_ping->udp.src_port = clib_host_to_net_u16 (src_port); + udp_ping->udp.dst_port = clib_host_to_net_u16 (dst_port); + udp_ping->udp.length = clib_host_to_net_u16 (sizeof (udp_ping_t)); + udp_ping->udp.checksum = 0; + udp_ping->ping_data.probe_marker1 = + clib_host_to_net_u32 (UDP_PING_PROBE_MARKER1); + udp_ping->ping_data.probe_marker2 = + clib_host_to_net_u32 (UDP_PING_PROBE_MARKER2); + udp_ping->ping_data.version = 1; + udp_ping->ping_data.msg_type = msg_type; + udp_ping->ping_data.flags = clib_host_to_net_u16 (0); + udp_ping->ping_data.tel_req_vec = clib_host_to_net_u16 (0); + udp_ping->ping_data.hop_limit = 254; + udp_ping->ping_data.hop_count = 0; + udp_ping->ping_data.reserve = clib_host_to_net_u16 (0); + udp_ping->ping_data.max_len = + udp_ping->ping_data.cur_len = clib_host_to_net_u16 (0); + udp_ping->ping_data.sender_handle = clib_host_to_net_u16 (ctx); + udp_ping->ping_data.seq_no = clib_host_to_net_u16 (0); + + return (sizeof (udp_ping_t)); +} + +/** + * @brief Frame IPv6 udp-ping probe packet. + * + * Creates IPv6 UDP-Ping probe packet along with iOAM headers. + * + */ +int +udp_ping_create_ip6_pak (u8 * buf, /*u16 len, */ + ip6_address_t src, ip6_address_t dst, + u16 src_port, u16 dst_port, u8 msg_type, u16 ctx) +{ + ip6_header_t *ip0; + ip6_hop_by_hop_header_t *hbh0; + //trace_profile *profile = NULL; + u16 hbh_len = 0, rnd_size = 0, ip0_len = 0, udp_len = 0; + u16 trace_len = 0, trace_data_size = 0; + u16 e2e_len = sizeof (ioam_e2e_option_t) - sizeof (ip6_hop_by_hop_option_t); + u8 *current = NULL; + ioam_trace_option_t *trace_option; + ioam_e2e_option_t *e2e; + + ip0 = (ip6_header_t *) buf; + + ip0->ip_version_traffic_class_and_flow_label = + clib_host_to_net_u32 (0x6 << 28); + + ip0->protocol = IP_PROTOCOL_IP6_HOP_BY_HOP_OPTIONS; + ip0->hop_limit = 255; + + ip0->src_address = src; + ip0->dst_address = dst; + + hbh0 = (ip6_hop_by_hop_header_t *) (ip0 + 1); + + /* Calculate hbh header len */ + //profile = trace_profile_find(); + trace_data_size = fetch_trace_data_size (TRACE_TYPE_IF_TS_APP); + /* We need 2 times data for trace as packet traverse back to source */ + trace_len = sizeof (ioam_trace_option_t) + + (5 * trace_data_size * 2) - sizeof (ip6_hop_by_hop_option_t); + //(profile->num_elts * trace_data_size * 2); + hbh_len = e2e_len + trace_len + sizeof (ip6_hop_by_hop_header_t); + rnd_size = (hbh_len + 7) & ~7; + + /* Length of header in 8 octet units, not incl first 8 octets */ + hbh0->length = (rnd_size >> 3) - 1; + hbh0->protocol = IP_PROTOCOL_UDP; + + /* Populate hbh header */ + current = (u8 *) (hbh0 + 1); + + /* Populate trace */ + trace_option = (ioam_trace_option_t *) current; + trace_option->hdr.type = HBH_OPTION_TYPE_IOAM_TRACE_DATA_LIST | + HBH_OPTION_TYPE_DATA_CHANGE_ENROUTE; + trace_option->hdr.length = trace_len; + trace_option->trace_hdr.ioam_trace_type = + TRACE_TYPE_IF_TS_APP & TRACE_TYPE_MASK; + + trace_option->trace_hdr.data_list_elts_left = 5 * 2; + //profile->num_elts * 2; + + current += trace_option->hdr.length + sizeof (ip6_hop_by_hop_option_t); + + /* Populate e2e */ + e2e = (ioam_e2e_option_t *) current; + e2e->hdr.type = HBH_OPTION_TYPE_IOAM_EDGE_TO_EDGE; + e2e->hdr.length = e2e_len; + + /* Move past hbh header */ + current = ((u8 *) hbh0) + ((hbh0->length + 1) << 3); + + /* Populate udp ping header */ + udp_len = udp_ping_fill_udp_data ((udp_ping_t *) current, + src_port, dst_port, msg_type, ctx); + + /* Calculate total length and set it in ip6 header */ + ip0_len = ((hbh0->length + 1) << 3) + udp_len; + //ip0_len = (len > ip0_len) ? len : ip0_len; + ip0->payload_length = clib_host_to_net_u16 (ip0_len); + + return (ip0_len + sizeof (ip6_header_t)); +} + +int +udp_ping_compare_flow (ip46_address_t src, ip46_address_t dst, + u16 start_src_port, u16 end_src_port, + u16 start_dst_port, u16 end_dst_port, + ip46_udp_ping_flow * flow) +{ + if ((0 == ip46_address_cmp (&flow->src, &src)) && + (0 == ip46_address_cmp (&flow->dst, &dst)) && + (flow->udp_data.start_src_port == start_src_port) && + (flow->udp_data.end_src_port == end_src_port) && + (flow->udp_data.start_dst_port == start_dst_port) && + (flow->udp_data.end_dst_port == end_dst_port)) + { + return 0; + } + + return -1; +} + +void +udp_ping_populate_flow (ip46_address_t src, ip46_address_t dst, + u16 start_src_port, u16 end_src_port, + u16 start_dst_port, u16 end_dst_port, + u16 interval, u8 fault_det, ip46_udp_ping_flow * flow) +{ + flow->src = src; + flow->dst = dst; + flow->udp_data.start_src_port = start_src_port; + flow->udp_data.end_src_port = end_src_port; + flow->udp_data.start_dst_port = start_dst_port; + flow->udp_data.end_dst_port = end_dst_port; + flow->udp_data.interval = interval; + flow->udp_data.next_send_time = 0; + flow->fault_det = fault_det; +} + +void +udp_ping_create_rewrite (ip46_udp_ping_flow * flow, u16 ctx) +{ + u16 src_port; + u16 dst_port; + u16 no_flows; + int i; + udp_ping_flow_data *stats; + + no_flows = + (flow->udp_data.end_dst_port - flow->udp_data.start_dst_port) + 1; + no_flows *= + ((flow->udp_data.end_src_port - flow->udp_data.start_src_port) + 1); + + vec_validate_aligned (flow->udp_data.stats, + no_flows - 1, CLIB_CACHE_LINE_BYTES); + + i = 0; + for (src_port = flow->udp_data.start_src_port; + src_port <= flow->udp_data.end_src_port; src_port++) + { + for (dst_port = flow->udp_data.start_dst_port; + dst_port <= flow->udp_data.end_dst_port; dst_port++) + { + u8 *rewrite = NULL; + + stats = flow->udp_data.stats + i; + ioam_analyse_init_data (&stats->analyse_data); + stats->analyse_data.is_free = 0; + + vec_validate (rewrite, UDP_PING_REWRITE_LEN - 1); + stats->ping_rewrite = rewrite; + stats->rewrite_len = + udp_ping_create_ip6_pak (rewrite, + flow->src.ip6, flow->dst.ip6, + src_port, dst_port, UDP_PING_PROBE, ctx); + /* For each flow we need to create ioam e2e flow */ + stats->flow_ctx = ioam_flow_add (1, (u8 *) "udp_ping"); //FIXME + i++; + } + } +} + +void +udp_ping_free_flow_data (ip46_udp_ping_flow * flow) +{ + int i; + udp_ping_flow_data *stats; + + for (i = 0; i < vec_len (flow->udp_data.stats); i++) + { + stats = flow->udp_data.stats + i; + vec_free (stats->ping_rewrite); + stats->ping_rewrite = NULL; + stats->rewrite_len = 0; + } + + vec_free (flow->udp_data.stats); + flow->udp_data.stats = NULL; +} + +/** + * @brief Create and send ipv6 udp-ping probe packet. + * + */ +void +udp_ping_send_ip6_pak (vlib_main_t * vm, ip46_udp_ping_flow * flow) +{ + u16 no_pak; + u32 *buffers = NULL; + int i; + vlib_buffer_t *b0; + udp_ping_flow_data *stats; + vlib_frame_t *nf = 0; + u32 *to_next; + vlib_node_t *next_node; + + next_node = vlib_get_node_by_name (vm, (u8 *) "ip6-lookup"); + nf = vlib_get_frame_to_node (vm, next_node->index); + nf->n_vectors = 0; + to_next = vlib_frame_vector_args (nf); + + no_pak = vec_len (flow->udp_data.stats); + vec_validate (buffers, (no_pak - 1)); + if (vlib_buffer_alloc (vm, buffers, vec_len (buffers)) != no_pak) + { + //Error + return; + } + + for (i = 0; i < no_pak; i++) + { + int bogus; + b0 = vlib_get_buffer (vm, buffers[i]); + stats = flow->udp_data.stats + i; + clib_memcpy (b0->data, stats->ping_rewrite, stats->rewrite_len); + b0->current_data = 0; + b0->current_length = stats->rewrite_len; + b0->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID; + + /* If session is going down, then set path down */ + if ((stats->retry != 0) && ((stats->retry % MAX_PING_RETRIES) == 0)) + ip6_ioam_analyse_set_paths_down (&stats->analyse_data); + + stats->retry++; + stats->analyse_data.pkt_sent++; + vnet_buffer (b0)->sw_if_index[VLIB_RX] = 0; + vnet_buffer (b0)->sw_if_index[VLIB_TX] = ~0; + vnet_buffer (b0)->l2_classify.opaque_index = stats->flow_ctx; + + ip6_header_t *ip6 = vlib_buffer_get_current (b0); + ip6_hop_by_hop_header_t *hbh = (ip6_hop_by_hop_header_t *) (ip6 + 1); + udp_header_t *udp = + (udp_header_t *) ((u8 *) hbh + ((hbh->length + 1) << 3)); + + /* If session is down, then set loopback flag in probe. + * This is for fault isolation. + */ + if (flow->fault_det && (stats->retry > MAX_PING_RETRIES)) + { + ioam_trace_option_t *opt = (ioam_trace_option_t *) + ip6_hbh_get_option (hbh, HBH_OPTION_TYPE_IOAM_TRACE_DATA_LIST); + ip6_hbh_ioam_trace_set_bit (opt, BIT_LOOPBACK); + } + + /* Checksum not pre-computed as we intend to vary packet length for every + * probe. its isnt done yet, but to be taken up later. + */ + udp->checksum = ip6_tcp_udp_icmp_compute_checksum (vm, b0, ip6, &bogus); + ASSERT (bogus == 0); + if (udp->checksum == 0) + udp->checksum = 0xffff; + + if (nf->n_vectors == VLIB_FRAME_SIZE) + { + vlib_put_frame_to_node (vm, next_node->index, nf); + nf = vlib_get_frame_to_node (vm, next_node->index); + nf->n_vectors = 0; + to_next = vlib_frame_vector_args (nf); + } + *to_next = buffers[i]; + nf->n_vectors++; + to_next++; + } + vlib_put_frame_to_node (vm, next_node->index, nf); + + flow->udp_data.next_send_time = + vlib_time_now (vm) + flow->udp_data.interval; +} + +/* + * fd.io coding-style-patch-verification: ON + * + * Local Variables: + * eval: (c-set-style "gnu") + * End: + */ |