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