ubus: use blobmsg_parse to validate data from network.interface:dump
[project/firewall3.git] / iptables.c
1 /*
2 * firewall3 - 3rd OpenWrt UCI firewall implementation
3 *
4 * Copyright (C) 2013 Jo-Philipp Wich <jow@openwrt.org>
5 *
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 */
18
19 #include "iptables.h"
20
21
22 static struct option base_opts[] = {
23 { .name = "match", .has_arg = 1, .val = 'm' },
24 { .name = "jump", .has_arg = 1, .val = 'j' },
25 { NULL }
26 };
27
28 static struct xtables_globals xtg = {
29 .option_offset = 0,
30 .program_version = "4",
31 .orig_opts = base_opts,
32 };
33
34 static struct xtables_globals xtg6 = {
35 .option_offset = 0,
36 .program_version = "6",
37 .orig_opts = base_opts,
38 };
39
40 /* Required by certain extensions like SNAT and DNAT */
41 int kernel_version = 0;
42
43 void
44 get_kernel_version(void)
45 {
46 static struct utsname uts;
47 int x = 0, y = 0, z = 0;
48
49 if (uname(&uts) == -1)
50 sprintf(uts.release, "3.0.0");
51
52 sscanf(uts.release, "%d.%d.%d", &x, &y, &z);
53 kernel_version = 0x10000 * x + 0x100 * y + z;
54 }
55
56 #ifdef DISABLE_IPV6
57 #undef __ipt_module
58 #define __ipt_module(x) libxt_##x##_init, libipt_##x##_init,
59 #else
60 #undef __ipt_module
61 #define __ipt_module(x) libxt_##x##_init, libipt_##x##_init, libip6t_##x##_init,
62 #endif
63
64 static void fw3_init_extensions(void)
65 {
66 int i;
67 void (*initfuncs[])(void) = { FW3_IPT_MODULES };
68
69 for (i = 0; i < sizeof(initfuncs)/sizeof(initfuncs[0]); i++)
70 if (initfuncs[i])
71 initfuncs[i]();
72 }
73
74 struct fw3_ipt_handle *
75 fw3_ipt_open(enum fw3_family family, enum fw3_table table)
76 {
77 struct fw3_ipt_handle *h;
78
79 h = fw3_alloc(sizeof(*h));
80
81 xtables_init();
82
83 if (family == FW3_FAMILY_V6)
84 {
85 #ifndef DISABLE_IPV6
86 h->family = FW3_FAMILY_V6;
87 h->table = table;
88 h->handle = ip6tc_init(fw3_flag_names[table]);
89
90 xtables_set_params(&xtg6);
91 xtables_set_nfproto(NFPROTO_IPV6);
92 #endif
93 }
94 else
95 {
96 h->family = FW3_FAMILY_V4;
97 h->table = table;
98 h->handle = iptc_init(fw3_flag_names[table]);
99
100 xtables_set_params(&xtg);
101 xtables_set_nfproto(NFPROTO_IPV4);
102 }
103
104 if (!h->handle)
105 {
106 free(h);
107 return NULL;
108 }
109
110 fw3_xt_reset();
111 fw3_init_extensions();
112
113 return h;
114 }
115
116 static void
117 debug(struct fw3_ipt_handle *h, const char *fmt, ...)
118 {
119 va_list ap;
120
121 printf("%s -t %s ", (h->family == FW3_FAMILY_V6) ? "ip6tables" : "iptables",
122 fw3_flag_names[h->table]);
123
124 va_start(ap, fmt);
125 vprintf(fmt, ap);
126 va_end(ap);
127 }
128
129 void
130 fw3_ipt_set_policy(struct fw3_ipt_handle *h, const char *chain,
131 enum fw3_flag policy)
132 {
133 if (fw3_pr_debug)
134 debug(h, "-P %s %s\n", chain, fw3_flag_names[policy]);
135
136 #ifndef DISABLE_IPV6
137 if (h->family == FW3_FAMILY_V6)
138 ip6tc_set_policy(chain, fw3_flag_names[policy], NULL, h->handle);
139 else
140 #endif
141 iptc_set_policy(chain, fw3_flag_names[policy], NULL, h->handle);
142 }
143
144 void
145 fw3_ipt_flush_chain(struct fw3_ipt_handle *h, const char *chain)
146 {
147 if (fw3_pr_debug)
148 debug(h, "-F %s\n", chain);
149
150 #ifndef DISABLE_IPV6
151 if (h->family == FW3_FAMILY_V6)
152 ip6tc_flush_entries(chain, h->handle);
153 else
154 #endif
155 iptc_flush_entries(chain, h->handle);
156 }
157
158 static void
159 delete_rules(struct fw3_ipt_handle *h, const char *target)
160 {
161 unsigned int num;
162 const struct ipt_entry *e;
163 const char *chain;
164 const char *t;
165 bool found;
166
167 #ifndef DISABLE_IPV6
168 if (h->family == FW3_FAMILY_V6)
169 {
170 for (chain = ip6tc_first_chain(h->handle);
171 chain != NULL;
172 chain = ip6tc_next_chain(h->handle))
173 {
174 do {
175 found = false;
176
177 const struct ip6t_entry *e6;
178 for (num = 0, e6 = ip6tc_first_rule(chain, h->handle);
179 e6 != NULL;
180 num++, e6 = ip6tc_next_rule(e6, h->handle))
181 {
182 t = ip6tc_get_target(e6, h->handle);
183
184 if (*t && !strcmp(t, target))
185 {
186 if (fw3_pr_debug)
187 debug(h, "-D %s %u\n", chain, num + 1);
188
189 ip6tc_delete_num_entry(chain, num, h->handle);
190 found = true;
191 break;
192 }
193 }
194 } while (found);
195 }
196 }
197 else
198 #endif
199 {
200 for (chain = iptc_first_chain(h->handle);
201 chain != NULL;
202 chain = iptc_next_chain(h->handle))
203 {
204 do {
205 found = false;
206
207 for (num = 0, e = iptc_first_rule(chain, h->handle);
208 e != NULL;
209 num++, e = iptc_next_rule(e, h->handle))
210 {
211 t = iptc_get_target(e, h->handle);
212
213 if (*t && !strcmp(t, target))
214 {
215 if (fw3_pr_debug)
216 debug(h, "-D %s %u\n", chain, num + 1);
217
218 iptc_delete_num_entry(chain, num, h->handle);
219 found = true;
220 break;
221 }
222 }
223 } while (found);
224 }
225 }
226 }
227
228 void
229 fw3_ipt_delete_chain(struct fw3_ipt_handle *h, const char *chain)
230 {
231 delete_rules(h, chain);
232
233 if (fw3_pr_debug)
234 debug(h, "-X %s\n", chain);
235
236 #ifndef DISABLE_IPV6
237 if (h->family == FW3_FAMILY_V6)
238 ip6tc_delete_chain(chain, h->handle);
239 else
240 #endif
241 iptc_delete_chain(chain, h->handle);
242 }
243
244 void
245 fw3_ipt_create_chain(struct fw3_ipt_handle *h, const char *fmt, ...)
246 {
247 char buf[32];
248 va_list ap;
249
250 va_start(ap, fmt);
251 vsnprintf(buf, sizeof(buf) - 1, fmt, ap);
252 va_end(ap);
253
254 if (fw3_pr_debug)
255 debug(h, "-N %s\n", buf);
256
257 iptc_create_chain(buf, h->handle);
258 }
259
260 void
261 fw3_ipt_flush(struct fw3_ipt_handle *h)
262 {
263 const char *chain;
264
265 #ifndef DISABLE_IPV6
266 if (h->family == FW3_FAMILY_V6)
267 {
268 for (chain = ip6tc_first_chain(h->handle);
269 chain != NULL;
270 chain = ip6tc_next_chain(h->handle))
271 {
272 ip6tc_flush_entries(chain, h->handle);
273 }
274
275 for (chain = ip6tc_first_chain(h->handle);
276 chain != NULL;
277 chain = ip6tc_next_chain(h->handle))
278 {
279 ip6tc_delete_chain(chain, h->handle);
280 }
281 }
282 else
283 #endif
284 {
285 for (chain = iptc_first_chain(h->handle);
286 chain != NULL;
287 chain = iptc_next_chain(h->handle))
288 {
289 iptc_flush_entries(chain, h->handle);
290 }
291
292 for (chain = iptc_first_chain(h->handle);
293 chain != NULL;
294 chain = iptc_next_chain(h->handle))
295 {
296 iptc_delete_chain(chain, h->handle);
297 }
298 }
299 }
300
301 void
302 fw3_ipt_commit(struct fw3_ipt_handle *h)
303 {
304 int rv;
305
306 #ifndef DISABLE_IPV6
307 if (h->family == FW3_FAMILY_V6)
308 {
309 rv = ip6tc_commit(h->handle);
310 if (!rv)
311 warn("ip6tc_commit(): %s", ip6tc_strerror(errno));
312 }
313 else
314 #endif
315 {
316 rv = iptc_commit(h->handle);
317 if (!rv)
318 warn("iptc_commit(): %s", iptc_strerror(errno));
319 }
320 }
321
322 void
323 fw3_ipt_close(struct fw3_ipt_handle *h)
324 {
325 if (h->libv)
326 {
327 while (h->libc > 0)
328 {
329 h->libc--;
330 dlclose(h->libv[h->libc]);
331 }
332
333 free(h->libv);
334 }
335
336 free(h);
337 }
338
339 struct fw3_ipt_rule *
340 fw3_ipt_rule_new(struct fw3_ipt_handle *h)
341 {
342 struct fw3_ipt_rule *r;
343
344 r = fw3_alloc(sizeof(*r));
345
346 r->h = h;
347 r->argv = fw3_alloc(sizeof(char *));
348 r->argv[r->argc++] = "fw3";
349
350 return r;
351 }
352
353
354 static bool
355 is_chain(struct fw3_ipt_handle *h, const char *name)
356 {
357 #ifndef DISABLE_IPV6
358 if (h->family == FW3_FAMILY_V6)
359 return ip6tc_is_chain(name, h->handle);
360 else
361 #endif
362 return iptc_is_chain(name, h->handle);
363 }
364
365 static char *
366 get_protoname(struct fw3_ipt_rule *r)
367 {
368 const struct xtables_pprot *pp;
369
370 if (r->protocol)
371 for (pp = xtables_chain_protos; pp->name; pp++)
372 if (pp->num == r->protocol)
373 return (char *)pp->name;
374
375 return NULL;
376 }
377
378 static bool
379 load_extension(struct fw3_ipt_handle *h, const char *name)
380 {
381 char path[256];
382 void *lib, **tmp;
383 const char *pfx = (h->family == FW3_FAMILY_V6) ? "libip6t" : "libipt";
384
385 snprintf(path, sizeof(path), "/usr/lib/iptables/libxt_%s.so", name);
386 if (!(lib = dlopen(path, RTLD_NOW)))
387 {
388 snprintf(path, sizeof(path), "/usr/lib/iptables/%s_%s.so", pfx, name);
389 lib = dlopen(path, RTLD_NOW);
390 }
391
392 if (!lib)
393 return false;
394
395 tmp = realloc(h->libv, sizeof(lib) * (h->libc + 1));
396
397 if (!tmp)
398 return false;
399
400 h->libv = tmp;
401 h->libv[h->libc++] = lib;
402
403 return true;
404 }
405
406 static struct xtables_match *
407 find_match(struct fw3_ipt_rule *r, const char *name)
408 {
409 struct xtables_match *m;
410
411 m = xtables_find_match(name, XTF_DONT_LOAD, &r->matches);
412
413 if (!m && load_extension(r->h, name))
414 m = xtables_find_match(name, XTF_DONT_LOAD, &r->matches);
415
416 return m;
417 }
418
419 static void
420 init_match(struct fw3_ipt_rule *r, struct xtables_match *m, bool no_clone)
421 {
422 size_t s;
423 struct xtables_globals *g;
424
425 if (!m)
426 return;
427
428 s = XT_ALIGN(sizeof(struct xt_entry_match)) + m->size;
429
430 m->m = fw3_alloc(s);
431
432 fw3_xt_set_match_name(m);
433
434 m->m->u.user.revision = m->revision;
435 m->m->u.match_size = s;
436
437 /* free previous userspace data */
438 fw3_xt_free_match_udata(m);
439
440 if (m->init)
441 m->init(m->m);
442
443 /* don't merge options if no_clone is set and this match is a clone */
444 if (no_clone && (m == m->next))
445 return;
446
447 /* merge option table */
448 g = (r->h->family == FW3_FAMILY_V6) ? &xtg6 : &xtg;
449 fw3_xt_merge_match_options(g, m);
450 }
451
452 static bool
453 need_protomatch(struct fw3_ipt_rule *r, const char *pname)
454 {
455 if (!pname)
456 return false;
457
458 if (!xtables_find_match(pname, XTF_DONT_LOAD, NULL))
459 return true;
460
461 return !r->protocol_loaded;
462 }
463
464 static struct xtables_match *
465 load_protomatch(struct fw3_ipt_rule *r)
466 {
467 const char *pname = get_protoname(r);
468
469 if (!need_protomatch(r, pname))
470 return NULL;
471
472 return find_match(r, pname);
473 }
474
475 static struct xtables_target *
476 find_target(struct fw3_ipt_rule *r, const char *name)
477 {
478 struct xtables_target *t;
479
480 if (is_chain(r->h, name))
481 return xtables_find_target(XT_STANDARD_TARGET, XTF_LOAD_MUST_SUCCEED);
482
483 t = xtables_find_target(name, XTF_DONT_LOAD);
484
485 if (!t && load_extension(r->h, name))
486 t = xtables_find_target(name, XTF_DONT_LOAD);
487
488 return t;
489 }
490
491 static struct xtables_target *
492 get_target(struct fw3_ipt_rule *r, const char *name)
493 {
494 size_t s;
495 struct xtables_target *t;
496 struct xtables_globals *g;
497
498 t = find_target(r, name);
499
500 if (!t)
501 return NULL;
502
503 s = XT_ALIGN(sizeof(struct xt_entry_target)) + t->size;
504 t->t = fw3_alloc(s);
505
506 fw3_xt_set_target_name(t, name);
507
508 t->t->u.user.revision = t->revision;
509 t->t->u.target_size = s;
510
511 /* free previous userspace data */
512 fw3_xt_free_target_udata(t);
513
514 if (t->init)
515 t->init(t->t);
516
517 /* merge option table */
518 g = (r->h->family == FW3_FAMILY_V6) ? &xtg6 : &xtg;
519 fw3_xt_merge_target_options(g, t);
520
521 r->target = t;
522
523 return t;
524 }
525
526 void
527 fw3_ipt_rule_proto(struct fw3_ipt_rule *r, struct fw3_protocol *proto)
528 {
529 uint32_t pr;
530
531 if (!proto || proto->any)
532 return;
533
534 pr = proto->protocol;
535
536 #ifndef DISABLE_IPV6
537 if (r->h->family == FW3_FAMILY_V6)
538 {
539 if (pr == 1)
540 pr = 58;
541
542 r->e6.ipv6.proto = pr;
543 r->e6.ipv6.flags |= IP6T_F_PROTO;
544
545 if (proto->invert)
546 r->e6.ipv6.invflags |= XT_INV_PROTO;
547 }
548 else
549 #endif
550 {
551 r->e.ip.proto = pr;
552
553 if (proto->invert)
554 r->e.ip.invflags |= XT_INV_PROTO;
555 }
556
557 r->protocol = pr;
558 }
559
560 void
561 fw3_ipt_rule_in_out(struct fw3_ipt_rule *r,
562 struct fw3_device *in, struct fw3_device *out)
563 {
564 #ifndef DISABLE_IPV6
565 if (r->h->family == FW3_FAMILY_V6)
566 {
567 if (in && !in->any)
568 {
569 xtables_parse_interface(in->name, r->e6.ipv6.iniface,
570 r->e6.ipv6.iniface_mask);
571
572 if (in->invert)
573 r->e6.ipv6.invflags |= IP6T_INV_VIA_IN;
574 }
575
576 if (out && !out->any)
577 {
578 xtables_parse_interface(out->name, r->e6.ipv6.outiface,
579 r->e6.ipv6.outiface_mask);
580
581 if (out->invert)
582 r->e6.ipv6.invflags |= IP6T_INV_VIA_OUT;
583 }
584 }
585 else
586 #endif
587 {
588 if (in && !in->any)
589 {
590 xtables_parse_interface(in->name, r->e.ip.iniface,
591 r->e.ip.iniface_mask);
592
593 if (in->invert)
594 r->e.ip.invflags |= IPT_INV_VIA_IN;
595 }
596
597 if (out && !out->any)
598 {
599 xtables_parse_interface(out->name, r->e.ip.outiface,
600 r->e.ip.outiface_mask);
601
602 if (out->invert)
603 r->e.ip.invflags |= IPT_INV_VIA_OUT;
604 }
605 }
606 }
607
608
609 static void
610 ip4prefix2mask(int prefix, struct in_addr *mask)
611 {
612 if (prefix > 0)
613 mask->s_addr = htonl(~((1 << (32 - prefix)) - 1));
614 else
615 mask->s_addr = 0;
616 }
617
618 #ifndef DISABLE_IPV6
619 static void
620 ip6prefix2mask(int prefix, struct in6_addr *mask)
621 {
622 char *p = (char *)mask;
623
624 if (prefix > 0)
625 {
626 memset(p, 0xff, prefix / 8);
627 memset(p + (prefix / 8) + 1, 0, (128 - prefix) / 8);
628 p[prefix / 8] = 0xff << (8 - (prefix & 7));
629 }
630 else
631 {
632 memset(mask, 0, sizeof(*mask));
633 }
634 }
635 #endif
636
637 void
638 fw3_ipt_rule_src_dest(struct fw3_ipt_rule *r,
639 struct fw3_address *src, struct fw3_address *dest)
640 {
641 if ((src && src->range) || (dest && dest->range))
642 {
643 fw3_ipt_rule_addarg(r, false, "-m", "iprange");
644 }
645
646 if (src && src->set)
647 {
648 if (src->range)
649 {
650 fw3_ipt_rule_addarg(r, src->invert, "--src-range",
651 fw3_address_to_string(src, false));
652 }
653 #ifndef DISABLE_IPV6
654 else if (r->h->family == FW3_FAMILY_V6)
655 {
656 r->e6.ipv6.src = src->address.v6;
657 ip6prefix2mask(src->mask, &r->e6.ipv6.smsk);
658
659 int i;
660 for (i = 0; i < 4; i++)
661 r->e6.ipv6.src.s6_addr32[i] &= r->e6.ipv6.smsk.s6_addr32[i];
662
663 if (src->invert)
664 r->e6.ipv6.invflags |= IP6T_INV_SRCIP;
665 }
666 #endif
667 else
668 {
669 r->e.ip.src = src->address.v4;
670 ip4prefix2mask(src->mask, &r->e.ip.smsk);
671
672 r->e.ip.src.s_addr &= r->e.ip.smsk.s_addr;
673
674 if (src->invert)
675 r->e.ip.invflags |= IPT_INV_SRCIP;
676 }
677 }
678
679 if (dest && dest->set)
680 {
681 if (dest->range)
682 {
683 fw3_ipt_rule_addarg(r, dest->invert, "--dst-range",
684 fw3_address_to_string(dest, false));
685 }
686 #ifndef DISABLE_IPV6
687 else if (r->h->family == FW3_FAMILY_V6)
688 {
689 r->e6.ipv6.dst = dest->address.v6;
690 ip6prefix2mask(dest->mask, &r->e6.ipv6.dmsk);
691
692 int i;
693 for (i = 0; i < 4; i++)
694 r->e6.ipv6.dst.s6_addr32[i] &= r->e6.ipv6.dmsk.s6_addr32[i];
695
696 if (dest->invert)
697 r->e6.ipv6.invflags |= IP6T_INV_DSTIP;
698 }
699 #endif
700 else
701 {
702 r->e.ip.dst = dest->address.v4;
703 ip4prefix2mask(dest->mask, &r->e.ip.dmsk);
704
705 r->e.ip.dst.s_addr &= r->e.ip.dmsk.s_addr;
706
707 if (dest->invert)
708 r->e.ip.invflags |= IPT_INV_DSTIP;
709 }
710 }
711 }
712
713 void
714 fw3_ipt_rule_sport_dport(struct fw3_ipt_rule *r,
715 struct fw3_port *sp, struct fw3_port *dp)
716 {
717 char buf[sizeof("65535:65535\0")];
718
719 if ((!sp || !sp->set) && (!dp || !dp->set))
720 return;
721
722 if (!get_protoname(r))
723 return;
724
725 if (sp && sp->set)
726 {
727 if (sp->port_min == sp->port_max)
728 sprintf(buf, "%u", sp->port_min);
729 else
730 sprintf(buf, "%u:%u", sp->port_min, sp->port_max);
731
732 fw3_ipt_rule_addarg(r, sp->invert, "--sport", buf);
733 }
734
735 if (dp && dp->set)
736 {
737 if (dp->port_min == dp->port_max)
738 sprintf(buf, "%u", dp->port_min);
739 else
740 sprintf(buf, "%u:%u", dp->port_min, dp->port_max);
741
742 fw3_ipt_rule_addarg(r, dp->invert, "--dport", buf);
743 }
744 }
745
746 void
747 fw3_ipt_rule_device(struct fw3_ipt_rule *r, const char *device, bool out)
748 {
749 if (device) {
750 struct fw3_device dev = { .any = false };
751 strncpy(dev.name, device, sizeof(dev.name) - 1);
752 fw3_ipt_rule_in_out(r, (out) ? NULL : &dev, (out) ? &dev : NULL);
753 }
754 }
755
756 void
757 fw3_ipt_rule_mac(struct fw3_ipt_rule *r, struct fw3_mac *mac)
758 {
759 char buf[sizeof("ff:ff:ff:ff:ff:ff\0")];
760 uint8_t *addr = mac->mac.ether_addr_octet;
761
762 if (!mac)
763 return;
764
765 sprintf(buf, "%02x:%02x:%02x:%02x:%02x:%02x",
766 addr[0], addr[1], addr[2], addr[3], addr[4], addr[5]);
767
768 fw3_ipt_rule_addarg(r, false, "-m", "mac");
769 fw3_ipt_rule_addarg(r, mac->invert, "--mac-source", buf);
770 }
771
772 void
773 fw3_ipt_rule_icmptype(struct fw3_ipt_rule *r, struct fw3_icmptype *icmp)
774 {
775 char buf[sizeof("255/255\0")];
776
777 if (!icmp)
778 return;
779
780 #ifndef DISABLE_IPV6
781 if (r->h->family == FW3_FAMILY_V6)
782 {
783 if (icmp->code6_min == 0 && icmp->code6_max == 0xFF)
784 sprintf(buf, "%u", icmp->type6);
785 else
786 sprintf(buf, "%u/%u", icmp->type6, icmp->code6_min);
787
788 fw3_ipt_rule_addarg(r, icmp->invert, "--icmpv6-type", buf);
789 }
790 else
791 #endif
792 {
793 if (icmp->code_min == 0 && icmp->code_max == 0xFF)
794 sprintf(buf, "%u", icmp->type);
795 else
796 sprintf(buf, "%u/%u", icmp->type, icmp->code_min);
797
798 fw3_ipt_rule_addarg(r, icmp->invert, "--icmp-type", buf);
799 }
800 }
801
802 void
803 fw3_ipt_rule_limit(struct fw3_ipt_rule *r, struct fw3_limit *limit)
804 {
805 char buf[sizeof("-4294967296/second\0")];
806
807 if (!limit || limit->rate <= 0)
808 return;
809
810 fw3_ipt_rule_addarg(r, false, "-m", "limit");
811
812 sprintf(buf, "%u/%s", limit->rate, fw3_limit_units[limit->unit]);
813 fw3_ipt_rule_addarg(r, limit->invert, "--limit", buf);
814
815 if (limit->burst > 0)
816 {
817 sprintf(buf, "%u", limit->burst);
818 fw3_ipt_rule_addarg(r, limit->invert, "--limit-burst", buf);
819 }
820 }
821
822 void
823 fw3_ipt_rule_ipset(struct fw3_ipt_rule *r, struct fw3_setmatch *match)
824 {
825 char buf[sizeof("dst,dst,dst\0")];
826 char *p = buf;
827 int i = 0;
828
829 struct fw3_ipset *set;
830 struct fw3_ipset_datatype *type;
831
832 if (!match || !match->set || !match->ptr)
833 return;
834
835 set = match->ptr;
836 list_for_each_entry(type, &set->datatypes, list)
837 {
838 if (i >= 3)
839 break;
840
841 if (p > buf)
842 *p++ = ',';
843
844 p += sprintf(p, "%s", match->dir[i] ? match->dir[i] : type->dir);
845 i++;
846 }
847
848 fw3_ipt_rule_addarg(r, false, "-m", "set");
849
850 fw3_ipt_rule_addarg(r, match->invert, "--match-set",
851 set->external ? set->external : set->name);
852
853 fw3_ipt_rule_addarg(r, false, buf, NULL);
854 }
855
856 void
857 fw3_ipt_rule_time(struct fw3_ipt_rule *r, struct fw3_time *time)
858 {
859 int i;
860 struct tm empty = { 0 };
861
862 char buf[84]; /* sizeof("1,2,3,...,30,31\0") */
863 char *p;
864
865 bool d1 = memcmp(&time->datestart, &empty, sizeof(empty));
866 bool d2 = memcmp(&time->datestop, &empty, sizeof(empty));
867
868 if (!d1 && !d2 && !time->timestart && !time->timestop &&
869 !(time->monthdays & 0xFFFFFFFE) && !(time->weekdays & 0xFE))
870 {
871 return;
872 }
873
874 fw3_ipt_rule_addarg(r, false, "-m", "time");
875
876 if (time->utc)
877 fw3_ipt_rule_addarg(r, false, "--utc", NULL);
878
879 if (d1)
880 {
881 strftime(buf, sizeof(buf), "%Y-%m-%dT%H:%M:%S", &time->datestart);
882 fw3_ipt_rule_addarg(r, false, "--datestart", buf);
883 }
884
885 if (d2)
886 {
887 strftime(buf, sizeof(buf), "%Y-%m-%dT%H:%M:%S", &time->datestop);
888 fw3_ipt_rule_addarg(r, false, "--datestop", buf);
889 }
890
891 if (time->timestart)
892 {
893 sprintf(buf, "%02d:%02d:%02d",
894 time->timestart / 3600,
895 time->timestart % 3600 / 60,
896 time->timestart % 60);
897
898 fw3_ipt_rule_addarg(r, false, "--timestart", buf);
899 }
900
901 if (time->timestop)
902 {
903 sprintf(buf, "%02d:%02d:%02d",
904 time->timestop / 3600,
905 time->timestop % 3600 / 60,
906 time->timestop % 60);
907
908 fw3_ipt_rule_addarg(r, false, "--timestop", buf);
909 }
910
911 if (time->monthdays & 0xFFFFFFFE)
912 {
913 for (i = 1, p = buf; i < 32; i++)
914 {
915 if (hasbit(time->monthdays, i))
916 {
917 if (p > buf)
918 *p++ = ',';
919
920 p += sprintf(p, "%u", i);
921 }
922 }
923
924 fw3_ipt_rule_addarg(r, hasbit(time->monthdays, 0), "--monthdays", buf);
925 }
926
927 if (time->weekdays & 0xFE)
928 {
929 for (i = 1, p = buf; i < 8; i++)
930 {
931 if (hasbit(time->weekdays, i))
932 {
933 if (p > buf)
934 *p++ = ',';
935
936 p += sprintf(p, "%u", i);
937 }
938 }
939
940 fw3_ipt_rule_addarg(r, hasbit(time->weekdays, 0), "--weekdays", buf);
941 }
942 }
943
944 void
945 fw3_ipt_rule_mark(struct fw3_ipt_rule *r, struct fw3_mark *mark)
946 {
947 char buf[sizeof("0xFFFFFFFF/0xFFFFFFFF\0")];
948
949 if (!mark || !mark->set)
950 return;
951
952 if (mark->mask < 0xFFFFFFFF)
953 sprintf(buf, "0x%x/0x%x", mark->mark, mark->mask);
954 else
955 sprintf(buf, "0x%x", mark->mark);
956
957 fw3_ipt_rule_addarg(r, false, "-m", "mark");
958 fw3_ipt_rule_addarg(r, mark->invert, "--mark", buf);
959 }
960
961 void
962 fw3_ipt_rule_comment(struct fw3_ipt_rule *r, const char *fmt, ...)
963 {
964 va_list ap;
965 char buf[256];
966
967 if (!fmt || !*fmt)
968 return;
969
970 va_start(ap, fmt);
971 vsnprintf(buf, sizeof(buf) - 1, fmt, ap);
972 va_end(ap);
973
974 fw3_ipt_rule_addarg(r, false, "-m", "comment");
975 fw3_ipt_rule_addarg(r, false, "--comment", buf);
976 }
977
978 void
979 fw3_ipt_rule_extra(struct fw3_ipt_rule *r, const char *extra)
980 {
981 char *p, **tmp, *s;
982
983 if (!extra || !*extra)
984 return;
985
986 s = fw3_strdup(extra);
987
988 for (p = strtok(s, " \t"); p; p = strtok(NULL, " \t"))
989 {
990 tmp = realloc(r->argv, (r->argc + 1) * sizeof(*r->argv));
991
992 if (!tmp)
993 break;
994
995 r->argv = tmp;
996 r->argv[r->argc++] = fw3_strdup(p);
997 }
998
999 free(s);
1000 }
1001
1002 #ifndef DISABLE_IPV6
1003 static void
1004 rule_print6(struct ip6t_entry *e)
1005 {
1006 char buf[INET6_ADDRSTRLEN];
1007 char *pname;
1008
1009 if (e->ipv6.flags & IP6T_F_PROTO)
1010 {
1011 if (e->ipv6.flags & XT_INV_PROTO)
1012 printf(" !");
1013
1014 pname = get_protoname(container_of(e, struct fw3_ipt_rule, e6));
1015
1016 if (pname)
1017 printf(" -p %s", pname);
1018 else
1019 printf(" -p %u", e->ipv6.proto);
1020 }
1021
1022 if (e->ipv6.iniface[0])
1023 {
1024 if (e->ipv6.flags & IP6T_INV_VIA_IN)
1025 printf(" !");
1026
1027 printf(" -i %s", e->ipv6.iniface);
1028 }
1029
1030 if (e->ipv6.outiface[0])
1031 {
1032 if (e->ipv6.flags & IP6T_INV_VIA_OUT)
1033 printf(" !");
1034
1035 printf(" -o %s", e->ipv6.outiface);
1036 }
1037
1038 if (memcmp(&e->ipv6.src, &in6addr_any, sizeof(struct in6_addr)))
1039 {
1040 if (e->ipv6.flags & IP6T_INV_SRCIP)
1041 printf(" !");
1042
1043 printf(" -s %s/%u", inet_ntop(AF_INET6, &e->ipv6.src, buf, sizeof(buf)),
1044 xtables_ip6mask_to_cidr(&e->ipv6.smsk));
1045 }
1046
1047 if (memcmp(&e->ipv6.dst, &in6addr_any, sizeof(struct in6_addr)))
1048 {
1049 if (e->ipv6.flags & IP6T_INV_DSTIP)
1050 printf(" !");
1051
1052 printf(" -d %s/%u", inet_ntop(AF_INET6, &e->ipv6.dst, buf, sizeof(buf)),
1053 xtables_ip6mask_to_cidr(&e->ipv6.dmsk));
1054 }
1055 }
1056 #endif
1057
1058 static void
1059 rule_print4(struct ipt_entry *e)
1060 {
1061 struct in_addr in_zero = { 0 };
1062 char buf[sizeof("255.255.255.255\0")];
1063 char *pname;
1064
1065 if (e->ip.proto)
1066 {
1067 if (e->ip.flags & XT_INV_PROTO)
1068 printf(" !");
1069
1070 pname = get_protoname(container_of(e, struct fw3_ipt_rule, e));
1071
1072 if (pname)
1073 printf(" -p %s", pname);
1074 else
1075 printf(" -p %u", e->ip.proto);
1076 }
1077
1078 if (e->ip.iniface[0])
1079 {
1080 if (e->ip.flags & IPT_INV_VIA_IN)
1081 printf(" !");
1082
1083 printf(" -i %s", e->ip.iniface);
1084 }
1085
1086 if (e->ip.outiface[0])
1087 {
1088 if (e->ip.flags & IPT_INV_VIA_OUT)
1089 printf(" !");
1090
1091 printf(" -o %s", e->ip.outiface);
1092 }
1093
1094 if (memcmp(&e->ip.src, &in_zero, sizeof(struct in_addr)))
1095 {
1096 if (e->ip.flags & IPT_INV_SRCIP)
1097 printf(" !");
1098
1099 printf(" -s %s/%u", inet_ntop(AF_INET, &e->ip.src, buf, sizeof(buf)),
1100 xtables_ipmask_to_cidr(&e->ip.smsk));
1101 }
1102
1103 if (memcmp(&e->ip.dst, &in_zero, sizeof(struct in_addr)))
1104 {
1105 if (e->ip.flags & IPT_INV_DSTIP)
1106 printf(" !");
1107
1108 printf(" -d %s/%u", inet_ntop(AF_INET, &e->ip.dst, buf, sizeof(buf)),
1109 xtables_ipmask_to_cidr(&e->ip.dmsk));
1110 }
1111 }
1112
1113 static void
1114 rule_print(struct fw3_ipt_rule *r, const char *prefix, const char *chain)
1115 {
1116 debug(r->h, "%s %s", prefix, chain);
1117
1118 #ifndef DISABLE_IPV6
1119 if (r->h->family == FW3_FAMILY_V6)
1120 rule_print6(&r->e6);
1121 else
1122 #endif
1123 rule_print4(&r->e);
1124
1125 fw3_xt_print_matches(&r->e.ip, r->matches);
1126 fw3_xt_print_target(&r->e.ip, r->target);
1127
1128 printf("\n");
1129 }
1130
1131 static bool
1132 parse_option(struct fw3_ipt_rule *r, int optc, bool inv)
1133 {
1134 struct xtables_rule_match *m;
1135 struct xtables_match *em;
1136
1137 /* is a target option */
1138 if (r->target && fw3_xt_has_target_parse(r->target) &&
1139 optc >= r->target->option_offset &&
1140 optc < (r->target->option_offset + 256))
1141 {
1142 xtables_option_tpcall(optc, r->argv, inv, r->target, &r->e);
1143 return false;
1144 }
1145
1146 /* try to dispatch argument to one of the match parsers */
1147 for (m = r->matches; m; m = m->next)
1148 {
1149 em = m->match;
1150
1151 if (m->completed || !fw3_xt_has_match_parse(em))
1152 continue;
1153
1154 if (optc < em->option_offset ||
1155 optc >= (em->option_offset + 256))
1156 continue;
1157
1158 xtables_option_mpcall(optc, r->argv, inv, em, &r->e);
1159 return false;
1160 }
1161
1162 /* unhandled option, might belong to a protocol match */
1163 if ((em = load_protomatch(r)) != NULL)
1164 {
1165 init_match(r, em, false);
1166
1167 r->protocol_loaded = true;
1168 optind--;
1169
1170 return true;
1171 }
1172
1173 if (optc == ':')
1174 warn("parse_option(): option '%s' needs argument", r->argv[optind-1]);
1175
1176 if (optc == '?')
1177 warn("parse_option(): unknown option '%s'", r->argv[optind-1]);
1178
1179 return false;
1180 }
1181
1182 void
1183 fw3_ipt_rule_addarg(struct fw3_ipt_rule *r, bool inv,
1184 const char *k, const char *v)
1185 {
1186 int n;
1187 char **tmp;
1188
1189 if (!k)
1190 return;
1191
1192 n = inv + !!k + !!v;
1193 tmp = realloc(r->argv, (r->argc + n) * sizeof(*tmp));
1194
1195 if (!tmp)
1196 return;
1197
1198 r->argv = tmp;
1199
1200 if (inv)
1201 r->argv[r->argc++] = fw3_strdup("!");
1202
1203 r->argv[r->argc++] = fw3_strdup(k);
1204
1205 if (v)
1206 r->argv[r->argc++] = fw3_strdup(v);
1207 }
1208
1209 static unsigned char *
1210 rule_mask(struct fw3_ipt_rule *r)
1211 {
1212 size_t s;
1213 unsigned char *p, *mask = NULL;
1214 struct xtables_rule_match *m;
1215
1216 #define SZ(x) XT_ALIGN(sizeof(struct x))
1217
1218 #ifndef DISABLE_IPV6
1219 if (r->h->family == FW3_FAMILY_V6)
1220 {
1221 s = SZ(ip6t_entry);
1222
1223 for (m = r->matches; m; m = m->next)
1224 s += SZ(ip6t_entry_match) + m->match->size;
1225
1226 s += SZ(ip6t_entry_target) + r->target->size;
1227
1228 mask = fw3_alloc(s);
1229 memset(mask, 0xFF, SZ(ip6t_entry));
1230 p = mask + SZ(ip6t_entry);
1231
1232 for (m = r->matches; m; m = m->next)
1233 {
1234 memset(p, 0xFF, SZ(ip6t_entry_match) + m->match->userspacesize);
1235 p += SZ(ip6t_entry_match) + m->match->size;
1236 }
1237
1238 memset(p, 0xFF, SZ(ip6t_entry_target) + r->target->userspacesize);
1239 }
1240 else
1241 #endif
1242 {
1243 s = SZ(ipt_entry);
1244
1245 for (m = r->matches; m; m = m->next)
1246 s += SZ(ipt_entry_match) + m->match->size;
1247
1248 s += SZ(ipt_entry_target) + r->target->size;
1249
1250 mask = fw3_alloc(s);
1251 memset(mask, 0xFF, SZ(ipt_entry));
1252 p = mask + SZ(ipt_entry);
1253
1254 for (m = r->matches; m; m = m->next)
1255 {
1256 memset(p, 0xFF, SZ(ipt_entry_match) + m->match->userspacesize);
1257 p += SZ(ipt_entry_match) + m->match->size;
1258 }
1259
1260 memset(p, 0xFF, SZ(ipt_entry_target) + r->target->userspacesize);
1261 }
1262
1263 return mask;
1264 }
1265
1266 static void *
1267 rule_build(struct fw3_ipt_rule *r)
1268 {
1269 size_t s;
1270 struct xtables_rule_match *m;
1271
1272 #ifndef DISABLE_IPV6
1273 if (r->h->family == FW3_FAMILY_V6)
1274 {
1275 struct ip6t_entry *e6;
1276
1277 s = XT_ALIGN(sizeof(struct ip6t_entry));
1278
1279 for (m = r->matches; m; m = m->next)
1280 s += m->match->m->u.match_size;
1281
1282 e6 = fw3_alloc(s + r->target->t->u.target_size);
1283
1284 memcpy(e6, &r->e6, sizeof(struct ip6t_entry));
1285
1286 e6->target_offset = s;
1287 e6->next_offset = s + r->target->t->u.target_size;
1288
1289 s = 0;
1290
1291 for (m = r->matches; m; m = m->next)
1292 {
1293 memcpy(e6->elems + s, m->match->m, m->match->m->u.match_size);
1294 s += m->match->m->u.match_size;
1295 }
1296
1297 memcpy(e6->elems + s, r->target->t, r->target->t->u.target_size);
1298
1299 return e6;
1300 }
1301 else
1302 #endif
1303 {
1304 struct ipt_entry *e;
1305
1306 s = XT_ALIGN(sizeof(struct ipt_entry));
1307
1308 for (m = r->matches; m; m = m->next)
1309 s += m->match->m->u.match_size;
1310
1311 e = fw3_alloc(s + r->target->t->u.target_size);
1312
1313 memcpy(e, &r->e, sizeof(struct ipt_entry));
1314
1315 e->target_offset = s;
1316 e->next_offset = s + r->target->t->u.target_size;
1317
1318 s = 0;
1319
1320 for (m = r->matches; m; m = m->next)
1321 {
1322 memcpy(e->elems + s, m->match->m, m->match->m->u.match_size);
1323 s += m->match->m->u.match_size;
1324 }
1325
1326 memcpy(e->elems + s, r->target->t, r->target->t->u.target_size);
1327
1328 return e;
1329 }
1330 }
1331
1332 void
1333 __fw3_ipt_rule_append(struct fw3_ipt_rule *r, bool repl, const char *fmt, ...)
1334 {
1335 void *rule;
1336 unsigned char *mask;
1337
1338 struct xtables_rule_match *m;
1339 struct xtables_match *em;
1340 struct xtables_target *et;
1341 struct xtables_globals *g;
1342
1343 int i, optc;
1344 bool inv = false;
1345 char buf[32];
1346 va_list ap;
1347
1348 va_start(ap, fmt);
1349 vsnprintf(buf, sizeof(buf) - 1, fmt, ap);
1350 va_end(ap);
1351
1352 g = (r->h->family == FW3_FAMILY_V6) ? &xtg6 : &xtg;
1353 g->opts = g->orig_opts;
1354
1355 optind = 0;
1356 opterr = 0;
1357
1358 while ((optc = getopt_long(r->argc, r->argv, "-:m:j:", g->opts,
1359 NULL)) != -1)
1360 {
1361 switch (optc)
1362 {
1363 case 'm':
1364 em = find_match(r, optarg);
1365
1366 if (!em)
1367 {
1368 warn("fw3_ipt_rule_append(): Can't find match '%s'", optarg);
1369 goto free;
1370 }
1371
1372 init_match(r, em, true);
1373 break;
1374
1375 case 'j':
1376 et = get_target(r, optarg);
1377
1378 if (!et)
1379 {
1380 warn("fw3_ipt_rule_append(): Can't find target '%s'", optarg);
1381 goto free;
1382 }
1383
1384 break;
1385
1386 case 1:
1387 if ((optarg[0] == '!') && (optarg[1] == '\0'))
1388 {
1389 optarg[0] = '\0';
1390 inv = true;
1391 continue;
1392 }
1393
1394 warn("fw3_ipt_rule_append(): Bad argument '%s'", optarg);
1395 goto free;
1396
1397 default:
1398 if (parse_option(r, optc, inv))
1399 continue;
1400 break;
1401 }
1402
1403 inv = false;
1404 }
1405
1406 for (m = r->matches; m; m = m->next)
1407 xtables_option_mfcall(m->match);
1408
1409 if (r->target)
1410 xtables_option_tfcall(r->target);
1411
1412 rule = rule_build(r);
1413
1414 #ifndef DISABLE_IPV6
1415 if (r->h->family == FW3_FAMILY_V6)
1416 {
1417 if (repl)
1418 {
1419 mask = rule_mask(r);
1420
1421 while (ip6tc_delete_entry(buf, rule, mask, r->h->handle))
1422 if (fw3_pr_debug)
1423 rule_print(r, "-D", buf);
1424
1425 free(mask);
1426 }
1427
1428 if (fw3_pr_debug)
1429 rule_print(r, "-A", buf);
1430
1431 if (!ip6tc_append_entry(buf, rule, r->h->handle))
1432 warn("ip6tc_append_entry(): %s", ip6tc_strerror(errno));
1433 }
1434 else
1435 #endif
1436 {
1437 if (repl)
1438 {
1439 mask = rule_mask(r);
1440
1441 while (iptc_delete_entry(buf, rule, mask, r->h->handle))
1442 if (fw3_pr_debug)
1443 rule_print(r, "-D", buf);
1444
1445 free(mask);
1446 }
1447
1448 if (fw3_pr_debug)
1449 rule_print(r, "-A", buf);
1450
1451 if (!iptc_append_entry(buf, rule, r->h->handle))
1452 warn("iptc_append_entry(): %s\n", iptc_strerror(errno));
1453 }
1454
1455 free(rule);
1456
1457 free:
1458 for (i = 1; i < r->argc; i++)
1459 free(r->argv[i]);
1460
1461 free(r->argv);
1462
1463 xtables_rule_matches_free(&r->matches);
1464
1465 if (r->target)
1466 free(r->target->t);
1467
1468 free(r);
1469
1470 /* reset all targets and matches */
1471 for (em = xtables_matches; em; em = em->next)
1472 em->mflags = 0;
1473
1474 for (et = xtables_targets; et; et = et->next)
1475 {
1476 et->tflags = 0;
1477 et->used = 0;
1478 }
1479
1480 xtables_free_opts(1);
1481 }
1482
1483 struct fw3_ipt_rule *
1484 fw3_ipt_rule_create(struct fw3_ipt_handle *handle, struct fw3_protocol *proto,
1485 struct fw3_device *in, struct fw3_device *out,
1486 struct fw3_address *src, struct fw3_address *dest)
1487 {
1488 struct fw3_ipt_rule *r;
1489
1490 r = fw3_ipt_rule_new(handle);
1491
1492 fw3_ipt_rule_proto(r, proto);
1493 fw3_ipt_rule_in_out(r, in, out);
1494 fw3_ipt_rule_src_dest(r, src, dest);
1495
1496 return r;
1497 }