BIER
[vpp.git] / src / vnet / bier / bier_imp_node.c
1 /*
2  * Copyright (c) 2016 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 <vnet/bier/bier_imp.h>
17 #include <vnet/bier/bier_hdr_inlines.h>
18 #include <vnet/ip/ip.h>
19
20 /**
21  * @brief A struct to hold tracing information for the BIER imposition
22  * node.
23  */
24 typedef struct bier_imp_trace_t_
25 {
26     /**
27      * BIER imposition object hit
28      */
29     index_t imp;
30
31     /**
32      * BIER hdr applied
33      */
34     bier_hdr_t hdr;
35 } bier_imp_trace_t;
36
37 always_inline uword
38 bier_imp_dpo_inline (vlib_main_t * vm,
39                      vlib_node_runtime_t * node,
40                      vlib_frame_t * from_frame,
41                      fib_protocol_t fproto,
42                      bier_hdr_proto_id_t bproto)
43 {
44     u32 n_left_from, next_index, * from, * to_next;
45
46     from = vlib_frame_vector_args (from_frame);
47     n_left_from = from_frame->n_vectors;
48
49     next_index = node->cached_next_index;
50
51     while (n_left_from > 0)
52     {
53         u32 n_left_to_next;
54
55         vlib_get_next_frame(vm, node, next_index, to_next, n_left_to_next);
56
57         while (n_left_from > 0 && n_left_to_next > 0)
58         {
59             vlib_buffer_t * b0;
60             bier_imp_t *bimp0;
61             bier_hdr_t *hdr0;
62             u32 bi0, bii0;
63             u32 next0;
64
65             bi0 = from[0];
66             to_next[0] = bi0;
67             from += 1;
68             to_next += 1;
69             n_left_from -= 1;
70             n_left_to_next -= 1;
71
72             b0 = vlib_get_buffer (vm, bi0);
73
74             bii0 = vnet_buffer(b0)->ip.adj_index[VLIB_TX];
75             bimp0 = bier_imp_get(bii0);
76
77             if (FIB_PROTOCOL_IP4 == fproto)
78             {
79                 /*
80                  * decrement the TTL on ingress to the BIER domain
81                  */
82                 ip4_header_t * ip0 = vlib_buffer_get_current(b0);
83                 u32 checksum0;
84
85                 checksum0 = ip0->checksum + clib_host_to_net_u16 (0x0100);
86                 checksum0 += checksum0 >= 0xffff;
87
88                 ip0->checksum = checksum0;
89                 ip0->ttl -= 1;
90
91                 /*
92                  * calculate an entropy
93                  */
94                 if (0 == vnet_buffer(b0)->ip.flow_hash)
95                 {
96                     vnet_buffer(b0)->ip.flow_hash =
97                         ip4_compute_flow_hash (ip0, IP_FLOW_HASH_DEFAULT);
98                 }
99             }
100             if (FIB_PROTOCOL_IP6 == fproto)
101             {
102                 /*
103                  * decrement the TTL on ingress to the BIER domain
104                  */
105                 ip6_header_t * ip0 = vlib_buffer_get_current(b0);
106
107                 ip0->hop_limit -= 1;
108
109                 /*
110                  * calculate an entropy
111                  */
112                 if (0 == vnet_buffer(b0)->ip.flow_hash)
113                 {
114                     vnet_buffer(b0)->ip.flow_hash =
115                         ip6_compute_flow_hash (ip0, IP_FLOW_HASH_DEFAULT);
116                 }
117             }
118
119             /* Paint the BIER header */
120             vlib_buffer_advance(b0, -(sizeof(bier_hdr_t) +
121                                       bier_hdr_len_id_to_num_bytes(bimp0->bi_tbl.bti_hdr_len)));
122             hdr0 = vlib_buffer_get_current(b0);
123             clib_memcpy(hdr0, &bimp0->bi_hdr,
124                         (sizeof(bier_hdr_t) +
125                          bier_hdr_len_id_to_num_bytes(bimp0->bi_tbl.bti_hdr_len)));
126             /*
127              * Fixup the entropy and protocol, both of which have a
128              * zero value post the paint job
129              */
130             hdr0->bh_oam_dscp_proto |=
131                 clib_host_to_net_u16(bproto << BIER_HDR_PROTO_FIELD_SHIFT);
132             hdr0->bh_first_word |=
133                 clib_host_to_net_u32((vnet_buffer(b0)->ip.flow_hash &
134                                       BIER_HDR_ENTROPY_FIELD_MASK) <<
135                                      BIER_HDR_ENTROPY_FIELD_SHIFT);
136
137             /* next node */
138             next0 = bimp0->bi_dpo[fproto].dpoi_next_node;
139             vnet_buffer(b0)->ip.adj_index[VLIB_TX] =
140                 bimp0->bi_dpo[fproto].dpoi_index;
141
142             if (PREDICT_FALSE(b0->flags & VLIB_BUFFER_IS_TRACED))
143             {
144                 bier_imp_trace_t *tr =
145                     vlib_add_trace (vm, node, b0, sizeof (*tr));
146                 tr->imp = bii0;
147                 tr->hdr = *hdr0;
148             }
149
150             vlib_validate_buffer_enqueue_x1(vm, node, next_index, to_next,
151                                             n_left_to_next, bi0, next0);
152         }
153         vlib_put_next_frame (vm, node, next_index, n_left_to_next);
154     }
155     return from_frame->n_vectors;
156 }
157
158 static u8 *
159 format_bier_imp_trace (u8 * s, va_list * args)
160 {
161     CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
162     CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
163     bier_imp_trace_t * t;
164     u32 indent;
165
166     t = va_arg (*args, bier_imp_trace_t *);
167     indent = format_get_indent (s);
168
169     s = format (s, "%U", format_bier_imp, t->imp, indent, BIER_SHOW_BRIEF);
170     return (s);
171 }
172
173 static uword
174 bier_imp_ip4 (vlib_main_t * vm,
175               vlib_node_runtime_t * node,
176               vlib_frame_t * frame)
177 {
178     return (bier_imp_dpo_inline(vm, node, frame,
179                                 FIB_PROTOCOL_IP4,
180                                 BIER_HDR_PROTO_IPV4));
181 }
182
183 VLIB_REGISTER_NODE (bier_imp_ip4_node) = {
184     .function = bier_imp_ip4,
185     .name = "bier-imp-ip4",
186     .vector_size = sizeof (u32),
187
188     .format_trace = format_bier_imp_trace,
189     .n_next_nodes = 1,
190     .next_nodes = {
191         [0] = "error-drop",
192     }
193 };
194 VLIB_NODE_FUNCTION_MULTIARCH (bier_imp_ip4_node, bier_imp_ip4)
195
196 static uword
197 bier_imp_ip6 (vlib_main_t * vm,
198               vlib_node_runtime_t * node,
199               vlib_frame_t * frame)
200 {
201     return (bier_imp_dpo_inline(vm, node, frame,
202                                 FIB_PROTOCOL_IP6,
203                                 BIER_HDR_PROTO_IPV6));
204 }
205
206 VLIB_REGISTER_NODE (bier_imp_ip6_node) = {
207     .function = bier_imp_ip6,
208     .name = "bier-imp-ip6",
209     .vector_size = sizeof (u32),
210
211     .format_trace = format_bier_imp_trace,
212     .n_next_nodes = 1,
213     .next_nodes = {
214         [0] = "error-drop",
215     }
216 };
217 VLIB_NODE_FUNCTION_MULTIARCH (bier_imp_ip6_node, bier_imp_ip6)