]>
Commit | Line | Data |
---|---|---|
e5c83d9b DS |
1 | /* |
2 | * Zebra connect code. | |
3 | * Copyright (C) 2018 Cumulus Networks, Inc. | |
4 | * Donald Sharp | |
5 | * | |
6 | * FRR is free software; you can redistribute it and/or modify it | |
7 | * under the terms of the GNU General Public License as published by the | |
8 | * Free Software Foundation; either version 2, or (at your option) any | |
9 | * later version. | |
10 | * | |
11 | * FRR is distributed in the hope that it will be useful, but | |
12 | * WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | * General Public License for more details. | |
15 | * | |
16 | * You should have received a copy of the GNU General Public License along | |
17 | * with this program; see the file COPYING; if not, write to the Free Software | |
18 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA | |
19 | */ | |
20 | #include <zebra.h> | |
21 | ||
22 | #include "thread.h" | |
23 | #include "command.h" | |
24 | #include "network.h" | |
25 | #include "prefix.h" | |
26 | #include "routemap.h" | |
27 | #include "table.h" | |
28 | #include "stream.h" | |
29 | #include "memory.h" | |
30 | #include "zclient.h" | |
31 | #include "filter.h" | |
32 | #include "plist.h" | |
33 | #include "log.h" | |
34 | #include "nexthop.h" | |
35 | #include "nexthop_group.h" | |
36 | ||
37 | #include "pbr_nht.h" | |
38 | #include "pbr_map.h" | |
39 | #include "pbr_memory.h" | |
40 | #include "pbr_zebra.h" | |
41 | #include "pbr_debug.h" | |
42 | ||
43 | DEFINE_MTYPE_STATIC(PBRD, PBR_INTERFACE, "PBR Interface") | |
44 | ||
45 | /* Zebra structure to hold current status. */ | |
d3765386 | 46 | struct zclient *zclient; |
e5c83d9b | 47 | |
b13e5ad6 | 48 | struct pbr_interface *pbr_if_new(struct interface *ifp) |
e5c83d9b DS |
49 | { |
50 | struct pbr_interface *pbr_ifp; | |
51 | ||
52 | zassert(ifp); | |
53 | zassert(!ifp->info); | |
54 | ||
55 | pbr_ifp = XCALLOC(MTYPE_PBR_INTERFACE, sizeof(*pbr_ifp)); | |
56 | ||
10a00758 DS |
57 | ifp->info = pbr_ifp; |
58 | return pbr_ifp; | |
e5c83d9b DS |
59 | } |
60 | ||
61 | /* Inteface addition message from zebra. */ | |
62 | static int interface_add(int command, struct zclient *zclient, | |
63 | zebra_size_t length, vrf_id_t vrf_id) | |
64 | { | |
65 | struct interface *ifp; | |
66 | ||
67 | ifp = zebra_interface_add_read(zclient->ibuf, vrf_id); | |
68 | ||
69 | if (!ifp) | |
70 | return 0; | |
71 | ||
10a00758 DS |
72 | if (!ifp->info) |
73 | pbr_if_new(ifp); | |
e5c83d9b DS |
74 | |
75 | return 0; | |
76 | } | |
77 | ||
78 | static int interface_delete(int command, struct zclient *zclient, | |
79 | zebra_size_t length, vrf_id_t vrf_id) | |
80 | { | |
81 | struct interface *ifp; | |
82 | struct stream *s; | |
83 | ||
84 | s = zclient->ibuf; | |
85 | /* zebra_interface_state_read () updates interface structure in iflist | |
86 | */ | |
87 | ifp = zebra_interface_state_read(s, vrf_id); | |
88 | ||
89 | if (ifp == NULL) | |
90 | return 0; | |
91 | ||
92 | if_set_index(ifp, IFINDEX_INTERNAL); | |
93 | ||
94 | return 0; | |
95 | } | |
96 | ||
97 | static int interface_address_add(int command, struct zclient *zclient, | |
98 | zebra_size_t length, vrf_id_t vrf_id) | |
99 | { | |
100 | zebra_interface_address_read(command, zclient->ibuf, vrf_id); | |
101 | ||
102 | return 0; | |
103 | } | |
104 | ||
105 | static int interface_address_delete(int command, struct zclient *zclient, | |
106 | zebra_size_t length, vrf_id_t vrf_id) | |
107 | { | |
108 | struct connected *c; | |
109 | ||
110 | c = zebra_interface_address_read(command, zclient->ibuf, vrf_id); | |
111 | ||
112 | if (!c) | |
113 | return 0; | |
114 | ||
115 | connected_free(c); | |
116 | return 0; | |
117 | } | |
118 | ||
119 | static int interface_state_up(int command, struct zclient *zclient, | |
120 | zebra_size_t length, vrf_id_t vrf_id) | |
121 | { | |
122 | ||
d7b3ad40 | 123 | zebra_interface_state_read(zclient->ibuf, vrf_id); |
e5c83d9b DS |
124 | |
125 | return 0; | |
126 | } | |
127 | ||
128 | static int interface_state_down(int command, struct zclient *zclient, | |
129 | zebra_size_t length, vrf_id_t vrf_id) | |
130 | { | |
131 | ||
132 | zebra_interface_state_read(zclient->ibuf, vrf_id); | |
133 | ||
134 | return 0; | |
135 | } | |
136 | ||
137 | static int route_notify_owner(int command, struct zclient *zclient, | |
138 | zebra_size_t length, vrf_id_t vrf_id) | |
139 | { | |
140 | struct prefix p; | |
141 | enum zapi_route_notify_owner note; | |
142 | uint32_t table_id; | |
143 | char buf[PREFIX_STRLEN]; | |
144 | ||
145 | prefix2str(&p, buf, sizeof(buf)); | |
146 | ||
147 | if (!zapi_route_notify_decode(zclient->ibuf, &p, &table_id, ¬e)) | |
148 | return -1; | |
149 | ||
150 | switch (note) { | |
151 | case ZAPI_ROUTE_FAIL_INSTALL: | |
152 | DEBUGD(&pbr_dbg_zebra, | |
153 | "%s: [%s] Route install failure for table: %u", | |
154 | __PRETTY_FUNCTION__, buf, table_id); | |
155 | break; | |
156 | case ZAPI_ROUTE_BETTER_ADMIN_WON: | |
157 | DEBUGD(&pbr_dbg_zebra, | |
158 | "%s: [%s] Route better admin distance won for table: %u", | |
159 | __PRETTY_FUNCTION__, buf, table_id); | |
160 | break; | |
161 | case ZAPI_ROUTE_INSTALLED: | |
162 | DEBUGD(&pbr_dbg_zebra, | |
163 | "%s: [%s] Route installed succeeded for table: %u", | |
164 | __PRETTY_FUNCTION__, buf, table_id); | |
165 | pbr_nht_route_installed_for_table(table_id); | |
166 | break; | |
167 | case ZAPI_ROUTE_REMOVED: | |
168 | DEBUGD(&pbr_dbg_zebra, | |
169 | "%s: [%s] Route Removed succeeded for table: %u", | |
170 | __PRETTY_FUNCTION__, buf, table_id); | |
171 | pbr_nht_route_removed_for_table(table_id); | |
172 | break; | |
173 | case ZAPI_ROUTE_REMOVE_FAIL: | |
174 | DEBUGD(&pbr_dbg_zebra, | |
175 | "%s: [%s] Route remove fail for table: %u", | |
176 | __PRETTY_FUNCTION__, buf, table_id); | |
177 | break; | |
178 | } | |
179 | ||
180 | return 0; | |
181 | } | |
182 | ||
183 | static int rule_notify_owner(int command, struct zclient *zclient, | |
184 | zebra_size_t length, vrf_id_t vrf_id) | |
185 | { | |
186 | uint32_t seqno, priority, unique; | |
187 | enum zapi_rule_notify_owner note; | |
188 | struct pbr_map_sequence *pbrms; | |
37c606ff | 189 | struct pbr_map_interface *pmi; |
e5c83d9b | 190 | ifindex_t ifi; |
37c606ff | 191 | uint64_t installed; |
e5c83d9b DS |
192 | |
193 | if (!zapi_rule_notify_decode(zclient->ibuf, &seqno, &priority, &unique, | |
194 | &ifi, ¬e)) | |
195 | return -1; | |
196 | ||
37c606ff DS |
197 | pmi = NULL; |
198 | pbrms = pbrms_lookup_unique(unique, ifi, &pmi); | |
e5c83d9b DS |
199 | if (!pbrms) { |
200 | DEBUGD(&pbr_dbg_zebra, | |
201 | "%s: Failure to lookup pbrms based upon %u", | |
202 | __PRETTY_FUNCTION__, unique); | |
203 | return 0; | |
204 | } | |
205 | ||
37c606ff DS |
206 | installed = 1 << pmi->install_bit; |
207 | ||
e5c83d9b DS |
208 | switch (note) { |
209 | case ZAPI_RULE_FAIL_INSTALL: | |
210 | DEBUGD(&pbr_dbg_zebra, "%s: Recieved RULE_FAIL_INSTALL", | |
211 | __PRETTY_FUNCTION__); | |
37c606ff | 212 | pbrms->installed &= ~installed; |
e5c83d9b DS |
213 | break; |
214 | case ZAPI_RULE_INSTALLED: | |
37c606ff | 215 | pbrms->installed |= installed; |
e5c83d9b DS |
216 | DEBUGD(&pbr_dbg_zebra, "%s: Recived RULE_INSTALLED", |
217 | __PRETTY_FUNCTION__); | |
218 | break; | |
373dd3b5 | 219 | case ZAPI_RULE_FAIL_REMOVE: |
e5c83d9b | 220 | case ZAPI_RULE_REMOVED: |
0f03639d | 221 | pbrms->installed &= ~installed; |
e5c83d9b DS |
222 | DEBUGD(&pbr_dbg_zebra, "%s: Received RULE REMOVED", |
223 | __PRETTY_FUNCTION__); | |
224 | break; | |
225 | } | |
226 | ||
227 | return 0; | |
228 | } | |
229 | ||
230 | static void zebra_connected(struct zclient *zclient) | |
231 | { | |
232 | zclient_send_reg_requests(zclient, VRF_DEFAULT); | |
233 | } | |
234 | ||
235 | static void route_add_helper(struct zapi_route *api, struct nexthop_group nhg, | |
236 | uint8_t install_afi) | |
237 | { | |
238 | struct zapi_nexthop *api_nh; | |
239 | struct nexthop *nhop; | |
240 | int i; | |
241 | ||
242 | api->prefix.family = install_afi; | |
243 | ||
244 | i = 0; | |
245 | for (ALL_NEXTHOPS(nhg, nhop)) { | |
246 | api_nh = &api->nexthops[i]; | |
247 | api_nh->vrf_id = nhop->vrf_id; | |
248 | api_nh->type = nhop->type; | |
249 | switch (nhop->type) { | |
250 | case NEXTHOP_TYPE_IPV4: | |
251 | api_nh->gate.ipv4 = nhop->gate.ipv4; | |
252 | break; | |
253 | case NEXTHOP_TYPE_IPV4_IFINDEX: | |
254 | api_nh->gate.ipv4 = nhop->gate.ipv4; | |
255 | api_nh->ifindex = nhop->ifindex; | |
256 | break; | |
257 | case NEXTHOP_TYPE_IFINDEX: | |
258 | api_nh->ifindex = nhop->ifindex; | |
259 | break; | |
260 | case NEXTHOP_TYPE_IPV6: | |
261 | memcpy(&api_nh->gate.ipv6, &nhop->gate.ipv6, 16); | |
262 | break; | |
263 | case NEXTHOP_TYPE_IPV6_IFINDEX: | |
264 | api_nh->ifindex = nhop->ifindex; | |
265 | memcpy(&api_nh->gate.ipv6, &nhop->gate.ipv6, 16); | |
266 | break; | |
267 | case NEXTHOP_TYPE_BLACKHOLE: | |
268 | api_nh->bh_type = nhop->bh_type; | |
269 | break; | |
270 | } | |
271 | i++; | |
272 | } | |
273 | api->nexthop_num = i; | |
274 | ||
275 | zclient_route_send(ZEBRA_ROUTE_ADD, zclient, api); | |
276 | } | |
277 | ||
278 | /* | |
279 | * This function assumes a default route is being | |
280 | * installed into the appropriate tableid | |
281 | */ | |
282 | void route_add(struct pbr_nexthop_group_cache *pnhgc, struct nexthop_group nhg, | |
283 | afi_t install_afi) | |
284 | { | |
285 | struct zapi_route api; | |
286 | ||
287 | memset(&api, 0, sizeof(api)); | |
288 | ||
289 | api.vrf_id = VRF_DEFAULT; | |
290 | api.type = ZEBRA_ROUTE_PBR; | |
291 | api.safi = SAFI_UNICAST; | |
292 | /* | |
293 | * Sending a default route | |
294 | */ | |
295 | api.tableid = pnhgc->table_id; | |
296 | SET_FLAG(api.flags, ZEBRA_FLAG_ALLOW_RECURSION); | |
297 | SET_FLAG(api.message, ZAPI_MESSAGE_TABLEID); | |
298 | SET_FLAG(api.message, ZAPI_MESSAGE_NEXTHOP); | |
299 | switch (install_afi) { | |
300 | case AFI_MAX: | |
301 | route_add_helper(&api, nhg, AF_INET); | |
302 | route_add_helper(&api, nhg, AF_INET6); | |
303 | break; | |
304 | case AFI_IP: | |
305 | route_add_helper(&api, nhg, AF_INET); | |
306 | break; | |
307 | case AFI_IP6: | |
308 | route_add_helper(&api, nhg, AF_INET6); | |
309 | break; | |
310 | case AFI_L2VPN: | |
311 | DEBUGD(&pbr_dbg_zebra, | |
312 | "%s: Asked to install unsupported route type: L2VPN", | |
313 | __PRETTY_FUNCTION__); | |
314 | break; | |
315 | } | |
316 | } | |
317 | ||
318 | /* | |
319 | * This function assumes a default route is being | |
320 | * removed from the appropriate tableid | |
321 | */ | |
322 | void route_delete(struct pbr_nexthop_group_cache *pnhgc, afi_t afi) | |
323 | { | |
324 | struct zapi_route api; | |
325 | ||
326 | memset(&api, 0, sizeof(api)); | |
327 | api.vrf_id = VRF_DEFAULT; | |
328 | api.type = ZEBRA_ROUTE_PBR; | |
329 | api.safi = SAFI_UNICAST; | |
330 | ||
331 | api.tableid = pnhgc->table_id; | |
332 | SET_FLAG(api.message, ZAPI_MESSAGE_TABLEID); | |
333 | ||
334 | switch (afi) { | |
335 | case AFI_IP: | |
336 | api.prefix.family = AF_INET; | |
337 | zclient_route_send(ZEBRA_ROUTE_DELETE, zclient, &api); | |
338 | break; | |
339 | case AFI_IP6: | |
340 | api.prefix.family = AF_INET6; | |
341 | zclient_route_send(ZEBRA_ROUTE_DELETE, zclient, &api); | |
342 | break; | |
343 | case AFI_MAX: | |
344 | api.prefix.family = AF_INET; | |
345 | zclient_route_send(ZEBRA_ROUTE_DELETE, zclient, &api); | |
346 | api.prefix.family = AF_INET6; | |
347 | zclient_route_send(ZEBRA_ROUTE_DELETE, zclient, &api); | |
348 | break; | |
349 | case AFI_L2VPN: | |
350 | DEBUGD(&pbr_dbg_zebra, | |
351 | "%s: Asked to delete unsupported route type: L2VPN", | |
352 | __PRETTY_FUNCTION__); | |
353 | break; | |
354 | } | |
e5c83d9b DS |
355 | } |
356 | ||
357 | static int pbr_zebra_nexthop_update(int command, struct zclient *zclient, | |
358 | zebra_size_t length, vrf_id_t vrf_id) | |
359 | { | |
360 | struct zapi_route nhr; | |
361 | char buf[PREFIX2STR_BUFFER]; | |
362 | uint32_t i; | |
363 | ||
364 | zapi_nexthop_update_decode(zclient->ibuf, &nhr); | |
365 | ||
366 | if (DEBUG_MODE_CHECK(&pbr_dbg_zebra, DEBUG_MODE_ALL)) { | |
367 | ||
368 | DEBUGD(&pbr_dbg_zebra, "%s: Received Nexthop update: %s", | |
369 | __PRETTY_FUNCTION__, | |
370 | prefix2str(&nhr.prefix, buf, sizeof(buf))); | |
371 | ||
372 | DEBUGD(&pbr_dbg_zebra, "%s: (\tNexthops(%u)", | |
373 | __PRETTY_FUNCTION__, nhr.nexthop_num); | |
374 | ||
375 | for (i = 0; i < nhr.nexthop_num; i++) { | |
376 | DEBUGD(&pbr_dbg_zebra, | |
377 | "%s: \tType: %d: vrf: %d, ifindex: %d gate: %s", | |
378 | __PRETTY_FUNCTION__, nhr.nexthops[i].type, | |
379 | nhr.nexthops[i].vrf_id, nhr.nexthops[i].ifindex, | |
380 | inet_ntoa(nhr.nexthops[i].gate.ipv4)); | |
381 | } | |
382 | } | |
383 | ||
384 | pbr_nht_nexthop_update(&nhr); | |
385 | return 1; | |
386 | } | |
387 | ||
388 | extern struct zebra_privs_t pbr_privs; | |
389 | ||
390 | void pbr_zebra_init(void) | |
391 | { | |
392 | struct zclient_options opt = { .receive_notify = true }; | |
393 | ||
394 | zclient = zclient_new_notify(master, &opt); | |
395 | ||
396 | zclient_init(zclient, ZEBRA_ROUTE_PBR, 0, &pbr_privs); | |
397 | zclient->zebra_connected = zebra_connected; | |
398 | zclient->interface_add = interface_add; | |
399 | zclient->interface_delete = interface_delete; | |
400 | zclient->interface_up = interface_state_up; | |
401 | zclient->interface_down = interface_state_down; | |
402 | zclient->interface_address_add = interface_address_add; | |
403 | zclient->interface_address_delete = interface_address_delete; | |
404 | zclient->route_notify_owner = route_notify_owner; | |
405 | zclient->rule_notify_owner = rule_notify_owner; | |
406 | zclient->nexthop_update = pbr_zebra_nexthop_update; | |
407 | } | |
408 | ||
409 | void pbr_send_rnh(struct nexthop *nhop, bool reg) | |
410 | { | |
411 | uint32_t command; | |
412 | struct prefix p; | |
413 | ||
414 | command = (reg) ? | |
415 | ZEBRA_NEXTHOP_REGISTER : ZEBRA_NEXTHOP_UNREGISTER; | |
416 | ||
417 | memset(&p, 0, sizeof(p)); | |
d3765386 | 418 | switch (nhop->type) { |
e5c83d9b DS |
419 | case NEXTHOP_TYPE_IFINDEX: |
420 | case NEXTHOP_TYPE_BLACKHOLE: | |
421 | return; | |
422 | case NEXTHOP_TYPE_IPV4: | |
423 | case NEXTHOP_TYPE_IPV4_IFINDEX: | |
424 | p.family = AF_INET; | |
425 | p.u.prefix4.s_addr = nhop->gate.ipv4.s_addr; | |
426 | p.prefixlen = 32; | |
427 | break; | |
428 | case NEXTHOP_TYPE_IPV6: | |
429 | case NEXTHOP_TYPE_IPV6_IFINDEX: | |
430 | p.family = AF_INET6; | |
431 | memcpy(&p.u.prefix6, &nhop->gate.ipv6, 16); | |
432 | p.prefixlen = 128; | |
433 | break; | |
434 | } | |
435 | ||
436 | if (zclient_send_rnh(zclient, command, &p, | |
437 | false, nhop->vrf_id) < 0) { | |
438 | zlog_warn("%s: Failure to send nexthop to zebra", | |
439 | __PRETTY_FUNCTION__); | |
440 | } | |
441 | } | |
442 | ||
443 | static void pbr_encode_pbr_map_sequence_prefix(struct stream *s, | |
444 | struct prefix *p, | |
49027ce8 | 445 | unsigned char family) |
e5c83d9b DS |
446 | { |
447 | struct prefix any; | |
448 | ||
449 | if (!p) { | |
450 | memset(&any, 0, sizeof(any)); | |
451 | any.family = family; | |
452 | p = &any; | |
453 | } | |
454 | ||
455 | stream_putc(s, p->family); | |
456 | stream_putc(s, p->prefixlen); | |
457 | stream_put(s, &p->u.prefix, prefix_blen(p)); | |
458 | } | |
459 | ||
460 | static void pbr_encode_pbr_map_sequence(struct stream *s, | |
461 | struct pbr_map_sequence *pbrms, | |
462 | struct interface *ifp) | |
463 | { | |
49027ce8 | 464 | unsigned char family; |
e5c83d9b DS |
465 | |
466 | family = AF_INET; | |
49027ce8 DS |
467 | if (pbrms->family) |
468 | family = pbrms->family; | |
e5c83d9b DS |
469 | |
470 | stream_putl(s, pbrms->seqno); | |
471 | stream_putl(s, pbrms->ruleno); | |
472 | stream_putl(s, pbrms->unique); | |
473 | pbr_encode_pbr_map_sequence_prefix(s, pbrms->src, family); | |
474 | stream_putw(s, 0); /* src port */ | |
475 | pbr_encode_pbr_map_sequence_prefix(s, pbrms->dst, family); | |
476 | stream_putw(s, 0); /* dst port */ | |
614827f8 | 477 | stream_putl(s, 0); /* fwmark */ |
e5c83d9b DS |
478 | if (pbrms->nhgrp_name) |
479 | stream_putl(s, pbr_nht_get_table(pbrms->nhgrp_name)); | |
480 | else if (pbrms->nhg) | |
481 | stream_putl(s, pbr_nht_get_table(pbrms->internal_nhg_name)); | |
482 | stream_putl(s, ifp->ifindex); | |
483 | } | |
484 | ||
b13e5ad6 DS |
485 | void pbr_send_pbr_map(struct pbr_map_sequence *pbrms, |
486 | struct pbr_map_interface *pmi, bool install) | |
e5c83d9b | 487 | { |
b13e5ad6 | 488 | struct pbr_map *pbrm = pbrms->parent; |
e5c83d9b | 489 | struct stream *s; |
10a00758 | 490 | uint64_t is_installed = (uint64_t)1 << pmi->install_bit; |
37c606ff DS |
491 | |
492 | is_installed &= pbrms->installed; | |
e5c83d9b | 493 | |
37c606ff DS |
494 | DEBUGD(&pbr_dbg_zebra, "%s: for %s %d(%" PRIu64 ")", |
495 | __PRETTY_FUNCTION__, pbrm->name, install, is_installed); | |
9b71ea4b DS |
496 | |
497 | /* | |
498 | * If we are installed and asked to do so again | |
499 | * just return. If we are not installed and asked | |
500 | * and asked to delete just return; | |
501 | */ | |
37c606ff | 502 | if (install && is_installed) |
9b71ea4b DS |
503 | return; |
504 | ||
37c606ff | 505 | if (!install && !is_installed) |
9b71ea4b | 506 | return; |
e5c83d9b DS |
507 | |
508 | s = zclient->obuf; | |
509 | stream_reset(s); | |
510 | ||
511 | zclient_create_header(s, | |
512 | install ? ZEBRA_RULE_ADD : ZEBRA_RULE_DELETE, | |
513 | VRF_DEFAULT); | |
514 | ||
b13e5ad6 DS |
515 | /* |
516 | * We are sending one item at a time at the moment | |
517 | */ | |
518 | stream_putl(s, 1); | |
e5c83d9b | 519 | |
b13e5ad6 DS |
520 | DEBUGD(&pbr_dbg_zebra, "%s: \t%s %s %d %s %u", |
521 | __PRETTY_FUNCTION__, install ? "Installing" : "Deleting", | |
522 | pbrm->name, install, pmi->ifp->name, pmi->delete); | |
e5c83d9b | 523 | |
b13e5ad6 | 524 | pbr_encode_pbr_map_sequence(s, pbrms, pmi->ifp); |
e5c83d9b | 525 | |
e5c83d9b DS |
526 | stream_putw_at(s, 0, stream_get_endp(s)); |
527 | ||
e5c83d9b DS |
528 | zclient_send_message(zclient); |
529 | } |