1 /* net/sched/sch_dsmark.c - Differentiated Services field marker */
3 /* Written 1998-2000 by Werner Almesberger, EPFL ICA */
6 #include <linux/config.h>
7 #include <linux/module.h>
8 #include <linux/init.h>
9 #include <linux/types.h>
10 #include <linux/string.h>
11 #include <linux/errno.h>
12 #include <linux/skbuff.h>
13 #include <linux/netdevice.h> /* for pkt_sched */
14 #include <linux/rtnetlink.h>
15 #include <net/pkt_sched.h>
16 #include <net/dsfield.h>
17 #include <net/inet_ecn.h>
18 #include <asm/byteorder.h>
22 #define DPRINTK(format,args...) printk(KERN_DEBUG format,##args)
24 #define DPRINTK(format,args...)
28 #define D2PRINTK(format,args...) printk(KERN_DEBUG format,##args)
30 #define D2PRINTK(format,args...)
34 #define PRIV(sch) qdisc_priv(sch)
38 * classid class marking
39 * ------- ----- -------
43 * x:y y>0 y+1 use entry [y]
45 * x:indices-1 indices use entry [indices-1]
47 * x:y y+1 use entry [y & (indices-1)]
49 * 0xffff 0x10000 use entry [indices-1]
53 #define NO_DEFAULT_INDEX (1 << 16)
55 struct dsmark_qdisc_data
{
57 struct tcf_proto
*filter_list
;
58 __u8
*mask
; /* "owns" the array */
61 __u32 default_index
; /* index range is 0...0xffff */
66 /* ------------------------- Class/flow operations ------------------------- */
69 static int dsmark_graft(struct Qdisc
*sch
,unsigned long arg
,
70 struct Qdisc
*new,struct Qdisc
**old
)
72 struct dsmark_qdisc_data
*p
= PRIV(sch
);
74 DPRINTK("dsmark_graft(sch %p,[qdisc %p],new %p,old %p)\n",sch
,p
,new,
78 new = qdisc_create_dflt(sch
->dev
, &pfifo_qdisc_ops
);
84 *old
= xchg(&p
->q
,new);
88 sch_tree_unlock(sch
); /* @@@ move up ? */
93 static struct Qdisc
*dsmark_leaf(struct Qdisc
*sch
, unsigned long arg
)
95 struct dsmark_qdisc_data
*p
= PRIV(sch
);
101 static unsigned long dsmark_get(struct Qdisc
*sch
,u32 classid
)
103 struct dsmark_qdisc_data
*p
__attribute__((unused
)) = PRIV(sch
);
105 DPRINTK("dsmark_get(sch %p,[qdisc %p],classid %x)\n",sch
,p
,classid
);
106 return TC_H_MIN(classid
)+1;
110 static unsigned long dsmark_bind_filter(struct Qdisc
*sch
,
111 unsigned long parent
, u32 classid
)
113 return dsmark_get(sch
,classid
);
117 static void dsmark_put(struct Qdisc
*sch
, unsigned long cl
)
122 static int dsmark_change(struct Qdisc
*sch
, u32 classid
, u32 parent
,
123 struct rtattr
**tca
, unsigned long *arg
)
125 struct dsmark_qdisc_data
*p
= PRIV(sch
);
126 struct rtattr
*opt
= tca
[TCA_OPTIONS
-1];
127 struct rtattr
*tb
[TCA_DSMARK_MAX
];
129 DPRINTK("dsmark_change(sch %p,[qdisc %p],classid %x,parent %x),"
130 "arg 0x%lx\n",sch
,p
,classid
,parent
,*arg
);
131 if (*arg
> p
->indices
)
133 if (!opt
|| rtattr_parse_nested(tb
, TCA_DSMARK_MAX
, opt
))
135 if (tb
[TCA_DSMARK_MASK
-1]) {
136 if (!RTA_PAYLOAD(tb
[TCA_DSMARK_MASK
-1]))
138 p
->mask
[*arg
-1] = *(__u8
*) RTA_DATA(tb
[TCA_DSMARK_MASK
-1]);
140 if (tb
[TCA_DSMARK_VALUE
-1]) {
141 if (!RTA_PAYLOAD(tb
[TCA_DSMARK_VALUE
-1]))
143 p
->value
[*arg
-1] = *(__u8
*) RTA_DATA(tb
[TCA_DSMARK_VALUE
-1]);
149 static int dsmark_delete(struct Qdisc
*sch
,unsigned long arg
)
151 struct dsmark_qdisc_data
*p
= PRIV(sch
);
153 if (!arg
|| arg
> p
->indices
)
155 p
->mask
[arg
-1] = 0xff;
161 static void dsmark_walk(struct Qdisc
*sch
,struct qdisc_walker
*walker
)
163 struct dsmark_qdisc_data
*p
= PRIV(sch
);
166 DPRINTK("dsmark_walk(sch %p,[qdisc %p],walker %p)\n",sch
,p
,walker
);
169 for (i
= 0; i
< p
->indices
; i
++) {
170 if (p
->mask
[i
] == 0xff && !p
->value
[i
])
172 if (walker
->count
>= walker
->skip
) {
173 if (walker
->fn(sch
, i
+1, walker
) < 0) {
184 static struct tcf_proto
**dsmark_find_tcf(struct Qdisc
*sch
,unsigned long cl
)
186 struct dsmark_qdisc_data
*p
= PRIV(sch
);
188 return &p
->filter_list
;
192 /* --------------------------- Qdisc operations ---------------------------- */
195 static int dsmark_enqueue(struct sk_buff
*skb
,struct Qdisc
*sch
)
197 struct dsmark_qdisc_data
*p
= PRIV(sch
);
198 struct tcf_result res
;
200 int ret
= NET_XMIT_POLICED
;
202 D2PRINTK("dsmark_enqueue(skb %p,sch %p,[qdisc %p])\n",skb
,sch
,p
);
203 if (p
->set_tc_index
) {
204 /* FIXME: Safe with non-linear skbs? --RR */
205 switch (skb
->protocol
) {
206 case __constant_htons(ETH_P_IP
):
207 skb
->tc_index
= ipv4_get_dsfield(skb
->nh
.iph
)
210 case __constant_htons(ETH_P_IPV6
):
211 skb
->tc_index
= ipv6_get_dsfield(skb
->nh
.ipv6h
)
219 result
= TC_POLICE_OK
; /* be nice to gcc */
220 if (TC_H_MAJ(skb
->priority
) == sch
->handle
) {
221 skb
->tc_index
= TC_H_MIN(skb
->priority
);
223 result
= tc_classify(skb
,p
->filter_list
,&res
);
224 D2PRINTK("result %d class 0x%04x\n",result
,res
.classid
);
226 #ifdef CONFIG_NET_CLS_POLICE
231 case TC_POLICE_RECLASSIFY
:
232 /* FIXME: what to do here ??? */
236 skb
->tc_index
= TC_H_MIN(res
.classid
);
238 case TC_POLICE_UNSPEC
:
241 if (p
->default_index
!= NO_DEFAULT_INDEX
)
242 skb
->tc_index
= p
->default_index
;
247 #ifdef CONFIG_NET_CLS_POLICE
248 result
== TC_POLICE_SHOT
||
251 ((ret
= p
->q
->enqueue(skb
,p
->q
)) != 0)) {
255 sch
->bstats
.bytes
+= skb
->len
;
256 sch
->bstats
.packets
++;
262 static struct sk_buff
*dsmark_dequeue(struct Qdisc
*sch
)
264 struct dsmark_qdisc_data
*p
= PRIV(sch
);
268 D2PRINTK("dsmark_dequeue(sch %p,[qdisc %p])\n",sch
,p
);
269 skb
= p
->q
->ops
->dequeue(p
->q
);
273 index
= skb
->tc_index
& (p
->indices
-1);
274 D2PRINTK("index %d->%d\n",skb
->tc_index
,index
);
275 switch (skb
->protocol
) {
276 case __constant_htons(ETH_P_IP
):
277 ipv4_change_dsfield(skb
->nh
.iph
,
278 p
->mask
[index
],p
->value
[index
]);
280 case __constant_htons(ETH_P_IPV6
):
281 ipv6_change_dsfield(skb
->nh
.ipv6h
,
282 p
->mask
[index
],p
->value
[index
]);
286 * Only complain if a change was actually attempted.
287 * This way, we can send non-IP traffic through dsmark
288 * and don't need yet another qdisc as a bypass.
290 if (p
->mask
[index
] != 0xff || p
->value
[index
])
291 printk(KERN_WARNING
"dsmark_dequeue: "
292 "unsupported protocol %d\n",
293 htons(skb
->protocol
));
300 static int dsmark_requeue(struct sk_buff
*skb
,struct Qdisc
*sch
)
303 struct dsmark_qdisc_data
*p
= PRIV(sch
);
305 D2PRINTK("dsmark_requeue(skb %p,sch %p,[qdisc %p])\n",skb
,sch
,p
);
306 if ((ret
= p
->q
->ops
->requeue(skb
, p
->q
)) == 0) {
308 sch
->qstats
.requeues
++;
316 static unsigned int dsmark_drop(struct Qdisc
*sch
)
318 struct dsmark_qdisc_data
*p
= PRIV(sch
);
321 DPRINTK("dsmark_reset(sch %p,[qdisc %p])\n",sch
,p
);
322 if (!p
->q
->ops
->drop
)
324 if (!(len
= p
->q
->ops
->drop(p
->q
)))
331 static int dsmark_init(struct Qdisc
*sch
,struct rtattr
*opt
)
333 struct dsmark_qdisc_data
*p
= PRIV(sch
);
334 struct rtattr
*tb
[TCA_DSMARK_MAX
];
337 DPRINTK("dsmark_init(sch %p,[qdisc %p],opt %p)\n",sch
,p
,opt
);
339 rtattr_parse(tb
,TCA_DSMARK_MAX
,RTA_DATA(opt
),RTA_PAYLOAD(opt
)) < 0 ||
340 !tb
[TCA_DSMARK_INDICES
-1] ||
341 RTA_PAYLOAD(tb
[TCA_DSMARK_INDICES
-1]) < sizeof(__u16
))
343 p
->indices
= *(__u16
*) RTA_DATA(tb
[TCA_DSMARK_INDICES
-1]);
346 for (tmp
= p
->indices
; tmp
!= 1; tmp
>>= 1) {
350 p
->default_index
= NO_DEFAULT_INDEX
;
351 if (tb
[TCA_DSMARK_DEFAULT_INDEX
-1]) {
352 if (RTA_PAYLOAD(tb
[TCA_DSMARK_DEFAULT_INDEX
-1]) < sizeof(__u16
))
355 *(__u16
*) RTA_DATA(tb
[TCA_DSMARK_DEFAULT_INDEX
-1]);
357 p
->set_tc_index
= !!tb
[TCA_DSMARK_SET_TC_INDEX
-1];
358 p
->mask
= kmalloc(p
->indices
*2,GFP_KERNEL
);
361 p
->value
= p
->mask
+p
->indices
;
362 memset(p
->mask
,0xff,p
->indices
);
363 memset(p
->value
,0,p
->indices
);
364 if (!(p
->q
= qdisc_create_dflt(sch
->dev
, &pfifo_qdisc_ops
)))
366 DPRINTK("dsmark_init: qdisc %p\n",&p
->q
);
371 static void dsmark_reset(struct Qdisc
*sch
)
373 struct dsmark_qdisc_data
*p
= PRIV(sch
);
375 DPRINTK("dsmark_reset(sch %p,[qdisc %p])\n",sch
,p
);
381 static void dsmark_destroy(struct Qdisc
*sch
)
383 struct dsmark_qdisc_data
*p
= PRIV(sch
);
384 struct tcf_proto
*tp
;
386 DPRINTK("dsmark_destroy(sch %p,[qdisc %p])\n",sch
,p
);
387 while (p
->filter_list
) {
389 p
->filter_list
= tp
->next
;
397 static int dsmark_dump_class(struct Qdisc
*sch
, unsigned long cl
,
398 struct sk_buff
*skb
, struct tcmsg
*tcm
)
400 struct dsmark_qdisc_data
*p
= PRIV(sch
);
401 unsigned char *b
= skb
->tail
;
404 DPRINTK("dsmark_dump_class(sch %p,[qdisc %p],class %ld\n",sch
,p
,cl
);
405 if (!cl
|| cl
> p
->indices
)
407 tcm
->tcm_handle
= TC_H_MAKE(TC_H_MAJ(sch
->handle
),cl
-1);
408 rta
= (struct rtattr
*) b
;
409 RTA_PUT(skb
,TCA_OPTIONS
,0,NULL
);
410 RTA_PUT(skb
,TCA_DSMARK_MASK
,1,&p
->mask
[cl
-1]);
411 RTA_PUT(skb
,TCA_DSMARK_VALUE
,1,&p
->value
[cl
-1]);
412 rta
->rta_len
= skb
->tail
-b
;
416 skb_trim(skb
,b
-skb
->data
);
420 static int dsmark_dump(struct Qdisc
*sch
, struct sk_buff
*skb
)
422 struct dsmark_qdisc_data
*p
= PRIV(sch
);
423 unsigned char *b
= skb
->tail
;
426 rta
= (struct rtattr
*) b
;
427 RTA_PUT(skb
,TCA_OPTIONS
,0,NULL
);
428 RTA_PUT(skb
,TCA_DSMARK_INDICES
,sizeof(__u16
),&p
->indices
);
429 if (p
->default_index
!= NO_DEFAULT_INDEX
) {
430 __u16 tmp
= p
->default_index
;
432 RTA_PUT(skb
,TCA_DSMARK_DEFAULT_INDEX
, sizeof(__u16
), &tmp
);
435 RTA_PUT(skb
, TCA_DSMARK_SET_TC_INDEX
, 0, NULL
);
436 rta
->rta_len
= skb
->tail
-b
;
440 skb_trim(skb
,b
-skb
->data
);
444 static struct Qdisc_class_ops dsmark_class_ops
= {
445 .graft
= dsmark_graft
,
449 .change
= dsmark_change
,
450 .delete = dsmark_delete
,
452 .tcf_chain
= dsmark_find_tcf
,
453 .bind_tcf
= dsmark_bind_filter
,
454 .unbind_tcf
= dsmark_put
,
455 .dump
= dsmark_dump_class
,
458 static struct Qdisc_ops dsmark_qdisc_ops
= {
460 .cl_ops
= &dsmark_class_ops
,
462 .priv_size
= sizeof(struct dsmark_qdisc_data
),
463 .enqueue
= dsmark_enqueue
,
464 .dequeue
= dsmark_dequeue
,
465 .requeue
= dsmark_requeue
,
468 .reset
= dsmark_reset
,
469 .destroy
= dsmark_destroy
,
472 .owner
= THIS_MODULE
,
475 static int __init
dsmark_module_init(void)
477 return register_qdisc(&dsmark_qdisc_ops
);
479 static void __exit
dsmark_module_exit(void)
481 unregister_qdisc(&dsmark_qdisc_ops
);
483 module_init(dsmark_module_init
)
484 module_exit(dsmark_module_exit
)
485 MODULE_LICENSE("GPL");