1
2
3
4#include <linux/module.h>
5#include <linux/types.h>
6#include <linux/kernel.h>
7#include <linux/string.h>
8#include <linux/errno.h>
9#include <linux/skbuff.h>
10#include <linux/rtnetlink.h>
11#include <linux/init.h>
12#include <linux/slab.h>
13#include <net/act_api.h>
14#include <net/netlink.h>
15#include <net/pkt_cls.h>
16#include <net/tc_act/tc_gate.h>
17
18static unsigned int gate_net_id;
19static struct tc_action_ops act_gate_ops;
20
21static ktime_t gate_get_time(struct tcf_gate *gact)
22{
23 ktime_t mono = ktime_get();
24
25 switch (gact->tk_offset) {
26 case TK_OFFS_MAX:
27 return mono;
28 default:
29 return ktime_mono_to_any(mono, gact->tk_offset);
30 }
31
32 return KTIME_MAX;
33}
34
35static void gate_get_start_time(struct tcf_gate *gact, ktime_t *start)
36{
37 struct tcf_gate_params *param = &gact->param;
38 ktime_t now, base, cycle;
39 u64 n;
40
41 base = ns_to_ktime(param->tcfg_basetime);
42 now = gate_get_time(gact);
43
44 if (ktime_after(base, now)) {
45 *start = base;
46 return;
47 }
48
49 cycle = param->tcfg_cycletime;
50
51 n = div64_u64(ktime_sub_ns(now, base), cycle);
52 *start = ktime_add_ns(base, (n + 1) * cycle);
53}
54
55static void gate_start_timer(struct tcf_gate *gact, ktime_t start)
56{
57 ktime_t expires;
58
59 expires = hrtimer_get_expires(&gact->hitimer);
60 if (expires == 0)
61 expires = KTIME_MAX;
62
63 start = min_t(ktime_t, start, expires);
64
65 hrtimer_start(&gact->hitimer, start, HRTIMER_MODE_ABS_SOFT);
66}
67
68static enum hrtimer_restart gate_timer_func(struct hrtimer *timer)
69{
70 struct tcf_gate *gact = container_of(timer, struct tcf_gate,
71 hitimer);
72 struct tcf_gate_params *p = &gact->param;
73 struct tcfg_gate_entry *next;
74 ktime_t close_time, now;
75
76 spin_lock(&gact->tcf_lock);
77
78 next = gact->next_entry;
79
80
81 gact->current_gate_status = next->gate_state ? GATE_ACT_GATE_OPEN : 0;
82 gact->current_entry_octets = 0;
83 gact->current_max_octets = next->maxoctets;
84
85 gact->current_close_time = ktime_add_ns(gact->current_close_time,
86 next->interval);
87
88 close_time = gact->current_close_time;
89
90 if (list_is_last(&next->list, &p->entries))
91 next = list_first_entry(&p->entries,
92 struct tcfg_gate_entry, list);
93 else
94 next = list_next_entry(next, list);
95
96 now = gate_get_time(gact);
97
98 if (ktime_after(now, close_time)) {
99 ktime_t cycle, base;
100 u64 n;
101
102 cycle = p->tcfg_cycletime;
103 base = ns_to_ktime(p->tcfg_basetime);
104 n = div64_u64(ktime_sub_ns(now, base), cycle);
105 close_time = ktime_add_ns(base, (n + 1) * cycle);
106 }
107
108 gact->next_entry = next;
109
110 hrtimer_set_expires(&gact->hitimer, close_time);
111
112 spin_unlock(&gact->tcf_lock);
113
114 return HRTIMER_RESTART;
115}
116
117static int tcf_gate_act(struct sk_buff *skb, const struct tc_action *a,
118 struct tcf_result *res)
119{
120 struct tcf_gate *gact = to_gate(a);
121
122 spin_lock(&gact->tcf_lock);
123
124 tcf_lastuse_update(&gact->tcf_tm);
125 bstats_update(&gact->tcf_bstats, skb);
126
127 if (unlikely(gact->current_gate_status & GATE_ACT_PENDING)) {
128 spin_unlock(&gact->tcf_lock);
129 return gact->tcf_action;
130 }
131
132 if (!(gact->current_gate_status & GATE_ACT_GATE_OPEN))
133 goto drop;
134
135 if (gact->current_max_octets >= 0) {
136 gact->current_entry_octets += qdisc_pkt_len(skb);
137 if (gact->current_entry_octets > gact->current_max_octets) {
138 gact->tcf_qstats.overlimits++;
139 goto drop;
140 }
141 }
142
143 spin_unlock(&gact->tcf_lock);
144
145 return gact->tcf_action;
146drop:
147 gact->tcf_qstats.drops++;
148 spin_unlock(&gact->tcf_lock);
149
150 return TC_ACT_SHOT;
151}
152
153static const struct nla_policy entry_policy[TCA_GATE_ENTRY_MAX + 1] = {
154 [TCA_GATE_ENTRY_INDEX] = { .type = NLA_U32 },
155 [TCA_GATE_ENTRY_GATE] = { .type = NLA_FLAG },
156 [TCA_GATE_ENTRY_INTERVAL] = { .type = NLA_U32 },
157 [TCA_GATE_ENTRY_IPV] = { .type = NLA_S32 },
158 [TCA_GATE_ENTRY_MAX_OCTETS] = { .type = NLA_S32 },
159};
160
161static const struct nla_policy gate_policy[TCA_GATE_MAX + 1] = {
162 [TCA_GATE_PARMS] =
163 NLA_POLICY_EXACT_LEN(sizeof(struct tc_gate)),
164 [TCA_GATE_PRIORITY] = { .type = NLA_S32 },
165 [TCA_GATE_ENTRY_LIST] = { .type = NLA_NESTED },
166 [TCA_GATE_BASE_TIME] = { .type = NLA_U64 },
167 [TCA_GATE_CYCLE_TIME] = { .type = NLA_U64 },
168 [TCA_GATE_CYCLE_TIME_EXT] = { .type = NLA_U64 },
169 [TCA_GATE_FLAGS] = { .type = NLA_U32 },
170 [TCA_GATE_CLOCKID] = { .type = NLA_S32 },
171};
172
173static int fill_gate_entry(struct nlattr **tb, struct tcfg_gate_entry *entry,
174 struct netlink_ext_ack *extack)
175{
176 u32 interval = 0;
177
178 entry->gate_state = nla_get_flag(tb[TCA_GATE_ENTRY_GATE]);
179
180 if (tb[TCA_GATE_ENTRY_INTERVAL])
181 interval = nla_get_u32(tb[TCA_GATE_ENTRY_INTERVAL]);
182
183 if (interval == 0) {
184 NL_SET_ERR_MSG(extack, "Invalid interval for schedule entry");
185 return -EINVAL;
186 }
187
188 entry->interval = interval;
189
190 if (tb[TCA_GATE_ENTRY_IPV])
191 entry->ipv = nla_get_s32(tb[TCA_GATE_ENTRY_IPV]);
192 else
193 entry->ipv = -1;
194
195 if (tb[TCA_GATE_ENTRY_MAX_OCTETS])
196 entry->maxoctets = nla_get_s32(tb[TCA_GATE_ENTRY_MAX_OCTETS]);
197 else
198 entry->maxoctets = -1;
199
200 return 0;
201}
202
203static int parse_gate_entry(struct nlattr *n, struct tcfg_gate_entry *entry,
204 int index, struct netlink_ext_ack *extack)
205{
206 struct nlattr *tb[TCA_GATE_ENTRY_MAX + 1] = { };
207 int err;
208
209 err = nla_parse_nested(tb, TCA_GATE_ENTRY_MAX, n, entry_policy, extack);
210 if (err < 0) {
211 NL_SET_ERR_MSG(extack, "Could not parse nested entry");
212 return -EINVAL;
213 }
214
215 entry->index = index;
216
217 return fill_gate_entry(tb, entry, extack);
218}
219
220static void release_entry_list(struct list_head *entries)
221{
222 struct tcfg_gate_entry *entry, *e;
223
224 list_for_each_entry_safe(entry, e, entries, list) {
225 list_del(&entry->list);
226 kfree(entry);
227 }
228}
229
230static int parse_gate_list(struct nlattr *list_attr,
231 struct tcf_gate_params *sched,
232 struct netlink_ext_ack *extack)
233{
234 struct tcfg_gate_entry *entry;
235 struct nlattr *n;
236 int err, rem;
237 int i = 0;
238
239 if (!list_attr)
240 return -EINVAL;
241
242 nla_for_each_nested(n, list_attr, rem) {
243 if (nla_type(n) != TCA_GATE_ONE_ENTRY) {
244 NL_SET_ERR_MSG(extack, "Attribute isn't type 'entry'");
245 continue;
246 }
247
248 entry = kzalloc(sizeof(*entry), GFP_ATOMIC);
249 if (!entry) {
250 NL_SET_ERR_MSG(extack, "Not enough memory for entry");
251 err = -ENOMEM;
252 goto release_list;
253 }
254
255 err = parse_gate_entry(n, entry, i, extack);
256 if (err < 0) {
257 kfree(entry);
258 goto release_list;
259 }
260
261 list_add_tail(&entry->list, &sched->entries);
262 i++;
263 }
264
265 sched->num_entries = i;
266
267 return i;
268
269release_list:
270 release_entry_list(&sched->entries);
271
272 return err;
273}
274
275static void gate_setup_timer(struct tcf_gate *gact, u64 basetime,
276 enum tk_offsets tko, s32 clockid,
277 bool do_init)
278{
279 if (!do_init) {
280 if (basetime == gact->param.tcfg_basetime &&
281 tko == gact->tk_offset &&
282 clockid == gact->param.tcfg_clockid)
283 return;
284
285 spin_unlock_bh(&gact->tcf_lock);
286 hrtimer_cancel(&gact->hitimer);
287 spin_lock_bh(&gact->tcf_lock);
288 }
289 gact->param.tcfg_basetime = basetime;
290 gact->param.tcfg_clockid = clockid;
291 gact->tk_offset = tko;
292 hrtimer_init(&gact->hitimer, clockid, HRTIMER_MODE_ABS_SOFT);
293 gact->hitimer.function = gate_timer_func;
294}
295
296static int tcf_gate_init(struct net *net, struct nlattr *nla,
297 struct nlattr *est, struct tc_action **a,
298 struct tcf_proto *tp, u32 flags,
299 struct netlink_ext_ack *extack)
300{
301 struct tc_action_net *tn = net_generic(net, gate_net_id);
302 enum tk_offsets tk_offset = TK_OFFS_TAI;
303 bool bind = flags & TCA_ACT_FLAGS_BIND;
304 struct nlattr *tb[TCA_GATE_MAX + 1];
305 struct tcf_chain *goto_ch = NULL;
306 u64 cycletime = 0, basetime = 0;
307 struct tcf_gate_params *p;
308 s32 clockid = CLOCK_TAI;
309 struct tcf_gate *gact;
310 struct tc_gate *parm;
311 int ret = 0, err;
312 u32 gflags = 0;
313 s32 prio = -1;
314 ktime_t start;
315 u32 index;
316
317 if (!nla)
318 return -EINVAL;
319
320 err = nla_parse_nested(tb, TCA_GATE_MAX, nla, gate_policy, extack);
321 if (err < 0)
322 return err;
323
324 if (!tb[TCA_GATE_PARMS])
325 return -EINVAL;
326
327 if (tb[TCA_GATE_CLOCKID]) {
328 clockid = nla_get_s32(tb[TCA_GATE_CLOCKID]);
329 switch (clockid) {
330 case CLOCK_REALTIME:
331 tk_offset = TK_OFFS_REAL;
332 break;
333 case CLOCK_MONOTONIC:
334 tk_offset = TK_OFFS_MAX;
335 break;
336 case CLOCK_BOOTTIME:
337 tk_offset = TK_OFFS_BOOT;
338 break;
339 case CLOCK_TAI:
340 tk_offset = TK_OFFS_TAI;
341 break;
342 default:
343 NL_SET_ERR_MSG(extack, "Invalid 'clockid'");
344 return -EINVAL;
345 }
346 }
347
348 parm = nla_data(tb[TCA_GATE_PARMS]);
349 index = parm->index;
350
351 err = tcf_idr_check_alloc(tn, &index, a, bind);
352 if (err < 0)
353 return err;
354
355 if (err && bind)
356 return 0;
357
358 if (!err) {
359 ret = tcf_idr_create(tn, index, est, a,
360 &act_gate_ops, bind, false, 0);
361 if (ret) {
362 tcf_idr_cleanup(tn, index);
363 return ret;
364 }
365
366 ret = ACT_P_CREATED;
367 } else if (!(flags & TCA_ACT_FLAGS_REPLACE)) {
368 tcf_idr_release(*a, bind);
369 return -EEXIST;
370 }
371
372 if (tb[TCA_GATE_PRIORITY])
373 prio = nla_get_s32(tb[TCA_GATE_PRIORITY]);
374
375 if (tb[TCA_GATE_BASE_TIME])
376 basetime = nla_get_u64(tb[TCA_GATE_BASE_TIME]);
377
378 if (tb[TCA_GATE_FLAGS])
379 gflags = nla_get_u32(tb[TCA_GATE_FLAGS]);
380
381 gact = to_gate(*a);
382 if (ret == ACT_P_CREATED)
383 INIT_LIST_HEAD(&gact->param.entries);
384
385 err = tcf_action_check_ctrlact(parm->action, tp, &goto_ch, extack);
386 if (err < 0)
387 goto release_idr;
388
389 spin_lock_bh(&gact->tcf_lock);
390 p = &gact->param;
391
392 if (tb[TCA_GATE_CYCLE_TIME])
393 cycletime = nla_get_u64(tb[TCA_GATE_CYCLE_TIME]);
394
395 if (tb[TCA_GATE_ENTRY_LIST]) {
396 err = parse_gate_list(tb[TCA_GATE_ENTRY_LIST], p, extack);
397 if (err < 0)
398 goto chain_put;
399 }
400
401 if (!cycletime) {
402 struct tcfg_gate_entry *entry;
403 ktime_t cycle = 0;
404
405 list_for_each_entry(entry, &p->entries, list)
406 cycle = ktime_add_ns(cycle, entry->interval);
407 cycletime = cycle;
408 if (!cycletime) {
409 err = -EINVAL;
410 goto chain_put;
411 }
412 }
413 p->tcfg_cycletime = cycletime;
414
415 if (tb[TCA_GATE_CYCLE_TIME_EXT])
416 p->tcfg_cycletime_ext =
417 nla_get_u64(tb[TCA_GATE_CYCLE_TIME_EXT]);
418
419 gate_setup_timer(gact, basetime, tk_offset, clockid,
420 ret == ACT_P_CREATED);
421 p->tcfg_priority = prio;
422 p->tcfg_flags = gflags;
423 gate_get_start_time(gact, &start);
424
425 gact->current_close_time = start;
426 gact->current_gate_status = GATE_ACT_GATE_OPEN | GATE_ACT_PENDING;
427
428 gact->next_entry = list_first_entry(&p->entries,
429 struct tcfg_gate_entry, list);
430
431 goto_ch = tcf_action_set_ctrlact(*a, parm->action, goto_ch);
432
433 gate_start_timer(gact, start);
434
435 spin_unlock_bh(&gact->tcf_lock);
436
437 if (goto_ch)
438 tcf_chain_put_by_act(goto_ch);
439
440 return ret;
441
442chain_put:
443 spin_unlock_bh(&gact->tcf_lock);
444
445 if (goto_ch)
446 tcf_chain_put_by_act(goto_ch);
447release_idr:
448
449
450
451 if (ret == ACT_P_CREATED)
452 gate_setup_timer(gact, gact->param.tcfg_basetime,
453 gact->tk_offset, gact->param.tcfg_clockid,
454 true);
455 tcf_idr_release(*a, bind);
456 return err;
457}
458
459static void tcf_gate_cleanup(struct tc_action *a)
460{
461 struct tcf_gate *gact = to_gate(a);
462 struct tcf_gate_params *p;
463
464 p = &gact->param;
465 hrtimer_cancel(&gact->hitimer);
466 release_entry_list(&p->entries);
467}
468
469static int dumping_entry(struct sk_buff *skb,
470 struct tcfg_gate_entry *entry)
471{
472 struct nlattr *item;
473
474 item = nla_nest_start_noflag(skb, TCA_GATE_ONE_ENTRY);
475 if (!item)
476 return -ENOSPC;
477
478 if (nla_put_u32(skb, TCA_GATE_ENTRY_INDEX, entry->index))
479 goto nla_put_failure;
480
481 if (entry->gate_state && nla_put_flag(skb, TCA_GATE_ENTRY_GATE))
482 goto nla_put_failure;
483
484 if (nla_put_u32(skb, TCA_GATE_ENTRY_INTERVAL, entry->interval))
485 goto nla_put_failure;
486
487 if (nla_put_s32(skb, TCA_GATE_ENTRY_MAX_OCTETS, entry->maxoctets))
488 goto nla_put_failure;
489
490 if (nla_put_s32(skb, TCA_GATE_ENTRY_IPV, entry->ipv))
491 goto nla_put_failure;
492
493 return nla_nest_end(skb, item);
494
495nla_put_failure:
496 nla_nest_cancel(skb, item);
497 return -1;
498}
499
500static int tcf_gate_dump(struct sk_buff *skb, struct tc_action *a,
501 int bind, int ref)
502{
503 unsigned char *b = skb_tail_pointer(skb);
504 struct tcf_gate *gact = to_gate(a);
505 struct tc_gate opt = {
506 .index = gact->tcf_index,
507 .refcnt = refcount_read(&gact->tcf_refcnt) - ref,
508 .bindcnt = atomic_read(&gact->tcf_bindcnt) - bind,
509 };
510 struct tcfg_gate_entry *entry;
511 struct tcf_gate_params *p;
512 struct nlattr *entry_list;
513 struct tcf_t t;
514
515 spin_lock_bh(&gact->tcf_lock);
516 opt.action = gact->tcf_action;
517
518 p = &gact->param;
519
520 if (nla_put(skb, TCA_GATE_PARMS, sizeof(opt), &opt))
521 goto nla_put_failure;
522
523 if (nla_put_u64_64bit(skb, TCA_GATE_BASE_TIME,
524 p->tcfg_basetime, TCA_GATE_PAD))
525 goto nla_put_failure;
526
527 if (nla_put_u64_64bit(skb, TCA_GATE_CYCLE_TIME,
528 p->tcfg_cycletime, TCA_GATE_PAD))
529 goto nla_put_failure;
530
531 if (nla_put_u64_64bit(skb, TCA_GATE_CYCLE_TIME_EXT,
532 p->tcfg_cycletime_ext, TCA_GATE_PAD))
533 goto nla_put_failure;
534
535 if (nla_put_s32(skb, TCA_GATE_CLOCKID, p->tcfg_clockid))
536 goto nla_put_failure;
537
538 if (nla_put_u32(skb, TCA_GATE_FLAGS, p->tcfg_flags))
539 goto nla_put_failure;
540
541 if (nla_put_s32(skb, TCA_GATE_PRIORITY, p->tcfg_priority))
542 goto nla_put_failure;
543
544 entry_list = nla_nest_start_noflag(skb, TCA_GATE_ENTRY_LIST);
545 if (!entry_list)
546 goto nla_put_failure;
547
548 list_for_each_entry(entry, &p->entries, list) {
549 if (dumping_entry(skb, entry) < 0)
550 goto nla_put_failure;
551 }
552
553 nla_nest_end(skb, entry_list);
554
555 tcf_tm_dump(&t, &gact->tcf_tm);
556 if (nla_put_64bit(skb, TCA_GATE_TM, sizeof(t), &t, TCA_GATE_PAD))
557 goto nla_put_failure;
558 spin_unlock_bh(&gact->tcf_lock);
559
560 return skb->len;
561
562nla_put_failure:
563 spin_unlock_bh(&gact->tcf_lock);
564 nlmsg_trim(skb, b);
565 return -1;
566}
567
568static int tcf_gate_walker(struct net *net, struct sk_buff *skb,
569 struct netlink_callback *cb, int type,
570 const struct tc_action_ops *ops,
571 struct netlink_ext_ack *extack)
572{
573 struct tc_action_net *tn = net_generic(net, gate_net_id);
574
575 return tcf_generic_walker(tn, skb, cb, type, ops, extack);
576}
577
578static void tcf_gate_stats_update(struct tc_action *a, u64 bytes, u64 packets,
579 u64 drops, u64 lastuse, bool hw)
580{
581 struct tcf_gate *gact = to_gate(a);
582 struct tcf_t *tm = &gact->tcf_tm;
583
584 tcf_action_update_stats(a, bytes, packets, drops, hw);
585 tm->lastuse = max_t(u64, tm->lastuse, lastuse);
586}
587
588static int tcf_gate_search(struct net *net, struct tc_action **a, u32 index)
589{
590 struct tc_action_net *tn = net_generic(net, gate_net_id);
591
592 return tcf_idr_search(tn, a, index);
593}
594
595static size_t tcf_gate_get_fill_size(const struct tc_action *act)
596{
597 return nla_total_size(sizeof(struct tc_gate));
598}
599
600static struct tc_action_ops act_gate_ops = {
601 .kind = "gate",
602 .id = TCA_ID_GATE,
603 .owner = THIS_MODULE,
604 .act = tcf_gate_act,
605 .dump = tcf_gate_dump,
606 .init = tcf_gate_init,
607 .cleanup = tcf_gate_cleanup,
608 .walk = tcf_gate_walker,
609 .stats_update = tcf_gate_stats_update,
610 .get_fill_size = tcf_gate_get_fill_size,
611 .lookup = tcf_gate_search,
612 .size = sizeof(struct tcf_gate),
613};
614
615static __net_init int gate_init_net(struct net *net)
616{
617 struct tc_action_net *tn = net_generic(net, gate_net_id);
618
619 return tc_action_net_init(net, tn, &act_gate_ops);
620}
621
622static void __net_exit gate_exit_net(struct list_head *net_list)
623{
624 tc_action_net_exit(net_list, gate_net_id);
625}
626
627static struct pernet_operations gate_net_ops = {
628 .init = gate_init_net,
629 .exit_batch = gate_exit_net,
630 .id = &gate_net_id,
631 .size = sizeof(struct tc_action_net),
632};
633
634static int __init gate_init_module(void)
635{
636 return tcf_register_action(&act_gate_ops, &gate_net_ops);
637}
638
639static void __exit gate_cleanup_module(void)
640{
641 tcf_unregister_action(&act_gate_ops, &gate_net_ops);
642}
643
644module_init(gate_init_module);
645module_exit(gate_cleanup_module);
646MODULE_LICENSE("GPL v2");
647
