1*a71a9546SAutomerger Merge Worker /*
2*a71a9546SAutomerger Merge Worker * Copyright (c) 2003-2013 Patrick McHardy <[email protected]>
3*a71a9546SAutomerger Merge Worker */
4*a71a9546SAutomerger Merge Worker
5*a71a9546SAutomerger Merge Worker #include <stdio.h>
6*a71a9546SAutomerger Merge Worker #include <xtables.h>
7*a71a9546SAutomerger Merge Worker #include <linux/netfilter/xt_CLASSIFY.h>
8*a71a9546SAutomerger Merge Worker #include <linux/pkt_sched.h>
9*a71a9546SAutomerger Merge Worker
10*a71a9546SAutomerger Merge Worker enum {
11*a71a9546SAutomerger Merge Worker O_SET_CLASS = 0,
12*a71a9546SAutomerger Merge Worker };
13*a71a9546SAutomerger Merge Worker
14*a71a9546SAutomerger Merge Worker static void
CLASSIFY_help(void)15*a71a9546SAutomerger Merge Worker CLASSIFY_help(void)
16*a71a9546SAutomerger Merge Worker {
17*a71a9546SAutomerger Merge Worker printf(
18*a71a9546SAutomerger Merge Worker "CLASSIFY target options:\n"
19*a71a9546SAutomerger Merge Worker "--set-class MAJOR:MINOR Set skb->priority value (always hexadecimal!)\n");
20*a71a9546SAutomerger Merge Worker }
21*a71a9546SAutomerger Merge Worker
22*a71a9546SAutomerger Merge Worker static const struct xt_option_entry CLASSIFY_opts[] = {
23*a71a9546SAutomerger Merge Worker {.name = "set-class", .id = O_SET_CLASS, .type = XTTYPE_STRING,
24*a71a9546SAutomerger Merge Worker .flags = XTOPT_MAND},
25*a71a9546SAutomerger Merge Worker XTOPT_TABLEEND,
26*a71a9546SAutomerger Merge Worker };
27*a71a9546SAutomerger Merge Worker
CLASSIFY_string_to_priority(const char * s,unsigned int * p)28*a71a9546SAutomerger Merge Worker static int CLASSIFY_string_to_priority(const char *s, unsigned int *p)
29*a71a9546SAutomerger Merge Worker {
30*a71a9546SAutomerger Merge Worker unsigned int i, j;
31*a71a9546SAutomerger Merge Worker
32*a71a9546SAutomerger Merge Worker if (sscanf(s, "%x:%x", &i, &j) != 2)
33*a71a9546SAutomerger Merge Worker return 1;
34*a71a9546SAutomerger Merge Worker
35*a71a9546SAutomerger Merge Worker *p = TC_H_MAKE(i<<16, j);
36*a71a9546SAutomerger Merge Worker return 0;
37*a71a9546SAutomerger Merge Worker }
38*a71a9546SAutomerger Merge Worker
CLASSIFY_parse(struct xt_option_call * cb)39*a71a9546SAutomerger Merge Worker static void CLASSIFY_parse(struct xt_option_call *cb)
40*a71a9546SAutomerger Merge Worker {
41*a71a9546SAutomerger Merge Worker struct xt_classify_target_info *clinfo = cb->data;
42*a71a9546SAutomerger Merge Worker
43*a71a9546SAutomerger Merge Worker xtables_option_parse(cb);
44*a71a9546SAutomerger Merge Worker if (CLASSIFY_string_to_priority(cb->arg, &clinfo->priority))
45*a71a9546SAutomerger Merge Worker xtables_error(PARAMETER_PROBLEM,
46*a71a9546SAutomerger Merge Worker "Bad class value \"%s\"", cb->arg);
47*a71a9546SAutomerger Merge Worker }
48*a71a9546SAutomerger Merge Worker
49*a71a9546SAutomerger Merge Worker static void
CLASSIFY_print_class(unsigned int priority,int numeric)50*a71a9546SAutomerger Merge Worker CLASSIFY_print_class(unsigned int priority, int numeric)
51*a71a9546SAutomerger Merge Worker {
52*a71a9546SAutomerger Merge Worker printf(" %x:%x", TC_H_MAJ(priority)>>16, TC_H_MIN(priority));
53*a71a9546SAutomerger Merge Worker }
54*a71a9546SAutomerger Merge Worker
55*a71a9546SAutomerger Merge Worker static void
CLASSIFY_print(const void * ip,const struct xt_entry_target * target,int numeric)56*a71a9546SAutomerger Merge Worker CLASSIFY_print(const void *ip,
57*a71a9546SAutomerger Merge Worker const struct xt_entry_target *target,
58*a71a9546SAutomerger Merge Worker int numeric)
59*a71a9546SAutomerger Merge Worker {
60*a71a9546SAutomerger Merge Worker const struct xt_classify_target_info *clinfo =
61*a71a9546SAutomerger Merge Worker (const struct xt_classify_target_info *)target->data;
62*a71a9546SAutomerger Merge Worker printf(" CLASSIFY set");
63*a71a9546SAutomerger Merge Worker CLASSIFY_print_class(clinfo->priority, numeric);
64*a71a9546SAutomerger Merge Worker }
65*a71a9546SAutomerger Merge Worker
66*a71a9546SAutomerger Merge Worker static void
CLASSIFY_save(const void * ip,const struct xt_entry_target * target)67*a71a9546SAutomerger Merge Worker CLASSIFY_save(const void *ip, const struct xt_entry_target *target)
68*a71a9546SAutomerger Merge Worker {
69*a71a9546SAutomerger Merge Worker const struct xt_classify_target_info *clinfo =
70*a71a9546SAutomerger Merge Worker (const struct xt_classify_target_info *)target->data;
71*a71a9546SAutomerger Merge Worker
72*a71a9546SAutomerger Merge Worker printf(" --set-class %.4x:%.4x",
73*a71a9546SAutomerger Merge Worker TC_H_MAJ(clinfo->priority)>>16, TC_H_MIN(clinfo->priority));
74*a71a9546SAutomerger Merge Worker }
75*a71a9546SAutomerger Merge Worker
76*a71a9546SAutomerger Merge Worker static void
CLASSIFY_arp_save(const void * ip,const struct xt_entry_target * target)77*a71a9546SAutomerger Merge Worker CLASSIFY_arp_save(const void *ip, const struct xt_entry_target *target)
78*a71a9546SAutomerger Merge Worker {
79*a71a9546SAutomerger Merge Worker const struct xt_classify_target_info *clinfo =
80*a71a9546SAutomerger Merge Worker (const struct xt_classify_target_info *)target->data;
81*a71a9546SAutomerger Merge Worker
82*a71a9546SAutomerger Merge Worker printf(" --set-class %x:%x",
83*a71a9546SAutomerger Merge Worker TC_H_MAJ(clinfo->priority)>>16, TC_H_MIN(clinfo->priority));
84*a71a9546SAutomerger Merge Worker }
85*a71a9546SAutomerger Merge Worker
86*a71a9546SAutomerger Merge Worker static void
CLASSIFY_arp_print(const void * ip,const struct xt_entry_target * target,int numeric)87*a71a9546SAutomerger Merge Worker CLASSIFY_arp_print(const void *ip,
88*a71a9546SAutomerger Merge Worker const struct xt_entry_target *target,
89*a71a9546SAutomerger Merge Worker int numeric)
90*a71a9546SAutomerger Merge Worker {
91*a71a9546SAutomerger Merge Worker CLASSIFY_arp_save(ip, target);
92*a71a9546SAutomerger Merge Worker }
93*a71a9546SAutomerger Merge Worker
CLASSIFY_xlate(struct xt_xlate * xl,const struct xt_xlate_tg_params * params)94*a71a9546SAutomerger Merge Worker static int CLASSIFY_xlate(struct xt_xlate *xl,
95*a71a9546SAutomerger Merge Worker const struct xt_xlate_tg_params *params)
96*a71a9546SAutomerger Merge Worker {
97*a71a9546SAutomerger Merge Worker const struct xt_classify_target_info *clinfo =
98*a71a9546SAutomerger Merge Worker (const struct xt_classify_target_info *)params->target->data;
99*a71a9546SAutomerger Merge Worker __u32 handle = clinfo->priority;
100*a71a9546SAutomerger Merge Worker
101*a71a9546SAutomerger Merge Worker xt_xlate_add(xl, "meta priority set ");
102*a71a9546SAutomerger Merge Worker
103*a71a9546SAutomerger Merge Worker switch (handle) {
104*a71a9546SAutomerger Merge Worker case TC_H_ROOT:
105*a71a9546SAutomerger Merge Worker xt_xlate_add(xl, "root");
106*a71a9546SAutomerger Merge Worker break;
107*a71a9546SAutomerger Merge Worker case TC_H_UNSPEC:
108*a71a9546SAutomerger Merge Worker xt_xlate_add(xl, "none");
109*a71a9546SAutomerger Merge Worker break;
110*a71a9546SAutomerger Merge Worker default:
111*a71a9546SAutomerger Merge Worker xt_xlate_add(xl, "%0x:%0x", TC_H_MAJ(handle) >> 16,
112*a71a9546SAutomerger Merge Worker TC_H_MIN(handle));
113*a71a9546SAutomerger Merge Worker break;
114*a71a9546SAutomerger Merge Worker }
115*a71a9546SAutomerger Merge Worker
116*a71a9546SAutomerger Merge Worker return 1;
117*a71a9546SAutomerger Merge Worker }
118*a71a9546SAutomerger Merge Worker
119*a71a9546SAutomerger Merge Worker static struct xtables_target classify_tg_reg[] = {
120*a71a9546SAutomerger Merge Worker {
121*a71a9546SAutomerger Merge Worker .family = NFPROTO_UNSPEC,
122*a71a9546SAutomerger Merge Worker .name = "CLASSIFY",
123*a71a9546SAutomerger Merge Worker .version = XTABLES_VERSION,
124*a71a9546SAutomerger Merge Worker .size = XT_ALIGN(sizeof(struct xt_classify_target_info)),
125*a71a9546SAutomerger Merge Worker .userspacesize = XT_ALIGN(sizeof(struct xt_classify_target_info)),
126*a71a9546SAutomerger Merge Worker .help = CLASSIFY_help,
127*a71a9546SAutomerger Merge Worker .print = CLASSIFY_print,
128*a71a9546SAutomerger Merge Worker .save = CLASSIFY_save,
129*a71a9546SAutomerger Merge Worker .x6_parse = CLASSIFY_parse,
130*a71a9546SAutomerger Merge Worker .x6_options = CLASSIFY_opts,
131*a71a9546SAutomerger Merge Worker .xlate = CLASSIFY_xlate,
132*a71a9546SAutomerger Merge Worker },
133*a71a9546SAutomerger Merge Worker {
134*a71a9546SAutomerger Merge Worker .family = NFPROTO_ARP,
135*a71a9546SAutomerger Merge Worker .name = "CLASSIFY",
136*a71a9546SAutomerger Merge Worker .version = XTABLES_VERSION,
137*a71a9546SAutomerger Merge Worker .size = XT_ALIGN(sizeof(struct xt_classify_target_info)),
138*a71a9546SAutomerger Merge Worker .userspacesize = XT_ALIGN(sizeof(struct xt_classify_target_info)),
139*a71a9546SAutomerger Merge Worker .help = CLASSIFY_help,
140*a71a9546SAutomerger Merge Worker .print = CLASSIFY_arp_print,
141*a71a9546SAutomerger Merge Worker .save = CLASSIFY_arp_save,
142*a71a9546SAutomerger Merge Worker .x6_parse = CLASSIFY_parse,
143*a71a9546SAutomerger Merge Worker .x6_options = CLASSIFY_opts,
144*a71a9546SAutomerger Merge Worker .xlate = CLASSIFY_xlate,
145*a71a9546SAutomerger Merge Worker }
146*a71a9546SAutomerger Merge Worker };
147*a71a9546SAutomerger Merge Worker
_init(void)148*a71a9546SAutomerger Merge Worker void _init(void)
149*a71a9546SAutomerger Merge Worker {
150*a71a9546SAutomerger Merge Worker xtables_register_targets(classify_tg_reg, ARRAY_SIZE(classify_tg_reg));
151*a71a9546SAutomerger Merge Worker }
152