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, see <http://www.gnu.org/licenses>.
18 */
19 /*
20 * based on iproute.c
21 */
22 /*
23 * Authors:
24 * Masahide NAKAMURA @USAGI
25 */
26
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <netdb.h>
31 #include <linux/netlink.h>
32 #include "utils.h"
33 #include "xfrm.h"
34 #include "ip_common.h"
35
36 //#define NLMSG_DELETEALL_BUF_SIZE (4096-512)
37 #define NLMSG_DELETEALL_BUF_SIZE 8192
38
39 /*
40 * Receiving buffer defines:
41 * nlmsg
42 * data = struct xfrm_userpolicy_info
43 * rtattr
44 * data = struct xfrm_user_tmpl[]
45 */
46 #define NLMSG_BUF_SIZE 4096
47 #define RTA_BUF_SIZE 2048
48 #define XFRM_TMPLS_BUF_SIZE 1024
49 #define CTX_BUF_SIZE 256
50
51 static void usage(void) __attribute__((noreturn));
52
usage(void)53 static void usage(void)
54 {
55 fprintf(stderr, "Usage: ip xfrm policy { add | update } SELECTOR dir DIR [ ctx CTX ]\n");
56 fprintf(stderr, " [ mark MARK [ mask MASK ] ] [ index INDEX ] [ ptype PTYPE ]\n");
57 fprintf(stderr, " [ action ACTION ] [ priority PRIORITY ] [ flag FLAG-LIST ]\n");
58 fprintf(stderr, " [ LIMIT-LIST ] [ TMPL-LIST ]\n");
59 fprintf(stderr, "Usage: ip xfrm policy { delete | get } { SELECTOR | index INDEX } dir DIR\n");
60 fprintf(stderr, " [ ctx CTX ] [ mark MARK [ mask MASK ] ] [ ptype PTYPE ]\n");
61 fprintf(stderr, "Usage: ip xfrm policy { deleteall | list } [ SELECTOR ] [ dir DIR ]\n");
62 fprintf(stderr, " [ index INDEX ] [ ptype PTYPE ] [ action ACTION ] [ priority PRIORITY ]\n");
63 fprintf(stderr, " [ flag FLAG-LIST ]\n");
64 fprintf(stderr, "Usage: ip xfrm policy flush [ ptype PTYPE ]\n");
65 fprintf(stderr, "Usage: ip xfrm policy count\n");
66 fprintf(stderr, "Usage: ip xfrm policy set [ hthresh4 LBITS RBITS ] [ hthresh6 LBITS RBITS ]\n");
67 fprintf(stderr, "SELECTOR := [ src ADDR[/PLEN] ] [ dst ADDR[/PLEN] ] [ dev DEV ] [ UPSPEC ]\n");
68 fprintf(stderr, "UPSPEC := proto { { ");
69 fprintf(stderr, "%s | ", strxf_proto(IPPROTO_TCP));
70 fprintf(stderr, "%s | ", strxf_proto(IPPROTO_UDP));
71 fprintf(stderr, "%s | ", strxf_proto(IPPROTO_SCTP));
72 fprintf(stderr, "%s", strxf_proto(IPPROTO_DCCP));
73 fprintf(stderr, " } [ sport PORT ] [ dport PORT ] |\n");
74 fprintf(stderr, " { ");
75 fprintf(stderr, "%s | ", strxf_proto(IPPROTO_ICMP));
76 fprintf(stderr, "%s | ", strxf_proto(IPPROTO_ICMPV6));
77 fprintf(stderr, "%s", strxf_proto(IPPROTO_MH));
78 fprintf(stderr, " } [ type NUMBER ] [ code NUMBER ] |\n");
79 fprintf(stderr, " %s", strxf_proto(IPPROTO_GRE));
80 fprintf(stderr, " [ key { DOTTED-QUAD | NUMBER } ] | PROTO }\n");
81 fprintf(stderr, "DIR := in | out | fwd\n");
82 fprintf(stderr, "PTYPE := main | sub\n");
83 fprintf(stderr, "ACTION := allow | block\n");
84 fprintf(stderr, "FLAG-LIST := [ FLAG-LIST ] FLAG\n");
85 fprintf(stderr, "FLAG := localok | icmp\n");
86 fprintf(stderr, "LIMIT-LIST := [ LIMIT-LIST ] limit LIMIT\n");
87 fprintf(stderr, "LIMIT := { time-soft | time-hard | time-use-soft | time-use-hard } SECONDS |\n");
88 fprintf(stderr, " { byte-soft | byte-hard } SIZE | { packet-soft | packet-hard } COUNT\n");
89 fprintf(stderr, "TMPL-LIST := [ TMPL-LIST ] tmpl TMPL\n");
90 fprintf(stderr, "TMPL := ID [ mode MODE ] [ reqid REQID ] [ level LEVEL ]\n");
91 fprintf(stderr, "ID := [ src ADDR ] [ dst ADDR ] [ proto XFRM-PROTO ] [ spi SPI ]\n");
92 fprintf(stderr, "XFRM-PROTO := ");
93 fprintf(stderr, "%s | ", strxf_xfrmproto(IPPROTO_ESP));
94 fprintf(stderr, "%s | ", strxf_xfrmproto(IPPROTO_AH));
95 fprintf(stderr, "%s | ", strxf_xfrmproto(IPPROTO_COMP));
96 fprintf(stderr, "%s | ", strxf_xfrmproto(IPPROTO_ROUTING));
97 fprintf(stderr, "%s\n", strxf_xfrmproto(IPPROTO_DSTOPTS));
98 fprintf(stderr, "MODE := transport | tunnel | beet | ro | in_trigger\n");
99 fprintf(stderr, "LEVEL := required | use\n");
100
101 exit(-1);
102 }
103
xfrm_policy_dir_parse(__u8 * dir,int * argcp,char *** argvp)104 static int xfrm_policy_dir_parse(__u8 *dir, int *argcp, char ***argvp)
105 {
106 int argc = *argcp;
107 char **argv = *argvp;
108
109 if (strcmp(*argv, "in") == 0)
110 *dir = XFRM_POLICY_IN;
111 else if (strcmp(*argv, "out") == 0)
112 *dir = XFRM_POLICY_OUT;
113 else if (strcmp(*argv, "fwd") == 0)
114 *dir = XFRM_POLICY_FWD;
115 else
116 invarg("DIR value is invalid", *argv);
117
118 *argcp = argc;
119 *argvp = argv;
120
121 return 0;
122 }
123
xfrm_policy_ptype_parse(__u8 * ptype,int * argcp,char *** argvp)124 static int xfrm_policy_ptype_parse(__u8 *ptype, int *argcp, char ***argvp)
125 {
126 int argc = *argcp;
127 char **argv = *argvp;
128
129 if (strcmp(*argv, "main") == 0)
130 *ptype = XFRM_POLICY_TYPE_MAIN;
131 else if (strcmp(*argv, "sub") == 0)
132 *ptype = XFRM_POLICY_TYPE_SUB;
133 else
134 invarg("PTYPE value is invalid", *argv);
135
136 *argcp = argc;
137 *argvp = argv;
138
139 return 0;
140 }
141
xfrm_policy_flag_parse(__u8 * flags,int * argcp,char *** argvp)142 static int xfrm_policy_flag_parse(__u8 *flags, int *argcp, char ***argvp)
143 {
144 int argc = *argcp;
145 char **argv = *argvp;
146 int len = strlen(*argv);
147
148 if (len > 2 && strncmp(*argv, "0x", 2) == 0) {
149 __u8 val = 0;
150
151 if (get_u8(&val, *argv, 16))
152 invarg("FLAG value is invalid", *argv);
153 *flags = val;
154 } else {
155 while (1) {
156 if (strcmp(*argv, "localok") == 0)
157 *flags |= XFRM_POLICY_LOCALOK;
158 else if (strcmp(*argv, "icmp") == 0)
159 *flags |= XFRM_POLICY_ICMP;
160 else {
161 PREV_ARG(); /* back track */
162 break;
163 }
164
165 if (!NEXT_ARG_OK())
166 break;
167 NEXT_ARG();
168 }
169 }
170
171 *argcp = argc;
172 *argvp = argv;
173
174 return 0;
175 }
176
xfrm_tmpl_parse(struct xfrm_user_tmpl * tmpl,int * argcp,char *** argvp)177 static int xfrm_tmpl_parse(struct xfrm_user_tmpl *tmpl,
178 int *argcp, char ***argvp)
179 {
180 int argc = *argcp;
181 char **argv = *argvp;
182 char *idp = NULL;
183
184 while (1) {
185 if (strcmp(*argv, "mode") == 0) {
186 NEXT_ARG();
187 xfrm_mode_parse(&tmpl->mode, &argc, &argv);
188 } else if (strcmp(*argv, "reqid") == 0) {
189 NEXT_ARG();
190 xfrm_reqid_parse(&tmpl->reqid, &argc, &argv);
191 } else if (strcmp(*argv, "level") == 0) {
192 NEXT_ARG();
193
194 if (strcmp(*argv, "required") == 0)
195 tmpl->optional = 0;
196 else if (strcmp(*argv, "use") == 0)
197 tmpl->optional = 1;
198 else
199 invarg("LEVEL value is invalid\n", *argv);
200
201 } else {
202 if (idp) {
203 PREV_ARG(); /* back track */
204 break;
205 }
206 idp = *argv;
207 preferred_family = AF_UNSPEC;
208 xfrm_id_parse(&tmpl->saddr, &tmpl->id, &tmpl->family,
209 0, &argc, &argv);
210 preferred_family = tmpl->family;
211 }
212
213 if (!NEXT_ARG_OK())
214 break;
215
216 NEXT_ARG();
217 }
218 if (argc == *argcp)
219 missarg("TMPL");
220
221 *argcp = argc;
222 *argvp = argv;
223
224 return 0;
225 }
226
xfrm_sctx_parse(char * ctxstr,char * s,struct xfrm_user_sec_ctx * sctx)227 int xfrm_sctx_parse(char *ctxstr, char *s,
228 struct xfrm_user_sec_ctx *sctx)
229 {
230 int slen;
231
232 slen = strlen(s) + 1;
233
234 sctx->exttype = XFRMA_SEC_CTX;
235 sctx->ctx_doi = 1;
236 sctx->ctx_alg = 1;
237 sctx->ctx_len = slen;
238 sctx->len = sizeof(struct xfrm_user_sec_ctx) + slen;
239 memcpy(ctxstr, s, slen);
240
241 return 0;
242 }
243
xfrm_policy_modify(int cmd,unsigned flags,int argc,char ** argv)244 static int xfrm_policy_modify(int cmd, unsigned flags, int argc, char **argv)
245 {
246 struct rtnl_handle rth;
247 struct {
248 struct nlmsghdr n;
249 struct xfrm_userpolicy_info xpinfo;
250 char buf[RTA_BUF_SIZE];
251 } req;
252 char *dirp = NULL;
253 char *selp = NULL;
254 char *ptypep = NULL;
255 char *sctxp = NULL;
256 struct xfrm_userpolicy_type upt;
257 char tmpls_buf[XFRM_TMPLS_BUF_SIZE];
258 int tmpls_len = 0;
259 struct xfrm_mark mark = {0, 0};
260 struct {
261 struct xfrm_user_sec_ctx sctx;
262 char str[CTX_BUF_SIZE];
263 } ctx;
264
265 memset(&req, 0, sizeof(req));
266 memset(&upt, 0, sizeof(upt));
267 memset(&tmpls_buf, 0, sizeof(tmpls_buf));
268 memset(&ctx, 0, sizeof(ctx));
269
270 req.n.nlmsg_len = NLMSG_LENGTH(sizeof(req.xpinfo));
271 req.n.nlmsg_flags = NLM_F_REQUEST|flags;
272 req.n.nlmsg_type = cmd;
273 req.xpinfo.sel.family = preferred_family;
274
275 req.xpinfo.lft.soft_byte_limit = XFRM_INF;
276 req.xpinfo.lft.hard_byte_limit = XFRM_INF;
277 req.xpinfo.lft.soft_packet_limit = XFRM_INF;
278 req.xpinfo.lft.hard_packet_limit = XFRM_INF;
279
280 while (argc > 0) {
281 if (strcmp(*argv, "dir") == 0) {
282 if (dirp)
283 duparg("dir", *argv);
284 dirp = *argv;
285
286 NEXT_ARG();
287 xfrm_policy_dir_parse(&req.xpinfo.dir, &argc, &argv);
288 } else if (strcmp(*argv, "ctx") == 0) {
289 char *context;
290
291 if (sctxp)
292 duparg("ctx", *argv);
293 sctxp = *argv;
294 NEXT_ARG();
295 context = *argv;
296 xfrm_sctx_parse((char *)&ctx.str, context, &ctx.sctx);
297 } else if (strcmp(*argv, "mark") == 0) {
298 xfrm_parse_mark(&mark, &argc, &argv);
299 } else if (strcmp(*argv, "index") == 0) {
300 NEXT_ARG();
301 if (get_u32(&req.xpinfo.index, *argv, 0))
302 invarg("INDEX value is invalid", *argv);
303 } else if (strcmp(*argv, "ptype") == 0) {
304 if (ptypep)
305 duparg("ptype", *argv);
306 ptypep = *argv;
307
308 NEXT_ARG();
309 xfrm_policy_ptype_parse(&upt.type, &argc, &argv);
310 } else if (strcmp(*argv, "action") == 0) {
311 NEXT_ARG();
312 if (strcmp(*argv, "allow") == 0)
313 req.xpinfo.action = XFRM_POLICY_ALLOW;
314 else if (strcmp(*argv, "block") == 0)
315 req.xpinfo.action = XFRM_POLICY_BLOCK;
316 else
317 invarg("ACTION value is invalid\n", *argv);
318 } else if (strcmp(*argv, "priority") == 0) {
319 NEXT_ARG();
320 if (get_u32(&req.xpinfo.priority, *argv, 0))
321 invarg("PRIORITY value is invalid", *argv);
322 } else if (strcmp(*argv, "flag") == 0) {
323 NEXT_ARG();
324 xfrm_policy_flag_parse(&req.xpinfo.flags, &argc,
325 &argv);
326 } else if (strcmp(*argv, "limit") == 0) {
327 NEXT_ARG();
328 xfrm_lifetime_cfg_parse(&req.xpinfo.lft, &argc, &argv);
329 } else if (strcmp(*argv, "tmpl") == 0) {
330 struct xfrm_user_tmpl *tmpl;
331
332 if (tmpls_len + sizeof(*tmpl) > sizeof(tmpls_buf)) {
333 fprintf(stderr, "Too many tmpls: buffer overflow\n");
334 exit(1);
335 }
336 tmpl = (struct xfrm_user_tmpl *)((char *)tmpls_buf + tmpls_len);
337
338 tmpl->family = preferred_family;
339 tmpl->aalgos = (~(__u32)0);
340 tmpl->ealgos = (~(__u32)0);
341 tmpl->calgos = (~(__u32)0);
342
343 NEXT_ARG();
344 xfrm_tmpl_parse(tmpl, &argc, &argv);
345
346 tmpls_len += sizeof(*tmpl);
347 } else {
348 if (selp)
349 duparg("unknown", *argv);
350 selp = *argv;
351
352 xfrm_selector_parse(&req.xpinfo.sel, &argc, &argv);
353 if (preferred_family == AF_UNSPEC)
354 preferred_family = req.xpinfo.sel.family;
355 }
356
357 argc--; argv++;
358 }
359
360 if (!dirp) {
361 fprintf(stderr, "Not enough information: DIR is required.\n");
362 exit(1);
363 }
364
365 if (ptypep) {
366 addattr_l(&req.n, sizeof(req), XFRMA_POLICY_TYPE,
367 (void *)&upt, sizeof(upt));
368 }
369
370 if (tmpls_len > 0) {
371 addattr_l(&req.n, sizeof(req), XFRMA_TMPL,
372 (void *)tmpls_buf, tmpls_len);
373 }
374
375 if (mark.m) {
376 int r = addattr_l(&req.n, sizeof(req.buf), XFRMA_MARK,
377 (void *)&mark, sizeof(mark));
378 if (r < 0) {
379 fprintf(stderr, "%s: XFRMA_MARK failed\n",__func__);
380 exit(1);
381 }
382 }
383
384 if (sctxp) {
385 addattr_l(&req.n, sizeof(req), XFRMA_SEC_CTX,
386 (void *)&ctx, ctx.sctx.len);
387 }
388
389 if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
390 exit(1);
391
392 if (req.xpinfo.sel.family == AF_UNSPEC)
393 req.xpinfo.sel.family = AF_INET;
394
395 if (rtnl_talk(&rth, &req.n, NULL, 0) < 0)
396 exit(2);
397
398 rtnl_close(&rth);
399
400 return 0;
401 }
402
xfrm_policy_filter_match(struct xfrm_userpolicy_info * xpinfo,__u8 ptype)403 static int xfrm_policy_filter_match(struct xfrm_userpolicy_info *xpinfo,
404 __u8 ptype)
405 {
406 if (!filter.use)
407 return 1;
408
409 if ((xpinfo->dir^filter.xpinfo.dir)&filter.dir_mask)
410 return 0;
411
412 if ((ptype^filter.ptype)&filter.ptype_mask)
413 return 0;
414
415 if (filter.sel_src_mask) {
416 if (xfrm_addr_match(&xpinfo->sel.saddr, &filter.xpinfo.sel.saddr,
417 filter.sel_src_mask))
418 return 0;
419 }
420
421 if (filter.sel_dst_mask) {
422 if (xfrm_addr_match(&xpinfo->sel.daddr, &filter.xpinfo.sel.daddr,
423 filter.sel_dst_mask))
424 return 0;
425 }
426
427 if ((xpinfo->sel.ifindex^filter.xpinfo.sel.ifindex)&filter.sel_dev_mask)
428 return 0;
429
430 if ((xpinfo->sel.proto^filter.xpinfo.sel.proto)&filter.upspec_proto_mask)
431 return 0;
432
433 if (filter.upspec_sport_mask) {
434 if ((xpinfo->sel.sport^filter.xpinfo.sel.sport)&filter.upspec_sport_mask)
435 return 0;
436 }
437
438 if (filter.upspec_dport_mask) {
439 if ((xpinfo->sel.dport^filter.xpinfo.sel.dport)&filter.upspec_dport_mask)
440 return 0;
441 }
442
443 if ((xpinfo->index^filter.xpinfo.index)&filter.index_mask)
444 return 0;
445
446 if ((xpinfo->action^filter.xpinfo.action)&filter.action_mask)
447 return 0;
448
449 if ((xpinfo->priority^filter.xpinfo.priority)&filter.priority_mask)
450 return 0;
451
452 if (filter.policy_flags_mask)
453 if ((xpinfo->flags & filter.xpinfo.flags) == 0)
454 return 0;
455
456 return 1;
457 }
458
xfrm_policy_print(const struct sockaddr_nl * who,struct nlmsghdr * n,void * arg)459 int xfrm_policy_print(const struct sockaddr_nl *who, struct nlmsghdr *n,
460 void *arg)
461 {
462 struct rtattr * tb[XFRMA_MAX+1];
463 struct rtattr * rta;
464 struct xfrm_userpolicy_info *xpinfo = NULL;
465 struct xfrm_user_polexpire *xpexp = NULL;
466 struct xfrm_userpolicy_id *xpid = NULL;
467 __u8 ptype = XFRM_POLICY_TYPE_MAIN;
468 FILE *fp = (FILE*)arg;
469 int len = n->nlmsg_len;
470
471 if (n->nlmsg_type != XFRM_MSG_NEWPOLICY &&
472 n->nlmsg_type != XFRM_MSG_DELPOLICY &&
473 n->nlmsg_type != XFRM_MSG_UPDPOLICY &&
474 n->nlmsg_type != XFRM_MSG_POLEXPIRE) {
475 fprintf(stderr, "Not a policy: %08x %08x %08x\n",
476 n->nlmsg_len, n->nlmsg_type, n->nlmsg_flags);
477 return 0;
478 }
479
480 if (n->nlmsg_type == XFRM_MSG_DELPOLICY) {
481 xpid = NLMSG_DATA(n);
482 len -= NLMSG_SPACE(sizeof(*xpid));
483 } else if (n->nlmsg_type == XFRM_MSG_POLEXPIRE) {
484 xpexp = NLMSG_DATA(n);
485 xpinfo = &xpexp->pol;
486 len -= NLMSG_SPACE(sizeof(*xpexp));
487 } else {
488 xpexp = NULL;
489 xpinfo = NLMSG_DATA(n);
490 len -= NLMSG_SPACE(sizeof(*xpinfo));
491 }
492
493 if (len < 0) {
494 fprintf(stderr, "BUG: wrong nlmsg len %d\n", len);
495 return -1;
496 }
497
498 if (n->nlmsg_type == XFRM_MSG_DELPOLICY)
499 rta = XFRMPID_RTA(xpid);
500 else if (n->nlmsg_type == XFRM_MSG_POLEXPIRE)
501 rta = XFRMPEXP_RTA(xpexp);
502 else
503 rta = XFRMP_RTA(xpinfo);
504
505 parse_rtattr(tb, XFRMA_MAX, rta, len);
506
507 if (tb[XFRMA_POLICY_TYPE]) {
508 struct xfrm_userpolicy_type *upt;
509
510 if (RTA_PAYLOAD(tb[XFRMA_POLICY_TYPE]) < sizeof(*upt)) {
511 fprintf(stderr, "too short XFRMA_POLICY_TYPE len\n");
512 return -1;
513 }
514 upt = (struct xfrm_userpolicy_type *)RTA_DATA(tb[XFRMA_POLICY_TYPE]);
515 ptype = upt->type;
516 }
517
518 if (xpinfo && !xfrm_policy_filter_match(xpinfo, ptype))
519 return 0;
520
521 if (n->nlmsg_type == XFRM_MSG_DELPOLICY)
522 fprintf(fp, "Deleted ");
523 else if (n->nlmsg_type == XFRM_MSG_UPDPOLICY)
524 fprintf(fp, "Updated ");
525 else if (n->nlmsg_type == XFRM_MSG_POLEXPIRE)
526 fprintf(fp, "Expired ");
527
528 if (n->nlmsg_type == XFRM_MSG_DELPOLICY) {
529 //xfrm_policy_id_print();
530 if (!tb[XFRMA_POLICY]) {
531 fprintf(stderr, "Buggy XFRM_MSG_DELPOLICY: no XFRMA_POLICY\n");
532 return -1;
533 }
534 if (RTA_PAYLOAD(tb[XFRMA_POLICY]) < sizeof(*xpinfo)) {
535 fprintf(stderr, "Buggy XFRM_MSG_DELPOLICY: too short XFRMA_POLICY len\n");
536 return -1;
537 }
538 xpinfo = (struct xfrm_userpolicy_info *)RTA_DATA(tb[XFRMA_POLICY]);
539 }
540
541 xfrm_policy_info_print(xpinfo, tb, fp, NULL, NULL);
542
543 if (n->nlmsg_type == XFRM_MSG_POLEXPIRE) {
544 fprintf(fp, "\t");
545 fprintf(fp, "hard %u", xpexp->hard);
546 fprintf(fp, "%s", _SL_);
547 }
548
549 if (oneline)
550 fprintf(fp, "\n");
551 fflush(fp);
552
553 return 0;
554 }
555
xfrm_policy_get_or_delete(int argc,char ** argv,int delete,void * res_nlbuf,size_t res_size)556 static int xfrm_policy_get_or_delete(int argc, char **argv, int delete,
557 void *res_nlbuf, size_t res_size)
558 {
559 struct rtnl_handle rth;
560 struct {
561 struct nlmsghdr n;
562 struct xfrm_userpolicy_id xpid;
563 char buf[RTA_BUF_SIZE];
564 } req;
565 char *dirp = NULL;
566 char *selp = NULL;
567 char *indexp = NULL;
568 char *ptypep = NULL;
569 char *sctxp = NULL;
570 struct xfrm_userpolicy_type upt;
571 struct xfrm_mark mark = {0, 0};
572 struct {
573 struct xfrm_user_sec_ctx sctx;
574 char str[CTX_BUF_SIZE];
575 } ctx;
576
577
578 memset(&req, 0, sizeof(req));
579 memset(&upt, 0, sizeof(upt));
580 memset(&ctx, 0, sizeof(ctx));
581
582 req.n.nlmsg_len = NLMSG_LENGTH(sizeof(req.xpid));
583 req.n.nlmsg_flags = NLM_F_REQUEST;
584 req.n.nlmsg_type = delete ? XFRM_MSG_DELPOLICY : XFRM_MSG_GETPOLICY;
585
586 while (argc > 0) {
587 if (strcmp(*argv, "dir") == 0) {
588 if (dirp)
589 duparg("dir", *argv);
590 dirp = *argv;
591
592 NEXT_ARG();
593 xfrm_policy_dir_parse(&req.xpid.dir, &argc, &argv);
594
595 } else if (strcmp(*argv, "ctx") == 0) {
596 char *context;
597
598 if (sctxp)
599 duparg("ctx", *argv);
600 sctxp = *argv;
601 NEXT_ARG();
602 context = *argv;
603 xfrm_sctx_parse((char *)&ctx.str, context, &ctx.sctx);
604 } else if (strcmp(*argv, "mark") == 0) {
605 xfrm_parse_mark(&mark, &argc, &argv);
606 } else if (strcmp(*argv, "index") == 0) {
607 if (indexp)
608 duparg("index", *argv);
609 indexp = *argv;
610
611 NEXT_ARG();
612 if (get_u32(&req.xpid.index, *argv, 0))
613 invarg("INDEX value is invalid", *argv);
614
615 } else if (strcmp(*argv, "ptype") == 0) {
616 if (ptypep)
617 duparg("ptype", *argv);
618 ptypep = *argv;
619
620 NEXT_ARG();
621 xfrm_policy_ptype_parse(&upt.type, &argc, &argv);
622
623 } else {
624 if (selp)
625 invarg("unknown", *argv);
626 selp = *argv;
627
628 xfrm_selector_parse(&req.xpid.sel, &argc, &argv);
629 if (preferred_family == AF_UNSPEC)
630 preferred_family = req.xpid.sel.family;
631
632 }
633
634 argc--; argv++;
635 }
636
637 if (!dirp) {
638 fprintf(stderr, "Not enough information: DIR is required.\n");
639 exit(1);
640 }
641 if (ptypep) {
642 addattr_l(&req.n, sizeof(req), XFRMA_POLICY_TYPE,
643 (void *)&upt, sizeof(upt));
644 }
645 if (!selp && !indexp) {
646 fprintf(stderr, "Not enough information: either SELECTOR or INDEX is required.\n");
647 exit(1);
648 }
649 if (selp && indexp)
650 duparg2("SELECTOR", "INDEX");
651
652 if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
653 exit(1);
654
655 if (req.xpid.sel.family == AF_UNSPEC)
656 req.xpid.sel.family = AF_INET;
657
658 if (mark.m & mark.v) {
659 int r = addattr_l(&req.n, sizeof(req.buf), XFRMA_MARK,
660 (void *)&mark, sizeof(mark));
661 if (r < 0) {
662 fprintf(stderr, "%s: XFRMA_MARK failed\n",__func__);
663 exit(1);
664 }
665 }
666
667 if (sctxp) {
668 addattr_l(&req.n, sizeof(req), XFRMA_SEC_CTX,
669 (void *)&ctx, ctx.sctx.len);
670 }
671
672 if (rtnl_talk(&rth, &req.n, res_nlbuf, res_size) < 0)
673 exit(2);
674
675 rtnl_close(&rth);
676
677 return 0;
678 }
679
xfrm_policy_delete(int argc,char ** argv)680 static int xfrm_policy_delete(int argc, char **argv)
681 {
682 return xfrm_policy_get_or_delete(argc, argv, 1, NULL, 0);
683 }
684
xfrm_policy_get(int argc,char ** argv)685 static int xfrm_policy_get(int argc, char **argv)
686 {
687 char buf[NLMSG_BUF_SIZE];
688 struct nlmsghdr *n = (struct nlmsghdr *)buf;
689
690 memset(buf, 0, sizeof(buf));
691
692 xfrm_policy_get_or_delete(argc, argv, 0, n, sizeof(buf));
693
694 if (xfrm_policy_print(NULL, n, (void*)stdout) < 0) {
695 fprintf(stderr, "An error :-)\n");
696 exit(1);
697 }
698
699 return 0;
700 }
701
702 /*
703 * With an existing policy of nlmsg, make new nlmsg for deleting the policy
704 * and store it to buffer.
705 */
xfrm_policy_keep(const struct sockaddr_nl * who,struct nlmsghdr * n,void * arg)706 static int xfrm_policy_keep(const struct sockaddr_nl *who,
707 struct nlmsghdr *n,
708 void *arg)
709 {
710 struct xfrm_buffer *xb = (struct xfrm_buffer *)arg;
711 struct rtnl_handle *rth = xb->rth;
712 struct xfrm_userpolicy_info *xpinfo = NLMSG_DATA(n);
713 int len = n->nlmsg_len;
714 struct rtattr *tb[XFRMA_MAX+1];
715 __u8 ptype = XFRM_POLICY_TYPE_MAIN;
716 struct nlmsghdr *new_n;
717 struct xfrm_userpolicy_id *xpid;
718
719 if (n->nlmsg_type != XFRM_MSG_NEWPOLICY) {
720 fprintf(stderr, "Not a policy: %08x %08x %08x\n",
721 n->nlmsg_len, n->nlmsg_type, n->nlmsg_flags);
722 return 0;
723 }
724
725 len -= NLMSG_LENGTH(sizeof(*xpinfo));
726 if (len < 0) {
727 fprintf(stderr, "BUG: wrong nlmsg len %d\n", len);
728 return -1;
729 }
730
731 parse_rtattr(tb, XFRMA_MAX, XFRMP_RTA(xpinfo), len);
732
733 if (tb[XFRMA_POLICY_TYPE]) {
734 struct xfrm_userpolicy_type *upt;
735
736 if (RTA_PAYLOAD(tb[XFRMA_POLICY_TYPE]) < sizeof(*upt)) {
737 fprintf(stderr, "too short XFRMA_POLICY_TYPE len\n");
738 return -1;
739 }
740 upt = (struct xfrm_userpolicy_type *)RTA_DATA(tb[XFRMA_POLICY_TYPE]);
741 ptype = upt->type;
742 }
743
744 if (!xfrm_policy_filter_match(xpinfo, ptype))
745 return 0;
746
747 if (xb->offset > xb->size) {
748 fprintf(stderr, "Policy buffer overflow\n");
749 return -1;
750 }
751
752 new_n = (struct nlmsghdr *)(xb->buf + xb->offset);
753 new_n->nlmsg_len = NLMSG_LENGTH(sizeof(*xpid));
754 new_n->nlmsg_flags = NLM_F_REQUEST;
755 new_n->nlmsg_type = XFRM_MSG_DELPOLICY;
756 new_n->nlmsg_seq = ++rth->seq;
757
758 xpid = NLMSG_DATA(new_n);
759 memcpy(&xpid->sel, &xpinfo->sel, sizeof(xpid->sel));
760 xpid->dir = xpinfo->dir;
761 xpid->index = xpinfo->index;
762
763 xb->offset += new_n->nlmsg_len;
764 xb->nlmsg_count ++;
765
766 return 0;
767 }
768
xfrm_policy_list_or_deleteall(int argc,char ** argv,int deleteall)769 static int xfrm_policy_list_or_deleteall(int argc, char **argv, int deleteall)
770 {
771 char *selp = NULL;
772 struct rtnl_handle rth;
773
774 if (argc > 0)
775 filter.use = 1;
776 filter.xpinfo.sel.family = preferred_family;
777
778 while (argc > 0) {
779 if (strcmp(*argv, "dir") == 0) {
780 NEXT_ARG();
781 xfrm_policy_dir_parse(&filter.xpinfo.dir, &argc, &argv);
782
783 filter.dir_mask = XFRM_FILTER_MASK_FULL;
784
785 } else if (strcmp(*argv, "index") == 0) {
786 NEXT_ARG();
787 if (get_u32(&filter.xpinfo.index, *argv, 0))
788 invarg("INDEX value is invalid", *argv);
789
790 filter.index_mask = XFRM_FILTER_MASK_FULL;
791
792 } else if (strcmp(*argv, "ptype") == 0) {
793 NEXT_ARG();
794 xfrm_policy_ptype_parse(&filter.ptype, &argc, &argv);
795
796 filter.ptype_mask = XFRM_FILTER_MASK_FULL;
797
798 } else if (strcmp(*argv, "action") == 0) {
799 NEXT_ARG();
800 if (strcmp(*argv, "allow") == 0)
801 filter.xpinfo.action = XFRM_POLICY_ALLOW;
802 else if (strcmp(*argv, "block") == 0)
803 filter.xpinfo.action = XFRM_POLICY_BLOCK;
804 else
805 invarg("ACTION value is invalid\n", *argv);
806
807 filter.action_mask = XFRM_FILTER_MASK_FULL;
808
809 } else if (strcmp(*argv, "priority") == 0) {
810 NEXT_ARG();
811 if (get_u32(&filter.xpinfo.priority, *argv, 0))
812 invarg("PRIORITY value is invalid", *argv);
813
814 filter.priority_mask = XFRM_FILTER_MASK_FULL;
815
816 } else if (strcmp(*argv, "flag") == 0) {
817 NEXT_ARG();
818 xfrm_policy_flag_parse(&filter.xpinfo.flags, &argc,
819 &argv);
820
821 filter.policy_flags_mask = XFRM_FILTER_MASK_FULL;
822
823 } else {
824 if (selp)
825 invarg("unknown", *argv);
826 selp = *argv;
827
828 xfrm_selector_parse(&filter.xpinfo.sel, &argc, &argv);
829 if (preferred_family == AF_UNSPEC)
830 preferred_family = filter.xpinfo.sel.family;
831
832 }
833
834 argc--; argv++;
835 }
836
837 if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
838 exit(1);
839
840 if (deleteall) {
841 struct xfrm_buffer xb;
842 char buf[NLMSG_DELETEALL_BUF_SIZE];
843 int i;
844
845 xb.buf = buf;
846 xb.size = sizeof(buf);
847 xb.rth = &rth;
848
849 for (i = 0; ; i++) {
850 struct {
851 struct nlmsghdr n;
852 char buf[NLMSG_BUF_SIZE];
853 } req = {
854 .n.nlmsg_len = NLMSG_HDRLEN,
855 .n.nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST,
856 .n.nlmsg_type = XFRM_MSG_GETPOLICY,
857 .n.nlmsg_seq = rth.dump = ++rth.seq,
858 };
859
860 xb.offset = 0;
861 xb.nlmsg_count = 0;
862
863 if (show_stats > 1)
864 fprintf(stderr, "Delete-all round = %d\n", i);
865
866 if (rtnl_send(&rth, (void *)&req, req.n.nlmsg_len) < 0) {
867 perror("Cannot send dump request");
868 exit(1);
869 }
870
871 if (rtnl_dump_filter(&rth, xfrm_policy_keep, &xb) < 0) {
872 fprintf(stderr, "Delete-all terminated\n");
873 exit(1);
874 }
875 if (xb.nlmsg_count == 0) {
876 if (show_stats > 1)
877 fprintf(stderr, "Delete-all completed\n");
878 break;
879 }
880
881 if (rtnl_send_check(&rth, xb.buf, xb.offset) < 0) {
882 perror("Failed to send delete-all request");
883 exit(1);
884 }
885 if (show_stats > 1)
886 fprintf(stderr, "Delete-all nlmsg count = %d\n", xb.nlmsg_count);
887
888 xb.offset = 0;
889 xb.nlmsg_count = 0;
890 }
891 } else {
892 struct {
893 struct nlmsghdr n;
894 char buf[NLMSG_BUF_SIZE];
895 } req = {
896 .n.nlmsg_len = NLMSG_HDRLEN,
897 .n.nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST,
898 .n.nlmsg_type = XFRM_MSG_GETPOLICY,
899 .n.nlmsg_seq = rth.dump = ++rth.seq,
900 };
901
902 if (rtnl_send(&rth, (void *)&req, req.n.nlmsg_len) < 0) {
903 perror("Cannot send dump request");
904 exit(1);
905 }
906
907 if (rtnl_dump_filter(&rth, xfrm_policy_print, stdout) < 0) {
908 fprintf(stderr, "Dump terminated\n");
909 exit(1);
910 }
911 }
912
913 rtnl_close(&rth);
914
915 exit(0);
916 }
917
print_spdinfo(struct nlmsghdr * n,void * arg)918 static int print_spdinfo( struct nlmsghdr *n, void *arg)
919 {
920 FILE *fp = (FILE*)arg;
921 __u32 *f = NLMSG_DATA(n);
922 struct rtattr * tb[XFRMA_SPD_MAX+1];
923 struct rtattr * rta;
924
925 int len = n->nlmsg_len;
926
927 len -= NLMSG_LENGTH(sizeof(__u32));
928 if (len < 0) {
929 fprintf(stderr, "SPDinfo: Wrong len %d\n", len);
930 return -1;
931 }
932
933 rta = XFRMSAPD_RTA(f);
934 parse_rtattr(tb, XFRMA_SPD_MAX, rta, len);
935
936 fprintf(fp,"\t SPD");
937 if (tb[XFRMA_SPD_INFO]) {
938 struct xfrmu_spdinfo *si;
939
940 if (RTA_PAYLOAD(tb[XFRMA_SPD_INFO]) < sizeof(*si)) {
941 fprintf(stderr, "SPDinfo: Wrong len %d\n", len);
942 return -1;
943 }
944 si = RTA_DATA(tb[XFRMA_SPD_INFO]);
945 fprintf(fp," IN %d", si->incnt);
946 fprintf(fp," OUT %d", si->outcnt);
947 fprintf(fp," FWD %d", si->fwdcnt);
948
949 if (show_stats) {
950 fprintf(fp," (Sock:");
951 fprintf(fp," IN %d", si->inscnt);
952 fprintf(fp," OUT %d", si->outscnt);
953 fprintf(fp," FWD %d", si->fwdscnt);
954 fprintf(fp,")");
955 }
956
957 fprintf(fp, "%s", _SL_);
958 }
959 if (show_stats > 1) {
960 struct xfrmu_spdhinfo *sh;
961
962 if (tb[XFRMA_SPD_HINFO]) {
963 if (RTA_PAYLOAD(tb[XFRMA_SPD_HINFO]) < sizeof(*sh)) {
964 fprintf(stderr, "SPDinfo: Wrong len %d\n", len);
965 return -1;
966 }
967 sh = RTA_DATA(tb[XFRMA_SPD_HINFO]);
968 fprintf(fp,"\t SPD buckets:");
969 fprintf(fp," count %d", sh->spdhcnt);
970 fprintf(fp," Max %d", sh->spdhmcnt);
971 fprintf(fp, "%s", _SL_);
972 }
973 if (tb[XFRMA_SPD_IPV4_HTHRESH]) {
974 struct xfrmu_spdhthresh *th;
975 if (RTA_PAYLOAD(tb[XFRMA_SPD_IPV4_HTHRESH]) < sizeof(*th)) {
976 fprintf(stderr, "SPDinfo: Wrong len %d\n", len);
977 return -1;
978 }
979 th = RTA_DATA(tb[XFRMA_SPD_IPV4_HTHRESH]);
980 fprintf(fp,"\t SPD IPv4 thresholds:");
981 fprintf(fp," local %d", th->lbits);
982 fprintf(fp," remote %d", th->rbits);
983 fprintf(fp, "%s", _SL_);
984
985 }
986 if (tb[XFRMA_SPD_IPV6_HTHRESH]) {
987 struct xfrmu_spdhthresh *th;
988 if (RTA_PAYLOAD(tb[XFRMA_SPD_IPV6_HTHRESH]) < sizeof(*th)) {
989 fprintf(stderr, "SPDinfo: Wrong len %d\n", len);
990 return -1;
991 }
992 th = RTA_DATA(tb[XFRMA_SPD_IPV6_HTHRESH]);
993 fprintf(fp,"\t SPD IPv6 thresholds:");
994 fprintf(fp," local %d", th->lbits);
995 fprintf(fp," remote %d", th->rbits);
996 fprintf(fp, "%s", _SL_);
997 }
998 }
999
1000 if (oneline)
1001 fprintf(fp, "\n");
1002
1003 return 0;
1004 }
1005
xfrm_spd_setinfo(int argc,char ** argv)1006 static int xfrm_spd_setinfo(int argc, char **argv)
1007 {
1008 struct rtnl_handle rth;
1009 struct {
1010 struct nlmsghdr n;
1011 __u32 flags;
1012 char buf[RTA_BUF_SIZE];
1013 } req;
1014
1015 char *thr4 = NULL;
1016 char *thr6 = NULL;
1017
1018 memset(&req, 0, sizeof(req));
1019
1020 req.n.nlmsg_len = NLMSG_LENGTH(sizeof(__u32));
1021 req.n.nlmsg_flags = NLM_F_REQUEST;
1022 req.n.nlmsg_type = XFRM_MSG_NEWSPDINFO;
1023 req.flags = 0XFFFFFFFF;
1024
1025 while (argc > 0) {
1026 if (strcmp(*argv, "hthresh4") == 0) {
1027 struct xfrmu_spdhthresh thr;
1028
1029 if (thr4)
1030 duparg("hthresh4", *argv);
1031 thr4 = *argv;
1032 NEXT_ARG();
1033 if (get_u8(&thr.lbits, *argv, 0) || thr.lbits > 32)
1034 invarg("hthresh4 LBITS value is invalid", *argv);
1035 NEXT_ARG();
1036 if (get_u8(&thr.rbits, *argv, 0) || thr.rbits > 32)
1037 invarg("hthresh4 RBITS value is invalid", *argv);
1038
1039 addattr_l(&req.n, sizeof(req), XFRMA_SPD_IPV4_HTHRESH,
1040 (void *)&thr, sizeof(thr));
1041 } else if (strcmp(*argv, "hthresh6") == 0) {
1042 struct xfrmu_spdhthresh thr;
1043
1044 if (thr6)
1045 duparg("hthresh6", *argv);
1046 thr6 = *argv;
1047 NEXT_ARG();
1048 if (get_u8(&thr.lbits, *argv, 0) || thr.lbits > 128)
1049 invarg("hthresh6 LBITS value is invalid", *argv);
1050 NEXT_ARG();
1051 if (get_u8(&thr.rbits, *argv, 0) || thr.rbits > 128)
1052 invarg("hthresh6 RBITS value is invalid", *argv);
1053
1054 addattr_l(&req.n, sizeof(req), XFRMA_SPD_IPV6_HTHRESH,
1055 (void *)&thr, sizeof(thr));
1056 } else {
1057 invarg("unknown", *argv);
1058 }
1059
1060 argc--; argv++;
1061 }
1062
1063 if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
1064 exit(1);
1065
1066 if (rtnl_talk(&rth, &req.n, NULL, 0) < 0)
1067 exit(2);
1068
1069 rtnl_close(&rth);
1070
1071 return 0;
1072 }
1073
xfrm_spd_getinfo(int argc,char ** argv)1074 static int xfrm_spd_getinfo(int argc, char **argv)
1075 {
1076 struct rtnl_handle rth;
1077 struct {
1078 struct nlmsghdr n;
1079 __u32 flags;
1080 char ans[128];
1081 } req;
1082
1083 memset(&req, 0, sizeof(req));
1084
1085 req.n.nlmsg_len = NLMSG_LENGTH(sizeof(__u32));
1086 req.n.nlmsg_flags = NLM_F_REQUEST;
1087 req.n.nlmsg_type = XFRM_MSG_GETSPDINFO;
1088 req.flags = 0XFFFFFFFF;
1089
1090 if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
1091 exit(1);
1092
1093 if (rtnl_talk(&rth, &req.n, &req.n, sizeof(req)) < 0)
1094 exit(2);
1095
1096 print_spdinfo(&req.n, (void*)stdout);
1097
1098 rtnl_close(&rth);
1099
1100 return 0;
1101 }
1102
xfrm_policy_flush(int argc,char ** argv)1103 static int xfrm_policy_flush(int argc, char **argv)
1104 {
1105 struct rtnl_handle rth;
1106 struct {
1107 struct nlmsghdr n;
1108 char buf[RTA_BUF_SIZE];
1109 } req;
1110 char *ptypep = NULL;
1111 struct xfrm_userpolicy_type upt;
1112
1113 memset(&req, 0, sizeof(req));
1114 memset(&upt, 0, sizeof(upt));
1115
1116 req.n.nlmsg_len = NLMSG_LENGTH(0); /* nlmsg data is nothing */
1117 req.n.nlmsg_flags = NLM_F_REQUEST;
1118 req.n.nlmsg_type = XFRM_MSG_FLUSHPOLICY;
1119
1120 while (argc > 0) {
1121 if (strcmp(*argv, "ptype") == 0) {
1122 if (ptypep)
1123 duparg("ptype", *argv);
1124 ptypep = *argv;
1125
1126 NEXT_ARG();
1127 xfrm_policy_ptype_parse(&upt.type, &argc, &argv);
1128 } else
1129 invarg("unknown", *argv);
1130
1131 argc--; argv++;
1132 }
1133
1134 if (ptypep) {
1135 addattr_l(&req.n, sizeof(req), XFRMA_POLICY_TYPE,
1136 (void *)&upt, sizeof(upt));
1137 }
1138
1139 if (rtnl_open_byproto(&rth, 0, NETLINK_XFRM) < 0)
1140 exit(1);
1141
1142 if (show_stats > 1)
1143 fprintf(stderr, "Flush policy\n");
1144
1145 if (rtnl_talk(&rth, &req.n, NULL, 0) < 0)
1146 exit(2);
1147
1148 rtnl_close(&rth);
1149
1150 return 0;
1151 }
1152
do_xfrm_policy(int argc,char ** argv)1153 int do_xfrm_policy(int argc, char **argv)
1154 {
1155 if (argc < 1)
1156 return xfrm_policy_list_or_deleteall(0, NULL, 0);
1157
1158 if (matches(*argv, "add") == 0)
1159 return xfrm_policy_modify(XFRM_MSG_NEWPOLICY, 0,
1160 argc-1, argv+1);
1161 if (matches(*argv, "update") == 0)
1162 return xfrm_policy_modify(XFRM_MSG_UPDPOLICY, 0,
1163 argc-1, argv+1);
1164 if (matches(*argv, "delete") == 0)
1165 return xfrm_policy_delete(argc-1, argv+1);
1166 if (matches(*argv, "deleteall") == 0 || matches(*argv, "delall") == 0)
1167 return xfrm_policy_list_or_deleteall(argc-1, argv+1, 1);
1168 if (matches(*argv, "list") == 0 || matches(*argv, "show") == 0
1169 || matches(*argv, "lst") == 0)
1170 return xfrm_policy_list_or_deleteall(argc-1, argv+1, 0);
1171 if (matches(*argv, "get") == 0)
1172 return xfrm_policy_get(argc-1, argv+1);
1173 if (matches(*argv, "flush") == 0)
1174 return xfrm_policy_flush(argc-1, argv+1);
1175 if (matches(*argv, "count") == 0)
1176 return xfrm_spd_getinfo(argc, argv);
1177 if (matches(*argv, "set") == 0)
1178 return xfrm_spd_setinfo(argc-1, argv+1);
1179 if (matches(*argv, "help") == 0)
1180 usage();
1181 fprintf(stderr, "Command \"%s\" is unknown, try \"ip xfrm policy help\".\n", *argv);
1182 exit(-1);
1183 }
1184