* limitations under the License.
*/
+#include <vnet/vnet.h>
+#include <vnet/plugin/plugin.h>
#include <lb/lb.h>
#include <vppinfra/byte_order.h>
-#include <vlibapi/api.h>
+#include <vppinfra/string.h>
+#include <vpp/api/types.h>
#include <vlibapi/api.h>
#include <vlibmemory/api.h>
-#include <vlibsocket/api.h>
-
-#define vl_msg_id(n,h) n,
-typedef enum {
-#include <lb/lb.api.h>
- /* We'll want to know how many messages IDs we need... */
- VL_MSG_FIRST_AVAILABLE,
-} vl_msg_id_t;
-#undef vl_msg_id
-
+#include <vpp/app/version.h>
+#include <vnet/format_fns.h>
+#include <vnet/ip/ip_types_api.h>
-/* define message structures */
-#define vl_typedefs
-#include <lb/lb.api.h>
-#undef vl_typedefs
-
-/* define generated endian-swappers */
-#define vl_endianfun
-#include <lb/lb.api.h>
-#undef vl_endianfun
+/* 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__)
-/* Get the API version number */
-#define vl_api_version(n,v) static u32 api_version=(v);
-#include <lb/lb.api.h>
-#undef vl_api_version
-
-#define vl_msg_name_crc_list
-#include <lb/lb.api.h>
-#undef vl_msg_name_crc_list
-
-static void
-setup_message_id_table (lb_main_t * lbm, api_main_t * am)
-{
-#define _(id,n,crc) \
- vl_msg_api_add_msg_name_crc (am, #n "_" #crc, id + lbm->msg_id_base);
- foreach_vl_msg_name_crc_lb;
-#undef _
-}
+#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_free (s); \
return handle;
-/*
- * A handy macro to set up a message reply.
- * Assumes that the following variables are available:
- * mp - pointer to request message
- * rmp - pointer to reply message type
- * rv - return value
- */
-
-#define REPLY_MACRO(t) \
-do { \
- unix_shared_memory_queue_t * q = \
- vl_api_client_index_to_input_queue (mp->client_index); \
- if (!q) \
- return; \
- \
- rmp = vl_msg_api_alloc (sizeof (*rmp)); \
- rmp->_vl_msg_id = ntohs((t)+lbm->msg_id_base); \
- rmp->context = mp->context; \
- rmp->retval = ntohl(rv); \
- \
- vl_msg_api_send_shmem (q, (u8 *)&rmp); \
-} while(0);
-
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,
- mp->sticky_buckets_per_core,
- mp->flow_timeout);
+ (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;
- ip46_address_t prefix;
- memcpy(&prefix.ip6, mp->ip_prefix, sizeof(prefix.ip6));
+ 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(&prefix, mp->prefix_length, &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;
- if (ip46_prefix_is_ip4(&prefix, mp->prefix_length)) {
- type = mp->is_gre4?LB_VIP_TYPE_IP4_GRE4:LB_VIP_TYPE_IP4_GRE6;
+ 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 {
- type = mp->is_gre4?LB_VIP_TYPE_IP6_GRE4:LB_VIP_TYPE_IP6_GRE6;
+ 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;
}
- rv = lb_vip_add(&prefix, mp->prefix_length, type,
- mp->new_flows_table_length, &vip_index);
+ 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_CONF_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_ip46_prefix,
- (ip46_address_t *)mp->ip_prefix, mp->prefix_length, IP46_TYPE_ANY);
- s = format (s, "%s ", mp->is_gre4?"gre4":"gre6");
- s = format (s, "%u ", mp->new_flows_table_length);
- s = format (s, "%s ", mp->is_del?"del":"add");
- FINISH;
+ REPLY_MACRO (VL_API_LB_ADD_DEL_VIP_REPLY);
}
static void
vl_api_lb_conf_reply_t * rmp;
int rv = 0;
u32 vip_index;
- if ((rv = lb_vip_find_index((ip46_address_t *)mp->vip_ip_prefix,
- mp->vip_prefix_length, &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, (ip46_address_t *)mp->as_address, 1);
+ rv = lb_vip_del_ass(vip_index, &as_address, 1, mp->is_flush);
else
- rv = lb_vip_add_ass(vip_index, (ip46_address_t *)mp->as_address, 1);
+ rv = lb_vip_add_ass(vip_index, &as_address, 1);
done:
- REPLY_MACRO (VL_API_LB_CONF_REPLY);
+ 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)
+static void
+vl_api_lb_vip_dump_t_handler
+(vl_api_lb_vip_dump_t * mp)
{
- u8 * s;
- s = format (0, "SCRIPT: lb_add_del_as ");
- s = format (s, "%U ", format_ip46_prefix,
- (ip46_address_t *)mp->vip_ip_prefix, mp->vip_prefix_length, IP46_TYPE_ANY);
- s = format (s, "%U ", format_ip46_address,
- (ip46_address_t *)mp->as_address, IP46_TYPE_ANY);
- s = format (s, "%s ", mp->is_del?"del":"add");
- FINISH;
+
+ 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);
+ }
+ });
+
+
}
-/* List of message types that this plugin understands */
-#define foreach_lb_plugin_api_msg \
-_(LB_CONF, lb_conf) \
-_(LB_ADD_DEL_VIP, lb_add_del_vip) \
-_(LB_ADD_DEL_AS, lb_add_del_as)
+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 clib_error_t * lb_api_init (vlib_main_t * vm)
+}
+
+static void
+vl_api_lb_as_dump_t_handler
+(vl_api_lb_as_dump_t * mp)
{
lb_main_t *lbm = &lb_main;
- u8 *name = format (0, "lb_%08x%c", api_version, 0);
- lbm->msg_id_base = vl_msg_api_get_msg_ids
- ((char *) name, VL_MSG_FIRST_AVAILABLE);
-
-#define _(N,n) \
- vl_msg_api_set_handlers((VL_API_##N + lbm->msg_id_base), \
- #n, \
- vl_api_##n##_t_handler, \
- vl_noop_handler, \
- vl_api_##n##_t_endian, \
- vl_api_##n##_t_print, \
- sizeof(vl_api_##n##_t), 1);
- foreach_lb_plugin_api_msg;
-#undef _
-
- /* Add our API messages to the global name_crc hash table */
- setup_message_id_table (lbm, &api_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);
+}
+
+#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;
}