/* * Copyright (c) 2016 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/ip/ip4_input.h> #include <vnet/ip/ip6_input.h> #include <vnet/dpo/mpls_disposition.h> #include <vnet/mpls/mpls.h> /* * pool of all MPLS Label DPOs */ mpls_disp_dpo_t *mpls_disp_dpo_pool; static mpls_disp_dpo_t * mpls_disp_dpo_alloc (void) { mpls_disp_dpo_t *mdd; pool_get_aligned(mpls_disp_dpo_pool, mdd, CLIB_CACHE_LINE_BYTES); memset(mdd, 0, sizeof(*mdd)); dpo_reset(&mdd->mdd_dpo); return (mdd); } static index_t mpls_disp_dpo_get_index (mpls_disp_dpo_t *mdd) { return (mdd - mpls_disp_dpo_pool); } index_t mpls_disp_dpo_create (dpo_proto_t payload_proto, fib_rpf_id_t rpf_id, const dpo_id_t *dpo) { mpls_disp_dpo_t *mdd; mdd = mpls_disp_dpo_alloc(); mdd->mdd_payload_proto = payload_proto; mdd->mdd_rpf_id = rpf_id; dpo_stack(DPO_MPLS_DISPOSITION, mdd->mdd_payload_proto, &mdd->mdd_dpo, dpo); return (mpls_disp_dpo_get_index(mdd)); } u8* format_mpls_disp_dpo (u8 *s, va_list *args) { index_t index = va_arg (*args, index_t); u32 indent = va_arg (*args, u32); mpls_disp_dpo_t *mdd; mdd = mpls_disp_dpo_get(index); s = format(s, "mpls-disposition:[%d]:[%U]", index, format_dpo_proto, mdd->mdd_payload_proto); s = format(s, "\n%U", format_white_space, indent); s = format(s, "%U", format_dpo_id, &mdd->mdd_dpo, indent+2); return (s); } static void mpls_disp_dpo_lock (dpo_id_t *dpo) { mpls_disp_dpo_t *mdd; mdd = mpls_disp_dpo_get(dpo->dpoi_index); mdd->mdd_locks++; } static void mpls_disp_dpo_unlock (dpo_id_t *dpo) { mpls_disp_dpo_t *mdd; mdd = mpls_disp_dpo_get(dpo->dpoi_index); mdd->mdd_locks--; if (0 == mdd->mdd_locks) { dpo_reset(&mdd->mdd_dpo); pool_put(mpls_disp_dpo_pool, mdd); } } /** * @brief A struct to hold tracing information for the MPLS label disposition * node. */ typedef struct mpls_label_disposition_trace_t_ { index_t mdd; } mpls_label_disposition_trace_t; extern vlib_node_registration_t ip4_mpls_label_disposition_node; extern vlib_node_registration_t ip6_mpls_label_disposition_node; always_inline uword mpls_label_disposition_inline (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * from_frame, u8 payload_is_ip4, u8 payload_is_ip6) { u32 n_left_from, next_index, * from, * to_next; vlib_node_runtime_t *error_node; if (payload_is_ip4) error_node = vlib_node_get_runtime (vm, ip4_mpls_label_disposition_node.index); else error_node = vlib_node_get_runtime (vm, ip6_mpls_label_disposition_node.index); from = vlib_frame_vector_args (from_frame); n_left_from = from_frame->n_vectors; next_index = node->cached_next_index; 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) { mpls_disp_dpo_t *mdd0, *mdd1; u32 bi0, mddi0, bi1, mddi1; vlib_buffer_t * b0, *b1; u32 next0, next1; bi0 = to_next[0] = from[0]; bi1 = to_next[1] = from[1]; /* 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, STORE); vlib_prefetch_buffer_header (p3, STORE); CLIB_PREFETCH (p2->data, sizeof (ip6_header_t), STORE); CLIB_PREFETCH (p3->data, sizeof (ip6_header_t), STORE); } from += 2; to_next += 2; n_left_from -= 2; n_left_to_next -= 2; b0 = vlib_get_buffer (vm, bi0); b1 = vlib_get_buffer (vm, bi1); /* dst lookup was done by ip4 lookup */ mddi0 = vnet_buffer(b0)->ip.adj_index[VLIB_TX]; mddi1 = vnet_buffer(b1)->ip.adj_index[VLIB_TX]; mdd0 = mpls_disp_dpo_get(mddi0); mdd1 = mpls_disp_dpo_get(mddi1); next0 = mdd0->mdd_dpo.dpoi_next_node; next1 = mdd1->mdd_dpo.dpoi_next_node; if (payload_is_ip4) { ip4_header_t *ip0, *ip1; ip0 = vlib_buffer_get_current (b0); ip1 = vlib_buffer_get_current (b1); /* * IPv4 input checks on the exposed IP header * including checksum */ ip4_input_check_x2 (vm, error_node, b0, b1, ip0, ip1, &next0, &next1, 1); } else if (payload_is_ip6) { ip6_header_t *ip0, *ip1; ip0 = vlib_buffer_get_current (b0); ip1 = vlib_buffer_get_current (b1); /* * IPv6 input checks on the exposed IP header */ ip6_input_check_x2 (vm, error_node, b0, b1, ip0, ip1, &next0, &next1); } vnet_buffer(b0)->ip.adj_index[VLIB_TX] = mdd0->mdd_dpo.dpoi_index; vnet_buffer(b1)->ip.adj_index[VLIB_TX] = mdd1->mdd_dpo.dpoi_index; vnet_buffer(b0)->ip.rpf_id = mdd0->mdd_rpf_id; vnet_buffer(b1)->ip.rpf_id = mdd1->mdd_rpf_id; if (PREDICT_FALSE(b0->flags & VLIB_BUFFER_IS_TRACED)) { mpls_label_disposition_trace_t *tr = vlib_add_trace (vm, node, b0, sizeof (*tr)); tr->mdd = mddi0; } if (PREDICT_FALSE(b1->flags & VLIB_BUFFER_IS_TRACED)) { mpls_label_disposition_trace_t *tr = vlib_add_trace (vm, node, b1, sizeof (*tr)); tr->mdd = mddi1; } vlib_validate_buffer_enqueue_x2(vm, node, next_index, to_next, n_left_to_next, bi0, bi1, next0, next1); } while (n_left_from > 0 && n_left_to_next > 0) { mpls_disp_dpo_t *mdd0; vlib_buffer_t * b0; u32 bi0, mddi0; u32 next0; bi0 = from[0]; to_next[0] = bi0; from += 1; to_next += 1; n_left_from -= 1; n_left_to_next -= 1; b0 = vlib_get_buffer (vm, bi0); /* dst lookup was done by ip4 lookup */ mddi0 = vnet_buffer(b0)->ip.adj_index[VLIB_TX]; mdd0 = mpls_disp_dpo_get(mddi0); next0 = mdd0->mdd_dpo.dpoi_next_node; if (payload_is_ip4) { ip4_header_t *ip0; ip0 = vlib_buffer_get_current (b0); /* * IPv4 input checks on the exposed IP header * including checksum */ ip4_input_check_x1 (vm, error_node, b0, ip0, &next0, 1); } else if (payload_is_ip6) { ip6_header_t *ip0; ip0 = vlib_buffer_get_current (b0); /* * IPv6 input checks on the exposed IP header */ ip6_input_check_x1 (vm, error_node, b0, ip0, &next0); } vnet_buffer(b0)->ip.adj_index[VLIB_TX] = mdd0->mdd_dpo.dpoi_index; vnet_buffer(b0)->ip.rpf_id = mdd0->mdd_rpf_id; if (PREDICT_FALSE(b0->flags & VLIB_BUFFER_IS_TRACED)) { mpls_label_disposition_trace_t *tr = vlib_add_trace (vm, node, b0, sizeof (*tr)); tr->mdd = mddi0; } vlib_validate_buffer_enqueue_x1(vm, node, next_index, to_next, n_left_to_next, bi0, next0); } vlib_put_next_frame (vm, node, next_index, n_left_to_next); } return from_frame->n_vectors; } static u8 * format_mpls_label_disposition_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 *); CLIB_UNUSED (mpls_label_disposition_trace_t * t); t = va_arg (*args, mpls_label_disposition_trace_t *); s = format(s, "disp:%d", t->mdd); return (s); } static uword ip4_mpls_label_disposition (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame) { return (mpls_label_disposition_inline(vm, node, frame, 1, 0)); } VLIB_REGISTER_NODE (ip4_mpls_label_disposition_node) = { .function = ip4_mpls_label_disposition, .name = "ip4-mpls-label-disposition", .vector_size = sizeof (u32), .format_trace = format_mpls_label_disposition_trace, .sibling_of = "ip4-input", .n_errors = IP4_N_ERROR, .error_strings = ip4_error_strings, }; VLIB_NODE_FUNCTION_MULTIARCH (ip4_mpls_label_disposition_node, ip4_mpls_label_disposition) static uword ip6_mpls_label_disposition (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame) { return (mpls_label_disposition_inline(vm, node, frame, 0, 1)); } VLIB_REGISTER_NODE (ip6_mpls_label_disposition_node) = { .function = ip6_mpls_label_disposition, .name = "ip6-mpls-label-disposition", .vector_size = sizeof (u32), .format_trace = format_mpls_label_disposition_trace, .sibling_of = "ip6-input", .n_errors = IP6_N_ERROR, .error_strings = ip6_error_strings, }; VLIB_NODE_FUNCTION_MULTIARCH (ip6_mpls_label_disposition_node, ip6_mpls_label_disposition) static void mpls_disp_dpo_mem_show (void) { fib_show_memory_usage("MPLS label", pool_elts(mpls_disp_dpo_pool), pool_len(mpls_disp_dpo_pool), sizeof(mpls_disp_dpo_t)); } const static dpo_vft_t mdd_vft = { .dv_lock = mpls_disp_dpo_lock, .dv_unlock = mpls_disp_dpo_unlock, .dv_format = format_mpls_disp_dpo, .dv_mem_show = mpls_disp_dpo_mem_show, }; const static char* const mpls_label_disp_ip4_nodes[] = { "ip4-mpls-label-disposition", NULL, }; const static char* const mpls_label_disp_ip6_nodes[] = { "ip6-mpls-label-disposition", NULL, }; const static char* const * const mpls_label_disp_nodes[DPO_PROTO_NUM] = { [DPO_PROTO_IP4] = mpls_label_disp_ip4_nodes, [DPO_PROTO_IP6] = mpls_label_disp_ip6_nodes, }; void mpls_disp_dpo_module_init (void) { dpo_register(DPO_MPLS_DISPOSITION, &mdd_vft, mpls_label_disp_nodes); }