2 * Copyright (c) 2016 Cisco and/or its affiliates.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at:
7 * http://www.apache.org/licenses/LICENSE-2.0
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
16 #include <vnet/vnet.h>
17 #include <vnet/plugin/plugin.h>
20 #include <vppinfra/byte_order.h>
21 #include <vppinfra/string.h>
22 #include <vpp/api/types.h>
23 #include <vlibapi/api.h>
24 #include <vlibmemory/api.h>
25 #include <vpp/app/version.h>
26 #include <vnet/format_fns.h>
27 #include <vnet/ip/ip_types_api.h>
29 /* define message IDs */
30 #include <lb/lb.api_enum.h>
31 #include <lb/lb.api_types.h>
33 #define vl_print(handle, ...) vlib_cli_output (handle, __VA_ARGS__)
35 #define REPLY_MSG_ID_BASE lbm->msg_id_base
36 #include <vlibapi/api_helper_macros.h>
38 /* Macro to finish up custom dump fns */
41 vl_print (handle, (char *)s); \
46 vl_api_lb_conf_t_handler
47 (vl_api_lb_conf_t * mp)
49 lb_main_t *lbm = &lb_main;
50 vl_api_lb_conf_reply_t * rmp;
51 u32 sticky_buckets_per_core, flow_timeout;
54 sticky_buckets_per_core = mp->sticky_buckets_per_core == ~0
55 ? lbm->per_cpu_sticky_buckets
56 : ntohl(mp->sticky_buckets_per_core);
57 flow_timeout = mp->flow_timeout == ~0
59 : ntohl(mp->flow_timeout);
61 rv = lb_conf((ip4_address_t *)&mp->ip4_src_address,
62 (ip6_address_t *)&mp->ip6_src_address,
63 sticky_buckets_per_core, flow_timeout);
65 REPLY_MACRO (VL_API_LB_CONF_REPLY);
68 static void *vl_api_lb_conf_t_print
69 (vl_api_lb_conf_t *mp, void * handle)
72 s = format (0, "SCRIPT: lb_conf ");
73 s = format (s, "%U ", format_ip4_address, (ip4_address_t *)&mp->ip4_src_address);
74 s = format (s, "%U ", format_ip6_address, (ip6_address_t *)&mp->ip6_src_address);
75 s = format (s, "%u ", mp->sticky_buckets_per_core);
76 s = format (s, "%u ", mp->flow_timeout);
82 vl_api_lb_add_del_vip_t_handler
83 (vl_api_lb_add_del_vip_t * mp)
85 lb_main_t *lbm = &lb_main;
86 vl_api_lb_conf_reply_t * rmp;
88 lb_vip_add_args_t args;
90 /* if port == 0, it means all-port VIP */
96 memcpy (&(args.prefix.ip6), &mp->pfx.address.un.ip6, sizeof(args.prefix.ip6));
100 if (!(rv = lb_vip_find_index(&(args.prefix), mp->pfx.len,
101 mp->protocol, ntohs(mp->port), &vip_index)))
102 rv = lb_vip_del(vip_index);
105 lb_vip_type_t type = 0;
107 if (ip46_prefix_is_ip4(&(args.prefix), mp->pfx.len)) {
108 if (mp->encap == LB_API_ENCAP_TYPE_GRE4)
109 type = LB_VIP_TYPE_IP4_GRE4;
110 else if (mp->encap == LB_API_ENCAP_TYPE_GRE6)
111 type = LB_VIP_TYPE_IP4_GRE6;
112 else if (mp->encap == LB_API_ENCAP_TYPE_L3DSR)
113 type = LB_VIP_TYPE_IP4_L3DSR;
114 else if (mp->encap == LB_API_ENCAP_TYPE_NAT4)
115 type = LB_VIP_TYPE_IP4_NAT4;
117 if (mp->encap == LB_API_ENCAP_TYPE_GRE4)
118 type = LB_VIP_TYPE_IP6_GRE4;
119 else if (mp->encap == LB_API_ENCAP_TYPE_GRE6)
120 type = LB_VIP_TYPE_IP6_GRE6;
121 else if (mp->encap == LB_API_ENCAP_TYPE_NAT6)
122 type = LB_VIP_TYPE_IP6_NAT6;
125 args.plen = mp->pfx.len;
126 args.protocol = mp->protocol;
127 args.port = ntohs(mp->port);
129 args.new_length = ntohl(mp->new_flows_table_length);
131 if (mp->encap == LB_API_ENCAP_TYPE_L3DSR) {
132 args.encap_args.dscp = (u8)(mp->dscp & 0x3F);
134 else if ((mp->encap == LB_API_ENCAP_TYPE_NAT4)
135 ||(mp->encap == LB_API_ENCAP_TYPE_NAT6)) {
136 args.encap_args.srv_type = mp->type;
137 args.encap_args.target_port = ntohs(mp->target_port);
140 rv = lb_vip_add(args, &vip_index);
142 REPLY_MACRO (VL_API_LB_ADD_DEL_VIP_REPLY);
145 static void *vl_api_lb_add_del_vip_t_print
146 (vl_api_lb_add_del_vip_t *mp, void * handle)
149 s = format (0, "SCRIPT: lb_add_del_vip ");
150 s = format (s, "%U", format_vl_api_prefix,
153 s = format (s, "%s ", (mp->encap == LB_API_ENCAP_TYPE_GRE4)? "gre4"
154 : (mp->encap == LB_API_ENCAP_TYPE_GRE6)? "gre6"
155 : (mp->encap == LB_API_ENCAP_TYPE_NAT4)? "nat4"
156 : (mp->encap == LB_API_ENCAP_TYPE_NAT6)? "nat6"
159 if (mp->encap==LB_API_ENCAP_TYPE_L3DSR)
161 s = format (s, "dscp %u ", mp->dscp);
164 if ((mp->encap==LB_API_ENCAP_TYPE_NAT4)
165 || (mp->encap==LB_API_ENCAP_TYPE_NAT6))
167 s = format (s, "type %u ", mp->type);
168 s = format (s, "port %u ", mp->port);
169 s = format (s, "target_port %u ", mp->target_port);
172 s = format (s, "%u ", mp->new_flows_table_length);
173 s = format (s, "%s ", mp->is_del?"del":"add");
178 vl_api_lb_add_del_as_t_handler
179 (vl_api_lb_add_del_as_t * mp)
181 lb_main_t *lbm = &lb_main;
182 vl_api_lb_conf_reply_t * rmp;
185 ip46_address_t vip_ip_prefix;
187 /* if port == 0, it means all-port VIP */
193 memcpy(&vip_ip_prefix.ip6, &mp->pfx.address.un.ip6,
194 sizeof(vip_ip_prefix.ip6));
196 ip46_address_t as_address;
198 memcpy(&as_address.ip6, &mp->as_address.un.ip6,
199 sizeof(as_address.ip6));
201 if ((rv = lb_vip_find_index(&vip_ip_prefix, mp->pfx.len,
202 mp->protocol, ntohs(mp->port), &vip_index)))
206 rv = lb_vip_del_ass(vip_index, &as_address, 1, mp->is_flush);
208 rv = lb_vip_add_ass(vip_index, &as_address, 1);
211 REPLY_MACRO (VL_API_LB_ADD_DEL_AS_REPLY);
214 static void *vl_api_lb_add_del_as_t_print
215 (vl_api_lb_add_del_as_t *mp, void * handle)
218 ip46_address_t address;
219 s = format (0, "SCRIPT: lb_add_del_as ");
220 s = format (s, "%U ", format_vl_api_prefix,
222 s = format(s, "%u ", mp->protocol);
223 if (ip_address_decode (&mp->as_address, &address) == IP46_TYPE_IP6)
224 s = format (s, "%U ", format_ip6_address,
225 (ip6_address_t *) & address.ip6);
227 s = format (s, "%U ", format_ip4_address,
228 (ip6_address_t *) & address.ip4);
229 s = format (s, "%s ", mp->is_del?"del":"add");
234 vl_api_lb_vip_dump_t_handler
235 (vl_api_lb_vip_dump_t * mp)
238 vl_api_registration_t *reg;
239 reg = vl_api_client_index_to_registration (mp->client_index);
243 lb_main_t *lbm = &lb_main;
244 vl_api_lb_vip_details_t * rmp;
248 /* construct vip list */
249 pool_foreach(vip, lbm->vips, {
251 if (vip != lbm->vips) {
252 msg_size = sizeof (*rmp);
253 rmp = vl_msg_api_alloc (msg_size);
254 memset (rmp, 0, msg_size);
256 htons (VL_API_LB_VIP_DETAILS + lbm->msg_id_base);
257 rmp->context = mp->context;
259 ip_address_encode(&vip->prefix, IP46_TYPE_ANY, &rmp->vip.pfx.address);
260 rmp->vip.pfx.len = vip->plen;
261 rmp->vip.protocol = htonl (vip->protocol);
262 rmp->vip.port = htons(vip->port);
263 rmp->encap = htonl(vip->type);
264 rmp->dscp = vip->encap_args.dscp;
265 rmp->srv_type = vip->encap_args.srv_type;
266 rmp->target_port = htons(vip->encap_args.target_port);
267 rmp->flow_table_length = htonl(vip->new_flow_table_mask + 1);
269 vl_api_send_msg (reg, (u8 *) rmp);
276 static void send_lb_as_details
277 (vl_api_registration_t * reg, u32 context, lb_vip_t * vip)
279 vl_api_lb_as_details_t *rmp;
280 lb_main_t *lbm = &lb_main;
285 /* construct as list under this vip */
288 pool_foreach(as_index, vip->as_indexes, {
289 /* Hide dummy As for specific VIP */
290 if (*as_index != 0) {
291 as = &lbm->ass[*as_index];
292 msg_size = sizeof (*rmp);
293 rmp = vl_msg_api_alloc (msg_size);
294 memset (rmp, 0, msg_size);
296 htons (VL_API_LB_AS_DETAILS + lbm->msg_id_base);
297 rmp->context = context;
298 ip_address_encode(&vip->prefix, IP46_TYPE_ANY, (vl_api_address_t *)&rmp->vip.pfx.address);
299 rmp->vip.pfx.len = vip->plen;
300 rmp->vip.protocol = htonl (vip->protocol);
301 rmp->vip.port = htons(vip->port);
302 ip_address_encode(&as->address, IP46_TYPE_ANY, &rmp->app_srv);
303 rmp->flags = as->flags;
304 rmp->in_use_since = htonl(as->last_used);
306 vl_api_send_msg (reg, (u8 *) rmp);
315 vl_api_lb_as_dump_t_handler
316 (vl_api_lb_as_dump_t * mp)
318 lb_main_t *lbm = &lb_main;
321 ip46_address_t prefix;
323 vl_api_registration_t *reg;
324 reg = vl_api_client_index_to_registration (mp->client_index);
328 clib_memcpy(&prefix.ip6, mp->pfx.address.un.ip6, sizeof(mp->pfx.address.un.ip6));
330 dump_all = (prefix.ip6.as_u64[0] == 0) && (prefix.ip6.as_u64[1] == 0);
333 pool_foreach(vip, lbm->vips,
336 || ((prefix.as_u64[0] == vip->prefix.as_u64[0])
337 && (prefix.as_u64[1] == vip->prefix.as_u64[1])
338 && (mp->protocol == vip->protocol)
339 && (mp->port == vip->port)) )
341 send_lb_as_details(reg, mp->context, vip);
348 vl_api_lb_flush_vip_t_handler
349 (vl_api_lb_flush_vip_t * mp)
351 lb_main_t *lbm = &lb_main;
353 ip46_address_t vip_prefix;
356 vl_api_lb_flush_vip_reply_t * rmp;
363 memcpy (&(vip_prefix.ip6), mp->pfx.address.un.ip6, sizeof(vip_prefix.ip6));
365 vip_plen = mp->pfx.len;
367 rv = lb_vip_find_index(&vip_prefix, vip_plen, mp->protocol,
368 ntohs(mp->port), &vip_index);
370 rv = lb_flush_vip_as(vip_index, ~0);
372 REPLY_MACRO (VL_API_LB_FLUSH_VIP_REPLY);
375 static void vl_api_lb_add_del_intf_nat4_t_handler
376 (vl_api_lb_add_del_intf_nat4_t * mp)
378 lb_main_t *lbm = &lb_main;
379 vl_api_lb_add_del_intf_nat4_reply_t *rmp;
380 u32 sw_if_index = ntohl (mp->sw_if_index);
384 is_del = !mp->is_add;
386 VALIDATE_SW_IF_INDEX (mp);
388 rv = lb_nat4_interface_add_del(sw_if_index, is_del);
390 BAD_SW_IF_INDEX_LABEL;
392 REPLY_MACRO (VL_API_LB_ADD_DEL_INTF_NAT4_REPLY);
395 static void vl_api_lb_add_del_intf_nat6_t_handler
396 (vl_api_lb_add_del_intf_nat6_t * mp)
398 lb_main_t *lbm = &lb_main;
399 vl_api_lb_add_del_intf_nat6_reply_t *rmp;
400 u32 sw_if_index = ntohl (mp->sw_if_index);
404 is_del = !mp->is_add;
406 VALIDATE_SW_IF_INDEX (mp);
408 rv = lb_nat6_interface_add_del(sw_if_index, is_del);
410 BAD_SW_IF_INDEX_LABEL;
412 REPLY_MACRO (VL_API_LB_ADD_DEL_INTF_NAT6_REPLY);
415 static void *vl_api_lb_flush_vip_t_print
416 (vl_api_lb_flush_vip_t *mp, void * handle)
419 s = format (0, "SCRIPT: lb_add_del_vip ");
420 s = format (s, "%U/%d", format_vl_api_address,
421 &mp->pfx.address, mp->pfx.len);
422 s = format (s, "protocol %u ", mp->protocol);
423 s = format (s, "port %u ", mp->port);
428 #include <lb/lb.api.c>
429 static clib_error_t * lb_api_init (vlib_main_t * vm)
431 lb_main_t * lbm = &lb_main;
434 lbm->vnet_main = vnet_get_main();
436 /* Ask for a correctly-sized block of API message decode slots */
437 lbm->msg_id_base = setup_message_id_table ();
442 VLIB_INIT_FUNCTION (lb_api_init);