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, clib_host_to_net_u32 (t->t64.sec),
42 clib_host_to_net_u32 (t->t64.nsec), t->tts_template, t->tts);
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 =
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)
73 case SR_PT_TTS_TEMPLATE_0:
74 hbh_opt_pt->cmd_stack[0].tts =
75 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_0;
77 case SR_PT_TTS_TEMPLATE_1:
78 hbh_opt_pt->cmd_stack[0].tts =
79 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_1;
81 case SR_PT_TTS_TEMPLATE_2:
82 hbh_opt_pt->cmd_stack[0].tts =
83 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_2;
85 case SR_PT_TTS_TEMPLATE_3:
86 hbh_opt_pt->cmd_stack[0].tts =
87 t64.nsec >> SR_PT_TTS_SHIFT_TEMPLATE_0;
97 VLIB_NODE_FN (sr_pt_node)
98 (vlib_main_t *vm, vlib_node_runtime_t *node, vlib_frame_t *from_frame)
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)
109 vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
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);
115 // Single loop for potentially the last three packets
116 while (n_left_from > 0 && n_left_to_next > 0)
122 ethernet_header_t *en0;
123 ip6_header_t *ip0 = 0;
131 b0 = vlib_get_buffer (vm, bi0);
132 iface = vnet_buffer (b0)->sw_if_index[VLIB_TX];
133 ls = sr_pt_find_iface (iface);
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;
141 if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
143 pt_trace_t *tr = vlib_add_trace (vm, node, b0, sizeof (*tr));
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;
153 vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
154 n_left_to_next, bi0, next0);
156 vlib_put_next_frame (vm, node, next_index, n_left_to_next);
159 return from_frame->n_vectors;
162 VLIB_REGISTER_NODE (sr_pt_node) = {
164 .vector_size = sizeof (u32),
165 .format_trace = format_pt_trace,
166 .type = VLIB_NODE_TYPE_INTERNAL,
169 .next_nodes = { [0] = "interface-output" },
172 VNET_FEATURE_INIT (sr_pt_node, static) = {
173 .arc_name = "ip6-output",