X-Git-Url: http://git.archive.openwrt.org/?p=project%2Ffirewall3.git;a=blobdiff_plain;f=utils.c;h=99310f731337879ee730f5a48a9c2c75522efdb4;hp=3558450d8831a96aa2a9ffa0763ea2a6a0eaf464;hb=d2a8a3f6a29746a3a07d2e3415737bcc2241afce;hpb=d5aa4e5d12fe80cf82da0ae4a0da595a8bfd83bb diff --git a/utils.c b/utils.c index 3558450..99310f7 100644 --- a/utils.c +++ b/utils.c @@ -132,6 +132,32 @@ info(const char* format, ...) fprintf(stderr, "\n"); } +void * +fw3_alloc(size_t size) +{ + void *mem; + + mem = calloc(1, size); + + if (!mem) + error("Out of memory while allocating %d bytes", size); + + return mem; +} + +char * +fw3_strdup(const char *s) +{ + char *ns; + + ns = strdup(s); + + if (!ns) + error("Out of memory while duplicating string '%s'", s); + + return ns; +} + const char * fw3_find_command(const char *cmd) { @@ -357,6 +383,21 @@ write_defaults_uci(struct uci_context *ctx, struct fw3_defaults *d, uci_add_section(ctx, dest, "defaults", &ptr.s); + ptr.o = NULL; + ptr.option = "input"; + ptr.value = fw3_flag_names[d->policy_input]; + uci_set(ctx, &ptr); + + ptr.o = NULL; + ptr.option = "output"; + ptr.value = fw3_flag_names[d->policy_output]; + uci_set(ctx, &ptr); + + ptr.o = NULL; + ptr.option = "forward"; + ptr.value = fw3_flag_names[d->policy_forward]; + uci_set(ctx, &ptr); + sprintf(buf, "0x%x", d->flags[0]); ptr.o = NULL; ptr.option = "__flags_v4"; @@ -455,7 +496,10 @@ write_zone_uci(struct uci_context *ctx, struct fw3_zone *z, if (dev->invert) p += sprintf(p, "!"); - p += sprintf(p, "%s", dev->name); + if (*dev->network) + p += sprintf(p, "%s@%s", dev->name, dev->network); + else + p += sprintf(p, "%s", dev->name); ptr.value = buf; uci_add_list(ctx, &ptr); @@ -490,20 +534,13 @@ static void write_ipset_uci(struct uci_context *ctx, struct fw3_ipset *s, struct uci_package *dest) { - enum fw3_family fam = FW3_FAMILY_ANY; + struct fw3_ipset_datatype *type; - char buf[sizeof("0xffffffff\0")]; + char buf[sizeof("65535-65535\0")]; struct uci_ptr ptr = { .p = dest }; - if (!s->enabled || (s->external && *s->external)) - return; - - if (fw3_no_family(s->flags[0]) && !fw3_no_family(s->flags[1])) - fam = FW3_FAMILY_V6; - else if (!fw3_no_family(s->flags[0]) && fw3_no_family(s->flags[1])) - fam = FW3_FAMILY_V4; - else if (fw3_no_family(s->flags[0]) && fw3_no_family(s->flags[1])) + if (!s->enabled || s->external) return; uci_add_section(ctx, dest, "ipset", &ptr.s); @@ -513,25 +550,38 @@ write_ipset_uci(struct uci_context *ctx, struct fw3_ipset *s, ptr.value = s->name; uci_set(ctx, &ptr); - if (fam != FW3_FAMILY_ANY) + ptr.o = NULL; + ptr.option = "storage"; + ptr.value = fw3_ipset_method_names[s->method]; + uci_set(ctx, &ptr); + + list_for_each_entry(type, &s->datatypes, list) { + sprintf(buf, "%s_%s", type->dest ? "dst" : "src", + fw3_ipset_type_names[type->type]); + ptr.o = NULL; - ptr.option = "family"; - ptr.value = fw3_flag_names[fam]; - uci_set(ctx, &ptr); + ptr.option = "match"; + ptr.value = buf; + uci_add_list(ctx, &ptr); } - sprintf(buf, "0x%x", s->flags[0]); - ptr.o = NULL; - ptr.option = "__flags_v4"; - ptr.value = buf; - uci_set(ctx, &ptr); + if (s->iprange.set) + { + ptr.o = NULL; + ptr.option = "iprange"; + ptr.value = fw3_address_to_string(&s->iprange, false); + uci_set(ctx, &ptr); + } - sprintf(buf, "0x%x", s->flags[1]); - ptr.o = NULL; - ptr.option = "__flags_v6"; - ptr.value = buf; - uci_set(ctx, &ptr); + if (s->portrange.set) + { + sprintf(buf, "%u-%u", s->portrange.port_min, s->portrange.port_max); + ptr.o = NULL; + ptr.option = "portrange"; + ptr.value = buf; + uci_set(ctx, &ptr); + } } void @@ -578,6 +628,7 @@ fw3_write_statefile(void *state) uci_unload(s->uci, p); } + fsync(fileno(sf)); fclose(sf); } } @@ -605,51 +656,30 @@ fw3_free_object(void *obj, const void *opts) free(obj); } - -bool -fw3_pr_rulespec(int table, int family, uint32_t *flags, uint32_t mask, - const struct fw3_rule_spec *r, const char *fmt, ...) +void +fw3_free_list(struct list_head *head) { - char buf[256]; - bool rv = false; - - va_list ap; - uint32_t f = flags ? flags[family == FW3_FAMILY_V6] : 0; + struct list_head *entry, *tmp; - if (mask) - f &= mask; + if (!head) + return; - for (; r->format; r++) + list_for_each_safe(entry, tmp, head) { - if (!fw3_is_family(r, family)) - continue; - - if (r->table != table) - continue; - - if ((r->flag != 0) && !hasbit(f, r->flag)) - continue; - - va_start(ap, fmt); - vsnprintf(buf, sizeof(buf), r->format, ap); - va_end(ap); - - fw3_pr(fmt, buf); - - rv = true; + list_del(entry); + free(entry); } - return rv; + free(head); } - bool fw3_hotplug(bool add, void *zone, void *device) { struct fw3_zone *z = zone; struct fw3_device *d = device; - if (!d->network) + if (!*d->network) return false; switch (fork()) @@ -673,7 +703,7 @@ fw3_hotplug(bool add, void *zone, void *device) clearenv(); setenv("ACTION", add ? "add" : "remove", 1); setenv("ZONE", z->name, 1); - setenv("INTERFACE", d->network->name, 1); + setenv("INTERFACE", d->network, 1); setenv("DEVICE", d->name, 1); execl(FW3_HOTPLUG, FW3_HOTPLUG, "firewall", NULL);