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