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 key
= jhash_1word(rule
->ifp
->ifindex
, key
);
150 key
= jhash_1word(0, key
);
152 if (rule
->rule
.filter
.fwmark
)
153 key
= jhash_1word(rule
->rule
.filter
.fwmark
, key
);
155 key
= jhash_1word(0, key
);
157 key
= jhash_1word(rule
->vrf_id
, key
);
159 return jhash_3words(rule
->rule
.filter
.src_port
,
160 rule
->rule
.filter
.dst_port
,
161 prefix_hash_key(&rule
->rule
.filter
.dst_ip
),
162 jhash_1word(rule
->rule
.unique
, key
));
165 bool zebra_pbr_rules_hash_equal(const void *arg1
, const void *arg2
)
167 const struct zebra_pbr_rule
*r1
, *r2
;
169 r1
= (const struct zebra_pbr_rule
*)arg1
;
170 r2
= (const struct zebra_pbr_rule
*)arg2
;
172 if (r1
->rule
.seq
!= r2
->rule
.seq
)
175 if (r1
->rule
.priority
!= r2
->rule
.priority
)
178 if (r1
->rule
.unique
!= r2
->rule
.unique
)
181 if (r1
->rule
.action
.table
!= r2
->rule
.action
.table
)
184 if (r1
->rule
.filter
.src_port
!= r2
->rule
.filter
.src_port
)
187 if (r1
->rule
.filter
.dst_port
!= r2
->rule
.filter
.dst_port
)
190 if (r1
->rule
.filter
.fwmark
!= r2
->rule
.filter
.fwmark
)
193 if (!prefix_same(&r1
->rule
.filter
.src_ip
, &r2
->rule
.filter
.src_ip
))
196 if (!prefix_same(&r1
->rule
.filter
.dst_ip
, &r2
->rule
.filter
.dst_ip
))
199 if (r1
->ifp
!= r2
->ifp
)
202 if (r1
->vrf_id
!= r2
->vrf_id
)
208 struct pbr_rule_unique_lookup
{
209 struct zebra_pbr_rule
*rule
;
211 struct interface
*ifp
;
215 static int pbr_rule_lookup_unique_walker(struct hash_bucket
*b
, void *data
)
217 struct pbr_rule_unique_lookup
*pul
= data
;
218 struct zebra_pbr_rule
*rule
= b
->data
;
220 if (pul
->unique
== rule
->rule
.unique
221 && pul
->ifp
== rule
->ifp
222 && pul
->vrf_id
== rule
->vrf_id
) {
224 return HASHWALK_ABORT
;
227 return HASHWALK_CONTINUE
;
230 static struct zebra_pbr_rule
*
231 pbr_rule_lookup_unique(struct zebra_pbr_rule
*zrule
)
233 struct pbr_rule_unique_lookup pul
;
235 pul
.unique
= zrule
->rule
.unique
;
236 pul
.ifp
= zrule
->ifp
;
238 pul
.vrf_id
= zrule
->vrf_id
;
239 hash_walk(zrouter
.rules_hash
, &pbr_rule_lookup_unique_walker
, &pul
);
244 void zebra_pbr_ipset_free(void *arg
)
246 struct zebra_pbr_ipset
*ipset
;
248 ipset
= (struct zebra_pbr_ipset
*)arg
;
249 hook_call(zebra_pbr_ipset_update
, 0, ipset
);
250 XFREE(MTYPE_TMP
, ipset
);
253 uint32_t zebra_pbr_ipset_hash_key(const void *arg
)
255 const struct zebra_pbr_ipset
*ipset
= arg
;
256 uint32_t *pnt
= (uint32_t *)&ipset
->ipset_name
;
257 uint32_t key
= jhash_1word(ipset
->vrf_id
, 0x63ab42de);
259 return jhash2(pnt
, ZEBRA_IPSET_NAME_HASH_SIZE
, key
);
262 bool zebra_pbr_ipset_hash_equal(const void *arg1
, const void *arg2
)
264 const struct zebra_pbr_ipset
*r1
, *r2
;
266 r1
= (const struct zebra_pbr_ipset
*)arg1
;
267 r2
= (const struct zebra_pbr_ipset
*)arg2
;
269 if (r1
->type
!= r2
->type
)
271 if (r1
->unique
!= r2
->unique
)
273 if (r1
->vrf_id
!= r2
->vrf_id
)
276 if (strncmp(r1
->ipset_name
, r2
->ipset_name
,
277 ZEBRA_IPSET_NAME_SIZE
))
282 void zebra_pbr_ipset_entry_free(void *arg
)
284 struct zebra_pbr_ipset_entry
*ipset
;
286 ipset
= (struct zebra_pbr_ipset_entry
*)arg
;
288 hook_call(zebra_pbr_ipset_entry_update
, 0, ipset
);
290 XFREE(MTYPE_TMP
, ipset
);
293 uint32_t zebra_pbr_ipset_entry_hash_key(const void *arg
)
295 const struct zebra_pbr_ipset_entry
*ipset
;
299 key
= prefix_hash_key(&ipset
->src
);
300 key
= jhash_1word(ipset
->unique
, key
);
301 key
= jhash_1word(prefix_hash_key(&ipset
->dst
), key
);
302 key
= jhash(&ipset
->dst_port_min
, 2, key
);
303 key
= jhash(&ipset
->dst_port_max
, 2, key
);
304 key
= jhash(&ipset
->src_port_min
, 2, key
);
305 key
= jhash(&ipset
->src_port_max
, 2, key
);
306 key
= jhash(&ipset
->proto
, 1, key
);
311 bool zebra_pbr_ipset_entry_hash_equal(const void *arg1
, const void *arg2
)
313 const struct zebra_pbr_ipset_entry
*r1
, *r2
;
315 r1
= (const struct zebra_pbr_ipset_entry
*)arg1
;
316 r2
= (const struct zebra_pbr_ipset_entry
*)arg2
;
318 if (r1
->unique
!= r2
->unique
)
321 if (!prefix_same(&r1
->src
, &r2
->src
))
324 if (!prefix_same(&r1
->dst
, &r2
->dst
))
327 if (r1
->src_port_min
!= r2
->src_port_min
)
330 if (r1
->src_port_max
!= r2
->src_port_max
)
333 if (r1
->dst_port_min
!= r2
->dst_port_min
)
336 if (r1
->dst_port_max
!= r2
->dst_port_max
)
339 if (r1
->proto
!= r2
->proto
)
344 void zebra_pbr_iptable_free(void *arg
)
346 struct zebra_pbr_iptable
*iptable
;
347 struct listnode
*node
, *nnode
;
350 iptable
= (struct zebra_pbr_iptable
*)arg
;
351 hook_call(zebra_pbr_iptable_update
, 0, iptable
);
353 for (ALL_LIST_ELEMENTS(iptable
->interface_name_list
,
354 node
, nnode
, name
)) {
355 XFREE(MTYPE_PBR_IPTABLE_IFNAME
, name
);
356 list_delete_node(iptable
->interface_name_list
,
359 XFREE(MTYPE_TMP
, iptable
);
362 uint32_t zebra_pbr_iptable_hash_key(const void *arg
)
364 const struct zebra_pbr_iptable
*iptable
= arg
;
365 uint32_t *pnt
= (uint32_t *)&(iptable
->ipset_name
);
368 key
= jhash2(pnt
, ZEBRA_IPSET_NAME_HASH_SIZE
,
370 key
= jhash_1word(iptable
->fwmark
, key
);
371 key
= jhash_1word(iptable
->pkt_len_min
, key
);
372 key
= jhash_1word(iptable
->pkt_len_max
, key
);
373 key
= jhash_1word(iptable
->tcp_flags
, key
);
374 key
= jhash_1word(iptable
->tcp_mask_flags
, key
);
375 key
= jhash_1word(iptable
->dscp_value
, key
);
376 key
= jhash_1word(iptable
->protocol
, key
);
377 key
= jhash_1word(iptable
->fragment
, key
);
378 key
= jhash_1word(iptable
->vrf_id
, key
);
380 return jhash_3words(iptable
->filter_bm
, iptable
->type
,
381 iptable
->unique
, key
);
384 bool zebra_pbr_iptable_hash_equal(const void *arg1
, const void *arg2
)
386 const struct zebra_pbr_iptable
*r1
, *r2
;
388 r1
= (const struct zebra_pbr_iptable
*)arg1
;
389 r2
= (const struct zebra_pbr_iptable
*)arg2
;
391 if (r1
->vrf_id
!= r2
->vrf_id
)
393 if (r1
->type
!= r2
->type
)
395 if (r1
->unique
!= r2
->unique
)
397 if (r1
->filter_bm
!= r2
->filter_bm
)
399 if (r1
->fwmark
!= r2
->fwmark
)
401 if (r1
->action
!= r2
->action
)
403 if (strncmp(r1
->ipset_name
, r2
->ipset_name
,
404 ZEBRA_IPSET_NAME_SIZE
))
406 if (r1
->pkt_len_min
!= r2
->pkt_len_min
)
408 if (r1
->pkt_len_max
!= r2
->pkt_len_max
)
410 if (r1
->tcp_flags
!= r2
->tcp_flags
)
412 if (r1
->tcp_mask_flags
!= r2
->tcp_mask_flags
)
414 if (r1
->dscp_value
!= r2
->dscp_value
)
416 if (r1
->fragment
!= r2
->fragment
)
418 if (r1
->protocol
!= r2
->protocol
)
423 static void *pbr_rule_alloc_intern(void *arg
)
425 struct zebra_pbr_rule
*zpr
;
426 struct zebra_pbr_rule
*new;
428 zpr
= (struct zebra_pbr_rule
*)arg
;
430 new = XCALLOC(MTYPE_TMP
, sizeof(*new));
432 memcpy(new, zpr
, sizeof(*zpr
));
437 void zebra_pbr_add_rule(struct zebra_pbr_rule
*rule
)
439 struct zebra_pbr_rule
*unique
=
440 pbr_rule_lookup_unique(rule
);
442 (void)hash_get(zrouter
.rules_hash
, rule
, pbr_rule_alloc_intern
);
443 (void)kernel_add_pbr_rule(rule
);
445 * Rule Replace semantics, if we have an old, install the
446 * new rule, look above, and then delete the old
449 zebra_pbr_del_rule(unique
);
452 void zebra_pbr_del_rule(struct zebra_pbr_rule
*rule
)
454 struct zebra_pbr_rule
*lookup
;
456 lookup
= hash_lookup(zrouter
.rules_hash
, rule
);
457 (void)kernel_del_pbr_rule(rule
);
460 hash_release(zrouter
.rules_hash
, lookup
);
461 XFREE(MTYPE_TMP
, lookup
);
463 zlog_debug("%s: Rule being deleted we know nothing about",
464 __PRETTY_FUNCTION__
);
467 static void zebra_pbr_cleanup_rules(struct hash_bucket
*b
, void *data
)
469 struct zebra_pbr_rule
*rule
= b
->data
;
472 if (rule
->sock
== *sock
) {
473 (void)kernel_del_pbr_rule(rule
);
474 if (hash_release(zrouter
.rules_hash
, rule
))
475 XFREE(MTYPE_TMP
, rule
);
478 "%s: Rule seq: %u is being cleaned but we can't find it in our tables",
479 __func__
, rule
->rule
.seq
);
483 static void zebra_pbr_cleanup_ipset(struct hash_bucket
*b
, void *data
)
485 struct zebra_pbr_ipset
*ipset
= b
->data
;
488 if (ipset
->sock
== *sock
) {
489 hook_call(zebra_pbr_ipset_update
, 0, ipset
);
490 hash_release(zrouter
.ipset_hash
, ipset
);
494 static void zebra_pbr_cleanup_ipset_entry(struct hash_bucket
*b
, void *data
)
496 struct zebra_pbr_ipset_entry
*ipset
= b
->data
;
499 if (ipset
->sock
== *sock
) {
500 hook_call(zebra_pbr_ipset_entry_update
, 0, ipset
);
501 hash_release(zrouter
.ipset_entry_hash
, ipset
);
505 static void zebra_pbr_cleanup_iptable(struct hash_bucket
*b
, void *data
)
507 struct zebra_pbr_iptable
*iptable
= b
->data
;
510 if (iptable
->sock
== *sock
) {
511 hook_call(zebra_pbr_iptable_update
, 0, iptable
);
512 hash_release(zrouter
.iptable_hash
, iptable
);
516 static int zebra_pbr_client_close_cleanup(struct zserv
*client
)
518 int sock
= client
->sock
;
522 hash_iterate(zrouter
.rules_hash
, zebra_pbr_cleanup_rules
, &sock
);
523 hash_iterate(zrouter
.iptable_hash
, zebra_pbr_cleanup_iptable
, &sock
);
524 hash_iterate(zrouter
.ipset_entry_hash
, zebra_pbr_cleanup_ipset_entry
,
526 hash_iterate(zrouter
.ipset_hash
, zebra_pbr_cleanup_ipset
, &sock
);
530 void zebra_pbr_init(void)
532 hook_register(zserv_client_close
, zebra_pbr_client_close_cleanup
);
535 static void *pbr_ipset_alloc_intern(void *arg
)
537 struct zebra_pbr_ipset
*zpi
;
538 struct zebra_pbr_ipset
*new;
540 zpi
= (struct zebra_pbr_ipset
*)arg
;
542 new = XCALLOC(MTYPE_TMP
, sizeof(struct zebra_pbr_ipset
));
544 memcpy(new, zpi
, sizeof(*zpi
));
549 void zebra_pbr_create_ipset(struct zebra_pbr_ipset
*ipset
)
553 (void)hash_get(zrouter
.ipset_hash
, ipset
, pbr_ipset_alloc_intern
);
554 ret
= hook_call(zebra_pbr_ipset_update
, 1, ipset
);
555 kernel_pbr_ipset_add_del_status(ipset
,
556 ret
? ZEBRA_DPLANE_INSTALL_SUCCESS
557 : ZEBRA_DPLANE_INSTALL_FAILURE
);
560 void zebra_pbr_destroy_ipset(struct zebra_pbr_ipset
*ipset
)
562 struct zebra_pbr_ipset
*lookup
;
564 lookup
= hash_lookup(zrouter
.ipset_hash
, ipset
);
565 hook_call(zebra_pbr_ipset_update
, 0, ipset
);
567 hash_release(zrouter
.ipset_hash
, lookup
);
568 XFREE(MTYPE_TMP
, lookup
);
571 "%s: IPSet Entry being deleted we know nothing about",
572 __PRETTY_FUNCTION__
);
575 struct pbr_ipset_name_lookup
{
576 struct zebra_pbr_ipset
*ipset
;
577 char ipset_name
[ZEBRA_IPSET_NAME_SIZE
];
580 const char *zebra_pbr_ipset_type2str(uint32_t type
)
582 return lookup_msg(ipset_type_msg
, type
,
583 "Unrecognized IPset Type");
586 static int zebra_pbr_ipset_pername_walkcb(struct hash_bucket
*bucket
, void *arg
)
588 struct pbr_ipset_name_lookup
*pinl
=
589 (struct pbr_ipset_name_lookup
*)arg
;
590 struct zebra_pbr_ipset
*zpi
= (struct zebra_pbr_ipset
*)bucket
->data
;
592 if (!strncmp(pinl
->ipset_name
, zpi
->ipset_name
,
593 ZEBRA_IPSET_NAME_SIZE
)) {
595 return HASHWALK_ABORT
;
597 return HASHWALK_CONTINUE
;
600 struct zebra_pbr_ipset
*zebra_pbr_lookup_ipset_pername(char *ipsetname
)
602 struct pbr_ipset_name_lookup pinl
;
603 struct pbr_ipset_name_lookup
*ptr
= &pinl
;
607 memset(ptr
, 0, sizeof(struct pbr_ipset_name_lookup
));
608 snprintf((char *)ptr
->ipset_name
, ZEBRA_IPSET_NAME_SIZE
, "%s",
610 hash_walk(zrouter
.ipset_hash
, zebra_pbr_ipset_pername_walkcb
, ptr
);
614 static void *pbr_ipset_entry_alloc_intern(void *arg
)
616 struct zebra_pbr_ipset_entry
*zpi
;
617 struct zebra_pbr_ipset_entry
*new;
619 zpi
= (struct zebra_pbr_ipset_entry
*)arg
;
621 new = XCALLOC(MTYPE_TMP
, sizeof(struct zebra_pbr_ipset_entry
));
623 memcpy(new, zpi
, sizeof(*zpi
));
628 void zebra_pbr_add_ipset_entry(struct zebra_pbr_ipset_entry
*ipset
)
632 (void)hash_get(zrouter
.ipset_entry_hash
, ipset
,
633 pbr_ipset_entry_alloc_intern
);
634 ret
= hook_call(zebra_pbr_ipset_entry_update
, 1, ipset
);
635 kernel_pbr_ipset_entry_add_del_status(ipset
,
636 ret
? ZEBRA_DPLANE_INSTALL_SUCCESS
637 : ZEBRA_DPLANE_INSTALL_FAILURE
);
640 void zebra_pbr_del_ipset_entry(struct zebra_pbr_ipset_entry
*ipset
)
642 struct zebra_pbr_ipset_entry
*lookup
;
644 lookup
= hash_lookup(zrouter
.ipset_entry_hash
, ipset
);
645 hook_call(zebra_pbr_ipset_entry_update
, 0, ipset
);
647 hash_release(zrouter
.ipset_entry_hash
, lookup
);
648 XFREE(MTYPE_TMP
, lookup
);
650 zlog_debug("%s: IPSet being deleted we know nothing about",
651 __PRETTY_FUNCTION__
);
654 static void *pbr_iptable_alloc_intern(void *arg
)
656 struct zebra_pbr_iptable
*zpi
;
657 struct zebra_pbr_iptable
*new;
659 zpi
= (struct zebra_pbr_iptable
*)arg
;
661 new = XCALLOC(MTYPE_TMP
, sizeof(struct zebra_pbr_iptable
));
663 memcpy(new, zpi
, sizeof(*zpi
));
668 void zebra_pbr_add_iptable(struct zebra_pbr_iptable
*iptable
)
672 (void)hash_get(zrouter
.iptable_hash
, iptable
, pbr_iptable_alloc_intern
);
673 ret
= hook_call(zebra_pbr_iptable_update
, 1, iptable
);
674 kernel_pbr_iptable_add_del_status(iptable
,
675 ret
? ZEBRA_DPLANE_INSTALL_SUCCESS
676 : ZEBRA_DPLANE_INSTALL_FAILURE
);
679 void zebra_pbr_del_iptable(struct zebra_pbr_iptable
*iptable
)
681 struct zebra_pbr_iptable
*lookup
;
683 lookup
= hash_lookup(zrouter
.iptable_hash
, iptable
);
684 hook_call(zebra_pbr_iptable_update
, 0, iptable
);
686 struct listnode
*node
, *nnode
;
689 hash_release(zrouter
.iptable_hash
, lookup
);
690 for (ALL_LIST_ELEMENTS(iptable
->interface_name_list
,
691 node
, nnode
, name
)) {
692 XFREE(MTYPE_PBR_IPTABLE_IFNAME
, name
);
693 list_delete_node(iptable
->interface_name_list
,
696 XFREE(MTYPE_TMP
, lookup
);
698 zlog_debug("%s: IPTable being deleted we know nothing about",
699 __PRETTY_FUNCTION__
);
703 * Handle success or failure of rule (un)install in the kernel.
705 void kernel_pbr_rule_add_del_status(struct zebra_pbr_rule
*rule
,
706 enum zebra_dplane_status res
)
709 case ZEBRA_DPLANE_INSTALL_SUCCESS
:
710 zsend_rule_notify_owner(rule
, ZAPI_RULE_INSTALLED
);
712 case ZEBRA_DPLANE_INSTALL_FAILURE
:
713 zsend_rule_notify_owner(rule
, ZAPI_RULE_FAIL_INSTALL
);
715 case ZEBRA_DPLANE_DELETE_SUCCESS
:
716 zsend_rule_notify_owner(rule
, ZAPI_RULE_REMOVED
);
718 case ZEBRA_DPLANE_DELETE_FAILURE
:
719 zsend_rule_notify_owner(rule
, ZAPI_RULE_FAIL_REMOVE
);
721 case ZEBRA_DPLANE_STATUS_NONE
:
727 * Handle success or failure of ipset (un)install in the kernel.
729 void kernel_pbr_ipset_add_del_status(struct zebra_pbr_ipset
*ipset
,
730 enum zebra_dplane_status res
)
733 case ZEBRA_DPLANE_INSTALL_SUCCESS
:
734 zsend_ipset_notify_owner(ipset
, ZAPI_IPSET_INSTALLED
);
736 case ZEBRA_DPLANE_INSTALL_FAILURE
:
737 zsend_ipset_notify_owner(ipset
, ZAPI_IPSET_FAIL_INSTALL
);
739 case ZEBRA_DPLANE_DELETE_SUCCESS
:
740 zsend_ipset_notify_owner(ipset
, ZAPI_IPSET_REMOVED
);
742 case ZEBRA_DPLANE_DELETE_FAILURE
:
743 zsend_ipset_notify_owner(ipset
, ZAPI_IPSET_FAIL_REMOVE
);
745 case ZEBRA_DPLANE_STATUS_NONE
:
751 * Handle success or failure of ipset (un)install in the kernel.
753 void kernel_pbr_ipset_entry_add_del_status(
754 struct zebra_pbr_ipset_entry
*ipset
,
755 enum zebra_dplane_status res
)
758 case ZEBRA_DPLANE_INSTALL_SUCCESS
:
759 zsend_ipset_entry_notify_owner(ipset
,
760 ZAPI_IPSET_ENTRY_INSTALLED
);
762 case ZEBRA_DPLANE_INSTALL_FAILURE
:
763 zsend_ipset_entry_notify_owner(ipset
,
764 ZAPI_IPSET_ENTRY_FAIL_INSTALL
);
766 case ZEBRA_DPLANE_DELETE_SUCCESS
:
767 zsend_ipset_entry_notify_owner(ipset
,
768 ZAPI_IPSET_ENTRY_REMOVED
);
770 case ZEBRA_DPLANE_DELETE_FAILURE
:
771 zsend_ipset_entry_notify_owner(ipset
,
772 ZAPI_IPSET_ENTRY_FAIL_REMOVE
);
774 case ZEBRA_DPLANE_STATUS_NONE
:
780 * Handle success or failure of ipset (un)install in the kernel.
782 void kernel_pbr_iptable_add_del_status(struct zebra_pbr_iptable
*iptable
,
783 enum zebra_dplane_status res
)
786 case ZEBRA_DPLANE_INSTALL_SUCCESS
:
787 zsend_iptable_notify_owner(iptable
, ZAPI_IPTABLE_INSTALLED
);
789 case ZEBRA_DPLANE_INSTALL_FAILURE
:
790 zsend_iptable_notify_owner(iptable
, ZAPI_IPTABLE_FAIL_INSTALL
);
792 case ZEBRA_DPLANE_DELETE_SUCCESS
:
793 zsend_iptable_notify_owner(iptable
,
794 ZAPI_IPTABLE_REMOVED
);
796 case ZEBRA_DPLANE_DELETE_FAILURE
:
797 zsend_iptable_notify_owner(iptable
,
798 ZAPI_IPTABLE_FAIL_REMOVE
);
800 case ZEBRA_DPLANE_STATUS_NONE
:
806 * Handle rule delete notification from kernel.
808 int kernel_pbr_rule_del(struct zebra_pbr_rule
*rule
)
813 struct zebra_pbr_ipset_entry_unique_display
{
814 struct zebra_pbr_ipset
*zpi
;
816 struct zebra_ns
*zns
;
819 struct zebra_pbr_env_display
{
820 struct zebra_ns
*zns
;
825 static const char *zebra_pbr_prefix2str(union prefixconstptr pu
,
828 const struct prefix
*p
= pu
.p
;
829 char buf
[PREFIX2STR_BUFFER
];
831 if (p
->family
== AF_INET
&& p
->prefixlen
== IPV4_MAX_PREFIXLEN
) {
832 snprintf(str
, size
, "%s", inet_ntop(p
->family
, &p
->u
.prefix
,
833 buf
, PREFIX2STR_BUFFER
));
836 return prefix2str(pu
, str
, size
);
839 static void zebra_pbr_display_icmp(struct vty
*vty
,
840 struct zebra_pbr_ipset_entry
*zpie
)
842 char decoded_str
[20];
845 /* range icmp type */
846 if (zpie
->src_port_max
|| zpie
->dst_port_max
) {
847 vty_out(vty
, ":icmp:[type <%d:%d>;code <%d:%d>",
848 zpie
->src_port_min
, zpie
->src_port_max
,
849 zpie
->dst_port_min
, zpie
->dst_port_max
);
851 port
= ((zpie
->src_port_min
<< 8) & 0xff00) +
852 (zpie
->dst_port_min
& 0xff);
853 memset(decoded_str
, 0, sizeof(decoded_str
));
854 sprintf(decoded_str
, "%d/%d",
857 vty_out(vty
, ":icmp:%s",
858 lookup_msg(icmp_typecode_str
,
863 static void zebra_pbr_display_port(struct vty
*vty
, uint32_t filter_bm
,
864 uint16_t port_min
, uint16_t port_max
,
867 if (!(filter_bm
& PBR_FILTER_PROTO
)) {
869 vty_out(vty
, ":udp/tcp:%d-%d",
872 vty_out(vty
, ":udp/tcp:%d",
876 vty_out(vty
, ":proto %d:%d-%d",
877 proto
, port_min
, port_max
);
879 vty_out(vty
, ":proto %d:%d",
884 static int zebra_pbr_show_ipset_entry_walkcb(struct hash_bucket
*bucket
,
887 struct zebra_pbr_ipset_entry_unique_display
*unique
=
888 (struct zebra_pbr_ipset_entry_unique_display
*)arg
;
889 struct zebra_pbr_ipset
*zpi
= unique
->zpi
;
890 struct vty
*vty
= unique
->vty
;
891 struct zebra_pbr_ipset_entry
*zpie
=
892 (struct zebra_pbr_ipset_entry
*)bucket
->data
;
893 uint64_t pkts
= 0, bytes
= 0;
896 if (zpie
->backpointer
!= zpi
)
897 return HASHWALK_CONTINUE
;
899 if ((zpi
->type
== IPSET_NET_NET
) ||
900 (zpi
->type
== IPSET_NET_PORT_NET
)) {
901 char buf
[PREFIX_STRLEN
];
903 zebra_pbr_prefix2str(&(zpie
->src
), buf
, sizeof(buf
));
904 vty_out(vty
, "\tfrom %s", buf
);
905 if (zpie
->filter_bm
& PBR_FILTER_SRC_PORT
&&
906 zpie
->proto
!= IPPROTO_ICMP
)
907 zebra_pbr_display_port(vty
, zpie
->filter_bm
,
911 vty_out(vty
, " to ");
912 zebra_pbr_prefix2str(&(zpie
->dst
), buf
, sizeof(buf
));
913 vty_out(vty
, "%s", buf
);
914 if (zpie
->filter_bm
& PBR_FILTER_DST_PORT
&&
915 zpie
->proto
!= IPPROTO_ICMP
)
916 zebra_pbr_display_port(vty
, zpie
->filter_bm
,
920 if (zpie
->proto
== IPPROTO_ICMP
)
921 zebra_pbr_display_icmp(vty
, zpie
);
922 } else if ((zpi
->type
== IPSET_NET
) ||
923 (zpi
->type
== IPSET_NET_PORT
)) {
924 char buf
[PREFIX_STRLEN
];
926 if (zpie
->filter_bm
& PBR_FILTER_SRC_IP
) {
927 zebra_pbr_prefix2str(&(zpie
->src
), buf
, sizeof(buf
));
928 vty_out(vty
, "\tfrom %s", buf
);
930 if (zpie
->filter_bm
& PBR_FILTER_SRC_PORT
&&
931 zpie
->proto
!= IPPROTO_ICMP
)
932 zebra_pbr_display_port(vty
, zpie
->filter_bm
,
936 if (zpie
->filter_bm
& PBR_FILTER_DST_IP
) {
937 zebra_pbr_prefix2str(&(zpie
->dst
), buf
, sizeof(buf
));
938 vty_out(vty
, "\tto %s", buf
);
940 if (zpie
->filter_bm
& PBR_FILTER_DST_PORT
&&
941 zpie
->proto
!= IPPROTO_ICMP
)
942 zebra_pbr_display_port(vty
, zpie
->filter_bm
,
946 if (zpie
->proto
== IPPROTO_ICMP
)
947 zebra_pbr_display_icmp(vty
, zpie
);
949 vty_out(vty
, " (%u)\n", zpie
->unique
);
951 ret
= hook_call(zebra_pbr_ipset_entry_get_stat
, zpie
, &pkts
,
954 vty_out(vty
, "\t pkts %" PRIu64
", bytes %" PRIu64
"\n",
956 return HASHWALK_CONTINUE
;
959 static int zebra_pbr_show_ipset_walkcb(struct hash_bucket
*bucket
, void *arg
)
961 struct zebra_pbr_env_display
*uniqueipset
=
962 (struct zebra_pbr_env_display
*)arg
;
963 struct zebra_pbr_ipset
*zpi
= (struct zebra_pbr_ipset
*)bucket
->data
;
964 struct zebra_pbr_ipset_entry_unique_display unique
;
965 struct vty
*vty
= uniqueipset
->vty
;
966 struct zebra_ns
*zns
= uniqueipset
->zns
;
968 vty_out(vty
, "IPset %s type %s\n", zpi
->ipset_name
,
969 zebra_pbr_ipset_type2str(zpi
->type
));
973 hash_walk(zrouter
.ipset_entry_hash
, zebra_pbr_show_ipset_entry_walkcb
,
976 return HASHWALK_CONTINUE
;
979 size_t zebra_pbr_tcpflags_snprintf(char *buffer
, size_t len
,
982 size_t len_written
= 0;
983 static struct message nt
= {0};
984 const struct message
*pnt
;
987 for (pnt
= tcp_value_str
;
988 memcmp(pnt
, &nt
, sizeof(struct message
)); pnt
++)
989 if (pnt
->key
& tcp_val
) {
990 len_written
+= snprintf(buffer
+ len_written
,
1001 void zebra_pbr_show_ipset_list(struct vty
*vty
, char *ipsetname
)
1003 struct zebra_pbr_ipset
*zpi
;
1004 struct zebra_ns
*zns
= zebra_ns_lookup(NS_DEFAULT
);
1005 struct zebra_pbr_ipset_entry_unique_display unique
;
1006 struct zebra_pbr_env_display uniqueipset
;
1009 zpi
= zebra_pbr_lookup_ipset_pername(ipsetname
);
1011 vty_out(vty
, "No IPset %s found\n", ipsetname
);
1014 vty_out(vty
, "IPset %s type %s\n", ipsetname
,
1015 zebra_pbr_ipset_type2str(zpi
->type
));
1020 hash_walk(zrouter
.ipset_entry_hash
,
1021 zebra_pbr_show_ipset_entry_walkcb
, &unique
);
1024 uniqueipset
.zns
= zns
;
1025 uniqueipset
.vty
= vty
;
1026 uniqueipset
.name
= NULL
;
1027 hash_walk(zrouter
.ipset_hash
, zebra_pbr_show_ipset_walkcb
,
1031 struct pbr_rule_fwmark_lookup
{
1032 struct zebra_pbr_rule
*ptr
;
1036 static int zebra_pbr_rule_lookup_fwmark_walkcb(struct hash_bucket
*bucket
,
1039 struct pbr_rule_fwmark_lookup
*iprule
=
1040 (struct pbr_rule_fwmark_lookup
*)arg
;
1041 struct zebra_pbr_rule
*zpr
= (struct zebra_pbr_rule
*)bucket
->data
;
1043 if (iprule
->fwmark
== zpr
->rule
.filter
.fwmark
) {
1045 return HASHWALK_ABORT
;
1047 return HASHWALK_CONTINUE
;
1050 static void zebra_pbr_show_iptable_unit(struct zebra_pbr_iptable
*iptable
,
1052 struct zebra_ns
*zns
)
1055 uint64_t pkts
= 0, bytes
= 0;
1057 vty_out(vty
, "IPtable %s action %s (%u)\n", iptable
->ipset_name
,
1058 iptable
->action
== ZEBRA_IPTABLES_DROP
? "drop" : "redirect",
1060 if (iptable
->type
== IPSET_NET_PORT
||
1061 iptable
->type
== IPSET_NET_PORT_NET
) {
1062 if (!(iptable
->filter_bm
& MATCH_ICMP_SET
)) {
1063 if (iptable
->filter_bm
& PBR_FILTER_DST_PORT
)
1064 vty_out(vty
, "\t lookup dst port\n");
1065 else if (iptable
->filter_bm
& PBR_FILTER_SRC_PORT
)
1066 vty_out(vty
, "\t lookup src port\n");
1069 if (iptable
->pkt_len_min
|| iptable
->pkt_len_max
) {
1070 if (!iptable
->pkt_len_max
)
1071 vty_out(vty
, "\t pkt len %u\n",
1072 iptable
->pkt_len_min
);
1074 vty_out(vty
, "\t pkt len [%u;%u]\n",
1075 iptable
->pkt_len_min
,
1076 iptable
->pkt_len_max
);
1078 if (iptable
->tcp_flags
|| iptable
->tcp_mask_flags
) {
1079 char tcp_flag_str
[64];
1080 char tcp_flag_mask_str
[64];
1082 zebra_pbr_tcpflags_snprintf(tcp_flag_str
,
1083 sizeof(tcp_flag_str
),
1084 iptable
->tcp_flags
);
1085 zebra_pbr_tcpflags_snprintf(tcp_flag_mask_str
,
1086 sizeof(tcp_flag_mask_str
),
1087 iptable
->tcp_mask_flags
);
1088 vty_out(vty
, "\t tcpflags [%s/%s]\n",
1089 tcp_flag_str
, tcp_flag_mask_str
);
1091 if (iptable
->filter_bm
& (MATCH_DSCP_SET
| MATCH_DSCP_INVERSE_SET
)) {
1092 vty_out(vty
, "\t dscp %s %d\n",
1093 iptable
->filter_bm
& MATCH_DSCP_INVERSE_SET
?
1094 "not" : "", iptable
->dscp_value
);
1096 if (iptable
->fragment
) {
1099 sprintf(val_str
, "%d", iptable
->fragment
);
1100 vty_out(vty
, "\t fragment%s %s\n",
1101 iptable
->filter_bm
& MATCH_FRAGMENT_INVERSE_SET
?
1102 " not" : "", lookup_msg(fragment_value_str
,
1103 iptable
->fragment
, val_str
));
1105 if (iptable
->protocol
) {
1106 vty_out(vty
, "\t protocol %d\n",
1109 ret
= hook_call(zebra_pbr_iptable_get_stat
, iptable
, &pkts
,
1111 if (ret
&& pkts
> 0)
1112 vty_out(vty
, "\t pkts %" PRIu64
", bytes %" PRIu64
"\n",
1114 if (iptable
->action
!= ZEBRA_IPTABLES_DROP
) {
1115 struct pbr_rule_fwmark_lookup prfl
;
1117 prfl
.fwmark
= iptable
->fwmark
;
1119 hash_walk(zrouter
.rules_hash
,
1120 &zebra_pbr_rule_lookup_fwmark_walkcb
, &prfl
);
1122 struct zebra_pbr_rule
*zpr
= prfl
.ptr
;
1124 vty_out(vty
, "\t table %u, fwmark %u\n",
1125 zpr
->rule
.action
.table
,
1131 static int zebra_pbr_show_iptable_walkcb(struct hash_bucket
*bucket
, void *arg
)
1133 struct zebra_pbr_iptable
*iptable
=
1134 (struct zebra_pbr_iptable
*)bucket
->data
;
1135 struct zebra_pbr_env_display
*env
= (struct zebra_pbr_env_display
*)arg
;
1136 struct vty
*vty
= env
->vty
;
1137 struct zebra_ns
*zns
= env
->zns
;
1138 char *iptable_name
= env
->name
;
1141 zebra_pbr_show_iptable_unit(iptable
, vty
, zns
);
1142 else if (!strncmp(iptable_name
,
1143 iptable
->ipset_name
,
1144 ZEBRA_IPSET_NAME_SIZE
))
1145 zebra_pbr_show_iptable_unit(iptable
, vty
, zns
);
1146 return HASHWALK_CONTINUE
;
1149 void zebra_pbr_show_iptable(struct vty
*vty
, char *iptable_name
)
1151 struct zebra_ns
*zns
= zebra_ns_lookup(NS_DEFAULT
);
1152 struct zebra_pbr_env_display env
;
1156 env
.name
= iptable_name
;
1157 hash_walk(zrouter
.iptable_hash
, zebra_pbr_show_iptable_walkcb
, &env
);
1160 void zebra_pbr_iptable_update_interfacelist(struct stream
*s
,
1161 struct zebra_pbr_iptable
*zpi
)
1163 uint32_t i
= 0, index
;
1164 struct interface
*ifp
;
1167 for (i
= 0; i
< zpi
->nb_interface
; i
++) {
1168 STREAM_GETL(s
, index
);
1169 ifp
= if_lookup_by_index(index
, zpi
->vrf_id
);
1172 name
= XSTRDUP(MTYPE_PBR_IPTABLE_IFNAME
, ifp
->name
);
1173 listnode_add(zpi
->interface_name_list
, name
);