Line data Source code
1 : /*
2 : * Copyright (c) 2016,2020 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:
6 : *
7 : * http://www.apache.org/licenses/LICENSE-2.0
8 : *
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.
14 : */
15 :
16 : #include <plugins/adl/adl.h>
17 :
18 : typedef struct {
19 : u32 next_index;
20 : u32 sw_if_index;
21 : } adl_input_trace_t;
22 :
23 : /* packet trace format function */
24 0 : static u8 * format_adl_input_trace (u8 * s, va_list * args)
25 : {
26 0 : CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
27 0 : CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
28 0 : adl_input_trace_t * t = va_arg (*args, adl_input_trace_t *);
29 :
30 0 : s = format (s, "ADL_INPUT: sw_if_index %d, next index %d",
31 : t->sw_if_index, t->next_index);
32 0 : return s;
33 : }
34 :
35 : #define foreach_adl_input_error \
36 : _(PROCESSED, "Allow/Deny packets processed")
37 :
38 : typedef enum {
39 : #define _(sym,str) ADL_INPUT_ERROR_##sym,
40 : foreach_adl_input_error
41 : #undef _
42 : ADL_INPUT_N_ERROR,
43 : } adl_input_error_t;
44 :
45 : static char * adl_input_error_strings[] = {
46 : #define _(sym,string) string,
47 : foreach_adl_input_error
48 : #undef _
49 : };
50 :
51 2301 : VLIB_NODE_FN (adl_input_node) (vlib_main_t * vm,
52 : vlib_node_runtime_t * node,
53 : vlib_frame_t * frame)
54 : {
55 : u32 n_left_from, * from, * to_next;
56 : adl_feature_type_t next_index;
57 1 : adl_main_t *am = &adl_main;
58 :
59 1 : from = vlib_frame_vector_args (frame);
60 1 : n_left_from = frame->n_vectors;
61 1 : next_index = node->cached_next_index;
62 :
63 2 : while (n_left_from > 0)
64 : {
65 : u32 n_left_to_next;
66 :
67 1 : vlib_get_next_frame (vm, node, next_index,
68 : to_next, n_left_to_next);
69 :
70 100 : while (n_left_from >= 4 && n_left_to_next >= 2)
71 : {
72 : u32 bi0, bi1;
73 : vlib_buffer_t * b0, * b1;
74 : u32 next0, next1;
75 : u32 sw_if_index0, sw_if_index1;
76 : ethernet_header_t * en0, * en1;
77 : adl_config_main_t * ccm0, * ccm1;
78 : u32 advance0, advance1;
79 : int proto0, proto1;
80 :
81 : /* Prefetch next iteration. */
82 : {
83 : vlib_buffer_t * p2, * p3;
84 :
85 99 : p2 = vlib_get_buffer (vm, from[2]);
86 99 : p3 = vlib_get_buffer (vm, from[3]);
87 :
88 99 : vlib_prefetch_buffer_header (p2, LOAD);
89 99 : vlib_prefetch_buffer_header (p3, LOAD);
90 :
91 99 : clib_prefetch_store (p2->data);
92 99 : clib_prefetch_store (p3->data);
93 : }
94 :
95 : /* speculatively enqueue b0 and b1 to the current next frame */
96 99 : to_next[0] = bi0 = from[0];
97 99 : to_next[1] = bi1 = from[1];
98 99 : from += 2;
99 99 : to_next += 2;
100 99 : n_left_from -= 2;
101 99 : n_left_to_next -= 2;
102 :
103 99 : b0 = vlib_get_buffer (vm, bi0);
104 99 : b1 = vlib_get_buffer (vm, bi1);
105 :
106 99 : en0 = vlib_buffer_get_current (b0);
107 99 : en1 = vlib_buffer_get_current (b1);
108 :
109 99 : sw_if_index0 = adl_buffer(b0)->sw_if_index[VLIB_RX];
110 99 : sw_if_index1 = adl_buffer(b1)->sw_if_index[VLIB_RX];
111 :
112 99 : proto0 = VNET_ADL_DEFAULT;
113 99 : proto1 = VNET_ADL_DEFAULT;
114 99 : advance0 = 0;
115 99 : advance1 = 0;
116 :
117 99 : if (en0->type == clib_host_to_net_u16(ETHERNET_TYPE_IP4))
118 : {
119 50 : proto0 = VNET_ADL_IP4;
120 50 : advance0 = sizeof(ethernet_header_t);
121 : }
122 49 : else if (en0->type == clib_host_to_net_u16(ETHERNET_TYPE_IP6))
123 : {
124 49 : proto0 = VNET_ADL_IP6;
125 49 : advance0 = sizeof(ethernet_header_t);
126 : }
127 :
128 99 : if (en1->type == clib_host_to_net_u16(ETHERNET_TYPE_IP4))
129 : {
130 50 : proto1 = VNET_ADL_IP4;
131 50 : advance1 = sizeof(ethernet_header_t);
132 : }
133 49 : else if (en1->type == clib_host_to_net_u16(ETHERNET_TYPE_IP6))
134 : {
135 49 : proto1 = VNET_ADL_IP6;
136 49 : advance1 = sizeof(ethernet_header_t);
137 : }
138 :
139 99 : ccm0 = am->adl_config_mains + proto0;
140 99 : ccm1 = am->adl_config_mains + proto1;
141 99 : adl_buffer(b0)->adl.current_config_index =
142 99 : ccm0->config_index_by_sw_if_index [sw_if_index0];
143 :
144 99 : adl_buffer(b1)->adl.current_config_index =
145 99 : ccm1->config_index_by_sw_if_index [sw_if_index1];
146 :
147 99 : vlib_buffer_advance (b0, advance0);
148 99 : vlib_buffer_advance (b1, advance1);
149 :
150 99 : vnet_get_config_data (&ccm0->config_main,
151 99 : &adl_buffer(b0)->adl.current_config_index,
152 : &next0, 0 /* bytes of config data */);
153 :
154 99 : vnet_get_config_data (&ccm1->config_main,
155 99 : &adl_buffer(b1)->adl.current_config_index,
156 : &next1, 0 /* bytes of config data */);
157 :
158 99 : if (PREDICT_FALSE((node->flags & VLIB_NODE_FLAG_TRACE)
159 : && (b0->flags & VLIB_BUFFER_IS_TRACED)))
160 : {
161 : adl_input_trace_t *t =
162 0 : vlib_add_trace (vm, node, b0, sizeof (*t));
163 0 : t->sw_if_index = sw_if_index0;
164 0 : t->next_index = next0;
165 : }
166 :
167 99 : if (PREDICT_FALSE((node->flags & VLIB_NODE_FLAG_TRACE)
168 : && (b1->flags & VLIB_BUFFER_IS_TRACED)))
169 : {
170 : adl_input_trace_t *t =
171 0 : vlib_add_trace (vm, node, b1, sizeof (*t));
172 0 : t->sw_if_index = sw_if_index1;
173 0 : t->next_index = next1;
174 : }
175 : /* verify speculative enqueues, maybe switch current next frame */
176 99 : vlib_validate_buffer_enqueue_x2 (vm, node, next_index,
177 : to_next, n_left_to_next,
178 : bi0, bi1, next0, next1);
179 : }
180 :
181 3 : while (n_left_from > 0 && n_left_to_next > 0)
182 : {
183 : u32 bi0;
184 : vlib_buffer_t * b0;
185 : u32 next0;
186 : u32 sw_if_index0;
187 : ethernet_header_t *en0;
188 : adl_config_main_t *ccm0;
189 : u32 advance0;
190 : int proto0;
191 :
192 : /* speculatively enqueue b0 to the current next frame */
193 2 : bi0 = from[0];
194 2 : to_next[0] = bi0;
195 2 : from += 1;
196 2 : to_next += 1;
197 2 : n_left_from -= 1;
198 2 : n_left_to_next -= 1;
199 :
200 2 : b0 = vlib_get_buffer (vm, bi0);
201 :
202 : /*
203 : * Direct from the driver, we should be at offset 0
204 : * aka at &b0->data[0]
205 : */
206 2 : ASSERT (b0->current_data == 0);
207 :
208 2 : en0 = vlib_buffer_get_current (b0);
209 :
210 2 : sw_if_index0 = adl_buffer(b0)->sw_if_index[VLIB_RX];
211 :
212 2 : proto0 = VNET_ADL_DEFAULT;
213 2 : advance0 = 0;
214 :
215 2 : if (en0->type == clib_host_to_net_u16(ETHERNET_TYPE_IP4))
216 : {
217 0 : proto0 = VNET_ADL_IP4;
218 0 : advance0 = sizeof(ethernet_header_t);
219 : }
220 2 : else if (en0->type == clib_host_to_net_u16(ETHERNET_TYPE_IP6))
221 : {
222 2 : proto0 = VNET_ADL_IP6;
223 2 : advance0 = sizeof(ethernet_header_t);
224 : }
225 :
226 2 : ccm0 = am->adl_config_mains + proto0;
227 2 : adl_buffer(b0)->adl.current_config_index =
228 2 : ccm0->config_index_by_sw_if_index [sw_if_index0];
229 :
230 2 : vlib_buffer_advance (b0, advance0);
231 :
232 2 : vnet_get_config_data (&ccm0->config_main,
233 2 : &adl_buffer(b0)->adl.current_config_index,
234 : &next0, 0 /* bytes of config data */);
235 :
236 2 : if (PREDICT_FALSE((node->flags & VLIB_NODE_FLAG_TRACE)
237 : && (b0->flags & VLIB_BUFFER_IS_TRACED)))
238 : {
239 : adl_input_trace_t *t =
240 0 : vlib_add_trace (vm, node, b0, sizeof (*t));
241 0 : t->sw_if_index = sw_if_index0;
242 0 : t->next_index = next0;
243 : }
244 :
245 : /* verify speculative enqueue, maybe switch current next frame */
246 2 : vlib_validate_buffer_enqueue_x1 (vm, node, next_index,
247 : to_next, n_left_to_next,
248 : bi0, next0);
249 : }
250 :
251 1 : vlib_put_next_frame (vm, node, next_index, n_left_to_next);
252 : }
253 1 : vlib_node_increment_counter (vm, adl_input_node.index,
254 1 : ADL_INPUT_ERROR_PROCESSED, frame->n_vectors);
255 1 : return frame->n_vectors;
256 : }
257 :
258 173420 : VLIB_REGISTER_NODE (adl_input_node) = {
259 : .name = "adl-input",
260 : .vector_size = sizeof (u32),
261 : .format_trace = format_adl_input_trace,
262 : .unformat_buffer = unformat_ethernet_header,
263 : .type = VLIB_NODE_TYPE_INTERNAL,
264 :
265 : .n_errors = ARRAY_LEN(adl_input_error_strings),
266 : .error_strings = adl_input_error_strings,
267 :
268 : .n_next_nodes = ADL_RX_N_FEATURES,
269 :
270 : /* edit / add dispositions here */
271 : .next_nodes = {
272 : [IP4_RX_ADL_ALLOWLIST] = "ip4-adl-allowlist",
273 : [IP6_RX_ADL_ALLOWLIST] = "ip6-adl-allowlist",
274 : [DEFAULT_RX_ADL_ALLOWLIST] = "default-adl-allowlist",
275 : [IP4_RX_ADL_INPUT] = "ip4-input",
276 : [IP6_RX_ADL_INPUT] = "ip6-input",
277 : [DEFAULT_RX_ADL_INPUT] = "ethernet-input",
278 : [RX_ADL_DROP] = "error-drop",
279 : },
280 : };
281 :
282 : #define foreach_adl_stub \
283 : _(default-adl-allowlist, default_adl_allowlist)
284 :
285 : #define _(n,f) \
286 : \
287 : static uword \
288 : f##_node_fn (vlib_main_t * vm, \
289 : vlib_node_runtime_t * node, \
290 : vlib_frame_t * frame) \
291 : { \
292 : clib_warning ("BUG: stub function called"); \
293 : return 0; \
294 : } \
295 : \
296 : VLIB_REGISTER_NODE (f##_input_node) = { \
297 : .function = f##_node_fn, \
298 : .name = #n, \
299 : .vector_size = sizeof (u32), \
300 : .type = VLIB_NODE_TYPE_INTERNAL, \
301 : \
302 : .n_errors = 0, \
303 : .error_strings = 0, \
304 : \
305 : .n_next_nodes = 0, \
306 : };
307 :
308 173420 : foreach_adl_stub;
|