* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
+#define _GNU_SOURCE /* RTLD_NEXT */
+
+/* include userspace headers */
+#include <dlfcn.h>
+#include <unistd.h>
+#include <getopt.h>
+#include <net/if.h>
+#include <netinet/in.h>
+#include <sys/utsname.h>
+#include <sys/socket.h>
+
+/* prevent indirect inclusion of kernel headers */
+#define _LINUX_IF_H
+#define _LINUX_IN_H
+#define _LINUX_IN6_H
+
+/* prevent libiptc from including kernel headers */
+#define _FWCHAINS_KERNEL_HEADERS_H
+
+/* finally include libiptc and xtables */
+#include <libiptc/libiptc.h>
+#include <libiptc/libip6tc.h>
+#include <xtables.h>
+
+#include "options.h"
+
+/* xtables interface */
+#if (XTABLES_VERSION_CODE >= 10)
+# include "xtables-10.h"
+#elif (XTABLES_VERSION_CODE == 5)
+# include "xtables-5.h"
+#else
+# error "Unsupported xtables version"
+#endif
+
#include "iptables.h"
+struct fw3_ipt_rule {
+ struct fw3_ipt_handle *h;
+
+ union {
+ struct ipt_entry e;
+ struct ip6t_entry e6;
+ };
+
+ struct xtables_rule_match *matches;
+ struct xtables_target *target;
+
+ int argc;
+ char **argv;
+
+ uint32_t protocol;
+ bool protocol_loaded;
+};
+
static struct option base_opts[] = {
{ .name = "match", .has_arg = 1, .val = 'm' },
{ .name = "jump", .has_arg = 1, .val = 'j' },
.option_offset = 0,
.program_version = "4",
.orig_opts = base_opts,
+#if XTABLES_VERSION_CODE > 10
+ .compat_rev = xtables_compatible_revision,
+#endif
};
static struct xtables_globals xtg6 = {
.option_offset = 0,
.program_version = "6",
.orig_opts = base_opts,
+#if XTABLES_VERSION_CODE > 10
+ .compat_rev = xtables_compatible_revision,
+#endif
};
+static struct {
+ bool retain;
+ int mcount, tcount;
+ struct xtables_match **matches;
+ struct xtables_target **targets;
+ void (*register_match)(struct xtables_match *);
+ void (*register_target)(struct xtables_target *);
+} xext;
+
+
/* Required by certain extensions like SNAT and DNAT */
int kernel_version = 0;
struct fw3_ipt_handle *
fw3_ipt_open(enum fw3_family family, enum fw3_table table)
{
+ int i;
struct fw3_ipt_handle *h;
h = fw3_alloc(sizeof(*h));
fw3_xt_reset();
fw3_init_extensions();
+ if (xext.register_match)
+ for (i = 0; i < xext.mcount; i++)
+ xext.register_match(xext.matches[i]);
+
+ if (xext.register_target)
+ for (i = 0; i < xext.tcount; i++)
+ xext.register_target(xext.targets[i]);
+
return h;
}
iptc_delete_chain(chain, h->handle);
}
-static int
-get_rule_id(const void *base, unsigned int start, unsigned int end)
+static bool
+has_rule_tag(const void *base, unsigned int start, unsigned int end)
{
- uint32_t id;
unsigned int i;
const struct xt_entry_match *em;
{
em = base + i;
- if (strcmp(em->u.user.name, "id"))
- continue;
-
- memcpy(&id, em->data, sizeof(id));
-
- if ((id & FW3_ID_MASK) != FW3_ID_MAGIC)
+ if (strcmp(em->u.user.name, "comment"))
continue;
- return (id & ~FW3_ID_MASK);
+ if (!memcmp(em->data, "!fw3", 4))
+ return true;
}
- return -1;
+ return false;
}
void
unsigned int num;
const struct ipt_entry *e;
bool found;
- int id;
#ifndef DISABLE_IPV6
if (h->family == FW3_FAMILY_V6)
e6 != NULL;
num++, e6 = ip6tc_next_rule(e6, h->handle))
{
- id = get_rule_id(e6, sizeof(*e6), e6->target_offset);
-
- if (id >= 0)
+ if (has_rule_tag(e6, sizeof(*e6), e6->target_offset))
{
if (fw3_pr_debug)
debug(h, "-D %s %u\n", chain, num + 1);
e != NULL;
num++, e = iptc_next_rule(e, h->handle))
{
- id = get_rule_id(e, sizeof(*e), e->target_offset);
-
- if (id >= 0)
+ if (has_rule_tag(e, sizeof(*e), e->target_offset))
{
if (fw3_pr_debug)
debug(h, "-D %s %u\n", chain, num + 1);
void
fw3_ipt_close(struct fw3_ipt_handle *h)
{
- if (h->libv)
- {
- while (h->libc > 0)
- {
- h->libc--;
- dlclose(h->libv[h->libc]);
- }
-
- free(h->libv);
- }
-
free(h);
}
r = fw3_alloc(sizeof(*r));
r->h = h;
- r->id = 0;
r->argv = fw3_alloc(sizeof(char *));
r->argv[r->argc++] = "fw3";
return NULL;
}
-static bool
-load_extension(struct fw3_ipt_handle *h, const char *name)
-{
- char path[256];
- void *lib, **tmp;
- const char *pfx = (h->family == FW3_FAMILY_V6) ? "libip6t" : "libipt";
-
- snprintf(path, sizeof(path), "/usr/lib/iptables/libxt_%s.so", name);
- if (!(lib = dlopen(path, RTLD_NOW)))
- {
- snprintf(path, sizeof(path), "/usr/lib/iptables/%s_%s.so", pfx, name);
- lib = dlopen(path, RTLD_NOW);
- }
-
- if (!lib)
- return false;
-
- tmp = realloc(h->libv, sizeof(lib) * (h->libc + 1));
-
- if (!tmp)
- return false;
-
- h->libv = tmp;
- h->libv[h->libc++] = lib;
-
- return true;
-}
-
static struct xtables_match *
find_match(struct fw3_ipt_rule *r, const char *name)
{
struct xtables_match *m;
- m = xtables_find_match(name, XTF_DONT_LOAD, &r->matches);
-
- if (!m && load_extension(r->h, name))
- m = xtables_find_match(name, XTF_DONT_LOAD, &r->matches);
+ xext.retain = true;
+ m = xtables_find_match(name, XTF_TRY_LOAD, &r->matches);
+ xext.retain = false;
return m;
}
{
struct xtables_target *t;
- if (is_chain(r->h, name))
- return xtables_find_target(XT_STANDARD_TARGET, XTF_LOAD_MUST_SUCCEED);
+ xext.retain = true;
- t = xtables_find_target(name, XTF_DONT_LOAD);
+ if (is_chain(r->h, name))
+ t = xtables_find_target(XT_STANDARD_TARGET, XTF_TRY_LOAD);
+ else
+ t = xtables_find_target(name, XTF_TRY_LOAD);
- if (!t && load_extension(r->h, name))
- t = xtables_find_target(name, XTF_DONT_LOAD);
+ xext.retain = false;
return t;
}
fw3_ipt_rule_addarg(r, false, "-m", "time");
- if (time->utc)
- fw3_ipt_rule_addarg(r, false, "--utc", NULL);
+ if (!time->utc)
+ fw3_ipt_rule_addarg(r, false, "--kerneltz", NULL);
if (d1)
{
}
}
+static void
+set_rule_tag(struct fw3_ipt_rule *r)
+{
+ int i;
+ char *p, **tmp;
+ const char *tag = "!fw3";
+
+ for (i = 0; i < r->argc; i++)
+ if (!strcmp(r->argv[i], "--comment") && (i + 1) < r->argc)
+ if (asprintf(&p, "%s: %s", tag, r->argv[i + 1]) > 0)
+ {
+ free(r->argv[i + 1]);
+ r->argv[i + 1] = p;
+ return;
+ }
+
+ tmp = realloc(r->argv, (r->argc + 4) * sizeof(*r->argv));
+
+ if (tmp)
+ {
+ r->argv = tmp;
+ r->argv[r->argc++] = fw3_strdup("-m");
+ r->argv[r->argc++] = fw3_strdup("comment");
+ r->argv[r->argc++] = fw3_strdup("--comment");
+ r->argv[r->argc++] = fw3_strdup(tag);
+ }
+}
+
void
__fw3_ipt_rule_append(struct fw3_ipt_rule *r, bool repl, const char *fmt, ...)
{
struct xtables_globals *g;
int i, optc;
- uint32_t id;
bool inv = false;
char buf[32];
va_list ap;
optind = 0;
opterr = 0;
- if (r->id >= 0)
- {
- em = find_match(r, "id");
-
- if (!em)
- {
- warn("fw3_ipt_rule_append(): Can't find match '%s'", "id");
- goto free;
- }
-
- init_match(r, em, true);
-
- id = FW3_ID_MAGIC | (r->id & ~FW3_ID_MASK);
- memcpy(em->m->data, &id, sizeof(id));
-
- em->mflags = 1;
- }
+ set_rule_tag(r);
while ((optc = getopt_long(r->argc, r->argv, "-:m:j:", g->opts,
NULL)) != -1)
return r;
}
+
+void
+xtables_register_match(struct xtables_match *me)
+{
+ int i;
+ static struct xtables_match **tmp;
+
+ if (!xext.register_match)
+ xext.register_match = dlsym(RTLD_NEXT, "xtables_register_match");
+
+ if (!xext.register_match)
+ return;
+
+ xext.register_match(me);
+
+ if (xext.retain)
+ {
+ for (i = 0; i < xext.mcount; i++)
+ if (xext.matches[i] == me)
+ return;
+
+ tmp = realloc(xext.matches, sizeof(me) * (xext.mcount + 1));
+
+ if (!tmp)
+ return;
+
+ xext.matches = tmp;
+ xext.matches[xext.mcount++] = me;
+ }
+}
+
+void
+xtables_register_target(struct xtables_target *me)
+{
+ int i;
+ static struct xtables_target **tmp;
+
+ if (!xext.register_target)
+ xext.register_target = dlsym(RTLD_NEXT, "xtables_register_target");
+
+ if (!xext.register_target)
+ return;
+
+ xext.register_target(me);
+
+ if (xext.retain)
+ {
+ for (i = 0; i < xext.tcount; i++)
+ if (xext.targets[i] == me)
+ return;
+
+ tmp = realloc(xext.targets, sizeof(me) * (xext.tcount + 1));
+
+ if (!tmp)
+ return;
+
+ xext.targets = tmp;
+ xext.targets[xext.tcount++] = me;
+ }
+}
projects / project / firewall3.git / blobdiff