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