text: /*
   * q_dsmark.c		Differentiated Services field marking.
   *
   * Hacked 1998,1999 by Werner Almesberger, EPFL ICA
   *
   */
  
  #include <stdio.h>
  #include <stdlib.h>
  #include <unistd.h>
  #include <syslog.h>
  #include <fcntl.h>
  #include <sys/socket.h>
  #include <netinet/in.h>
  #include <arpa/inet.h>
  #include <string.h>
  
  #include "utils.h"
  #include "tc_util.h"
  
  
  #define usage() return(-1)
  
  
  static void explain(void)
  {
  	fprintf(stderr,"Usage: dsmark indices INDICES [ default_index "
  	    "DEFAULT_INDEX ] [ set_tc_index ]\n");
  }
  
  
  static int dsmark_parse_opt(struct qdisc_util *qu, int argc, char **argv,
      struct nlmsghdr *n)
  {
  	struct rtattr *tail;
  	__u16 ind;
  	char *end;
  	int dflt,set_tc_index;
  
  	ind = set_tc_index = 0;
  	dflt = -1;
  	while (argc > 0) {
  		if (!strcmp(*argv,"indices")) {
  			NEXT_ARG();
  			ind = strtoul(*argv,&end,0);
  			if (*end) {
  				explain();
  				return -1;
  			}
  		}
  		else if (!strcmp(*argv,"default_index") || !strcmp(*argv,
  		    "default")) {
  			NEXT_ARG();
  			dflt = strtoul(*argv,&end,0);
  			if (*end) {
  				explain();
  				return -1;
  			}
  		}
  		else if (!strcmp(*argv,"set_tc_index")) {
  			set_tc_index = 1;
  		}
  		else {
  			explain();
  			return -1;
  		}
  		argc--;
  		argv++;
  	}
  	if (!ind) {
  		explain();
  		return -1;
  	}
  	tail = NLMSG_TAIL(n);
  	addattr_l(n,1024,TCA_OPTIONS,NULL,0);
  	addattr_l(n,1024,TCA_DSMARK_INDICES,&ind,sizeof(ind));
  	if (dflt != -1) {
  	    __u16 tmp = dflt;
  
  	    addattr_l(n,1024,TCA_DSMARK_DEFAULT_INDEX,&tmp,sizeof(tmp));
  	}
  	if (set_tc_index) addattr_l(n,1024,TCA_DSMARK_SET_TC_INDEX,NULL,0);
  	tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
  	return 0;
  }
  
  
  static void explain_class(void)
  {
  	fprintf(stderr, "Usage: ... dsmark [ mask MASK ] [ value VALUE ]\n");
  }
  
  
  static int dsmark_parse_class_opt(struct qdisc_util *qu, int argc, char **argv,
     struct nlmsghdr *n)
  {
  	struct rtattr *tail;
  	__u8 tmp;
  	char *end;
  
  	tail = NLMSG_TAIL(n);
  	addattr_l(n,1024,TCA_OPTIONS,NULL,0);
  	while (argc > 0) {
  		if (!strcmp(*argv,"mask")) {
  			NEXT_ARG();
  			tmp = strtoul(*argv,&end,0);
  			if (*end) {
  				explain_class();
  				return -1;
  			}
  			addattr_l(n,1024,TCA_DSMARK_MASK,&tmp,1);
  		}
  		else if (!strcmp(*argv,"value")) {
  			NEXT_ARG();
  			tmp = strtoul(*argv,&end,0);
  			if (*end) {
  				explain_class();
  				return -1;
  			}
  			addattr_l(n,1024,TCA_DSMARK_VALUE,&tmp,1);
  		}
  		else {
  			explain_class();
  			return -1;
  		}
  		argc--;
  		argv++;
  	}
  	tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
  	return 0;
  }
  
  
  
  static int dsmark_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
  {
  	struct rtattr *tb[TCA_DSMARK_MAX+1];
  
  	if (!opt) return 0;
  	memset(tb, 0, sizeof(tb));
  	parse_rtattr(tb, TCA_DSMARK_MAX, RTA_DATA(opt), RTA_PAYLOAD(opt));
  	if (tb[TCA_DSMARK_MASK]) {
  		if (!RTA_PAYLOAD(tb[TCA_DSMARK_MASK]))
  			fprintf(stderr,"dsmark: empty mask\n");
  		else fprintf(f,"mask 0x%02x ",
  			    *(__u8 *) RTA_DATA(tb[TCA_DSMARK_MASK]));
  	}
  	if (tb[TCA_DSMARK_VALUE]) {
  		if (!RTA_PAYLOAD(tb[TCA_DSMARK_VALUE]))
  			fprintf(stderr,"dsmark: empty value\n");
  		else fprintf(f,"value 0x%02x ",
  			    *(__u8 *) RTA_DATA(tb[TCA_DSMARK_VALUE]));
  	}
  	if (tb[TCA_DSMARK_INDICES]) {
  		if (RTA_PAYLOAD(tb[TCA_DSMARK_INDICES]) < sizeof(__u16))
  			fprintf(stderr,"dsmark: indices too short\n");
  		else fprintf(f,"indices 0x%04x ",
  			    *(__u16 *) RTA_DATA(tb[TCA_DSMARK_INDICES]));
  	}
  	if (tb[TCA_DSMARK_DEFAULT_INDEX]) {
  		if (RTA_PAYLOAD(tb[TCA_DSMARK_DEFAULT_INDEX]) < sizeof(__u16))
  			fprintf(stderr,"dsmark: default_index too short\n");
  		else fprintf(f,"default_index 0x%04x ",
  			    *(__u16 *) RTA_DATA(tb[TCA_DSMARK_DEFAULT_INDEX]));
  	}
  	if (tb[TCA_DSMARK_SET_TC_INDEX]) fprintf(f,"set_tc_index ");
  	return 0;
  }
  
  
  struct qdisc_util dsmark_qdisc_util = {
  	.id		= "dsmark",
  	.parse_qopt	= dsmark_parse_opt,
  	.print_qopt	= dsmark_print_opt,
  	.parse_copt	= dsmark_parse_class_opt,
  	.print_copt	= dsmark_print_opt,
  };
  