|
Packit Service |
fa126c |
#include <errno.h>
|
|
Packit Service |
fa126c |
#include <stdbool.h>
|
|
Packit Service |
fa126c |
#include <string.h>
|
|
Packit Service |
fa126c |
#include <stdio.h>
|
|
Packit Service |
fa126c |
#include <stdint.h>
|
|
Packit Service |
fa126c |
#include <xtables.h>
|
|
Packit Service |
fa126c |
#include <linux/netfilter/xt_connlabel.h>
|
|
Packit Service |
fa126c |
#include <libnetfilter_conntrack/libnetfilter_conntrack.h>
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
enum {
|
|
Packit Service |
fa126c |
O_LABEL = 0,
|
|
Packit Service |
fa126c |
O_SET = 1,
|
|
Packit Service |
fa126c |
};
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static struct nfct_labelmap *map;
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static void connlabel_mt_help(void)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
puts(
|
|
Packit Service |
fa126c |
"connlabel match options:\n"
|
|
Packit Service |
fa126c |
"[!] --label name Match if label has been set on connection\n"
|
|
Packit Service |
fa126c |
" --set Set label on connection");
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static const struct xt_option_entry connlabel_mt_opts[] = {
|
|
Packit Service |
fa126c |
{.name = "label", .id = O_LABEL, .type = XTTYPE_STRING,
|
|
Packit Service |
fa126c |
.min = 1, .flags = XTOPT_MAND|XTOPT_INVERT},
|
|
Packit Service |
fa126c |
{.name = "set", .id = O_SET, .type = XTTYPE_NONE},
|
|
Packit Service |
fa126c |
XTOPT_TABLEEND,
|
|
Packit Service |
fa126c |
};
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
/* cannot do this via _init, else static builds might spew error message
|
|
Packit Service |
fa126c |
* for every iptables invocation.
|
|
Packit Service |
fa126c |
*/
|
|
Packit Service |
fa126c |
static void connlabel_open(void)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
const char *fname;
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
if (map)
|
|
Packit Service |
fa126c |
return;
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
map = nfct_labelmap_new(NULL);
|
|
Packit Service |
fa126c |
if (map != NULL)
|
|
Packit Service |
fa126c |
return;
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
fname = nfct_labels_get_path();
|
|
Packit Service |
fa126c |
if (errno) {
|
|
Packit Service |
fa126c |
xtables_error(RESOURCE_PROBLEM,
|
|
Packit Service |
fa126c |
"cannot open %s: %s", fname, strerror(errno));
|
|
Packit Service |
fa126c |
} else {
|
|
Packit Service |
fa126c |
xtables_error(RESOURCE_PROBLEM,
|
|
Packit Service |
fa126c |
"cannot parse %s: no labels found", fname);
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static void connlabel_mt_parse(struct xt_option_call *cb)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
struct xt_connlabel_mtinfo *info = cb->data;
|
|
Packit Service |
fa126c |
int tmp;
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
connlabel_open();
|
|
Packit Service |
fa126c |
xtables_option_parse(cb);
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
switch (cb->entry->id) {
|
|
Packit Service |
fa126c |
case O_LABEL:
|
|
Packit Service |
fa126c |
tmp = nfct_labelmap_get_bit(map, cb->arg);
|
|
Packit Service |
fa126c |
if (tmp < 0)
|
|
Packit Service |
fa126c |
xtables_error(PARAMETER_PROBLEM, "label '%s' not found", cb->arg);
|
|
Packit Service |
fa126c |
info->bit = tmp;
|
|
Packit Service |
fa126c |
if (cb->invert)
|
|
Packit Service |
fa126c |
info->options |= XT_CONNLABEL_OP_INVERT;
|
|
Packit Service |
fa126c |
break;
|
|
Packit Service |
fa126c |
case O_SET:
|
|
Packit Service |
fa126c |
info->options |= XT_CONNLABEL_OP_SET;
|
|
Packit Service |
fa126c |
break;
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static const char *connlabel_get_name(int b)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
const char *name;
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
connlabel_open();
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
name = nfct_labelmap_get_name(map, b);
|
|
Packit Service |
fa126c |
if (name && strcmp(name, ""))
|
|
Packit Service |
fa126c |
return name;
|
|
Packit Service |
fa126c |
return NULL;
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static void
|
|
Packit Service |
fa126c |
connlabel_mt_print_op(const struct xt_connlabel_mtinfo *info, const char *prefix)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
if (info->options & XT_CONNLABEL_OP_SET)
|
|
Packit Service |
fa126c |
printf(" %sset", prefix);
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static void
|
|
Packit Service |
fa126c |
connlabel_mt_print(const void *ip, const struct xt_entry_match *match, int numeric)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
const struct xt_connlabel_mtinfo *info = (const void *)match->data;
|
|
Packit Service |
fa126c |
const char *name = connlabel_get_name(info->bit);
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
printf(" connlabel");
|
|
Packit Service |
fa126c |
if (info->options & XT_CONNLABEL_OP_INVERT)
|
|
Packit Service |
fa126c |
printf(" !");
|
|
Packit Service |
fa126c |
if (numeric || name == NULL) {
|
|
Packit Service |
fa126c |
printf(" %u", info->bit);
|
|
Packit Service |
fa126c |
} else {
|
|
Packit Service |
fa126c |
printf(" '%s'", name);
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
connlabel_mt_print_op(info, "");
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static void
|
|
Packit Service |
fa126c |
connlabel_mt_save(const void *ip, const struct xt_entry_match *match)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
const struct xt_connlabel_mtinfo *info = (const void *)match->data;
|
|
Packit Service |
fa126c |
const char *name = connlabel_get_name(info->bit);
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
if (info->options & XT_CONNLABEL_OP_INVERT)
|
|
Packit Service |
fa126c |
printf(" !");
|
|
Packit Service |
fa126c |
if (name)
|
|
Packit Service |
fa126c |
printf(" --label \"%s\"", name);
|
|
Packit Service |
fa126c |
else
|
|
Packit Service |
fa126c |
printf(" --label \"%u\"", info->bit);
|
|
Packit Service |
fa126c |
connlabel_mt_print_op(info, "--");
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static int connlabel_mt_xlate(struct xt_xlate *xl,
|
|
Packit Service |
fa126c |
const struct xt_xlate_mt_params *params)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
const struct xt_connlabel_mtinfo *info =
|
|
Packit Service |
fa126c |
(const void *)params->match->data;
|
|
Packit Service |
fa126c |
const char *name = connlabel_get_name(info->bit);
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
if (name == NULL)
|
|
Packit Service |
fa126c |
return 0;
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
if (info->options & XT_CONNLABEL_OP_SET)
|
|
Packit Service |
fa126c |
xt_xlate_add(xl, "ct label set %s ", name);
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
xt_xlate_add(xl, "ct label ");
|
|
Packit Service |
fa126c |
if (info->options & XT_CONNLABEL_OP_INVERT)
|
|
Packit Service |
fa126c |
xt_xlate_add(xl, "and %s != ", name);
|
|
Packit Service |
fa126c |
xt_xlate_add(xl, "%s", name);
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
return 1;
|
|
Packit Service |
fa126c |
}
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
static struct xtables_match connlabel_mt_reg = {
|
|
Packit Service |
fa126c |
.family = NFPROTO_UNSPEC,
|
|
Packit Service |
fa126c |
.name = "connlabel",
|
|
Packit Service |
fa126c |
.version = XTABLES_VERSION,
|
|
Packit Service |
fa126c |
.size = XT_ALIGN(sizeof(struct xt_connlabel_mtinfo)),
|
|
Packit Service |
fa126c |
.userspacesize = offsetof(struct xt_connlabel_mtinfo, bit),
|
|
Packit Service |
fa126c |
.help = connlabel_mt_help,
|
|
Packit Service |
fa126c |
.print = connlabel_mt_print,
|
|
Packit Service |
fa126c |
.save = connlabel_mt_save,
|
|
Packit Service |
fa126c |
.x6_parse = connlabel_mt_parse,
|
|
Packit Service |
fa126c |
.x6_options = connlabel_mt_opts,
|
|
Packit Service |
fa126c |
.xlate = connlabel_mt_xlate,
|
|
Packit Service |
fa126c |
};
|
|
Packit Service |
fa126c |
|
|
Packit Service |
fa126c |
void _init(void)
|
|
Packit Service |
fa126c |
{
|
|
Packit Service |
fa126c |
xtables_register_match(&connlabel_mt_reg);
|
|
Packit Service |
fa126c |
}
|