Linux Audio

Check our new training course

Loading...
v3.1
  1/*
  2 * net/sched/cls_flow.c		Generic flow classifier
  3 *
  4 * Copyright (c) 2007, 2008 Patrick McHardy <kaber@trash.net>
  5 *
  6 * This program is free software; you can redistribute it and/or
  7 * modify it under the terms of the GNU General Public License
  8 * as published by the Free Software Foundation; either version 2
  9 * of the License, or (at your option) any later version.
 10 */
 11
 12#include <linux/kernel.h>
 13#include <linux/init.h>
 14#include <linux/list.h>
 15#include <linux/jhash.h>
 16#include <linux/random.h>
 17#include <linux/pkt_cls.h>
 18#include <linux/skbuff.h>
 19#include <linux/in.h>
 20#include <linux/ip.h>
 21#include <linux/ipv6.h>
 22#include <linux/if_vlan.h>
 23#include <linux/slab.h>
 
 
 24
 25#include <net/pkt_cls.h>
 26#include <net/ip.h>
 27#include <net/route.h>
 28#if defined(CONFIG_NF_CONNTRACK) || defined(CONFIG_NF_CONNTRACK_MODULE)
 
 
 29#include <net/netfilter/nf_conntrack.h>
 30#endif
 31
 32struct flow_head {
 33	struct list_head	filters;
 
 34};
 35
 36struct flow_filter {
 37	struct list_head	list;
 38	struct tcf_exts		exts;
 39	struct tcf_ematch_tree	ematches;
 
 40	struct timer_list	perturb_timer;
 41	u32			perturb_period;
 42	u32			handle;
 43
 44	u32			nkeys;
 45	u32			keymask;
 46	u32			mode;
 47	u32			mask;
 48	u32			xor;
 49	u32			rshift;
 50	u32			addend;
 51	u32			divisor;
 52	u32			baseclass;
 53	u32			hashrnd;
 54};
 55
 56static const struct tcf_ext_map flow_ext_map = {
 57	.action	= TCA_FLOW_ACT,
 58	.police	= TCA_FLOW_POLICE,
 59};
 60
 61static inline u32 addr_fold(void *addr)
 62{
 63	unsigned long a = (unsigned long)addr;
 64
 65	return (a & 0xFFFFFFFF) ^ (BITS_PER_LONG > 32 ? a >> 32 : 0);
 66}
 67
 68static u32 flow_get_src(struct sk_buff *skb)
 69{
 70	switch (skb->protocol) {
 71	case htons(ETH_P_IP):
 72		if (pskb_network_may_pull(skb, sizeof(struct iphdr)))
 73			return ntohl(ip_hdr(skb)->saddr);
 74		break;
 75	case htons(ETH_P_IPV6):
 76		if (pskb_network_may_pull(skb, sizeof(struct ipv6hdr)))
 77			return ntohl(ipv6_hdr(skb)->saddr.s6_addr32[3]);
 78		break;
 79	}
 80
 81	return addr_fold(skb->sk);
 82}
 83
 84static u32 flow_get_dst(struct sk_buff *skb)
 85{
 86	switch (skb->protocol) {
 87	case htons(ETH_P_IP):
 88		if (pskb_network_may_pull(skb, sizeof(struct iphdr)))
 89			return ntohl(ip_hdr(skb)->daddr);
 90		break;
 91	case htons(ETH_P_IPV6):
 92		if (pskb_network_may_pull(skb, sizeof(struct ipv6hdr)))
 93			return ntohl(ipv6_hdr(skb)->daddr.s6_addr32[3]);
 94		break;
 95	}
 96
 97	return addr_fold(skb_dst(skb)) ^ (__force u16)skb->protocol;
 98}
 99
100static u32 flow_get_proto(struct sk_buff *skb)
 
101{
102	switch (skb->protocol) {
103	case htons(ETH_P_IP):
104		return pskb_network_may_pull(skb, sizeof(struct iphdr)) ?
105		       ip_hdr(skb)->protocol : 0;
106	case htons(ETH_P_IPV6):
107		return pskb_network_may_pull(skb, sizeof(struct ipv6hdr)) ?
108		       ipv6_hdr(skb)->nexthdr : 0;
109	default:
110		return 0;
111	}
112}
113
114static u32 flow_get_proto_src(struct sk_buff *skb)
 
115{
116	switch (skb->protocol) {
117	case htons(ETH_P_IP): {
118		struct iphdr *iph;
119		int poff;
120
121		if (!pskb_network_may_pull(skb, sizeof(*iph)))
122			break;
123		iph = ip_hdr(skb);
124		if (ip_is_fragment(iph))
125			break;
126		poff = proto_ports_offset(iph->protocol);
127		if (poff >= 0 &&
128		    pskb_network_may_pull(skb, iph->ihl * 4 + 2 + poff)) {
129			iph = ip_hdr(skb);
130			return ntohs(*(__be16 *)((void *)iph + iph->ihl * 4 +
131						 poff));
132		}
133		break;
134	}
135	case htons(ETH_P_IPV6): {
136		struct ipv6hdr *iph;
137		int poff;
138
139		if (!pskb_network_may_pull(skb, sizeof(*iph)))
140			break;
141		iph = ipv6_hdr(skb);
142		poff = proto_ports_offset(iph->nexthdr);
143		if (poff >= 0 &&
144		    pskb_network_may_pull(skb, sizeof(*iph) + poff + 2)) {
145			iph = ipv6_hdr(skb);
146			return ntohs(*(__be16 *)((void *)iph + sizeof(*iph) +
147						 poff));
148		}
149		break;
150	}
151	}
152
153	return addr_fold(skb->sk);
154}
155
156static u32 flow_get_proto_dst(struct sk_buff *skb)
 
