/* * 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/vnet.h> #include <vnet/plugin/plugin.h> #include <lb/lb.h> #include <vppinfra/byte_order.h> #include <vppinfra/string.h> #include <vpp/api/types.h> #include <vlibapi/api.h> #include <vlibmemory/api.h> #include <vpp/app/version.h> #include <vnet/format_fns.h> #include <vnet/ip/ip_types_api.h> /* define message IDs */ #include <lb/lb.api_enum.h> #include <lb/lb.api_types.h> #define vl_print(handle, ...) vlib_cli_output (handle, __VA_ARGS__) #define REPLY_MSG_ID_BASE lbm->msg_id_base #include <vlibapi/api_helper_macros.h> /* Macro to finish up custom dump fns */ #define FINISH \ vec_add1 (s, 0); \ vl_print (handle, (char *)s); \ vec_free (s); \ return handle; static void vl_api_lb_conf_t_handler (vl_api_lb_conf_t * mp) { lb_main_t *lbm = &lb_main; vl_api_lb_conf_reply_t * rmp; u32 sticky_buckets_per_core, flow_timeout; int rv = 0; sticky_buckets_per_core = mp->sticky_buckets_per_core == ~0 ? lbm->per_cpu_sticky_buckets : ntohl(mp->sticky_buckets_per_core); flow_timeout = mp->flow_timeout == ~0 ? lbm->flow_timeout : ntohl(mp->flow_timeout); rv = lb_conf((ip4_address_t *)&mp->ip4_src_address, (ip6_address_t *)&mp->ip6_src_address, sticky_buckets_per_core, flow_timeout); REPLY_MACRO (VL_API_LB_CONF_REPLY); } static void *vl_api_lb_conf_t_print (vl_api_lb_conf_t *mp, void * handle) { u8 * s; s = format (0, "SCRIPT: lb_conf "); s = format (s, "%U ", format_ip4_address, (ip4_address_t *)&mp->ip4_src_address); s = format (s, "%U ", format_ip6_address, (ip6_address_t *)&mp->ip6_src_address); s = format (s, "%u ", mp->sticky_buckets_per_core); s = format (s, "%u ", mp->flow_timeout); FINISH; } static void vl_api_lb_add_del_vip_t_handler (vl_api_lb_add_del_vip_t * mp) { lb_main_t *lbm = &lb_main; vl_api_lb_conf_reply_t * rmp; int rv = 0; lb_vip_add_args_t args; /* if port == 0, it means all-port VIP */ if (mp->port == 0) { mp->protocol = ~0; } ip_address_decode (&mp->pfx.address, &(args.prefix)); if (mp->is_del) { u32 vip_index; if (!(rv = lb_vip_find_index(&(args.prefix), mp->pfx.len, mp->protocol, ntohs(mp->port), &vip_index))) rv = lb_vip_del(vip_index); } else { u32 vip_index; lb_vip_type_t type = 0; if (ip46_prefix_is_ip4(&(args.prefix), mp->pfx.len)) { if (mp->encap == LB_API_ENCAP_TYPE_GRE4) type = LB_VIP_TYPE_IP4_GRE4; else if (mp->encap == LB_API_ENCAP_TYPE_GRE6) type = LB_VIP_TYPE_IP4_GRE6; else if (mp->encap == LB_API_ENCAP_TYPE_L3DSR) type = LB_VIP_TYPE_IP4_L3DSR; else if (mp->encap == LB_API_ENCAP_TYPE_NAT4) type = LB_VIP_TYPE_IP4_NAT4; } else { if (mp->encap == LB_API_ENCAP_TYPE_GRE4) type = LB_VIP_TYPE_IP6_GRE4; else if (mp->encap == LB_API_ENCAP_TYPE_GRE6) type = LB_VIP_TYPE_IP6_GRE6; else if (mp->encap == LB_API_ENCAP_TYPE_NAT6) type = LB_VIP_TYPE_IP6_NAT6; } args.plen = mp->pfx.len; args.protocol = mp->protocol; args.port = ntohs(mp->port); args.type = type; args.new_length = ntohl(mp->new_flows_table_length); if (mp->encap == LB_API_ENCAP_TYPE_L3DSR) { args.encap_args.dscp = (u8)(mp->dscp & 0x3F); } else if ((mp->encap == LB_API_ENCAP_TYPE_NAT4) ||(mp->encap == LB_API_ENCAP_TYPE_NAT6)) { args.encap_args.srv_type = mp->type; args.encap_args.target_port = ntohs(mp->target_port); } rv = lb_vip_add(args, &vip_index); } REPLY_MACRO (VL_API_LB_ADD_DEL_VIP_REPLY); } static void *vl_api_lb_add_del_vip_t_print (vl_api_lb_add_del_vip_t *mp, void * handle) { u8 * s; s = format (0, "SCRIPT: lb_add_del_vip "); s = format (s, "%U", format_vl_api_prefix, &mp->pfx); s = format (s, "%s ", (mp->encap == LB_API_ENCAP_TYPE_GRE4)? "gre4" : (mp->encap == LB_API_ENCAP_TYPE_GRE6)? "gre6" : (mp->encap == LB_API_ENCAP_TYPE_NAT4)? "nat4" : (mp->encap == LB_API_ENCAP_TYPE_NAT6)? "nat6" : "l3dsr"); if (mp->encap==LB_API_ENCAP_TYPE_L3DSR) { s = format (s, "dscp %u ", mp->dscp); } if ((mp->encap==LB_API_ENCAP_TYPE_NAT4) || (mp->encap==LB_API_ENCAP_TYPE_NAT6)) { s = format (s, "type %u ", mp->type); s = format (s, "port %u ", mp->port); s = format (s, "target_port %u ", mp->target_port); } s = format (s, "%u ", mp->new_flows_table_length); s = format (s, "%s ", mp->is_del?"del":"add"); FINISH; } static void vl_api_lb_add_del_as_t_handler (vl_api_lb_add_del_as_t * mp) { lb_main_t *lbm = &lb_main; vl_api_lb_conf_reply_t * rmp; int rv = 0; u32 vip_index; ip46_address_t vip_ip_prefix; ip46_address_t as_address; /* if port == 0, it means all-port VIP */ if (mp->port == 0) { mp->protocol = ~0; } ip_address_decode (&mp->pfx.address, &vip_ip_prefix); ip_address_decode (&mp->as_address, &as_address); if ((rv = lb_vip_find_index(&vip_ip_prefix, mp->pfx.len, mp->protocol, ntohs(mp->port), &vip_index))) goto done; if (mp->is_del) rv = lb_vip_del_ass(vip_index, &as_address, 1, mp->is_flush); else rv = lb_vip_add_ass(vip_index, &as_address, 1); done: REPLY_MACRO (VL_API_LB_ADD_DEL_AS_REPLY); } static void *vl_api_lb_add_del_as_t_print (vl_api_lb_add_del_as_t *mp, void * handle) { u8 * s; ip46_address_t address; s = format (0, "SCRIPT: lb_add_del_as "); s = format (s, "%U ", format_vl_api_prefix, &mp->pfx); s = format(s, "%u ", mp->protocol); if (ip_address_decode (&mp->as_address, &address) == IP46_TYPE_IP6) s = format (s, "%U ", format_ip6_address, (ip6_address_t *) & address.ip6); else s = format (s, "%U ", format_ip4_address, (ip6_address_t *) & address.ip4); s = format (s, "%s ", mp->is_del?"del":"add"); FINISH; } static void vl_api_lb_vip_dump_t_handler (vl_api_lb_vip_dump_t * mp) { vl_api_registration_t *reg; reg = vl_api_client_index_to_registration (mp->client_index); if (!reg) return; lb_main_t *lbm = &lb_main; vl_api_lb_vip_details_t * rmp; int msg_size = 0; lb_vip_t *vip = 0; /* construct vip list */ pool_foreach(vip, lbm->vips, { /* Hide placeholder VIP */ if (vip != lbm->vips) { msg_size = sizeof (*rmp); rmp = vl_msg_api_alloc (msg_size); memset (rmp, 0, msg_size); rmp->_vl_msg_id = htons (VL_API_LB_VIP_DETAILS + lbm->msg_id_base); rmp->context = mp->context; ip_address_encode(&vip->prefix, IP46_TYPE_ANY, &rmp->vip.pfx.address); rmp->vip.pfx.len = vip->plen; rmp->vip.protocol = htonl (vip->protocol); rmp->vip.port = htons(vip->port); rmp->encap = htonl(vip->type); rmp->dscp = vip->encap_args.dscp; rmp->srv_type = vip->encap_args.srv_type; rmp->target_port = htons(vip->encap_args.target_port); rmp->flow_table_length = htonl(vip->new_flow_table_mask + 1); vl_api_send_msg (reg, (u8 *) rmp); } }); } static void send_lb_as_details (vl_api_registration_t * reg, u32 context, lb_vip_t * vip) { vl_api_lb_as_details_t *rmp; lb_main_t *lbm = &lb_main; int msg_size = 0; u32 *as_index; u32 asindex = 0; /* construct as list under this vip */ lb_as_t *as; pool_foreach(as_index, vip->as_indexes, { /* Hide placeholder As for specific VIP */ if (*as_index != 0) { as = &lbm->ass[*as_index]; msg_size = sizeof (*rmp); rmp = vl_msg_api_alloc (msg_size); memset (rmp, 0, msg_size); rmp->_vl_msg_id = htons (VL_API_LB_AS_DETAILS + lbm->msg_id_base); rmp->context = context; ip_address_encode(&vip->prefix, IP46_TYPE_ANY, (vl_api_address_t *)&rmp->vip.pfx.address); rmp->vip.pfx.len = vip->plen; rmp->vip.protocol = htonl (vip->protocol); rmp->vip.port = htons(vip->port); ip_address_encode(&as->address, IP46_TYPE_ANY, &rmp->app_srv); rmp->flags = as->flags; rmp->in_use_since = htonl(as->last_used); vl_api_send_msg (reg, (u8 *) rmp); asindex++; } }); } static void vl_api_lb_as_dump_t_handler (vl_api_lb_as_dump_t * mp) { lb_main_t *lbm = &lb_main; lb_vip_t *vip = 0; u8 dump_all = 0; ip46_address_t prefix; vl_api_registration_t *reg; reg = vl_api_client_index_to_registration (mp->client_index); if (!reg) return; clib_memcpy(&prefix.ip6, mp->pfx.address.un.ip6, sizeof(mp->pfx.address.un.ip6)); dump_all = (prefix.ip6.as_u64[0] == 0) && (prefix.ip6.as_u64[1] == 0); /* *INDENT-OFF* */ pool_foreach(vip, lbm->vips, ({ if ( dump_all || ((prefix.as_u64[0] == vip->prefix.as_u64[0]) && (prefix.as_u64[1] == vip->prefix.as_u64[1]) && (mp->protocol == vip->protocol) && (mp->port == vip->port)) ) { send_lb_as_details(reg, mp->context, vip); } })); /* *INDENT-ON* */ } static void vl_api_lb_flush_vip_t_handler (vl_api_lb_flush_vip_t * mp) { lb_main_t *lbm = &lb_main; int rv = 0; ip46_address_t vip_prefix; u8 vip_plen; u32 vip_index; vl_api_lb_flush_vip_reply_t * rmp; if (mp->port == 0) { mp->protocol = ~0; } memcpy (&(vip_prefix.ip6), mp->pfx.address.un.ip6, sizeof(vip_prefix.ip6)); vip_plen = mp->pfx.len; rv = lb_vip_find_index(&vip_prefix, vip_plen, mp->protocol, ntohs(mp->port), &vip_index); rv = lb_flush_vip_as(vip_index, ~0); REPLY_MACRO (VL_API_LB_FLUSH_VIP_REPLY); } static void vl_api_lb_add_del_intf_nat4_t_handler (vl_api_lb_add_del_intf_nat4_t * mp) { lb_main_t *lbm = &lb_main; vl_api_lb_add_del_intf_nat4_reply_t *rmp; u32 sw_if_index = ntohl (mp->sw_if_index); u8 is_del; int rv = 0; is_del = !mp->is_add; VALIDATE_SW_IF_INDEX (mp); rv = lb_nat4_interface_add_del(sw_if_index, is_del); BAD_SW_IF_INDEX_LABEL; REPLY_MACRO (VL_API_LB_ADD_DEL_INTF_NAT4_REPLY); } static void vl_api_lb_add_del_intf_nat6_t_handler (vl_api_lb_add_del_intf_nat6_t * mp) { lb_main_t *lbm = &lb_main; vl_api_lb_add_del_intf_nat6_reply_t *rmp; u32 sw_if_index = ntohl (mp->sw_if_index); u8 is_del; int rv = 0; is_del = !mp->is_add; VALIDATE_SW_IF_INDEX (mp); rv = lb_nat6_interface_add_del(sw_if_index, is_del); BAD_SW_IF_INDEX_LABEL; REPLY_MACRO (VL_API_LB_ADD_DEL_INTF_NAT6_REPLY); } static void *vl_api_lb_flush_vip_t_print (vl_api_lb_flush_vip_t *mp, void * handle) { u8 * s; s = format (0, "SCRIPT: lb_add_del_vip "); s = format (s, "%U/%d", format_vl_api_address, &mp->pfx.address, mp->pfx.len); s = format (s, "protocol %u ", mp->protocol); s = format (s, "port %u ", mp->port); FINISH; } #include <lb/lb.api.c> static clib_error_t * lb_api_init (vlib_main_t * vm) { lb_main_t * lbm = &lb_main; lbm->vlib_main = vm; lbm->vnet_main = vnet_get_main(); /* Ask for a correctly-sized block of API message decode slots */ lbm->msg_id_base = setup_message_id_table (); return 0; } VLIB_INIT_FUNCTION (lb_api_init);