iproute2/ip/iprule.c
<<
>>
Prefs
   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
  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 ]\n"
  48                "            [ fwmark FWMARK[/MASK] ]\n"
  49                "            [ iif STRING ] [ oif STRING ] [ pref NUMBER ] [ l3mdev ]\n"
  50                "            [ uidrange NUMBER-NUMBER ]\n"
  51                "            [ ipproto PROTOCOL ]\n"
  52                "            [ sport [ NUMBER | NUMBER-NUMBER ]\n"
  53                "            [ dport [ NUMBER | NUMBER-NUMBER ] ]\n"
  54                "ACTION := [ table TABLE_ID ]\n"
  55                "          [ protocol PROTO ]\n"
  56                "          [ nat ADDRESS ]\n"
  57                "          [ realms [SRCREALM/]DSTREALM ]\n"
  58                "          [ goto NUMBER ]\n"
  59                "          SUPPRESSOR\n"
  60                "SUPPRESSOR := [ suppress_prefixlength NUMBER ]\n"
  61                "              [ suppress_ifgroup DEVGROUP ]\n"
  62                "TABLE_ID := [ local | main | default | NUMBER ]\n");
  63        exit(-1);
  64}
  65
  66static struct
  67{
  68        int not;
  69        int l3mdev;
  70        int iifmask, oifmask, uidrange;
  71        unsigned int tb;
  72        unsigned int tos, tosmask;
  73        unsigned int pref, prefmask;
  74        unsigned int fwmark, fwmask;
  75        uint64_t tun_id;
  76        char iif[IFNAMSIZ];
  77        char oif[IFNAMSIZ];
  78        struct fib_rule_uid_range range;
  79        inet_prefix src;
  80        inet_prefix dst;
  81        int protocol;
  82        int protocolmask;
  83        struct fib_rule_port_range sport;
  84        struct fib_rule_port_range dport;
  85        __u8 ipproto;
  86} filter;
  87
  88static inline int frh_get_table(struct fib_rule_hdr *frh, struct rtattr **tb)
  89{
  90        __u32 table = frh->table;
  91        if (tb[RTA_TABLE])
  92                table = rta_getattr_u32(tb[RTA_TABLE]);
  93        return table;
  94}
  95
  96static bool filter_nlmsg(struct nlmsghdr *n, struct rtattr **tb, int host_len)
  97{
  98        struct fib_rule_hdr *frh = NLMSG_DATA(n);
  99        __u32 table;
 100
 101        if (preferred_family != AF_UNSPEC && frh->family != preferred_family)
 102                return false;
 103
 104        if (filter.prefmask &&
 105            filter.pref ^ (tb[FRA_PRIORITY] ? rta_getattr_u32(tb[FRA_PRIORITY]) : 0))
 106                return false;
 107        if (filter.not && !(frh->flags & FIB_RULE_INVERT))
 108                return false;
 109
 110        if (filter.src.family) {
 111                inet_prefix *f_src = &filter.src;
 112
 113                if (f_src->family != frh->family ||
 114                    f_src->bitlen > frh->src_len)
 115                        return false;
 116
 117                if (inet_addr_match_rta(f_src, tb[FRA_SRC]))
 118                        return false;
 119        }
 120
 121        if (filter.dst.family) {
 122                inet_prefix *f_dst = &filter.dst;
 123
 124                if (f_dst->family != frh->family ||
 125                    f_dst->bitlen > frh->dst_len)
 126                        return false;
 127
 128                if (inet_addr_match_rta(f_dst, tb[FRA_DST]))
 129                        return false;
 130        }
 131
 132        if (filter.tosmask && filter.tos ^ frh->tos)
 133                return false;
 134
 135        if (filter.fwmark) {
 136                __u32 mark = 0;
 137
 138                if (tb[FRA_FWMARK])
 139                        mark = rta_getattr_u32(tb[FRA_FWMARK]);
 140                if (filter.fwmark ^ mark)
 141                        return false;
 142        }
 143        if (filter.fwmask) {
 144                __u32 mask = 0;
 145
 146                if (tb[FRA_FWMASK])
 147                        mask = rta_getattr_u32(tb[FRA_FWMASK]);
 148                if (filter.fwmask ^ mask)
 149                        return false;
 150        }
 151
 152        if (filter.iifmask) {
 153                if (tb[FRA_IFNAME]) {
 154                        if (strcmp(filter.iif, rta_getattr_str(tb[FRA_IFNAME])) != 0)
 155                                return false;
 156                } else {
 157                        return false;
 158                }
 159        }
 160
 161        if (filter.oifmask) {
 162                if (tb[FRA_OIFNAME]) {
 163                        if (strcmp(filter.oif, rta_getattr_str(tb[FRA_OIFNAME])) != 0)
 164                                return false;
 165                } else {
 166                        return false;
 167                }
 168        }
 169
 170        if (filter.l3mdev && !(tb[FRA_L3MDEV] && rta_getattr_u8(tb[FRA_L3MDEV])))
 171                return false;
 172
 173        if (filter.uidrange) {
 174                struct fib_rule_uid_range *r = RTA_DATA(tb[FRA_UID_RANGE]);
 175
 176                if (!tb[FRA_UID_RANGE] ||
 177                    r->start != filter.range.start ||
 178                    r->end != filter.range.end)
 179                        return false;
 180        }
 181
 182        if (filter.ipproto) {
 183                __u8 ipproto = 0;
 184
 185                if (tb[FRA_IP_PROTO])
 186                        ipproto = rta_getattr_u8(tb[FRA_IP_PROTO]);
 187                if (filter.ipproto != ipproto)
 188                        return false;
 189        }
 190
 191        if (filter.sport.start) {
 192                const struct fib_rule_port_range *r;
 193
 194                if (!tb[FRA_SPORT_RANGE])
 195                        return false;
 196
 197                r = RTA_DATA(tb[FRA_SPORT_RANGE]);
 198                if (r->start != filter.sport.start ||
 199                    r->end != filter.sport.end)
 200                        return false;
 201        }
 202
 203        if (filter.dport.start) {
 204                const struct fib_rule_port_range *r;
 205
 206                if (!tb[FRA_DPORT_RANGE])
 207                        return false;
 208
 209                r = RTA_DATA(tb[FRA_DPORT_RANGE]);
 210                if (r->start != filter.dport.start ||
 211                    r->end != filter.dport.end)
 212                        return false;
 213        }
 214
 215        if (filter.tun_id) {
 216                __u64 tun_id = 0;
 217
 218                if (tb[FRA_TUN_ID]) {
 219                        tun_id = ntohll(rta_getattr_u64(tb[FRA_TUN_ID]));
 220                        if (filter.tun_id != tun_id)
 221                                return false;
 222                } else {
 223                        return false;
 224                }
 225        }
 226
 227        table = frh_get_table(frh, tb);
 228        if (filter.tb > 0 && filter.tb ^ table)
 229                return false;
 230
 231        return true;
 232}
 233
 234int print_rule(struct nlmsghdr *n, void *arg)
 235{
 236        FILE *fp = arg;
 237        struct fib_rule_hdr *frh = NLMSG_DATA(n);
 238        int len = n->nlmsg_len;
 239        int host_len = -1;
 240        __u32 table, prio = 0;
 241        struct rtattr *tb[FRA_MAX+1];
 242        SPRINT_BUF(b1);
 243
 244        if (n->nlmsg_type != RTM_NEWRULE && n->nlmsg_type != RTM_DELRULE)
 245                return 0;
 246
 247        len -= NLMSG_LENGTH(sizeof(*frh));
 248        if (len < 0)
 249                return -1;
 250
 251        parse_rtattr(tb, FRA_MAX, RTM_RTA(frh), len);
 252
 253        host_len = af_bit_len(frh->family);
 254
 255        if (!filter_nlmsg(n, tb, host_len))
 256                return 0;
 257
 258        open_json_object(NULL);
 259        if (n->nlmsg_type == RTM_DELRULE)
 260                print_bool(PRINT_ANY, "deleted", "Deleted ", true);
 261
 262        if (tb[FRA_PRIORITY])
 263                prio = rta_getattr_u32(tb[FRA_PRIORITY]);
 264
 265        print_uint(PRINT_ANY, "priority", "%u:\t", prio);
 266
 267        if (frh->flags & FIB_RULE_INVERT)
 268                print_null(PRINT_ANY, "not", "not ", NULL);
 269
 270        if (tb[FRA_SRC]) {
 271                const char *src = rt_addr_n2a_rta(frh->family, tb[FRA_SRC]);
 272
 273                print_string(PRINT_FP, NULL, "from ", NULL);
 274                print_color_string(PRINT_ANY, ifa_family_color(frh->family),
 275                                   "src", "%s", src);
 276                if (frh->src_len != host_len)
 277                        print_uint(PRINT_ANY, "srclen", "/%u", frh->src_len);
 278        } else if (frh->src_len) {
 279                print_string(PRINT_ANY, "src", "from %s", "0");
 280                print_uint(PRINT_ANY, "srclen", "/%u", frh->src_len);
 281        } else {
 282                print_string(PRINT_ANY, "src", "from %s", "all");
 283        }
 284
 285        if (tb[FRA_DST]) {
 286                const char *dst = rt_addr_n2a_rta(frh->family, tb[FRA_DST]);
 287
 288                print_string(PRINT_FP, NULL, " to ", NULL);
 289                print_color_string(PRINT_ANY, ifa_family_color(frh->family),
 290                                   "dst", "%s", dst);
 291                if (frh->dst_len != host_len)
 292                        print_uint(PRINT_ANY, "dstlen", "/%u", frh->dst_len);
 293        } else if (frh->dst_len) {
 294                print_string(PRINT_ANY, "dst", " to %s", "0");
 295                print_uint(PRINT_ANY, "dstlen", "/%u", frh->dst_len);
 296        }
 297
 298        if (frh->tos) {
 299                print_string(PRINT_ANY, "tos",
 300                             " tos %s",
 301                             rtnl_dsfield_n2a(frh->tos, b1, sizeof(b1)));
 302        }
 303
 304        if (tb[FRA_FWMARK] || tb[FRA_FWMASK]) {
 305                __u32 mark = 0, mask = 0;
 306
 307                if (tb[FRA_FWMARK])
 308                        mark = rta_getattr_u32(tb[FRA_FWMARK]);
 309
 310                if (tb[FRA_FWMASK] &&
 311                    (mask = rta_getattr_u32(tb[FRA_FWMASK])) != 0xFFFFFFFF) {
 312                        print_0xhex(PRINT_ANY, "fwmark", " fwmark %#llx", mark);
 313                        print_0xhex(PRINT_ANY, "fwmask", "/%#llx", mask);
 314                } else {
 315                        print_0xhex(PRINT_ANY, "fwmark", " fwmark %#llx", mark);
 316                }
 317        }
 318
 319        if (tb[FRA_IFNAME]) {
 320                if (!is_json_context())
 321                        fprintf(fp, " iif ");
 322                print_color_string(PRINT_ANY, COLOR_IFNAME,
 323                                   "iif", "%s",
 324                                   rta_getattr_str(tb[FRA_IFNAME]));
 325
 326                if (frh->flags & FIB_RULE_IIF_DETACHED)
 327                        print_null(PRINT_ANY, "iif_detached", " [detached]",
 328                                   NULL);
 329        }
 330
 331        if (tb[FRA_OIFNAME]) {
 332                if (!is_json_context())
 333                        fprintf(fp, " oif ");
 334
 335                print_color_string(PRINT_ANY, COLOR_IFNAME, "oif", "%s",
 336                                   rta_getattr_str(tb[FRA_OIFNAME]));
 337
 338                if (frh->flags & FIB_RULE_OIF_DETACHED)
 339                        print_null(PRINT_ANY, "oif_detached", " [detached]",
 340                                   NULL);
 341        }
 342
 343        if (tb[FRA_L3MDEV]) {
 344                __u8 mdev = rta_getattr_u8(tb[FRA_L3MDEV]);
 345
 346                if (mdev)
 347                        print_null(PRINT_ANY, "l3mdev",
 348                                   " lookup [l3mdev-table]", NULL);
 349        }
 350
 351        if (tb[FRA_UID_RANGE]) {
 352                struct fib_rule_uid_range *r = RTA_DATA(tb[FRA_UID_RANGE]);
 353
 354                print_uint(PRINT_ANY, "uid_start", " uidrange %u", r->start);
 355                print_uint(PRINT_ANY, "uid_end", "-%u", r->end);
 356        }
 357
 358        if (tb[FRA_IP_PROTO]) {
 359                SPRINT_BUF(pbuf);
 360                print_string(PRINT_ANY, "ipproto", " ipproto %s",
 361                             inet_proto_n2a(rta_getattr_u8(tb[FRA_IP_PROTO]),
 362                                            pbuf, sizeof(pbuf)));
 363        }
 364
 365        if (tb[FRA_SPORT_RANGE]) {
 366                struct fib_rule_port_range *r = RTA_DATA(tb[FRA_SPORT_RANGE]);
 367
 368                if (r->start == r->end) {
 369                        print_uint(PRINT_ANY, "sport", " sport %u", r->start);
 370                } else {
 371                        print_uint(PRINT_ANY, "sport_start", " sport %u",
 372                                   r->start);
 373                        print_uint(PRINT_ANY, "sport_end", "-%u", r->end);
 374                }
 375        }
 376
 377        if (tb[FRA_DPORT_RANGE]) {
 378                struct fib_rule_port_range *r = RTA_DATA(tb[FRA_DPORT_RANGE]);
 379
 380                if (r->start == r->end) {
 381                        print_uint(PRINT_ANY, "dport", " dport %u", r->start);
 382                } else {
 383                        print_uint(PRINT_ANY, "dport_start", " dport %u",
 384                                   r->start);
 385                        print_uint(PRINT_ANY, "dport_end", "-%u", r->end);
 386                }
 387        }
 388
 389        if (tb[FRA_TUN_ID]) {
 390                __u64 tun_id = ntohll(rta_getattr_u64(tb[FRA_TUN_ID]));
 391
 392                print_u64(PRINT_ANY, "tun_id", " tun_id %llu", tun_id);
 393        }
 394
 395        table = frh_get_table(frh, tb);
 396        if (table) {
 397                print_string(PRINT_ANY, "table",
 398                             " lookup %s",
 399                             rtnl_rttable_n2a(table, b1, sizeof(b1)));
 400
 401                if (tb[FRA_SUPPRESS_PREFIXLEN]) {
 402                        int pl = rta_getattr_u32(tb[FRA_SUPPRESS_PREFIXLEN]);
 403
 404                        if (pl != -1)
 405                                print_int(PRINT_ANY, "suppress_prefixlen",
 406                                          " suppress_prefixlength %d", pl);
 407                }
 408
 409                if (tb[FRA_SUPPRESS_IFGROUP]) {
 410                        int group = rta_getattr_u32(tb[FRA_SUPPRESS_IFGROUP]);
 411
 412                        if (group != -1) {
 413                                const char *grname
 414                                        = rtnl_group_n2a(group, b1, sizeof(b1));
 415
 416                                print_string(PRINT_ANY, "suppress_ifgroup",
 417                                             " suppress_ifgroup %s", grname);
 418                        }
 419                }
 420        }
 421
 422        if (tb[FRA_FLOW]) {
 423                __u32 to = rta_getattr_u32(tb[FRA_FLOW]);
 424                __u32 from = to>>16;
 425
 426                to &= 0xFFFF;
 427                if (from)
 428                        print_string(PRINT_ANY,
 429                                     "flow_from", " realms %s/",
 430                                     rtnl_rtrealm_n2a(from, b1, sizeof(b1)));
 431                else
 432                        print_string(PRINT_FP, NULL, " realms ", NULL);
 433
 434                print_string(PRINT_ANY, "flow_to", "%s",
 435                             rtnl_rtrealm_n2a(to, b1, sizeof(b1)));
 436        }
 437
 438        if (frh->action == RTN_NAT) {
 439                if (tb[RTA_GATEWAY]) {
 440                        const char *gateway;
 441
 442                        gateway = format_host_rta(frh->family, tb[RTA_GATEWAY]);
 443
 444                        print_string(PRINT_ANY, "nat_gateway",
 445                                     " map-to %s", gateway);
 446                } else {
 447                        print_null(PRINT_ANY, "masquerade", " masquerade", NULL);
 448                }
 449        } else if (frh->action == FR_ACT_GOTO) {
 450                if (tb[FRA_GOTO])
 451                        print_uint(PRINT_ANY, "goto", " goto %u",
 452                                   rta_getattr_u32(tb[FRA_GOTO]));
 453                else
 454                        print_string(PRINT_ANY, "goto", " goto %s", "none");
 455
 456                if (frh->flags & FIB_RULE_UNRESOLVED)
 457                        print_null(PRINT_ANY, "unresolved",
 458                                   " [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
 480static __u32 rule_dump_magic = 0x71706986;
 481
 482static 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
 500static 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
 513static 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
 554static 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
 728static 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
 748static 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
 765static 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
 773static 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
1024int 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
1050int 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}
1072