1 /* 2 * Copyright (c) 2005-2013 Patrick McHardy <kaber (at) trash.net> 3 */ 4 5 #include <stdbool.h> 6 #include <stdint.h> 7 #include <stdio.h> 8 #include <string.h> 9 #include <netdb.h> 10 #include <xtables.h> 11 #include <linux/netfilter/xt_policy.h> 12 13 enum { 14 O_DIRECTION = 0, 15 O_POLICY, 16 O_STRICT, 17 O_REQID, 18 O_SPI, 19 O_PROTO, 20 O_MODE, 21 O_TUNNELSRC, 22 O_TUNNELDST, 23 O_NEXT, 24 F_STRICT = 1 << O_STRICT, 25 }; 26 27 static void policy_help(void) 28 { 29 printf( 30 "policy match options:\n" 31 " --dir in|out match policy applied during decapsulation/\n" 32 " policy to be applied during encapsulation\n" 33 " --pol none|ipsec match policy\n" 34 " --strict match entire policy instead of single element\n" 35 " at any position\n" 36 "These options may be used repeatedly, to describe policy elements:\n" 37 "[!] --reqid reqid match reqid\n" 38 "[!] --spi spi match SPI\n" 39 "[!] --proto proto match protocol (ah/esp/ipcomp)\n" 40 "[!] --mode mode match mode (transport/tunnel)\n" 41 "[!] --tunnel-src addr/mask match tunnel source\n" 42 "[!] --tunnel-dst addr/mask match tunnel destination\n" 43 " --next begin next element in policy\n"); 44 } 45 46 static const struct xt_option_entry policy_opts[] = { 47 {.name = "dir", .id = O_DIRECTION, .type = XTTYPE_STRING}, 48 {.name = "pol", .id = O_POLICY, .type = XTTYPE_STRING}, 49 {.name = "strict", .id = O_STRICT, .type = XTTYPE_NONE}, 50 {.name = "reqid", .id = O_REQID, .type = XTTYPE_UINT32, 51 .flags = XTOPT_MULTI | XTOPT_INVERT}, 52 {.name = "spi", .id = O_SPI, .type = XTTYPE_UINT32, 53 .flags = XTOPT_MULTI | XTOPT_INVERT}, 54 {.name = "tunnel-src", .id = O_TUNNELSRC, .type = XTTYPE_HOSTMASK, 55 .flags = XTOPT_MULTI | XTOPT_INVERT}, 56 {.name = "tunnel-dst", .id = O_TUNNELDST, .type = XTTYPE_HOSTMASK, 57 .flags = XTOPT_MULTI | XTOPT_INVERT}, 58 {.name = "proto", .id = O_PROTO, .type = XTTYPE_PROTOCOL, 59 .flags = XTOPT_MULTI | XTOPT_INVERT}, 60 {.name = "mode", .id = O_MODE, .type = XTTYPE_STRING, 61 .flags = XTOPT_MULTI | XTOPT_INVERT}, 62 {.name = "next", .id = O_NEXT, .type = XTTYPE_NONE, 63 .flags = XTOPT_MULTI, .also = F_STRICT}, 64 XTOPT_TABLEEND, 65 }; 66 67 static int parse_direction(const char *s) 68 { 69 if (strcmp(s, "in") == 0) 70 return XT_POLICY_MATCH_IN; 71 if (strcmp(s, "out") == 0) 72 return XT_POLICY_MATCH_OUT; 73 xtables_error(PARAMETER_PROBLEM, "policy_match: invalid dir \"%s\"", s); 74 } 75 76 static int parse_policy(const char *s) 77 { 78 if (strcmp(s, "none") == 0) 79 return XT_POLICY_MATCH_NONE; 80 if (strcmp(s, "ipsec") == 0) 81 return 0; 82 xtables_error(PARAMETER_PROBLEM, "policy match: invalid policy \"%s\"", s); 83 } 84 85 static int parse_mode(const char *s) 86 { 87 if (strcmp(s, "transport") == 0) 88 return XT_POLICY_MODE_TRANSPORT; 89 if (strcmp(s, "tunnel") == 0) 90 return XT_POLICY_MODE_TUNNEL; 91 xtables_error(PARAMETER_PROBLEM, "policy match: invalid mode \"%s\"", s); 92 } 93 94 static void policy_parse(struct xt_option_call *cb) 95 { 96 struct xt_policy_info *info = cb->data; 97 struct xt_policy_elem *e = &info->pol[info->len]; 98 99 xtables_option_parse(cb); 100 switch (cb->entry->id) { 101 case O_DIRECTION: 102 info->flags |= parse_direction(cb->arg); 103 break; 104 case O_POLICY: 105 info->flags |= parse_policy(cb->arg); 106 break; 107 case O_STRICT: 108 info->flags |= XT_POLICY_MATCH_STRICT; 109 break; 110 case O_REQID: 111 if (e->match.reqid) 112 xtables_error(PARAMETER_PROBLEM, 113 "policy match: double --reqid option"); 114 e->match.reqid = 1; 115 e->invert.reqid = cb->invert; 116 e->reqid = cb->val.u32; 117 break; 118 case O_SPI: 119 if (e->match.spi) 120 xtables_error(PARAMETER_PROBLEM, 121 "policy match: double --spi option"); 122 e->match.spi = 1; 123 e->invert.spi = cb->invert; 124 e->spi = cb->val.u32; 125 break; 126 case O_TUNNELSRC: 127 if (e->match.saddr) 128 xtables_error(PARAMETER_PROBLEM, 129 "policy match: double --tunnel-src option"); 130 131 e->match.saddr = 1; 132 e->invert.saddr = cb->invert; 133 memcpy(&e->saddr, &cb->val.haddr, sizeof(cb->val.haddr)); 134 memcpy(&e->smask, &cb->val.hmask, sizeof(cb->val.hmask)); 135 break; 136 case O_TUNNELDST: 137 if (e->match.daddr) 138 xtables_error(PARAMETER_PROBLEM, 139 "policy match: double --tunnel-dst option"); 140 e->match.daddr = 1; 141 e->invert.daddr = cb->invert; 142 memcpy(&e->daddr, &cb->val.haddr, sizeof(cb->val.haddr)); 143 memcpy(&e->dmask, &cb->val.hmask, sizeof(cb->val.hmask)); 144 break; 145 case O_PROTO: 146 if (e->match.proto) 147 xtables_error(PARAMETER_PROBLEM, 148 "policy match: double --proto option"); 149 e->proto = cb->val.protocol; 150 if (e->proto != IPPROTO_AH && e->proto != IPPROTO_ESP && 151 e->proto != IPPROTO_COMP) 152 xtables_error(PARAMETER_PROBLEM, 153 "policy match: protocol must be ah/esp/ipcomp"); 154 e->match.proto = 1; 155 e->invert.proto = cb->invert; 156 break; 157 case O_MODE: 158 if (e->match.mode) 159 xtables_error(PARAMETER_PROBLEM, 160 "policy match: double --mode option"); 161 e->match.mode = 1; 162 e->invert.mode = cb->invert; 163 e->mode = parse_mode(cb->arg); 164 break; 165 case O_NEXT: 166 if (++info->len == XT_POLICY_MAX_ELEM) 167 xtables_error(PARAMETER_PROBLEM, 168 "policy match: maximum policy depth reached"); 169 break; 170 } 171 } 172 173 static void policy_check(struct xt_fcheck_call *cb) 174 { 175 struct xt_policy_info *info = cb->data; 176 const struct xt_policy_elem *e; 177 int i; 178 179 /* 180 * The old "no parameters given" check is carried out 181 * by testing for --dir. 182 */ 183 if (!(info->flags & (XT_POLICY_MATCH_IN | XT_POLICY_MATCH_OUT))) 184 xtables_error(PARAMETER_PROBLEM, 185 "policy match: neither --dir in nor --dir out specified"); 186 187 if (info->flags & XT_POLICY_MATCH_NONE) { 188 if (info->flags & XT_POLICY_MATCH_STRICT) 189 xtables_error(PARAMETER_PROBLEM, 190 "policy match: policy none but --strict given"); 191 192 if (info->len != 0) 193 xtables_error(PARAMETER_PROBLEM, 194 "policy match: policy none but policy given"); 195 } else 196 info->len++; /* increase len by 1, no --next after last element */ 197 198 /* 199 * This is already represented with O_NEXT requiring F_STRICT in the 200 * options table, but will keep this code as a comment for reference. 201 * 202 if (!(info->flags & XT_POLICY_MATCH_STRICT) && info->len > 1) 203 xtables_error(PARAMETER_PROBLEM, 204 "policy match: multiple elements but no --strict"); 205 */ 206 207 for (i = 0; i < info->len; i++) { 208 e = &info->pol[i]; 209 210 if (info->flags & XT_POLICY_MATCH_STRICT && 211 !(e->match.reqid || e->match.spi || e->match.saddr || 212 e->match.daddr || e->match.proto || e->match.mode)) 213 xtables_error(PARAMETER_PROBLEM, 214 "policy match: empty policy element %u. " 215 "--strict is in effect, but at least one of " 216 "reqid, spi, tunnel-src, tunnel-dst, proto or " 217 "mode is required.", i); 218 219 if ((e->match.saddr || e->match.daddr) 220 && ((e->mode == XT_POLICY_MODE_TUNNEL && e->invert.mode) || 221 (e->mode == XT_POLICY_MODE_TRANSPORT && !e->invert.mode))) 222 xtables_error(PARAMETER_PROBLEM, 223 "policy match: --tunnel-src/--tunnel-dst " 224 "is only valid in tunnel mode"); 225 } 226 } 227 228 static void print_mode(const char *prefix, uint8_t mode, int numeric) 229 { 230 printf(" %smode ", prefix); 231 232 switch (mode) { 233 case XT_POLICY_MODE_TRANSPORT: 234 printf("transport"); 235 break; 236 case XT_POLICY_MODE_TUNNEL: 237 printf("tunnel"); 238 break; 239 default: 240 printf("???"); 241 break; 242 } 243 } 244 245 static void print_proto(const char *prefix, uint8_t proto, int numeric) 246 { 247 const struct protoent *p = NULL; 248 249 printf(" %sproto ", prefix); 250 if (!numeric) 251 p = getprotobynumber(proto); 252 if (p != NULL) 253 printf("%s", p->p_name); 254 else 255 printf("%u", proto); 256 } 257 258 #define PRINT_INVERT(x) \ 259 do { \ 260 if (x) \ 261 printf(" !"); \ 262 } while(0) 263 264 static void print_entry(const char *prefix, const struct xt_policy_elem *e, 265 bool numeric, uint8_t family) 266 { 267 if (e->match.reqid) { 268 PRINT_INVERT(e->invert.reqid); 269 printf(" %sreqid %u", prefix, e->reqid); 270 } 271 if (e->match.spi) { 272 PRINT_INVERT(e->invert.spi); 273 printf(" %sspi 0x%x", prefix, e->spi); 274 } 275 if (e->match.proto) { 276 PRINT_INVERT(e->invert.proto); 277 print_proto(prefix, e->proto, numeric); 278 } 279 if (e->match.mode) { 280 PRINT_INVERT(e->invert.mode); 281 print_mode(prefix, e->mode, numeric); 282 } 283 if (e->match.daddr) { 284 PRINT_INVERT(e->invert.daddr); 285 if (family == NFPROTO_IPV6) 286 printf(" %stunnel-dst %s%s", prefix, 287 xtables_ip6addr_to_numeric(&e->daddr.a6), 288 xtables_ip6mask_to_numeric(&e->dmask.a6)); 289 else 290 printf(" %stunnel-dst %s%s", prefix, 291 xtables_ipaddr_to_numeric(&e->daddr.a4), 292 xtables_ipmask_to_numeric(&e->dmask.a4)); 293 } 294 if (e->match.saddr) { 295 PRINT_INVERT(e->invert.saddr); 296 if (family == NFPROTO_IPV6) 297 printf(" %stunnel-src %s%s", prefix, 298 xtables_ip6addr_to_numeric(&e->saddr.a6), 299 xtables_ip6mask_to_numeric(&e->smask.a6)); 300 else 301 printf(" %stunnel-src %s%s", prefix, 302 xtables_ipaddr_to_numeric(&e->saddr.a4), 303 xtables_ipmask_to_numeric(&e->smask.a4)); 304 } 305 } 306 307 static void print_flags(const char *prefix, const struct xt_policy_info *info) 308 { 309 if (info->flags & XT_POLICY_MATCH_IN) 310 printf(" %sdir in", prefix); 311 else 312 printf(" %sdir out", prefix); 313 314 if (info->flags & XT_POLICY_MATCH_NONE) 315 printf(" %spol none", prefix); 316 else 317 printf(" %spol ipsec", prefix); 318 319 if (info->flags & XT_POLICY_MATCH_STRICT) 320 printf(" %sstrict", prefix); 321 } 322 323 static void policy4_print(const void *ip, const struct xt_entry_match *match, 324 int numeric) 325 { 326 const struct xt_policy_info *info = (void *)match->data; 327 unsigned int i; 328 329 printf(" policy match"); 330 print_flags("", info); 331 for (i = 0; i < info->len; i++) { 332 if (info->len > 1) 333 printf(" [%u]", i); 334 print_entry("", &info->pol[i], numeric, NFPROTO_IPV4); 335 } 336 } 337 338 static void policy6_print(const void *ip, const struct xt_entry_match *match, 339 int numeric) 340 { 341 const struct xt_policy_info *info = (void *)match->data; 342 unsigned int i; 343 344 printf(" policy match"); 345 print_flags("", info); 346 for (i = 0; i < info->len; i++) { 347 if (info->len > 1) 348 printf(" [%u]", i); 349 print_entry("", &info->pol[i], numeric, NFPROTO_IPV6); 350 } 351 } 352 353 static void policy4_save(const void *ip, const struct xt_entry_match *match) 354 { 355 const struct xt_policy_info *info = (void *)match->data; 356 unsigned int i; 357 358 print_flags("--", info); 359 for (i = 0; i < info->len; i++) { 360 print_entry("--", &info->pol[i], false, NFPROTO_IPV4); 361 if (i + 1 < info->len) 362 printf(" --next"); 363 } 364 } 365 366 static void policy6_save(const void *ip, const struct xt_entry_match *match) 367 { 368 const struct xt_policy_info *info = (void *)match->data; 369 unsigned int i; 370 371 print_flags("--", info); 372 for (i = 0; i < info->len; i++) { 373 print_entry("--", &info->pol[i], false, NFPROTO_IPV6); 374 if (i + 1 < info->len) 375 printf(" --next"); 376 } 377 } 378 379 static struct xtables_match policy_mt_reg[] = { 380 { 381 .name = "policy", 382 .version = XTABLES_VERSION, 383 .family = NFPROTO_IPV4, 384 .size = XT_ALIGN(sizeof(struct xt_policy_info)), 385 .userspacesize = XT_ALIGN(sizeof(struct xt_policy_info)), 386 .help = policy_help, 387 .x6_parse = policy_parse, 388 .x6_fcheck = policy_check, 389 .print = policy4_print, 390 .save = policy4_save, 391 .x6_options = policy_opts, 392 }, 393 { 394 .name = "policy", 395 .version = XTABLES_VERSION, 396 .family = NFPROTO_IPV6, 397 .size = XT_ALIGN(sizeof(struct xt_policy_info)), 398 .userspacesize = XT_ALIGN(sizeof(struct xt_policy_info)), 399 .help = policy_help, 400 .x6_parse = policy_parse, 401 .x6_fcheck = policy_check, 402 .print = policy6_print, 403 .save = policy6_save, 404 .x6_options = policy_opts, 405 }, 406 }; 407 408 void _init(void) 409 { 410 xtables_register_matches(policy_mt_reg, ARRAY_SIZE(policy_mt_reg)); 411 } 412