Home | History | Annotate | Download | only in extensions
      1 /*
      2  * Copyright (c) 2003-2013 Patrick McHardy <kaber (at) trash.net>
      3  */
      4 
      5 #include <stdio.h>
      6 #include <xtables.h>
      7 #include <linux/netfilter/xt_CLASSIFY.h>
      8 #include <linux/pkt_sched.h>
      9 
     10 enum {
     11 	O_SET_CLASS = 0,
     12 };
     13 
     14 static void
     15 CLASSIFY_help(void)
     16 {
     17 	printf(
     18 "CLASSIFY target options:\n"
     19 "--set-class MAJOR:MINOR    Set skb->priority value (always hexadecimal!)\n");
     20 }
     21 
     22 static const struct xt_option_entry CLASSIFY_opts[] = {
     23 	{.name = "set-class", .id = O_SET_CLASS, .type = XTTYPE_STRING,
     24 	 .flags = XTOPT_MAND},
     25 	XTOPT_TABLEEND,
     26 };
     27 
     28 static int CLASSIFY_string_to_priority(const char *s, unsigned int *p)
     29 {
     30 	unsigned int i, j;
     31 
     32 	if (sscanf(s, "%x:%x", &i, &j) != 2)
     33 		return 1;
     34 
     35 	*p = TC_H_MAKE(i<<16, j);
     36 	return 0;
     37 }
     38 
     39 static void CLASSIFY_parse(struct xt_option_call *cb)
     40 {
     41 	struct xt_classify_target_info *clinfo = cb->data;
     42 
     43 	xtables_option_parse(cb);
     44 	if (CLASSIFY_string_to_priority(cb->arg, &clinfo->priority))
     45 		xtables_error(PARAMETER_PROBLEM,
     46 			   "Bad class value \"%s\"", cb->arg);
     47 }
     48 
     49 static void
     50 CLASSIFY_print_class(unsigned int priority, int numeric)
     51 {
     52 	printf(" %x:%x", TC_H_MAJ(priority)>>16, TC_H_MIN(priority));
     53 }
     54 
     55 static void
     56 CLASSIFY_print(const void *ip,
     57       const struct xt_entry_target *target,
     58       int numeric)
     59 {
     60 	const struct xt_classify_target_info *clinfo =
     61 		(const struct xt_classify_target_info *)target->data;
     62 	printf(" CLASSIFY set");
     63 	CLASSIFY_print_class(clinfo->priority, numeric);
     64 }
     65 
     66 static void
     67 CLASSIFY_save(const void *ip, const struct xt_entry_target *target)
     68 {
     69 	const struct xt_classify_target_info *clinfo =
     70 		(const struct xt_classify_target_info *)target->data;
     71 
     72 	printf(" --set-class %.4x:%.4x",
     73 	       TC_H_MAJ(clinfo->priority)>>16, TC_H_MIN(clinfo->priority));
     74 }
     75 
     76 static struct xtables_target classify_target = {
     77 	.family		= NFPROTO_UNSPEC,
     78 	.name		= "CLASSIFY",
     79 	.version	= XTABLES_VERSION,
     80 	.size		= XT_ALIGN(sizeof(struct xt_classify_target_info)),
     81 	.userspacesize	= XT_ALIGN(sizeof(struct xt_classify_target_info)),
     82 	.help		= CLASSIFY_help,
     83 	.print		= CLASSIFY_print,
     84 	.save		= CLASSIFY_save,
     85 	.x6_parse	= CLASSIFY_parse,
     86 	.x6_options	= CLASSIFY_opts,
     87 };
     88 
     89 void _init(void)
     90 {
     91 	xtables_register_target(&classify_target);
     92 }
     93