|
Packit |
7b22a4 |
#include <stdio.h>
|
|
Packit |
7b22a4 |
#include <xtables.h>
|
|
Packit |
7b22a4 |
#include <linux/netfilter/xt_cpu.h>
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
enum {
|
|
Packit |
7b22a4 |
O_CPU = 0,
|
|
Packit |
7b22a4 |
};
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
static void cpu_help(void)
|
|
Packit |
7b22a4 |
{
|
|
Packit |
7b22a4 |
printf(
|
|
Packit |
7b22a4 |
"cpu match options:\n"
|
|
Packit |
7b22a4 |
"[!] --cpu number Match CPU number\n");
|
|
Packit |
7b22a4 |
}
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
static const struct xt_option_entry cpu_opts[] = {
|
|
Packit |
7b22a4 |
{.name = "cpu", .id = O_CPU, .type = XTTYPE_UINT32,
|
|
Packit |
7b22a4 |
.flags = XTOPT_INVERT | XTOPT_MAND | XTOPT_PUT,
|
|
Packit |
7b22a4 |
XTOPT_POINTER(struct xt_cpu_info, cpu)},
|
|
Packit |
7b22a4 |
XTOPT_TABLEEND,
|
|
Packit |
7b22a4 |
};
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
static void cpu_parse(struct xt_option_call *cb)
|
|
Packit |
7b22a4 |
{
|
|
Packit |
7b22a4 |
struct xt_cpu_info *cpuinfo = cb->data;
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
xtables_option_parse(cb);
|
|
Packit |
7b22a4 |
if (cb->invert)
|
|
Packit |
7b22a4 |
cpuinfo->invert = true;
|
|
Packit |
7b22a4 |
}
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
static void
|
|
Packit |
7b22a4 |
cpu_print(const void *ip, const struct xt_entry_match *match, int numeric)
|
|
Packit |
7b22a4 |
{
|
|
Packit |
7b22a4 |
const struct xt_cpu_info *info = (void *)match->data;
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
printf(" cpu %s%u", info->invert ? "! ":"", info->cpu);
|
|
Packit |
7b22a4 |
}
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
static void cpu_save(const void *ip, const struct xt_entry_match *match)
|
|
Packit |
7b22a4 |
{
|
|
Packit |
7b22a4 |
const struct xt_cpu_info *info = (void *)match->data;
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
printf("%s --cpu %u", info->invert ? " !" : "", info->cpu);
|
|
Packit |
7b22a4 |
}
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
static int cpu_xlate(struct xt_xlate *xl,
|
|
Packit |
7b22a4 |
const struct xt_xlate_mt_params *params)
|
|
Packit |
7b22a4 |
{
|
|
Packit |
7b22a4 |
const struct xt_cpu_info *info = (void *)params->match->data;
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
xt_xlate_add(xl, "cpu%s %u", info->invert ? " !=" : "", info->cpu);
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
return 1;
|
|
Packit |
7b22a4 |
}
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
static struct xtables_match cpu_match = {
|
|
Packit |
7b22a4 |
.family = NFPROTO_UNSPEC,
|
|
Packit |
7b22a4 |
.name = "cpu",
|
|
Packit |
7b22a4 |
.version = XTABLES_VERSION,
|
|
Packit |
7b22a4 |
.size = XT_ALIGN(sizeof(struct xt_cpu_info)),
|
|
Packit |
7b22a4 |
.userspacesize = XT_ALIGN(sizeof(struct xt_cpu_info)),
|
|
Packit |
7b22a4 |
.help = cpu_help,
|
|
Packit |
7b22a4 |
.print = cpu_print,
|
|
Packit |
7b22a4 |
.save = cpu_save,
|
|
Packit |
7b22a4 |
.x6_parse = cpu_parse,
|
|
Packit |
7b22a4 |
.x6_options = cpu_opts,
|
|
Packit |
7b22a4 |
.xlate = cpu_xlate,
|
|
Packit |
7b22a4 |
};
|
|
Packit |
7b22a4 |
|
|
Packit |
7b22a4 |
void _init(void)
|
|
Packit |
7b22a4 |
{
|
|
Packit |
7b22a4 |
xtables_register_match(&cpu_match);
|
|
Packit |
7b22a4 |
}
|