1*a71a9546SAutomerger Merge Worker #include <stdio.h>
2*a71a9546SAutomerger Merge Worker #include <xtables.h>
3*a71a9546SAutomerger Merge Worker #include <linux/netfilter/xt_cpu.h>
4*a71a9546SAutomerger Merge Worker
5*a71a9546SAutomerger Merge Worker enum {
6*a71a9546SAutomerger Merge Worker O_CPU = 0,
7*a71a9546SAutomerger Merge Worker };
8*a71a9546SAutomerger Merge Worker
cpu_help(void)9*a71a9546SAutomerger Merge Worker static void cpu_help(void)
10*a71a9546SAutomerger Merge Worker {
11*a71a9546SAutomerger Merge Worker printf(
12*a71a9546SAutomerger Merge Worker "cpu match options:\n"
13*a71a9546SAutomerger Merge Worker "[!] --cpu number Match CPU number\n");
14*a71a9546SAutomerger Merge Worker }
15*a71a9546SAutomerger Merge Worker
16*a71a9546SAutomerger Merge Worker static const struct xt_option_entry cpu_opts[] = {
17*a71a9546SAutomerger Merge Worker {.name = "cpu", .id = O_CPU, .type = XTTYPE_UINT32,
18*a71a9546SAutomerger Merge Worker .flags = XTOPT_INVERT | XTOPT_MAND | XTOPT_PUT,
19*a71a9546SAutomerger Merge Worker XTOPT_POINTER(struct xt_cpu_info, cpu)},
20*a71a9546SAutomerger Merge Worker XTOPT_TABLEEND,
21*a71a9546SAutomerger Merge Worker };
22*a71a9546SAutomerger Merge Worker
cpu_parse(struct xt_option_call * cb)23*a71a9546SAutomerger Merge Worker static void cpu_parse(struct xt_option_call *cb)
24*a71a9546SAutomerger Merge Worker {
25*a71a9546SAutomerger Merge Worker struct xt_cpu_info *cpuinfo = cb->data;
26*a71a9546SAutomerger Merge Worker
27*a71a9546SAutomerger Merge Worker xtables_option_parse(cb);
28*a71a9546SAutomerger Merge Worker if (cb->invert)
29*a71a9546SAutomerger Merge Worker cpuinfo->invert = true;
30*a71a9546SAutomerger Merge Worker }
31*a71a9546SAutomerger Merge Worker
32*a71a9546SAutomerger Merge Worker static void
cpu_print(const void * ip,const struct xt_entry_match * match,int numeric)33*a71a9546SAutomerger Merge Worker cpu_print(const void *ip, const struct xt_entry_match *match, int numeric)
34*a71a9546SAutomerger Merge Worker {
35*a71a9546SAutomerger Merge Worker const struct xt_cpu_info *info = (void *)match->data;
36*a71a9546SAutomerger Merge Worker
37*a71a9546SAutomerger Merge Worker printf(" cpu %s%u", info->invert ? "! ":"", info->cpu);
38*a71a9546SAutomerger Merge Worker }
39*a71a9546SAutomerger Merge Worker
cpu_save(const void * ip,const struct xt_entry_match * match)40*a71a9546SAutomerger Merge Worker static void cpu_save(const void *ip, const struct xt_entry_match *match)
41*a71a9546SAutomerger Merge Worker {
42*a71a9546SAutomerger Merge Worker const struct xt_cpu_info *info = (void *)match->data;
43*a71a9546SAutomerger Merge Worker
44*a71a9546SAutomerger Merge Worker printf("%s --cpu %u", info->invert ? " !" : "", info->cpu);
45*a71a9546SAutomerger Merge Worker }
46*a71a9546SAutomerger Merge Worker
cpu_xlate(struct xt_xlate * xl,const struct xt_xlate_mt_params * params)47*a71a9546SAutomerger Merge Worker static int cpu_xlate(struct xt_xlate *xl,
48*a71a9546SAutomerger Merge Worker const struct xt_xlate_mt_params *params)
49*a71a9546SAutomerger Merge Worker {
50*a71a9546SAutomerger Merge Worker const struct xt_cpu_info *info = (void *)params->match->data;
51*a71a9546SAutomerger Merge Worker
52*a71a9546SAutomerger Merge Worker xt_xlate_add(xl, "cpu%s %u", info->invert ? " !=" : "", info->cpu);
53*a71a9546SAutomerger Merge Worker
54*a71a9546SAutomerger Merge Worker return 1;
55*a71a9546SAutomerger Merge Worker }
56*a71a9546SAutomerger Merge Worker
57*a71a9546SAutomerger Merge Worker static struct xtables_match cpu_match = {
58*a71a9546SAutomerger Merge Worker .family = NFPROTO_UNSPEC,
59*a71a9546SAutomerger Merge Worker .name = "cpu",
60*a71a9546SAutomerger Merge Worker .version = XTABLES_VERSION,
61*a71a9546SAutomerger Merge Worker .size = XT_ALIGN(sizeof(struct xt_cpu_info)),
62*a71a9546SAutomerger Merge Worker .userspacesize = XT_ALIGN(sizeof(struct xt_cpu_info)),
63*a71a9546SAutomerger Merge Worker .help = cpu_help,
64*a71a9546SAutomerger Merge Worker .print = cpu_print,
65*a71a9546SAutomerger Merge Worker .save = cpu_save,
66*a71a9546SAutomerger Merge Worker .x6_parse = cpu_parse,
67*a71a9546SAutomerger Merge Worker .x6_options = cpu_opts,
68*a71a9546SAutomerger Merge Worker .xlate = cpu_xlate,
69*a71a9546SAutomerger Merge Worker };
70*a71a9546SAutomerger Merge Worker
_init(void)71*a71a9546SAutomerger Merge Worker void _init(void)
72*a71a9546SAutomerger Merge Worker {
73*a71a9546SAutomerger Merge Worker xtables_register_match(&cpu_match);
74*a71a9546SAutomerger Merge Worker }
75