]> git.proxmox.com Git - mirror_iproute2.git/blame - tc/q_tbf.c
tc: m_action: Improve conversion to C99 style initializers
[mirror_iproute2.git] / tc / q_tbf.c
CommitLineData
aba5acdf
SH
1/*
2 * q_tbf.c TBF.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version
7 * 2 of the License, or (at your option) any later version.
8 *
9 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
10 *
11 */
12
13#include <stdio.h>
14#include <stdlib.h>
15#include <unistd.h>
16#include <syslog.h>
17#include <fcntl.h>
18#include <sys/socket.h>
19#include <netinet/in.h>
20#include <arpa/inet.h>
21#include <string.h>
22
23#include "utils.h"
24#include "tc_util.h"
25
26static void explain(void)
27{
28 fprintf(stderr, "Usage: ... tbf limit BYTES burst BYTES[/BYTES] rate KBPS [ mtu BYTES[/BYTES] ]\n");
2c42579f 29 fprintf(stderr, " [ peakrate KBPS ] [ latency TIME ] ");
292f29b4 30 fprintf(stderr, "[ overhead BYTES ] [ linklayer TYPE ]\n");
aba5acdf
SH
31}
32
3bed7bb7 33static void explain1(const char *arg, const char *val)
aba5acdf 34{
3bed7bb7 35 fprintf(stderr, "tbf: illegal value for \"%s\": \"%s\"\n", arg, val);
aba5acdf
SH
36}
37
38
aba5acdf
SH
39static int tbf_parse_opt(struct qdisc_util *qu, int argc, char **argv, struct nlmsghdr *n)
40{
32a121cb 41 int ok = 0;
aba5acdf
SH
42 struct tc_tbf_qopt opt;
43 __u32 rtab[256];
44 __u32 ptab[256];
32a121cb
SH
45 unsigned buffer = 0, mtu = 0, mpu = 0, latency = 0;
46 int Rcell_log = -1, Pcell_log = -1;
47 unsigned short overhead = 0;
292f29b4 48 unsigned int linklayer = LINKLAYER_ETHERNET; /* Assume ethernet */
aba5acdf 49 struct rtattr *tail;
ddc6243e 50 __u64 rate64 = 0, prate64 = 0;
aba5acdf
SH
51
52 memset(&opt, 0, sizeof(opt));
53
54 while (argc > 0) {
55 if (matches(*argv, "limit") == 0) {
56 NEXT_ARG();
3bed7bb7
KR
57 if (opt.limit) {
58 fprintf(stderr, "tbf: duplicate \"limit\" specification\n");
59 return -1;
60 }
61 if (latency) {
62 fprintf(stderr, "tbf: specifying both \"latency\" and \"limit\" is not allowed\n");
aba5acdf
SH
63 return -1;
64 }
65 if (get_size(&opt.limit, *argv)) {
3bed7bb7 66 explain1("limit", *argv);
aba5acdf
SH
67 return -1;
68 }
69 ok++;
70 } else if (matches(*argv, "latency") == 0) {
71 NEXT_ARG();
3bed7bb7
KR
72 if (latency) {
73 fprintf(stderr, "tbf: duplicate \"latency\" specification\n");
74 return -1;
75 }
76 if (opt.limit) {
77 fprintf(stderr, "tbf: specifying both \"limit\" and \"/latency\" is not allowed\n");
aba5acdf
SH
78 return -1;
79 }
8f34caaf 80 if (get_time(&latency, *argv)) {
3bed7bb7 81 explain1("latency", *argv);
aba5acdf
SH
82 return -1;
83 }
84 ok++;
85 } else if (matches(*argv, "burst") == 0 ||
86 strcmp(*argv, "buffer") == 0 ||
87 strcmp(*argv, "maxburst") == 0) {
3bed7bb7 88 const char *parm_name = *argv;
32a121cb 89
aba5acdf
SH
90 NEXT_ARG();
91 if (buffer) {
3bed7bb7 92 fprintf(stderr, "tbf: duplicate \"buffer/burst/maxburst\" specification\n");
aba5acdf
SH
93 return -1;
94 }
95 if (get_size_and_cell(&buffer, &Rcell_log, *argv) < 0) {
3bed7bb7 96 explain1(parm_name, *argv);
aba5acdf
SH
97 return -1;
98 }
99 ok++;
100 } else if (strcmp(*argv, "mtu") == 0 ||
101 strcmp(*argv, "minburst") == 0) {
3bed7bb7 102 const char *parm_name = *argv;
32a121cb 103
aba5acdf
SH
104 NEXT_ARG();
105 if (mtu) {
3bed7bb7 106 fprintf(stderr, "tbf: duplicate \"mtu/minburst\" specification\n");
aba5acdf
SH
107 return -1;
108 }
109 if (get_size_and_cell(&mtu, &Pcell_log, *argv) < 0) {
3bed7bb7 110 explain1(parm_name, *argv);
aba5acdf
SH
111 return -1;
112 }
113 ok++;
114 } else if (strcmp(*argv, "mpu") == 0) {
115 NEXT_ARG();
116 if (mpu) {
3bed7bb7 117 fprintf(stderr, "tbf: duplicate \"mpu\" specification\n");
aba5acdf
SH
118 return -1;
119 }
120 if (get_size(&mpu, *argv)) {
3bed7bb7 121 explain1("mpu", *argv);
aba5acdf
SH
122 return -1;
123 }
124 ok++;
125 } else if (strcmp(*argv, "rate") == 0) {
126 NEXT_ARG();
ddc6243e 127 if (rate64) {
3bed7bb7 128 fprintf(stderr, "tbf: duplicate \"rate\" specification\n");
aba5acdf
SH
129 return -1;
130 }
ddc6243e 131 if (get_rate64(&rate64, *argv)) {
3bed7bb7 132 explain1("rate", *argv);
aba5acdf
SH
133 return -1;
134 }
135 ok++;
136 } else if (matches(*argv, "peakrate") == 0) {
137 NEXT_ARG();
ddc6243e 138 if (prate64) {
3bed7bb7 139 fprintf(stderr, "tbf: duplicate \"peakrate\" specification\n");
aba5acdf
SH
140 return -1;
141 }
ddc6243e 142 if (get_rate64(&prate64, *argv)) {
3bed7bb7 143 explain1("peakrate", *argv);
aba5acdf
SH
144 return -1;
145 }
146 ok++;
2c42579f
JDB
147 } else if (matches(*argv, "overhead") == 0) {
148 NEXT_ARG();
149 if (overhead) {
3bed7bb7 150 fprintf(stderr, "tbf: duplicate \"overhead\" specification\n");
2c42579f
JDB
151 return -1;
152 }
153 if (get_u16(&overhead, *argv, 10)) {
3bed7bb7 154 explain1("overhead", *argv); return -1;
2c42579f 155 }
292f29b4
JDB
156 } else if (matches(*argv, "linklayer") == 0) {
157 NEXT_ARG();
158 if (get_linklayer(&linklayer, *argv)) {
3bed7bb7 159 explain1("linklayer", *argv); return -1;
292f29b4 160 }
aba5acdf
SH
161 } else if (strcmp(*argv, "help") == 0) {
162 explain();
163 return -1;
164 } else {
3bed7bb7 165 fprintf(stderr, "tbf: unknown parameter \"%s\"\n", *argv);
aba5acdf
SH
166 explain();
167 return -1;
168 }
169 argc--; argv++;
170 }
171
32a121cb 172 int verdict = 0;
aba5acdf 173
32a121cb
SH
174 /* Be nice to the user: try to emit all error messages in
175 * one go rather than reveal one more problem when a
176 * previous one has been fixed.
177 */
ddc6243e 178 if (rate64 == 0) {
3bed7bb7
KR
179 fprintf(stderr, "tbf: the \"rate\" parameter is mandatory.\n");
180 verdict = -1;
181 }
182 if (!buffer) {
183 fprintf(stderr, "tbf: the \"burst\" parameter is mandatory.\n");
184 verdict = -1;
aba5acdf 185 }
ddc6243e 186 if (prate64) {
aba5acdf 187 if (!mtu) {
3bed7bb7
KR
188 fprintf(stderr, "tbf: when \"peakrate\" is specified, \"mtu\" must also be specified.\n");
189 verdict = -1;
aba5acdf
SH
190 }
191 }
192
193 if (opt.limit == 0 && latency == 0) {
3bed7bb7
KR
194 fprintf(stderr, "tbf: either \"limit\" or \"latency\" is required.\n");
195 verdict = -1;
aba5acdf
SH
196 }
197
32a121cb
SH
198 if (verdict != 0) {
199 explain();
200 return verdict;
201 }
3bed7bb7 202
ddc6243e
YY
203 opt.rate.rate = (rate64 >= (1ULL << 32)) ? ~0U : rate64;
204 opt.peakrate.rate = (prate64 >= (1ULL << 32)) ? ~0U : prate64;
205
aba5acdf 206 if (opt.limit == 0) {
ddc6243e 207 double lim = rate64*(double)latency/TIME_UNITS_PER_SEC + buffer;
32a121cb 208
ddc6243e
YY
209 if (prate64) {
210 double lim2 = prate64*(double)latency/TIME_UNITS_PER_SEC + mtu;
32a121cb 211
aba5acdf
SH
212 if (lim2 < lim)
213 lim = lim2;
214 }
215 opt.limit = lim;
216 }
217
2c42579f
JDB
218 opt.rate.mpu = mpu;
219 opt.rate.overhead = overhead;
292f29b4 220 if (tc_calc_rtable(&opt.rate, rtab, Rcell_log, mtu, linklayer) < 0) {
3bed7bb7 221 fprintf(stderr, "tbf: failed to calculate rate table.\n");
aba5acdf
SH
222 return -1;
223 }
224 opt.buffer = tc_calc_xmittime(opt.rate.rate, buffer);
d5f46f9c 225
aba5acdf 226 if (opt.peakrate.rate) {
2c42579f
JDB
227 opt.peakrate.mpu = mpu;
228 opt.peakrate.overhead = overhead;
292f29b4 229 if (tc_calc_rtable(&opt.peakrate, ptab, Pcell_log, mtu, linklayer) < 0) {
3bed7bb7 230 fprintf(stderr, "tbf: failed to calculate peak rate table.\n");
aba5acdf
SH
231 return -1;
232 }
233 opt.mtu = tc_calc_xmittime(opt.peakrate.rate, mtu);
aba5acdf
SH
234 }
235
228569c3 236 tail = NLMSG_TAIL(n);
aba5acdf
SH
237 addattr_l(n, 1024, TCA_OPTIONS, NULL, 0);
238 addattr_l(n, 2024, TCA_TBF_PARMS, &opt, sizeof(opt));
a01de0a3 239 addattr_l(n, 2124, TCA_TBF_BURST, &buffer, sizeof(buffer));
ddc6243e
YY
240 if (rate64 >= (1ULL << 32))
241 addattr_l(n, 2124, TCA_TBF_RATE64, &rate64, sizeof(rate64));
aba5acdf 242 addattr_l(n, 3024, TCA_TBF_RTAB, rtab, 1024);
ddc6243e
YY
243 if (opt.peakrate.rate) {
244 if (prate64 >= (1ULL << 32))
245 addattr_l(n, 3124, TCA_TBF_PRATE64, &prate64, sizeof(prate64));
a01de0a3 246 addattr_l(n, 3224, TCA_TBF_PBURST, &mtu, sizeof(mtu));
aba5acdf 247 addattr_l(n, 4096, TCA_TBF_PTAB, ptab, 1024);
ddc6243e 248 }
228569c3 249 tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
aba5acdf
SH
250 return 0;
251}
252
253static int tbf_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
254{
ddc6243e 255 struct rtattr *tb[TCA_TBF_MAX+1];
aba5acdf 256 struct tc_tbf_qopt *qopt;
3e92ff52 257 unsigned int linklayer;
aba5acdf
SH
258 double buffer, mtu;
259 double latency;
ddc6243e 260 __u64 rate64 = 0, prate64 = 0;
32a121cb 261
aba5acdf
SH
262 SPRINT_BUF(b1);
263 SPRINT_BUF(b2);
3e92ff52 264 SPRINT_BUF(b3);
aba5acdf
SH
265
266 if (opt == NULL)
267 return 0;
268
ddc6243e 269 parse_rtattr_nested(tb, TCA_TBF_MAX, opt);
aba5acdf
SH
270
271 if (tb[TCA_TBF_PARMS] == NULL)
272 return -1;
273
274 qopt = RTA_DATA(tb[TCA_TBF_PARMS]);
275 if (RTA_PAYLOAD(tb[TCA_TBF_PARMS]) < sizeof(*qopt))
276 return -1;
ddc6243e
YY
277 rate64 = qopt->rate.rate;
278 if (tb[TCA_TBF_RATE64] &&
279 RTA_PAYLOAD(tb[TCA_TBF_RATE64]) >= sizeof(rate64))
280 rate64 = rta_getattr_u64(tb[TCA_TBF_RATE64]);
281 fprintf(f, "rate %s ", sprint_rate(rate64, b1));
282 buffer = tc_calc_xmitsize(rate64, qopt->buffer);
aba5acdf
SH
283 if (show_details) {
284 fprintf(f, "burst %s/%u mpu %s ", sprint_size(buffer, b1),
285 1<<qopt->rate.cell_log, sprint_size(qopt->rate.mpu, b2));
286 } else {
287 fprintf(f, "burst %s ", sprint_size(buffer, b1));
288 }
289 if (show_raw)
290 fprintf(f, "[%08x] ", qopt->buffer);
ddc6243e
YY
291 prate64 = qopt->peakrate.rate;
292 if (tb[TCA_TBF_PRATE64] &&
293 RTA_PAYLOAD(tb[TCA_TBF_PRATE64]) >= sizeof(prate64))
294 prate64 = rta_getattr_u64(tb[TCA_TBF_PRATE64]);
295 if (prate64) {
296 fprintf(f, "peakrate %s ", sprint_rate(prate64, b1));
aba5acdf 297 if (qopt->mtu || qopt->peakrate.mpu) {
ddc6243e 298 mtu = tc_calc_xmitsize(prate64, qopt->mtu);
aba5acdf
SH
299 if (show_details) {
300 fprintf(f, "mtu %s/%u mpu %s ", sprint_size(mtu, b1),
301 1<<qopt->peakrate.cell_log, sprint_size(qopt->peakrate.mpu, b2));
302 } else {
303 fprintf(f, "minburst %s ", sprint_size(mtu, b1));
304 }
305 if (show_raw)
306 fprintf(f, "[%08x] ", qopt->mtu);
307 }
308 }
309
ddc6243e
YY
310 latency = TIME_UNITS_PER_SEC*(qopt->limit/(double)rate64) - tc_core_tick2time(qopt->buffer);
311 if (prate64) {
312 double lat2 = TIME_UNITS_PER_SEC*(qopt->limit/(double)prate64) - tc_core_tick2time(qopt->mtu);
32a121cb 313
aba5acdf
SH
314 if (lat2 > latency)
315 latency = lat2;
316 }
3ff10e82
SL
317 if (latency >= 0.0)
318 fprintf(f, "lat %s ", sprint_time(latency, b1));
319 if (show_raw || latency < 0.0)
320 fprintf(f, "limit %s ", sprint_size(qopt->limit, b1));
aba5acdf 321
2c42579f
JDB
322 if (qopt->rate.overhead) {
323 fprintf(f, "overhead %d", qopt->rate.overhead);
324 }
3e92ff52
JDB
325 linklayer = (qopt->rate.linklayer & TC_LINKLAYER_MASK);
326 if (linklayer > TC_LINKLAYER_ETHERNET || show_details)
327 fprintf(f, "linklayer %s ", sprint_linklayer(linklayer, b3));
2c42579f 328
aba5acdf
SH
329 return 0;
330}
331
95812b56 332struct qdisc_util tbf_qdisc_util = {
f2f99e2e
SH
333 .id = "tbf",
334 .parse_qopt = tbf_parse_opt,
335 .print_qopt = tbf_print_opt,
aba5acdf 336};