/* * decap.c: gtpu tunnel decap packet processing * * Copyright (c) 2017 Intel 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 #include #include vlib_node_registration_t gtpu4_input_node; vlib_node_registration_t gtpu6_input_node; typedef struct { u32 next_index; u32 tunnel_index; u32 error; u32 teid; } gtpu_rx_trace_t; static u8 * format_gtpu_rx_trace (u8 * s, va_list * args) { CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *); CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *); gtpu_rx_trace_t * t = va_arg (*args, gtpu_rx_trace_t *); if (t->tunnel_index != ~0) { s = format (s, "GTPU decap from gtpu_tunnel%d teid %d next %d error %d", t->tunnel_index, t->teid, t->next_index, t->error); } else { s = format (s, "GTPU decap error - tunnel for teid %d does not exist", t->teid); } return s; } always_inline u32 validate_gtpu_fib (vlib_buffer_t *b, gtpu_tunnel_t *t, u32 is_ip4) { u32 fib_index, sw_if_index; sw_if_index = vnet_buffer (b)->sw_if_index[VLIB_RX]; if (is_ip4) fib_index = (vnet_buffer (b)->sw_if_index[VLIB_TX] == (u32) ~ 0) ? vec_elt (ip4_main.fib_index_by_sw_if_index, sw_if_index) : vnet_buffer (b)->sw_if_index[VLIB_TX]; else fib_index = (vnet_buffer (b)->sw_if_index[VLIB_TX] == (u32) ~ 0) ? vec_elt (ip6_main.fib_index_by_sw_if_index, sw_if_index) : vnet_buffer (b)->sw_if_index[VLIB_TX]; return (fib_index == t->encap_fib_index); } always_inline uword gtpu_input (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * from_frame, u32 is_ip4) { u32 n_left_from, next_index, * from, * to_next; gtpu_main_t * gtm = >pu_main; vnet_main_t * vnm = gtm->vnet_main; vnet_interface_main_t * im = &vnm->interface_main; u32 last_tunnel_index = ~0; gtpu4_tunnel_key_t last_key4; gtpu6_tunnel_key_t last_key6; u32 pkts_decapsulated = 0; u32 thread_index = vlib_get_thread_index(); u32 stats_sw_if_index, stats_n_packets, stats_n_bytes; if (is_ip4) last_key4.as_u64 = ~0; else memset (&last_key6, 0xff, sizeof (last_key6)); from = vlib_frame_vector_args (from_frame); n_left_from = from_frame->n_vectors; next_index = node->cached_next_index; stats_sw_if_index = node->runtime_data[0]; stats_n_packets = stats_n_bytes = 0; while (n_left_from > 0) { u32 n_left_to_next; vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next); while (n_left_from >= 4 && n_left_to_next >= 2) { u32 bi0, bi1; vlib_buffer_t * b0, * b1; u32 next0, next1; ip4_header_t * ip4_0, * ip4_1; ip6_header_t * ip6_0, * ip6_1; gtpu_header_t * gtpu0, * gtpu1; u32 gtpu_hdr_len0 = 0, gtpu_hdr_len1 =0 ; uword * p0, * p1; u32 tunnel_index0, tunnel_index1; gtpu_tunnel_t * t0, * t1, * mt0 = NULL, * mt1 = NULL; gtpu4_tunnel_key_t key4_0, key4_1; gtpu6_tunnel_key_t key6_0, key6_1; u32 error0, error1; u32 sw_if_index0, sw_if_index1, len0, len1; /* Prefetch next iteration. */ { vlib_buffer_t * p2, * p3; p2 = vlib_get_buffer (vm, from[2]); p3 = vlib_get_buffer (vm, from[3]); vlib_prefetch_buffer_header (p2, LOAD); vlib_prefetch_buffer_header (p3, LOAD); CLIB_PREFETCH (p2->data, 2*CLIB_CACHE_LINE_BYTES, LOAD); CLIB_PREFETCH (p3->data, 2*CLIB_CACHE_LINE_BYTES, LOAD); } bi0 = from[0]; bi1 = from[1]; to_next[0] = bi0; to_next[1] = bi1; from += 2; to_next += 2; n_left_to_next -= 2; n_left_from -= 2; b0 = vlib_get_buffer (vm, bi0); b1 = vlib_get_buffer (vm, bi1); /* udp leaves current_data pointing at the gtpu header */ gtpu0 = vlib_buffer_get_current (b0); gtpu1 = vlib_buffer_get_current (b1); if (is_ip4) { vlib_buffer_advance (b0, -(word)(sizeof(udp_header_t)+sizeof(ip4_header_t))); vlib_buffer_advance (b1, -(word)(sizeof(udp_header_t)+sizeof(ip4_header_t))); ip4_0 = vlib_buffer_get_current (b0); ip4_1 = vlib_buffer_get_current (b1); } else { vlib_buffer_advance (b0, -(word)(sizeof(udp_header_t)+sizeof(ip6_header_t))); vlib_buffer_advance (b1, -(word)(sizeof(udp_header_t)+sizeof(ip6_header_t))); ip6_0 = vlib_buffer_get_current (b0); ip6_1 = vlib_buffer_get_current (b1); } /* pop (ip, udp, gtpu) */ if (is_ip4) { vlib_buffer_advance (b0, sizeof(*ip4_0)+sizeof(udp_header_t)); vlib_buffer_advance (b1, sizeof(*ip4_1)+sizeof(udp_header_t)); } else { vlib_buffer_advance (b0, sizeof(*ip6_0)+sizeof(udp_header_t)); vlib_buffer_advance (b1, sizeof(*ip6_1)+sizeof(udp_header_t)); } tunnel_index0 = ~0; error0 = 0; tunnel_index1 = ~0; error1 = 0; if (PREDICT_FALSE ((gtpu0->ver_flags & GTPU_VER_MASK) != GTPU_V1_VER)) { error0 = GTPU_ERROR_BAD_VER; next0 = GTPU_INPUT_NEXT_DROP; goto trace0; } /* Manipulate packet 0 */ if (is_ip4) { key4_0.src = ip4_0->src_address.as_u32; key4_0.teid = gtpu0->teid; /* Make sure GTPU tunnel exist according to packet SIP and teid * SIP identify a GTPU path, and teid identify a tunnel in a given GTPU path */ if (PREDICT_FALSE (key4_0.as_u64 != last_key4.as_u64)) { p0 = hash_get (gtm->gtpu4_tunnel_by_key, key4_0.as_u64); if (PREDICT_FALSE (p0 == NULL)) { error0 = GTPU_ERROR_NO_SUCH_TUNNEL; next0 = GTPU_INPUT_NEXT_DROP; goto trace0; } last_key4.as_u64 = key4_0.as_u64; tunnel_index0 = last_tunnel_index = p0[0]; } else tunnel_index0 = last_tunnel_index; t0 = pool_elt_at_index (gtm->tunnels, tunnel_index0); /* Validate GTPU tunnel encap-fib index agaist packet */ if (PREDICT_FALSE (validate_gtpu_fib (b0, t0, is_ip4) == 0)) { error0 = GTPU_ERROR_NO_SUCH_TUNNEL; next0 = GTPU_INPUT_NEXT_DROP; goto trace0; } /* Validate GTPU tunnel SIP against packet DIP */ if (PREDICT_TRUE (ip4_0->dst_address.as_u32 == t0->src.ip4.as_u32)) goto next0; /* valid packet */ if (PREDICT_FALSE (ip4_address_is_multicast (&ip4_0->dst_address))) { key4_0.src = ip4_0->dst_address.as_u32; key4_0.teid = gtpu0->teid; /* Make sure mcast GTPU tunnel exist by packet DIP and teid */ p0 = hash_get (gtm->gtpu4_tunnel_by_key, key4_0.as_u64); if (PREDICT_TRUE (p0 != NULL)) { mt0 = pool_elt_at_index (gtm->tunnels, p0[0]); goto next0; /* valid packet */ } } error0 = GTPU_ERROR_NO_SUCH_TUNNEL; next0 = GTPU_INPUT_NEXT_DROP; goto trace0; } else /* !is_ip4 */ { key6_0.src.as_u64[0] = ip6_0->src_address.as_u64[0]; key6_0.src.as_u64[1] = ip6_0->src_address.as_u64[1]; key6_0.teid = gtpu0->teid; /* Make sure GTPU tunnel exist according to packet SIP and teid * SIP identify a GTPU path, and teid identify a tunnel in a given GTPU path */ if (PREDICT_FALSE (memcmp(&key6_0, &last_key6, sizeof(last_key6)) != 0)) { p0 = hash_get_mem (gtm->gtpu6_tunnel_by_key, &key6_0); if (PREDICT_FALSE (p0 == NULL)) { error0 = GTPU_ERROR_NO_SUCH_TUNNEL; next0 = GTPU_INPUT_NEXT_DROP; goto trace0; } clib_memcpy (&last_key6, &key6_0, sizeof(key6_0)); tunnel_index0 = last_tunnel_index = p0[0]; } else tunnel_index0 = last_tunnel_index; t0 = pool_elt_at_index (gtm->tunnels, tunnel_index0); /* Validate GTPU tunnel encap-fib index agaist packet */ if (PREDICT_FALSE (validate_gtpu_fib (b0, t0, is_ip4) == 0)) { error0 = GTPU_ERROR_NO_SUCH_TUNNEL; next0 = GTPU_INPUT_NEXT_DROP; goto trace0; } /* Validate GTPU tunnel SIP against packet DIP */ if (PREDICT_TRUE (ip6_address_is_equal (&ip6_0->dst_address, &t0->src.ip6))) goto next0; /* valid packet */ if (PREDICT_FALSE (ip6_address_is_multicast (&ip6_0->dst_address))) { key6_0.src.as_u64[0] = ip6_0->dst_address.as_u64[0]; key6_0.src.as_u64[1] = ip6_0->dst_address.as_u64[1]; key6_0.teid = gtpu0->teid; p0 = hash_get_mem (gtm->gtpu6_tunnel_by_key, &key6_0); if (PREDICT_TRUE (p0 != NULL)) { mt0 = pool_elt_at_index (gtm->tunnels, p0[0]); goto next0; /* valid packet */ } } error0 = GTPU_ERROR_NO_SUCH_TUNNEL; next0 = GTPU_INPUT_NEXT_DROP; goto trace0; } next0: /* Manipulate gtpu header */ if (PREDICT_FALSE((gtpu0->ver_flags & GTPU_E_S_PN_BIT) != 0)) { gtpu_hdr_len0 = sizeof(gtpu_header_t); /* Manipulate Sequence Number and N-PDU Number */ /* TBD */ /* Manipulate Next Extension Header */ /* TBD */ } else { gtpu_hdr_len0 = sizeof(gtpu_header_t) - 4; } /* Pop gtpu header */ vlib_buffer_advance (b0, gtpu_hdr_len0); next0 = t0->decap_next_index; sw_if_index0 = t0->sw_if_index; len0 = vlib_buffer_length_in_chain (vm, b0); /* Required to make the l2 tag push / pop code work on l2 subifs */ if (PREDICT_TRUE(next0 == GTPU_INPUT_NEXT_L2_INPUT)) vnet_update_l2_len (b0); /* Set packet input sw_if_index to unicast GTPU tunnel for learning */ vnet_buffer(b0)->sw_if_index[VLIB_RX] = sw_if_index0; sw_if_index0 = (mt0) ? mt0->sw_if_index : sw_if_index0; pkts_decapsulated ++; stats_n_packets += 1; stats_n_bytes += len0; /* Batch stats increment on the same gtpu tunnel so counter is not incremented per packet */ if (PREDICT_FALSE (sw_if_index0 != stats_sw_if_index)) { stats_n_packets -= 1; stats_n_bytes -= len0; if (stats_n_packets) vlib_increment_combined_counter (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_RX, thread_index, stats_sw_if_index, stats_n_packets, stats_n_bytes); stats_n_packets = 1; stats_n_bytes = len0; stats_sw_if_index = sw_if_index0; } trace0: b0->error = error0 ? node->errors[error0] : 0; if (PREDICT_FALSE(b0->flags & VLIB_BUFFER_IS_TRACED)) { gtpu_rx_trace_t *tr = vlib_add_trace (vm, node, b0, sizeof (*tr)); tr->next_index = next0; tr->error = error0; tr->tunnel_index = tunnel_index0; tr->teid = clib_net_to_host_u32(gtpu0->teid); } if (PREDICT_FALSE ((gtpu1->ver_flags & GTPU_VER_MASK) != GTPU_V1_VER)) { error1 = GTPU_ERROR_BAD_VER; next1 = GTPU_INPUT_NEXT_DROP; goto trace1; } /* Manipulate packet 1 */ if (is_ip4) { key4_1.src = ip4_1->src_address.as_u32; key4_1.teid = gtpu1->teid; /* Make sure GTPU tunnel exist according to packet SIP and teid * SIP identify a GTPU path, and teid identify a tunnel in a given GTPU path */ if (PREDICT_FALSE (key4_1.as_u64 != last_key4.as_u64)) { p1 = hash_get (gtm->gtpu4_tunnel_by_key, key4_1.as_u64); if (PREDICT_FALSE (p1 == NULL)) { error1 = GTPU_ERROR_NO_SUCH_TUNNEL; next1 = GTPU_INPUT_NEXT_DROP; goto trace1; } last_key4.as_u64 = key4_1.as_u64; tunnel_index1 = last_tunnel_index = p1[0]; } else tunnel_index1 = last_tunnel_index; t1 = pool_elt_at_index (gtm->tunnels, tunnel_index1); /* Validate GTPU tunnel encap-fib index agaist packet */ if (PREDICT_FALSE (validate_gtpu_fib (b1, t1, is_ip4) == 0)) { error1 = GTPU_ERROR_NO_SUCH_TUNNEL; next1 = GTPU_INPUT_NEXT_DROP; goto trace1; } /* Validate GTPU tunnel SIP against packet DIP */ if (PREDICT_TRUE (ip4_1->dst_address.as_u32 == t1->src.ip4.as_u32)) goto next1; /* valid packet */ if (PREDICT_FALSE (ip4_address_is_multicast (&ip4_1->dst_address))) { key4_1.src = ip4_1->dst_address.as_u32; key4_1.teid = gtpu1->teid; /* Make sure mcast GTPU tunnel exist by packet DIP and teid */ p1 = hash_get (gtm->gtpu4_tunnel_by_key, key4_1.as_u64); if (PREDICT_TRUE (p1 != NULL)) { mt1 = pool_elt_at_index (gtm->tunnels, p1[0]); goto next1; /* valid packet */ } } error1 = GTPU_ERROR_NO_SUCH_TUNNEL; next1 = GTPU_INPUT_NEXT_DROP; goto trace1; } else /* !is_ip4 */ { key6_1.src.as_u64[0] = ip6_1->src_address.as_u64[0]; key6_1.src.as_u64[1] = ip6_1->src_address.as_u64[1]; key6_1.teid = gtpu1->teid; /* Make sure GTPU tunnel exist according to packet SIP and teid * SIP identify a GTPU path, and teid identify a tunnel in a given GTPU path */ if (PREDICT_FALSE (memcmp(&key6_1, &last_key6, sizeof(last_key6)) != 0)) { p1 = hash_get_mem (gtm->gtpu6_tunnel_by_key, &key6_1); if (PREDICT_FALSE (p1 == NULL)) { error1 = GTPU_ERROR_NO_SUCH_TUNNEL; next1 = GTPU_INPUT_NEXT_DROP; goto trace1; } clib_memcpy (&last_key6, &key6_1, sizeof(key6_1)); tunnel_index1 = last_tunnel_index = p1[0]; } else tunnel_index1 = last_tunnel_index; t1 = pool_elt_at_index (gtm->tunnels, tunnel_index1); /* Validate GTPU tunnel encap-fib index agaist packet */ if (PREDICT_FALSE (validate_gtpu_fib (b1, t1, is_ip4) == 0)) { error1 = GTPU_ERROR_NO_SUCH_TUNNEL; next1 = GTPU_INPUT_NEXT_DROP; goto trace1; } /* Validate GTPU tunnel SIP against packet DIP */ if (PREDICT_TRUE (ip6_address_is_equal (&ip6_1->dst_address, &t1->src.ip6))) goto next1; /* valid packet */ if (PREDICT_FALSE (ip6_address_is_multicast (&ip6_1->dst_address))) { key6_1.src.as_u64[0] = ip6_1->dst_address.as_u64[0]; key6_1.src.as_u64[1] = ip6_1->dst_address.as_u64[1]; key6_1.teid = gtpu1->teid; p1 = hash_get_mem (gtm->gtpu6_tunnel_by_key, &key6_1); if (PREDICT_TRUE (p1 != NULL)) { mt1 = pool_elt_at_index (gtm->tunnels, p1[0]); goto next1;
# Copyright (c) 2019 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.

"""Stream profile for T-rex traffic generator.

Stream profile:
 - Two streams sent in directions 0 --> 1 and 1 --> 0 at the same time.
 - Packet: ETH / IP /
 - Direction 0 --> 1:
   - Destination MAC address: 52:54:00:00:nf_id:01
   - Source IP address range:      10.10.10.1 - 10.10.10.254
   - Destination IP address range: 20.20.20.1
 - Direction 1 --> 0:
    - Destination MAC address: 52:54:00:00:nf_id:02
   - Source IP address range:      20.20.20.1 - 20.20.20.254
   - Destination IP address range: 10.10.10.1
"""

from trex_stl_lib.api import *
from profile_trex_stateless_base_class import TrafficStreamsBaseClass


class TrafficStreams(TrafficStreamsBaseClass):
    """Stream profile."""

    def __init__(self):
        """Initialization and setting of streams' parameters."""

        super(TrafficStreamsBaseClass, self).__init__()

        # Service density parameters.
        self.nf_chains = 4
        self.nf_nodes = 6

        # MACs used in packet headers.
        self.p1_dst_start_mac = '52:54:00:00:00:01'
        self.p2_dst_start_mac = '52:54:00:00:00:02'

        # IPs used in packet headers.
        self.p1_src_start_ip = '10.10.10.1'
        self.p1_src_end_ip = '10.10.10.254'
        self.p1_dst_start_ip = '20.20.20.1'

        self.p2_src_start_ip = '20.20.20.1'
        self.p2_src_end_ip = '20.20.20.254'
        self.p2_dst_start_ip = '10.10.10.1'

    def define_packets(self):
        """Defines the packets to be sent from the traffic generator.

        Packet definition: | ETH | IP |

        :returns: Packets to be sent from the traffic generator.
        :rtype: tuple
        """

        # Direction 0 --> 1
        base_pkt_a = (Ether(dst=self.p1_dst_start_mac) /
                      IP(src=self.p1_src_start_ip,
                         dst=self.p1_dst_start_ip,
                         proto=61))
        # Direction 1 --> 0
        base_pkt_b = (Ether(dst=self.p2_dst_start_mac) /
                      IP(src=self.p2_src_start_ip,
                         dst=self.p2_dst_start_ip,
                         proto=61))

        # Direction 0 --> 1
        vm1 = STLScVmRaw([STLVmFlowVar(name="mac_dst",
                                       min_value=1,
                                       max_value=self.nf_chains*self.nf_nodes,
                                       size=1, step=self.nf_nodes, op="inc"),
                          STLVmWrFlowVar(fv_name="mac_dst", pkt_offset=4),
                          STLVmFlowVar(name="src",
                                       min_value=self.p1_src_start_ip,
                                       max_value=self.p1_src_end_ip,
                                       size=4, op="inc"),
                          STLVmWrFlowVar(fv_name="src", pkt_offset="IP.src"),
                          STLVmFixIpv4(offset="IP")])
        # Direction 1 --> 0
        vm2 = STLScVmRaw([STLVmFlowVar(name="mac_dst",
                                       min_value=self.nf_nodes,
                                       max_value=self.nf_chains*self.nf_nodes,
                                       size=1, step=self.nf_nodes, op="inc"),
                          STLVmWrFlowVar(fv_name="mac_dst", pkt_offset=4),
                          STLVmFlowVar(name="src",
                                       min_value=self.p2_src_start_ip,
                                       max_value=self.p2_src_end_ip,
                                       size=4, op="inc"),
                          STLVmWrFlowVar(fv_name="src", pkt_offset="IP.src"),
                          STLVmFixIpv4(offset="IP")])

        return base_pkt_a, base_pkt_b, vm1, vm2


def register():
    """Register this traffic profile to T-rex.

    Do not change this function.

    :return: Traffic streams.
    :rtype: Object
    """
    return TrafficStreams()
YPASS_NEXT_DROP] = "error-drop", [IP_GTPU_BYPASS_NEXT_GTPU] = "gtpu4-input", }, .format_buffer = format_ip4_header, .format_trace = format_ip4_forward_next_trace, }; VLIB_NODE_FUNCTION_MULTIARCH (ip4_gtpu_bypass_node,ip4_gtpu_bypass) /* Dummy init function to get us linked in. */ clib_error_t * ip4_gtpu_bypass_init (vlib_main_t * vm) { return 0; } VLIB_INIT_FUNCTION (ip4_gtpu_bypass_init); static uword ip6_gtpu_bypass (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame) { return ip_gtpu_bypass_inline (vm, node, frame, /* is_ip4 */ 0); } VLIB_REGISTER_NODE (ip6_gtpu_bypass_node) = { .function = ip6_gtpu_bypass, .name = "ip6-gtpu-bypass", .vector_size = sizeof (u32), .n_next_nodes = IP_GTPU_BYPASS_N_NEXT, .next_nodes = { [IP_GTPU_BYPASS_NEXT_DROP] = "error-drop", [IP_GTPU_BYPASS_NEXT_GTPU] = "gtpu6-input", }, .format_buffer = format_ip6_header, .format_trace = format_ip6_forward_next_trace, }; VLIB_NODE_FUNCTION_MULTIARCH (ip6_gtpu_bypass_node,ip6_gtpu_bypass) /* Dummy init function to get us linked in. */ clib_error_t * ip6_gtpu_bypass_init (vlib_main_t * vm) { return 0; } VLIB_INIT_FUNCTION (ip6_gtpu_bypass_init);