tc/q_dsmark.c

   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 };