ip/xfrm_policy.c

   1 /* $USAGI: $ */
   2
   3 /*
   4  * Copyright (C)2004 USAGI/WIDE Project
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20 /*
  21  * based on iproute.c
  22  */
  23 /*
  24  * Authors:
  25  *      Masahide NAKAMURA @USAGI
  26  */
  27
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30 #include <string.h>
  31 #include <netdb.h>
  32 #include <linux/netlink.h>
  33 #include <linux/xfrm.h>
  34 #include "utils.h"
  35 #include "xfrm.h"
  36 #include "ip_common.h"
  37
  38 //#define NLMSG_FLUSH_BUF_SIZE (4096-512)
  39 #define NLMSG_FLUSH_BUF_SIZE 8192
  40
  41 /*
  42  * Receiving buffer defines:
  43  * nlmsg
  44  *   data = struct xfrm_userpolicy_info
  45  *   rtattr
  46  *     data = struct xfrm_user_tmpl[]
  47  */
  48 #define NLMSG_BUF_SIZE 4096
  49 #define RTA_BUF_SIZE 2048
  50 #define XFRM_TMPLS_BUF_SIZE 1024
  51
  52 static void usage(void) __attribute__((noreturn));
  53
  54 static void usage(void)
  55 {
  56         fprintf(stderr, "Usage: ip xfrm policy { add | update } dir DIR SELECTOR [ index INDEX ] \n");
  57         fprintf(stderr, "        [ action ACTION ] [ priority PRIORITY ] [ LIMIT-LIST ] [ TMPL-LIST ]\n");
  58         fprintf(stderr, "Usage: ip xfrm policy { delete | get } dir DIR [ SELECTOR | index INDEX ]\n");
  59         fprintf(stderr, "Usage: ip xfrm policy { flush | list } [ dir DIR ] [ SELECTOR ]\n");
  60         fprintf(stderr, "        [ index INDEX ] [ action ACTION ] [ priority PRIORITY ]\n");
  61         fprintf(stderr, "DIR := [ in | out | fwd ]\n");
  62
  63         fprintf(stderr, "SELECTOR := src ADDR[/PLEN] dst ADDR[/PLEN] [ UPSPEC ] [ dev DEV ]\n");
  64
  65         fprintf(stderr, "UPSPEC := proto PROTO [ [ sport PORT ] [ dport PORT ] |\n");
  66         fprintf(stderr, "                        [ type NUMBER ] [ code NUMBER ] ]\n");
  67
  68         //fprintf(stderr, "DEV - device name(default=none)\n");
  69
  70         fprintf(stderr, "ACTION := [ allow | block ](default=allow)\n");
  71
  72         //fprintf(stderr, "PRIORITY - priority value(default=0)\n");
  73
  74         fprintf(stderr, "LIMIT-LIST := [ LIMIT-LIST ] | [ limit LIMIT ]\n");
  75         fprintf(stderr, "LIMIT := [ [time-soft|time-hard|time-use-soft|time-use-hard] SECONDS ] |\n");
  76         fprintf(stderr, "         [ [byte-soft|byte-hard] SIZE ] | [ [packet-soft|packet-hard] NUMBER ]\n");
  77
  78         fprintf(stderr, "TMPL-LIST := [ TMPL-LIST ] | [ tmpl TMPL ]\n");
  79         fprintf(stderr, "TMPL := ID [ mode MODE ] [ reqid REQID ] [ level LEVEL ]\n");
  80         fprintf(stderr, "ID := [ src ADDR ] [ dst ADDR ] [ proto XFRM_PROTO ] [ spi SPI ]\n");
  81
  82         //fprintf(stderr, "XFRM_PROTO := [ esp | ah | comp ]\n");
  83         fprintf(stderr, "XFRM_PROTO := [ ");
  84         fprintf(stderr, "%s | ", strxf_xfrmproto(IPPROTO_ESP));
  85         fprintf(stderr, "%s | ", strxf_xfrmproto(IPPROTO_AH));
  86         fprintf(stderr, "%s", strxf_xfrmproto(IPPROTO_COMP));
  87         fprintf(stderr, " ]\n");
  88
  89         fprintf(stderr, "MODE := [ transport | tunnel ](default=transport)\n");
  90         //fprintf(stderr, "REQID - number(default=0)\n");
  91         fprintf(stderr, "LEVEL := [ required | use ](default=required)\n");
  92
  93         exit(-1);
  94 }
  95
  96 static int xfrm_policy_dir_parse(__u8 *dir, int *argcp, char ***argvp)
  97 {
  98         int argc = *argcp;
  99         char **argv = *argvp;
 100
 101         if (strcmp(*argv, "in") == 0)
 102                 *dir = XFRM_POLICY_IN;
 103         else if (strcmp(*argv, "out") == 0)
 104                 *dir = XFRM_POLICY_OUT;
 105         else if (strcmp(*argv, "fwd") == 0)
 106                 *dir = XFRM_POLICY_FWD;
 107         else
 108                 invarg("\"DIR\" is invalid", *argv);
 109
 110         *argcp = argc;
 111         *argvp = argv;
 112
 113         return 0;
 114 }
 115
 116 static int xfrm_tmpl_parse(struct xfrm_user_tmpl *tmpl,
 117                            int *argcp, char ***argvp)
 118 {
 119         int argc = *argcp;
 120         char **argv = *argvp;
 121         char *idp = NULL;
 122
 123         while (1) {
 124                 if (strcmp(*argv, "mode") == 0) {
 125                         NEXT_ARG();
 126                         xfrm_mode_parse(&tmpl->mode,  &argc, &argv);
 127                 } else if (strcmp(*argv, "reqid") == 0) {
 128                         NEXT_ARG();
 129                         xfrm_reqid_parse(&tmpl->reqid, &argc, &argv);
 130                 } else if (strcmp(*argv, "level") == 0) {
 131                         NEXT_ARG();
 132
 133                         if (strcmp(*argv, "required") == 0)
 134                                 tmpl->optional = 0;
 135                         else if (strcmp(*argv, "use") == 0)
 136                                 tmpl->optional = 1;
 137                         else
 138                                 invarg("\"LEVEL\" is invalid\n", *argv);
 139
 140                 } else {
 141                         if (idp) {
 142                                 PREV_ARG(); /* back track */
 143                                 break;
 144                         }
 145                         idp = *argv;
 146                         xfrm_id_parse(&tmpl->saddr, &tmpl->id, &tmpl->family,
 147                                       0, &argc, &argv);
 148                         if (preferred_family == AF_UNSPEC)
 149                                 preferred_family = tmpl->family;
 150                 }
 151
 152                 if (!NEXT_ARG_OK())
 153                         break;
 154
 155                 NEXT_ARG();
 156         }
 157         if (argc == *argcp)
 158                 missarg("TMPL");
 159
 160         *argcp = argc;
 161         *argvp = argv;
 162
 163         return 0;
 164 }
 165
 166 static int xfrm_policy_modify(int cmd, unsigned flags, int argc, char **argv)
 167 {
 168         struct rtnl_handle rth;
 169         struct {
 170                 struct nlmsghdr                 n;
 171                 struct xfrm_userpolicy_info     xpinfo;
 172                 char                            buf[RTA_BUF_SIZE];
 173         } req;
 174         char *dirp = NULL;
 175         char *selp = NULL;
 176         char tmpls_buf[XFRM_TMPLS_BUF_SIZE];
 177         int tmpls_len = 0;
 178
 179         memset(&req, 0, sizeof(req));
 180         memset(&tmpls_buf, 0, sizeof(tmpls_buf));
 181
 182         req.n.nlmsg_len = NLMSG_LENGTH(sizeof(req.xpinfo));
 183         req.n.nlmsg_flags = NLM_F_REQUEST|flags;
 184         req.n.nlmsg_type = cmd;
 185         req.xpinfo.sel.family = preferred_family;
 186
 187         req.xpinfo.lft.soft_byte_limit = XFRM_INF;
 188         req.xpinfo.lft.hard_byte_limit = XFRM_INF;
 189         req.xpinfo.lft.soft_packet_limit = XFRM_INF;
 190         req.xpinfo.lft.hard_packet_limit = XFRM_INF;
 191
 192         while (argc > 0) {
 193                 if (strcmp(*argv, "dir") == 0) {
 194                         if (dirp)
 195                                 duparg("dir", *argv);
 196                         dirp = *argv;
 197
 198                         NEXT_ARG();
 199                         xfrm_policy_dir_parse(&req.xpinfo.dir, &argc, &argv);
 200
 201                         filter.dir_mask = XFRM_FILTER_MASK_FULL;
 202
 203                 } else if (strcmp(*argv, "index") == 0) {
 204                         NEXT_ARG();
 205                         if (get_u32(&req.xpinfo.index, *argv, 0))
 206                                 invarg("\"INDEX\" is invalid", *argv);
 207
 208                         filter.index_mask = XFRM_FILTER_MASK_FULL;
 209
 210                 } else if (strcmp(*argv, "action") == 0) {
 211                         NEXT_ARG();
 212                         if (strcmp(*argv, "allow") == 0)
 213                                 req.xpinfo.action = XFRM_POLICY_ALLOW;
 214                         else if (strcmp(*argv, "block") == 0)
 215                                 req.xpinfo.action = XFRM_POLICY_BLOCK;
 216                         else
 217                                 invarg("\"action\" value is invalid\n", *argv);
 218
 219                         filter.action_mask = XFRM_FILTER_MASK_FULL;
 220
 221                 } else if (strcmp(*argv, "priority") == 0) {
 222                         NEXT_ARG();
 223                         if (get_u32(&req.xpinfo.priority, *argv, 0))
 224                                 invarg("\"PRIORITY\" is invalid", *argv);
 225
 226                         filter.priority_mask = XFRM_FILTER_MASK_FULL;
 227
 228                 } else if (strcmp(*argv, "limit") == 0) {
 229                         NEXT_ARG();
 230                         xfrm_lifetime_cfg_parse(&req.xpinfo.lft, &argc, &argv);
 231                 } else if (strcmp(*argv, "tmpl") == 0) {
 232                         struct xfrm_user_tmpl *tmpl;
 233
 234                         if (tmpls_len + sizeof(*tmpl) > sizeof(tmpls_buf)) {
 235                                 fprintf(stderr, "Too many tmpls: buffer overflow\n");
 236                                 exit(1);
 237                         }
 238                         tmpl = (struct xfrm_user_tmpl *)((char *)tmpls_buf + tmpls_len);
 239
 240                         tmpl->family = preferred_family;
 241                         tmpl->aalgos = (~(__u32)0);
 242                         tmpl->ealgos = (~(__u32)0);
 243                         tmpl->calgos = (~(__u32)0);
 244
 245                         NEXT_ARG();
 246                         xfrm_tmpl_parse(tmpl, &argc, &argv);
 247
 248                         tmpls_len += sizeof(*tmpl);
 249                 } else {
 250                         if (selp)
 251                                 duparg("unknown", *argv);
 252                         selp = *argv;
 253
 254                         xfrm_selector_parse(&req.xpinfo.sel, &argc, &argv);
 255                         if (preferred_family == AF_UNSPEC)
 256                                 preferred_family = req.xpinfo.sel.family;
 257                 }
 258
 259                 argc--; argv++;
 260         }
 261
 262         if (!dirp) {
 263                 fprintf(stderr, "Not enough information: \"DIR\" is required.\n");
 264                 exit(1);
 265         }
 266
 267         if (tmpls_len > 0) {
 268                 addattr_l(&req.n, sizeof(req), XFRMA_TMPL,
 269                           (void *)tmpls_buf, tmpls_len);
 270         }
 271
 272         if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
 273                 exit(1);
 274
 275         if (req.xpinfo.sel.family == AF_UNSPEC)
 276                 req.xpinfo.sel.family = AF_INET;
 277
 278         if (rtnl_talk(&rth, &req.n, 0, 0, NULL, NULL, NULL) < 0)
 279                 exit(2);
 280
 281         rtnl_close(&rth);
 282
 283         return 0;
 284 }
 285
 286 static int xfrm_policy_filter_match(struct xfrm_userpolicy_info *xpinfo)
 287 {
 288         if (!filter.use)
 289                 return 1;
 290
 291         if ((xpinfo->dir^filter.xpinfo.dir)&filter.dir_mask)
 292                 return 0;
 293
 294         if (filter.sel_src_mask) {
 295                 if (memcmp(&xpinfo->sel.saddr, &filter.xpinfo.sel.saddr,
 296                            filter.sel_src_mask) != 0)
 297                         return 0;
 298                 if (xpinfo->sel.prefixlen_s != filter.xpinfo.sel.prefixlen_s)
 299                         return 0;
 300         }
 301
 302         if (filter.sel_dst_mask) {
 303                 if (memcmp(&xpinfo->sel.daddr, &filter.xpinfo.sel.daddr,
 304                            filter.sel_dst_mask) != 0)
 305                         return 0;
 306                 if (xpinfo->sel.prefixlen_d != filter.xpinfo.sel.prefixlen_d)
 307                         return 0;
 308         }
 309
 310         if ((xpinfo->sel.ifindex^filter.xpinfo.sel.ifindex)&filter.sel_dev_mask)
 311                 return 0;
 312
 313         if ((xpinfo->sel.proto^filter.xpinfo.sel.proto)&filter.upspec_proto_mask)
 314                 return 0;
 315
 316         if (filter.upspec_sport_mask) {
 317                 if ((xpinfo->sel.sport^filter.xpinfo.sel.sport)&filter.upspec_sport_mask)
 318                         return 0;
 319         }
 320
 321         if (filter.upspec_dport_mask) {
 322                 if ((xpinfo->sel.dport^filter.xpinfo.sel.dport)&filter.upspec_dport_mask)
 323                         return 0;
 324         }
 325
 326         if ((xpinfo->index^filter.xpinfo.index)&filter.index_mask)
 327                 return 0;
 328
 329         if ((xpinfo->action^filter.xpinfo.action)&filter.action_mask)
 330                 return 0;
 331
 332         if ((xpinfo->priority^filter.xpinfo.priority)&filter.priority_mask)
 333                 return 0;
 334
 335         return 1;
 336 }
 337
 338 static int xfrm_policy_print(const struct sockaddr_nl *who,
 339                              struct nlmsghdr *n, void *arg)
 340 {
 341         FILE *fp = (FILE*)arg;
 342         struct xfrm_userpolicy_info *xpinfo = NLMSG_DATA(n);
 343         int len = n->nlmsg_len;
 344         struct rtattr * tb[XFRM_MAX_DEPTH];
 345         int ntb;
 346
 347         if (n->nlmsg_type != XFRM_MSG_NEWPOLICY &&
 348             n->nlmsg_type != XFRM_MSG_DELPOLICY) {
 349                 fprintf(stderr, "Not a policy: %08x %08x %08x\n",
 350                         n->nlmsg_len, n->nlmsg_type, n->nlmsg_flags);
 351                 return 0;
 352         }
 353
 354         len -= NLMSG_LENGTH(sizeof(*xpinfo));
 355         if (len < 0) {
 356                 fprintf(stderr, "BUG: wrong nlmsg len %d\n", len);
 357                 return -1;
 358         }
 359
 360         if (!xfrm_policy_filter_match(xpinfo))
 361                 return 0;
 362
 363         memset(tb, 0, sizeof(tb));
 364         ntb = parse_rtattr_byindex(tb, XFRM_MAX_DEPTH, XFRMP_RTA(xpinfo), len);
 365
 366         if (n->nlmsg_type == XFRM_MSG_DELPOLICY)
 367                 fprintf(fp, "Deleted ");
 368
 369         xfrm_selector_print(&xpinfo->sel, preferred_family, fp, NULL);
 370
 371         fprintf(fp, "\t");
 372         fprintf(fp, "dir ");
 373         switch (xpinfo->dir) {
 374         case XFRM_POLICY_IN:
 375                 fprintf(fp, "in");
 376                 break;
 377         case XFRM_POLICY_OUT:
 378                 fprintf(fp, "out");
 379                 break;
 380         case XFRM_POLICY_FWD:
 381                 fprintf(fp, "fwd");
 382                 break;
 383         default:
 384                 fprintf(fp, "%d", xpinfo->dir);
 385                 break;
 386         }
 387         fprintf(fp, " ");
 388
 389         switch (xpinfo->action) {
 390         case XFRM_POLICY_ALLOW:
 391                 if (show_stats > 0)
 392                         fprintf(fp, "action allow ");
 393                 break;
 394         case XFRM_POLICY_BLOCK:
 395                 fprintf(fp, "action block ");
 396                 break;
 397         default:
 398                 fprintf(fp, "action %d ", xpinfo->action);
 399                 break;
 400         }
 401
 402         if (show_stats)
 403                 fprintf(fp, "index %u ", xpinfo->index);
 404         fprintf(fp, "priority %u ", xpinfo->priority);
 405         if (show_stats > 0) {
 406                 fprintf(fp, "share %s ", strxf_share(xpinfo->share));
 407                 fprintf(fp, "flags 0x%s", strxf_flags(xpinfo->flags));
 408         }
 409         fprintf(fp, "%s", _SL_);
 410
 411         if (show_stats > 0)
 412                 xfrm_lifetime_print(&xpinfo->lft, &xpinfo->curlft, fp, "\t");
 413
 414         xfrm_xfrma_print(tb, ntb, xpinfo->sel.family, fp, "\t");
 415
 416         if (oneline)
 417                 fprintf(fp, "\n");
 418
 419         return 0;
 420 }
 421
 422 static int xfrm_policy_get_or_delete(int argc, char **argv, int delete,
 423                                      void *res_nlbuf)
 424 {
 425         struct rtnl_handle rth;
 426         struct {
 427                 struct nlmsghdr                 n;
 428                 struct xfrm_userpolicy_id       xpid;
 429         } req;
 430         char *dirp = NULL;
 431         char *selp = NULL;
 432         char *indexp = NULL;
 433
 434         memset(&req, 0, sizeof(req));
 435
 436         req.n.nlmsg_len = NLMSG_LENGTH(sizeof(req.xpid));
 437         req.n.nlmsg_flags = NLM_F_REQUEST;
 438         req.n.nlmsg_type = delete ? XFRM_MSG_DELPOLICY : XFRM_MSG_GETPOLICY;
 439
 440         while (argc > 0) {
 441                 if (strcmp(*argv, "dir") == 0) {
 442                         if (dirp)
 443                                 duparg("dir", *argv);
 444                         dirp = *argv;
 445
 446                         NEXT_ARG();
 447                         xfrm_policy_dir_parse(&req.xpid.dir, &argc, &argv);
 448
 449                 } else if (strcmp(*argv, "index") == 0) {
 450                         if (indexp)
 451                                 duparg("index", *argv);
 452                         indexp = *argv;
 453
 454                         NEXT_ARG();
 455                         if (get_u32(&req.xpid.index, *argv, 0))
 456                                 invarg("\"INDEX\" is invalid", *argv);
 457
 458                 } else {
 459                         if (selp)
 460                                 invarg("unknown", *argv);
 461                         selp = *argv;
 462
 463                         xfrm_selector_parse(&req.xpid.sel, &argc, &argv);
 464                         if (preferred_family == AF_UNSPEC)
 465                                 preferred_family = req.xpid.sel.family;
 466
 467                 }
 468
 469                 argc--; argv++;
 470         }
 471
 472         if (!dirp) {
 473                 fprintf(stderr, "Not enough information: \"DIR\" is required.\n");
 474                 exit(1);
 475         }
 476         if (!selp && !indexp) {
 477                 fprintf(stderr, "Not enough information: either \"SELECTOR\" or \"INDEX\" is required.\n");
 478                 exit(1);
 479         }
 480         if (selp && indexp)
 481                 duparg2("SELECTOR", "INDEX");
 482
 483         if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
 484                 exit(1);
 485
 486         if (req.xpid.sel.family == AF_UNSPEC)
 487                 req.xpid.sel.family = AF_INET;
 488
 489         if (rtnl_talk(&rth, &req.n, 0, 0, res_nlbuf, NULL, NULL) < 0)
 490                 exit(2);
 491
 492         rtnl_close(&rth);
 493
 494         return 0;
 495 }
 496
 497 static int xfrm_policy_delete(int argc, char **argv)
 498 {
 499         return xfrm_policy_get_or_delete(argc, argv, 1, NULL);
 500 }
 501
 502 static int xfrm_policy_get(int argc, char **argv)
 503 {
 504         char buf[NLMSG_BUF_SIZE];
 505         struct nlmsghdr *n = (struct nlmsghdr *)buf;
 506
 507         memset(buf, 0, sizeof(buf));
 508
 509         xfrm_policy_get_or_delete(argc, argv, 0, n);
 510
 511         if (xfrm_policy_print(NULL, n, (void*)stdout) < 0) {
 512                 fprintf(stderr, "An error :-)\n");
 513                 exit(1);
 514         }
 515
 516         return 0;
 517 }
 518
 519 /*
 520  * With an existing policy of nlmsg, make new nlmsg for deleting the policy
 521  * and store it to buffer.
 522  */
 523 static int xfrm_policy_keep(const struct sockaddr_nl *who,
 524                             struct nlmsghdr *n,
 525                             void *arg)
 526 {
 527         struct xfrm_buffer *xb = (struct xfrm_buffer *)arg;
 528         struct rtnl_handle *rth = xb->rth;
 529         struct xfrm_userpolicy_info *xpinfo = NLMSG_DATA(n);
 530         int len = n->nlmsg_len;
 531         struct nlmsghdr *new_n;
 532         struct xfrm_userpolicy_id *xpid;
 533
 534         if (n->nlmsg_type != XFRM_MSG_NEWPOLICY) {
 535                 fprintf(stderr, "Not a policy: %08x %08x %08x\n",
 536                         n->nlmsg_len, n->nlmsg_type, n->nlmsg_flags);
 537                 return 0;
 538         }
 539
 540         len -= NLMSG_LENGTH(sizeof(*xpinfo));
 541         if (len < 0) {
 542                 fprintf(stderr, "BUG: wrong nlmsg len %d\n", len);
 543                 return -1;
 544         }
 545
 546         if (!xfrm_policy_filter_match(xpinfo))
 547                 return 0;
 548
 549         if (xb->offset > xb->size) {
 550                 fprintf(stderr, "Flush buffer overflow\n");
 551                 return -1;
 552         }
 553
 554         new_n = (struct nlmsghdr *)(xb->buf + xb->offset);
 555         new_n->nlmsg_len = NLMSG_LENGTH(sizeof(*xpid));
 556         new_n->nlmsg_flags = NLM_F_REQUEST;
 557         new_n->nlmsg_type = XFRM_MSG_DELPOLICY;
 558         new_n->nlmsg_seq = ++rth->seq;
 559
 560         xpid = NLMSG_DATA(new_n);
 561         memcpy(&xpid->sel, &xpinfo->sel, sizeof(xpid->sel));
 562         xpid->dir = xpinfo->dir;
 563         xpid->index = xpinfo->index;
 564
 565         xb->offset += new_n->nlmsg_len;
 566         xb->nlmsg_count ++;
 567
 568         return 0;
 569 }
 570
 571 static int xfrm_policy_list_or_flush(int argc, char **argv, int flush)
 572 {
 573         char *selp = NULL;
 574         struct rtnl_handle rth;
 575
 576         if (argc > 0)
 577                 filter.use = 1;
 578         filter.xpinfo.sel.family = preferred_family;
 579
 580         while (argc > 0) {
 581                 if (strcmp(*argv, "dir") == 0) {
 582                         NEXT_ARG();
 583                         xfrm_policy_dir_parse(&filter.xpinfo.dir, &argc, &argv);
 584
 585                         filter.dir_mask = XFRM_FILTER_MASK_FULL;
 586
 587                 } else if (strcmp(*argv, "index") == 0) {
 588                         NEXT_ARG();
 589                         if (get_u32(&filter.xpinfo.index, *argv, 0))
 590                                 invarg("\"INDEX\" is invalid", *argv);
 591
 592                         filter.index_mask = XFRM_FILTER_MASK_FULL;
 593
 594                 } else if (strcmp(*argv, "action") == 0) {
 595                         NEXT_ARG();
 596                         if (strcmp(*argv, "allow") == 0)
 597                                 filter.xpinfo.action = XFRM_POLICY_ALLOW;
 598                         else if (strcmp(*argv, "block") == 0)
 599                                 filter.xpinfo.action = XFRM_POLICY_BLOCK;
 600                         else
 601                                 invarg("\"ACTION\" is invalid\n", *argv);
 602
 603                         filter.action_mask = XFRM_FILTER_MASK_FULL;
 604
 605                 } else if (strcmp(*argv, "priority") == 0) {
 606                         NEXT_ARG();
 607                         if (get_u32(&filter.xpinfo.priority, *argv, 0))
 608                                 invarg("\"PRIORITY\" is invalid", *argv);
 609
 610                         filter.priority_mask = XFRM_FILTER_MASK_FULL;
 611
 612                 } else {
 613                         if (selp)
 614                                 invarg("unknown", *argv);
 615                         selp = *argv;
 616
 617                         xfrm_selector_parse(&filter.xpinfo.sel, &argc, &argv);
 618                         if (preferred_family == AF_UNSPEC)
 619                                 preferred_family = filter.xpinfo.sel.family;
 620
 621                 }
 622
 623                 argc--; argv++;
 624         }
 625
 626         if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
 627                 exit(1);
 628
 629         if (flush) {
 630                 struct xfrm_buffer xb;
 631                 char buf[NLMSG_FLUSH_BUF_SIZE];
 632                 int i;
 633
 634                 xb.buf = buf;
 635                 xb.size = sizeof(buf);
 636                 xb.rth = &rth;
 637
 638                 for (i = 0; ; i++) {
 639                         xb.offset = 0;
 640                         xb.nlmsg_count = 0;
 641
 642                         if (show_stats > 1)
 643                                 fprintf(stderr, "Flush round = %d\n", i);
 644
 645                         if (rtnl_wilddump_request(&rth, preferred_family, XFRM_MSG_GETPOLICY) < 0) {
 646                                 perror("Cannot send dump request");
 647                                 exit(1);
 648                         }
 649
 650                         if (rtnl_dump_filter(&rth, xfrm_policy_keep, &xb, NULL, NULL) < 0) {
 651                                 fprintf(stderr, "Flush terminated\n");
 652                                 exit(1);
 653                         }
 654                         if (xb.nlmsg_count == 0) {
 655                                 if (show_stats > 1)
 656                                         fprintf(stderr, "Flush completed\n");
 657                                 break;
 658                         }
 659
 660                         if (rtnl_send(&rth, xb.buf, xb.offset) < 0) {
 661                                 perror("Failed to send flush request\n");
 662                                 exit(1);
 663                         }
 664                         if (show_stats > 1)
 665                                 fprintf(stderr, "Flushed nlmsg count = %d\n", xb.nlmsg_count);
 666
 667                         xb.offset = 0;
 668                         xb.nlmsg_count = 0;
 669                 }
 670         } else {
 671                 if (rtnl_wilddump_request(&rth, preferred_family, XFRM_MSG_GETPOLICY) < 0) {
 672                         perror("Cannot send dump request");
 673                         exit(1);
 674                 }
 675
 676                 if (rtnl_dump_filter(&rth, xfrm_policy_print, stdout, NULL, NULL) < 0) {
 677                         fprintf(stderr, "Dump terminated\n");
 678                         exit(1);
 679                 }
 680         }
 681
 682         rtnl_close(&rth);
 683
 684         exit(0);
 685 }
 686
 687 static int xfrm_policy_flush_all(void)
 688 {
 689         struct rtnl_handle rth;
 690         struct {
 691                 struct nlmsghdr n;
 692         } req;
 693
 694         memset(&req, 0, sizeof(req));
 695
 696         req.n.nlmsg_len = NLMSG_LENGTH(0); /* nlmsg data is nothing */
 697         req.n.nlmsg_flags = NLM_F_REQUEST;
 698         req.n.nlmsg_type = XFRM_MSG_FLUSHPOLICY;
 699
 700         if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
 701                 exit(1);
 702
 703         if (show_stats > 1)
 704                 fprintf(stderr, "Flush all\n");
 705
 706         if (rtnl_talk(&rth, &req.n, 0, 0, NULL, NULL, NULL) < 0)
 707                 exit(2);
 708
 709         rtnl_close(&rth);
 710
 711         return 0;
 712 }
 713
 714 int do_xfrm_policy(int argc, char **argv)
 715 {
 716         if (argc < 1)
 717                 return xfrm_policy_list_or_flush(0, NULL, 0);
 718
 719         if (matches(*argv, "add") == 0)
 720                 return xfrm_policy_modify(XFRM_MSG_NEWPOLICY, 0,
 721                                           argc-1, argv+1);
 722         if (matches(*argv, "update") == 0)
 723                 return xfrm_policy_modify(XFRM_MSG_UPDPOLICY, 0,
 724                                           argc-1, argv+1);
 725         if (matches(*argv, "delete") == 0 || matches(*argv, "del") == 0)
 726                 return xfrm_policy_delete(argc-1, argv+1);
 727         if (matches(*argv, "list") == 0 || matches(*argv, "show") == 0
 728             || matches(*argv, "lst") == 0)
 729                 return xfrm_policy_list_or_flush(argc-1, argv+1, 0);
 730         if (matches(*argv, "get") == 0)
 731                 return xfrm_policy_get(argc-1, argv+1);
 732         if (matches(*argv, "flush") == 0) {
 733                 if (argc-1 < 1)
 734                         return xfrm_policy_flush_all();
 735                 else
 736                         return xfrm_policy_list_or_flush(argc-1, argv+1, 1);
 737         }
 738         if (matches(*argv, "help") == 0)
 739                 usage();
 740         fprintf(stderr, "Command \"%s\" is unknown, try \"ip xfrm policy help\".\n", *argv);
 741         exit(-1);
 742 }