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>
34 #define REPLY_MSG_ID_BASE lbm->msg_id_base
35 #include <vlibapi/api_helper_macros.h>
39 vlib_cli_output (handle, (char *) s); \
44 vl_api_lb_conf_t_handler
45 (vl_api_lb_conf_t * mp)
47 lb_main_t *lbm = &lb_main;
48 vl_api_lb_conf_reply_t * rmp;
49 u32 sticky_buckets_per_core, flow_timeout;
52 sticky_buckets_per_core = mp->sticky_buckets_per_core == ~0
53 ? lbm->per_cpu_sticky_buckets
54 : ntohl(mp->sticky_buckets_per_core);
55 flow_timeout = mp->flow_timeout == ~0
57 : ntohl(mp->flow_timeout);
59 rv = lb_conf((ip4_address_t *)&mp->ip4_src_address,
60 (ip6_address_t *)&mp->ip6_src_address,
61 sticky_buckets_per_core, flow_timeout);
63 REPLY_MACRO (VL_API_LB_CONF_REPLY);
67 vl_api_lb_add_del_vip_t_handler
68 (vl_api_lb_add_del_vip_t * mp)
70 lb_main_t *lbm = &lb_main;
71 vl_api_lb_conf_reply_t * rmp;
73 lb_vip_add_args_t args;
75 /* if port == 0, it means all-port VIP */
81 ip_address_decode (&mp->pfx.address, &(args.prefix));
85 if (!(rv = lb_vip_find_index(&(args.prefix), mp->pfx.len,
86 mp->protocol, ntohs(mp->port), &vip_index)))
87 rv = lb_vip_del(vip_index);
90 lb_vip_type_t type = 0;
92 if (ip46_prefix_is_ip4(&(args.prefix), mp->pfx.len)) {
93 if (mp->encap == LB_API_ENCAP_TYPE_GRE4)
94 type = LB_VIP_TYPE_IP4_GRE4;
95 else if (mp->encap == LB_API_ENCAP_TYPE_GRE6)
96 type = LB_VIP_TYPE_IP4_GRE6;
97 else if (mp->encap == LB_API_ENCAP_TYPE_L3DSR)
98 type = LB_VIP_TYPE_IP4_L3DSR;
99 else if (mp->encap == LB_API_ENCAP_TYPE_NAT4)
100 type = LB_VIP_TYPE_IP4_NAT4;
102 if (mp->encap == LB_API_ENCAP_TYPE_GRE4)
103 type = LB_VIP_TYPE_IP6_GRE4;
104 else if (mp->encap == LB_API_ENCAP_TYPE_GRE6)
105 type = LB_VIP_TYPE_IP6_GRE6;
106 else if (mp->encap == LB_API_ENCAP_TYPE_NAT6)
107 type = LB_VIP_TYPE_IP6_NAT6;
110 args.plen = mp->pfx.len;
111 args.protocol = mp->protocol;
112 args.port = ntohs(mp->port);
114 args.new_length = ntohl(mp->new_flows_table_length);
116 if (mp->encap == LB_API_ENCAP_TYPE_L3DSR) {
117 args.encap_args.dscp = (u8)(mp->dscp & 0x3F);
119 else if ((mp->encap == LB_API_ENCAP_TYPE_NAT4)
120 ||(mp->encap == LB_API_ENCAP_TYPE_NAT6)) {
121 args.encap_args.srv_type = mp->type;
122 args.encap_args.target_port = ntohs(mp->target_port);
125 rv = lb_vip_add(args, &vip_index);
127 REPLY_MACRO (VL_API_LB_ADD_DEL_VIP_REPLY);
131 vl_api_lb_add_del_as_t_handler
132 (vl_api_lb_add_del_as_t * mp)
134 lb_main_t *lbm = &lb_main;
135 vl_api_lb_conf_reply_t * rmp;
138 ip46_address_t vip_ip_prefix;
139 ip46_address_t as_address;
141 /* if port == 0, it means all-port VIP */
146 ip_address_decode (&mp->pfx.address, &vip_ip_prefix);
147 ip_address_decode (&mp->as_address, &as_address);
149 if ((rv = lb_vip_find_index(&vip_ip_prefix, mp->pfx.len,
150 mp->protocol, ntohs(mp->port), &vip_index)))
154 rv = lb_vip_del_ass(vip_index, &as_address, 1, mp->is_flush);
156 rv = lb_vip_add_ass(vip_index, &as_address, 1);
159 REPLY_MACRO (VL_API_LB_ADD_DEL_AS_REPLY);
163 vl_api_lb_vip_dump_t_handler
164 (vl_api_lb_vip_dump_t * mp)
167 vl_api_registration_t *reg;
168 reg = vl_api_client_index_to_registration (mp->client_index);
172 lb_main_t *lbm = &lb_main;
173 vl_api_lb_vip_details_t * rmp;
177 /* construct vip list */
178 pool_foreach (vip, lbm->vips) {
179 /* Hide placeholder VIP */
180 if (vip != lbm->vips) {
181 msg_size = sizeof (*rmp);
182 rmp = vl_msg_api_alloc (msg_size);
183 memset (rmp, 0, msg_size);
185 htons (VL_API_LB_VIP_DETAILS + lbm->msg_id_base);
186 rmp->context = mp->context;
188 ip_address_encode(&vip->prefix, IP46_TYPE_ANY, &rmp->vip.pfx.address);
189 rmp->vip.pfx.len = vip->plen;
190 rmp->vip.protocol = htonl (vip->protocol);
191 rmp->vip.port = htons(vip->port);
192 rmp->encap = htonl(vip->type);
193 rmp->dscp = vip->encap_args.dscp;
194 rmp->srv_type = vip->encap_args.srv_type;
195 rmp->target_port = htons(vip->encap_args.target_port);
196 rmp->flow_table_length = htonl(vip->new_flow_table_mask + 1);
198 vl_api_send_msg (reg, (u8 *) rmp);
205 static void send_lb_as_details
206 (vl_api_registration_t * reg, u32 context, lb_vip_t * vip)
208 vl_api_lb_as_details_t *rmp;
209 lb_main_t *lbm = &lb_main;
214 /* construct as list under this vip */
217 pool_foreach (as_index, vip->as_indexes) {
218 /* Hide placeholder As for specific VIP */
219 if (*as_index != 0) {
220 as = &lbm->ass[*as_index];
221 msg_size = sizeof (*rmp);
222 rmp = vl_msg_api_alloc (msg_size);
223 memset (rmp, 0, msg_size);
225 htons (VL_API_LB_AS_DETAILS + lbm->msg_id_base);
226 rmp->context = context;
227 ip_address_encode(&vip->prefix, IP46_TYPE_ANY, (vl_api_address_t *)&rmp->vip.pfx.address);
228 rmp->vip.pfx.len = vip->plen;
229 rmp->vip.protocol = htonl (vip->protocol);
230 rmp->vip.port = htons(vip->port);
231 ip_address_encode(&as->address, IP46_TYPE_ANY, &rmp->app_srv);
232 rmp->flags = as->flags;
233 rmp->in_use_since = htonl(as->last_used);
235 vl_api_send_msg (reg, (u8 *) rmp);
244 vl_api_lb_as_dump_t_handler
245 (vl_api_lb_as_dump_t * mp)
247 lb_main_t *lbm = &lb_main;
250 ip46_address_t prefix;
252 vl_api_registration_t *reg;
253 reg = vl_api_client_index_to_registration (mp->client_index);
257 clib_memcpy(&prefix.ip6, mp->pfx.address.un.ip6, sizeof(mp->pfx.address.un.ip6));
259 dump_all = (prefix.ip6.as_u64[0] == 0) && (prefix.ip6.as_u64[1] == 0);
262 pool_foreach (vip, lbm->vips)
265 || ((prefix.as_u64[0] == vip->prefix.as_u64[0])
266 && (prefix.as_u64[1] == vip->prefix.as_u64[1])
267 && (mp->protocol == vip->protocol)
268 && (mp->port == vip->port)) )
270 send_lb_as_details(reg, mp->context, vip);
277 vl_api_lb_flush_vip_t_handler
278 (vl_api_lb_flush_vip_t * mp)
280 lb_main_t *lbm = &lb_main;
282 ip46_address_t vip_prefix;
285 vl_api_lb_flush_vip_reply_t * rmp;
292 memcpy (&(vip_prefix.ip6), mp->pfx.address.un.ip6, sizeof(vip_prefix.ip6));
294 vip_plen = mp->pfx.len;
296 rv = lb_vip_find_index(&vip_prefix, vip_plen, mp->protocol,
297 ntohs(mp->port), &vip_index);
299 rv = lb_flush_vip_as(vip_index, ~0);
301 REPLY_MACRO (VL_API_LB_FLUSH_VIP_REPLY);
304 static void vl_api_lb_add_del_intf_nat4_t_handler
305 (vl_api_lb_add_del_intf_nat4_t * mp)
307 lb_main_t *lbm = &lb_main;
308 vl_api_lb_add_del_intf_nat4_reply_t *rmp;
309 u32 sw_if_index = ntohl (mp->sw_if_index);
313 is_del = !mp->is_add;
315 VALIDATE_SW_IF_INDEX (mp);
317 rv = lb_nat4_interface_add_del(sw_if_index, is_del);
319 BAD_SW_IF_INDEX_LABEL;
321 REPLY_MACRO (VL_API_LB_ADD_DEL_INTF_NAT4_REPLY);
324 static void vl_api_lb_add_del_intf_nat6_t_handler
325 (vl_api_lb_add_del_intf_nat6_t * mp)
327 lb_main_t *lbm = &lb_main;
328 vl_api_lb_add_del_intf_nat6_reply_t *rmp;
329 u32 sw_if_index = ntohl (mp->sw_if_index);
333 is_del = !mp->is_add;
335 VALIDATE_SW_IF_INDEX (mp);
337 rv = lb_nat6_interface_add_del(sw_if_index, is_del);
339 BAD_SW_IF_INDEX_LABEL;
341 REPLY_MACRO (VL_API_LB_ADD_DEL_INTF_NAT6_REPLY);
344 #include <lb/lb.api.c>
345 static clib_error_t * lb_api_init (vlib_main_t * vm)
347 lb_main_t * lbm = &lb_main;
350 lbm->vnet_main = vnet_get_main();
352 /* Ask for a correctly-sized block of API message decode slots */
353 lbm->msg_id_base = setup_message_id_table ();
358 VLIB_INIT_FUNCTION (lb_api_init);