Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 1 | #include <stdio.h> |
Yasuyuki KOZAKAI | e4cc20b | 2007-08-04 08:23:13 +0000 | [diff] [blame] | 2 | #include <xtables.h> |
Yasuyuki KOZAKAI | e4cc20b | 2007-08-04 08:23:13 +0000 | [diff] [blame] | 3 | #include <linux/netfilter/xt_CLASSIFY.h> |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 4 | #include <linux/pkt_sched.h> |
| 5 | |
Jan Engelhardt | 35459f0 | 2011-03-01 20:14:16 +0100 | [diff] [blame] | 6 | enum { |
| 7 | O_SET_CLASS = 0, |
| 8 | }; |
| 9 | |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 10 | static void |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 11 | CLASSIFY_help(void) |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 12 | { |
| 13 | printf( |
Jan Engelhardt | 8b7c64d | 2008-04-15 11:48:25 +0200 | [diff] [blame] | 14 | "CLASSIFY target options:\n" |
Jan Engelhardt | c9ccba5 | 2009-04-04 18:43:15 +0200 | [diff] [blame] | 15 | "--set-class MAJOR:MINOR Set skb->priority value (always hexadecimal!)\n"); |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 16 | } |
| 17 | |
Jan Engelhardt | 35459f0 | 2011-03-01 20:14:16 +0100 | [diff] [blame] | 18 | static const struct xt_option_entry CLASSIFY_opts[] = { |
| 19 | {.name = "set-class", .id = O_SET_CLASS, .type = XTTYPE_STRING, |
| 20 | .flags = XTOPT_MAND}, |
| 21 | XTOPT_TABLEEND, |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 22 | }; |
| 23 | |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 24 | static int CLASSIFY_string_to_priority(const char *s, unsigned int *p) |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 25 | { |
| 26 | unsigned int i, j; |
| 27 | |
| 28 | if (sscanf(s, "%x:%x", &i, &j) != 2) |
| 29 | return 1; |
| 30 | |
| 31 | *p = TC_H_MAKE(i<<16, j); |
| 32 | return 0; |
| 33 | } |
| 34 | |
Jan Engelhardt | 35459f0 | 2011-03-01 20:14:16 +0100 | [diff] [blame] | 35 | static void CLASSIFY_parse(struct xt_option_call *cb) |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 36 | { |
Jan Engelhardt | 35459f0 | 2011-03-01 20:14:16 +0100 | [diff] [blame] | 37 | struct xt_classify_target_info *clinfo = cb->data; |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 38 | |
Jan Engelhardt | 35459f0 | 2011-03-01 20:14:16 +0100 | [diff] [blame] | 39 | xtables_option_parse(cb); |
| 40 | if (CLASSIFY_string_to_priority(cb->arg, &clinfo->priority)) |
Jan Engelhardt | 1829ed4 | 2009-02-21 03:29:44 +0100 | [diff] [blame] | 41 | xtables_error(PARAMETER_PROBLEM, |
Jan Engelhardt | 35459f0 | 2011-03-01 20:14:16 +0100 | [diff] [blame] | 42 | "Bad class value \"%s\"", cb->arg); |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 43 | } |
| 44 | |
| 45 | static void |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 46 | CLASSIFY_print_class(unsigned int priority, int numeric) |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 47 | { |
Jan Engelhardt | 7386635 | 2010-12-18 02:04:59 +0100 | [diff] [blame] | 48 | printf(" %x:%x", TC_H_MAJ(priority)>>16, TC_H_MIN(priority)); |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 49 | } |
| 50 | |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 51 | static void |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 52 | CLASSIFY_print(const void *ip, |
Yasuyuki KOZAKAI | 193df8e | 2007-07-24 05:57:28 +0000 | [diff] [blame] | 53 | const struct xt_entry_target *target, |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 54 | int numeric) |
| 55 | { |
Yasuyuki KOZAKAI | e4cc20b | 2007-08-04 08:23:13 +0000 | [diff] [blame] | 56 | const struct xt_classify_target_info *clinfo = |
| 57 | (const struct xt_classify_target_info *)target->data; |
Jan Engelhardt | 7386635 | 2010-12-18 02:04:59 +0100 | [diff] [blame] | 58 | printf(" CLASSIFY set"); |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 59 | CLASSIFY_print_class(clinfo->priority, numeric); |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 60 | } |
| 61 | |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 62 | static void |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 63 | CLASSIFY_save(const void *ip, const struct xt_entry_target *target) |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 64 | { |
Yasuyuki KOZAKAI | e4cc20b | 2007-08-04 08:23:13 +0000 | [diff] [blame] | 65 | const struct xt_classify_target_info *clinfo = |
| 66 | (const struct xt_classify_target_info *)target->data; |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 67 | |
Jan Engelhardt | 7386635 | 2010-12-18 02:04:59 +0100 | [diff] [blame] | 68 | printf(" --set-class %.4x:%.4x", |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 69 | TC_H_MAJ(clinfo->priority)>>16, TC_H_MIN(clinfo->priority)); |
| 70 | } |
| 71 | |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 72 | static struct xtables_target classify_target = { |
Jan Engelhardt | 4297936 | 2009-06-01 11:56:23 +0200 | [diff] [blame] | 73 | .family = NFPROTO_UNSPEC, |
Yasuyuki KOZAKAI | e4cc20b | 2007-08-04 08:23:13 +0000 | [diff] [blame] | 74 | .name = "CLASSIFY", |
Jan Engelhardt | 8b7c64d | 2008-04-15 11:48:25 +0200 | [diff] [blame] | 75 | .version = XTABLES_VERSION, |
Yasuyuki KOZAKAI | e4cc20b | 2007-08-04 08:23:13 +0000 | [diff] [blame] | 76 | .size = XT_ALIGN(sizeof(struct xt_classify_target_info)), |
| 77 | .userspacesize = XT_ALIGN(sizeof(struct xt_classify_target_info)), |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 78 | .help = CLASSIFY_help, |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 79 | .print = CLASSIFY_print, |
| 80 | .save = CLASSIFY_save, |
Jan Engelhardt | 35459f0 | 2011-03-01 20:14:16 +0100 | [diff] [blame] | 81 | .x6_parse = CLASSIFY_parse, |
| 82 | .x6_options = CLASSIFY_opts, |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 83 | }; |
| 84 | |
| 85 | void _init(void) |
| 86 | { |
László Attila Tóth | 7211888 | 2007-10-08 05:12:42 +0000 | [diff] [blame] | 87 | xtables_register_target(&classify_target); |
Joszef Kadlecsik | 6ab626b | 2003-04-11 10:14:10 +0000 | [diff] [blame] | 88 | } |