1 /* SPDX-License-Identifier: Apache-2.0
2 * Copyright(c) 2022 Cisco Systems, Inc.
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>
13 * @brief PT node trace
27 format_pt_trace (u8 *s, va_list *args)
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 *);
37 "Behavior Midpoint, outgoing interface %U, outgoing interface id %u, "
38 "outgoing interface load %u, t64_sec %u, t64_nsec %u, tts_template "
40 format_vnet_sw_if_index_name, vnet_get_main (), t->iface, t->id,
41 t->load, htobe32 (t->t64.sec), htobe32 (t->t64.nsec), t->tts_template,
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)
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;
59 if (ip0->protocol == IP_PROTOCOL_IP6_HOP_BY_HOP_OPTIONS)
61 hbh = (void *) (ip0 + 1);
62 hbh_opt = (void *) (hbh + 1);
63 if (hbh_opt->type == IP6_HBH_PT_TYPE)
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 = htobe16 (ls->id << 4);
69 hbh_opt_pt->cmd_stack[0].oif_oil |= ls->egress_load;
70 switch (ls->tts_template)
72 case SR_PT_TTS_TEMPLATE_0:
73 hbh_opt_pt->cmd_stack[0].tts =
74 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_0;
76 case SR_PT_TTS_TEMPLATE_1:
77 hbh_opt_pt->cmd_stack[0].tts =
78 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_1;
80 case SR_PT_TTS_TEMPLATE_2:
81 hbh_opt_pt->cmd_stack[0].tts =
82 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_2;
84 case SR_PT_TTS_TEMPLATE_3:
85 hbh_opt_pt->cmd_stack[0].tts =
86 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_0;
96 VLIB_NODE_FN (sr_pt_node)
97 (vlib_main_t *vm, vlib_node_runtime_t *node, vlib_frame_t *from_frame)
99 u32 n_left_from, next_index, *from, *to_next;
100 from = vlib_frame_vector_args (from_frame);
101 n_left_from = from_frame->n_vectors;
102 next_index = node->cached_next_index;
103 u8 pt_behavior = ~(u8) 0;
104 sr_pt_iface_t *ls = 0;
105 while (n_left_from > 0)
108 vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
110 // Single loop for potentially the last three packets
111 while (n_left_from > 0 && n_left_to_next > 0)
117 ethernet_header_t *en0;
118 ip6_header_t *ip0 = 0;
127 b0 = vlib_get_buffer (vm, bi0);
128 iface = vnet_buffer (b0)->sw_if_index[VLIB_TX];
129 ls = sr_pt_find_iface (iface);
132 en0 = vlib_buffer_get_current (b0);
133 ip0 = (void *) (en0 + 1);
134 unix_time_now_nsec_fraction (&t64.sec, &t64.nsec);
135 pt_midpoint_processing (vm, node, b0, ip0, ls, t64);
136 pt_behavior = PT_BEHAVIOR_MID;
138 if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
140 pt_trace_t *tr = vlib_add_trace (vm, node, b0, sizeof (*tr));
143 tr->load = ls->egress_load;
144 tr->tts_template = ls->tts_template;
145 tr->t64.sec = t64.sec;
146 tr->t64.nsec = t64.nsec;
147 tr->tts = t64.nsec >> 20;
148 tr->behavior = pt_behavior;
150 vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
151 n_left_to_next, bi0, next0);
153 vlib_put_next_frame (vm, node, next_index, n_left_to_next);
156 return from_frame->n_vectors;
159 VLIB_REGISTER_NODE (sr_pt_node) = {
161 .vector_size = sizeof (u32),
162 .format_trace = format_pt_trace,
163 .type = VLIB_NODE_TYPE_INTERNAL,
166 .next_nodes = { [0] = "interface-output" },
169 VNET_FEATURE_INIT (sr_pt_node, static) = {
170 .arc_name = "ip6-output",
172 .runs_after = VNET_FEATURES ("ip6-lookup"),