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:
7 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 * The load-balance object represents an ECMP choice. The buckets of a load
18 * balance object point to the sub-graph after the choice is made.
19 * THe load-balance object is also object type returned from a FIB table lookup.
20 * As such it needs to represent the case where there is only one coice. It may
21 * seem like overkill to use a load-balance object in this case, but the reason
22 * is for performance. If the load-balance object were not the result of the FIB
23 * lookup, then some other object would be. The case where there was ECMP
24 * this other object would need a load-balance as a parent and hence just add
25 * an unnecessary indirection.
27 * It is also the object in the DP that represents a via-fib-entry in a recursive
32 #ifndef __LOAD_BALANCE_H__
33 #define __LOAD_BALANCE_H__
35 #include <vlib/vlib.h>
36 #include <vnet/ip/lookup.h>
37 #include <vnet/dpo/dpo.h>
38 #include <vnet/fib/fib_types.h>
39 #include <vnet/fib/fib_entry.h>
44 typedef struct load_balance_main_t_
46 vlib_combined_counter_main_t lbm_to_counters;
47 vlib_combined_counter_main_t lbm_via_counters;
48 } load_balance_main_t;
50 extern load_balance_main_t load_balance_main;
53 * The number of buckets that a load-balance object can have and still
54 * fit in one cache-line
56 #define LB_NUM_INLINE_BUCKETS 4
59 * @brief One path from an [EU]CMP set that the client wants to add to a
62 typedef struct load_balance_path_t_ {
64 * ID of the Data-path object.
69 * The index of the FIB path
71 fib_node_index_t path_index;
74 * weight for the path.
77 } load_balance_path_t;
80 * Flags controlling load-balance creation and modification
82 typedef enum load_balance_attr_t_ {
83 LOAD_BALANCE_ATTR_USES_MAP = 0,
84 LOAD_BALANCE_ATTR_STICKY = 1,
85 } load_balance_attr_t;
87 #define LOAD_BALANCE_ATTR_NAMES { \
88 [LOAD_BALANCE_ATTR_USES_MAP] = "uses-map", \
89 [LOAD_BALANCE_ATTR_STICKY] = "sticky", \
92 #define FOR_EACH_LOAD_BALANCE_ATTR(_attr) \
93 for (_attr = 0; _attr <= LOAD_BALANCE_ATTR_STICKY; _attr++)
95 typedef enum load_balance_flags_t_ {
96 LOAD_BALANCE_FLAG_NONE = 0,
97 LOAD_BALANCE_FLAG_USES_MAP = (1 << 0),
98 LOAD_BALANCE_FLAG_STICKY = (1 << 1),
99 } __attribute__((packed)) load_balance_flags_t;
102 * The FIB DPO provieds;
103 * - load-balancing over the next DPOs in the chain/graph
104 * - per-route counters
106 typedef struct load_balance_t_ {
108 * required for pool_get_aligned.
109 * memebers used in the switch path come first!
111 CLIB_CACHE_LINE_ALIGN_MARK(cacheline0);
114 * number of buckets in the load-balance. always a power of 2.
118 * number of buckets in the load-balance - 1. used in the switch path
119 * as part of the hash calculation.
121 u16 lb_n_buckets_minus_1;
124 * The protocol of packets that traverse this LB.
125 * need in combination with the flow hash config to determine how to hash.
128 dpo_proto_t lb_proto;
131 * Flags concenring the LB's creation and modification
133 load_balance_flags_t lb_flags;
136 * Flags from the load-balance's associated fib_entry_t
138 fib_entry_flag_t lb_fib_entry_flags;
141 * The number of locks, which is approximately the number of users,
142 * of this load-balance.
143 * Load-balance objects of via-entries are heavily shared by recursives,
144 * so the lock count is a u32.
149 * index of the load-balance map, INVALID if this LB does not use one
154 * This is the index of the uRPF list for this LB
159 * the hash config to use when selecting a bucket. this is a u16
161 flow_hash_config_t lb_hash_config;
164 * Vector of buckets containing the next DPOs, sized as lbo_num
166 dpo_id_t *lb_buckets;
169 * The rest of the cache line is used for buckets. In the common case
170 * where there there are less than 4 buckets, then the buckets are
171 * on the same cachlie and we save ourselves a pointer dereferance in
174 dpo_id_t lb_buckets_inline[LB_NUM_INLINE_BUCKETS];
177 STATIC_ASSERT(sizeof(load_balance_t) <= CLIB_CACHE_LINE_BYTES,
178 "A load_balance object size exceeds one cachline");
181 * Flags controlling load-balance formatting/display
183 typedef enum load_balance_format_flags_t_ {
184 LOAD_BALANCE_FORMAT_NONE,
185 LOAD_BALANCE_FORMAT_DETAIL = (1 << 0),
186 } load_balance_format_flags_t;
188 extern index_t load_balance_create(u32 num_buckets,
189 dpo_proto_t lb_proto,
190 flow_hash_config_t fhc);
191 extern void load_balance_multipath_update(
193 const load_balance_path_t * raw_next_hops,
194 load_balance_flags_t flags);
196 extern void load_balance_set_bucket(index_t lbi,
198 const dpo_id_t *next);
199 extern void load_balance_set_urpf(index_t lbi,
201 extern void load_balance_set_fib_entry_flags(index_t lbi,
202 fib_entry_flag_t flags);
203 extern index_t load_balance_get_urpf(index_t lbi);
205 extern u8* format_load_balance(u8 * s, va_list * args);
207 extern const dpo_id_t *load_balance_get_bucket(index_t lbi,
209 extern int load_balance_is_drop(const dpo_id_t *dpo);
210 extern u16 load_balance_n_buckets(index_t lbi);
212 extern f64 load_balance_get_multipath_tolerance(void);
215 * The encapsulation breakages are for fast DP access
217 extern load_balance_t *load_balance_pool;
218 static inline load_balance_t*
219 load_balance_get (index_t lbi)
221 return (pool_elt_at_index(load_balance_pool, lbi));
224 #define LB_HAS_INLINE_BUCKETS(_lb) \
225 ((_lb)->lb_n_buckets <= LB_NUM_INLINE_BUCKETS)
227 static inline const dpo_id_t *
228 load_balance_get_bucket_i (const load_balance_t *lb,
231 ASSERT(bucket < lb->lb_n_buckets);
233 if (PREDICT_TRUE(LB_HAS_INLINE_BUCKETS(lb)))
235 return (&lb->lb_buckets_inline[bucket]);
239 return (&lb->lb_buckets[bucket]);
243 extern void load_balance_module_init(void);