21 #include <netlink-private/netlink.h>
22 #include <netlink-private/tc.h>
23 #include <netlink/netlink.h>
24 #include <netlink/utils.h>
25 #include <netlink-private/route/tc-api.h>
26 #include <netlink/route/qdisc.h>
27 #include <netlink/route/qdisc/netem.h>
30 #define SCH_NETEM_ATTR_LATENCY 0x0001
31 #define SCH_NETEM_ATTR_LIMIT 0x0002
32 #define SCH_NETEM_ATTR_LOSS 0x0004
33 #define SCH_NETEM_ATTR_GAP 0x0008
34 #define SCH_NETEM_ATTR_DUPLICATE 0x0010
35 #define SCH_NETEM_ATTR_JITTER 0x0020
36 #define SCH_NETEM_ATTR_DELAY_CORR 0x0040
37 #define SCH_NETEM_ATTR_LOSS_CORR 0x0080
38 #define SCH_NETEM_ATTR_DUP_CORR 0x0100
39 #define SCH_NETEM_ATTR_RO_PROB 0x0200
40 #define SCH_NETEM_ATTR_RO_CORR 0x0400
41 #define SCH_NETEM_ATTR_CORRUPT_PROB 0x0800
42 #define SCH_NETEM_ATTR_CORRUPT_CORR 0x1000
43 #define SCH_NETEM_ATTR_DIST 0x2000
46 static struct nla_policy netem_policy[TCA_NETEM_MAX+1] = {
47 [TCA_NETEM_CORR] = { .
minlen =
sizeof(
struct tc_netem_corr) },
48 [TCA_NETEM_REORDER] = { .minlen =
sizeof(
struct tc_netem_reorder) },
49 [TCA_NETEM_CORRUPT] = { .minlen =
sizeof(
struct tc_netem_corrupt) },
52 static int netem_msg_parser(
struct rtnl_tc *tc,
void *data)
54 struct rtnl_netem *netem = data;
55 struct tc_netem_qopt *opts;
58 if (tc->tc_opts->d_size <
sizeof(*opts))
61 opts = (
struct tc_netem_qopt *) tc->tc_opts->d_data;
62 netem->qnm_latency = opts->latency;
63 netem->qnm_limit = opts->limit;
64 netem->qnm_loss = opts->loss;
65 netem->qnm_gap = opts->gap;
66 netem->qnm_duplicate = opts->duplicate;
67 netem->qnm_jitter = opts->jitter;
69 netem->qnm_mask = (SCH_NETEM_ATTR_LATENCY | SCH_NETEM_ATTR_LIMIT |
70 SCH_NETEM_ATTR_LOSS | SCH_NETEM_ATTR_GAP |
71 SCH_NETEM_ATTR_DUPLICATE | SCH_NETEM_ATTR_JITTER);
73 len = tc->tc_opts->d_size -
sizeof(*opts);
76 struct nlattr *tb[TCA_NETEM_MAX+1];
78 err =
nla_parse(tb, TCA_NETEM_MAX, (
struct nlattr *)
79 (tc->tc_opts->d_data +
sizeof(*opts)),
86 if (tb[TCA_NETEM_CORR]) {
87 struct tc_netem_corr cor;
89 nla_memcpy(&cor, tb[TCA_NETEM_CORR],
sizeof(cor));
90 netem->qnm_corr.nmc_delay = cor.delay_corr;
91 netem->qnm_corr.nmc_loss = cor.loss_corr;
92 netem->qnm_corr.nmc_duplicate = cor.dup_corr;
94 netem->qnm_mask |= (SCH_NETEM_ATTR_DELAY_CORR |
95 SCH_NETEM_ATTR_LOSS_CORR |
96 SCH_NETEM_ATTR_DUP_CORR);
99 if (tb[TCA_NETEM_REORDER]) {
100 struct tc_netem_reorder ro;
102 nla_memcpy(&ro, tb[TCA_NETEM_REORDER],
sizeof(ro));
103 netem->qnm_ro.nmro_probability = ro.probability;
104 netem->qnm_ro.nmro_correlation = ro.correlation;
106 netem->qnm_mask |= (SCH_NETEM_ATTR_RO_PROB |
107 SCH_NETEM_ATTR_RO_CORR);
110 if (tb[TCA_NETEM_CORRUPT]) {
111 struct tc_netem_corrupt corrupt;
113 nla_memcpy(&corrupt, tb[TCA_NETEM_CORRUPT],
sizeof(corrupt));
114 netem->qnm_crpt.nmcr_probability = corrupt.probability;
115 netem->qnm_crpt.nmcr_correlation = corrupt.correlation;
117 netem->qnm_mask |= (SCH_NETEM_ATTR_CORRUPT_PROB |
118 SCH_NETEM_ATTR_CORRUPT_CORR);
122 netem->qnm_dist.dist_data = NULL;
123 netem->qnm_dist.dist_size = 0;
129 static void netem_free_data(
struct rtnl_tc *tc,
void *data)
131 struct rtnl_netem *netem = data;
136 free(netem->qnm_dist.dist_data);
139 static void netem_dump_line(
struct rtnl_tc *tc,
void *data,
142 struct rtnl_netem *netem = data;
145 nl_dump(p,
"limit %d", netem->qnm_limit);
148 static int netem_msg_fill_raw(
struct rtnl_tc *tc,
void *data,
152 struct tc_netem_qopt opts;
153 struct tc_netem_corr cor;
154 struct tc_netem_reorder reorder;
155 struct tc_netem_corrupt corrupt;
156 struct rtnl_netem *netem = data;
158 unsigned char set_correlation = 0, set_reorder = 0,
159 set_corrupt = 0, set_dist = 0;
164 memset(&opts, 0,
sizeof(opts));
165 memset(&cor, 0,
sizeof(cor));
166 memset(&reorder, 0,
sizeof(reorder));
167 memset(&corrupt, 0,
sizeof(corrupt));
169 msg->nm_nlh->nlmsg_flags |= NLM_F_REQUEST;
171 if ( netem->qnm_ro.nmro_probability != 0 ) {
172 if (netem->qnm_latency == 0) {
173 return -NLE_MISSING_ATTR;
175 if (netem->qnm_gap == 0) netem->qnm_gap = 1;
177 else if ( netem->qnm_gap ) {
178 return -NLE_MISSING_ATTR;
181 if ( netem->qnm_corr.nmc_delay != 0 ) {
182 if ( netem->qnm_latency == 0 || netem->qnm_jitter == 0) {
183 return -NLE_MISSING_ATTR;
188 if ( netem->qnm_corr.nmc_loss != 0 ) {
189 if ( netem->qnm_loss == 0 ) {
190 return -NLE_MISSING_ATTR;
195 if ( netem->qnm_corr.nmc_duplicate != 0 ) {
196 if ( netem->qnm_duplicate == 0 ) {
197 return -NLE_MISSING_ATTR;
202 if ( netem->qnm_ro.nmro_probability != 0 ) set_reorder = 1;
203 else if ( netem->qnm_ro.nmro_correlation != 0 ) {
204 return -NLE_MISSING_ATTR;
207 if ( netem->qnm_crpt.nmcr_probability != 0 ) set_corrupt = 1;
208 else if ( netem->qnm_crpt.nmcr_correlation != 0 ) {
209 return -NLE_MISSING_ATTR;
212 if ( netem->qnm_dist.dist_data && netem->qnm_dist.dist_size ) {
213 if (netem->qnm_latency == 0 || netem->qnm_jitter == 0) {
214 return -NLE_MISSING_ATTR;
218 int new_msg_len = msg->nm_size + netem->qnm_dist.dist_size *
219 sizeof(netem->qnm_dist.dist_data[0]);
221 msg->nm_nlh = (
struct nlmsghdr *) realloc(msg->nm_nlh, new_msg_len);
222 if ( msg->nm_nlh == NULL )
224 msg->nm_size = new_msg_len;
229 opts.latency = netem->qnm_latency;
230 opts.limit = netem->qnm_limit ? netem->qnm_limit : 1000;
231 opts.loss = netem->qnm_loss;
232 opts.gap = netem->qnm_gap;
233 opts.duplicate = netem->qnm_duplicate;
234 opts.jitter = netem->qnm_jitter;
236 NLA_PUT(msg, TCA_OPTIONS,
sizeof(opts), &opts);
238 if ( set_correlation ) {
239 cor.delay_corr = netem->qnm_corr.nmc_delay;
240 cor.loss_corr = netem->qnm_corr.nmc_loss;
241 cor.dup_corr = netem->qnm_corr.nmc_duplicate;
243 NLA_PUT(msg, TCA_NETEM_CORR,
sizeof(cor), &cor);
247 reorder.probability = netem->qnm_ro.nmro_probability;
248 reorder.correlation = netem->qnm_ro.nmro_correlation;
250 NLA_PUT(msg, TCA_NETEM_REORDER,
sizeof(reorder), &reorder);
254 corrupt.probability = netem->qnm_crpt.nmcr_probability;
255 corrupt.correlation = netem->qnm_crpt.nmcr_correlation;
257 NLA_PUT(msg, TCA_NETEM_CORRUPT,
sizeof(corrupt), &corrupt);
261 NLA_PUT(msg, TCA_NETEM_DELAY_DIST,
262 netem->qnm_dist.dist_size *
sizeof(netem->qnm_dist.dist_data[0]),
263 netem->qnm_dist.dist_data);
270 struct nlattr* head = (
struct nlattr *)(NLMSG_DATA(msg->nm_nlh) +
271 NLMSG_LENGTH(
sizeof(
struct tcmsg)) - NLMSG_ALIGNTO);
273 struct nlattr* tail = (
struct nlattr *)(((
void *) (msg->nm_nlh)) +
274 NLMSG_ALIGN(msg->nm_nlh->nlmsg_len));
276 int old_len = head->nla_len;
277 head->nla_len = (
void *)tail - (
void *)head;
278 msg->nm_nlh->nlmsg_len += (head->nla_len - old_len);
298 struct rtnl_netem *netem;
303 netem->qnm_limit = limit;
304 netem->qnm_mask |= SCH_NETEM_ATTR_LIMIT;
314 struct rtnl_netem *netem;
319 if (netem->qnm_mask & SCH_NETEM_ATTR_LIMIT)
320 return netem->qnm_limit;
340 struct rtnl_netem *netem;
345 netem->qnm_gap = gap;
346 netem->qnm_mask |= SCH_NETEM_ATTR_GAP;
356 struct rtnl_netem *netem;
361 if (netem->qnm_mask & SCH_NETEM_ATTR_GAP)
362 return netem->qnm_gap;
375 struct rtnl_netem *netem;
380 netem->qnm_ro.nmro_probability = prob;
381 netem->qnm_mask |= SCH_NETEM_ATTR_RO_PROB;
391 struct rtnl_netem *netem;
396 if (netem->qnm_mask & SCH_NETEM_ATTR_RO_PROB)
397 return netem->qnm_ro.nmro_probability;
410 struct rtnl_netem *netem;
415 netem->qnm_ro.nmro_correlation = prob;
416 netem->qnm_mask |= SCH_NETEM_ATTR_RO_CORR;
426 struct rtnl_netem *netem;
431 if (netem->qnm_mask & SCH_NETEM_ATTR_RO_CORR)
432 return netem->qnm_ro.nmro_correlation;
452 struct rtnl_netem *netem;
457 netem->qnm_crpt.nmcr_probability = prob;
458 netem->qnm_mask |= SCH_NETEM_ATTR_CORRUPT_PROB;
468 struct rtnl_netem *netem;
473 if (netem->qnm_mask & SCH_NETEM_ATTR_CORRUPT_PROB)
474 return netem->qnm_crpt.nmcr_probability;
487 struct rtnl_netem *netem;
492 netem->qnm_crpt.nmcr_correlation = prob;
493 netem->qnm_mask |= SCH_NETEM_ATTR_CORRUPT_CORR;
503 struct rtnl_netem *netem;
508 if (netem->qnm_mask & SCH_NETEM_ATTR_CORRUPT_CORR)
509 return netem->qnm_crpt.nmcr_correlation;
529 struct rtnl_netem *netem;
534 netem->qnm_loss = prob;
535 netem->qnm_mask |= SCH_NETEM_ATTR_LOSS;
545 struct rtnl_netem *netem;
550 if (netem->qnm_mask & SCH_NETEM_ATTR_LOSS)
551 return netem->qnm_loss;
564 struct rtnl_netem *netem;
569 netem->qnm_corr.nmc_loss = prob;
570 netem->qnm_mask |= SCH_NETEM_ATTR_LOSS_CORR;
580 struct rtnl_netem *netem;
585 if (netem->qnm_mask & SCH_NETEM_ATTR_LOSS_CORR)
586 return netem->qnm_corr.nmc_loss;
606 struct rtnl_netem *netem;
611 netem->qnm_duplicate = prob;
612 netem->qnm_mask |= SCH_NETEM_ATTR_DUPLICATE;
622 struct rtnl_netem *netem;
627 if (netem->qnm_mask & SCH_NETEM_ATTR_DUPLICATE)
628 return netem->qnm_duplicate;
641 struct rtnl_netem *netem;
646 netem->qnm_corr.nmc_duplicate = prob;
647 netem->qnm_mask |= SCH_NETEM_ATTR_DUP_CORR;
657 struct rtnl_netem *netem;
662 if (netem->qnm_mask & SCH_NETEM_ATTR_DUP_CORR)
663 return netem->qnm_corr.nmc_duplicate;
683 struct rtnl_netem *netem;
689 netem->qnm_mask |= SCH_NETEM_ATTR_LATENCY;
699 struct rtnl_netem *netem;
704 if (netem->qnm_mask & SCH_NETEM_ATTR_LATENCY)
718 struct rtnl_netem *netem;
724 netem->qnm_mask |= SCH_NETEM_ATTR_JITTER;
734 struct rtnl_netem *netem;
739 if (netem->qnm_mask & SCH_NETEM_ATTR_JITTER)
752 struct rtnl_netem *netem;
757 netem->qnm_corr.nmc_delay = prob;
758 netem->qnm_mask |= SCH_NETEM_ATTR_DELAY_CORR;
768 struct rtnl_netem *netem;
773 if (netem->qnm_mask & SCH_NETEM_ATTR_DELAY_CORR)
774 return netem->qnm_corr.nmc_delay;
786 struct rtnl_netem *netem;
791 if (netem->qnm_mask & SCH_NETEM_ATTR_DIST)
792 return netem->qnm_dist.dist_size;
805 struct rtnl_netem *netem;
810 if (netem->qnm_mask & SCH_NETEM_ATTR_DIST) {
811 *dist_ptr = netem->qnm_dist.dist_data;
824 struct rtnl_netem *netem;
835 char dist_suffix[] =
".dist";
838 char *test_suffix = strstr(dist_type, dist_suffix);
839 if (test_suffix != NULL && strlen(test_suffix) == 5)
840 strcpy(dist_suffix,
"");
843 char *test_path[] = {
"",
"./",
"/usr/lib/tc/",
"/usr/local/lib/tc/" };
845 for (i = 0; i < ARRAY_SIZE(test_path); i++) {
846 snprintf(name, NAME_MAX,
"%s%s%s", test_path[i], dist_type, dist_suffix);
847 if ((f = fopen(name,
"r")))
852 return -nl_syserr2nlerr(errno);
854 netem->qnm_dist.dist_data = (int16_t *) calloc (MAXDIST,
sizeof(int16_t));
856 line = (
char *) calloc (
sizeof(
char), len + 1);
858 while (getline(&line, &len, f) != -1) {
861 if (*line ==
'\n' || *line ==
'#')
864 for (p = line; ; p = endp) {
865 long x = strtol(p, &endp, 0);
866 if (endp == p)
break;
873 netem->qnm_dist.dist_data[n++] = x;
879 netem->qnm_dist.dist_size = n;
880 netem->qnm_mask |= SCH_NETEM_ATTR_DIST;
888 static struct rtnl_tc_ops netem_ops = {
890 .to_type = RTNL_TC_TYPE_QDISC,
891 .to_size =
sizeof(
struct rtnl_netem),
892 .to_msg_parser = netem_msg_parser,
893 .to_free_data = netem_free_data,
895 .to_msg_fill_raw = netem_msg_fill_raw,
898 static void __init netem_init(
void)
903 static void __exit netem_exit(
void)