157{
158	switch (skb->protocol) {
159	case htons(ETH_P_IP): {
160		struct iphdr *iph;
161		int poff;
162
163		if (!pskb_network_may_pull(skb, sizeof(*iph)))
164			break;
165		iph = ip_hdr(skb);
166		if (ip_is_fragment(iph))
167			break;
168		poff = proto_ports_offset(iph->protocol);
169		if (poff >= 0 &&
170		    pskb_network_may_pull(skb, iph->ihl * 4 + 4 + poff)) {
171			iph = ip_hdr(skb);
172			return ntohs(*(__be16 *)((void *)iph + iph->ihl * 4 +
173						 2 + poff));
174		}
175		break;
176	}
177	case htons(ETH_P_IPV6): {
178		struct ipv6hdr *iph;
179		int poff;
180
181		if (!pskb_network_may_pull(skb, sizeof(*iph)))
182			break;
183		iph = ipv6_hdr(skb);
184		poff = proto_ports_offset(iph->nexthdr);
185		if (poff >= 0 &&
186		    pskb_network_may_pull(skb, sizeof(*iph) + poff + 4)) {
187			iph = ipv6_hdr(skb);
188			return ntohs(*(__be16 *)((void *)iph + sizeof(*iph) +
189						 poff + 2));
190		}
191		break;
192	}
193	}
194
195	return addr_fold(skb_dst(skb)) ^ (__force u16)skb->protocol;
196}
197
198static u32 flow_get_iif(const struct sk_buff *skb)
199{
200	return skb->skb_iif;
201}
202
203static u32 flow_get_priority(const struct sk_buff *skb)
204{
205	return skb->priority;
206}
207
208static u32 flow_get_mark(const struct sk_buff *skb)
209{
210	return skb->mark;
211}
212
213static u32 flow_get_nfct(const struct sk_buff *skb)
214{
215#if defined(CONFIG_NF_CONNTRACK) || defined(CONFIG_NF_CONNTRACK_MODULE)
216	return addr_fold(skb->nfct);
217#else
218	return 0;
219#endif
220}
221
222#if defined(CONFIG_NF_CONNTRACK) || defined(CONFIG_NF_CONNTRACK_MODULE)
223#define CTTUPLE(skb, member)						\
224({									\
225	enum ip_conntrack_info ctinfo;					\
226	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);			\
227	if (ct == NULL)							\
228		goto fallback;						\
229	ct->tuplehash[CTINFO2DIR(ctinfo)].tuple.member;			\
230})
231#else
232#define CTTUPLE(skb, member)						\
233({									\
234	goto fallback;							\
235	0;								\
236})
237#endif
238
239static u32 flow_get_nfct_src(struct sk_buff *skb)
 
240{
241	switch (skb->protocol) {
242	case htons(ETH_P_IP):
243		return ntohl(CTTUPLE(skb, src.u3.ip));
244	case htons(ETH_P_IPV6):
245		return ntohl(CTTUPLE(skb, src.u3.ip6[3]));
246	}
247fallback:
248	return flow_get_src(skb);
249}
250
251static u32 flow_get_nfct_dst(struct sk_buff *skb)
 
252{
253	switch (skb->protocol) {
254	case htons(ETH_P_IP):
255		return ntohl(CTTUPLE(skb, dst.u3.ip));
256	case htons(ETH_P_IPV6):
257		return ntohl(CTTUPLE(skb, dst.u3.ip6[3]));
258	}
259fallback:
260	return flow_get_dst(skb);
261}
262
263static u32 flow_get_nfct_proto_src(struct sk_buff *skb)
 
264{
265	return ntohs(CTTUPLE(skb, src.u.all));
266fallback:
267	return flow_get_proto_src(skb);
268}
269
270static u32 flow_get_nfct_proto_dst(struct sk_buff *skb)
 
