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.
20 lb_vip_command_fn (vlib_main_t * vm,
21 unformat_input_t * input, vlib_cli_command_t * cmd)
23 unformat_input_t _line_input, *line_input = &_line_input;
24 lb_vip_add_args_t args;
30 u32 srv_type = LB_SRV_TYPE_CLUSTERIP;
32 clib_error_t *error = 0;
34 args.new_length = 1024;
36 if (!unformat_user (input, unformat_line_input, line_input))
39 if (!unformat(line_input, "%U", unformat_ip46_prefix, &(args.prefix),
40 &(args.plen), IP46_TYPE_ANY, &(args.plen))) {
41 error = clib_error_return (0, "invalid vip prefix: '%U'",
42 format_unformat_error, line_input);
46 while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
48 if (unformat(line_input, "new_len %d", &(args.new_length)))
50 else if (unformat(line_input, "del"))
52 else if (unformat(line_input, "protocol tcp"))
54 args.protocol = (u8)IP_PROTOCOL_TCP;
56 else if (unformat(line_input, "protocol udp"))
58 args.protocol = (u8)IP_PROTOCOL_UDP;
60 else if (unformat(line_input, "port %d", &port))
62 else if (unformat(line_input, "encap gre4"))
63 encap = LB_ENCAP_TYPE_GRE4;
64 else if (unformat(line_input, "encap gre6"))
65 encap = LB_ENCAP_TYPE_GRE6;
66 else if (unformat(line_input, "encap l3dsr"))
67 encap = LB_ENCAP_TYPE_L3DSR;
68 else if (unformat(line_input, "encap nat4"))
69 encap = LB_ENCAP_TYPE_NAT4;
70 else if (unformat(line_input, "encap nat6"))
71 encap = LB_ENCAP_TYPE_NAT6;
72 else if (unformat(line_input, "dscp %d", &dscp))
74 else if (unformat(line_input, "type clusterip"))
75 srv_type = LB_SRV_TYPE_CLUSTERIP;
76 else if (unformat(line_input, "type nodeport"))
77 srv_type = LB_SRV_TYPE_NODEPORT;
78 else if (unformat(line_input, "target_port %d", &target_port))
81 error = clib_error_return (0, "parse error: '%U'",
82 format_unformat_error, line_input);
87 /* if port == 0, it means all-port VIP */
95 args.port = (u16)port;
98 if ((encap != LB_ENCAP_TYPE_L3DSR) && (dscp != ~0))
100 error = clib_error_return(0, "lb_vip_add error: "
101 "should not configure dscp for none L3DSR.");
105 if ((encap == LB_ENCAP_TYPE_L3DSR) && (dscp >= 64))
107 error = clib_error_return(0, "lb_vip_add error: "
108 "dscp for L3DSR should be less than 64.");
112 if (ip46_prefix_is_ip4(&(args.prefix), (args.plen)))
114 if (encap == LB_ENCAP_TYPE_GRE4)
115 args.type = LB_VIP_TYPE_IP4_GRE4;
116 else if (encap == LB_ENCAP_TYPE_GRE6)
117 args.type = LB_VIP_TYPE_IP4_GRE6;
118 else if (encap == LB_ENCAP_TYPE_L3DSR)
119 args.type = LB_VIP_TYPE_IP4_L3DSR;
120 else if (encap == LB_ENCAP_TYPE_NAT4)
121 args.type = LB_VIP_TYPE_IP4_NAT4;
122 else if (encap == LB_ENCAP_TYPE_NAT6)
124 error = clib_error_return(0, "currently does not support NAT46");
130 if (encap == LB_ENCAP_TYPE_GRE4)
131 args.type = LB_VIP_TYPE_IP6_GRE4;
132 else if (encap == LB_ENCAP_TYPE_GRE6)
133 args.type = LB_VIP_TYPE_IP6_GRE6;
134 else if (encap == LB_ENCAP_TYPE_NAT6)
135 args.type = LB_VIP_TYPE_IP6_NAT6;
136 else if (encap == LB_ENCAP_TYPE_NAT4)
138 error = clib_error_return(0, "currently does not support NAT64");
143 lb_garbage_collection();
147 if (encap == LB_ENCAP_TYPE_L3DSR) {
148 args.encap_args.dscp = (u8)(dscp & 0x3F);
150 else if ((encap == LB_ENCAP_TYPE_NAT4)
151 || (encap == LB_ENCAP_TYPE_NAT6))
153 args.encap_args.srv_type = (u8) srv_type;
154 args.encap_args.target_port = (u16) target_port;
157 if ((ret = lb_vip_add(args, &index))) {
158 error = clib_error_return (0, "lb_vip_add error %d", ret);
161 vlib_cli_output(vm, "lb_vip_add ok %d", index);
164 if ((ret = lb_vip_find_index(&(args.prefix), args.plen,
165 args.protocol, args.port, &index))) {
166 error = clib_error_return (0, "lb_vip_find_index error %d", ret);
168 } else if ((ret = lb_vip_del(index))) {
169 error = clib_error_return (0, "lb_vip_del error %d", ret);
175 unformat_free (line_input);
180 VLIB_CLI_COMMAND (lb_vip_command, static) =
183 .short_help = "lb vip <prefix> "
184 "[protocol (tcp|udp) port <n>] "
185 "[encap (gre6|gre4|l3dsr|nat4|nat6)] "
187 "[type (nodeport|clusterip) target_port <n>] "
188 "[new_len <n>] [del]",
189 .function = lb_vip_command_fn,
192 static clib_error_t *
193 lb_as_command_fn (vlib_main_t * vm,
194 unformat_input_t * input, vlib_cli_command_t * cmd)
196 unformat_input_t _line_input, *line_input = &_line_input;
197 ip46_address_t vip_prefix, as_addr;
199 ip46_address_t *as_array = 0;
206 clib_error_t *error = 0;
208 if (!unformat_user (input, unformat_line_input, line_input))
211 if (!unformat(line_input, "%U", unformat_ip46_prefix,
212 &vip_prefix, &vip_plen, IP46_TYPE_ANY))
214 error = clib_error_return (0, "invalid as address: '%U'",
215 format_unformat_error, line_input);
219 while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
221 if (unformat(line_input, "%U", unformat_ip46_address,
222 &as_addr, IP46_TYPE_ANY))
224 vec_add1(as_array, as_addr);
226 else if (unformat(line_input, "del"))
230 else if (unformat(line_input, "flush"))
234 else if (unformat(line_input, "protocol tcp"))
236 protocol = (u8)IP_PROTOCOL_TCP;
238 else if (unformat(line_input, "protocol udp"))
240 protocol = (u8)IP_PROTOCOL_UDP;
242 else if (unformat(line_input, "port %d", &port))
245 error = clib_error_return (0, "parse error: '%U'",
246 format_unformat_error, line_input);
251 /* If port == 0, it means all-port VIP */
257 if ((ret = lb_vip_find_index(&vip_prefix, vip_plen, protocol,
258 (u16)port, &vip_index))){
259 error = clib_error_return (0, "lb_vip_find_index error %d", ret);
263 if (!vec_len(as_array)) {
264 error = clib_error_return (0, "No AS address provided");
268 lb_garbage_collection();
269 clib_warning("vip index is %d", vip_index);
272 if ((ret = lb_vip_del_ass(vip_index, as_array, vec_len(as_array), flush)))
274 error = clib_error_return (0, "lb_vip_del_ass error %d", ret);
278 if ((ret = lb_vip_add_ass(vip_index, as_array, vec_len(as_array))))
280 error = clib_error_return (0, "lb_vip_add_ass error %d", ret);
286 unformat_free (line_input);
292 VLIB_CLI_COMMAND (lb_as_command, static) =
295 .short_help = "lb as <vip-prefix> [protocol (tcp|udp) port <n>]"
296 " [<address> [<address> [...]]] [del] [flush]",
297 .function = lb_as_command_fn,
300 static clib_error_t *
301 lb_conf_command_fn (vlib_main_t * vm,
302 unformat_input_t * input, vlib_cli_command_t * cmd)
304 lb_main_t *lbm = &lb_main;
305 unformat_input_t _line_input, *line_input = &_line_input;
306 ip4_address_t ip4 = lbm->ip4_src_address;
307 ip6_address_t ip6 = lbm->ip6_src_address;
308 u32 per_cpu_sticky_buckets = lbm->per_cpu_sticky_buckets;
309 u32 per_cpu_sticky_buckets_log2 = 0;
310 u32 flow_timeout = lbm->flow_timeout;
312 clib_error_t *error = 0;
314 if (!unformat_user (input, unformat_line_input, line_input))
317 while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
319 if (unformat(line_input, "ip4-src-address %U", unformat_ip4_address, &ip4))
321 else if (unformat(line_input, "ip6-src-address %U", unformat_ip6_address, &ip6))
323 else if (unformat(line_input, "buckets %d", &per_cpu_sticky_buckets))
325 else if (unformat(line_input, "buckets-log2 %d", &per_cpu_sticky_buckets_log2)) {
326 if (per_cpu_sticky_buckets_log2 >= 32)
327 return clib_error_return (0, "buckets-log2 value is too high");
328 per_cpu_sticky_buckets = 1 << per_cpu_sticky_buckets_log2;
329 } else if (unformat(line_input, "timeout %d", &flow_timeout))
332 error = clib_error_return (0, "parse error: '%U'",
333 format_unformat_error, line_input);
338 lb_garbage_collection();
340 if ((ret = lb_conf(&ip4, &ip6, per_cpu_sticky_buckets, flow_timeout))) {
341 error = clib_error_return (0, "lb_conf error %d", ret);
346 unformat_free (line_input);
351 VLIB_CLI_COMMAND (lb_conf_command, static) =
354 .short_help = "lb conf [ip4-src-address <addr>] [ip6-src-address <addr>] [buckets <n>] [timeout <s>]",
355 .function = lb_conf_command_fn,
358 static clib_error_t *
359 lb_show_command_fn (vlib_main_t * vm,
360 unformat_input_t * input, vlib_cli_command_t * cmd)
362 vlib_cli_output(vm, "%U", format_lb_main);
367 VLIB_CLI_COMMAND (lb_show_command, static) =
370 .short_help = "show lb",
371 .function = lb_show_command_fn,
374 static clib_error_t *
375 lb_show_vips_command_fn (vlib_main_t * vm,
376 unformat_input_t * input, vlib_cli_command_t * cmd)
378 unformat_input_t line_input;
379 lb_main_t *lbm = &lb_main;
383 if (!unformat_user (input, unformat_line_input, &line_input))
386 if (unformat(&line_input, "verbose"))
390 pool_foreach(vip, lbm->vips, {
391 if (vip != lbm->vips) {
392 vlib_cli_output(vm, "%U\n", verbose?format_lb_vip_detailed:format_lb_vip, vip);
396 unformat_free (&line_input);
400 VLIB_CLI_COMMAND (lb_show_vips_command, static) =
402 .path = "show lb vips",
403 .short_help = "show lb vips [verbose]",
404 .function = lb_show_vips_command_fn,
407 static clib_error_t *
408 lb_set_interface_nat_command_fn (vlib_main_t * vm,
409 unformat_input_t * input,
410 vlib_cli_command_t * cmd,
413 unformat_input_t _line_input, *line_input = &_line_input;
414 vnet_main_t * vnm = vnet_get_main();
415 clib_error_t * error = 0;
416 u32 _sw_if_index, *sw_if_index = &_sw_if_index;
417 u32 * inside_sw_if_indices = 0;
420 /* Get a line of input. */
421 if (!unformat_user (input, unformat_line_input, line_input))
424 while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
426 if (unformat (line_input, "in %U", unformat_vnet_sw_interface,
428 vec_add1 (inside_sw_if_indices, *sw_if_index);
429 else if (unformat (line_input, "del"))
433 error = clib_error_return (0, "unknown input '%U'",
434 format_unformat_error, line_input);
439 vec_foreach (sw_if_index, inside_sw_if_indices)
443 if (lb_nat4_interface_add_del (*sw_if_index, is_del))
445 error = clib_error_return(
446 0, "%s %U failed", is_del ? "del" : "add",
447 format_vnet_sw_interface_name, vnm,
448 vnet_get_sw_interface (vnm, *sw_if_index));
454 if (lb_nat6_interface_add_del (*sw_if_index, is_del))
456 error = clib_error_return(
457 0, "%s %U failed", is_del ? "del" : "add",
458 format_vnet_sw_interface_name, vnm,
459 vnet_get_sw_interface (vnm, *sw_if_index));
466 unformat_free (line_input);
467 vec_free (inside_sw_if_indices);
472 static clib_error_t *
473 lb_set_interface_nat4_command_fn (vlib_main_t * vm,
474 unformat_input_t * input,
475 vlib_cli_command_t * cmd)
477 return lb_set_interface_nat_command_fn(vm, input, cmd, 0);
480 VLIB_CLI_COMMAND (lb_set_interface_nat4_command, static) = {
481 .path = "lb set interface nat4",
482 .function = lb_set_interface_nat4_command_fn,
483 .short_help = "lb set interface nat4 in <intfc> [del]",
486 static clib_error_t *
487 lb_set_interface_nat6_command_fn (vlib_main_t * vm,
488 unformat_input_t * input,
489 vlib_cli_command_t * cmd)
491 return lb_set_interface_nat_command_fn(vm, input, cmd, 1);
494 VLIB_CLI_COMMAND (lb_set_interface_nat6_command, static) = {
495 .path = "lb set interface nat6",
496 .function = lb_set_interface_nat6_command_fn,
497 .short_help = "lb set interface nat6 in <intfc> [del]",
500 static clib_error_t *
501 lb_flowtable_flush_command_fn (vlib_main_t * vm,
502 unformat_input_t * input, vlib_cli_command_t * cmd)
504 lb_flush_vip_as(~0, 0);
509 static clib_error_t *
510 lb_flush_vip_command_fn (vlib_main_t * vm,
511 unformat_input_t * input,
512 vlib_cli_command_t * cmd)
514 unformat_input_t _line_input, *line_input = &_line_input;
516 ip46_address_t vip_prefix;
521 clib_error_t *error = 0;
523 if (!unformat_user (input, unformat_line_input, line_input))
526 if (!unformat(line_input, "%U", unformat_ip46_prefix, &vip_prefix,
527 &vip_plen, IP46_TYPE_ANY, &vip_plen)) {
528 error = clib_error_return (0, "invalid vip prefix: '%U'",
529 format_unformat_error, line_input);
533 while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
535 if (unformat(line_input, "protocol tcp"))
537 protocol = (u8)IP_PROTOCOL_TCP;
539 else if (unformat(line_input, "protocol udp"))
541 protocol = (u8)IP_PROTOCOL_UDP;
543 else if (unformat(line_input, "port %d", &port))
552 if ((ret = lb_vip_find_index(&vip_prefix, vip_plen, protocol,
553 (u16)port, &vip_index))){
554 error = clib_error_return (0, "lb_vip_find_index error %d", ret);
558 if ((ret = lb_flush_vip_as(vip_index, ~0)))
560 error = clib_error_return (0, "lb_flush_vip error %d", ret);
564 vlib_cli_output(vm, "lb_flush_vip ok %d", vip_index);
568 unformat_free (line_input);
574 * flush lb flowtable as per vip
576 VLIB_CLI_COMMAND (lb_flush_vip_command, static) =
578 .path = "lb flush vip",
579 .short_help = "lb flush vip <prefix> "
580 "[protocol (tcp|udp) port <n>]",
581 .function = lb_flush_vip_command_fn,
585 * flush all lb flowtables
586 * This is indented for debug and unit-tests purposes only
588 VLIB_CLI_COMMAND (lb_flowtable_flush_command, static) =
590 .path = "test lb flowtable flush",
591 .short_help = "test lb flowtable flush",
592 .function = lb_flowtable_flush_command_fn,