1 /* Zebra Policy Based Routing (PBR) main handling.
2 * Copyright (C) 2018 Cumulus Networks, Inc.
4 * This file is part of FRR.
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
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.
16 * You should have received a copy of the GNU General Public License
17 * along with FRR; see the file COPYING. If not, write to the Free
18 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
29 #include "zebra/zebra_router.h"
30 #include "zebra/zebra_pbr.h"
32 #include "zebra/zapi_msg.h"
33 #include "zebra/zebra_memory.h"
34 #include "zebra/zserv.h"
37 DEFINE_MTYPE_STATIC(ZEBRA
, PBR_IPTABLE_IFNAME
, "PBR interface list")
40 static const struct message ipset_type_msg
[] = {
41 {IPSET_NET_PORT_NET
, "net,port,net"},
42 {IPSET_NET_PORT
, "net,port"},
43 {IPSET_NET_NET
, "net,net"},
48 const struct message icmp_typecode_str
[] = {
49 { 0 << 8, "echo-reply"},
51 { 3 << 8, "network-unreachable"},
52 { (3 << 8) + 1, "host-unreachable"},
53 { (3 << 8) + 2, "protocol-unreachable"},
54 { (3 << 8) + 3, "port-unreachable"},
55 { (3 << 8) + 4, "fragmentation-needed"},
56 { (3 << 8) + 5, "source-route-failed"},
57 { (3 << 8) + 6, "network-unknown"},
58 { (3 << 8) + 7, "host-unknown"},
59 { (3 << 8) + 9, "network-prohibited"},
60 { (3 << 8) + 10, "host-prohibited"},
61 { (3 << 8) + 11, "TOS-network-unreachable"},
62 { (3 << 8) + 12, "TOS-host-unreachable"},
63 { (3 << 8) + 13, "communication-prohibited"},
64 { (3 << 8) + 14, "host-precedence-violation"},
65 { (3 << 8) + 15, "precedence-cutoff"},
66 { 4 << 8, "source-quench"},
67 { 5 << 8, "network-redirect"},
68 { (5 << 8) + 1, "host-redirect"},
69 { (5 << 8) + 2, "TOS-network-redirect"},
70 { (5 << 8) + 3, "TOS-host-redirect"},
71 { 8 << 8, "echo-request"},
73 { 9 << 8, "router-advertisement"},
74 { 10 << 8, "router-solicitation"},
75 { 11 << 8, "ttl-zero-during-transit"},
76 { (11 << 8) + 1, "ttl-zero-during-reassembly"},
77 { 12 << 8, "ip-header-bad"},
78 { (12 << 8) + 1, "required-option-missing"},
79 { 13 << 8, "timestamp-request"},
80 { 14 << 8, "timestamp-reply"},
81 { 17 << 8, "address-mask-request"},
82 { 18 << 8, "address-mask-reply"},
87 static const struct message tcp_value_str
[] = {
88 {TCP_HEADER_FIN
, "FIN"},
89 {TCP_HEADER_SYN
, "SYN"},
90 {TCP_HEADER_RST
, "RST"},
91 {TCP_HEADER_PSH
, "PSH"},
92 {TCP_HEADER_ACK
, "ACK"},
93 {TCP_HEADER_URG
, "URG"},
97 static const struct message fragment_value_str
[] = {
100 {4, "first-fragment"},
101 {8, "last-fragment"},
105 /* static function declarations */
106 DEFINE_HOOK(zebra_pbr_ipset_entry_get_stat
,
107 (struct zebra_pbr_ipset_entry
*ipset
, uint64_t *pkts
,
109 (ipset
, pkts
, bytes
))
111 DEFINE_HOOK(zebra_pbr_iptable_get_stat
,
112 (struct zebra_pbr_iptable
*iptable
, uint64_t *pkts
,
114 (iptable
, pkts
, bytes
))
116 DEFINE_HOOK(zebra_pbr_iptable_update
,
117 (int cmd
, struct zebra_pbr_iptable
*iptable
), (cmd
, iptable
));
119 DEFINE_HOOK(zebra_pbr_ipset_entry_update
,
120 (int cmd
, struct zebra_pbr_ipset_entry
*ipset
), (cmd
, ipset
));
122 DEFINE_HOOK(zebra_pbr_ipset_update
,
123 (int cmd
, struct zebra_pbr_ipset
*ipset
), (cmd
, ipset
));
125 /* Private functions */
127 /* Public functions */
128 void zebra_pbr_rules_free(void *arg
)
130 struct zebra_pbr_rule
*rule
;
132 rule
= (struct zebra_pbr_rule
*)arg
;
134 (void)kernel_del_pbr_rule(rule
);
135 XFREE(MTYPE_TMP
, rule
);
138 uint32_t zebra_pbr_rules_hash_key(const void *arg
)
140 const struct zebra_pbr_rule
*rule
;
144 key
= jhash_3words(rule
->rule
.seq
, rule
->rule
.priority
,
145 rule
->rule
.action
.table
,
146 prefix_hash_key(&rule
->rule
.filter
.src_ip
));
148 if (rule
->rule
.filter
.fwmark
)
149 key
= jhash_3words(rule
->rule
.filter
.fwmark
, rule
->vrf_id
,
150 rule
->rule
.ifindex
, key
);
152 key
= jhash_2words(rule
->vrf_id
, rule
->rule
.ifindex
, key
);
154 return jhash_3words(rule
->rule
.filter
.src_port
,
155 rule
->rule
.filter
.dst_port
,
156 prefix_hash_key(&rule
->rule
.filter
.dst_ip
),
157 jhash_1word(rule
->rule
.unique
, key
));
160 bool zebra_pbr_rules_hash_equal(const void *arg1
, const void *arg2
)
162 const struct zebra_pbr_rule
*r1
, *r2
;
164 r1
= (const struct zebra_pbr_rule
*)arg1
;
165 r2
= (const struct zebra_pbr_rule
*)arg2
;
167 if (r1
->rule
.seq
!= r2
->rule
.seq
)
170 if (r1
->rule
.priority
!= r2
->rule
.priority
)
173 if (r1
->rule
.unique
!= r2
->rule
.unique
)
176 if (r1
->rule
.action
.table
!= r2
->rule
.action
.table
)
179 if (r1
->rule
.filter
.src_port
!= r2
->rule
.filter
.src_port
)
182 if (r1
->rule
.filter
.dst_port
!= r2
->rule
.filter
.dst_port
)
185 if (r1
->rule
.filter
.fwmark
!= r2
->rule
.filter
.fwmark
)
188 if (!prefix_same(&r1
->rule
.filter
.src_ip
, &r2
->rule
.filter
.src_ip
))
191 if (!prefix_same(&r1
->rule
.filter
.dst_ip
, &r2
->rule
.filter
.dst_ip
))
194 if (r1
->rule
.ifindex
!= r2
->rule
.ifindex
)
197 if (r1
->vrf_id
!= r2
->vrf_id
)
203 struct pbr_rule_unique_lookup
{
204 struct zebra_pbr_rule
*rule
;
210 static int pbr_rule_lookup_unique_walker(struct hash_bucket
*b
, void *data
)
212 struct pbr_rule_unique_lookup
*pul
= data
;
213 struct zebra_pbr_rule
*rule
= b
->data
;
215 if (pul
->unique
== rule
->rule
.unique
216 && pul
->ifindex
== rule
->rule
.ifindex
217 && pul
->vrf_id
== rule
->vrf_id
) {
219 return HASHWALK_ABORT
;
222 return HASHWALK_CONTINUE
;
225 static struct zebra_pbr_rule
*
226 pbr_rule_lookup_unique(struct zebra_pbr_rule
*zrule
)
228 struct pbr_rule_unique_lookup pul
;
230 pul
.unique
= zrule
->rule
.unique
;
231 pul
.ifindex
= zrule
->rule
.ifindex
;
233 pul
.vrf_id
= zrule
->vrf_id
;
234 hash_walk(zrouter
.rules_hash
, &pbr_rule_lookup_unique_walker
, &pul
);
239 void zebra_pbr_ipset_free(void *arg
)
241 struct zebra_pbr_ipset
*ipset
;
243 ipset
= (struct zebra_pbr_ipset
*)arg
;
244 hook_call(zebra_pbr_ipset_update
, 0, ipset
);
245 XFREE(MTYPE_TMP
, ipset
);
248 uint32_t zebra_pbr_ipset_hash_key(const void *arg
)
250 const struct zebra_pbr_ipset
*ipset
= arg
;
251 uint32_t *pnt
= (uint32_t *)&ipset
->ipset_name
;
252 uint32_t key
= jhash_1word(ipset
->vrf_id
, 0x63ab42de);
254 return jhash2(pnt
, ZEBRA_IPSET_NAME_HASH_SIZE
, key
);
257 bool zebra_pbr_ipset_hash_equal(const void *arg1
, const void *arg2
)
259 const struct zebra_pbr_ipset
*r1
, *r2
;
261 r1
= (const struct zebra_pbr_ipset
*)arg1
;
262 r2
= (const struct zebra_pbr_ipset
*)arg2
;
264 if (r1
->type
!= r2
->type
)
266 if (r1
->unique
!= r2
->unique
)
268 if (r1
->vrf_id
!= r2
->vrf_id
)
271 if (strncmp(r1
->ipset_name
, r2
->ipset_name
,
272 ZEBRA_IPSET_NAME_SIZE
))
277 void zebra_pbr_ipset_entry_free(void *arg
)
279 struct zebra_pbr_ipset_entry
*ipset
;
281 ipset
= (struct zebra_pbr_ipset_entry
*)arg
;
283 hook_call(zebra_pbr_ipset_entry_update
, 0, ipset
);
285 XFREE(MTYPE_TMP
, ipset
);
288 uint32_t zebra_pbr_ipset_entry_hash_key(const void *arg
)
290 const struct zebra_pbr_ipset_entry
*ipset
;
294 key
= prefix_hash_key(&ipset
->src
);
295 key
= jhash_1word(ipset
->unique
, key
);
296 key
= jhash_1word(prefix_hash_key(&ipset
->dst
), key
);
297 key
= jhash(&ipset
->dst_port_min
, 2, key
);
298 key
= jhash(&ipset
->dst_port_max
, 2, key
);
299 key
= jhash(&ipset
->src_port_min
, 2, key
);
300 key
= jhash(&ipset
->src_port_max
, 2, key
);
301 key
= jhash(&ipset
->proto
, 1, key
);
306 bool zebra_pbr_ipset_entry_hash_equal(const void *arg1
, const void *arg2
)
308 const struct zebra_pbr_ipset_entry
*r1
, *r2
;
310 r1
= (const struct zebra_pbr_ipset_entry
*)arg1
;
311 r2
= (const struct zebra_pbr_ipset_entry
*)arg2
;
313 if (r1
->unique
!= r2
->unique
)
316 if (!prefix_same(&r1
->src
, &r2
->src
))
319 if (!prefix_same(&r1
->dst
, &r2
->dst
))
322 if (r1
->src_port_min
!= r2
->src_port_min
)
325 if (r1
->src_port_max
!= r2
->src_port_max
)
328 if (r1
->dst_port_min
!= r2
->dst_port_min
)
331 if (r1
->dst_port_max
!= r2
->dst_port_max
)
334 if (r1
->proto
!= r2
->proto
)
339 void zebra_pbr_iptable_free(void *arg
)
341 struct zebra_pbr_iptable
*iptable
;
342 struct listnode
*node
, *nnode
;
345 iptable
= (struct zebra_pbr_iptable
*)arg
;
346 hook_call(zebra_pbr_iptable_update
, 0, iptable
);
348 if (iptable
->interface_name_list
) {
349 for (ALL_LIST_ELEMENTS(iptable
->interface_name_list
, node
,
351 XFREE(MTYPE_PBR_IPTABLE_IFNAME
, name
);
352 list_delete_node(iptable
->interface_name_list
, node
);
354 list_delete(&iptable
->interface_name_list
);
356 XFREE(MTYPE_TMP
, iptable
);
359 uint32_t zebra_pbr_iptable_hash_key(const void *arg
)
361 const struct zebra_pbr_iptable
*iptable
= arg
;
362 uint32_t *pnt
= (uint32_t *)&(iptable
->ipset_name
);
365 key
= jhash2(pnt
, ZEBRA_IPSET_NAME_HASH_SIZE
,
367 key
= jhash_1word(iptable
->fwmark
, key
);
368 key
= jhash_1word(iptable
->pkt_len_min
, key
);
369 key
= jhash_1word(iptable
->pkt_len_max
, key
);
370 key
= jhash_1word(iptable
->tcp_flags
, key
);
371 key
= jhash_1word(iptable
->tcp_mask_flags
, key
);
372 key
= jhash_1word(iptable
->dscp_value
, key
);
373 key
= jhash_1word(iptable
->protocol
, key
);
374 key
= jhash_1word(iptable
->fragment
, key
);
375 key
= jhash_1word(iptable
->vrf_id
, key
);
377 return jhash_3words(iptable
->filter_bm
, iptable
->type
,
378 iptable
->unique
, key
);
381 bool zebra_pbr_iptable_hash_equal(const void *arg1
, const void *arg2
)
383 const struct zebra_pbr_iptable
*r1
, *r2
;
385 r1
= (const struct zebra_pbr_iptable
*)arg1
;
386 r2
= (const struct zebra_pbr_iptable
*)arg2
;
388 if (r1
->vrf_id
!= r2
->vrf_id
)
390 if (r1
->type
!= r2
->type
)
392 if (r1
->unique
!= r2
->unique
)
394 if (r1
->filter_bm
!= r2
->filter_bm
)
396 if (r1
->fwmark
!= r2
->fwmark
)
398 if (r1
->action
!= r2
->action
)
400 if (strncmp(r1
->ipset_name
, r2
->ipset_name
,
401 ZEBRA_IPSET_NAME_SIZE
))
403 if (r1
->pkt_len_min
!= r2
->pkt_len_min
)
405 if (r1
->pkt_len_max
!= r2
->pkt_len_max
)
407 if (r1
->tcp_flags
!= r2
->tcp_flags
)
409 if (r1
->tcp_mask_flags
!= r2
->tcp_mask_flags
)
411 if (r1
->dscp_value
!= r2
->dscp_value
)
413 if (r1
->fragment
!= r2
->fragment
)
415 if (r1
->protocol
!= r2
->protocol
)
420 static void *pbr_rule_alloc_intern(void *arg
)
422 struct zebra_pbr_rule
*zpr
;
423 struct zebra_pbr_rule
*new;
425 zpr
= (struct zebra_pbr_rule
*)arg
;
427 new = XCALLOC(MTYPE_TMP
, sizeof(*new));
429 memcpy(new, zpr
, sizeof(*zpr
));
434 void zebra_pbr_add_rule(struct zebra_pbr_rule
*rule
)
436 struct zebra_pbr_rule
*unique
=
437 pbr_rule_lookup_unique(rule
);
439 (void)hash_get(zrouter
.rules_hash
, rule
, pbr_rule_alloc_intern
);
440 (void)kernel_add_pbr_rule(rule
);
442 * Rule Replace semantics, if we have an old, install the
443 * new rule, look above, and then delete the old
446 zebra_pbr_del_rule(unique
);
449 void zebra_pbr_del_rule(struct zebra_pbr_rule
*rule
)
451 struct zebra_pbr_rule
*lookup
;
453 lookup
= hash_lookup(zrouter
.rules_hash
, rule
);
454 (void)kernel_del_pbr_rule(rule
);
457 hash_release(zrouter
.rules_hash
, lookup
);
458 XFREE(MTYPE_TMP
, lookup
);
460 zlog_debug("%s: Rule being deleted we know nothing about",
461 __PRETTY_FUNCTION__
);
464 static void zebra_pbr_cleanup_rules(struct hash_bucket
*b
, void *data
)
466 struct zebra_pbr_rule
*rule
= b
->data
;
469 if (rule
->sock
== *sock
) {
470 (void)kernel_del_pbr_rule(rule
);
471 if (hash_release(zrouter
.rules_hash
, rule
))
472 XFREE(MTYPE_TMP
, rule
);
475 "%s: Rule seq: %u is being cleaned but we can't find it in our tables",
476 __func__
, rule
->rule
.seq
);
480 static void zebra_pbr_cleanup_ipset(struct hash_bucket
*b
, void *data
)
482 struct zebra_pbr_ipset
*ipset
= b
->data
;
485 if (ipset
->sock
== *sock
) {
486 hook_call(zebra_pbr_ipset_update
, 0, ipset
);
487 hash_release(zrouter
.ipset_hash
, ipset
);
491 static void zebra_pbr_cleanup_ipset_entry(struct hash_bucket
*b
, void *data
)
493 struct zebra_pbr_ipset_entry
*ipset
= b
->data
;
496 if (ipset
->sock
== *sock
) {
497 hook_call(zebra_pbr_ipset_entry_update
, 0, ipset
);
498 hash_release(zrouter
.ipset_entry_hash
, ipset
);
502 static void zebra_pbr_cleanup_iptable(struct hash_bucket
*b
, void *data
)
504 struct zebra_pbr_iptable
*iptable
= b
->data
;
507 if (iptable
->sock
== *sock
) {
508 hook_call(zebra_pbr_iptable_update
, 0, iptable
);
509 hash_release(zrouter
.iptable_hash
, iptable
);
513 static int zebra_pbr_client_close_cleanup(struct zserv
*client
)
515 int sock
= client
->sock
;
519 hash_iterate(zrouter
.rules_hash
, zebra_pbr_cleanup_rules
, &sock
);
520 hash_iterate(zrouter
.iptable_hash
, zebra_pbr_cleanup_iptable
, &sock
);
521 hash_iterate(zrouter
.ipset_entry_hash
, zebra_pbr_cleanup_ipset_entry
,
523 hash_iterate(zrouter
.ipset_hash
, zebra_pbr_cleanup_ipset
, &sock
);
527 void zebra_pbr_init(void)
529 hook_register(zserv_client_close
, zebra_pbr_client_close_cleanup
);
532 static void *pbr_ipset_alloc_intern(void *arg
)
534 struct zebra_pbr_ipset
*zpi
;
535 struct zebra_pbr_ipset
*new;
537 zpi
= (struct zebra_pbr_ipset
*)arg
;
539 new = XCALLOC(MTYPE_TMP
, sizeof(struct zebra_pbr_ipset
));
541 memcpy(new, zpi
, sizeof(*zpi
));
546 void zebra_pbr_create_ipset(struct zebra_pbr_ipset
*ipset
)
550 (void)hash_get(zrouter
.ipset_hash
, ipset
, pbr_ipset_alloc_intern
);
551 ret
= hook_call(zebra_pbr_ipset_update
, 1, ipset
);
552 kernel_pbr_ipset_add_del_status(ipset
,
553 ret
? ZEBRA_DPLANE_INSTALL_SUCCESS
554 : ZEBRA_DPLANE_INSTALL_FAILURE
);
557 void zebra_pbr_destroy_ipset(struct zebra_pbr_ipset
*ipset
)
559 struct zebra_pbr_ipset
*lookup
;
561 lookup
= hash_lookup(zrouter
.ipset_hash
, ipset
);
562 hook_call(zebra_pbr_ipset_update
, 0, ipset
);
564 hash_release(zrouter
.ipset_hash
, lookup
);
565 XFREE(MTYPE_TMP
, lookup
);
568 "%s: IPSet Entry being deleted we know nothing about",
569 __PRETTY_FUNCTION__
);
572 struct pbr_ipset_name_lookup
{
573 struct zebra_pbr_ipset
*ipset
;
574 char ipset_name
[ZEBRA_IPSET_NAME_SIZE
];
577 const char *zebra_pbr_ipset_type2str(uint32_t type
)
579 return lookup_msg(ipset_type_msg
, type
,
580 "Unrecognized IPset Type");
583 static int zebra_pbr_ipset_pername_walkcb(struct hash_bucket
*bucket
, void *arg
)
585 struct pbr_ipset_name_lookup
*pinl
=
586 (struct pbr_ipset_name_lookup
*)arg
;
587 struct zebra_pbr_ipset
*zpi
= (struct zebra_pbr_ipset
*)bucket
->data
;
589 if (!strncmp(pinl
->ipset_name
, zpi
->ipset_name
,
590 ZEBRA_IPSET_NAME_SIZE
)) {
592 return HASHWALK_ABORT
;
594 return HASHWALK_CONTINUE
;
597 struct zebra_pbr_ipset
*zebra_pbr_lookup_ipset_pername(char *ipsetname
)
599 struct pbr_ipset_name_lookup pinl
;
600 struct pbr_ipset_name_lookup
*ptr
= &pinl
;
604 memset(ptr
, 0, sizeof(struct pbr_ipset_name_lookup
));
605 snprintf((char *)ptr
->ipset_name
, ZEBRA_IPSET_NAME_SIZE
, "%s",
607 hash_walk(zrouter
.ipset_hash
, zebra_pbr_ipset_pername_walkcb
, ptr
);
611 static void *pbr_ipset_entry_alloc_intern(void *arg
)
613 struct zebra_pbr_ipset_entry
*zpi
;
614 struct zebra_pbr_ipset_entry
*new;
616 zpi
= (struct zebra_pbr_ipset_entry
*)arg
;
618 new = XCALLOC(MTYPE_TMP
, sizeof(struct zebra_pbr_ipset_entry
));
620 memcpy(new, zpi
, sizeof(*zpi
));
625 void zebra_pbr_add_ipset_entry(struct zebra_pbr_ipset_entry
*ipset
)
629 (void)hash_get(zrouter
.ipset_entry_hash
, ipset
,
630 pbr_ipset_entry_alloc_intern
);
631 ret
= hook_call(zebra_pbr_ipset_entry_update
, 1, ipset
);
632 kernel_pbr_ipset_entry_add_del_status(ipset
,
633 ret
? ZEBRA_DPLANE_INSTALL_SUCCESS
634 : ZEBRA_DPLANE_INSTALL_FAILURE
);
637 void zebra_pbr_del_ipset_entry(struct zebra_pbr_ipset_entry
*ipset
)
639 struct zebra_pbr_ipset_entry
*lookup
;
641 lookup
= hash_lookup(zrouter
.ipset_entry_hash
, ipset
);
642 hook_call(zebra_pbr_ipset_entry_update
, 0, ipset
);
644 hash_release(zrouter
.ipset_entry_hash
, lookup
);
645 XFREE(MTYPE_TMP
, lookup
);
647 zlog_debug("%s: IPSet being deleted we know nothing about",
648 __PRETTY_FUNCTION__
);
651 static void *pbr_iptable_alloc_intern(void *arg
)
653 struct zebra_pbr_iptable
*zpi
;
654 struct zebra_pbr_iptable
*new;
656 zpi
= (struct zebra_pbr_iptable
*)arg
;
658 new = XCALLOC(MTYPE_TMP
, sizeof(struct zebra_pbr_iptable
));
660 memcpy(new, zpi
, sizeof(*zpi
));
665 void zebra_pbr_add_iptable(struct zebra_pbr_iptable
*iptable
)
669 (void)hash_get(zrouter
.iptable_hash
, iptable
, pbr_iptable_alloc_intern
);
670 ret
= hook_call(zebra_pbr_iptable_update
, 1, iptable
);
671 kernel_pbr_iptable_add_del_status(iptable
,
672 ret
? ZEBRA_DPLANE_INSTALL_SUCCESS
673 : ZEBRA_DPLANE_INSTALL_FAILURE
);
676 void zebra_pbr_del_iptable(struct zebra_pbr_iptable
*iptable
)
678 struct zebra_pbr_iptable
*lookup
;
680 lookup
= hash_lookup(zrouter
.iptable_hash
, iptable
);
681 hook_call(zebra_pbr_iptable_update
, 0, iptable
);
683 struct listnode
*node
, *nnode
;
686 hash_release(zrouter
.iptable_hash
, lookup
);
687 for (ALL_LIST_ELEMENTS(iptable
->interface_name_list
,
688 node
, nnode
, name
)) {
689 XFREE(MTYPE_PBR_IPTABLE_IFNAME
, name
);
690 list_delete_node(iptable
->interface_name_list
,
693 list_delete(&iptable
->interface_name_list
);
694 XFREE(MTYPE_TMP
, lookup
);
696 zlog_debug("%s: IPTable being deleted we know nothing about",
697 __PRETTY_FUNCTION__
);
701 * Handle success or failure of rule (un)install in the kernel.
703 void kernel_pbr_rule_add_del_status(struct zebra_pbr_rule
*rule
,
704 enum zebra_dplane_status res
)
707 case ZEBRA_DPLANE_INSTALL_SUCCESS
:
708 zsend_rule_notify_owner(rule
, ZAPI_RULE_INSTALLED
);
710 case ZEBRA_DPLANE_INSTALL_FAILURE
:
711 zsend_rule_notify_owner(rule
, ZAPI_RULE_FAIL_INSTALL
);
713 case ZEBRA_DPLANE_DELETE_SUCCESS
:
714 zsend_rule_notify_owner(rule
, ZAPI_RULE_REMOVED
);
716 case ZEBRA_DPLANE_DELETE_FAILURE
:
717 zsend_rule_notify_owner(rule
, ZAPI_RULE_FAIL_REMOVE
);
719 case ZEBRA_DPLANE_STATUS_NONE
:
725 * Handle success or failure of ipset (un)install in the kernel.
727 void kernel_pbr_ipset_add_del_status(struct zebra_pbr_ipset
*ipset
,
728 enum zebra_dplane_status res
)
731 case ZEBRA_DPLANE_INSTALL_SUCCESS
:
732 zsend_ipset_notify_owner(ipset
, ZAPI_IPSET_INSTALLED
);
734 case ZEBRA_DPLANE_INSTALL_FAILURE
:
735 zsend_ipset_notify_owner(ipset
, ZAPI_IPSET_FAIL_INSTALL
);
737 case ZEBRA_DPLANE_DELETE_SUCCESS
:
738 zsend_ipset_notify_owner(ipset
, ZAPI_IPSET_REMOVED
);
740 case ZEBRA_DPLANE_DELETE_FAILURE
:
741 zsend_ipset_notify_owner(ipset
, ZAPI_IPSET_FAIL_REMOVE
);
743 case ZEBRA_DPLANE_STATUS_NONE
:
749 * Handle success or failure of ipset (un)install in the kernel.
751 void kernel_pbr_ipset_entry_add_del_status(
752 struct zebra_pbr_ipset_entry
*ipset
,
753 enum zebra_dplane_status res
)
756 case ZEBRA_DPLANE_INSTALL_SUCCESS
:
757 zsend_ipset_entry_notify_owner(ipset
,
758 ZAPI_IPSET_ENTRY_INSTALLED
);
760 case ZEBRA_DPLANE_INSTALL_FAILURE
:
761 zsend_ipset_entry_notify_owner(ipset
,
762 ZAPI_IPSET_ENTRY_FAIL_INSTALL
);
764 case ZEBRA_DPLANE_DELETE_SUCCESS
:
765 zsend_ipset_entry_notify_owner(ipset
,
766 ZAPI_IPSET_ENTRY_REMOVED
);
768 case ZEBRA_DPLANE_DELETE_FAILURE
:
769 zsend_ipset_entry_notify_owner(ipset
,
770 ZAPI_IPSET_ENTRY_FAIL_REMOVE
);
772 case ZEBRA_DPLANE_STATUS_NONE
:
778 * Handle success or failure of ipset (un)install in the kernel.
780 void kernel_pbr_iptable_add_del_status(struct zebra_pbr_iptable
*iptable
,
781 enum zebra_dplane_status res
)
784 case ZEBRA_DPLANE_INSTALL_SUCCESS
:
785 zsend_iptable_notify_owner(iptable
, ZAPI_IPTABLE_INSTALLED
);
787 case ZEBRA_DPLANE_INSTALL_FAILURE
:
788 zsend_iptable_notify_owner(iptable
, ZAPI_IPTABLE_FAIL_INSTALL
);
790 case ZEBRA_DPLANE_DELETE_SUCCESS
:
791 zsend_iptable_notify_owner(iptable
,
792 ZAPI_IPTABLE_REMOVED
);
794 case ZEBRA_DPLANE_DELETE_FAILURE
:
795 zsend_iptable_notify_owner(iptable
,
796 ZAPI_IPTABLE_FAIL_REMOVE
);
798 case ZEBRA_DPLANE_STATUS_NONE
:
804 * Handle rule delete notification from kernel.
806 int kernel_pbr_rule_del(struct zebra_pbr_rule
*rule
)
811 struct zebra_pbr_ipset_entry_unique_display
{
812 struct zebra_pbr_ipset
*zpi
;
814 struct zebra_ns
*zns
;
817 struct zebra_pbr_env_display
{
818 struct zebra_ns
*zns
;
823 static const char *zebra_pbr_prefix2str(union prefixconstptr pu
,
826 const struct prefix
*p
= pu
.p
;
827 char buf
[PREFIX2STR_BUFFER
];
829 if (p
->family
== AF_INET
&& p
->prefixlen
== IPV4_MAX_PREFIXLEN
) {
830 snprintf(str
, size
, "%s", inet_ntop(p
->family
, &p
->u
.prefix
,
831 buf
, PREFIX2STR_BUFFER
));
834 return prefix2str(pu
, str
, size
);
837 static void zebra_pbr_display_icmp(struct vty
*vty
,
838 struct zebra_pbr_ipset_entry
*zpie
)
840 char decoded_str
[20];
843 /* range icmp type */
844 if (zpie
->src_port_max
|| zpie
->dst_port_max
) {
845 vty_out(vty
, ":icmp:[type <%d:%d>;code <%d:%d>",
846 zpie
->src_port_min
, zpie
->src_port_max
,
847 zpie
->dst_port_min
, zpie
->dst_port_max
);
849 port
= ((zpie
->src_port_min
<< 8) & 0xff00) +
850 (zpie
->dst_port_min
& 0xff);
851 memset(decoded_str
, 0, sizeof(decoded_str
));
852 sprintf(decoded_str
, "%d/%d",
855 vty_out(vty
, ":icmp:%s",
856 lookup_msg(icmp_typecode_str
,
861 static void zebra_pbr_display_port(struct vty
*vty
, uint32_t filter_bm
,
862 uint16_t port_min
, uint16_t port_max
,
865 if (!(filter_bm
& PBR_FILTER_PROTO
)) {
867 vty_out(vty
, ":udp/tcp:%d-%d",
870 vty_out(vty
, ":udp/tcp:%d",
874 vty_out(vty
, ":proto %d:%d-%d",
875 proto
, port_min
, port_max
);
877 vty_out(vty
, ":proto %d:%d",
882 static int zebra_pbr_show_ipset_entry_walkcb(struct hash_bucket
*bucket
,
885 struct zebra_pbr_ipset_entry_unique_display
*unique
=
886 (struct zebra_pbr_ipset_entry_unique_display
*)arg
;
887 struct zebra_pbr_ipset
*zpi
= unique
->zpi
;
888 struct vty
*vty
= unique
->vty
;
889 struct zebra_pbr_ipset_entry
*zpie
=
890 (struct zebra_pbr_ipset_entry
*)bucket
->data
;
891 uint64_t pkts
= 0, bytes
= 0;
894 if (zpie
->backpointer
!= zpi
)
895 return HASHWALK_CONTINUE
;
897 if ((zpi
->type
== IPSET_NET_NET
) ||
898 (zpi
->type
== IPSET_NET_PORT_NET
)) {
899 char buf
[PREFIX_STRLEN
];
901 zebra_pbr_prefix2str(&(zpie
->src
), buf
, sizeof(buf
));
902 vty_out(vty
, "\tfrom %s", buf
);
903 if (zpie
->filter_bm
& PBR_FILTER_SRC_PORT
&&
904 zpie
->proto
!= IPPROTO_ICMP
)
905 zebra_pbr_display_port(vty
, zpie
->filter_bm
,
909 vty_out(vty
, " to ");
910 zebra_pbr_prefix2str(&(zpie
->dst
), buf
, sizeof(buf
));
911 vty_out(vty
, "%s", buf
);
912 if (zpie
->filter_bm
& PBR_FILTER_DST_PORT
&&
913 zpie
->proto
!= IPPROTO_ICMP
)
914 zebra_pbr_display_port(vty
, zpie
->filter_bm
,
918 if (zpie
->proto
== IPPROTO_ICMP
)
919 zebra_pbr_display_icmp(vty
, zpie
);
920 } else if ((zpi
->type
== IPSET_NET
) ||
921 (zpi
->type
== IPSET_NET_PORT
)) {
922 char buf
[PREFIX_STRLEN
];
924 if (zpie
->filter_bm
& PBR_FILTER_SRC_IP
) {
925 zebra_pbr_prefix2str(&(zpie
->src
), buf
, sizeof(buf
));
926 vty_out(vty
, "\tfrom %s", buf
);
928 if (zpie
->filter_bm
& PBR_FILTER_SRC_PORT
&&
929 zpie
->proto
!= IPPROTO_ICMP
)
930 zebra_pbr_display_port(vty
, zpie
->filter_bm
,
934 if (zpie
->filter_bm
& PBR_FILTER_DST_IP
) {
935 zebra_pbr_prefix2str(&(zpie
->dst
), buf
, sizeof(buf
));
936 vty_out(vty
, "\tto %s", buf
);
938 if (zpie
->filter_bm
& PBR_FILTER_DST_PORT
&&
939 zpie
->proto
!= IPPROTO_ICMP
)
940 zebra_pbr_display_port(vty
, zpie
->filter_bm
,
944 if (zpie
->proto
== IPPROTO_ICMP
)
945 zebra_pbr_display_icmp(vty
, zpie
);
947 vty_out(vty
, " (%u)\n", zpie
->unique
);
949 ret
= hook_call(zebra_pbr_ipset_entry_get_stat
, zpie
, &pkts
,
952 vty_out(vty
, "\t pkts %" PRIu64
", bytes %" PRIu64
"\n",
954 return HASHWALK_CONTINUE
;
957 static int zebra_pbr_show_ipset_walkcb(struct hash_bucket
*bucket
, void *arg
)
959 struct zebra_pbr_env_display
*uniqueipset
=
960 (struct zebra_pbr_env_display
*)arg
;
961 struct zebra_pbr_ipset
*zpi
= (struct zebra_pbr_ipset
*)bucket
->data
;
962 struct zebra_pbr_ipset_entry_unique_display unique
;
963 struct vty
*vty
= uniqueipset
->vty
;
964 struct zebra_ns
*zns
= uniqueipset
->zns
;
966 vty_out(vty
, "IPset %s type %s\n", zpi
->ipset_name
,
967 zebra_pbr_ipset_type2str(zpi
->type
));
971 hash_walk(zrouter
.ipset_entry_hash
, zebra_pbr_show_ipset_entry_walkcb
,
974 return HASHWALK_CONTINUE
;
977 size_t zebra_pbr_tcpflags_snprintf(char *buffer
, size_t len
,
980 size_t len_written
= 0;
981 static struct message nt
= {0};
982 const struct message
*pnt
;
985 for (pnt
= tcp_value_str
;
986 memcmp(pnt
, &nt
, sizeof(struct message
)); pnt
++)
987 if (pnt
->key
& tcp_val
) {
988 len_written
+= snprintf(buffer
+ len_written
,
999 void zebra_pbr_show_ipset_list(struct vty
*vty
, char *ipsetname
)
1001 struct zebra_pbr_ipset
*zpi
;
1002 struct zebra_ns
*zns
= zebra_ns_lookup(NS_DEFAULT
);
1003 struct zebra_pbr_ipset_entry_unique_display unique
;
1004 struct zebra_pbr_env_display uniqueipset
;
1007 zpi
= zebra_pbr_lookup_ipset_pername(ipsetname
);
1009 vty_out(vty
, "No IPset %s found\n", ipsetname
);
1012 vty_out(vty
, "IPset %s type %s\n", ipsetname
,
1013 zebra_pbr_ipset_type2str(zpi
->type
));
1018 hash_walk(zrouter
.ipset_entry_hash
,
1019 zebra_pbr_show_ipset_entry_walkcb
, &unique
);
1022 uniqueipset
.zns
= zns
;
1023 uniqueipset
.vty
= vty
;
1024 uniqueipset
.name
= NULL
;
1025 hash_walk(zrouter
.ipset_hash
, zebra_pbr_show_ipset_walkcb
,
1029 struct pbr_rule_fwmark_lookup
{
1030 struct zebra_pbr_rule
*ptr
;
1034 static int zebra_pbr_rule_lookup_fwmark_walkcb(struct hash_bucket
*bucket
,
1037 struct pbr_rule_fwmark_lookup
*iprule
=
1038 (struct pbr_rule_fwmark_lookup
*)arg
;
1039 struct zebra_pbr_rule
*zpr
= (struct zebra_pbr_rule
*)bucket
->data
;
1041 if (iprule
->fwmark
== zpr
->rule
.filter
.fwmark
) {
1043 return HASHWALK_ABORT
;
1045 return HASHWALK_CONTINUE
;
1048 static void zebra_pbr_show_iptable_unit(struct zebra_pbr_iptable
*iptable
,
1050 struct zebra_ns
*zns
)
1053 uint64_t pkts
= 0, bytes
= 0;
1055 vty_out(vty
, "IPtable %s action %s (%u)\n", iptable
->ipset_name
,
1056 iptable
->action
== ZEBRA_IPTABLES_DROP
? "drop" : "redirect",
1058 if (iptable
->type
== IPSET_NET_PORT
||
1059 iptable
->type
== IPSET_NET_PORT_NET
) {
1060 if (!(iptable
->filter_bm
& MATCH_ICMP_SET
)) {
1061 if (iptable
->filter_bm
& PBR_FILTER_DST_PORT
)
1062 vty_out(vty
, "\t lookup dst port\n");
1063 else if (iptable
->filter_bm
& PBR_FILTER_SRC_PORT
)
1064 vty_out(vty
, "\t lookup src port\n");
1067 if (iptable
->pkt_len_min
|| iptable
->pkt_len_max
) {
1068 if (!iptable
->pkt_len_max
)
1069 vty_out(vty
, "\t pkt len %u\n",
1070 iptable
->pkt_len_min
);
1072 vty_out(vty
, "\t pkt len [%u;%u]\n",
1073 iptable
->pkt_len_min
,
1074 iptable
->pkt_len_max
);
1076 if (iptable
->tcp_flags
|| iptable
->tcp_mask_flags
) {
1077 char tcp_flag_str
[64];
1078 char tcp_flag_mask_str
[64];
1080 zebra_pbr_tcpflags_snprintf(tcp_flag_str
,
1081 sizeof(tcp_flag_str
),
1082 iptable
->tcp_flags
);
1083 zebra_pbr_tcpflags_snprintf(tcp_flag_mask_str
,
1084 sizeof(tcp_flag_mask_str
),
1085 iptable
->tcp_mask_flags
);
1086 vty_out(vty
, "\t tcpflags [%s/%s]\n",
1087 tcp_flag_str
, tcp_flag_mask_str
);
1089 if (iptable
->filter_bm
& (MATCH_DSCP_SET
| MATCH_DSCP_INVERSE_SET
)) {
1090 vty_out(vty
, "\t dscp %s %d\n",
1091 iptable
->filter_bm
& MATCH_DSCP_INVERSE_SET
?
1092 "not" : "", iptable
->dscp_value
);
1094 if (iptable
->fragment
) {
1097 sprintf(val_str
, "%d", iptable
->fragment
);
1098 vty_out(vty
, "\t fragment%s %s\n",
1099 iptable
->filter_bm
& MATCH_FRAGMENT_INVERSE_SET
?
1100 " not" : "", lookup_msg(fragment_value_str
,
1101 iptable
->fragment
, val_str
));
1103 if (iptable
->protocol
) {
1104 vty_out(vty
, "\t protocol %d\n",
1107 ret
= hook_call(zebra_pbr_iptable_get_stat
, iptable
, &pkts
,
1109 if (ret
&& pkts
> 0)
1110 vty_out(vty
, "\t pkts %" PRIu64
", bytes %" PRIu64
"\n",
1112 if (iptable
->action
!= ZEBRA_IPTABLES_DROP
) {
1113 struct pbr_rule_fwmark_lookup prfl
;
1115 prfl
.fwmark
= iptable
->fwmark
;
1117 hash_walk(zrouter
.rules_hash
,
1118 &zebra_pbr_rule_lookup_fwmark_walkcb
, &prfl
);
1120 struct zebra_pbr_rule
*zpr
= prfl
.ptr
;
1122 vty_out(vty
, "\t table %u, fwmark %u\n",
1123 zpr
->rule
.action
.table
,
1129 static int zebra_pbr_show_iptable_walkcb(struct hash_bucket
*bucket
, void *arg
)
1131 struct zebra_pbr_iptable
*iptable
=
1132 (struct zebra_pbr_iptable
*)bucket
->data
;
1133 struct zebra_pbr_env_display
*env
= (struct zebra_pbr_env_display
*)arg
;
1134 struct vty
*vty
= env
->vty
;
1135 struct zebra_ns
*zns
= env
->zns
;
1136 char *iptable_name
= env
->name
;
1139 zebra_pbr_show_iptable_unit(iptable
, vty
, zns
);
1140 else if (!strncmp(iptable_name
,
1141 iptable
->ipset_name
,
1142 ZEBRA_IPSET_NAME_SIZE
))
1143 zebra_pbr_show_iptable_unit(iptable
, vty
, zns
);
1144 return HASHWALK_CONTINUE
;
1147 void zebra_pbr_show_iptable(struct vty
*vty
, char *iptable_name
)
1149 struct zebra_ns
*zns
= zebra_ns_lookup(NS_DEFAULT
);
1150 struct zebra_pbr_env_display env
;
1154 env
.name
= iptable_name
;
1155 hash_walk(zrouter
.iptable_hash
, zebra_pbr_show_iptable_walkcb
, &env
);
1158 void zebra_pbr_iptable_update_interfacelist(struct stream
*s
,
1159 struct zebra_pbr_iptable
*zpi
)
1161 uint32_t i
= 0, index
;
1162 struct interface
*ifp
;
1165 for (i
= 0; i
< zpi
->nb_interface
; i
++) {
1166 STREAM_GETL(s
, index
);
1167 ifp
= if_lookup_by_index(index
, zpi
->vrf_id
);
1170 name
= XSTRDUP(MTYPE_PBR_IPTABLE_IFNAME
, ifp
->name
);
1171 listnode_add(zpi
->interface_name_list
, name
);