LCOV - code coverage report
Current view: top level - plugins/adl - node.c (source / functions) Hit Total Coverage
Test: coverage-filtered.info Lines: 88 105 83.8 %
Date: 2023-10-26 01:39:38 Functions: 9 14 64.3 %

          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;

Generated by: LCOV version 1.14