/* * Copyright (c) 2015 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. */ /* * pg_cli.c: packet generator cli * * Copyright (c) 2008 Eliot Dresselhaus * * Permission is hereby granted, free of charge, to any person obtaining * a copy of this software and associated documentation files (the * "Software"), to deal in the Software without restriction, including * without limitation the rights to use, copy, modify, merge, publish, * distribute, sublicense, and/or sell copies of the Software, and to * permit persons to whom the Software is furnished to do so, subject to * the following conditions: * * The above copyright notice and this permission notice shall be * included in all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ #include #include #include #ifdef CLIB_UNIX #include #endif /* Root of all packet generator cli commands. */ /* *INDENT-OFF* */ VLIB_CLI_COMMAND (vlib_cli_pg_command, static) = { .path = "packet-generator", .short_help = "Packet generator commands", }; /* *INDENT-ON* */ void pg_enable_disable (u32 stream_index, int is_enable) { pg_main_t *pg = &pg_main; pg_stream_t *s; if (stream_index == ~0) { /* No stream specified: enable/disable all streams. */ /* *INDENT-OFF* */ pool_foreach (s, pg->streams, ({ pg_stream_enable_disable (pg, s, is_enable); })); /* *INDENT-ON* */ } else { /* enable/disable specified stream. */ s = pool_elt_at_index (pg->streams, stream_index); pg_stream_enable_disable (pg, s, is_enable); } } clib_error_t * pg_capture (pg_capture_args_t * a) { pg_main_t *pg = &pg_main; pg_interface_t *pi; if (a->is_enabled == 1) { struct stat sb; if (stat ((char *) a->pcap_file_name, &sb) != -1) return clib_error_return (0, "Cannot create pcap file"); } pi = pool_elt_at_index (pg->interfaces, a->dev_instance); vec_free (pi->pcap_file_name); memset (&pi->pcap_main, 0, sizeof (pi->pcap_main)); if (a->is_enabled == 0) return 0; pi->pcap_file_name = a->pcap_file_name; pi->pcap_main.file_name = (char *) pi->pcap_file_name; pi->pcap_main.n_packets_to_capture = a->count; pi->pcap_main.packet_type = PCAP_PACKET_TYPE_ethernet; return 0; } static clib_error_t * enable_disable_stream (vlib_main_t * vm, unformat_input_t * input, vlib_cli_command_t * cmd) { pg_main_t *pg = &pg_main; int is_enable = cmd->function_arg != 0; u32 stream_index = ~0; if (unformat (input, "%U", unformat_eof)) ; else if (unformat (input, "%U", unformat_hash_vec_string, pg->stream_index_by_name, &stream_index)) ; else return clib_error_create ("unknown input `%U'", format_unformat_error, input); pg_enable_disable (stream_index, is_enable); return 0; } /* *INDENT-OFF* */ VLIB_CLI_COMMAND (enable_streams_cli, static) = { .path = "packet-generator enable-stream", .short_help = "Enable packet generator streams", .function = enable_disable_stream, .function_arg = 1, /* is_enable */ }; /* *INDENT-ON* */ /* *INDENT-OFF* */ VLIB_CLI_COMMAND (disable_streams_cli, static) = { .path = "packet-generator disable-stream", .short_help = "Disable packet generator streams", .function = enable_disable_stream, .function_arg = 0, /* is_enable */ }; /* *INDENT-ON* */ static u8 * format_pg_stream (u8 * s, va_list * va) { pg_stream_t *t = va_arg (*va, pg_stream_t *); u8 *v; if (!t) return format (s, "%=16s%=12s%=16s%s", "Name", "Enabled", "Count", "Parameters"); s = format (s, "%-16v%=12s%16Ld", t->name, pg_stream_is_enabled (t) ? "Yes" : "No", t->n_packets_generated); v = 0; v = format (v, "limit %Ld, ", t->n_packets_limit); v = format (v, "rate %.2e pps, ", t->rate_packets_per_second); v = format (v, "size %d%c%d, ", t->min_packet_bytes, t->packet_size_edit_type == PG_EDIT_RANDOM ? '+' : '-', t->max_packet_bytes); v = format (v, "buffer-size %d, ", t->buffer_bytes); v = format (v, "worker %d, ", t->worker_index); if (v) { s = format (s, " %v", v); vec_free (v); } return s; } static clib_error_t * show_streams (vlib_main_t * vm, unformat_input_t * input, vlib_cli_command_t * cmd) { pg_main_t *pg = &pg_main; pg_stream_t *s; if (pool_elts (pg->streams) == 0) { vlib_cli_output (vm, "no streams currently defined"); goto done; } vlib_cli_output (vm, "%U", format_pg_stream, 0); /* *INDENT-OFF* */ pool_foreach (s, pg->streams, ({ vlib_cli_output (vm, "%U", format_pg_stream, s); })); /* *INDENT-ON* */ done: return 0; } /* *INDENT-OFF* */ VLIB_CLI_COMMAND (show_streams_cli, static) = { .path = "show packet-generator", .short_help = "Show packet generator streams", .function = show_streams, }; /* *INDENT-ON* */ static clib_error_t * pg_pcap_read (pg_stream_t * s, char *file_name) { #ifndef CLIB_UNIX return clib_error_return (0, "no pcap support"); #else pcap_main_t pm; clib_error_t *error; memset (&pm, 0, sizeof (pm)); pm.file_name = file_name; error = pcap_read (&pm); s->replay_packet_templates = pm.packets_read; s->min_packet_bytes = pm.min_packet_bytes; s->max_packet_bytes = pm.max_packet_bytes; s->buffer_bytes = pm.max_packet_bytes; /* For PCAP buffers we never re-use buffers. */ s->flags |= PG_STREAM_FLAGS_DISABLE_BUFFER_RECYCLE; if (s->n_packets_limit == 0) s->n_packets_limit = vec_len (pm.packets_read); return error; #endif /* CLIB_UNIX */ } static uword unformat_pg_stream_parameter (unformat_input_t * input, va_list * args) { pg_stream_t *s = va_arg (*args, pg_stream_t *); f64 x; if (unformat (input, "limit %f", &x)) s->n_packets_limit = x; else if (unformat (input, "rate %f", &x)) s->rate_packets_per_second = x; else if (unformat (input, "size %d-%d", &s->min_packet_bytes, &s->max_packet_bytes)) s->packet_size_edit_type = PG_EDIT_INCREMENT; else if (unformat (input, "size %d+%d", &s->min_packet_bytes, &s->max_packet_bytes)) s->packet_size_edit_type = PG_EDIT_RANDOM; else if (unformat (input, "buffer-size %d", &s->buffer_bytes)) ; else return 0; return 1; } static clib_error_t * validate_stream (pg_stream_t * s) { if (s->max_packet_bytes < s->min_packet_bytes) return clib_error_create ("max-size < min-size"); if (s->buffer_bytes >= 4096 || s->buffer_b