]>
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 | ||
aba5acdf SH |
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 | } | |
228569c3 | 71 | tail = NLMSG_TAIL(n); |
aba5acdf SH |
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); | |
228569c3 | 80 | tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail; |
aba5acdf SH |
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 | ||
228569c3 | 98 | tail = NLMSG_TAIL(n); |
aba5acdf SH |
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 | } | |
228569c3 | 126 | tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail; |
aba5acdf SH |
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 ", | |
ff24746c | 143 | rta_getattr_u8(tb[TCA_DSMARK_MASK])); |
aba5acdf SH |
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 ", | |
ff24746c | 149 | rta_getattr_u8(tb[TCA_DSMARK_VALUE])); |
aba5acdf SH |
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 ", | |
ff24746c | 155 | rta_getattr_u16(tb[TCA_DSMARK_INDICES])); |
aba5acdf SH |
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 ", | |
ff24746c | 161 | rta_getattr_u16(tb[TCA_DSMARK_DEFAULT_INDEX])); |
aba5acdf SH |
162 | } |
163 | if (tb[TCA_DSMARK_SET_TC_INDEX]) fprintf(f,"set_tc_index "); | |
164 | return 0; | |
165 | } | |
166 | ||
167 | ||
95812b56 | 168 | struct qdisc_util dsmark_qdisc_util = { |
f2f99e2e SH |
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, | |
aba5acdf | 174 | }; |