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 DEFAULT_INDEX ] [ set_tc_index ]\n");
  25 }
  26
  27
  28 static int dsmark_parse_opt(struct qdisc_util *qu, int argc, char **argv,
  29     struct nlmsghdr *n)
  30 {
  31         struct rtattr *tail;
  32         __u16 ind;
  33         char *end;
  34         int dflt, set_tc_index;
  35
  36         ind = set_tc_index = 0;
  37         dflt = -1;
  38         while (argc > 0) {
  39                 if (!strcmp(*argv, "indices")) {
  40                         NEXT_ARG();
  41                         ind = strtoul(*argv, &end, 0);
  42                         if (*end) {
  43                                 explain();
  44                                 return -1;
  45                         }
  46                 } else if (!strcmp(*argv,"default_index") || !strcmp(*argv,
  47                     "default")) {
  48                         NEXT_ARG();
  49                         dflt = strtoul(*argv, &end, 0);
  50                         if (*end) {
  51                                 explain();
  52                                 return -1;
  53                         }
  54                 } else if (!strcmp(*argv,"set_tc_index")) {
  55                         set_tc_index = 1;
  56                 } else {
  57                         explain();
  58                         return -1;
  59                 }
  60                 argc--;
  61                 argv++;
  62         }
  63         if (!ind) {
  64                 explain();
  65                 return -1;
  66         }
  67         tail = NLMSG_TAIL(n);
  68         addattr_l(n, 1024, TCA_OPTIONS, NULL, 0);
  69         addattr_l(n, 1024, TCA_DSMARK_INDICES, &ind, sizeof(ind));
  70         if (dflt != -1) {
  71             __u16 tmp = dflt;
  72
  73             addattr_l(n, 1024, TCA_DSMARK_DEFAULT_INDEX, &tmp, sizeof(tmp));
  74         }
  75         if (set_tc_index) addattr_l(n, 1024, TCA_DSMARK_SET_TC_INDEX, NULL, 0);
  76         tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
  77         return 0;
  78 }
  79
  80
  81 static void explain_class(void)
  82 {
  83         fprintf(stderr, "Usage: ... dsmark [ mask MASK ] [ value VALUE ]\n");
  84 }
  85
  86
  87 static int dsmark_parse_class_opt(struct qdisc_util *qu, int argc, char **argv,
  88    struct nlmsghdr *n)
  89 {
  90         struct rtattr *tail;
  91         __u8 tmp;
  92         char *end;
  93
  94         tail = NLMSG_TAIL(n);
  95         addattr_l(n, 1024, TCA_OPTIONS, NULL, 0);
  96         while (argc > 0) {
  97                 if (!strcmp(*argv, "mask")) {
  98                         NEXT_ARG();
  99                         tmp = strtoul(*argv, &end, 0);
 100                         if (*end) {
 101                                 explain_class();
 102                                 return -1;
 103                         }
 104                         addattr_l(n, 1024, TCA_DSMARK_MASK, &tmp, 1);
 105                 } else if (!strcmp(*argv,"value")) {
 106                         NEXT_ARG();
 107                         tmp = strtoul(*argv, &end, 0);
 108                         if (*end) {
 109                                 explain_class();
 110                                 return -1;
 111                         }
 112                         addattr_l(n, 1024, TCA_DSMARK_VALUE, &tmp, 1);
 113                 } else {
 114                         explain_class();
 115                         return -1;
 116                 }
 117                 argc--;
 118                 argv++;
 119         }
 120         tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
 121         return 0;
 122 }
 123
 124
 125
 126 static int dsmark_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
 127 {
 128         struct rtattr *tb[TCA_DSMARK_MAX+1];
 129
 130         if (!opt) return 0;
 131         parse_rtattr(tb, TCA_DSMARK_MAX, RTA_DATA(opt), RTA_PAYLOAD(opt));
 132         if (tb[TCA_DSMARK_MASK]) {
 133                 if (!RTA_PAYLOAD(tb[TCA_DSMARK_MASK]))
 134                         fprintf(stderr, "dsmark: empty mask\n");
 135                 else fprintf(f, "mask 0x%02x ",
 136                             rta_getattr_u8(tb[TCA_DSMARK_MASK]));
 137         }
 138         if (tb[TCA_DSMARK_VALUE]) {
 139                 if (!RTA_PAYLOAD(tb[TCA_DSMARK_VALUE]))
 140                         fprintf(stderr, "dsmark: empty value\n");
 141                 else fprintf(f, "value 0x%02x ",
 142                             rta_getattr_u8(tb[TCA_DSMARK_VALUE]));
 143         }
 144         if (tb[TCA_DSMARK_INDICES]) {
 145                 if (RTA_PAYLOAD(tb[TCA_DSMARK_INDICES]) < sizeof(__u16))
 146                         fprintf(stderr, "dsmark: indices too short\n");
 147                 else fprintf(f, "indices 0x%04x ",
 148                             rta_getattr_u16(tb[TCA_DSMARK_INDICES]));
 149         }
 150         if (tb[TCA_DSMARK_DEFAULT_INDEX]) {
 151                 if (RTA_PAYLOAD(tb[TCA_DSMARK_DEFAULT_INDEX]) < sizeof(__u16))
 152                         fprintf(stderr, "dsmark: default_index too short\n");
 153                 else fprintf(f, "default_index 0x%04x ",
 154                             rta_getattr_u16(tb[TCA_DSMARK_DEFAULT_INDEX]));
 155         }
 156         if (tb[TCA_DSMARK_SET_TC_INDEX]) fprintf(f, "set_tc_index ");
 157         return 0;
 158 }
 159
 160
 161 struct qdisc_util dsmark_qdisc_util = {
 162         .id             = "dsmark",
 163         .parse_qopt     = dsmark_parse_opt,
 164         .print_qopt     = dsmark_print_opt,
 165         .parse_copt     = dsmark_parse_class_opt,
 166         .print_copt     = dsmark_print_opt,
 167 };