Home | History | Annotate | Download | only in tc
      1 /*
      2  * q_dsmark.c		Differentiated Services field marking.
      3  *
      4  * Hacked 1998,1999 by Werner Almesberger, EPFL ICA
      5  *
      6  */
      7 
      8 #include <stdio.h>
      9 #include <stdlib.h>
     10 #include <unistd.h>
     11 #include <syslog.h>
     12 #include <fcntl.h>
     13 #include <sys/socket.h>
     14 #include <netinet/in.h>
     15 #include <arpa/inet.h>
     16 #include <string.h>
     17 
     18 #include "utils.h"
     19 #include "tc_util.h"
     20 
     21 
     22 static void explain(void)
     23 {
     24 	fprintf(stderr,"Usage: dsmark indices INDICES [ default_index "
     25 	    "DEFAULT_INDEX ] [ set_tc_index ]\n");
     26 }
     27 
     28 
     29 static int dsmark_parse_opt(struct qdisc_util *qu, int argc, char **argv,
     30     struct nlmsghdr *n)
     31 {
     32 	struct rtattr *tail;
     33 	__u16 ind;
     34 	char *end;
     35 	int dflt,set_tc_index;
     36 
     37 	ind = set_tc_index = 0;
     38 	dflt = -1;
     39 	while (argc > 0) {
     40 		if (!strcmp(*argv,"indices")) {
     41 			NEXT_ARG();
     42 			ind = strtoul(*argv,&end,0);
     43 			if (*end) {
     44 				explain();
     45 				return -1;
     46 			}
     47 		}
     48 		else if (!strcmp(*argv,"default_index") || !strcmp(*argv,
     49 		    "default")) {
     50 			NEXT_ARG();
     51 			dflt = strtoul(*argv,&end,0);
     52 			if (*end) {
     53 				explain();
     54 				return -1;
     55 			}
     56 		}
     57 		else if (!strcmp(*argv,"set_tc_index")) {
     58 			set_tc_index = 1;
     59 		}
     60 		else {
     61 			explain();
     62 			return -1;
     63 		}
     64 		argc--;
     65 		argv++;
     66 	}
     67 	if (!ind) {
     68 		explain();
     69 		return -1;
     70 	}
     71 	tail = NLMSG_TAIL(n);
     72 	addattr_l(n,1024,TCA_OPTIONS,NULL,0);
     73 	addattr_l(n,1024,TCA_DSMARK_INDICES,&ind,sizeof(ind));
     74 	if (dflt != -1) {
     75 	    __u16 tmp = dflt;
     76 
     77 	    addattr_l(n,1024,TCA_DSMARK_DEFAULT_INDEX,&tmp,sizeof(tmp));
     78 	}
     79 	if (set_tc_index) addattr_l(n,1024,TCA_DSMARK_SET_TC_INDEX,NULL,0);
     80 	tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
     81 	return 0;
     82 }
     83 
     84 
     85 static void explain_class(void)
     86 {
     87 	fprintf(stderr, "Usage: ... dsmark [ mask MASK ] [ value VALUE ]\n");
     88 }
     89 
     90 
     91 static int dsmark_parse_class_opt(struct qdisc_util *qu, int argc, char **argv,
     92    struct nlmsghdr *n)
     93 {
     94 	struct rtattr *tail;
     95 	__u8 tmp;
     96 	char *end;
     97 
     98 	tail = NLMSG_TAIL(n);
     99 	addattr_l(n,1024,TCA_OPTIONS,NULL,0);
    100 	while (argc > 0) {
    101 		if (!strcmp(*argv,"mask")) {
    102 			NEXT_ARG();
    103 			tmp = strtoul(*argv,&end,0);
    104 			if (*end) {
    105 				explain_class();
    106 				return -1;
    107 			}
    108 			addattr_l(n,1024,TCA_DSMARK_MASK,&tmp,1);
    109 		}
    110 		else if (!strcmp(*argv,"value")) {
    111 			NEXT_ARG();
    112 			tmp = strtoul(*argv,&end,0);
    113 			if (*end) {
    114 				explain_class();
    115 				return -1;
    116 			}
    117 			addattr_l(n,1024,TCA_DSMARK_VALUE,&tmp,1);
    118 		}
    119 		else {
    120 			explain_class();
    121 			return -1;
    122 		}
    123 		argc--;
    124 		argv++;
    125 	}
    126 	tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
    127 	return 0;
    128 }
    129 
    130 
    131 
    132 static int dsmark_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
    133 {
    134 	struct rtattr *tb[TCA_DSMARK_MAX+1];
    135 
    136 	if (!opt) return 0;
    137 	memset(tb, 0, sizeof(tb));
    138 	parse_rtattr(tb, TCA_DSMARK_MAX, RTA_DATA(opt), RTA_PAYLOAD(opt));
    139 	if (tb[TCA_DSMARK_MASK]) {
    140 		if (!RTA_PAYLOAD(tb[TCA_DSMARK_MASK]))
    141 			fprintf(stderr,"dsmark: empty mask\n");
    142 		else fprintf(f,"mask 0x%02x ",
    143 			    *(__u8 *) RTA_DATA(tb[TCA_DSMARK_MASK]));
    144 	}
    145 	if (tb[TCA_DSMARK_VALUE]) {
    146 		if (!RTA_PAYLOAD(tb[TCA_DSMARK_VALUE]))
    147 			fprintf(stderr,"dsmark: empty value\n");
    148 		else fprintf(f,"value 0x%02x ",
    149 			    *(__u8 *) RTA_DATA(tb[TCA_DSMARK_VALUE]));
    150 	}
    151 	if (tb[TCA_DSMARK_INDICES]) {
    152 		if (RTA_PAYLOAD(tb[TCA_DSMARK_INDICES]) < sizeof(__u16))
    153 			fprintf(stderr,"dsmark: indices too short\n");
    154 		else fprintf(f,"indices 0x%04x ",
    155 			    *(__u16 *) RTA_DATA(tb[TCA_DSMARK_INDICES]));
    156 	}
    157 	if (tb[TCA_DSMARK_DEFAULT_INDEX]) {
    158 		if (RTA_PAYLOAD(tb[TCA_DSMARK_DEFAULT_INDEX]) < sizeof(__u16))
    159 			fprintf(stderr,"dsmark: default_index too short\n");
    160 		else fprintf(f,"default_index 0x%04x ",
    161 			    *(__u16 *) RTA_DATA(tb[TCA_DSMARK_DEFAULT_INDEX]));
    162 	}
    163 	if (tb[TCA_DSMARK_SET_TC_INDEX]) fprintf(f,"set_tc_index ");
    164 	return 0;
    165 }
    166 
    167 
    168 struct qdisc_util dsmark_qdisc_util = {
    169 	.id		= "dsmark",
    170 	.parse_qopt	= dsmark_parse_opt,
    171 	.print_qopt	= dsmark_print_opt,
    172 	.parse_copt	= dsmark_parse_class_opt,
    173 	.print_copt	= dsmark_print_opt,
    174 };
    175