ip/iprule.c

   1 /*
   2  * iprule.c             "ip rule".
   3  *
   4  *              This program is free software; you can redistribute it and/or
   5  *              modify it under the terms of the GNU General Public License
   6  *              as published by the Free Software Foundation; either version
   7  *              2 of the License, or (at your option) any later version.
   8  *
   9  * Authors:     Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
  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
  31 enum list_action {
  32         IPRULE_LIST,
  33         IPRULE_FLUSH,
  34         IPRULE_SAVE,
  35 };
  36
  37 extern struct rtnl_handle rth;
  38
  39 static void usage(void) __attribute__((noreturn));
  40
  41 static 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
  65 static 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
  87 static 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
  95 static 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
 233 int 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
 278                         print_string(PRINT_FP, NULL, " ", NULL);
 279         } else if (frh->src_len) {
 280                 print_string(PRINT_ANY, "src", "from %s", "0");
 281                 print_uint(PRINT_ANY, "srclen", "/%u ", frh->src_len);
 282         } else {
 283                 print_string(PRINT_ANY, "src", "from %s ", "all");
 284         }
 285
 286         if (tb[FRA_DST]) {
 287                 const char *dst = rt_addr_n2a_rta(frh->family, tb[FRA_DST]);
 288
 289                 print_string(PRINT_FP, NULL, "to ", NULL);
 290                 print_color_string(PRINT_ANY, ifa_family_color(frh->family),
 291                                    "dst", "%s", dst);
 292                 if (frh->dst_len != host_len)
 293                         print_uint(PRINT_ANY, "dstlen", "/%u ", frh->dst_len);
 294                 else
 295                         print_string(PRINT_FP, NULL, " ", NULL);
 296         } else if (frh->dst_len) {
 297                 print_string(PRINT_ANY, "dst", "to %s", "0");
 298                 print_uint(PRINT_ANY, "dstlen", "/%u ", frh->dst_len);
 299         }
 300
 301         if (frh->tos) {
 302                 print_string(PRINT_ANY, "tos",
 303                              "tos %s ",
 304                              rtnl_dsfield_n2a(frh->tos, b1, sizeof(b1)));
 305         }
 306
 307         if (tb[FRA_FWMARK] || tb[FRA_FWMASK]) {
 308                 __u32 mark = 0, mask = 0;
 309
 310                 if (tb[FRA_FWMARK])
 311                         mark = rta_getattr_u32(tb[FRA_FWMARK]);
 312
 313                 if (tb[FRA_FWMASK] &&
 314                     (mask = rta_getattr_u32(tb[FRA_FWMASK])) != 0xFFFFFFFF) {
 315                         print_0xhex(PRINT_ANY, "fwmark", "fwmark %#llx", mark);
 316                         print_0xhex(PRINT_ANY, "fwmask", "/%#llx ", mask);
 317                 } else {
 318                         print_0xhex(PRINT_ANY, "fwmark", "fwmark %#llx ", mark);
 319                 }
 320         }
 321
 322         if (tb[FRA_IFNAME]) {
 323                 if (!is_json_context())
 324                         fprintf(fp, "iif ");
 325                 print_color_string(PRINT_ANY, COLOR_IFNAME,
 326                                    "iif", "%s ",
 327                                    rta_getattr_str(tb[FRA_IFNAME]));
 328
 329                 if (frh->flags & FIB_RULE_IIF_DETACHED)
 330                         print_null(PRINT_ANY, "iif_detached", "[detached] ",
 331                                    NULL);
 332         }
 333
 334         if (tb[FRA_OIFNAME]) {
 335                 if (!is_json_context())
 336                         fprintf(fp, "oif ");
 337
 338                 print_color_string(PRINT_ANY, COLOR_IFNAME, "oif", "%s ",
 339                                    rta_getattr_str(tb[FRA_OIFNAME]));
 340
 341                 if (frh->flags & FIB_RULE_OIF_DETACHED)
 342                         print_null(PRINT_ANY, "oif_detached", "[detached] ",
 343                                    NULL);
 344         }
 345
 346         if (tb[FRA_L3MDEV]) {
 347                 __u8 mdev = rta_getattr_u8(tb[FRA_L3MDEV]);
 348
 349                 if (mdev)
 350                         print_null(PRINT_ANY, "l3mdev",
 351                                    "lookup [l3mdev-table] ", NULL);
 352         }
 353
 354         if (tb[FRA_UID_RANGE]) {
 355                 struct fib_rule_uid_range *r = RTA_DATA(tb[FRA_UID_RANGE]);
 356
 357                 print_uint(PRINT_ANY, "uid_start", "uidrange %u", r->start);
 358                 print_uint(PRINT_ANY, "uid_end", "-%u ", r->end);
 359         }
 360
 361         if (tb[FRA_IP_PROTO]) {
 362                 SPRINT_BUF(pbuf);
 363                 print_string(PRINT_ANY, "ipproto", "ipproto %s ",
 364                              inet_proto_n2a(rta_getattr_u8(tb[FRA_IP_PROTO]),
 365                                             pbuf, sizeof(pbuf)));
 366         }
 367
 368         if (tb[FRA_SPORT_RANGE]) {
 369                 struct fib_rule_port_range *r = RTA_DATA(tb[FRA_SPORT_RANGE]);
 370
 371                 if (r->start == r->end) {
 372                         print_uint(PRINT_ANY, "sport", "sport %u ", r->start);
 373                 } else {
 374                         print_uint(PRINT_ANY, "sport_start", "sport %u",
 375                                    r->start);
 376                         print_uint(PRINT_ANY, "sport_end", "-%u ", r->end);
 377                 }
 378         }
 379
 380         if (tb[FRA_DPORT_RANGE]) {
 381                 struct fib_rule_port_range *r = RTA_DATA(tb[FRA_DPORT_RANGE]);
 382
 383                 if (r->start == r->end) {
 384                         print_uint(PRINT_ANY, "dport", "dport %u ", r->start);
 385                 } else {
 386                         print_uint(PRINT_ANY, "dport_start", "dport %u",
 387                                    r->start);
 388                         print_uint(PRINT_ANY, "dport_end", "-%u ", r->end);
 389                 }
 390         }
 391
 392         if (tb[FRA_TUN_ID]) {
 393                 __u64 tun_id = ntohll(rta_getattr_u64(tb[FRA_TUN_ID]));
 394
 395                 print_u64(PRINT_ANY, "tun_id", "tun_id %llu ", tun_id);
 396         }
 397
 398         table = frh_get_table(frh, tb);
 399         if (table) {
 400                 print_string(PRINT_ANY, "table",
 401                              "lookup %s ",
 402                              rtnl_rttable_n2a(table, b1, sizeof(b1)));
 403
 404                 if (tb[FRA_SUPPRESS_PREFIXLEN]) {
 405                         int pl = rta_getattr_u32(tb[FRA_SUPPRESS_PREFIXLEN]);
 406
 407                         if (pl != -1)
 408                                 print_int(PRINT_ANY, "suppress_prefixlen",
 409                                           "suppress_prefixlength %d ", pl);
 410                 }
 411
 412                 if (tb[FRA_SUPPRESS_IFGROUP]) {
 413                         int group = rta_getattr_u32(tb[FRA_SUPPRESS_IFGROUP]);
 414
 415                         if (group != -1) {
 416                                 const char *grname
 417                                         = rtnl_group_n2a(group, b1, sizeof(b1));
 418
 419                                 print_string(PRINT_ANY, "suppress_ifgroup",
 420                                              "suppress_ifgroup %s ", grname);
 421                         }
 422                 }
 423         }
 424
 425         if (tb[FRA_FLOW]) {
 426                 __u32 to = rta_getattr_u32(tb[FRA_FLOW]);
 427                 __u32 from = to>>16;
 428
 429                 to &= 0xFFFF;
 430                 if (from)
 431                         print_string(PRINT_ANY,
 432                                      "flow_from", "realms %s/",
 433                                      rtnl_rtrealm_n2a(from, b1, sizeof(b1)));
 434
 435                 print_string(PRINT_ANY, "flow_to", "%s ",
 436                              rtnl_rtrealm_n2a(to, b1, sizeof(b1)));
 437         }
 438
 439         if (frh->action == RTN_NAT) {
 440                 if (tb[RTA_GATEWAY]) {
 441                         const char *gateway;
 442
 443                         gateway = format_host_rta(frh->family, tb[RTA_GATEWAY]);
 444
 445                         print_string(PRINT_ANY, "nat_gateway",
 446                                      "map-to %s ", gateway);
 447                 } else {
 448                         print_null(PRINT_ANY, "masquerade", "masquerade", NULL);
 449                 }
 450         } else if (frh->action == FR_ACT_GOTO) {
 451                 if (tb[FRA_GOTO])
 452                         print_uint(PRINT_ANY, "goto", "goto %u",
 453                                    rta_getattr_u32(tb[FRA_GOTO]));
 454                 else
 455                         print_string(PRINT_ANY, "goto", "goto %s", "none");
 456
 457                 if (frh->flags & FIB_RULE_UNRESOLVED)
 458                         print_null(PRINT_ANY, "unresolved", "unresolved", NULL);
 459         } else if (frh->action == FR_ACT_NOP) {
 460                 print_null(PRINT_ANY, "nop", "nop", NULL);
 461         } else if (frh->action != FR_ACT_TO_TBL) {
 462                 print_string(PRINT_ANY, "action", "%s",
 463                              rtnl_rtntype_n2a(frh->action, b1, sizeof(b1)));
 464         }
 465
 466         if (tb[FRA_PROTOCOL]) {
 467                 __u8 protocol = rta_getattr_u8(tb[FRA_PROTOCOL]);
 468
 469                 if ((protocol && protocol != RTPROT_KERNEL) || show_details > 0) {
 470                         print_string(PRINT_ANY, "protocol", " proto %s ",
 471                                      rtnl_rtprot_n2a(protocol, b1, sizeof(b1)));
 472                 }
 473         }
 474         print_string(PRINT_FP, NULL, "\n", "");
 475         close_json_object();
 476         fflush(fp);
 477         return 0;
 478 }
 479
 480 static __u32 rule_dump_magic = 0x71706986;
 481
 482 static int save_rule_prep(void)
 483 {
 484         int ret;
 485
 486         if (isatty(STDOUT_FILENO)) {
 487                 fprintf(stderr, "Not sending a binary stream to stdout\n");
 488                 return -1;
 489         }
 490
 491         ret = write(STDOUT_FILENO, &rule_dump_magic, sizeof(rule_dump_magic));
 492         if (ret != sizeof(rule_dump_magic)) {
 493                 fprintf(stderr, "Can't write magic to dump file\n");
 494                 return -1;
 495         }
 496
 497         return 0;
 498 }
 499
 500 static int save_rule(struct nlmsghdr *n, void *arg)
 501 {
 502         int ret;
 503
 504         ret = write(STDOUT_FILENO, n, n->nlmsg_len);
 505         if ((ret > 0) && (ret != n->nlmsg_len)) {
 506                 fprintf(stderr, "Short write while saving nlmsg\n");
 507                 ret = -EIO;
 508         }
 509
 510         return ret == n->nlmsg_len ? 0 : ret;
 511 }
 512
 513 static int flush_rule(struct nlmsghdr *n, void *arg)
 514 {
 515         struct rtnl_handle rth2;
 516         struct fib_rule_hdr *frh = NLMSG_DATA(n);
 517         int len = n->nlmsg_len;
 518         struct rtattr *tb[FRA_MAX+1];
 519         int host_len = -1;
 520
 521         len -= NLMSG_LENGTH(sizeof(*frh));
 522         if (len < 0)
 523                 return -1;
 524
 525         parse_rtattr(tb, FRA_MAX, RTM_RTA(frh), len);
 526
 527         host_len = af_bit_len(frh->family);
 528         if (!filter_nlmsg(n, tb, host_len))
 529                 return 0;
 530
 531         if (tb[FRA_PROTOCOL]) {
 532                 __u8 protocol = rta_getattr_u8(tb[FRA_PROTOCOL]);
 533
 534                 if ((filter.protocol ^ protocol) & filter.protocolmask)
 535                         return 0;
 536         }
 537
 538         if (tb[FRA_PRIORITY]) {
 539                 n->nlmsg_type = RTM_DELRULE;
 540                 n->nlmsg_flags = NLM_F_REQUEST;
 541
 542                 if (rtnl_open(&rth2, 0) < 0)
 543                         return -1;
 544
 545                 if (rtnl_talk(&rth2, n, NULL) < 0)
 546                         return -2;
 547
 548                 rtnl_close(&rth2);
 549         }
 550
 551         return 0;
 552 }
 553
 554 static int iprule_list_flush_or_save(int argc, char **argv, int action)
 555 {
 556         rtnl_filter_t filter_fn;
 557         int af = preferred_family;
 558
 559         if (af == AF_UNSPEC)
 560                 af = AF_INET;
 561
 562         if (action == IPRULE_SAVE && argc > 0) {
 563                 fprintf(stderr, "\"ip rule save\" does not take any arguments.\n");
 564                 return -1;
 565         }
 566
 567         switch (action) {
 568         case IPRULE_SAVE:
 569                 if (save_rule_prep())
 570                         return -1;
 571                 filter_fn = save_rule;
 572                 break;
 573         case IPRULE_FLUSH:
 574                 filter_fn = flush_rule;
 575                 break;
 576         default:
 577                 filter_fn = print_rule;
 578         }
 579
 580         memset(&filter, 0, sizeof(filter));
 581
 582         while (argc > 0) {
 583                 if (matches(*argv, "preference") == 0 ||
 584                     matches(*argv, "order") == 0 ||
 585                     matches(*argv, "priority") == 0) {
 586                         __u32 pref;
 587
 588                         NEXT_ARG();
 589                         if (get_u32(&pref, *argv, 0))
 590                                 invarg("preference value is invalid\n", *argv);
 591                         filter.pref = pref;
 592                         filter.prefmask = 1;
 593                 } else if (strcmp(*argv, "not") == 0) {
 594                         filter.not = 1;
 595                 } else if (strcmp(*argv, "tos") == 0) {
 596                         __u32 tos;
 597
 598                         NEXT_ARG();
 599                         if (rtnl_dsfield_a2n(&tos, *argv))
 600                                 invarg("TOS value is invalid\n", *argv);
 601                         filter.tos = tos;
 602                         filter.tosmask = 1;
 603                 } else if (strcmp(*argv, "fwmark") == 0) {
 604                         char *slash;
 605                         __u32 fwmark, fwmask;
 606
 607                         NEXT_ARG();
 608                         slash = strchr(*argv, '/');
 609                         if (slash != NULL)
 610                                 *slash = '\0';
 611                         if (get_u32(&fwmark, *argv, 0))
 612                                 invarg("fwmark value is invalid\n", *argv);
 613                         filter.fwmark = fwmark;
 614                         if (slash) {
 615                                 if (get_u32(&fwmask, slash+1, 0))
 616                                         invarg("fwmask value is invalid\n",
 617                                                slash+1);
 618                                 filter.fwmask = fwmask;
 619                         }
 620                 } else if (strcmp(*argv, "dev") == 0 ||
 621                            strcmp(*argv, "iif") == 0) {
 622                         NEXT_ARG();
 623                         if (get_ifname(filter.iif, *argv))
 624                                 invarg("\"iif\"/\"dev\" not a valid ifname", *argv);
 625                         filter.iifmask = 1;
 626                 } else if (strcmp(*argv, "oif") == 0) {
 627                         NEXT_ARG();
 628                         if (get_ifname(filter.oif, *argv))
 629                                 invarg("\"oif\" not a valid ifname", *argv);
 630                         filter.oifmask = 1;
 631                 } else if (strcmp(*argv, "l3mdev") == 0) {
 632                         filter.l3mdev = 1;
 633                 } else if (strcmp(*argv, "uidrange") == 0) {
 634                         NEXT_ARG();
 635                         filter.uidrange = 1;
 636                         if (sscanf(*argv, "%u-%u",
 637                                    &filter.range.start,
 638                                    &filter.range.end) != 2)
 639                                 invarg("invalid UID range\n", *argv);
 640
 641                 } else if (matches(*argv, "tun_id") == 0) {
 642                         __u64 tun_id;
 643
 644                         NEXT_ARG();
 645                         if (get_u64(&tun_id, *argv, 0))
 646                                 invarg("\"tun_id\" value is invalid\n", *argv);
 647                         filter.tun_id = tun_id;
 648                 } else if (matches(*argv, "lookup") == 0 ||
 649                            matches(*argv, "table") == 0) {
 650                         __u32 tid;
 651
 652                         NEXT_ARG();
 653                         if (rtnl_rttable_a2n(&tid, *argv))
 654                                 invarg("table id value is invalid\n", *argv);
 655                         filter.tb = tid;
 656                 } else if (matches(*argv, "from") == 0 ||
 657                            matches(*argv, "src") == 0) {
 658                         NEXT_ARG();
 659                         if (get_prefix(&filter.src, *argv, af))
 660                                 invarg("from value is invalid\n", *argv);
 661                 } else if (matches(*argv, "protocol") == 0) {
 662                         __u32 prot;
 663                         NEXT_ARG();
 664                         filter.protocolmask = -1;
 665                         if (rtnl_rtprot_a2n(&prot, *argv)) {
 666                                 if (strcmp(*argv, "all") != 0)
 667                                         invarg("invalid \"protocol\"\n", *argv);
 668                                 prot = 0;
 669                                 filter.protocolmask = 0;
 670                         }
 671                         filter.protocol = prot;
 672                 } else if (strcmp(*argv, "ipproto") == 0) {
 673                         int ipproto;
 674
 675                         NEXT_ARG();
 676                         ipproto = inet_proto_a2n(*argv);
 677                         if (ipproto < 0)
 678                                 invarg("Invalid \"ipproto\" value\n", *argv);
 679                         filter.ipproto = ipproto;
 680                 } else if (strcmp(*argv, "sport") == 0) {
 681                         struct fib_rule_port_range r;
 682                         int ret;
 683
 684                         NEXT_ARG();
 685                         ret = sscanf(*argv, "%hu-%hu", &r.start, &r.end);
 686                         if (ret == 1)
 687                                 r.end = r.start;
 688                         else if (ret != 2)
 689                                 invarg("invalid port range\n", *argv);
 690                         filter.sport = r;
 691                 } else if (strcmp(*argv, "dport") == 0) {
 692                         struct fib_rule_port_range r;
 693                         int ret;
 694
 695                         NEXT_ARG();
 696                         ret = sscanf(*argv, "%hu-%hu", &r.start, &r.end);
 697                         if (ret == 1)
 698                                 r.end = r.start;
 699                         else if (ret != 2)
 700                                 invarg("invalid dport range\n", *argv);
 701                         filter.dport = r;
 702                 } else{
 703                         if (matches(*argv, "dst") == 0 ||
 704                             matches(*argv, "to") == 0) {
 705                                 NEXT_ARG();
 706                         }
 707                         if (get_prefix(&filter.dst, *argv, af))
 708                                 invarg("to value is invalid\n", *argv);
 709                 }
 710                 argc--; argv++;
 711         }
 712
 713         if (rtnl_ruledump_req(&rth, af) < 0) {
 714                 perror("Cannot send dump request");
 715                 return 1;
 716         }
 717
 718         new_json_obj(json);
 719         if (rtnl_dump_filter(&rth, filter_fn, stdout) < 0) {
 720                 fprintf(stderr, "Dump terminated\n");
 721                 return 1;
 722         }
 723         delete_json_obj();
 724
 725         return 0;
 726 }
 727
 728 static int rule_dump_check_magic(void)
 729 {
 730         int ret;
 731         __u32 magic = 0;
 732
 733         if (isatty(STDIN_FILENO)) {
 734                 fprintf(stderr, "Can't restore rule dump from a terminal\n");
 735                 return -1;
 736         }
 737
 738         ret = fread(&magic, sizeof(magic), 1, stdin);
 739         if (magic != rule_dump_magic) {
 740                 fprintf(stderr, "Magic mismatch (%d elems, %x magic)\n",
 741                         ret, magic);
 742                 return -1;
 743         }
 744
 745         return 0;
 746 }
 747
 748 static int restore_handler(struct rtnl_ctrl_data *ctrl,
 749                            struct nlmsghdr *n, void *arg)
 750 {
 751         int ret;
 752
 753         n->nlmsg_flags |= NLM_F_REQUEST | NLM_F_CREATE | NLM_F_ACK;
 754
 755         ll_init_map(&rth);
 756
 757         ret = rtnl_talk(&rth, n, NULL);
 758         if ((ret < 0) && (errno == EEXIST))
 759                 ret = 0;
 760
 761         return ret;
 762 }
 763
 764
 765 static int iprule_restore(void)
 766 {
 767         if (rule_dump_check_magic())
 768                 exit(-1);
 769
 770         exit(rtnl_from_file(stdin, &restore_handler, NULL));
 771 }
 772
 773 static int iprule_modify(int cmd, int argc, char **argv)
 774 {
 775         int l3mdev_rule = 0;
 776         int table_ok = 0;
 777         __u32 tid = 0;
 778         struct {
 779                 struct nlmsghdr n;
 780                 struct fib_rule_hdr     frh;
 781                 char                    buf[1024];
 782         } req = {
 783                 .n.nlmsg_type = cmd,
 784                 .n.nlmsg_len = NLMSG_LENGTH(sizeof(struct fib_rule_hdr)),
 785                 .n.nlmsg_flags = NLM_F_REQUEST,
 786                 .frh.family = preferred_family,
 787                 .frh.action = FR_ACT_UNSPEC,
 788         };
 789
 790         if (cmd == RTM_NEWRULE) {
 791                 if (argc == 0) {
 792                         fprintf(stderr,
 793                                 "\"ip rule add\" requires arguments.\n");
 794                         return -1;
 795                 }
 796                 req.n.nlmsg_flags |= NLM_F_CREATE|NLM_F_EXCL;
 797                 req.frh.action = FR_ACT_TO_TBL;
 798         }
 799
 800         if (cmd == RTM_DELRULE && argc == 0) {
 801                 fprintf(stderr, "\"ip rule del\" requires arguments.\n");
 802                 return -1;
 803         }
 804
 805         while (argc > 0) {
 806                 if (strcmp(*argv, "not") == 0) {
 807                         req.frh.flags |= FIB_RULE_INVERT;
 808                 } else if (strcmp(*argv, "from") == 0) {
 809                         inet_prefix dst;
 810
 811                         NEXT_ARG();
 812                         get_prefix(&dst, *argv, req.frh.family);
 813                         req.frh.src_len = dst.bitlen;
 814                         addattr_l(&req.n, sizeof(req), FRA_SRC,
 815                                   &dst.data, dst.bytelen);
 816                 } else if (strcmp(*argv, "to") == 0) {
 817                         inet_prefix dst;
 818
 819                         NEXT_ARG();
 820                         get_prefix(&dst, *argv, req.frh.family);
 821                         req.frh.dst_len = dst.bitlen;
 822                         addattr_l(&req.n, sizeof(req), FRA_DST,
 823                                   &dst.data, dst.bytelen);
 824                 } else if (matches(*argv, "preference") == 0 ||
 825                            matches(*argv, "order") == 0 ||
 826                            matches(*argv, "priority") == 0) {
 827                         __u32 pref;
 828
 829                         NEXT_ARG();
 830                         if (get_u32(&pref, *argv, 0))
 831                                 invarg("preference value is invalid\n", *argv);
 832                         addattr32(&req.n, sizeof(req), FRA_PRIORITY, pref);
 833                 } else if (strcmp(*argv, "tos") == 0 ||
 834                            matches(*argv, "dsfield") == 0) {
 835                         __u32 tos;
 836
 837                         NEXT_ARG();
 838                         if (rtnl_dsfield_a2n(&tos, *argv))
 839                                 invarg("TOS value is invalid\n", *argv);
 840                         req.frh.tos = tos;
 841                 } else if (strcmp(*argv, "fwmark") == 0) {
 842                         char *slash;
 843                         __u32 fwmark, fwmask;
 844
 845                         NEXT_ARG();
 846
 847                         slash = strchr(*argv, '/');
 848                         if (slash != NULL)
 849                                 *slash = '\0';
 850                         if (get_u32(&fwmark, *argv, 0))
 851                                 invarg("fwmark value is invalid\n", *argv);
 852                         addattr32(&req.n, sizeof(req), FRA_FWMARK, fwmark);
 853                         if (slash) {
 854                                 if (get_u32(&fwmask, slash+1, 0))
 855                                         invarg("fwmask value is invalid\n",
 856                                                slash+1);
 857                                 addattr32(&req.n, sizeof(req),
 858                                           FRA_FWMASK, fwmask);
 859                         }
 860                 } else if (matches(*argv, "realms") == 0) {
 861                         __u32 realm;
 862
 863                         NEXT_ARG();
 864                         if (get_rt_realms_or_raw(&realm, *argv))
 865                                 invarg("invalid realms\n", *argv);
 866                         addattr32(&req.n, sizeof(req), FRA_FLOW, realm);
 867                 } else if (matches(*argv, "protocol") == 0) {
 868                         __u32 proto;
 869
 870                         NEXT_ARG();
 871                         if (rtnl_rtprot_a2n(&proto, *argv))
 872                                 invarg("\"protocol\" value is invalid\n", *argv);
 873                         addattr8(&req.n, sizeof(req), FRA_PROTOCOL, proto);
 874                 } else if (matches(*argv, "tun_id") == 0) {
 875                         __u64 tun_id;
 876
 877                         NEXT_ARG();
 878                         if (get_be64(&tun_id, *argv, 0))
 879                                 invarg("\"tun_id\" value is invalid\n", *argv);
 880                         addattr64(&req.n, sizeof(req), FRA_TUN_ID, tun_id);
 881                 } else if (matches(*argv, "table") == 0 ||
 882                            strcmp(*argv, "lookup") == 0) {
 883                         NEXT_ARG();
 884                         if (rtnl_rttable_a2n(&tid, *argv))
 885                                 invarg("invalid table ID\n", *argv);
 886                         if (tid < 256)
 887                                 req.frh.table = tid;
 888                         else {
 889                                 req.frh.table = RT_TABLE_UNSPEC;
 890                                 addattr32(&req.n, sizeof(req), FRA_TABLE, tid);
 891                         }
 892                         table_ok = 1;
 893                 } else if (matches(*argv, "suppress_prefixlength") == 0 ||
 894                            strcmp(*argv, "sup_pl") == 0) {
 895                         int pl;
 896
 897                         NEXT_ARG();
 898                         if (get_s32(&pl, *argv, 0) || pl < 0)
 899                                 invarg("suppress_prefixlength value is invalid\n",
 900                                        *argv);
 901                         addattr32(&req.n, sizeof(req),
 902                                   FRA_SUPPRESS_PREFIXLEN, pl);
 903                 } else if (matches(*argv, "suppress_ifgroup") == 0 ||
 904                            strcmp(*argv, "sup_group") == 0) {
 905                         NEXT_ARG();
 906                         int group;
 907
 908                         if (rtnl_group_a2n(&group, *argv))
 909                                 invarg("Invalid \"suppress_ifgroup\" value\n",
 910                                        *argv);
 911                         addattr32(&req.n, sizeof(req),
 912                                   FRA_SUPPRESS_IFGROUP, group);
 913                 } else if (strcmp(*argv, "dev") == 0 ||
 914                            strcmp(*argv, "iif") == 0) {
 915                         NEXT_ARG();
 916                         if (check_ifname(*argv))
 917                                 invarg("\"iif\"/\"dev\" not a valid ifname", *argv);
 918                         addattr_l(&req.n, sizeof(req), FRA_IFNAME,
 919                                   *argv, strlen(*argv)+1);
 920                 } else if (strcmp(*argv, "oif") == 0) {
 921                         NEXT_ARG();
 922                         if (check_ifname(*argv))
 923                                 invarg("\"oif\" not a valid ifname", *argv);
 924                         addattr_l(&req.n, sizeof(req), FRA_OIFNAME,
 925                                   *argv, strlen(*argv)+1);
 926                 } else if (strcmp(*argv, "l3mdev") == 0) {
 927                         addattr8(&req.n, sizeof(req), FRA_L3MDEV, 1);
 928                         table_ok = 1;
 929                         l3mdev_rule = 1;
 930                 } else if (strcmp(*argv, "uidrange") == 0) {
 931                         struct fib_rule_uid_range r;
 932
 933                         NEXT_ARG();
 934                         if (sscanf(*argv, "%u-%u", &r.start, &r.end) != 2)
 935                                 invarg("invalid UID range\n", *argv);
 936                         addattr_l(&req.n, sizeof(req), FRA_UID_RANGE, &r,
 937                                   sizeof(r));
 938                 } else if (strcmp(*argv, "nat") == 0 ||
 939                            matches(*argv, "map-to") == 0) {
 940                         NEXT_ARG();
 941                         fprintf(stderr, "Warning: route NAT is deprecated\n");
 942                         addattr32(&req.n, sizeof(req), RTA_GATEWAY,
 943                                   get_addr32(*argv));
 944                         req.frh.action = RTN_NAT;
 945                 } else if (strcmp(*argv, "ipproto") == 0) {
 946                         int ipproto;
 947
 948                         NEXT_ARG();
 949                         ipproto = inet_proto_a2n(*argv);
 950                         if (ipproto < 0)
 951                                 invarg("Invalid \"ipproto\" value\n",
 952                                        *argv);
 953                         addattr8(&req.n, sizeof(req), FRA_IP_PROTO, ipproto);
 954                 } else if (strcmp(*argv, "sport") == 0) {
 955                         struct fib_rule_port_range r;
 956                         int ret = 0;
 957
 958                         NEXT_ARG();
 959                         ret = sscanf(*argv, "%hu-%hu", &r.start, &r.end);
 960                         if (ret == 1)
 961                                 r.end = r.start;
 962                         else if (ret != 2)
 963                                 invarg("invalid port range\n", *argv);
 964                         addattr_l(&req.n, sizeof(req), FRA_SPORT_RANGE, &r,
 965                                   sizeof(r));
 966                 } else if (strcmp(*argv, "dport") == 0) {
 967                         struct fib_rule_port_range r;
 968                         int ret = 0;
 969
 970                         NEXT_ARG();
 971                         ret = sscanf(*argv, "%hu-%hu", &r.start, &r.end);
 972                         if (ret == 1)
 973                                 r.end = r.start;
 974                         else if (ret != 2)
 975                                 invarg("invalid dport range\n", *argv);
 976                         addattr_l(&req.n, sizeof(req), FRA_DPORT_RANGE, &r,
 977                                   sizeof(r));
 978                 } else {
 979                         int type;
 980
 981                         if (strcmp(*argv, "type") == 0)
 982                                 NEXT_ARG();
 983
 984                         if (matches(*argv, "help") == 0)
 985                                 usage();
 986                         else if (matches(*argv, "goto") == 0) {
 987                                 __u32 target;
 988
 989                                 type = FR_ACT_GOTO;
 990                                 NEXT_ARG();
 991                                 if (get_u32(&target, *argv, 0))
 992                                         invarg("invalid target\n", *argv);
 993                                 addattr32(&req.n, sizeof(req),
 994                                           FRA_GOTO, target);
 995                         } else if (matches(*argv, "nop") == 0)
 996                                 type = FR_ACT_NOP;
 997                         else if (rtnl_rtntype_a2n(&type, *argv))
 998                                 invarg("Failed to parse rule type", *argv);
 999                         req.frh.action = type;
1000                         table_ok = 1;
1001                 }
1002                 argc--;
1003                 argv++;
1004         }
1005
1006         if (l3mdev_rule && tid != 0) {
1007                 fprintf(stderr,
1008                         "table can not be specified for l3mdev rules\n");
1009                 return -EINVAL;
1010         }
1011
1012         if (req.frh.family == AF_UNSPEC)
1013                 req.frh.family = AF_INET;
1014
1015         if (!table_ok && cmd == RTM_NEWRULE)
1016                 req.frh.table = RT_TABLE_MAIN;
1017
1018         if (rtnl_talk(&rth, &req.n, NULL) < 0)
1019                 return -2;
1020
1021         return 0;
1022 }
1023
1024 int do_iprule(int argc, char **argv)
1025 {
1026         if (argc < 1) {
1027                 return iprule_list_flush_or_save(0, NULL, IPRULE_LIST);
1028         } else if (matches(argv[0], "list") == 0 ||
1029                    matches(argv[0], "lst") == 0 ||
1030                    matches(argv[0], "show") == 0) {
1031                 return iprule_list_flush_or_save(argc-1, argv+1, IPRULE_LIST);
1032         } else if (matches(argv[0], "save") == 0) {
1033                 return iprule_list_flush_or_save(argc-1, argv+1, IPRULE_SAVE);
1034         } else if (matches(argv[0], "restore") == 0) {
1035                 return iprule_restore();
1036         } else if (matches(argv[0], "add") == 0) {
1037                 return iprule_modify(RTM_NEWRULE, argc-1, argv+1);
1038         } else if (matches(argv[0], "delete") == 0) {
1039                 return iprule_modify(RTM_DELRULE, argc-1, argv+1);
1040         } else if (matches(argv[0], "flush") == 0) {
1041                 return iprule_list_flush_or_save(argc-1, argv+1, IPRULE_FLUSH);
1042         } else if (matches(argv[0], "help") == 0)
1043                 usage();
1044
1045         fprintf(stderr,
1046                 "Command \"%s\" is unknown, try \"ip rule help\".\n", *argv);
1047         exit(-1);
1048 }
1049
1050 int do_multirule(int argc, char **argv)
1051 {
1052         switch (preferred_family) {
1053         case AF_UNSPEC:
1054         case AF_INET:
1055                 preferred_family = RTNL_FAMILY_IPMR;
1056                 break;
1057         case AF_INET6:
1058                 preferred_family = RTNL_FAMILY_IP6MR;
1059                 break;
1060         case RTNL_FAMILY_IPMR:
1061         case RTNL_FAMILY_IP6MR:
1062                 break;
1063         default:
1064                 fprintf(stderr,
1065                         "Multicast rules are only supported for IPv4/IPv6, was: %i\n",
1066                         preferred_family);
1067                 exit(-1);
1068         }
1069
1070         return do_iprule(argc, argv);
1071 }