diff options
Diffstat (limited to 'src/plugins/unittest/vlib_test.c')
-rw-r--r-- | src/plugins/unittest/vlib_test.c | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/src/plugins/unittest/vlib_test.c b/src/plugins/unittest/vlib_test.c new file mode 100644 index 00000000000..275584e01eb --- /dev/null +++ b/src/plugins/unittest/vlib_test.c @@ -0,0 +1,137 @@ +/* + * Copyright (c) 2019 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. + */ + +#include <vlib/vlib.h> + +u8 *vlib_validate_buffers (vlib_main_t * vm, + u32 * buffers, + uword next_buffer_stride, + uword n_buffers, + vlib_buffer_known_state_t known_state, + uword follow_buffer_next); + +static clib_error_t * +test_vlib_command_fn (vlib_main_t * vm, + unformat_input_t * input, vlib_cli_command_t * cmd) +{ + u32 bi; + u8 *res; + u32 allocated; + vlib_buffer_t *b; + vlib_buffer_t *last_b; + u8 junk[4] = { 1, 2, 3, 4 }; + vlib_packet_template_t _t, *t = &_t; + u8 *data_copy = 0; + + /* Cover vlib_packet_template_get_packet */ + t->packet_data = format (0, "silly packet data"); + t->min_n_buffers_each_alloc = 1; + t->name = (u8 *) "test template"; + + if (vlib_packet_template_get_packet (vm, t, &bi)) + vlib_buffer_free_one (vm, bi); + + vec_free (t->packet_data); + + /* Get a buffer */ + allocated = vlib_buffer_alloc (vm, &bi, 1); + if (allocated != 1) + return clib_error_return (0, "Buffer allocation failure!"); + + b = vlib_get_buffer (vm, bi); + + /* Force buffer allocation */ + b->current_length = 2048; + last_b = b; + vlib_buffer_chain_append_data_with_alloc (vm, b, &last_b, + junk, ARRAY_LEN (junk)); + + /* Cover vlib_buffer_length_in_chain_slow_path(...) */ + b->flags &= ~(VLIB_BUFFER_TOTAL_LENGTH_VALID); + vlib_cli_output (vm, "buffer length %d", + vlib_buffer_length_in_chain (vm, b)); + b->flags &= ~(VLIB_BUFFER_TOTAL_LENGTH_VALID); + vlib_cli_output (vm, "%u", vlib_buffer_index_length_in_chain (vm, bi)); + + /* Add more data. Eat Mor Chikin. */ + vlib_buffer_add_data (vm, &bi, junk, ARRAY_LEN (junk)); + + /* Dump the resulting two-chunk pkt */ + vlib_cli_output (vm, "%U", format_vlib_buffer_and_data, b); + + vec_validate (data_copy, vlib_buffer_length_in_chain (vm, b) - 1); + vlib_cli_output (vm, "%u", vlib_buffer_contents (vm, bi, data_copy)); + vec_free (data_copy); + + /* Cover simple functions in buffer.h / buffer_funcs.h */ + vlib_cli_output (vm, "%llx", vlib_buffer_get_va (b)); + vlib_cli_output (vm, "%llx", vlib_buffer_get_current_va (b)); + vlib_cli_output (vm, "%d", vlib_buffer_has_space (b, 100ll)); + vlib_buffer_reset (b); + vlib_cli_output (vm, "%llx", vlib_buffer_get_tail (b)); + vlib_buffer_put_uninit (b, 0); + vlib_buffer_push_uninit (b, 0); + vlib_buffer_make_headroom (b, 0); + (void) vlib_buffer_pull (b, 0); + vlib_cli_output (vm, "%llx", vlib_buffer_get_pa (vm, b)); + vlib_cli_output (vm, "%llx", vlib_buffer_get_current_pa (vm, b)); + + /* Validate it one way */ + res = vlib_validate_buffer (vm, bi, 1 /* follow_buffer_next */ ); + if (res) + return clib_error_return (0, "%v", res); + + /* Validate it a different way */ + res = vlib_validate_buffers (vm, &bi, 0 /* stride */ , + 1, VLIB_BUFFER_KNOWN_ALLOCATED, + 1 /* follow_buffer_next */ ); + if (res) + return clib_error_return (0, "%v", res); + + /* Free it */ + vlib_buffer_free_one (vm, bi); + /* It will be free */ + res = vlib_validate_buffers (vm, &bi, 0 /* stride */ , + 1, VLIB_BUFFER_KNOWN_FREE, + 1 /* follow_buffer_next */ ); + if (res) + return clib_error_return (0, "%v", res); + + /* It will not be allocated, exercise error path */ + res = vlib_validate_buffers (vm, &bi, 0 /* stride */ , + 1, VLIB_BUFFER_KNOWN_ALLOCATED, + 1 /* follow_buffer_next */ ); + if (res) + return clib_error_return (0, "%v", res); + + return 0; +} + +/* *INDENT-OFF* */ +VLIB_CLI_COMMAND (test_vlib_command, static) = +{ + .path = "test vlib", + .short_help = "vlib code coverate unit test", + .function = test_vlib_command_fn, +}; +/* *INDENT-ON* */ + +/* + * fd.io coding-style-patch-verification: ON + * + * Local Variables: + * eval: (c-set-style "gnu") + * End: + */ |