0cd90680efa707e930ab0276eb2d6b582e9d15bf
[project/firewall3.git] / options.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 "options.h"
20 #include "ubus.h"
21
22
23 static bool
24 put_value(void *ptr, void *val, int elem_size, bool is_list)
25 {
26         void *copy;
27
28         if (is_list)
29         {
30                 copy = malloc(elem_size);
31
32                 if (!copy)
33                         return false;
34
35                 memcpy(copy, val, elem_size);
36                 list_add_tail((struct list_head *)copy, (struct list_head *)ptr);
37                 return true;
38         }
39
40         memcpy(ptr, val, elem_size);
41         return false;
42 }
43
44 static bool
45 parse_enum(void *ptr, const char *val, const char **values, int min, int max)
46 {
47         int i, l = strlen(val);
48
49         if (l > 0)
50         {
51                 for (i = 0; i <= (max - min); i++)
52                 {
53                         if (!strncasecmp(val, values[i], l))
54                         {
55                                 *((int *)ptr) = min + i;
56                                 return true;
57                         }
58                 }
59         }
60
61         return false;
62 }
63
64
65 const char *fw3_flag_names[__FW3_FLAG_MAX] = {
66         "filter",
67         "nat",
68         "mangle",
69         "raw",
70
71         "IPv4",
72         "IPv6",
73
74         "ACCEPT",
75         "REJECT",
76         "DROP",
77         "NOTRACK",
78         "MARK",
79         "DNAT",
80         "SNAT",
81
82         "ACCEPT",
83         "REJECT",
84         "DROP",
85 };
86
87 const char *fw3_limit_units[__FW3_LIMIT_UNIT_MAX] = {
88         "second",
89         "minute",
90         "hour",
91         "day",
92 };
93
94 const char *fw3_ipset_method_names[__FW3_IPSET_METHOD_MAX] = {
95         "(bug)",
96         "bitmap",
97         "hash",
98         "list",
99 };
100
101 const char *fw3_ipset_type_names[__FW3_IPSET_TYPE_MAX] = {
102         "(bug)",
103         "ip",
104         "port",
105         "mac",
106         "net",
107         "set",
108 };
109
110 static const char *weekdays[] = {
111         "monday",
112         "tuesday",
113         "wednesday",
114         "thursday",
115         "friday",
116         "saturday",
117         "sunday",
118 };
119
120 static const char *include_types[] = {
121         "script",
122         "restore",
123 };
124
125 static const char *reflection_sources[] = {
126         "internal",
127         "external",
128 };
129
130
131 bool
132 fw3_parse_bool(void *ptr, const char *val, bool is_list)
133 {
134         if (!strcmp(val, "true") || !strcmp(val, "yes") || !strcmp(val, "1"))
135                 *((bool *)ptr) = true;
136         else
137                 *((bool *)ptr) = false;
138
139         return true;
140 }
141
142 bool
143 fw3_parse_int(void *ptr, const char *val, bool is_list)
144 {
145         int n = strtol(val, NULL, 0);
146
147         if (errno == ERANGE || errno == EINVAL)
148                 return false;
149
150         *((int *)ptr) = n;
151
152         return true;
153 }
154
155 bool
156 fw3_parse_string(void *ptr, const char *val, bool is_list)
157 {
158         *((char **)ptr) = (char *)val;
159         return true;
160 }
161
162 bool
163 fw3_parse_target(void *ptr, const char *val, bool is_list)
164 {
165         return parse_enum(ptr, val, &fw3_flag_names[FW3_FLAG_ACCEPT],
166                           FW3_FLAG_ACCEPT, FW3_FLAG_SNAT);
167 }
168
169 bool
170 fw3_parse_limit(void *ptr, const char *val, bool is_list)
171 {
172         struct fw3_limit *limit = ptr;
173         enum fw3_limit_unit u = FW3_LIMIT_UNIT_SECOND;
174         char *e;
175         int n;
176
177         if (*val == '!')
178         {
179                 limit->invert = true;
180                 while (isspace(*++val));
181         }
182
183         n = strtol(val, &e, 10);
184
185         if (errno == ERANGE || errno == EINVAL)
186                 return false;
187
188         if (*e && *e++ != '/')
189                 return false;
190
191         if (!strlen(e))
192                 return false;
193
194         if (!parse_enum(&u, e, fw3_limit_units, 0, FW3_LIMIT_UNIT_DAY))
195                 return false;
196
197         limit->rate = n;
198         limit->unit = u;
199
200         return true;
201 }
202
203 bool
204 fw3_parse_device(void *ptr, const char *val, bool is_list)
205 {
206         char *p;
207         struct fw3_device dev = { };
208
209         if (*val == '*')
210         {
211                 dev.set = true;
212                 dev.any = true;
213                 put_value(ptr, &dev, sizeof(dev), is_list);
214                 return true;
215         }
216
217         if (*val == '!')
218         {
219                 dev.invert = true;
220                 while (isspace(*++val));
221         }
222
223         if ((p = strchr(val, '@')) != NULL)
224         {
225                 *p++ = 0;
226                 snprintf(dev.network, sizeof(dev.network), "%s", p);
227         }
228
229         if (*val)
230                 snprintf(dev.name, sizeof(dev.name), "%s", val);
231         else
232                 return false;
233
234         dev.set = true;
235         put_value(ptr, &dev, sizeof(dev), is_list);
236         return true;
237 }
238
239 bool
240 fw3_parse_address(void *ptr, const char *val, bool is_list)
241 {
242         struct fw3_address addr = { };
243         struct in_addr v4;
244         struct in6_addr v6;
245         char *p, *s, *e;
246         int i, m = -1;
247
248         if (*val == '!')
249         {
250                 addr.invert = true;
251                 while (isspace(*++val));
252         }
253
254         s = strdup(val);
255
256         if (!s)
257                 return false;
258
259         if ((p = strchr(s, '/')) != NULL)
260         {
261                 *p++ = 0;
262                 m = strtoul(p, &e, 10);
263
264                 if ((e == p) || (*e != 0))
265                 {
266                         if (strchr(s, ':') || !inet_pton(AF_INET, p, &v4))
267                         {
268                                 free(s);
269                                 return false;
270                         }
271
272                         for (i = 0, m = 32; !(v4.s_addr & 1) && (i < 32); i++)
273                         {
274                                 m--;
275                                 v4.s_addr >>= 1;
276                         }
277                 }
278         }
279         else if ((p = strchr(s, '-')) != NULL)
280         {
281                 *p++ = 0;
282
283                 if (inet_pton(AF_INET6, p, &v6))
284                 {
285                         addr.family = FW3_FAMILY_V6;
286                         addr.address2.v6 = v6;
287                         addr.range = true;
288                 }
289                 else if (inet_pton(AF_INET, p, &v4))
290                 {
291                         addr.family = FW3_FAMILY_V4;
292                         addr.address2.v4 = v4;
293                         addr.range = true;
294                 }
295                 else
296                 {
297                         free(s);
298                         return false;
299                 }
300         }
301
302         if (inet_pton(AF_INET6, s, &v6))
303         {
304                 addr.family = FW3_FAMILY_V6;
305                 addr.address.v6 = v6;
306                 addr.mask = (m >= 0) ? m : 128;
307         }
308         else if (inet_pton(AF_INET, s, &v4))
309         {
310                 addr.family = FW3_FAMILY_V4;
311                 addr.address.v4 = v4;
312                 addr.mask = (m >= 0) ? m : 32;
313         }
314         else
315         {
316                 free(s);
317                 return false;
318         }
319
320         free(s);
321         addr.set = true;
322         put_value(ptr, &addr, sizeof(addr), is_list);
323         return true;
324 }
325
326 bool
327 fw3_parse_network(void *ptr, const char *val, bool is_list)
328 {
329         struct fw3_device dev = { };
330         struct fw3_address *addr;
331         struct list_head *addr_list;
332
333         if (!fw3_parse_address(ptr, val, is_list))
334         {
335                 if (!fw3_parse_device(&dev, val, false))
336                         return false;
337
338                 addr_list = fw3_ubus_address(dev.name);
339
340                 if (addr_list)
341                 {
342                         list_for_each_entry(addr, addr_list, list)
343                         {
344                                 addr->invert = dev.invert;
345
346                                 if (!put_value(ptr, addr, sizeof(*addr), is_list))
347                                         break;
348                         }
349
350                         fw3_ubus_address_free(addr_list);
351                 }
352         }
353
354         return true;
355 }
356
357 bool
358 fw3_parse_mac(void *ptr, const char *val, bool is_list)
359 {
360         struct fw3_mac addr = { };
361         struct ether_addr *mac;
362
363         if (*val == '!')
364         {
365                 addr.invert = true;
366                 while (isspace(*++val));
367         }
368
369         if ((mac = ether_aton(val)) != NULL)
370         {
371                 addr.mac = *mac;
372                 addr.set = true;
373
374                 put_value(ptr, &addr, sizeof(addr), is_list);
375                 return true;
376         }
377
378         return false;
379 }
380
381 bool
382 fw3_parse_port(void *ptr, const char *val, bool is_list)
383 {
384         struct fw3_port range = { };
385         uint16_t n;
386         uint16_t m;
387         char *p;
388
389         if (*val == '!')
390         {
391                 range.invert = true;
392                 while (isspace(*++val));
393         }
394
395         n = strtoul(val, &p, 10);
396
397         if (errno == ERANGE || errno == EINVAL)
398                 return false;
399
400         if (*p && *p != '-' && *p != ':')
401                 return false;
402
403         if (*p)
404         {
405                 m = strtoul(++p, NULL, 10);
406
407                 if (errno == ERANGE || errno == EINVAL || m < n)
408                         return false;
409
410                 range.port_min = n;
411                 range.port_max = m;
412         }
413         else
414         {
415                 range.port_min = n;
416                 range.port_max = n;
417         }
418
419         range.set = true;
420         put_value(ptr, &range, sizeof(range), is_list);
421         return true;
422 }
423
424 bool
425 fw3_parse_family(void *ptr, const char *val, bool is_list)
426 {
427         if (!strcmp(val, "any"))
428                 *((enum fw3_family *)ptr) = FW3_FAMILY_ANY;
429         else if (!strcmp(val, "inet") || strrchr(val, '4'))
430                 *((enum fw3_family *)ptr) = FW3_FAMILY_V4;
431         else if (!strcmp(val, "inet6") || strrchr(val, '6'))
432                 *((enum fw3_family *)ptr) = FW3_FAMILY_V6;
433         else
434                 return false;
435
436         return true;
437 }
438
439 bool
440 fw3_parse_icmptype(void *ptr, const char *val, bool is_list)
441 {
442         struct fw3_icmptype icmp = { };
443         bool v4 = false;
444         bool v6 = false;
445         char *p;
446         int i;
447
448         for (i = 0; i < ARRAY_SIZE(fw3_icmptype_list_v4); i++)
449         {
450                 if (!strcmp(val, fw3_icmptype_list_v4[i].name))
451                 {
452                         icmp.type     = fw3_icmptype_list_v4[i].type;
453                         icmp.code_min = fw3_icmptype_list_v4[i].code_min;
454                         icmp.code_max = fw3_icmptype_list_v4[i].code_max;
455
456                         v4 = true;
457                         break;
458                 }
459         }
460
461         for (i = 0; i < ARRAY_SIZE(fw3_icmptype_list_v6); i++)
462         {
463                 if (!strcmp(val, fw3_icmptype_list_v6[i].name))
464                 {
465                         icmp.type6     = fw3_icmptype_list_v6[i].type;
466                         icmp.code6_min = fw3_icmptype_list_v6[i].code_min;
467                         icmp.code6_max = fw3_icmptype_list_v6[i].code_max;
468
469                         v6 = true;
470                         break;
471                 }
472         }
473
474         if (!v4 && !v6)
475         {
476                 i = strtoul(val, &p, 10);
477
478                 if ((p == val) || (*p != '/' && *p != 0) || (i > 0xFF))
479                         return false;
480
481                 icmp.type = i;
482
483                 if (*p == '/')
484                 {
485                         val = ++p;
486                         i = strtoul(val, &p, 10);
487
488                         if ((p == val) || (*p != 0) || (i > 0xFF))
489                                 return false;
490
491                         icmp.code_min = i;
492                         icmp.code_max = i;
493                 }
494                 else
495                 {
496                         icmp.code_min = 0;
497                         icmp.code_max = 0xFF;
498                 }
499
500                 icmp.type6     = icmp.type;
501                 icmp.code6_min = icmp.code_max;
502                 icmp.code6_max = icmp.code_max;
503
504                 v4 = true;
505                 v6 = true;
506         }
507
508         icmp.family = (v4 && v6) ? FW3_FAMILY_ANY
509                                  : (v6 ? FW3_FAMILY_V6 : FW3_FAMILY_V4);
510
511         put_value(ptr, &icmp, sizeof(icmp), is_list);
512         return true;
513 }
514
515 bool
516 fw3_parse_protocol(void *ptr, const char *val, bool is_list)
517 {
518         struct fw3_protocol proto = { };
519         struct protoent *ent;
520
521         if (*val == '!')
522         {
523                 proto.invert = true;
524                 while (isspace(*++val));
525         }
526
527         if (!strcmp(val, "all"))
528         {
529                 proto.any = true;
530                 put_value(ptr, &proto, sizeof(proto), is_list);
531                 return true;
532         }
533         else if (!strcmp(val, "icmpv6"))
534         {
535                 val = "ipv6-icmp";
536         }
537         else if (!strcmp(val, "tcpudp"))
538         {
539                 proto.protocol = 6;
540                 if (put_value(ptr, &proto, sizeof(proto), is_list))
541                 {
542                         proto.protocol = 17;
543                         put_value(ptr, &proto, sizeof(proto), is_list);
544                 }
545
546                 return true;
547         }
548
549         ent = getprotobyname(val);
550
551         if (ent)
552         {
553                 proto.protocol = ent->p_proto;
554                 put_value(ptr, &proto, sizeof(proto), is_list);
555                 return true;
556         }
557
558         proto.protocol = strtoul(val, NULL, 10);
559
560         if (errno == ERANGE || errno == EINVAL)
561                 return false;
562
563         put_value(ptr, &proto, sizeof(proto), is_list);
564         return true;
565 }
566
567 bool
568 fw3_parse_ipset_method(void *ptr, const char *val, bool is_list)
569 {
570         return parse_enum(ptr, val, &fw3_ipset_method_names[FW3_IPSET_METHOD_BITMAP],
571                           FW3_IPSET_METHOD_BITMAP, FW3_IPSET_METHOD_LIST);
572 }
573
574 bool
575 fw3_parse_ipset_datatype(void *ptr, const char *val, bool is_list)
576 {
577         struct fw3_ipset_datatype type = { };
578
579         if (!strncmp(val, "dest_", 5))
580         {
581                 val += 5;
582                 type.dest = true;
583         }
584         else if (!strncmp(val, "dst_", 4))
585         {
586                 val += 4;
587                 type.dest = true;
588         }
589         else if (!strncmp(val, "src_", 4))
590         {
591                 val += 4;
592                 type.dest = false;
593         }
594
595         if (parse_enum(&type.type, val, &fw3_ipset_type_names[FW3_IPSET_TYPE_IP],
596                        FW3_IPSET_TYPE_IP, FW3_IPSET_TYPE_SET))
597         {
598                 put_value(ptr, &type, sizeof(type), is_list);
599                 return true;
600         }
601
602         return false;
603 }
604
605 bool
606 fw3_parse_date(void *ptr, const char *val, bool is_list)
607 {
608         unsigned int year = 1970, mon = 1, day = 1, hour = 0, min = 0, sec = 0;
609         struct tm tm = { 0 };
610         char *p;
611
612         year = strtoul(val, &p, 10);
613         if ((*p != '-' && *p) || year < 1970 || year > 2038)
614                 goto fail;
615         else if (!*p)
616                 goto ret;
617
618         mon = strtoul(++p, &p, 10);
619         if ((*p != '-' && *p) || mon > 12)
620                 goto fail;
621         else if (!*p)
622                 goto ret;
623
624         day = strtoul(++p, &p, 10);
625         if ((*p != 'T' && *p) || day > 31)
626                 goto fail;
627         else if (!*p)
628                 goto ret;
629
630         hour = strtoul(++p, &p, 10);
631         if ((*p != ':' && *p) || hour > 23)
632                 goto fail;
633         else if (!*p)
634                 goto ret;
635
636         min = strtoul(++p, &p, 10);
637         if ((*p != ':' && *p) || min > 59)
638                 goto fail;
639         else if (!*p)
640                 goto ret;
641
642         sec = strtoul(++p, &p, 10);
643         if (*p || sec > 59)
644                 goto fail;
645
646 ret:
647         tm.tm_year = year - 1900;
648         tm.tm_mon  = mon - 1;
649         tm.tm_mday = day;
650         tm.tm_hour = hour;
651         tm.tm_min  = min;
652         tm.tm_sec  = sec;
653
654         if (mktime(&tm) >= 0)
655         {
656                 *((struct tm *)ptr) = tm;
657                 return true;
658         }
659
660 fail:
661         return false;
662 }
663
664 bool
665 fw3_parse_time(void *ptr, const char *val, bool is_list)
666 {
667         unsigned int hour = 0, min = 0, sec = 0;
668         char *p;
669
670         hour = strtoul(val, &p, 10);
671         if (*p != ':' || hour > 23)
672                 goto fail;
673
674         min = strtoul(++p, &p, 10);
675         if ((*p != ':' && *p) || min > 59)
676                 goto fail;
677         else if (!*p)
678                 goto ret;
679
680         sec = strtoul(++p, &p, 10);
681         if (*p || sec > 59)
682                 goto fail;
683
684 ret:
685         *((int *)ptr) = 60 * 60 * hour + 60 * min + sec;
686         return true;
687
688 fail:
689         return false;
690 }
691
692 bool
693 fw3_parse_weekdays(void *ptr, const char *val, bool is_list)
694 {
695         unsigned int w = 0;
696         char *p, *s;
697
698         if (*val == '!')
699         {
700                 setbit(*(uint8_t *)ptr, 0);
701                 while (isspace(*++val));
702         }
703
704         if (!(s = strdup(val)))
705                 return false;
706
707         for (p = strtok(s, " \t"); p; p = strtok(NULL, " \t"))
708         {
709                 if (!parse_enum(&w, p, weekdays, 1, 7))
710                 {
711                         w = strtoul(p, &p, 10);
712
713                         if (*p || w < 1 || w > 7)
714                         {
715                                 free(s);
716                                 return false;
717                         }
718                 }
719
720                 setbit(*(uint8_t *)ptr, w);
721         }
722
723         free(s);
724         return true;
725 }
726
727 bool
728 fw3_parse_monthdays(void *ptr, const char *val, bool is_list)
729 {
730         unsigned int d;
731         char *p, *s;
732
733         if (*val == '!')
734         {
735                 setbit(*(uint32_t *)ptr, 0);
736                 while (isspace(*++val));
737         }
738
739         if (!(s = strdup(val)))
740                 return false;
741
742         for (p = strtok(s, " \t"); p; p = strtok(NULL, " \t"))
743         {
744                 d = strtoul(p, &p, 10);
745
746                 if (*p || d < 1 || d > 31)
747                 {
748                         free(s);
749                         return false;
750                 }
751
752                 setbit(*(uint32_t *)ptr, d);
753         }
754
755         free(s);
756         return true;
757 }
758
759 bool
760 fw3_parse_include_type(void *ptr, const char *val, bool is_list)
761 {
762         return parse_enum(ptr, val, include_types,
763                           FW3_INC_TYPE_SCRIPT, FW3_INC_TYPE_RESTORE);
764 }
765
766 bool
767 fw3_parse_reflection_source(void *ptr, const char *val, bool is_list)
768 {
769         return parse_enum(ptr, val, reflection_sources,
770                           FW3_REFLECTION_INTERNAL, FW3_REFLECTION_EXTERNAL);
771 }
772
773 bool
774 fw3_parse_mark(void *ptr, const char *val, bool is_list)
775 {
776         uint32_t n;
777         char *s, *e;
778         struct fw3_mark *m = ptr;
779
780         if (*val == '!')
781         {
782                 m->invert = true;
783                 while (isspace(*++val));
784         }
785
786         if ((s = strchr(val, '/')) != NULL)
787                 *s++ = 0;
788
789         n = strtoul(val, &e, 0);
790
791         if (e == val || *e)
792                 return false;
793
794         m->mark = n;
795         m->mask = 0xFFFFFFFF;
796
797         if (s)
798         {
799                 n = strtoul(s, &e, 0);
800
801                 if (e == s || *e)
802                         return false;
803
804                 m->mask = n;
805         }
806
807         m->set = true;
808         return true;
809 }
810
811
812 void
813 fw3_parse_options(void *s, const struct fw3_option *opts,
814                   struct uci_section *section)
815 {
816         char *p, *v;
817         bool known;
818         struct uci_element *e, *l;
819         struct uci_option *o;
820         const struct fw3_option *opt;
821         struct list_head *dest;
822
823         uci_foreach_element(&section->options, e)
824         {
825                 o = uci_to_option(e);
826                 known = false;
827
828                 for (opt = opts; opt->name; opt++)
829                 {
830                         if (!opt->parse)
831                                 continue;
832
833                         if (strcmp(opt->name, e->name))
834                                 continue;
835
836                         if (o->type == UCI_TYPE_LIST)
837                         {
838                                 if (!opt->elem_size)
839                                 {
840                                         warn_elem(e, "must not be a list");
841                                 }
842                                 else
843                                 {
844                                         dest = (struct list_head *)((char *)s + opt->offset);
845
846                                         uci_foreach_element(&o->v.list, l)
847                                         {
848                                                 if (!l->name)
849                                                         continue;
850
851                                                 if (!opt->parse(dest, l->name, true))
852                                                 {
853                                                         warn_elem(e, "has invalid value '%s'", l->name);
854                                                         continue;
855                                                 }
856                                         }
857                                 }
858                         }
859                         else
860                         {
861                                 v = o->v.string;
862
863                                 if (!v)
864                                         continue;
865
866                                 if (!opt->elem_size)
867                                 {
868                                         if (!opt->parse((char *)s + opt->offset, o->v.string, false))
869                                                 warn_elem(e, "has invalid value '%s'", o->v.string);
870                                 }
871                                 else
872                                 {
873                                         dest = (struct list_head *)((char *)s + opt->offset);
874
875                                         for (p = strtok(v, " \t"); p != NULL; p = strtok(NULL, " \t"))
876                                         {
877                                                 if (!opt->parse(dest, p, true))
878                                                 {
879                                                         warn_elem(e, "has invalid value '%s'", p);
880                                                         continue;
881                                                 }
882                                         }
883                                 }
884                         }
885
886                         known = true;
887                         break;
888                 }
889
890                 if (!known)
891                         warn_elem(e, "is unknown");
892         }
893 }
894
895
896 const char *
897 fw3_address_to_string(struct fw3_address *address, bool allow_invert)
898 {
899         char *p, ip[INET6_ADDRSTRLEN];
900         static char buf[INET6_ADDRSTRLEN * 2 + 2];
901
902         p = buf;
903
904         if (address->invert && allow_invert)
905                 p += sprintf(p, "!");
906
907         inet_ntop(address->family == FW3_FAMILY_V4 ? AF_INET : AF_INET6,
908                   &address->address.v4, ip, sizeof(ip));
909
910         p += sprintf(p, "%s", ip);
911
912         if (address->range)
913         {
914                 inet_ntop(address->family == FW3_FAMILY_V4 ? AF_INET : AF_INET6,
915                           &address->address2.v4, ip, sizeof(ip));
916
917                 p += sprintf(p, "-%s", ip);
918         }
919         else
920         {
921                 p += sprintf(p, "/%u", address->mask);
922         }
923
924         return buf;
925 }
926
927 void
928 fw3_format_src_dest(struct fw3_address *src, struct fw3_address *dest)
929 {
930         if ((src && src->range) || (dest && dest->range))
931                 fw3_pr(" -m iprange");
932
933         if (src && src->set)
934         {
935                 fw3_pr(" %s%s %s", src->invert ? "! " : "",
936                                    src->range ? "--src-range" : "-s",
937                                    fw3_address_to_string(src, false));
938         }
939
940         if (dest && dest->set)
941         {
942                 fw3_pr(" %s%s %s", dest->invert ? "! " : "",
943                                    dest->range ? "--dst-range" : "-d",
944                                    fw3_address_to_string(dest, false));
945         }
946 }
947
948 void
949 fw3_format_sport_dport(struct fw3_port *sp, struct fw3_port *dp)
950 {
951         if (sp && sp->set)
952         {
953                 if (sp->port_min == sp->port_max)
954                         fw3_pr(" %s--sport %u", sp->invert ? "! " : "", sp->port_min);
955                 else
956                         fw3_pr(" %s--sport %u:%u",
957                                sp->invert ? "! " : "", sp->port_min, sp->port_max);
958         }
959
960         if (dp && dp->set)
961         {
962                 if (dp->port_min == dp->port_max)
963                         fw3_pr(" %s--dport %u", dp->invert ? "! " : "", dp->port_min);
964                 else
965                         fw3_pr(" %s--dport %u:%u",
966                                dp->invert ? "! " : "", dp->port_min, dp->port_max);
967         }
968 }
969
970 void
971 fw3_format_mac(struct fw3_mac *mac)
972 {
973         if (!mac)
974                 return;
975
976         fw3_pr(" -m mac %s--mac-source %s",
977                mac->invert ? "! " : "", ether_ntoa(&mac->mac));
978 }
979
980 void
981 fw3_format_protocol(struct fw3_protocol *proto, enum fw3_family family)
982 {
983         uint16_t pr;
984
985         if (!proto)
986                 return;
987
988         pr = proto->protocol;
989
990         if (pr == 1 && family == FW3_FAMILY_V6)
991                 pr = 58;
992
993         if (proto->any)
994                 fw3_pr(" -p all");
995         else
996                 fw3_pr(" %s-p %u", proto->invert ? "! " : "", pr);
997 }
998
999 void
1000 fw3_format_icmptype(struct fw3_icmptype *icmp, enum fw3_family family)
1001 {
1002         if (!icmp)
1003                 return;
1004
1005         if (family != FW3_FAMILY_V6)
1006         {
1007                 if (icmp->code_min == 0 && icmp->code_max == 0xFF)
1008                         fw3_pr(" %s--icmp-type %u", icmp->invert ? "! " : "", icmp->type);
1009                 else
1010                         fw3_pr(" %s--icmp-type %u/%u",
1011                                    icmp->invert ? "! " : "", icmp->type, icmp->code_min);
1012         }
1013         else
1014         {
1015                 if (icmp->code6_min == 0 && icmp->code6_max == 0xFF)
1016                         fw3_pr(" %s--icmpv6-type %u", icmp->invert ? "! " : "", icmp->type6);
1017                 else
1018                         fw3_pr(" %s--icmpv6-type %u/%u",
1019                                    icmp->invert ? "! " : "", icmp->type6, icmp->code6_min);
1020         }
1021 }
1022
1023 void
1024 fw3_format_limit(struct fw3_limit *limit)
1025 {
1026         if (!limit)
1027                 return;
1028
1029         if (limit->rate > 0)
1030         {
1031                 fw3_pr(" -m limit %s--limit %u/%s",
1032                        limit->invert ? "! " : "",
1033                        limit->rate, fw3_limit_units[limit->unit]);
1034
1035                 if (limit->burst > 0)
1036                         fw3_pr(" --limit-burst %u", limit->burst);
1037         }
1038 }
1039
1040 void
1041 fw3_format_ipset(struct fw3_ipset *ipset, bool invert)
1042 {
1043         bool first = true;
1044         const char *name = NULL;
1045         struct fw3_ipset_datatype *type;
1046
1047         if (!ipset)
1048                 return;
1049
1050         name = ipset->external ? ipset->external : ipset->name;
1051
1052         fw3_pr(" -m set %s--match-set %s", invert ? "! " : "", name);
1053
1054         list_for_each_entry(type, &ipset->datatypes, list)
1055         {
1056                 fw3_pr("%c%s", first ? ' ' : ',', type->dest ? "dst" : "src");
1057                 first = false;
1058         }
1059 }
1060
1061 void
1062 fw3_format_time(struct fw3_time *time)
1063 {
1064         int i;
1065         struct tm empty = { 0 };
1066         char buf[sizeof("9999-99-99T23:59:59\0")];
1067         bool d1 = memcmp(&time->datestart, &empty, sizeof(empty));
1068         bool d2 = memcmp(&time->datestop, &empty, sizeof(empty));
1069         bool first;
1070
1071         if (!d1 && !d2 && !time->timestart && !time->timestop &&
1072             !(time->monthdays & 0xFFFFFFFE) && !(time->weekdays & 0xFE))
1073         {
1074                 return;
1075         }
1076
1077         fw3_pr(" -m time");
1078
1079         if (time->utc)
1080                 fw3_pr(" --utc");
1081
1082         if (d1)
1083         {
1084                 strftime(buf, sizeof(buf), "%Y-%m-%dT%H:%M:%S", &time->datestart);
1085                 fw3_pr(" --datestart %s", buf);
1086         }
1087
1088         if (d2)
1089         {
1090                 strftime(buf, sizeof(buf), "%Y-%m-%dT%H:%M:%S", &time->datestop);
1091                 fw3_pr(" --datestop %s", buf);
1092         }
1093
1094         if (time->timestart)
1095         {
1096                 fw3_pr(" --timestart %02d:%02d:%02d",
1097                        time->timestart / 3600,
1098                        time->timestart % 3600 / 60,
1099                        time->timestart % 60);
1100         }
1101
1102         if (time->timestop)
1103         {
1104                 fw3_pr(" --timestop %02d:%02d:%02d",
1105                        time->timestop / 3600,
1106                        time->timestop % 3600 / 60,
1107                        time->timestop % 60);
1108         }
1109
1110         if (time->monthdays & 0xFFFFFFFE)
1111         {
1112                 fw3_pr(" %s--monthdays", hasbit(time->monthdays, 0) ? "! " : "");
1113
1114                 for (i = 1, first = true; i < 32; i++)
1115                 {
1116                         if (hasbit(time->monthdays, i))
1117                         {
1118                                 fw3_pr("%c%u", first ? ' ' : ',', i);
1119                                 first = false;
1120                         }
1121                 }
1122         }
1123
1124         if (time->weekdays & 0xFE)
1125         {
1126                 fw3_pr(" %s--weekdays", hasbit(time->weekdays, 0) ? "! " : "");
1127
1128                 for (i = 1, first = true; i < 8; i++)
1129                 {
1130                         if (hasbit(time->weekdays, i))
1131                         {
1132                                 fw3_pr("%c%u", first ? ' ' : ',', i);
1133                                 first = false;
1134                         }
1135                 }
1136         }
1137 }
1138
1139 void
1140 fw3_format_mark(struct fw3_mark *mark)
1141 {
1142         if (!mark->set)
1143                 return;
1144
1145         fw3_pr(" -m mark %s--mark 0x%x", mark->invert ? "! " : "", mark->mark);
1146
1147         if (mark->mask < 0xFFFFFFFF)
1148                 fw3_pr("/0x%x", mark->mask);
1149 }
1150
1151 void
1152 __fw3_format_comment(const char *comment, ...)
1153 {
1154         va_list ap;
1155         int len = 0;
1156         const char *c;
1157
1158         if (!comment || !*comment)
1159                 return;
1160
1161         fw3_pr(" -m comment --comment \"");
1162
1163         c = comment;
1164
1165         va_start(ap, comment);
1166
1167         do
1168         {
1169                 while (*c)
1170                 {
1171                         switch (*c)
1172                         {
1173                         case '"':
1174                         case '$':
1175                         case '`':
1176                         case '\\':
1177                                 fw3_pr("\\");
1178                                 /* fall through */
1179
1180                         default:
1181                                 fw3_pr("%c", *c);
1182                                 break;
1183                         }
1184
1185                         c++;
1186
1187                         if (len++ >= 255)
1188                                 goto end;
1189                 }
1190
1191                 c = va_arg(ap, const char *);
1192         }
1193         while (c);
1194
1195 end:
1196         va_end(ap);
1197         fw3_pr("\"");
1198 }
1199
1200 void
1201 fw3_format_extra(const char *extra)
1202 {
1203         if (!extra || !*extra)
1204                 return;
1205
1206         fw3_pr(" %s", extra);
1207 }