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