14c7310fbf3ea29d56b7fb6fe07e3d71200f866a
[vpp.git] / vnet / vnet / fib / fib_entry_src_mpls.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/mpls/mpls_types.h>
17 #include <vnet/dpo/drop_dpo.h>
18
19 #include <vnet/fib/fib_table.h>
20 #include <vnet/fib/fib_entry.h>
21 #include <vnet/fib/fib_entry_src.h>
22 #include <vnet/fib/mpls_fib.h>
23
24 /**
25  * Source initialisation Function 
26  */
27 static void
28 fib_entry_src_mpls_init (fib_entry_src_t *src)
29 {
30     mpls_eos_bit_t eos;
31
32     src->fes_flags = FIB_ENTRY_SRC_FLAG_NONE;
33     src->mpls.fesm_label = MPLS_LABEL_INVALID;
34
35     FOR_EACH_MPLS_EOS_BIT(eos)
36     {
37         src->mpls.fesm_lfes[eos] = FIB_NODE_INDEX_INVALID;
38     }
39 }
40
41 /**
42  * Source deinitialisation Function 
43  */
44 static void
45 fib_entry_src_mpls_deinit (fib_entry_src_t *src)
46 {
47 }
48
49 static void
50 fib_entry_src_mpls_remove (fib_entry_src_t *src)
51 {
52     src->fes_pl = FIB_NODE_INDEX_INVALID;
53     src->mpls.fesm_label = MPLS_LABEL_INVALID;
54 }
55
56 static void
57 fib_entry_src_mpls_add (fib_entry_src_t *src,
58                         const fib_entry_t *entry,
59                         fib_entry_flag_t flags,
60                         fib_protocol_t proto,
61                         const dpo_id_t *dpo)
62 {
63     src->fes_pl =
64         fib_path_list_create_special(proto,
65                                      FIB_PATH_LIST_FLAG_DROP,
66                                      drop_dpo_get(fib_proto_to_dpo(proto)));
67 }
68
69 static void
70 fib_entry_src_mpls_set_data (fib_entry_src_t *src,
71                              const fib_entry_t *entry,
72                              const void *data)
73 {
74     fib_protocol_t payload_proto;
75     fib_node_index_t fei;
76     mpls_label_t label;
77     mpls_eos_bit_t eos;
78
79     /*
80      * post MPLS table alloc and the possible rea-alloc of fib entrys
81      * the entry pointer will no longer be valid. so save its index
82      */
83     payload_proto = entry->fe_prefix.fp_proto;
84     fei = fib_entry_get_index(entry);
85     label = *(mpls_label_t*)data;
86
87     if (MPLS_LABEL_INVALID == label)
88     {
89         /*
90          * removing the local label
91          */
92         FOR_EACH_MPLS_EOS_BIT(eos)
93         {
94             fib_table_entry_delete_index(src->mpls.fesm_lfes[eos],
95                                          FIB_SOURCE_SPECIAL);
96         }
97         fib_table_unlock(MPLS_FIB_DEFAULT_TABLE_ID, FIB_PROTOCOL_MPLS);
98         src->mpls.fesm_label = label;
99     }
100     else
101     {
102         fib_prefix_t prefix = {
103             .fp_proto = FIB_PROTOCOL_MPLS,
104             .fp_label = label,
105         };
106         fib_node_index_t fib_index;
107         dpo_id_t dpo = DPO_INVALID;
108
109         /*
110          * adding a new local label. make sure the MPLS fib exists.
111          */
112         if (MPLS_LABEL_INVALID == src->mpls.fesm_label)
113         {
114             fib_index =
115                 fib_table_find_or_create_and_lock(FIB_PROTOCOL_MPLS,
116                                                   MPLS_FIB_DEFAULT_TABLE_ID);
117         }
118         else
119         {
120             fib_index = mpls_fib_index_from_table_id(MPLS_FIB_DEFAULT_TABLE_ID);
121
122             /*
123              * if this is a change in label, reomve the old one first
124              */
125             if (src->mpls.fesm_label != label)
126             {
127                 FOR_EACH_MPLS_EOS_BIT(eos)
128                 {
129                     ASSERT(FIB_NODE_INDEX_INVALID != src->mpls.fesm_lfes[eos]);
130                     fib_table_entry_delete_index(src->mpls.fesm_lfes[eos],
131                                                  FIB_SOURCE_SPECIAL);
132                 }
133             }
134         }
135
136         src->mpls.fesm_label = label;
137
138         FOR_EACH_MPLS_EOS_BIT(eos)
139         {
140             prefix.fp_eos = eos;
141             prefix.fp_payload_proto = fib_proto_to_dpo(payload_proto);
142
143             fib_entry_contribute_forwarding(fei,
144                                             (eos ?
145                                              FIB_FORW_CHAIN_TYPE_MPLS_EOS :
146                                              FIB_FORW_CHAIN_TYPE_MPLS_NON_EOS),
147                                             &dpo);
148             src->mpls.fesm_lfes[eos] = 
149                 fib_table_entry_special_dpo_add(fib_index,
150                                                 &prefix,
151                                                 FIB_SOURCE_SPECIAL,
152                                                 FIB_ENTRY_FLAG_EXCLUSIVE,
153                                                 &dpo);
154             dpo_reset(&dpo);
155         }
156     }
157 }
158
159 static const void *
160 fib_entry_src_mpls_get_data (fib_entry_src_t *src,
161                              const fib_entry_t *entry)
162 {
163     return (&(src->mpls.fesm_label));
164 }
165
166 static u8*
167 fib_entry_src_mpls_format (fib_entry_src_t *src,
168                            u8* s)
169 {
170     return (format(s, "MPLS local-label:%d", src->mpls.fesm_label));
171 }
172
173 const static fib_entry_src_vft_t mpls_src_vft = {
174     .fesv_init = fib_entry_src_mpls_init,
175     .fesv_deinit = fib_entry_src_mpls_deinit,
176     .fesv_add = fib_entry_src_mpls_add,
177     .fesv_remove = fib_entry_src_mpls_remove,
178     .fesv_format = fib_entry_src_mpls_format,
179     .fesv_set_data = fib_entry_src_mpls_set_data,
180     .fesv_get_data = fib_entry_src_mpls_get_data,
181     /*
182      * .fesv_fwd_update = fib_entry_src_mpls_fwd_update,
183      *  When the forwarding for the IP entry is updated, any MPLS chains
184      * it has created are also updated. Since the MPLS entry will have already
185      * installed that chain/load-balance there is no need to update the netry
186      * FIXME: later: propagate any walk to the children of the MPLS entry. for SR
187      */
188 };
189
190 void
191 fib_entry_src_mpls_register (void)
192 {
193     fib_entry_src_register(FIB_SOURCE_MPLS, &mpls_src_vft);    
194 }
195
196