1
2
3
4
5
6
7
8
9
10
11
12
13#include <stdio.h>
14#include <stdlib.h>
15#include <unistd.h>
16#include <fcntl.h>
17#include <sys/socket.h>
18#include <netinet/in.h>
19#include <netinet/ip.h>
20#include <arpa/inet.h>
21#include <string.h>
22#include <linux/if.h>
23#include <linux/fib_rules.h>
24#include <errno.h>
25
26#include "rt_names.h"
27#include "utils.h"
28#include "ip_common.h"
29#include "json_print.h"
30
31enum list_action {
32 IPRULE_LIST,
33 IPRULE_FLUSH,
34 IPRULE_SAVE,
35};
36
37extern struct rtnl_handle rth;
38
39static void usage(void) __attribute__((noreturn));
40
41static void usage(void)
42{
43 fprintf(stderr,
44 "Usage: ip rule { add | del } SELECTOR ACTION\n"
45 " ip rule { flush | save | restore }\n"
46 " ip rule [ list [ SELECTOR ]]\n"
47 "SELECTOR := [ not ] [ from PREFIX ] [ to PREFIX ] [ tos TOS ] [ fwmark FWMARK[/MASK] ]\n"
48 " [ iif STRING ] [ oif STRING ] [ pref NUMBER ] [ l3mdev ]\n"
49 " [ uidrange NUMBER-NUMBER ]\n"
50 " [ ipproto PROTOCOL ]\n"
51 " [ sport [ NUMBER | NUMBER-NUMBER ]\n"
52 " [ dport [ NUMBER | NUMBER-NUMBER ] ]\n"
53 "ACTION := [ table TABLE_ID ]\n"
54 " [ protocol PROTO ]\n"
55 " [ nat ADDRESS ]\n"
56 " [ realms [SRCREALM/]DSTREALM ]\n"
57 " [ goto NUMBER ]\n"
58 " SUPPRESSOR\n"
59 "SUPPRESSOR := [ suppress_prefixlength NUMBER ]\n"
60 " [ suppress_ifgroup DEVGROUP ]\n"
61 "TABLE_ID := [ local | main | default | NUMBER ]\n");
62 exit(-1);
63}
64
65static struct
66{
67 int not;
68 int l3mdev;
69 int iifmask, oifmask, uidrange;
70 unsigned int tb;
71 unsigned int tos, tosmask;
72 unsigned int pref, prefmask;
73 unsigned int fwmark, fwmask;
74 uint64_t tun_id;
75 char iif[IFNAMSIZ];
76 char oif[IFNAMSIZ];
77 struct fib_rule_uid_range range;
78 inet_prefix src;
79 inet_prefix dst;
80 int protocol;
81 int protocolmask;
82 struct fib_rule_port_range sport;
83 struct fib_rule_port_range dport;
84 __u8 ipproto;
85} filter;
86
87static inline int frh_get_table(struct fib_rule_hdr *frh, struct rtattr **tb)
88{
89 __u32 table = frh->table;
90 if (tb[RTA_TABLE])
91 table = rta_getattr_u32(tb[RTA_TABLE]);
92 return table;
93}
94
95static bool filter_nlmsg(struct nlmsghdr *n, struct rtattr **tb, int host_len)
96{
97 struct fib_rule_hdr *frh = NLMSG_DATA(n);
98 __u32 table;
99
100 if (preferred_family != AF_UNSPEC && frh->family != preferred_family)
101 return false;
102
103 if (filter.prefmask &&
104 filter.pref ^ (tb[FRA_PRIORITY] ? rta_getattr_u32(tb[FRA_PRIORITY]) : 0))
105 return false;
106 if (filter.not && !(frh->flags & FIB_RULE_INVERT))
107 return false;
108
109 if (filter.src.family) {
110 inet_prefix *f_src = &filter.src;
111
112 if (f_src->family != frh->family ||
113 f_src->bitlen > frh->src_len)
114 return false;
115
116 if (inet_addr_match_rta(f_src, tb[FRA_SRC]))
117 return false;
118 }
119
120 if (filter.dst.family) {
121 inet_prefix *f_dst = &filter.dst;
122
123 if (f_dst->family != frh->family ||
124 f_dst->bitlen > frh->dst_len)
125 return false;
126
127 if (inet_addr_match_rta(f_dst, tb[FRA_DST]))
128 return false;
129 }
130
131 if (filter.tosmask && filter.tos ^ frh->tos)
132 return false;
133
134 if (filter.fwmark) {
135 __u32 mark = 0;
136
137 if (tb[FRA_FWMARK])
138 mark = rta_getattr_u32(tb[FRA_FWMARK]);
139 if (filter.fwmark ^ mark)
140 return false;
141 }
142 if (filter.fwmask) {
143 __u32 mask = 0;
144
145 if (tb[FRA_FWMASK])
146 mask = rta_getattr_u32(tb[FRA_FWMASK]);
147 if (filter.fwmask ^ mask)
148 return false;
149 }
150
151 if (filter.iifmask) {
152 if (tb[FRA_IFNAME]) {
153 if (strcmp(filter.iif, rta_getattr_str(tb[FRA_IFNAME])) != 0)
154 return false;
155 } else {
156 return false;
157 }
158 }
159
160 if (filter.oifmask) {
161 if (tb[FRA_OIFNAME]) {
162 if (strcmp(filter.oif, rta_getattr_str(tb[FRA_OIFNAME])) != 0)
163 return false;
164 } else {
165 return false;
166 }
167 }
168
169 if (filter.l3mdev && !(tb[FRA_L3MDEV] && rta_getattr_u8(tb[FRA_L3MDEV])))
170 return false;
171
172 if (filter.uidrange) {
173 struct fib_rule_uid_range *r = RTA_DATA(tb[FRA_UID_RANGE]);
174
175 if (!tb[FRA_UID_RANGE] ||
176 r->start != filter.range.start ||
177 r->end != filter.range.end)
178 return false;
179 }
180
181 if (filter.ipproto) {
182 __u8 ipproto = 0;
183
184 if (tb[FRA_IP_PROTO])
185 ipproto = rta_getattr_u8(tb[FRA_IP_PROTO]);
186 if (filter.ipproto != ipproto)
187 return false;
188 }
189
190 if (filter.sport.start) {
191 const struct fib_rule_port_range *r;
192
193 if (!tb[FRA_SPORT_RANGE])
194 return false;
195
196 r = RTA_DATA(tb[FRA_SPORT_RANGE]);
197 if (r->start != filter.sport.start ||
198 r->end != filter.sport.end)
199 return false;
200 }
201
202 if (filter.dport.start) {
203 const struct fib_rule_port_range *r;
204
205 if (!tb[FRA_DPORT_RANGE])
206 return false;
207
208 r = RTA_DATA(tb[FRA_DPORT_RANGE]);
209 if (r->start != filter.dport.start ||
210 r->end != filter.dport.end)
211 return false;
212 }
213
214 if (filter.tun_id) {
215 __u64 tun_id = 0;
216
217 if (tb[FRA_TUN_ID]) {
218 tun_id = ntohll(rta_getattr_u64(tb[FRA_TUN_ID]));
219 if (filter.tun_id != tun_id)
220 return false;
221 } else {
222 return false;
223 }
224 }
225
226 table = frh_get_table(frh, tb);
227 if (filter.tb > 0 && filter.tb ^ table)
228 return false;
229
230 return true;
231}
232
233int print_rule(struct nlmsghdr *n, void *arg)
234{
235 FILE *fp = arg;
236 struct fib_rule_hdr *frh = NLMSG_DATA(n);
237 int len = n->nlmsg_len;
238 int host_len = -1;
239 __u32 table, prio = 0;
240 struct rtattr *tb[FRA_MAX+1];
241 SPRINT_BUF(b1);
242
243 if (n->nlmsg_type != RTM_NEWRULE && n->nlmsg_type != RTM_DELRULE)
244 return 0;
245
246 len -= NLMSG_LENGTH(sizeof(*frh));
247 if (len < 0)
248 return -1;
249
250 parse_rtattr(tb, FRA_MAX, RTM_RTA(frh), len);
251
252 host_len = af_bit_len(frh->family);
253
254 if (!filter_nlmsg(n, tb, host_len))
255 return 0;
256
257 open_json_object(NULL);
258 if (n->nlmsg_type == RTM_DELRULE)
259 print_bool(PRINT_ANY, "deleted", "Deleted ", true);
260
261 if (tb[FRA_PRIORITY])
262 prio = rta_getattr_u32(tb[FRA_PRIORITY]);
263
264 print_uint(PRINT_ANY, "priority", "%u:\t", prio);
265
266 if (frh->flags & FIB_RULE_INVERT)
267 print_null(PRINT_ANY, "not", "not ", NULL);
268
269 if (tb[FRA_SRC]) {
270 const char *src = rt_addr_n2a_rta(frh->family, tb[FRA_SRC]);
271
272 print_string(PRINT_FP, NULL, "from ", NULL);
273 print_color_string(PRINT_ANY, ifa_family_color(frh->family),
274 "src", "%s", src);
275 if (frh->src_len != host_len)
276 print_uint(PRINT_ANY, "srclen", "/%u", frh->src_len);
277 } else if (frh->src_len) {
278 print_string(PRINT_ANY, "src", "from %s", "0");
279 print_uint(PRINT_ANY, "srclen", "/%u", frh->src_len);
280 } else {
281 print_string(PRINT_ANY, "src", "from %s", "all");
282 }
283
284 if (tb[FRA_DST]) {
285 const char *dst = rt_addr_n2a_rta(frh->family, tb[FRA_DST]);
286
287 print_string(PRINT_FP, NULL, " to ", NULL);
288 print_color_string(PRINT_ANY, ifa_family_color(frh->family),
289 "dst", "%s", dst);
290 if (frh->dst_len != host_len)
291 print_uint(PRINT_ANY, "dstlen", "/%u", frh->dst_len);
292 } else if (frh->dst_len) {
293 print_string(PRINT_ANY, "dst", " to %s", "0");
294 print_uint(PRINT_ANY, "dstlen", "/%u", frh->dst_len);
295 }
296
297 if (frh->tos) {
298 print_string(PRINT_ANY, "tos",
299 " tos %s",
300 rtnl_dsfield_n2a(frh->tos, b1, sizeof(b1)));
301 }
302
303 if (tb[FRA_FWMARK] || tb[FRA_FWMASK]) {
304 __u32 mark = 0, mask = 0;
305
306 if (tb[FRA_FWMARK])
307 mark = rta_getattr_u32(tb[FRA_FWMARK]);
308
309 if (tb[FRA_FWMASK] &&
310 (mask = rta_getattr_u32(tb[FRA_FWMASK])) != 0xFFFFFFFF) {
311 print_0xhex(PRINT_ANY, "fwmark", " fwmark %#llx", mark);
312 print_0xhex(PRINT_ANY, "fwmask", "/%#llx", mask);
313 } else {
314 print_0xhex(PRINT_ANY, "fwmark", " fwmark %#llx", mark);
315 }
316 }
317
318 if (tb[FRA_IFNAME]) {
319 if (!is_json_context())
320 fprintf(fp, " iif ");
321 print_color_string(PRINT_ANY, COLOR_IFNAME,
322 "iif", "%s",
323 rta_getattr_str(tb[FRA_IFNAME]));
324
325 if (frh->flags & FIB_RULE_IIF_DETACHED)
326 print_null(PRINT_ANY, "iif_detached", " [detached]",
327 NULL);
328 }
329
330 if (tb[FRA_OIFNAME]) {
331 if (!is_json_context())
332 fprintf(fp, " oif ");
333
334 print_color_string(PRINT_ANY, COLOR_IFNAME, "oif", "%s",
335 rta_getattr_str(tb[FRA_OIFNAME]));
336
337 if (frh->flags & FIB_RULE_OIF_DETACHED)
338 print_null(PRINT_ANY, "oif_detached", " [detached]",
339 NULL);
340 }
341
342 if (tb[FRA_L3MDEV]) {
343 __u8 mdev = rta_getattr_u8(tb[FRA_L3MDEV]);
344
345 if (mdev)
346 print_null(PRINT_ANY, "l3mdev",
347 " lookup [l3mdev-table]", NULL);
348 }
349
350 if (tb[FRA_UID_RANGE]) {
351 struct fib_rule_uid_range *r = RTA_DATA(tb[FRA_UID_RANGE]);
352
353 print_uint(PRINT_ANY, "uid_start", " uidrange %u", r->start);
354 print_uint(PRINT_ANY, "uid_end", "-%u", r->end);
355 }
356
357 if (tb[FRA_IP_PROTO]) {
358 SPRINT_BUF(pbuf);
359 print_string(PRINT_ANY, "ipproto", " ipproto %s",
360 inet_proto_n2a(rta_getattr_u8(tb[FRA_IP_PROTO]),
361 pbuf, sizeof(pbuf)));
362 }
363
364 if (tb[FRA_SPORT_RANGE]) {
365 struct fib_rule_port_range *r = RTA_DATA(tb[FRA_SPORT_RANGE]);
366
367 if (r->start == r->end) {
368 print_uint(PRINT_ANY, "sport", " sport %u", r->start);
369 } else {
370 print_uint(PRINT_ANY, "sport_start", " sport %u",
371 r->start);
372 print_uint(PRINT_ANY, "sport_end", "-%u", r->end);
373 }
374 }
375
376 if (tb[FRA_DPORT_RANGE]) {
377 struct fib_rule_port_range *r = RTA_DATA(tb[FRA_DPORT_RANGE]);
378
379 if (r->start == r->end) {
380 print_uint(PRINT_ANY, "dport", " dport %u", r->start);
381 } else {
382 print_uint(PRINT_ANY, "dport_start", " dport %u",
383 r->start);
384 print_uint(PRINT_ANY, "dport_end", "-%u", r->end);
385 }
386 }
387
388 if (tb[FRA_TUN_ID]) {
389 __u64 tun_id = ntohll(rta_getattr_u64(tb[FRA_TUN_ID]));
390
391 print_u64(PRINT_ANY, "tun_id", " tun_id %llu", tun_id);
392 }
393
394 table = frh_get_table(frh, tb);
395 if (table) {
396 print_string(PRINT_ANY, "table",
397 " lookup %s",
398 rtnl_rttable_n2a(table, b1, sizeof(b1)));
399
400 if (tb[FRA_SUPPRESS_PREFIXLEN]) {
401 int pl = rta_getattr_u32(tb[FRA_SUPPRESS_PREFIXLEN]);
402
403 if (pl != -1)
404 print_int(PRINT_ANY, "suppress_prefixlen",
405 " suppress_prefixlength %d", pl);
406 }
407
408 if (tb[FRA_SUPPRESS_IFGROUP]) {
409 int group = rta_getattr_u32(tb[FRA_SUPPRESS_IFGROUP]);
410
411 if (group != -1) {
412 const char *grname
413 = rtnl_group_n2a(group, b1, sizeof(b1));
414
415 print_string(PRINT_ANY, "suppress_ifgroup",
416 " suppress_ifgroup %s", grname);
417 }
418 }
419 }
420
421 if (tb[FRA_FLOW]) {
422 __u32 to = rta_getattr_u32(tb[FRA_FLOW]);
423 __u32 from = to>>16;
424
425 to &= 0xFFFF;
426 if (from)
427 print_string(PRINT_ANY,
428 "flow_from", " realms %s/",
429 rtnl_rtrealm_n2a(from, b1, sizeof(b1)));
430 else
431 print_string(PRINT_FP, NULL, " realms ", NULL);
432
433 print_string(PRINT_ANY, "flow_to", "%s",
434 rtnl_rtrealm_n2a(to, b1, sizeof(b1)));
435 }
436
437 if (frh->action == RTN_NAT) {
438 if (tb[RTA_GATEWAY]) {
439 const char *gateway;
440
441 gateway = format_host_rta(frh->family, tb[RTA_GATEWAY]);
442
443 print_string(PRINT_ANY, "nat_gateway",
444 " map-to %s", gateway);
445 } else {
446 print_null(PRINT_ANY, "masquerade", " masquerade", NULL);
447 }
448 } else if (frh->action == FR_ACT_GOTO) {
449 if (tb[FRA_GOTO])
450 print_uint(PRINT_ANY, "goto", " goto %u",
451 rta_getattr_u32(tb[FRA_GOTO]));
452 else
453 print_string(PRINT_ANY, "goto", " goto %s", "none");
454
455 if (frh->flags & FIB_RULE_UNRESOLVED)
456 print_null(PRINT_ANY, "unresolved",
457 " [unresolved]", NULL);
458 } else if (frh->action == FR_ACT_NOP) {
459 print_null(PRINT_ANY, "nop", " nop", NULL);
460 } else if (frh->action != FR_ACT_TO_TBL) {
461 print_string(PRINT_ANY, "action", " %s",
462 rtnl_rtntype_n2a(frh->action, b1, sizeof(b1)));
463 }
464
465 if (tb[FRA_PROTOCOL]) {
466 __u8 protocol = rta_getattr_u8(tb[FRA_PROTOCOL]);
467
468 if ((protocol && protocol != RTPROT_KERNEL) || show_details > 0) {
469 print_string(PRINT_ANY, "protocol", " proto %s",
470 rtnl_rtprot_n2a(protocol, b1, sizeof(b1)));
471 }
472 }
473 print_string(PRINT_FP, NULL, "\n", "");
474 close_json_object();
475 fflush(fp);
476 return 0;
477}
478
479static __u32 rule_dump_magic = 0x71706986;
480
481static int save_rule_prep(void)
482{
483 int ret;
484
485 if (isatty(STDOUT_FILENO)) {
486 fprintf(stderr, "Not sending a binary stream to stdout\n");
487 return -1;
488 }
489
490 ret = write(STDOUT_FILENO, &rule_dump_magic, sizeof(rule_dump_magic));
491 if (ret != sizeof(rule_dump_magic)) {
492 fprintf(stderr, "Can't write magic to dump file\n");
493 return -1;
494 }
495
496 return 0;
497}
498
499static int save_rule(struct nlmsghdr *n, void *arg)
500{
501 int ret;
502
503 ret = write(STDOUT_FILENO, n, n->nlmsg_len);
504 if ((ret > 0) && (ret != n->nlmsg_len)) {
505 fprintf(stderr, "Short write while saving nlmsg\n");
506 ret = -EIO;
507 }
508
509 return ret == n->nlmsg_len ? 0 : ret;
510}
511
512static int flush_rule(struct nlmsghdr *n, void *arg)
513{
514 struct rtnl_handle rth2;
515 struct fib_rule_hdr *frh = NLMSG_DATA(n);
516 int len = n->nlmsg_len;
517 struct rtattr *tb[FRA_MAX+1];
518 int host_len = -1;
519
520 len -= NLMSG_LENGTH(sizeof(*frh));
521 if (len < 0)
522 return -1;
523
524 parse_rtattr(tb, FRA_MAX, RTM_RTA(frh), len);
525
526 host_len = af_bit_len(frh->family);
527 if (!filter_nlmsg(n, tb, host_len))
528 return 0;
529
530 if (tb[FRA_PROTOCOL]) {
531 __u8 protocol = rta_getattr_u8(tb[FRA_PROTOCOL]);
532
533 if ((filter.protocol ^ protocol) & filter.protocolmask)
534 return 0;
535 }
536
537 if (tb[FRA_PRIORITY]) {
538 n->nlmsg_type = RTM_DELRULE;
539 n->nlmsg_flags = NLM_F_REQUEST;
540
541 if (rtnl_open(&rth2, 0) < 0)
542 return -1;
543
544 if (rtnl_talk(&rth2, n, NULL) < 0)
545 return -2;
546
547 rtnl_close(&rth2);
548 }
549
550 return 0;
551}
552
553static int iprule_list_flush_or_save(int argc, char **argv, int action)
554{
555 rtnl_filter_t filter_fn;
556 int af = preferred_family;
557
558 if (af == AF_UNSPEC)
559 af = AF_INET;
560
561 if (action == IPRULE_SAVE && argc > 0) {
562 fprintf(stderr, "\"ip rule save\" does not take any arguments.\n");
563 return -1;
564 }
565
566 switch (action) {
567 case IPRULE_SAVE:
568 if (save_rule_prep())
569 return -1;
570 filter_fn = save_rule;
571 break;
572 case IPRULE_FLUSH:
573 filter_fn = flush_rule;
574 break;
575 default:
576 filter_fn = print_rule;
577 }
578
579 memset(&filter, 0, sizeof(filter));
580
581 while (argc > 0) {
582 if (matches(*argv, "preference") == 0 ||
583 matches(*argv, "order") == 0 ||
584 matches(*argv, "priority") == 0) {
585 __u32 pref;
586
587 NEXT_ARG();
588 if (get_u32(&pref, *argv, 0))
589 invarg("preference value is invalid\n", *argv);
590 filter.pref = pref;
591 filter.prefmask = 1;
592 } else if (strcmp(*argv, "not") == 0) {
593 filter.not = 1;
594 } else if (strcmp(*argv, "tos") == 0) {
595 __u32 tos;
596
597 NEXT_ARG();
598 if (rtnl_dsfield_a2n(&tos, *argv))
599 invarg("TOS value is invalid\n", *argv);
600 filter.tos = tos;
601 filter.tosmask = 1;
602 } else if (strcmp(*argv, "fwmark") == 0) {
603 char *slash;
604 __u32 fwmark, fwmask;
605
606 NEXT_ARG();
607 slash = strchr(*argv, '/');
608 if (slash != NULL)
609 *slash = '\0';
610 if (get_u32(&fwmark, *argv, 0))
611 invarg("fwmark value is invalid\n", *argv);
612 filter.fwmark = fwmark;
613 if (slash) {
614 if (get_u32(&fwmask, slash+1, 0))
615 invarg("fwmask value is invalid\n",
616 slash+1);
617 filter.fwmask = fwmask;
618 }
619 } else if (strcmp(*argv, "dev") == 0 ||
620 strcmp(*argv, "iif") == 0) {
621 NEXT_ARG();
622 if (get_ifname(filter.iif, *argv))
623 invarg("\"iif\"/\"dev\" not a valid ifname", *argv);
624 filter.iifmask = 1;
625 } else if (strcmp(*argv, "oif") == 0) {
626 NEXT_ARG();
627 if (get_ifname(filter.oif, *argv))
628 invarg("\"oif\" not a valid ifname", *argv);
629 filter.oifmask = 1;
630 } else if (strcmp(*argv, "l3mdev") == 0) {
631 filter.l3mdev = 1;
632 } else if (strcmp(*argv, "uidrange") == 0) {
633 NEXT_ARG();
634 filter.uidrange = 1;
635 if (sscanf(*argv, "%u-%u",
636 &filter.range.start,
637 &filter.range.end) != 2)
638 invarg("invalid UID range\n", *argv);
639
640 } else if (matches(*argv, "tun_id") == 0) {
641 __u64 tun_id;
642
643 NEXT_ARG();
644 if (get_u64(&tun_id, *argv, 0))
645 invarg("\"tun_id\" value is invalid\n", *argv);
646 filter.tun_id = tun_id;
647 } else if (matches(*argv, "lookup") == 0 ||
648 matches(*argv, "table") == 0) {
649 __u32 tid;
650
651 NEXT_ARG();
652 if (rtnl_rttable_a2n(&tid, *argv))
653 invarg("table id value is invalid\n", *argv);
654 filter.tb = tid;
655 } else if (matches(*argv, "from") == 0 ||
656 matches(*argv, "src") == 0) {
657 NEXT_ARG();
658 if (get_prefix(&filter.src, *argv, af))
659 invarg("from value is invalid\n", *argv);
660 } else if (matches(*argv, "protocol") == 0) {
661 __u32 prot;
662 NEXT_ARG();
663 filter.protocolmask = -1;
664 if (rtnl_rtprot_a2n(&prot, *argv)) {
665 if (strcmp(*argv, "all") != 0)
666 invarg("invalid \"protocol\"\n", *argv);
667 prot = 0;
668 filter.protocolmask = 0;
669 }
670 filter.protocol = prot;
671 } else if (strcmp(*argv, "ipproto") == 0) {
672 int ipproto;
673
674 NEXT_ARG();
675 ipproto = inet_proto_a2n(*argv);
676 if (ipproto < 0)
677 invarg("Invalid \"ipproto\" value\n", *argv);
678 filter.ipproto = ipproto;
679 } else if (strcmp(*argv, "sport") == 0) {
680 struct fib_rule_port_range r;
681 int ret;
682
683 NEXT_ARG();
684 ret = sscanf(*argv, "%hu-%hu", &r.start, &r.end);
685 if (ret == 1)
686 r.end = r.start;
687 else if (ret != 2)
688 invarg("invalid port range\n", *argv);
689 filter.sport = r;
690 } else if (strcmp(*argv, "dport") == 0) {
691 struct fib_rule_port_range r;
692 int ret;
693
694 NEXT_ARG();
695 ret = sscanf(*argv, "%hu-%hu", &r.start, &r.end);
696 if (ret == 1)
697 r.end = r.start;
698 else if (ret != 2)
699 invarg("invalid dport range\n", *argv);
700 filter.dport = r;
701 } else{
702 if (matches(*argv, "dst") == 0 ||
703 matches(*argv, "to") == 0) {
704 NEXT_ARG();
705 }
706 if (get_prefix(&filter.dst, *argv, af))
707 invarg("to value is invalid\n", *argv);
708 }
709 argc--; argv++;
710 }
711
712 if (rtnl_ruledump_req(&rth, af) < 0) {
713 perror("Cannot send dump request");
714 return 1;
715 }
716
717 new_json_obj(json);
718 if (rtnl_dump_filter(&rth, filter_fn, stdout) < 0) {
719 fprintf(stderr, "Dump terminated\n");
720 return 1;
721 }
722 delete_json_obj();
723
724 return 0;
725}
726
727static int rule_dump_check_magic(void)
728{
729 int ret;
730 __u32 magic = 0;
731
732 if (isatty(STDIN_FILENO)) {
733 fprintf(stderr, "Can't restore rule dump from a terminal\n");
734 return -1;
735 }
736
737 ret = fread(&magic, sizeof(magic), 1, stdin);
738 if (magic != rule_dump_magic) {
739 fprintf(stderr, "Magic mismatch (%d elems, %x magic)\n",
740 ret, magic);
741 return -1;
742 }
743
744 return 0;
745}
746
747static int restore_handler(struct rtnl_ctrl_data *ctrl,
748 struct nlmsghdr *n, void *arg)
749{
750 int ret;
751
752 n->nlmsg_flags |= NLM_F_REQUEST | NLM_F_CREATE | NLM_F_ACK;
753
754 ll_init_map(&rth);
755
756 ret = rtnl_talk(&rth, n, NULL);
757 if ((ret < 0) && (errno == EEXIST))
758 ret = 0;
759
760 return ret;
761}
762
763
764static int iprule_restore(void)
765{
766 if (rule_dump_check_magic())
767 exit(-1);
768
769 exit(rtnl_from_file(stdin, &restore_handler, NULL));
770}
771
772static int iprule_modify(int cmd, int argc, char **argv)
773{
774 int l3mdev_rule = 0;
775 int table_ok = 0;
776 __u32 tid = 0;
777 struct {
778 struct nlmsghdr n;
779 struct fib_rule_hdr frh;
780 char buf[1024];
781 } req = {
782 .n.nlmsg_type = cmd,
783 .n.nlmsg_len = NLMSG_LENGTH(sizeof(struct fib_rule_hdr)),
784 .n.nlmsg_flags = NLM_F_REQUEST,
785 .frh.family = preferred_family,
786 .frh.action = FR_ACT_UNSPEC,
787 };
788
789 if (cmd == RTM_NEWRULE) {
790 if (argc == 0) {
791 fprintf(stderr,
792 "\"ip rule add\" requires arguments.\n");
793 return -1;
794 }
795 req.n.nlmsg_flags |= NLM_F_CREATE|NLM_F_EXCL;
796 req.frh.action = FR_ACT_TO_TBL;
797 }
798
799 if (cmd == RTM_DELRULE && argc == 0) {
800 fprintf(stderr, "\"ip rule del\" requires arguments.\n");
801 return -1;
802 }
803
804 while (argc > 0) {
805 if (strcmp(*argv, "not") == 0) {
806 req.frh.flags |= FIB_RULE_INVERT;
807 } else if (strcmp(*argv, "from") == 0) {
808 inet_prefix dst;
809
810 NEXT_ARG();
811 get_prefix(&dst, *argv, req.frh.family);
812 req.frh.src_len = dst.bitlen;
813 addattr_l(&req.n, sizeof(req), FRA_SRC,
814 &dst.data, dst.bytelen);
815 } else if (strcmp(*argv, "to") == 0) {
816 inet_prefix dst;
817
818 NEXT_ARG();
819 get_prefix(&dst, *argv, req.frh.family);
820 req.frh.dst_len = dst.bitlen;
821 addattr_l(&req.n, sizeof(req), FRA_DST,
822 &dst.data, dst.bytelen);
823 } else if (matches(*argv, "preference") == 0 ||
824 matches(*argv, "order") == 0 ||
825 matches(*argv, "priority") == 0) {
826 __u32 pref;
827
828 NEXT_ARG();
829 if (get_u32(&pref, *argv, 0))
830 invarg("preference value is invalid\n", *argv);
831 addattr32(&req.n, sizeof(req), FRA_PRIORITY, pref);
832 } else if (strcmp(*argv, "tos") == 0 ||
833 matches(*argv, "dsfield") == 0) {
834 __u32 tos;
835
836 NEXT_ARG();
837 if (rtnl_dsfield_a2n(&tos, *argv))
838 invarg("TOS value is invalid\n", *argv);
839 req.frh.tos = tos;
840 } else if (strcmp(*argv, "fwmark") == 0) {
841 char *slash;
842 __u32 fwmark, fwmask;
843
844 NEXT_ARG();
845
846 slash = strchr(*argv, '/');
847 if (slash != NULL)
848 *slash = '\0';
849 if (get_u32(&fwmark, *argv, 0))
850 invarg("fwmark value is invalid\n", *argv);
851 addattr32(&req.n, sizeof(req), FRA_FWMARK, fwmark);
852 if (slash) {
853 if (get_u32(&fwmask, slash+1, 0))
854 invarg("fwmask value is invalid\n",
855 slash+1);
856 addattr32(&req.n, sizeof(req),
857 FRA_FWMASK, fwmask);
858 }
859 } else if (matches(*argv, "realms") == 0) {
860 __u32 realm;
861
862 NEXT_ARG();
863 if (get_rt_realms_or_raw(&realm, *argv))
864 invarg("invalid realms\n", *argv);
865 addattr32(&req.n, sizeof(req), FRA_FLOW, realm);
866 } else if (matches(*argv, "protocol") == 0) {
867 __u32 proto;
868
869 NEXT_ARG();
870 if (rtnl_rtprot_a2n(&proto, *argv))
871 invarg("\"protocol\" value is invalid\n", *argv);
872 addattr8(&req.n, sizeof(req), FRA_PROTOCOL, proto);
873 } else if (matches(*argv, "tun_id") == 0) {
874 __u64 tun_id;
875
876 NEXT_ARG();
877 if (get_be64(&tun_id, *argv, 0))
878 invarg("\"tun_id\" value is invalid\n", *argv);
879 addattr64(&req.n, sizeof(req), FRA_TUN_ID, tun_id);
880 } else if (matches(*argv, "table") == 0 ||
881 strcmp(*argv, "lookup") == 0) {
882 NEXT_ARG();
883 if (rtnl_rttable_a2n(&tid, *argv))
884 invarg("invalid table ID\n", *argv);
885 if (tid < 256)
886 req.frh.table = tid;
887 else {
888 req.frh.table = RT_TABLE_UNSPEC;
889 addattr32(&req.n, sizeof(req), FRA_TABLE, tid);
890 }
891 table_ok = 1;
892 } else if (matches(*argv, "suppress_prefixlength") == 0 ||
893 strcmp(*argv, "sup_pl") == 0) {
894 int pl;
895
896 NEXT_ARG();
897 if (get_s32(&pl, *argv, 0) || pl < 0)
898 invarg("suppress_prefixlength value is invalid\n",
899 *argv);
900 addattr32(&req.n, sizeof(req),
901 FRA_SUPPRESS_PREFIXLEN, pl);
902 } else if (matches(*argv, "suppress_ifgroup") == 0 ||
903 strcmp(*argv, "sup_group") == 0) {
904 NEXT_ARG();
905 int group;
906
907 if (rtnl_group_a2n(&group, *argv))
908 invarg("Invalid \"suppress_ifgroup\" value\n",
909 *argv);
910 addattr32(&req.n, sizeof(req),
911 FRA_SUPPRESS_IFGROUP, group);
912 } else if (strcmp(*argv, "dev") == 0 ||
913 strcmp(*argv, "iif") == 0) {
914 NEXT_ARG();
915 if (check_ifname(*argv))
916 invarg("\"iif\"/\"dev\" not a valid ifname", *argv);
917 addattr_l(&req.n, sizeof(req), FRA_IFNAME,
918 *argv, strlen(*argv)+1);
919 } else if (strcmp(*argv, "oif") == 0) {
920 NEXT_ARG();
921 if (check_ifname(*argv))
922 invarg("\"oif\" not a valid ifname", *argv);
923 addattr_l(&req.n, sizeof(req), FRA_OIFNAME,
924 *argv, strlen(*argv)+1);
925 } else if (strcmp(*argv, "l3mdev") == 0) {
926 addattr8(&req.n, sizeof(req), FRA_L3MDEV, 1);
927 table_ok = 1;
928 l3mdev_rule = 1;
929 } else if (strcmp(*argv, "uidrange") == 0) {
930 struct fib_rule_uid_range r;
931
932 NEXT_ARG();
933 if (sscanf(*argv, "%u-%u", &r.start, &r.end) != 2)
934 invarg("invalid UID range\n", *argv);
935 addattr_l(&req.n, sizeof(req), FRA_UID_RANGE, &r,
936 sizeof(r));
937 } else if (strcmp(*argv, "nat") == 0 ||
938 matches(*argv, "map-to") == 0) {
939 NEXT_ARG();
940 fprintf(stderr, "Warning: route NAT is deprecated\n");
941 addattr32(&req.n, sizeof(req), RTA_GATEWAY,
942 get_addr32(*argv));
943 req.frh.action = RTN_NAT;
944 } else if (strcmp(*argv, "ipproto") == 0) {
945 int ipproto;
946
947 NEXT_ARG();
948 ipproto = inet_proto_a2n(*argv);
949 if (ipproto < 0)
950 invarg("Invalid \"ipproto\" value\n",
951 *argv);
952 addattr8(&req.n, sizeof(req), FRA_IP_PROTO, ipproto);
953 } else if (strcmp(*argv, "sport") == 0) {
954 struct fib_rule_port_range r;
955 int ret = 0;
956
957 NEXT_ARG();
958 ret = sscanf(*argv, "%hu-%hu", &r.start, &r.end);
959 if (ret == 1)
960 r.end = r.start;
961 else if (ret != 2)
962 invarg("invalid port range\n", *argv);
963 addattr_l(&req.n, sizeof(req), FRA_SPORT_RANGE, &r,
964 sizeof(r));
965 } else if (strcmp(*argv, "dport") == 0) {
966 struct fib_rule_port_range r;
967 int ret = 0;
968
969 NEXT_ARG();
970 ret = sscanf(*argv, "%hu-%hu", &r.start, &r.end);
971 if (ret == 1)
972 r.end = r.start;
973 else if (ret != 2)
974 invarg("invalid dport range\n", *argv);
975 addattr_l(&req.n, sizeof(req), FRA_DPORT_RANGE, &r,
976 sizeof(r));
977 } else {
978 int type;
979
980 if (strcmp(*argv, "type") == 0)
981 NEXT_ARG();
982
983 if (matches(*argv, "help") == 0)
984 usage();
985 else if (matches(*argv, "goto") == 0) {
986 __u32 target;
987
988 type = FR_ACT_GOTO;
989 NEXT_ARG();
990 if (get_u32(&target, *argv, 0))
991 invarg("invalid target\n", *argv);
992 addattr32(&req.n, sizeof(req),
993 FRA_GOTO, target);
994 } else if (matches(*argv, "nop") == 0)
995 type = FR_ACT_NOP;
996 else if (rtnl_rtntype_a2n(&type, *argv))
997 invarg("Failed to parse rule type", *argv);
998 req.frh.action = type;
999 table_ok = 1;
1000 }
1001 argc--;
1002 argv++;
1003 }
1004
1005 if (l3mdev_rule && tid != 0) {
1006 fprintf(stderr,
1007 "table can not be specified for l3mdev rules\n");
1008 return -EINVAL;
1009 }
1010
1011 if (req.frh.family == AF_UNSPEC)
1012 req.frh.family = AF_INET;
1013
1014 if (!table_ok && cmd == RTM_NEWRULE)
1015 req.frh.table = RT_TABLE_MAIN;
1016
1017 if (rtnl_talk(&rth, &req.n, NULL) < 0)
1018 return -2;
1019
1020 return 0;
1021}
1022
1023int do_iprule(int argc, char **argv)
1024{
1025 if (argc < 1) {
1026 return iprule_list_flush_or_save(0, NULL, IPRULE_LIST);
1027 } else if (matches(argv[0], "list") == 0 ||
1028 matches(argv[0], "lst") == 0 ||
1029 matches(argv[0], "show") == 0) {
1030 return iprule_list_flush_or_save(argc-1, argv+1, IPRULE_LIST);
1031 } else if (matches(argv[0], "save") == 0) {
1032 return iprule_list_flush_or_save(argc-1, argv+1, IPRULE_SAVE);
1033 } else if (matches(argv[0], "restore") == 0) {
1034 return iprule_restore();
1035 } else if (matches(argv[0], "add") == 0) {
1036 return iprule_modify(RTM_NEWRULE, argc-1, argv+1);
1037 } else if (matches(argv[0], "delete") == 0) {
1038 return iprule_modify(RTM_DELRULE, argc-1, argv+1);
1039 } else if (matches(argv[0], "flush") == 0) {
1040 return iprule_list_flush_or_save(argc-1, argv+1, IPRULE_FLUSH);
1041 } else if (matches(argv[0], "help") == 0)
1042 usage();
1043
1044 fprintf(stderr,
1045 "Command \"%s\" is unknown, try \"ip rule help\".\n", *argv);
1046 exit(-1);
1047}
1048
1049int do_multirule(int argc, char **argv)
1050{
1051 switch (preferred_family) {
1052 case AF_UNSPEC:
1053 case AF_INET:
1054 preferred_family = RTNL_FAMILY_IPMR;
1055 break;
1056 case AF_INET6:
1057 preferred_family = RTNL_FAMILY_IP6MR;
1058 break;
1059 case RTNL_FAMILY_IPMR:
1060 case RTNL_FAMILY_IP6MR:
1061 break;
1062 default:
1063 fprintf(stderr,
1064 "Multicast rules are only supported for IPv4/IPv6, was: %i\n",
1065 preferred_family);
1066 exit(-1);
1067 }
1068
1069 return do_iprule(argc, argv);
1070}
1071
