diff options
Diffstat (limited to 'src/plugins/unittest/mpcap_node.c')
| -rw-r--r-- | src/plugins/unittest/mpcap_node.c | 260 |
1 files changed, 260 insertions, 0 deletions
diff --git a/src/plugins/unittest/mpcap_node.c b/src/plugins/unittest/mpcap_node.c new file mode 100644 index 00000000000..8c0bc8677be --- /dev/null +++ b/src/plugins/unittest/mpcap_node.c @@ -0,0 +1,260 @@ +/* + * mpcap_node.c - unit-test plugin mapped pcap test node + * + * Copyright (c) <current-year> <your-organization> + * 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> +#include <vnet/vnet.h> +#include <vnet/pg/pg.h> +#include <vppinfra/error.h> +#include <vppinfra/mpcap.h> +#include <vnet/ethernet/ethernet.h> + +static mpcap_main_t test_mpcap_main = { + .file_name = "/tmp/mpcap_unittest.pcap", + .n_packets_to_capture = 15, + .packet_type = MPCAP_PACKET_TYPE_ethernet, +}; + +typedef struct +{ + u32 sw_if_index; + u32 next_index; +} mpcap_trace_t; + +#ifndef CLIB_MARCH_VARIANT + +/* packet trace format function */ +static u8 * +format_mpcap_trace (u8 * s, va_list * args) +{ + CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *); + CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *); + mpcap_trace_t *t = va_arg (*args, mpcap_trace_t *); + + s = format (s, "MPCAP: sw_if_index %d, next index %d\n", + t->sw_if_index, t->next_index); + return s; +} + +vlib_node_registration_t mpcap_node; + +#endif /* CLIB_MARCH_VARIANT */ + +#define foreach_mpcap_error \ +_(PROCESSED, "Mapped pcap packets processed") + +typedef enum +{ +#define _(sym,str) MPCAP_ERROR_##sym, + foreach_mpcap_error +#undef _ + MPCAP_N_ERROR, +} mpcap_error_t; + +#ifndef CLIB_MARCH_VARIANT +static char *mpcap_error_strings[] = { +#define _(sym,string) string, + foreach_mpcap_error +#undef _ +}; +#endif /* CLIB_MARCH_VARIANT */ + +typedef enum +{ + MPCAP_NEXT_DROP, + MPCAP_N_NEXT, +} mpcap_next_t; + +always_inline uword +mpcap_inline (vlib_main_t * vm, + vlib_node_runtime_t * node, vlib_frame_t * frame, + int is_ip4, int is_trace) +{ + u32 n_left_from, *from; + vlib_buffer_t *bufs[VLIB_FRAME_SIZE], **b; + u16 nexts[VLIB_FRAME_SIZE], *next; + mpcap_main_t *mm = &test_mpcap_main; + int i = 0; + f64 now = vlib_time_now (vm); + u32 error_index = node->errors[MPCAP_ERROR_PROCESSED]; + + if (!(mm->flags & MPCAP_FLAG_INIT_DONE)) + { + clib_error_t *error; + + error = mpcap_init (mm); + if (error) + clib_error_report (error); + mm->flags |= MPCAP_FLAG_INIT_DONE; + } + + from = vlib_frame_vector_args (frame); + n_left_from = frame->n_vectors; + + vlib_get_buffers (vm, from, bufs, n_left_from); + b = bufs; + next = nexts; + + while (n_left_from >= 4) + { + /* Prefetch next iteration. */ + if (PREDICT_TRUE (n_left_from >= 8)) + { + vlib_prefetch_buffer_header (b[4], STORE); + vlib_prefetch_buffer_header (b[5], STORE); + vlib_prefetch_buffer_header (b[6], STORE); + vlib_prefetch_buffer_header (b[7], STORE); + CLIB_PREFETCH (b[4]->data, CLIB_CACHE_LINE_BYTES, STORE); + CLIB_PREFETCH (b[5]->data, CLIB_CACHE_LINE_BYTES, STORE); + CLIB_PREFETCH (b[6]->data, CLIB_CACHE_LINE_BYTES, STORE); + CLIB_PREFETCH (b[7]->data, CLIB_CACHE_LINE_BYTES, STORE); + } + + next[0] = 0; + next[1] = 0; + next[2] = 0; + next[3] = 0; + + b[0]->error = error_index; + b[1]->error = error_index; + b[2]->error = error_index; + b[3]->error = error_index; + + mpcap_add_buffer (mm, vm, now, from[i++], + vlib_buffer_length_in_chain (vm, b[0])); + + mpcap_add_buffer (mm, vm, now, from[i++], + vlib_buffer_length_in_chain (vm, b[1])); + + mpcap_add_buffer (mm, vm, now, from[i++], + vlib_buffer_length_in_chain (vm, b[2])); + + mpcap_add_buffer (mm, vm, now, from[i++], + vlib_buffer_length_in_chain (vm, b[3])); + + if (is_trace) + { + if (b[0]->flags & VLIB_BUFFER_IS_TRACED) + { + mpcap_trace_t *t = vlib_add_trace (vm, node, b[0], sizeof (*t)); + t->next_index = next[0]; + t->sw_if_index = vnet_buffer (b[0])->sw_if_index[VLIB_RX]; + } + if (b[1]->flags & VLIB_BUFFER_IS_TRACED) + { + mpcap_trace_t *t = vlib_add_trace (vm, node, b[1], sizeof (*t)); + t->next_index = next[1]; + t->sw_if_index = vnet_buffer (b[1])->sw_if_index[VLIB_RX]; + } + if (b[2]->flags & VLIB_BUFFER_IS_TRACED) + { + mpcap_trace_t *t = vlib_add_trace (vm, node, b[2], sizeof (*t)); + t->next_index = next[2]; + t->sw_if_index = vnet_buffer (b[2])->sw_if_index[VLIB_RX]; + } + if (b[3]->flags & VLIB_BUFFER_IS_TRACED) + { + mpcap_trace_t *t = vlib_add_trace (vm, node, b[3], sizeof (*t)); + t->next_index = next[3]; + t->sw_if_index = vnet_buffer (b[3])->sw_if_index[VLIB_RX]; + } + } + + b += 4; + next += 4; + n_left_from -= 4; + } + + while (n_left_from > 0) + { + mpcap_add_buffer (mm, vm, now, from[i++], + vlib_buffer_length_in_chain (vm, b[0])); + next[0] = 0; + b[0]->error = error_index; + + if (is_trace) + { + if (b[0]->flags & VLIB_BUFFER_IS_TRACED) + { + mpcap_trace_t *t = vlib_add_trace (vm, node, b[0], sizeof (*t)); + t->next_index = next[0]; + t->sw_if_index = vnet_buffer (b[0])->sw_if_index[VLIB_RX]; + } + } + + b += 1; + next += 1; + n_left_from -= 1; + } + + vlib_buffer_enqueue_to_next (vm, node, from, nexts, frame->n_vectors); + + return frame->n_vectors; +} + +VLIB_NODE_FN (mpcap_node) (vlib_main_t * vm, vlib_node_runtime_t * node, + vlib_frame_t * frame) +{ + if (PREDICT_FALSE (node->flags & VLIB_NODE_FLAG_TRACE)) + return mpcap_inline (vm, node, frame, 1 /* is_ip4 */ , + 1 /* is_trace */ ); + else + return mpcap_inline (vm, node, frame, 1 /* is_ip4 */ , + 0 /* is_trace */ ); +} + +/* *INDENT-OFF* */ +#ifndef CLIB_MARCH_VARIANT +VLIB_REGISTER_NODE (mpcap_node) = +{ + .name = "mpcap-unittest", + .vector_size = sizeof (u32), + .format_trace = format_mpcap_trace, + .format_buffer = format_ethernet_header_with_length, + .unformat_buffer = unformat_ethernet_header, + + .type = VLIB_NODE_TYPE_INTERNAL, + + .n_errors = ARRAY_LEN(mpcap_error_strings), + .error_strings = mpcap_error_strings, + + .n_next_nodes = MPCAP_N_NEXT, + + /* edit / add dispositions here */ + .next_nodes = + { + [MPCAP_NEXT_DROP] = "error-drop", + }, +}; + +static clib_error_t * +mpcap_node_init (vlib_main_t *vm) +{ + /* So we can send ethernet/IP packets from the pg... */ + ethernet_setup_node (vm, mpcap_node.index); + return 0; +} +VLIB_INIT_FUNCTION (mpcap_node_init); + +#endif /* CLIB_MARCH_VARIANT */ +/* *INDENT-ON* */ + +/* + * fd.io coding-style-patch-verification: ON + * + * Local Variables: + * eval: (c-set-style "gnu") + * End: + */ |