1
2
3
4
5#include <linux/module.h>
6#include <linux/slab.h>
7#include <linux/init.h>
8#include <linux/interrupt.h>
9#include <linux/string.h>
10#include <linux/errno.h>
11#include <linux/skbuff.h>
12#include <linux/atmdev.h>
13#include <linux/atmclip.h>
14#include <linux/rtnetlink.h>
15#include <linux/file.h>
16#include <net/netlink.h>
17#include <net/pkt_sched.h>
18#include <net/pkt_cls.h>
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41#define VCC2FLOW(vcc) ((struct atm_flow_data *) ((vcc)->user_back))
42
43struct atm_flow_data {
44 struct Qdisc_class_common common;
45 struct Qdisc *q;
46 struct tcf_proto __rcu *filter_list;
47 struct tcf_block *block;
48 struct atm_vcc *vcc;
49 void (*old_pop)(struct atm_vcc *vcc,
50 struct sk_buff *skb);
51 struct atm_qdisc_data *parent;
52 struct socket *sock;
53 int ref;
54 struct gnet_stats_basic_packed bstats;
55 struct gnet_stats_queue qstats;
56 struct list_head list;
57 struct atm_flow_data *excess;
58
59 int hdr_len;
60 unsigned char hdr[0];
61};
62
63struct atm_qdisc_data {
64 struct atm_flow_data link;
65 struct list_head flows;
66
67 struct tasklet_struct task;
68};
69
70
71
72static inline struct atm_flow_data *lookup_flow(struct Qdisc *sch, u32 classid)
73{
74 struct atm_qdisc_data *p = qdisc_priv(sch);
75 struct atm_flow_data *flow;
76
77 list_for_each_entry(flow, &p->flows, list) {
78 if (flow->common.classid == classid)
79 return flow;
80 }
81 return NULL;
82}
83
84static int atm_tc_graft(struct Qdisc *sch, unsigned long arg,
85 struct Qdisc *new, struct Qdisc **old)
86{
87 struct atm_qdisc_data *p = qdisc_priv(sch);
88 struct atm_flow_data *flow = (struct atm_flow_data *)arg;
89
90 pr_debug("atm_tc_graft(sch %p,[qdisc %p],flow %p,new %p,old %p)\n",
91 sch, p, flow, new, old);
92 if (list_empty(&flow->list))
93 return -EINVAL;
94 if (!new)
95 new = &noop_qdisc;
96 *old = flow->q;
97 flow->q = new;
98 if (*old)
99 qdisc_reset(*old);
100 return 0;
101}
102
103static struct Qdisc *atm_tc_leaf(struct Qdisc *sch, unsigned long cl)
104{
105 struct atm_flow_data *flow = (struct atm_flow_data *)cl;
106
107 pr_debug("atm_tc_leaf(sch %p,flow %p)\n", sch, flow);
108 return flow ? flow->q : NULL;
109}
110
111static unsigned long atm_tc_find(struct Qdisc *sch, u32 classid)
112{
113 struct atm_qdisc_data *p __maybe_unused = qdisc_priv(sch);
114 struct atm_flow_data *flow;
115
116 pr_debug("%s(sch %p,[qdisc %p],classid %x)\n", __func__, sch, p, classid);
117 flow = lookup_flow(sch, classid);
118 pr_debug("%s: flow %p\n", __func__, flow);
119 return (unsigned long)flow;
120}
121
122static unsigned long atm_tc_bind_filter(struct Qdisc *sch,
123 unsigned long parent, u32 classid)
124{
125 struct atm_qdisc_data *p __maybe_unused = qdisc_priv(sch);
126 struct atm_flow_data *flow;
127
128 pr_debug("%s(sch %p,[qdisc %p],classid %x)\n", __func__, sch, p, classid);
129 flow = lookup_flow(sch, classid);
130 if (flow)
131 flow->ref++;
132 pr_debug("%s: flow %p\n", __func__, flow);
133 return (unsigned long)flow;
134}
135
136
137
138
139
140
141static void atm_tc_put(struct Qdisc *sch, unsigned long cl)
142{
143 struct atm_qdisc_data *p = qdisc_priv(sch);
144 struct atm_flow_data *flow = (struct atm_flow_data *)cl;
145
146 pr_debug("atm_tc_put(sch %p,[qdisc %p],flow %p)\n", sch, p, flow);
147 if (--flow->ref)
148 return;
149 pr_debug("atm_tc_put: destroying\n");
150 list_del_init(&flow->list);
151 pr_debug("atm_tc_put: qdisc %p\n", flow->q);
152 qdisc_destroy(flow->q);
153 tcf_block_put(flow->block);
154 if (flow->sock) {
155 pr_debug("atm_tc_put: f_count %ld\n",
156 file_count(flow->sock->file));
157 flow->vcc->pop = flow->old_pop;
158 sockfd_put(flow->sock);
159 }
160 if (flow->excess)
161 atm_tc_put(sch, (unsigned long)flow->excess);
162 if (flow != &p->link)
163 kfree(flow);
164
165
166
167
168}
169
170static void sch_atm_pop(struct atm_vcc *vcc, struct sk_buff *skb)
171{
172 struct atm_qdisc_data *p = VCC2FLOW(vcc)->parent;
173
174 pr_debug("sch_atm_pop(vcc %p,skb %p,[qdisc %p])\n", vcc, skb, p);
175 VCC2FLOW(vcc)->old_pop(vcc, skb);
176 tasklet_schedule(&p->task);
177}
178
179static const u8 llc_oui_ip[] = {
180 0xaa,
181 0xaa,
182 0x03,
183 0x00,
184 0x00, 0x00,
185 0x08, 0x00
186};
187
188static const struct nla_policy atm_policy[TCA_ATM_MAX + 1] = {
189 [TCA_ATM_FD] = { .type = NLA_U32 },
190 [TCA_ATM_EXCESS] = { .type = NLA_U32 },
191};
192
193static int atm_tc_change(struct Qdisc *sch, u32 classid, u32 parent,
194 struct nlattr **tca, unsigned long *arg)
195{
196 struct atm_qdisc_data *p = qdisc_priv(sch);
197 struct atm_flow_data *flow = (struct atm_flow_data *)*arg;
198 struct atm_flow_data *excess = NULL;
199 struct nlattr *opt = tca[TCA_OPTIONS];
200 struct nlattr *tb[TCA_ATM_MAX + 1];
201 struct socket *sock;
202 int fd, error, hdr_len;
203 void *hdr;
204
205 pr_debug("atm_tc_change(sch %p,[qdisc %p],classid %x,parent %x,"
206 "flow %p,opt %p)\n", sch, p, classid, parent, flow, opt);
207
208
209
210 if (parent && parent != TC_H_ROOT && parent != sch->handle)
211 return -EINVAL;
212
213
214
215
216
217
218
219 if (flow)
220 return -EBUSY;
221 if (opt == NULL)
222 return -EINVAL;
223
224 error = nla_parse_nested(tb, TCA_ATM_MAX, opt, atm_policy, NULL);
225 if (error < 0)
226 return error;
227
228 if (!tb[TCA_ATM_FD])
229 return -EINVAL;
230 fd = nla_get_u32(tb[TCA_ATM_FD]);
231 pr_debug("atm_tc_change: fd %d\n", fd);
232 if (tb[TCA_ATM_HDR]) {
233 hdr_len = nla_len(tb[TCA_ATM_HDR]);
234 hdr = nla_data(tb[TCA_ATM_HDR]);
235 } else {
236 hdr_len = RFC1483LLC_LEN;
237 hdr = NULL;
238 }
239 if (!tb[TCA_ATM_EXCESS])
240 excess = NULL;
241 else {
242 excess = (struct atm_flow_data *)
243 atm_tc_find(sch, nla_get_u32(tb[TCA_ATM_EXCESS]));
244 if (!excess)
245 return -ENOENT;
246 }
247 pr_debug("atm_tc_change: type %d, payload %d, hdr_len %d\n",
248 opt->nla_type, nla_len(opt), hdr_len);
249 sock = sockfd_lookup(fd, &error);
250 if (!sock)
251 return error;
252 pr_debug("atm_tc_change: f_count %ld\n", file_count(sock->file));
253 if (sock->ops->family != PF_ATMSVC && sock->ops->family != PF_ATMPVC) {
254 error = -EPROTOTYPE;
255 goto err_out;
256 }
257
258
259 if (classid) {
260 if (TC_H_MAJ(classid ^ sch->handle)) {
261 pr_debug("atm_tc_change: classid mismatch\n");
262 error = -EINVAL;
263 goto err_out;
264 }
265 } else {
266 int i;
267 unsigned long cl;
268
269 for (i = 1; i < 0x8000; i++) {
270 classid = TC_H_MAKE(sch->handle, 0x8000 | i);
271 cl = atm_tc_find(sch, classid);
272 if (!cl)
273 break;
274 }
275 }
276 pr_debug("atm_tc_change: new id %x\n", classid);
277 flow = kzalloc(sizeof(struct atm_flow_data) + hdr_len, GFP_KERNEL);
278 pr_debug("atm_tc_change: flow %p\n", flow);
279 if (!flow) {
280 error = -ENOBUFS;
281 goto err_out;
282 }
283
284 error = tcf_block_get(&flow->block, &flow->filter_list);
285 if (error) {
286 kfree(flow);
287 goto err_out;
288 }
289
290 flow->q = qdisc_create_dflt(sch->dev_queue, &pfifo_qdisc_ops, classid);
291 if (!flow->q)
292 flow->q = &noop_qdisc;
293 pr_debug("atm_tc_change: qdisc %p\n", flow->q);
294 flow->sock = sock;
295 flow->vcc = ATM_SD(sock);
296 flow->vcc->user_back = flow;
297 pr_debug("atm_tc_change: vcc %p\n", flow->vcc);
298 flow->old_pop = flow->vcc->pop;
299 flow->parent = p;
300 flow->vcc->pop = sch_atm_pop;
301 flow->common.classid = classid;
302 flow->ref = 1;
303 flow->excess = excess;
304 list_add(&flow->list, &p->link.list);
305 flow->hdr_len = hdr_len;
306 if (hdr)
307 memcpy(flow->hdr, hdr, hdr_len);
308 else
309 memcpy(flow->hdr, llc_oui_ip, sizeof(llc_oui_ip));
310 *arg = (unsigned long)flow;
311 return 0;
312err_out:
313 sockfd_put(sock);
314 return error;
315}
316
317static int atm_tc_delete(struct Qdisc *sch, unsigned long arg)
318{
319 struct atm_qdisc_data *p = qdisc_priv(sch);
320 struct atm_flow_data *flow = (struct atm_flow_data *)arg;
321
322 pr_debug("atm_tc_delete(sch %p,[qdisc %p],flow %p)\n", sch, p, flow);
323 if (list_empty(&flow->list))
324 return -EINVAL;
325 if (rcu_access_pointer(flow->filter_list) || flow == &p->link)
326 return -EBUSY;
327
328
329
330
331 if (flow->ref < 2) {
332 pr_err("atm_tc_delete: flow->ref == %d\n", flow->ref);
333 return -EINVAL;
334 }
335 if (flow->ref > 2)
336 return -EBUSY;
337 atm_tc_put(sch, arg);
338 return 0;
339}
340
341static void atm_tc_walk(struct Qdisc *sch, struct qdisc_walker *walker)
342{
343 struct atm_qdisc_data *p = qdisc_priv(sch);
344 struct atm_flow_data *flow;
345
346 pr_debug("atm_tc_walk(sch %p,[qdisc %p],walker %p)\n", sch, p, walker);
347 if (walker->stop)
348 return;
349 list_for_each_entry(flow, &p->flows, list) {
350 if (walker->count >= walker->skip &&
351 walker->fn(sch, (unsigned long)flow, walker) < 0) {
352 walker->stop = 1;
353 break;
354 }
355 walker->count++;
356 }
357}
358
359static struct tcf_block *atm_tc_tcf_block(struct Qdisc *sch, unsigned long cl)
360{
361 struct atm_qdisc_data *p = qdisc_priv(sch);
362 struct atm_flow_data *flow = (struct atm_flow_data *)cl;
363
364 pr_debug("atm_tc_find_tcf(sch %p,[qdisc %p],flow %p)\n", sch, p, flow);
365 return flow ? flow->block : p->link.block;
366}
367
368
369
370static int atm_tc_enqueue(struct sk_buff *skb, struct Qdisc *sch,
371 struct sk_buff **to_free)
372{
373 struct atm_qdisc_data *p = qdisc_priv(sch);
374 struct atm_flow_data *flow;
375 struct tcf_result res;
376 int result;
377 int ret = NET_XMIT_SUCCESS | __NET_XMIT_BYPASS;
378
379 pr_debug("atm_tc_enqueue(skb %p,sch %p,[qdisc %p])\n", skb, sch, p);
380 result = TC_ACT_OK;
381 flow = NULL;
382 if (TC_H_MAJ(skb->priority) != sch->handle ||
383 !(flow = (struct atm_flow_data *)atm_tc_find(sch, skb->priority))) {
384 struct tcf_proto *fl;
385
386 list_for_each_entry(flow, &p->flows, list) {
387 fl = rcu_dereference_bh(flow->filter_list);
388 if (fl) {
389 result = tcf_classify(skb, fl, &res, true);
390 if (result < 0)
391 continue;
392 flow = (struct atm_flow_data *)res.class;
393 if (!flow)
394 flow = lookup_flow(sch, res.classid);
395 goto done;
396 }
397 }
398 flow = NULL;
399done:
400 ;
401 }
402 if (!flow) {
403 flow = &p->link;
404 } else {
405 if (flow->vcc)
406 ATM_SKB(skb)->atm_options = flow->vcc->atm_options;
407
408#ifdef CONFIG_NET_CLS_ACT
409 switch (result) {
410 case TC_ACT_QUEUED:
411 case TC_ACT_STOLEN:
412 case TC_ACT_TRAP:
413 __qdisc_drop(skb, to_free);
414 return NET_XMIT_SUCCESS | __NET_XMIT_STOLEN;
415 case TC_ACT_SHOT:
416 __qdisc_drop(skb, to_free);
417 goto drop;
418 case TC_ACT_RECLASSIFY:
419 if (flow->excess)
420 flow = flow->excess;
421 else
422 ATM_SKB(skb)->atm_options |= ATM_ATMOPT_CLP;
423 break;
424 }
425#endif
426 }
427
428 ret = qdisc_enqueue(skb, flow->q, to_free);
429 if (ret != NET_XMIT_SUCCESS) {
430drop: __maybe_unused
431 if (net_xmit_drop_count(ret)) {
432 qdisc_qstats_drop(sch);
433 if (flow)
434 flow->qstats.drops++;
435 }
436 return ret;
437 }
438
439
440
441
442
443
444
445
446
447 if (flow == &p->link) {
448 sch->q.qlen++;
449 return NET_XMIT_SUCCESS;
450 }
451 tasklet_schedule(&p->task);
452 return NET_XMIT_SUCCESS | __NET_XMIT_BYPASS;
453}
454
455
456
457
458
459
460
461
462static void sch_atm_dequeue(unsigned long data)
463{
464 struct Qdisc *sch = (struct Qdisc *)data;
465 struct atm_qdisc_data *p = qdisc_priv(sch);
466 struct atm_flow_data *flow;
467 struct sk_buff *skb;
468
469 pr_debug("sch_atm_dequeue(sch %p,[qdisc %p])\n", sch, p);
470 list_for_each_entry(flow, &p->flows, list) {
471 if (flow == &p->link)
472 continue;
473
474
475
476
477 while ((skb = flow->q->ops->peek(flow->q))) {
478 if (!atm_may_send(flow->vcc, skb->truesize))
479 break;
480
481 skb = qdisc_dequeue_peeked(flow->q);
482 if (unlikely(!skb))
483 break;
484
485 qdisc_bstats_update(sch, skb);
486 bstats_update(&flow->bstats, skb);
487 pr_debug("atm_tc_dequeue: sending on class %p\n", flow);
488
489 skb_pull(skb, skb_network_offset(skb));
490 if (skb_headroom(skb) < flow->hdr_len) {
491 struct sk_buff *new;
492
493 new = skb_realloc_headroom(skb, flow->hdr_len);
494 dev_kfree_skb(skb);
495 if (!new)
496 continue;
497 skb = new;
498 }
499 pr_debug("sch_atm_dequeue: ip %p, data %p\n",
500 skb_network_header(skb), skb->data);
501 ATM_SKB(skb)->vcc = flow->vcc;
502 memcpy(skb_push(skb, flow->hdr_len), flow->hdr,
503 flow->hdr_len);
504 refcount_add(skb->truesize,
505 &sk_atm(flow->vcc)->sk_wmem_alloc);
506
507 flow->vcc->send(flow->vcc, skb);
508 }
509 }
510}
511
512static struct sk_buff *atm_tc_dequeue(struct Qdisc *sch)
513{
514 struct atm_qdisc_data *p = qdisc_priv(sch);
515 struct sk_buff *skb;
516
517 pr_debug("atm_tc_dequeue(sch %p,[qdisc %p])\n", sch, p);
518 tasklet_schedule(&p->task);
519 skb = qdisc_dequeue_peeked(p->link.q);
520 if (skb)
521 sch->q.qlen--;
522 return skb;
523}
524
525static struct sk_buff *atm_tc_peek(struct Qdisc *sch)
526{
527 struct atm_qdisc_data *p = qdisc_priv(sch);
528
529 pr_debug("atm_tc_peek(sch %p,[qdisc %p])\n", sch, p);
530
531 return p->link.q->ops->peek(p->link.q);
532}
533
534static int atm_tc_init(struct Qdisc *sch, struct nlattr *opt)
535{
536 struct atm_qdisc_data *p = qdisc_priv(sch);
537 int err;
538
539 pr_debug("atm_tc_init(sch %p,[qdisc %p],opt %p)\n", sch, p, opt);
540 INIT_LIST_HEAD(&p->flows);
541 INIT_LIST_HEAD(&p->link.list);
542 list_add(&p->link.list, &p->flows);
543 p->link.q = qdisc_create_dflt(sch->dev_queue,
544 &pfifo_qdisc_ops, sch->handle);
545 if (!p->link.q)
546 p->link.q = &noop_qdisc;
547 pr_debug("atm_tc_init: link (%p) qdisc %p\n", &p->link, p->link.q);
548
549 err = tcf_block_get(&p->link.block, &p->link.filter_list);
550 if (err)
551 return err;
552
553 p->link.vcc = NULL;
554 p->link.sock = NULL;
555 p->link.common.classid = sch->handle;
556 p->link.ref = 1;
557 tasklet_init(&p->task, sch_atm_dequeue, (unsigned long)sch);
558 return 0;
559}
560
561static void atm_tc_reset(struct Qdisc *sch)
562{
563 struct atm_qdisc_data *p = qdisc_priv(sch);
564 struct atm_flow_data *flow;
565
566 pr_debug("atm_tc_reset(sch %p,[qdisc %p])\n", sch, p);
567 list_for_each_entry(flow, &p->flows, list)
568 qdisc_reset(flow->q);
569 sch->q.qlen = 0;
570}
571
572static void atm_tc_destroy(struct Qdisc *sch)
573{
574 struct atm_qdisc_data *p = qdisc_priv(sch);
575 struct atm_flow_data *flow, *tmp;
576
577 pr_debug("atm_tc_destroy(sch %p,[qdisc %p])\n", sch, p);
578 list_for_each_entry(flow, &p->flows, list) {
579 tcf_block_put(flow->block);
580 flow->block = NULL;
581 }
582
583 list_for_each_entry_safe(flow, tmp, &p->flows, list) {
584 if (flow->ref > 1)
585 pr_err("atm_destroy: %p->ref = %d\n", flow, flow->ref);
586 atm_tc_put(sch, (unsigned long)flow);
587 }
588 tasklet_kill(&p->task);
589}
590
591static int atm_tc_dump_class(struct Qdisc *sch, unsigned long cl,
592 struct sk_buff *skb, struct tcmsg *tcm)
593{
594 struct atm_qdisc_data *p = qdisc_priv(sch);
595 struct atm_flow_data *flow = (struct atm_flow_data *)cl;
596 struct nlattr *nest;
597
598 pr_debug("atm_tc_dump_class(sch %p,[qdisc %p],flow %p,skb %p,tcm %p)\n",
599 sch, p, flow, skb, tcm);
600 if (list_empty(&flow->list))
601 return -EINVAL;
602 tcm->tcm_handle = flow->common.classid;
603 tcm->tcm_info = flow->q->handle;
604
605 nest = nla_nest_start(skb, TCA_OPTIONS);
606 if (nest == NULL)
607 goto nla_put_failure;
608
609 if (nla_put(skb, TCA_ATM_HDR, flow->hdr_len, flow->hdr))
610 goto nla_put_failure;
611 if (flow->vcc) {
612 struct sockaddr_atmpvc pvc;
613 int state;
614
615 memset(&pvc, 0, sizeof(pvc));
616 pvc.sap_family = AF_ATMPVC;
617 pvc.sap_addr.itf = flow->vcc->dev ? flow->vcc->dev->number : -1;
618 pvc.sap_addr.vpi = flow->vcc->vpi;
619 pvc.sap_addr.vci = flow->vcc->vci;
620 if (nla_put(skb, TCA_ATM_ADDR, sizeof(pvc), &pvc))
621 goto nla_put_failure;
622 state = ATM_VF2VS(flow->vcc->flags);
623 if (nla_put_u32(skb, TCA_ATM_STATE, state))
624 goto nla_put_failure;
625 }
626 if (flow->excess) {
627 if (nla_put_u32(skb, TCA_ATM_EXCESS, flow->common.classid))
628 goto nla_put_failure;
629 } else {
630 if (nla_put_u32(skb, TCA_ATM_EXCESS, 0))
631 goto nla_put_failure;
632 }
633 return nla_nest_end(skb, nest);
634
635nla_put_failure:
636 nla_nest_cancel(skb, nest);
637 return -1;
638}
639static int
640atm_tc_dump_class_stats(struct Qdisc *sch, unsigned long arg,
641 struct gnet_dump *d)
642{
643 struct atm_flow_data *flow = (struct atm_flow_data *)arg;
644
645 if (gnet_stats_copy_basic(qdisc_root_sleeping_running(sch),
646 d, NULL, &flow->bstats) < 0 ||
647 gnet_stats_copy_queue(d, NULL, &flow->qstats, flow->q->q.qlen) < 0)
648 return -1;
649
650 return 0;
651}
652
653static int atm_tc_dump(struct Qdisc *sch, struct sk_buff *skb)
654{
655 return 0;
656}
657
658static const struct Qdisc_class_ops atm_class_ops = {
659 .graft = atm_tc_graft,
660 .leaf = atm_tc_leaf,
661 .find = atm_tc_find,
662 .change = atm_tc_change,
663 .delete = atm_tc_delete,
664 .walk = atm_tc_walk,
665 .tcf_block = atm_tc_tcf_block,
666 .bind_tcf = atm_tc_bind_filter,
667 .unbind_tcf = atm_tc_put,
668 .dump = atm_tc_dump_class,
669 .dump_stats = atm_tc_dump_class_stats,
670};
671
672static struct Qdisc_ops atm_qdisc_ops __read_mostly = {
673 .cl_ops = &atm_class_ops,
674 .id = "atm",
675 .priv_size = sizeof(struct atm_qdisc_data),
676 .enqueue = atm_tc_enqueue,
677 .dequeue = atm_tc_dequeue,
678 .peek = atm_tc_peek,
679 .init = atm_tc_init,
680 .reset = atm_tc_reset,
681 .destroy = atm_tc_destroy,
682 .dump = atm_tc_dump,
683 .owner = THIS_MODULE,
684};
685
686static int __init atm_init(void)
687{
688 return register_qdisc(&atm_qdisc_ops);
689}
690
691static void __exit atm_exit(void)
692{
693 unregister_qdisc(&atm_qdisc_ops);
694}
695
696module_init(atm_init)
697module_exit(atm_exit)
698MODULE_LICENSE("GPL");
699