2 * Copyright (c) 2018 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 <vlib/vlib.h>
17 #include <vnet/vnet.h>
18 #include <vppinfra/error.h>
20 #include <vnet/feature/feature.h>
21 #include <vnet/ethernet/ethernet.h>
23 #ifndef CLIB_MARCH_VARIANT
25 vnet_sw_interface_stats_collect_enable_disable (u32 sw_if_index, u8 enable)
27 ethernet_interface_t *eif;
28 vnet_sw_interface_t *si;
32 vnm = vnet_get_main ();
34 si = vnet_get_sw_interface (vnm, sw_if_index);
37 * only ethernet HW interfaces are supported at this time
39 if (si->type != VNET_SW_INTERFACE_TYPE_HARDWARE)
41 return (VNET_API_ERROR_INVALID_VALUE);
44 eif = ethernet_get_interface (em, si->hw_if_index);
48 return (VNET_API_ERROR_FEATURE_DISABLED);
51 vnet_feature_enable_disable ("device-input", "stats-collect-rx",
52 sw_if_index, enable, 0, 0);
53 vnet_feature_enable_disable ("interface-output", "stats-collect-tx",
54 sw_if_index, enable, 0, 0);
58 #endif /* CLIB_MARCH_VARIANT */
61 format_stats_collect_trace (u8 * s, va_list * args)
63 CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
64 CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
69 #define inc_counter(ctype, rx_tx) \
73 static_always_inline uword
74 stats_collect_inline (vlib_main_t * vm,
75 vlib_node_runtime_t * node,
76 vlib_frame_t * frame, vlib_rx_or_tx_t rxtx)
78 vnet_interface_counter_type_t ct;
79 u32 n_left_from, *from, *to_next;
82 u32 stats_n_packets[VNET_N_COMBINED_INTERFACE_COUNTER] = { 0 };
83 u64 stats_n_bytes[VNET_N_COMBINED_INTERFACE_COUNTER] = { 0 };
85 from = vlib_frame_vector_args (frame);
86 n_left_from = frame->n_vectors;
87 next_index = node->cached_next_index;
89 while (n_left_from > 0)
93 vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
95 while (n_left_from > 0 && n_left_to_next > 0)
102 /* speculatively enqueue b0 to the current next frame */
103 to_next[0] = bi0 = from[0];
109 b0 = vlib_get_buffer (vm, bi0);
110 sw_if_index = vnet_buffer (b0)->sw_if_index[rxtx];
115 eh_dst_addr_to_rx_ctype (vlib_buffer_get_current (b0));
120 eh_dst_addr_to_tx_ctype (vlib_buffer_get_current (b0));
123 stats_n_bytes[b0_ctype] += vlib_buffer_length_in_chain (vm, b0);
124 stats_n_packets[b0_ctype] += 1;
126 vnet_feature_next (&next0, b0);
128 vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
129 n_left_to_next, bi0, next0);
134 foreach_rx_combined_interface_counter (ct)
136 vlib_increment_combined_counter
137 (vnet_main.interface_main.combined_sw_if_counters + ct,
138 vlib_get_thread_index (),
139 sw_if_index, stats_n_packets[ct], stats_n_bytes[ct]);
144 foreach_tx_combined_interface_counter (ct)
146 vlib_increment_combined_counter
147 (vnet_main.interface_main.combined_sw_if_counters + ct,
148 vlib_get_thread_index (),
149 sw_if_index, stats_n_packets[ct], stats_n_bytes[ct]);
153 vlib_put_next_frame (vm, node, next_index, n_left_to_next);
156 return frame->n_vectors;
159 VLIB_NODE_FN (stats_collect_rx_node) (vlib_main_t * vm,
160 vlib_node_runtime_t * node,
161 vlib_frame_t * frame)
163 return stats_collect_inline (vm, node, frame, VLIB_RX);
166 VLIB_NODE_FN (stats_collect_tx_node) (vlib_main_t * vm,
167 vlib_node_runtime_t * node,
168 vlib_frame_t * frame)
170 return stats_collect_inline (vm, node, frame, VLIB_TX);
174 VLIB_REGISTER_NODE (stats_collect_rx_node) = {
175 .vector_size = sizeof (u32),
176 .format_trace = format_stats_collect_trace,
177 .type = VLIB_NODE_TYPE_INTERNAL,
180 .name = "stats-collect-rx",
183 VLIB_REGISTER_NODE (stats_collect_tx_node) = {
184 .vector_size = sizeof (u32),
185 .format_trace = format_stats_collect_trace,
186 .type = VLIB_NODE_TYPE_INTERNAL,
189 .name = "stats-collect-tx",
192 VNET_FEATURE_INIT (stats_collect_rx_node, static) = {
193 .arc_name = "device-input",
194 .node_name = "stats-collect-rx",
195 .runs_before = VNET_FEATURES ("ethernet-input"),
198 VNET_FEATURE_INIT (stats_collect_tx_node, static) = {
199 .arc_name = "interface-output",
200 .node_name = "stats-collect-tx",
201 .runs_before = VNET_FEATURES ("interface-tx"),
206 static clib_error_t *
207 stats_collect_init (vlib_main_t * vm)
212 VLIB_INIT_FUNCTION (stats_collect_init);
216 * fd.io coding-style-patch-verification: ON
219 * eval: (c-set-style "gnu")