src/vnet/srv6/sr_pt_node.c

   1 /* SPDX-License-Identifier: Apache-2.0
   2  * Copyright(c) 2022 Cisco Systems, Inc.
   3  */
   4
   5 #include <vnet/fib/ip6_fib.h>
   6 #include <vnet/dpo/load_balance.h>
   7 #include <vnet/l2/feat_bitmap.h>
   8 #include <vnet/fib/fib_table.h>
   9 #include <vnet/srv6/sr.h>
  10 #include <vnet/srv6/sr_pt.h>
  11
  12 /**
  13  * @brief PT node trace
  14  */
  15 typedef struct
  16 {
  17   u32 iface;
  18   u16 id;
  19   u8 load;
  20   timestamp_64_t t64;
  21   u8 tts_template;
  22   u8 tts;
  23   u8 behavior;
  24 } pt_trace_t;
  25
  26 static u8 *
  27 format_pt_trace (u8 *s, va_list *args)
  28 {
  29   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
  30   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
  31   pt_trace_t *t = va_arg (*args, pt_trace_t *);
  32   switch (t->behavior)
  33     {
  34     case PT_BEHAVIOR_MID:
  35       s = format (
  36         s,
  37         "Behavior Midpoint, outgoing interface %U, outgoing interface id %u, "
  38         "outgoing interface load %u, t64_sec %u, t64_nsec %u, tts_template "
  39         "%u, tts %u",
  40         format_vnet_sw_if_index_name, vnet_get_main (), t->iface, t->id,
  41         t->load, clib_host_to_net_u32 (t->t64.sec),
  42         clib_host_to_net_u32 (t->t64.nsec), t->tts_template, t->tts);
  43       break;
  44     default:
  45       break;
  46     }
  47   return s;
  48 }
  49
  50 static_always_inline void
  51 pt_midpoint_processing (vlib_main_t *vm, vlib_node_runtime_t *node,
  52                         vlib_buffer_t *b0, ip6_header_t *ip0,
  53                         sr_pt_iface_t *ls, timestamp_64_t t64)
  54 {
  55   ip6_hop_by_hop_header_t *hbh;
  56   ip6_hop_by_hop_option_t *hbh_opt;
  57   ip6_hop_by_hop_option_pt_t *hbh_opt_pt;
  58
  59   if (ip0->protocol == IP_PROTOCOL_IP6_HOP_BY_HOP_OPTIONS)
  60     {
  61       hbh = (void *) (ip0 + 1);
  62       hbh_opt = (void *) (hbh + 1);
  63       if (hbh_opt->type == IP6_HBH_PT_TYPE)
  64         {
  65           hbh_opt_pt = (void *) (hbh_opt + 1);
  66           clib_memcpy_fast (&hbh_opt_pt->cmd_stack[1],
  67                             &hbh_opt_pt->cmd_stack[0], 33);
  68           hbh_opt_pt->cmd_stack[0].oif_oil =
  69             clib_host_to_net_u16 (ls->id << 4);
  70           hbh_opt_pt->cmd_stack[0].oif_oil |= ls->egress_load;
  71           switch (ls->tts_template)
  72             {
  73             case SR_PT_TTS_TEMPLATE_0:
  74               hbh_opt_pt->cmd_stack[0].tts =
  75                 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_0;
  76               break;
  77             case SR_PT_TTS_TEMPLATE_1:
  78               hbh_opt_pt->cmd_stack[0].tts =
  79                 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_1;
  80               break;
  81             case SR_PT_TTS_TEMPLATE_2:
  82               hbh_opt_pt->cmd_stack[0].tts =
  83                 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_2;
  84               break;
  85             case SR_PT_TTS_TEMPLATE_3:
  86               hbh_opt_pt->cmd_stack[0].tts =
  87                 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_0;
  88               break;
  89             default:
  90               break;
  91             }
  92         }
  93     }
  94   return;
  95 }
  96
  97 VLIB_NODE_FN (sr_pt_node)
  98 (vlib_main_t *vm, vlib_node_runtime_t *node, vlib_frame_t *from_frame)
  99 {
 100   u32 n_left_from, next_index, *from, *to_next;
 101   from = vlib_frame_vector_args (from_frame);
 102   n_left_from = from_frame->n_vectors;
 103   next_index = node->cached_next_index;
 104   u8 pt_behavior = ~(u8) 0;
 105   sr_pt_iface_t *ls = 0;
 106   while (n_left_from > 0)
 107     {
 108       u32 n_left_to_next;
 109       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
 110
 111       // Getting the timestamp (one for each batch of packets)
 112       timestamp_64_t t64 = {};
 113       unix_time_now_nsec_fraction (&t64.sec, &t64.nsec);
 114
 115       // Single loop for potentially the last three packets
 116       while (n_left_from > 0 && n_left_to_next > 0)
 117         {
 118           u32 bi0;
 119           u32 iface;
 120           vlib_buffer_t *b0;
 121           u32 next0 = 0;
 122           ethernet_header_t *en0;
 123           ip6_header_t *ip0 = 0;
 124           bi0 = from[0];
 125           to_next[0] = bi0;
 126           from += 1;
 127           to_next += 1;
 128           n_left_from -= 1;
 129           n_left_to_next -= 1;
 130
 131           b0 = vlib_get_buffer (vm, bi0);
 132           iface = vnet_buffer (b0)->sw_if_index[VLIB_TX];
 133           ls = sr_pt_find_iface (iface);
 134           if (ls)
 135             {
 136               en0 = vlib_buffer_get_current (b0);
 137               ip0 = (void *) (en0 + 1);
 138               pt_midpoint_processing (vm, node, b0, ip0, ls, t64);
 139               pt_behavior = PT_BEHAVIOR_MID;
 140             }
 141           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
 142             {
 143               pt_trace_t *tr = vlib_add_trace (vm, node, b0, sizeof (*tr));
 144               tr->iface = iface;
 145               tr->id = ls->id;
 146               tr->load = ls->egress_load;
 147               tr->tts_template = ls->tts_template;
 148               tr->t64.sec = t64.sec;
 149               tr->t64.nsec = t64.nsec;
 150               tr->tts = t64.nsec >> 20;
 151               tr->behavior = pt_behavior;
 152             }
 153           vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
 154                                            n_left_to_next, bi0, next0);
 155         }
 156       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
 157     }
 158
 159   return from_frame->n_vectors;
 160 }
 161
 162 VLIB_REGISTER_NODE (sr_pt_node) = {
 163   .name = "pt",
 164   .vector_size = sizeof (u32),
 165   .format_trace = format_pt_trace,
 166   .type = VLIB_NODE_TYPE_INTERNAL,
 167   .n_errors = 0,
 168   .n_next_nodes = 1,
 169   .next_nodes = { [0] = "interface-output" },
 170 };
 171
 172 VNET_FEATURE_INIT (sr_pt_node, static) = {
 173   .arc_name = "ip6-output",
 174   .node_name = "pt",
 175 };