271{
272	return ntohs(CTTUPLE(skb, dst.u.all));
273fallback:
274	return flow_get_proto_dst(skb);
275}
276
277static u32 flow_get_rtclassid(const struct sk_buff *skb)
278{
279#ifdef CONFIG_IP_ROUTE_CLASSID
280	if (skb_dst(skb))
281		return skb_dst(skb)->tclassid;
282#endif
283	return 0;
284}
285
286static u32 flow_get_skuid(const struct sk_buff *skb)
287{
288	if (skb->sk && skb->sk->sk_socket && skb->sk->sk_socket->file)
289		return skb->sk->sk_socket->file->f_cred->fsuid;
 
 
 
 
 
290	return 0;
291}
292
293static u32 flow_get_skgid(const struct sk_buff *skb)
294{
295	if (skb->sk && skb->sk->sk_socket && skb->sk->sk_socket->file)
296		return skb->sk->sk_socket->file->f_cred->fsgid;
 
 
 
 
 
297	return 0;
298}
299
300static u32 flow_get_vlan_tag(const struct sk_buff *skb)
301{
302	u16 uninitialized_var(tag);
303
304	if (vlan_get_tag(skb, &tag) < 0)
305		return 0;
306	return tag & VLAN_VID_MASK;
307}
308
309static u32 flow_get_rxhash(struct sk_buff *skb)
310{
311	return skb_get_rxhash(skb);
312}
313
314static u32 flow_key_get(struct sk_buff *skb, int key)
315{
316	switch (key) {
317	case FLOW_KEY_SRC:
318		return flow_get_src(skb);
319	case FLOW_KEY_DST:
320		return flow_get_dst(skb);
321	case FLOW_KEY_PROTO:
322		return flow_get_proto(skb);
323	case FLOW_KEY_PROTO_SRC:
324		return flow_get_proto_src(skb);
325	case FLOW_KEY_PROTO_DST:
326		return flow_get_proto_dst(skb);
327	case FLOW_KEY_IIF:
328		return flow_get_iif(skb);
329	case FLOW_KEY_PRIORITY:
330		return flow_get_priority(skb);
331	case FLOW_KEY_MARK:
332		return flow_get_mark(skb);
333	case FLOW_KEY_NFCT:
334		return flow_get_nfct(skb);
335	case FLOW_KEY_NFCT_SRC:
336		return flow_get_nfct_src(skb);
337	case FLOW_KEY_NFCT_DST:
338		return flow_get_nfct_dst(skb);
339	case FLOW_KEY_NFCT_PROTO_SRC:
340		return flow_get_nfct_proto_src(skb);
341	case FLOW_KEY_NFCT_PROTO_DST:
342		return flow_get_nfct_proto_dst(skb);
343	case FLOW_KEY_RTCLASSID:
344		return flow_get_rtclassid(skb);
345	case FLOW_KEY_SKUID:
346		return flow_get_skuid(skb);
347	case FLOW_KEY_SKGID:
348		return flow_get_skgid(skb);
349	case FLOW_KEY_VLAN_TAG:
350		return flow_get_vlan_tag(skb);
351	case FLOW_KEY_RXHASH:
352		return flow_get_rxhash(skb);
353	default:
354		WARN_ON(1);
355		return 0;
356	}
357}
358
 
 
 
 
 
 
 
 
 
 
359static int flow_classify(struct sk_buff *skb, const struct tcf_proto *tp,
360			 struct tcf_result *res)
361{
362	struct flow_head *head = tp->root;
363	struct flow_filter *f;
364	u32 keymask;
365	u32 classid;
366	unsigned int n, key;
367	int r;
368
369	list_for_each_entry(f, &head->filters, list) {
370		u32 keys[f->nkeys];
 
371
372		if (!tcf_em_tree_match(skb, &f->ematches, NULL))
373			continue;
374
375		keymask = f->keymask;
 
 
376
377		for (n = 0; n < f->nkeys; n++) {
378			key = ffs(keymask) - 1;
379			keymask &= ~(1 << key);
380			keys[n] = flow_key_get(skb, key);
381		}
382
383		if (f->mode == FLOW_MODE_HASH)
384			classid = jhash2(keys, f->nkeys, f->hashrnd);
385		else {
386			classid = keys[0];
387			classid = (classid & f->mask) ^ f->xor;
388			classid = (classid >> f->rshift) + f->addend;
389		}
390
391		if (f->divisor)
392			classid %= f->divisor;
393
394		res->class   = 0;
395		res->classid = TC_H_MAKE(f->baseclass, f->baseclass + classid);
396
397		r = tcf_exts_exec(skb, &f->exts, res);
398		if (r < 0)
399			continue;
400		return r;
401	}
402	return -1;
403}
404
405static void flow_perturbation(unsigned long arg)
406{
407	struct flow_filter *f = (struct flow_filter *)arg;
408
409	get_random_bytes(&f->hashrnd, 4);
410	if (f->perturb_period)
411		mod_timer(&f->perturb_timer, jiffies + f->perturb_period);
412}
413
414static const struct nla_policy flow_policy[TCA_FLOW_MAX + 1] = {
415	[TCA_FLOW_KEYS]		= { .type = NLA_U32 },
416	[TCA_FLOW_MODE]		= { .type = NLA_U32 },
417	[TCA_FLOW_BASECLASS]	= { .type = NLA_U32 },
418	[TCA_FLOW_RSHIFT]	= { .type = NLA_U32 },
419	[TCA_FLOW_ADDEND]	= { .type = NLA_U32 },
420	[TCA_FLOW_MASK]		= { .type = NLA_U32 },
421	[TCA_FLOW_XOR]		= { .type = NLA_U32 },
422	[TCA_FLOW_DIVISOR]	= { .type = NLA_U32 },
423	[TCA_FLOW_ACT]		= { .type = NLA_NESTED },
424	[TCA_FLOW_POLICE]	= { .type = NLA_NESTED },
425	[TCA_FLOW_EMATCHES]	= { .type = NLA_NESTED },
426	[TCA_FLOW_PERTURB]	= { .type = NLA_U32 },
427};
428
429static int flow_change(struct tcf_proto *tp, unsigned long base,
 
 
 
 
 
 
 
 
 
 
 
430		       u32 handle, struct nlattr **tca,
431		       unsigned long *arg)
432{
433	struct flow_head *head = tp->root;
434	struct flow_filter *f;
435	struct nlattr *opt = tca[TCA_OPTIONS];
436	struct nlattr *tb[TCA_FLOW_MAX + 1];
437	struct tcf_exts e;
438	struct tcf_ematch_tree t;
439	unsigned int nkeys = 0;
440	unsigned int perturb_period = 0;
441	u32 baseclass = 0;
442	u32 keymask = 0;
443	u32 mode;
444	int err;
445
446	if (opt == NULL)
447		return -EINVAL;
448
449	err = nla_parse_nested(tb, TCA_FLOW_MAX, opt, flow_policy);
450	if (err < 0)
451		return err;
452
453	if (tb[TCA_FLOW_BASECLASS]) {
454		baseclass = nla_get_u32(tb[TCA_FLOW_BASECLASS]);
455		if (TC_H_MIN(baseclass) == 0)
456			return -EINVAL;
457	}
458
459	if (tb[TCA_FLOW_KEYS]) {
460		keymask = nla_get_u32(tb[TCA_FLOW_KEYS]);
461
462		nkeys = hweight32(keymask);
463		if (nkeys == 0)
464			return -EINVAL;
465
466		if (fls(keymask) - 1 > FLOW_KEY_MAX)
467			return -EOPNOTSUPP;
 
 
 
 
468	}
469
470	err = tcf_exts_validate(tp, tb, tca[TCA_RATE], &e, &flow_ext_map);
471	if (err < 0)
472		return err;
 
 
 
473
474	err = tcf_em_tree_validate(tp, tb[TCA_FLOW_EMATCHES], &t);
475	if (err < 0)
476		goto err1;
477
478	f = (struct flow_filter *)*arg;
479	if (f != NULL) {
 
 
 
 
 
 
 
 
 
480		err = -EINVAL;
481		if (f->handle != handle && handle)
482			goto err2;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
483
484		mode = f->mode;
485		if (tb[TCA_FLOW_MODE])
486			mode = nla_get_u32(tb[TCA_FLOW_MODE]);
487		if (mode != FLOW_MODE_HASH && nkeys > 1)
488			goto err2;
489
490		if (mode == FLOW_MODE_HASH)
491			perturb_period = f->perturb_period;
492		if (tb[TCA_FLOW_PERTURB]) {
493			if (mode != FLOW_MODE_HASH)
494				goto err2;
495			perturb_period = nla_get_u32(tb[TCA_FLOW_PERTURB]) * HZ;
496		}
497	} else {
498		err = -EINVAL;
499		if (!handle)
500			goto err2;
501		if (!tb[TCA_FLOW_KEYS])
502			goto err2;
503
504		mode = FLOW_MODE_MAP;
505		if (tb[TCA_FLOW_MODE])
506			mode = nla_get_u32(tb[TCA_FLOW_MODE]);
507		if (mode != FLOW_MODE_HASH && nkeys > 1)
508			goto err2;
509
510		if (tb[TCA_FLOW_PERTURB]) {
511			if (mode != FLOW_MODE_HASH)
512				goto err2;
513			perturb_period = nla_get_u32(tb[TCA_FLOW_PERTURB]) * HZ;
514		}
515
516		if (TC_H_MAJ(baseclass) == 0)
517			baseclass = TC_H_MAKE(tp->q->handle, baseclass);
518		if (TC_H_MIN(baseclass) == 0)
519			baseclass = TC_H_MAKE(baseclass, 1);
520
521		err = -ENOBUFS;
522		f = kzalloc(sizeof(*f), GFP_KERNEL);
523		if (f == NULL)
524			goto err2;
525
526		f->handle = handle;
527		f->mask	  = ~0U;
528
529		get_random_bytes(&f->hashrnd, 4);
530		f->perturb_timer.function = flow_perturbation;
531		f->perturb_timer.data = (unsigned long)f;
532		init_timer_deferrable(&f->perturb_timer);
533	}
534
535	tcf_exts_change(tp, &f->exts, &e);
536	tcf_em_tree_change(tp, &f->ematches, &t);
 
 
 
 
537
538	tcf_tree_lock(tp);
539
540	if (tb[TCA_FLOW_KEYS]) {
541		f->keymask = keymask;
542		f->nkeys   = nkeys;
543	}
544
545	f->mode = mode;
546
547	if (tb[TCA_FLOW_MASK])
548		f->mask = nla_get_u32(tb[TCA_FLOW_MASK]);
549	if (tb[TCA_FLOW_XOR])
550		f->xor = nla_get_u32(tb[TCA_FLOW_XOR]);
551	if (tb[TCA_FLOW_RSHIFT])
552		f->rshift = nla_get_u32(tb[TCA_FLOW_RSHIFT]);
553	if (tb[TCA_FLOW_ADDEND])
554		f->addend = nla_get_u32(tb[TCA_FLOW_ADDEND]);
555
556	if (tb[TCA_FLOW_DIVISOR])
557		f->divisor = nla_get_u32(tb[TCA_FLOW_DIVISOR]);
558	if (baseclass)
559		f->baseclass = baseclass;
560
561	f->perturb_period = perturb_period;
562	del_timer(&f->perturb_timer);
563	if (perturb_period)
564		mod_timer(&f->perturb_timer, jiffies + perturb_period);
565
566	if (*arg == 0)
567		list_add_tail(&f->list, &head->filters);
 
 
568
569	tcf_tree_unlock(tp);
570
571	*arg = (unsigned long)f;
 
572	return 0;
573
 
 
574err2:
575	tcf_em_tree_destroy(tp, &t);
 
576err1:
577	tcf_exts_destroy(tp, &e);
578	return err;
579}
580
581static void flow_destroy_filter(struct tcf_proto *tp, struct flow_filter *f)
582{
583	del_timer_sync(&f->perturb_timer);
584	tcf_exts_destroy(tp, &f->exts);
585	tcf_em_tree_destroy(tp, &f->ematches);
586	kfree(f);
587}
588
589static int flow_delete(struct tcf_proto *tp, unsigned long arg)
590{
591	struct flow_filter *f = (struct flow_filter *)arg;
592
593	tcf_tree_lock(tp);
594	list_del(&f->list);
595	tcf_tree_unlock(tp);
596	flow_destroy_filter(tp, f);
597	return 0;
598}
599
600static int flow_init(struct tcf_proto *tp)
601{
602	struct flow_head *head;
603
604	head = kzalloc(sizeof(*head), GFP_KERNEL);
605	if (head == NULL)
606		return -ENOBUFS;
607	INIT_LIST_HEAD(&head->filters);
608	tp->root = head;
609	return 0;
610}
611
612static void flow_destroy(struct tcf_proto *tp)
613{
614	struct flow_head *head = tp->root;
615	struct flow_filter *f, *next;
616
 
 
 
617	list_for_each_entry_safe(f, next, &head->filters, list) {
618		list_del(&f->list);
619		flow_destroy_filter(tp, f);
620	}
621	kfree(head);
 
622}
623
624static unsigned long flow_get(struct tcf_proto *tp, u32 handle)
625{
626	struct flow_head *head = tp->root;
627	struct flow_filter *f;
628
629	list_for_each_entry(f, &head->filters, list)
630		if (f->handle == handle)
631			return (unsigned long)f;
632	return 0;
633}
634
635static void flow_put(struct tcf_proto *tp, unsigned long f)
636{
637}
638
639static int flow_dump(struct tcf_proto *tp, unsigned long fh,
640		     struct sk_buff *skb, struct tcmsg *t)
641{
642	struct flow_filter *f = (struct flow_filter *)fh;
643	struct nlattr *nest;
644
645	if (f == NULL)
646		return skb->len;
647
648	t->tcm_handle = f->handle;
649
650	nest = nla_nest_start(skb, TCA_OPTIONS);
651	if (nest == NULL)
652		goto nla_put_failure;
653
654	NLA_PUT_U32(skb, TCA_FLOW_KEYS, f->keymask);
655	NLA_PUT_U32(skb, TCA_FLOW_MODE, f->mode);
 
656
657	if (f->mask != ~0 || f->xor != 0) {
658		NLA_PUT_U32(skb, TCA_FLOW_MASK, f->mask);
659		NLA_PUT_U32(skb, TCA_FLOW_XOR, f->xor);
 
660	}
661	if (f->rshift)
662		NLA_PUT_U32(skb, TCA_FLOW_RSHIFT, f->rshift);
663	if (f->addend)
664		NLA_PUT_U32(skb, TCA_FLOW_ADDEND, f->addend);
665
666	if (f->divisor)
667		NLA_PUT_U32(skb, TCA_FLOW_DIVISOR, f->divisor);
668	if (f->baseclass)
669		NLA_PUT_U32(skb, TCA_FLOW_BASECLASS, f->baseclass);
670
671	if (f->perturb_period)
672		NLA_PUT_U32(skb, TCA_FLOW_PERTURB, f->perturb_period / HZ);
 
 
 
 
 
 
 
 
673
674	if (tcf_exts_dump(skb, &f->exts, &flow_ext_map) < 0)
675		goto nla_put_failure;
676#ifdef CONFIG_NET_EMATCH
677	if (f->ematches.hdr.nmatches &&
678	    tcf_em_tree_dump(skb, &f->ematches, TCA_FLOW_EMATCHES) < 0)
679		goto nla_put_failure;
680#endif
681	nla_nest_end(skb, nest);
682
683	if (tcf_exts_dump_stats(skb, &f->exts, &flow_ext_map) < 0)
684		goto nla_put_failure;
685
686	return skb->len;
687
688nla_put_failure:
689	nlmsg_trim(skb, nest);
690	return -1;
691}
692
693static void flow_walk(struct tcf_proto *tp, struct tcf_walker *arg)
694{
695	struct flow_head *head = tp->root;
696	struct flow_filter *f;
697
698	list_for_each_entry(f, &head->filters, list) {
699		if (arg->count < arg->skip)
700			goto skip;
701		if (arg->fn(tp, (unsigned long)f, arg) < 0) {
702			arg->stop = 1;
703			break;
704		}
705skip:
706		arg->count++;
707	}
708}
709
710static struct tcf_proto_ops cls_flow_ops __read_mostly = {
711	.kind		= "flow",
712	.classify	= flow_classify,
713	.init		= flow_init,
714	.destroy	= flow_destroy,
715	.change		= flow_change,
716	.delete		= flow_delete,
717	.get		= flow_get,
718	.put		= flow_put,
719	.dump		= flow_dump,
720	.walk		= flow_walk,
721	.owner		= THIS_MODULE,
722};
723
724static int __init cls_flow_init(void)
725{
726	return register_tcf_proto_ops(&cls_flow_ops);
727}
728
729static void __exit cls_flow_exit(void)
730{
731	unregister_tcf_proto_ops(&cls_flow_ops);
732}
733
734module_init(cls_flow_init);
735module_exit(cls_flow_exit);
736
737MODULE_LICENSE("GPL");
738MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
739MODULE_DESCRIPTION("TC flow classifier");
v4.10.11
  1/*
  2 * net/sched/cls_flow.c		Generic flow classifier
  3 *
  4 * Copyright (c) 2007, 2008 Patrick McHardy <kaber@trash.net>
  5 *
  6 * This program is free software; you can redistribute it and/or
  7 * modify it under the terms of the GNU General Public License
  8 * as published by the Free Software Foundation; either version 2
  9 * of the License, or (at your option) any later version.
 10 */
 11
 12#include <linux/kernel.h>
 13#include <linux/init.h>
 14#include <linux/list.h>
 15#include <linux/jhash.h>
 16#include <linux/random.h>
 17#include <linux/pkt_cls.h>
 18#include <linux/skbuff.h>
 19#include <linux/in.h>
 20#include <linux/ip.h>
 21#include <linux/ipv6.h>
 22#include <linux/if_vlan.h>
 23#include <linux/slab.h>
 24#include <linux/module.h>
 25#include <net/inet_sock.h>
 26
 27#include <net/pkt_cls.h>
 28#include <net/ip.h>
 29#include <net/route.h>
 30#include <net/flow_dissector.h>
 31
 32#if IS_ENABLED(CONFIG_NF_CONNTRACK)
 33#include <net/netfilter/nf_conntrack.h>
 34#endif
 35
 36struct flow_head {
 37	struct list_head	filters;
 38	struct rcu_head		rcu;
 39};
 40
 41struct flow_filter {
 42	struct list_head	list;
 43	struct tcf_exts		exts;
 44	struct tcf_ematch_tree	ematches;
 45	struct tcf_proto	*tp;
 46	struct timer_list	perturb_timer;
 47	u32			perturb_period;
 48	u32			handle;
 49
 50	u32			nkeys;
 51	u32			keymask;
 52	u32			mode;
 53	u32			mask;
 54	u32			xor;
 55	u32			rshift;
 56	u32			addend;
 57	u32			divisor;
 58	u32			baseclass;
 59	u32			hashrnd;
 60	struct rcu_head		rcu;
 
 
 
 
 61};
 62
 63static inline u32 addr_fold(void *addr)
 64{
 65	unsigned long a = (unsigned long)addr;
 66
 67	return (a & 0xFFFFFFFF) ^ (BITS_PER_LONG > 32 ? a >> 32 : 0);
 68}
 69
 70static u32 flow_get_src(const struct sk_buff *skb, const struct flow_keys *flow)
 71{
 72	__be32 src = flow_get_u32_src(flow);
 73
 74	if (src)
 75		return ntohl(src);
 
 
 
 
 
 
 76
 77	return addr_fold(skb->sk);
 78}
 79
 80static u32 flow_get_dst(const struct sk_buff *skb, const struct flow_keys *flow)
 81{
 82	__be32 dst = flow_get_u32_dst(flow);
 83
 84	if (dst)
 85		return ntohl(dst);
 
 
 
 
 
 
 86
 87	return addr_fold(skb_dst(skb)) ^ (__force u16) tc_skb_protocol(skb);
 88}
 89
 90static u32 flow_get_proto(const struct sk_buff *skb,
 91			  const struct flow_keys *flow)
 92{
 93	return flow->basic.ip_proto;
 
 
 
 
 
 
 
 
 
 94}
 95
 96static u32 flow_get_proto_src(const struct sk_buff *skb,
 97			      const struct flow_keys *flow)
 98{
 99	if (flow->ports.ports)
100		return ntohs(flow->ports.src);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
101
102	return addr_fold(skb->sk);
103}
104
105static u32 flow_get_proto_dst(const struct sk_buff *skb,
106			      const struct flow_keys *flow)
107{
108	if (flow->ports.ports)
109		return ntohs(flow->ports.dst);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
110
111	return addr_fold(skb_dst(skb)) ^ (__force u16) tc_skb_protocol(skb);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
112}
113
114static u32 flow_get_iif(const struct sk_buff *skb)
115{
116	return skb->skb_iif;
117}
118
119static u32 flow_get_priority(const struct sk_buff *skb)
120{
121	return skb->priority;
122}
123
124static u32 flow_get_mark(const struct sk_buff *skb)
125{
126	return skb->mark;
127}
128
129static u32 flow_get_nfct(const struct sk_buff *skb)
130{
131#if IS_ENABLED(CONFIG_NF_CONNTRACK)
132	return addr_fold(skb->nfct);
133#else
134	return 0;
135#endif
136}
137
138#if IS_ENABLED(CONFIG_NF_CONNTRACK)
139#define CTTUPLE(skb, member)						\
140({									\
141	enum ip_conntrack_info ctinfo;					\
142	const struct nf_conn *ct = nf_ct_get(skb, &ctinfo);		\
143	if (ct == NULL)							\
144		goto fallback;						\
145	ct->tuplehash[CTINFO2DIR(ctinfo)].tuple.member;			\
146})
147#else
148#define CTTUPLE(skb, member)						\
149({									\
150	goto fallback;							\
151	0;								\
152})
153#endif
154
155static u32 flow_get_nfct_src(const struct sk_buff *skb,
156			     const struct flow_keys *flow)
157{
158	switch (tc_skb_protocol(skb)) {
159	case htons(ETH_P_IP):
160		return ntohl(CTTUPLE(skb, src.u3.ip));
161	case htons(ETH_P_IPV6):
162		return ntohl(CTTUPLE(skb, src.u3.ip6[3]));
163	}
164fallback:
165	return flow_get_src(skb, flow);
166}
167
168static u32 flow_get_nfct_dst(const struct sk_buff *skb,
169			     const struct flow_keys *flow)
170{
171	switch (tc_skb_protocol(skb)) {
172	case htons(ETH_P_IP):
173		return ntohl(CTTUPLE(skb, dst.u3.ip));
174	case htons(ETH_P_IPV6):
175		return ntohl(CTTUPLE(skb, dst.u3.ip6[3]));
176	}
177fallback:
178	return flow_get_dst(skb, flow);
179}
180
181static u32 flow_get_nfct_proto_src(const struct sk_buff *skb,
182				   const struct flow_keys *flow)
183{
184	return ntohs(CTTUPLE(skb, src.u.all));
185fallback:
186	return flow_get_proto_src(skb, flow);
187}
188
189static u32 flow_get_nfct_proto_dst(const struct sk_buff *skb,
190				   const struct flow_keys *flow)
191{
192	return ntohs(CTTUPLE(skb, dst.u.all));
193fallback:
194	return flow_get_proto_dst(skb, flow);
195}
196
197static u32 flow_get_rtclassid(const struct sk_buff *skb)
198{
199#ifdef CONFIG_IP_ROUTE_CLASSID
200	if (skb_dst(skb))
201		return skb_dst(skb)->tclassid;
202#endif
203	return 0;
204}
205
206static u32 flow_get_skuid(const struct sk_buff *skb)
207{
208	struct sock *sk = skb_to_full_sk(skb);
209
210	if (sk && sk->sk_socket && sk->sk_socket->file) {
211		kuid_t skuid = sk->sk_socket->file->f_cred->fsuid;
212
213		return from_kuid(&init_user_ns, skuid);
214	}
215	return 0;
216}
217
218static u32 flow_get_skgid(const struct sk_buff *skb)
219{
220	struct sock *sk = skb_to_full_sk(skb);
221
222	if (sk && sk->sk_socket && sk->sk_socket->file) {
223		kgid_t skgid = sk->sk_socket->file->f_cred->fsgid;
224
225		return from_kgid(&init_user_ns, skgid);
226	}
227	return 0;
228}
229
230static u32 flow_get_vlan_tag(const struct sk_buff *skb)
231{
232	u16 uninitialized_var(tag);
233
234	if (vlan_get_tag(skb, &tag) < 0)
235		return 0;
236	return tag & VLAN_VID_MASK;
237}
238
239static u32 flow_get_rxhash(struct sk_buff *skb)
240{
241	return skb_get_hash(skb);
242}
243
244static u32 flow_key_get(struct sk_buff *skb, int key, struct flow_keys *flow)
245{
246	switch (key) {
247	case FLOW_KEY_SRC:
248		return flow_get_src(skb, flow);
249	case FLOW_KEY_DST:
250		return flow_get_dst(skb, flow);
251	case FLOW_KEY_PROTO:
252		return flow_get_proto(skb, flow);
253	case FLOW_KEY_PROTO_SRC:
254		return flow_get_proto_src(skb, flow);
255	case FLOW_KEY_PROTO_DST:
256		return flow_get_proto_dst(skb, flow);
257	case FLOW_KEY_IIF:
258		return flow_get_iif(skb);
259	case FLOW_KEY_PRIORITY:
260		return flow_get_priority(skb);
261	case FLOW_KEY_MARK:
262		return flow_get_mark(skb);
263	case FLOW_KEY_NFCT:
264		return flow_get_nfct(skb);
265	case FLOW_KEY_NFCT_SRC:
266		return flow_get_nfct_src(skb, flow);
267	case FLOW_KEY_NFCT_DST:
268		return flow_get_nfct_dst(skb, flow);
269	case FLOW_KEY_NFCT_PROTO_SRC:
270		return flow_get_nfct_proto_src(skb, flow);
271	case FLOW_KEY_NFCT_PROTO_DST:
272		return flow_get_nfct_proto_dst(skb, flow);
273	case FLOW_KEY_RTCLASSID:
274		return flow_get_rtclassid(skb);
275	case FLOW_KEY_SKUID:
276		return flow_get_skuid(skb);
277	case FLOW_KEY_SKGID:
278		return flow_get_skgid(skb);
279	case FLOW_KEY_VLAN_TAG:
280		return flow_get_vlan_tag(skb);
281	case FLOW_KEY_RXHASH:
282		return flow_get_rxhash(skb);
283	default:
284		WARN_ON(1);
285		return 0;
286	}
287}
288
289#define FLOW_KEYS_NEEDED ((1 << FLOW_KEY_SRC) | 		\
290			  (1 << FLOW_KEY_DST) |			\
291			  (1 << FLOW_KEY_PROTO) |		\
292			  (1 << FLOW_KEY_PROTO_SRC) |		\
293			  (1 << FLOW_KEY_PROTO_DST) | 		\
294			  (1 << FLOW_KEY_NFCT_SRC) |		\
295			  (1 << FLOW_KEY_NFCT_DST) |		\
296			  (1 << FLOW_KEY_NFCT_PROTO_SRC) |	\
297			  (1 << FLOW_KEY_NFCT_PROTO_DST))
298
299static int flow_classify(struct sk_buff *skb, const struct tcf_proto *tp,
300			 struct tcf_result *res)
301{
302	struct flow_head *head = rcu_dereference_bh(tp->root);
303	struct flow_filter *f;
304	u32 keymask;
305	u32 classid;
306	unsigned int n, key;
307	int r;
308
309	list_for_each_entry_rcu(f, &head->filters, list) {
310		u32 keys[FLOW_KEY_MAX + 1];
311		struct flow_keys flow_keys;
312
313		if (!tcf_em_tree_match(skb, &f->ematches, NULL))
314			continue;
315
316		keymask = f->keymask;
317		if (keymask & FLOW_KEYS_NEEDED)
318			skb_flow_dissect_flow_keys(skb, &flow_keys, 0);
319
320		for (n = 0; n < f->nkeys; n++) {
321			key = ffs(keymask) - 1;
322			keymask &= ~(1 << key);
323			keys[n] = flow_key_get(skb, key, &flow_keys);
324		}
325
326		if (f->mode == FLOW_MODE_HASH)
327			classid = jhash2(keys, f->nkeys, f->hashrnd);
328		else {
329			classid = keys[0];
330			classid = (classid & f->mask) ^ f->xor;
331			classid = (classid >> f->rshift) + f->addend;
332		}
333
334		if (f->divisor)
335			classid %= f->divisor;
336
337		res->class   = 0;
338		res->classid = TC_H_MAKE(f->baseclass, f->baseclass + classid);
339
340		r = tcf_exts_exec(skb, &f->exts, res);
341		if (r < 0)
342			continue;
343		return r;
344	}
345	return -1;
346}
347
348static void flow_perturbation(unsigned long arg)
349{
350	struct flow_filter *f = (struct flow_filter *)arg;
351
352	get_random_bytes(&f->hashrnd, 4);
353	if (f->perturb_period)
354		mod_timer(&f->perturb_timer, jiffies + f->perturb_period);
355}
356
357static const struct nla_policy flow_policy[TCA_FLOW_MAX + 1] = {
358	[TCA_FLOW_KEYS]		= { .type = NLA_U32 },
359	[TCA_FLOW_MODE]		= { .type = NLA_U32 },
360	[TCA_FLOW_BASECLASS]	= { .type = NLA_U32 },
361	[TCA_FLOW_RSHIFT]	= { .type = NLA_U32 },
362	[TCA_FLOW_ADDEND]	= { .type = NLA_U32 },
363	[TCA_FLOW_MASK]		= { .type = NLA_U32 },
364	[TCA_FLOW_XOR]		= { .type = NLA_U32 },
365	[TCA_FLOW_DIVISOR]	= { .type = NLA_U32 },
366	[TCA_FLOW_ACT]		= { .type = NLA_NESTED },
367	[TCA_FLOW_POLICE]	= { .type = NLA_NESTED },
368	[TCA_FLOW_EMATCHES]	= { .type = NLA_NESTED },
369	[TCA_FLOW_PERTURB]	= { .type = NLA_U32 },
370};
371
372static void flow_destroy_filter(struct rcu_head *head)
373{
374	struct flow_filter *f = container_of(head, struct flow_filter, rcu);
375
376	del_timer_sync(&f->perturb_timer);
377	tcf_exts_destroy(&f->exts);
378	tcf_em_tree_destroy(&f->ematches);
379	kfree(f);
380}
381
382static int flow_change(struct net *net, struct sk_buff *in_skb,
383		       struct tcf_proto *tp, unsigned long base,
384		       u32 handle, struct nlattr **tca,
385		       unsigned long *arg, bool ovr)
386{
387	struct flow_head *head = rtnl_dereference(tp->root);
388	struct flow_filter *fold, *fnew;
389	struct nlattr *opt = tca[TCA_OPTIONS];
390	struct nlattr *tb[TCA_FLOW_MAX + 1];
391	struct tcf_exts e;
392	struct tcf_ematch_tree t;
393	unsigned int nkeys = 0;
394	unsigned int perturb_period = 0;
395	u32 baseclass = 0;
396	u32 keymask = 0;
397	u32 mode;
398	int err;
399
400	if (opt == NULL)
401		return -EINVAL;
402
403	err = nla_parse_nested(tb, TCA_FLOW_MAX, opt, flow_policy);
404	if (err < 0)
405		return err;
406
407	if (tb[TCA_FLOW_BASECLASS]) {
408		baseclass = nla_get_u32(tb[TCA_FLOW_BASECLASS]);
409		if (TC_H_MIN(baseclass) == 0)
410			return -EINVAL;
411	}
412
413	if (tb[TCA_FLOW_KEYS]) {
414		keymask = nla_get_u32(tb[TCA_FLOW_KEYS]);
415
416		nkeys = hweight32(keymask);
417		if (nkeys == 0)
418			return -EINVAL;
419
420		if (fls(keymask) - 1 > FLOW_KEY_MAX)
421			return -EOPNOTSUPP;
422
423		if ((keymask & (FLOW_KEY_SKUID|FLOW_KEY_SKGID)) &&
424		    sk_user_ns(NETLINK_CB(in_skb).sk) != &init_user_ns)
425			return -EOPNOTSUPP;
426	}
427
428	err = tcf_exts_init(&e, TCA_FLOW_ACT, TCA_FLOW_POLICE);
429	if (err < 0)
430		goto err1;
431	err = tcf_exts_validate(net, tp, tb, tca[TCA_RATE], &e, ovr);
432	if (err < 0)
433		goto err1;
434
435	err = tcf_em_tree_validate(tp, tb[TCA_FLOW_EMATCHES], &t);
436	if (err < 0)
437		goto err1;
438
439	err = -ENOBUFS;
440	fnew = kzalloc(sizeof(*fnew), GFP_KERNEL);
441	if (!fnew)
442		goto err2;
443
444	err = tcf_exts_init(&fnew->exts, TCA_FLOW_ACT, TCA_FLOW_POLICE);
445	if (err < 0)
446		goto err3;
447
448	fold = (struct flow_filter *)*arg;
449	if (fold) {
450		err = -EINVAL;
451		if (fold->handle != handle && handle)
452			goto err3;
453
454		/* Copy fold into fnew */
455		fnew->tp = fold->tp;
456		fnew->handle = fold->handle;
457		fnew->nkeys = fold->nkeys;
458		fnew->keymask = fold->keymask;
459		fnew->mode = fold->mode;
460		fnew->mask = fold->mask;
461		fnew->xor = fold->xor;
462		fnew->rshift = fold->rshift;
463		fnew->addend = fold->addend;
464		fnew->divisor = fold->divisor;
465		fnew->baseclass = fold->baseclass;
466		fnew->hashrnd = fold->hashrnd;
467
468		mode = fold->mode;
469		if (tb[TCA_FLOW_MODE])
470			mode = nla_get_u32(tb[TCA_FLOW_MODE]);
471		if (mode != FLOW_MODE_HASH && nkeys > 1)
472			goto err3;
473
474		if (mode == FLOW_MODE_HASH)
475			perturb_period = fold->perturb_period;
476		if (tb[TCA_FLOW_PERTURB]) {
477			if (mode != FLOW_MODE_HASH)
478				goto err3;
479			perturb_period = nla_get_u32(tb[TCA_FLOW_PERTURB]) * HZ;
480		}
481	} else {
482		err = -EINVAL;
483		if (!handle)
484			goto err3;
485		if (!tb[TCA_FLOW_KEYS])
486			goto err3;
487
488		mode = FLOW_MODE_MAP;
489		if (tb[TCA_FLOW_MODE])
490			mode = nla_get_u32(tb[TCA_FLOW_MODE]);
491		if (mode != FLOW_MODE_HASH && nkeys > 1)
492			goto err3;
493
494		if (tb[TCA_FLOW_PERTURB]) {
495			if (mode != FLOW_MODE_HASH)
496				goto err3;
497			perturb_period = nla_get_u32(tb[TCA_FLOW_PERTURB]) * HZ;
498		}
499
500		if (TC_H_MAJ(baseclass) == 0)
501			baseclass = TC_H_MAKE(tp->q->handle, baseclass);
502		if (TC_H_MIN(baseclass) == 0)
503			baseclass = TC_H_MAKE(baseclass, 1);
504
505		fnew->handle = handle;
506		fnew->mask  = ~0U;
507		fnew->tp = tp;
508		get_random_bytes(&fnew->hashrnd, 4);
 
 
 
 
 
 
 
 
509	}
510
511	fnew->perturb_timer.function = flow_perturbation;
512	fnew->perturb_timer.data = (unsigned long)fnew;
513	init_timer_deferrable(&fnew->perturb_timer);
514
515	tcf_exts_change(tp, &fnew->exts, &e);
516	tcf_em_tree_change(tp, &fnew->ematches, &t);
517
518	netif_keep_dst(qdisc_dev(tp->q));
519
520	if (tb[TCA_FLOW_KEYS]) {
521		fnew->keymask = keymask;
522		fnew->nkeys   = nkeys;
523	}
524
525	fnew->mode = mode;
526
527	if (tb[TCA_FLOW_MASK])
528		fnew->mask = nla_get_u32(tb[TCA_FLOW_MASK]);
529	if (tb[TCA_FLOW_XOR])
530		fnew->xor = nla_get_u32(tb[TCA_FLOW_XOR]);
531	if (tb[TCA_FLOW_RSHIFT])
532		fnew->rshift = nla_get_u32(tb[TCA_FLOW_RSHIFT]);
533	if (tb[TCA_FLOW_ADDEND])
534		fnew->addend = nla_get_u32(tb[TCA_FLOW_ADDEND]);
535
536	if (tb[TCA_FLOW_DIVISOR])
537		fnew->divisor = nla_get_u32(tb[TCA_FLOW_DIVISOR]);
538	if (baseclass)
539		fnew->baseclass = baseclass;
540
541	fnew->perturb_period = perturb_period;
 
542	if (perturb_period)
543		mod_timer(&fnew->perturb_timer, jiffies + perturb_period);
544
545	if (*arg == 0)
546		list_add_tail_rcu(&fnew->list, &head->filters);
547	else
548		list_replace_rcu(&fold->list, &fnew->list);
549
550	*arg = (unsigned long)fnew;
551
552	if (fold)
553		call_rcu(&fold->rcu, flow_destroy_filter);
554	return 0;
555
556err3:
557	tcf_exts_destroy(&fnew->exts);
558err2:
559	tcf_em_tree_destroy(&t);
560	kfree(fnew);
561err1:
562	tcf_exts_destroy(&e);
563	return err;
564}
565
 
 
 
 
 
 
 
 
566static int flow_delete(struct tcf_proto *tp, unsigned long arg)
567{
568	struct flow_filter *f = (struct flow_filter *)arg;
569
570	list_del_rcu(&f->list);
571	call_rcu(&f->rcu, flow_destroy_filter);
 
 
572	return 0;
573}
574
575static int flow_init(struct tcf_proto *tp)
576{
577	struct flow_head *head;
578
579	head = kzalloc(sizeof(*head), GFP_KERNEL);
580	if (head == NULL)
581		return -ENOBUFS;
582	INIT_LIST_HEAD(&head->filters);
583	rcu_assign_pointer(tp->root, head);
584	return 0;
585}
586
587static bool flow_destroy(struct tcf_proto *tp, bool force)
588{
589	struct flow_head *head = rtnl_dereference(tp->root);
590	struct flow_filter *f, *next;
591
592	if (!force && !list_empty(&head->filters))
593		return false;
594
595	list_for_each_entry_safe(f, next, &head->filters, list) {
596		list_del_rcu(&f->list);
597		call_rcu(&f->rcu, flow_destroy_filter);
598	}
599	kfree_rcu(head, rcu);
600	return true;
601}
602
603static unsigned long flow_get(struct tcf_proto *tp, u32 handle)
604{
605	struct flow_head *head = rtnl_dereference(tp->root);
606	struct flow_filter *f;
607
608	list_for_each_entry(f, &head->filters, list)
609		if (f->handle == handle)
610			return (unsigned long)f;
611	return 0;
612}
613
614static int flow_dump(struct net *net, struct tcf_proto *tp, unsigned long fh,
 
 
 
 
615		     struct sk_buff *skb, struct tcmsg *t)
616{
617	struct flow_filter *f = (struct flow_filter *)fh;
618	struct nlattr *nest;
619
620	if (f == NULL)
621		return skb->len;
622
623	t->tcm_handle = f->handle;
624
625	nest = nla_nest_start(skb, TCA_OPTIONS);
626	if (nest == NULL)
627		goto nla_put_failure;
628
629	if (nla_put_u32(skb, TCA_FLOW_KEYS, f->keymask) ||
630	    nla_put_u32(skb, TCA_FLOW_MODE, f->mode))
631		goto nla_put_failure;
632
633	if (f->mask != ~0 || f->xor != 0) {
634		if (nla_put_u32(skb, TCA_FLOW_MASK, f->mask) ||
635		    nla_put_u32(skb, TCA_FLOW_XOR, f->xor))
636			goto nla_put_failure;
637	}
638	if (f->rshift &&
639	    nla_put_u32(skb, TCA_FLOW_RSHIFT, f->rshift))
640		goto nla_put_failure;
641	if (f->addend &&
642	    nla_put_u32(skb, TCA_FLOW_ADDEND, f->addend))
643		goto nla_put_failure;
 
 
 
644
645	if (f->divisor &&
646	    nla_put_u32(skb, TCA_FLOW_DIVISOR, f->divisor))
647		goto nla_put_failure;
648	if (f->baseclass &&
649	    nla_put_u32(skb, TCA_FLOW_BASECLASS, f->baseclass))
650		goto nla_put_failure;
651
652	if (f->perturb_period &&
653	    nla_put_u32(skb, TCA_FLOW_PERTURB, f->perturb_period / HZ))
654		goto nla_put_failure;
655
656	if (tcf_exts_dump(skb, &f->exts) < 0)
657		goto nla_put_failure;
658#ifdef CONFIG_NET_EMATCH
659	if (f->ematches.hdr.nmatches &&
660	    tcf_em_tree_dump(skb, &f->ematches, TCA_FLOW_EMATCHES) < 0)
661		goto nla_put_failure;
662#endif
663	nla_nest_end(skb, nest);
664
665	if (tcf_exts_dump_stats(skb, &f->exts) < 0)
666		goto nla_put_failure;
667
668	return skb->len;
669
670nla_put_failure:
671	nla_nest_cancel(skb, nest);
672	return -1;
673}
674
675static void flow_walk(struct tcf_proto *tp, struct tcf_walker *arg)
676{
677	struct flow_head *head = rtnl_dereference(tp->root);
678	struct flow_filter *f;
679
680	list_for_each_entry(f, &head->filters, list) {
681		if (arg->count < arg->skip)
682			goto skip;
683		if (arg->fn(tp, (unsigned long)f, arg) < 0) {
684			arg->stop = 1;
685			break;
686		}
687skip:
688		arg->count++;
689	}
690}
691
692static struct tcf_proto_ops cls_flow_ops __read_mostly = {
693	.kind		= "flow",
694	.classify	= flow_classify,
695	.init		= flow_init,
696	.destroy	= flow_destroy,
697	.change		= flow_change,
698	.delete		= flow_delete,
699	.get		= flow_get,
 
700	.dump		= flow_dump,
701	.walk		= flow_walk,
702	.owner		= THIS_MODULE,
703};
704
705static int __init cls_flow_init(void)
706{
707	return register_tcf_proto_ops(&cls_flow_ops);
708}
709
710static void __exit cls_flow_exit(void)
711{
712	unregister_tcf_proto_ops(&cls_flow_ops);
713}
714
715module_init(cls_flow_init);
716module_exit(cls_flow_exit);
717
718MODULE_LICENSE("GPL");
719MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
720MODULE_DESCRIPTION("TC flow classifier");