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
explain(void)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
dsmark_parse_opt(struct qdisc_util * qu,int argc,char ** argv,struct nlmsghdr * n)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
explain_class(void)85 static void explain_class(void)
86 {
87 fprintf(stderr, "Usage: ... dsmark [ mask MASK ] [ value VALUE ]\n");
88 }
89
90
dsmark_parse_class_opt(struct qdisc_util * qu,int argc,char ** argv,struct nlmsghdr * n)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
dsmark_print_opt(struct qdisc_util * qu,FILE * f,struct rtattr * opt)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 rta_getattr_u8(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 rta_getattr_u8(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 rta_getattr_u16(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 rta_getattr_u16(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