2 * Copyright (C) 2018 Cumulus Networks, Inc.
5 * This file is part of FRR.
7 * FRR is free software; you can redistribute it and/or modify it
8 * under the terms of the GNU General Public License as published by the
9 * Free Software Foundation; either version 2, or (at your option) any
12 * FRR is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with FRR; see the file COPYING. If not, write to the Free
19 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
24 #include "zebra_router.h"
25 #include "zebra_memory.h"
26 #include "zebra_pbr.h"
28 struct zebra_router zrouter
;
31 zebra_router_table_entry_compare(const struct zebra_router_table
*e1
,
32 const struct zebra_router_table
*e2
);
34 RB_GENERATE(zebra_router_table_head
, zebra_router_table
,
35 zebra_router_table_entry
, zebra_router_table_entry_compare
);
39 zebra_router_table_entry_compare(const struct zebra_router_table
*e1
,
40 const struct zebra_router_table
*e2
)
42 if (e1
->tableid
< e2
->tableid
)
44 if (e1
->tableid
> e2
->tableid
)
46 if (e1
->ns_id
< e2
->ns_id
)
48 if (e1
->ns_id
> e2
->ns_id
)
50 if (e1
->afi
< e2
->afi
)
52 if (e1
->afi
> e2
->afi
)
54 return (e1
->safi
- e2
->safi
);
58 struct route_table
*zebra_router_find_table(struct zebra_vrf
*zvrf
,
59 uint32_t tableid
, afi_t afi
,
62 struct zebra_router_table finder
;
63 struct zebra_router_table
*zrt
;
65 memset(&finder
, 0, sizeof(finder
));
68 finder
.tableid
= tableid
;
69 finder
.ns_id
= zvrf
->zns
->ns_id
;
70 zrt
= RB_FIND(zebra_router_table_head
, &zrouter
.tables
, &finder
);
78 struct route_table
*zebra_router_get_table(struct zebra_vrf
*zvrf
,
79 uint32_t tableid
, afi_t afi
,
82 struct zebra_router_table finder
;
83 struct zebra_router_table
*zrt
;
84 rib_table_info_t
*info
;
86 memset(&finder
, 0, sizeof(finder
));
89 finder
.tableid
= tableid
;
90 finder
.ns_id
= zvrf
->zns
->ns_id
;
91 zrt
= RB_FIND(zebra_router_table_head
, &zrouter
.tables
, &finder
);
96 zrt
= XCALLOC(MTYPE_ZEBRA_NS
, sizeof(*zrt
));
97 zrt
->tableid
= tableid
;
99 zrt
->ns_id
= zvrf
->zns
->ns_id
;
101 (afi
== AFI_IP6
) ? srcdest_table_init() : route_table_init();
103 info
= XCALLOC(MTYPE_RIB_TABLE_INFO
, sizeof(*info
));
106 info
->safi
= SAFI_UNICAST
;
107 route_table_set_info(zrt
->table
, info
);
108 zrt
->table
->cleanup
= zebra_rtable_node_cleanup
;
110 RB_INSERT(zebra_router_table_head
, &zrouter
.tables
, zrt
);
114 unsigned long zebra_router_score_proto(uint8_t proto
, unsigned short instance
)
116 struct zebra_router_table
*zrt
;
117 unsigned long cnt
= 0;
119 RB_FOREACH (zrt
, zebra_router_table_head
, &zrouter
.tables
) {
120 if (zrt
->ns_id
!= NS_DEFAULT
)
122 cnt
+= rib_score_proto_table(proto
, instance
, zrt
->table
);
127 void zebra_router_sweep_route(void)
129 struct zebra_router_table
*zrt
;
131 RB_FOREACH (zrt
, zebra_router_table_head
, &zrouter
.tables
) {
132 if (zrt
->ns_id
!= NS_DEFAULT
)
134 rib_sweep_table(zrt
->table
);
138 static void zebra_router_free_table(struct zebra_router_table
*zrt
)
142 rib_close_table(zrt
->table
);
144 table_info
= route_table_get_info(zrt
->table
);
145 route_table_finish(zrt
->table
);
146 XFREE(MTYPE_RIB_TABLE_INFO
, table_info
);
147 XFREE(MTYPE_ZEBRA_NS
, zrt
);
150 void zebra_router_terminate(void)
152 struct zebra_router_table
*zrt
, *tmp
;
154 RB_FOREACH_SAFE (zrt
, zebra_router_table_head
, &zrouter
.tables
, tmp
) {
155 RB_REMOVE(zebra_router_table_head
, &zrouter
.tables
, zrt
);
156 zebra_router_free_table(zrt
);
159 hash_clean(zrouter
.rules_hash
, zebra_pbr_rules_free
);
160 hash_free(zrouter
.rules_hash
);
162 hash_clean(zrouter
.ipset_entry_hash
, zebra_pbr_ipset_entry_free
),
163 hash_clean(zrouter
.ipset_hash
, zebra_pbr_ipset_free
);
164 hash_free(zrouter
.ipset_hash
);
165 hash_free(zrouter
.ipset_entry_hash
);
166 hash_clean(zrouter
.iptable_hash
, zebra_pbr_iptable_free
);
167 hash_free(zrouter
.iptable_hash
);
170 void zebra_router_init(void)
172 zrouter
.l3vni_table
= NULL
;
174 zrouter
.rules_hash
= hash_create_size(8, zebra_pbr_rules_hash_key
,
175 zebra_pbr_rules_hash_equal
,
179 hash_create_size(8, zebra_pbr_ipset_hash_key
,
180 zebra_pbr_ipset_hash_equal
, "IPset Hash");
182 zrouter
.ipset_entry_hash
= hash_create_size(
183 8, zebra_pbr_ipset_entry_hash_key
,
184 zebra_pbr_ipset_entry_hash_equal
, "IPset Hash Entry");
186 zrouter
.iptable_hash
= hash_create_size(8, zebra_pbr_iptable_hash_key
,
187 zebra_pbr_iptable_hash_equal
,
188 "IPtable Hash Entry");