blob: 455fc8f83d0ae0ab48ba97fc2a11ce99e8e00784 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * net/sched/cls_route.c ROUTE4 classifier.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version
7 * 2 of the License, or (at your option) any later version.
8 *
9 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
10 */
11
12#include <linux/module.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090013#include <linux/slab.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070014#include <linux/types.h>
15#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070016#include <linux/string.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070017#include <linux/errno.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070018#include <linux/skbuff.h>
Patrick McHardy0ba48052007-07-02 22:49:07 -070019#include <net/dst.h>
20#include <net/route.h>
21#include <net/netlink.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070022#include <net/act_api.h>
23#include <net/pkt_cls.h>
24
25/*
Eric Dumazetcc7ec452011-01-19 19:26:56 +000026 * 1. For now we assume that route tags < 256.
27 * It allows to use direct table lookups, instead of hash tables.
28 * 2. For now we assume that "from TAG" and "fromdev DEV" statements
29 * are mutually exclusive.
30 * 3. "to TAG from ANY" has higher priority, than "to ANY from XXX"
Linus Torvalds1da177e2005-04-16 15:20:36 -070031 */
Eric Dumazetcc7ec452011-01-19 19:26:56 +000032struct route4_fastmap {
John Fastabend1109c002014-09-12 20:07:50 -070033 struct route4_filter *filter;
34 u32 id;
35 int iif;
Linus Torvalds1da177e2005-04-16 15:20:36 -070036};
37
Eric Dumazetcc7ec452011-01-19 19:26:56 +000038struct route4_head {
John Fastabend1109c002014-09-12 20:07:50 -070039 struct route4_fastmap fastmap[16];
40 struct route4_bucket __rcu *table[256 + 1];
41 struct rcu_head rcu;
Linus Torvalds1da177e2005-04-16 15:20:36 -070042};
43
Eric Dumazetcc7ec452011-01-19 19:26:56 +000044struct route4_bucket {
Linus Torvalds1da177e2005-04-16 15:20:36 -070045 /* 16 FROM buckets + 16 IIF buckets + 1 wildcard bucket */
John Fastabend1109c002014-09-12 20:07:50 -070046 struct route4_filter __rcu *ht[16 + 16 + 1];
47 struct rcu_head rcu;
Linus Torvalds1da177e2005-04-16 15:20:36 -070048};
49
Eric Dumazetcc7ec452011-01-19 19:26:56 +000050struct route4_filter {
John Fastabend1109c002014-09-12 20:07:50 -070051 struct route4_filter __rcu *next;
Linus Torvalds1da177e2005-04-16 15:20:36 -070052 u32 id;
53 int iif;
54
55 struct tcf_result res;
56 struct tcf_exts exts;
57 u32 handle;
58 struct route4_bucket *bkt;
John Fastabend1109c002014-09-12 20:07:50 -070059 struct tcf_proto *tp;
60 struct rcu_head rcu;
Linus Torvalds1da177e2005-04-16 15:20:36 -070061};
62
Eric Dumazetcc7ec452011-01-19 19:26:56 +000063#define ROUTE4_FAILURE ((struct route4_filter *)(-1L))
Linus Torvalds1da177e2005-04-16 15:20:36 -070064
Eric Dumazetcc7ec452011-01-19 19:26:56 +000065static inline int route4_fastmap_hash(u32 id, int iif)
Linus Torvalds1da177e2005-04-16 15:20:36 -070066{
Eric Dumazetcc7ec452011-01-19 19:26:56 +000067 return id & 0xF;
Linus Torvalds1da177e2005-04-16 15:20:36 -070068}
69
John Fastabend1109c002014-09-12 20:07:50 -070070static DEFINE_SPINLOCK(fastmap_lock);
Eric Dumazetcc7ec452011-01-19 19:26:56 +000071static void
John Fastabend1109c002014-09-12 20:07:50 -070072route4_reset_fastmap(struct route4_head *head)
Linus Torvalds1da177e2005-04-16 15:20:36 -070073{
John Fastabend1109c002014-09-12 20:07:50 -070074 spin_lock_bh(&fastmap_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -070075 memset(head->fastmap, 0, sizeof(head->fastmap));
John Fastabend1109c002014-09-12 20:07:50 -070076 spin_unlock_bh(&fastmap_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -070077}
78
Eric Dumazetcc7ec452011-01-19 19:26:56 +000079static void
Linus Torvalds1da177e2005-04-16 15:20:36 -070080route4_set_fastmap(struct route4_head *head, u32 id, int iif,
81 struct route4_filter *f)
82{
83 int h = route4_fastmap_hash(id, iif);
Eric Dumazetcc7ec452011-01-19 19:26:56 +000084
John Fastabend1109c002014-09-12 20:07:50 -070085 /* fastmap updates must look atomic to aling id, iff, filter */
86 spin_lock_bh(&fastmap_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -070087 head->fastmap[h].id = id;
88 head->fastmap[h].iif = iif;
89 head->fastmap[h].filter = f;
John Fastabend1109c002014-09-12 20:07:50 -070090 spin_unlock_bh(&fastmap_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -070091}
92
Eric Dumazetcc7ec452011-01-19 19:26:56 +000093static inline int route4_hash_to(u32 id)
Linus Torvalds1da177e2005-04-16 15:20:36 -070094{
Eric Dumazetcc7ec452011-01-19 19:26:56 +000095 return id & 0xFF;
Linus Torvalds1da177e2005-04-16 15:20:36 -070096}
97
Eric Dumazetcc7ec452011-01-19 19:26:56 +000098static inline int route4_hash_from(u32 id)
Linus Torvalds1da177e2005-04-16 15:20:36 -070099{
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000100 return (id >> 16) & 0xF;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700101}
102
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000103static inline int route4_hash_iif(int iif)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700104{
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000105 return 16 + ((iif >> 16) & 0xF);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700106}
107
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000108static inline int route4_hash_wild(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700109{
110 return 32;
111}
112
113#define ROUTE4_APPLY_RESULT() \
114{ \
115 *res = f->res; \
116 if (tcf_exts_is_available(&f->exts)) { \
117 int r = tcf_exts_exec(skb, &f->exts, res); \
118 if (r < 0) { \
119 dont_cache = 1; \
120 continue; \
121 } \
122 return r; \
123 } else if (!dont_cache) \
124 route4_set_fastmap(head, id, iif, f); \
125 return 0; \
126}
127
Eric Dumazetdc7f9f62011-07-05 23:25:42 +0000128static int route4_classify(struct sk_buff *skb, const struct tcf_proto *tp,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700129 struct tcf_result *res)
130{
John Fastabend1109c002014-09-12 20:07:50 -0700131 struct route4_head *head = rcu_dereference_bh(tp->root);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700132 struct dst_entry *dst;
133 struct route4_bucket *b;
134 struct route4_filter *f;
135 u32 id, h;
136 int iif, dont_cache = 0;
137
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000138 dst = skb_dst(skb);
139 if (!dst)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700140 goto failure;
141
142 id = dst->tclassid;
143 if (head == NULL)
144 goto old_method;
145
David S. Miller92101b32012-07-23 16:29:00 -0700146 iif = inet_iif(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700147
148 h = route4_fastmap_hash(id, iif);
John Fastabend1109c002014-09-12 20:07:50 -0700149
150 spin_lock(&fastmap_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700151 if (id == head->fastmap[h].id &&
152 iif == head->fastmap[h].iif &&
153 (f = head->fastmap[h].filter) != NULL) {
John Fastabend1109c002014-09-12 20:07:50 -0700154 if (f == ROUTE4_FAILURE) {
155 spin_unlock(&fastmap_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700156 goto failure;
John Fastabend1109c002014-09-12 20:07:50 -0700157 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700158
159 *res = f->res;
John Fastabend1109c002014-09-12 20:07:50 -0700160 spin_unlock(&fastmap_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700161 return 0;
162 }
John Fastabend1109c002014-09-12 20:07:50 -0700163 spin_unlock(&fastmap_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700164
165 h = route4_hash_to(id);
166
167restart:
John Fastabend1109c002014-09-12 20:07:50 -0700168 b = rcu_dereference_bh(head->table[h]);
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000169 if (b) {
John Fastabend1109c002014-09-12 20:07:50 -0700170 for (f = rcu_dereference_bh(b->ht[route4_hash_from(id)]);
171 f;
172 f = rcu_dereference_bh(f->next))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700173 if (f->id == id)
174 ROUTE4_APPLY_RESULT();
175
John Fastabend1109c002014-09-12 20:07:50 -0700176 for (f = rcu_dereference_bh(b->ht[route4_hash_iif(iif)]);
177 f;
178 f = rcu_dereference_bh(f->next))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700179 if (f->iif == iif)
180 ROUTE4_APPLY_RESULT();
181
John Fastabend1109c002014-09-12 20:07:50 -0700182 for (f = rcu_dereference_bh(b->ht[route4_hash_wild()]);
183 f;
184 f = rcu_dereference_bh(f->next))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700185 ROUTE4_APPLY_RESULT();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700186 }
187 if (h < 256) {
188 h = 256;
189 id &= ~0xFFFF;
190 goto restart;
191 }
192
193 if (!dont_cache)
194 route4_set_fastmap(head, id, iif, ROUTE4_FAILURE);
195failure:
196 return -1;
197
198old_method:
199 if (id && (TC_H_MAJ(id) == 0 ||
200 !(TC_H_MAJ(id^tp->q->handle)))) {
201 res->classid = id;
202 res->class = 0;
203 return 0;
204 }
205 return -1;
206}
207
208static inline u32 to_hash(u32 id)
209{
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000210 u32 h = id & 0xFF;
211
212 if (id & 0x8000)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700213 h += 256;
214 return h;
215}
216
217static inline u32 from_hash(u32 id)
218{
219 id &= 0xFFFF;
220 if (id == 0xFFFF)
221 return 32;
222 if (!(id & 0x8000)) {
223 if (id > 255)
224 return 256;
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000225 return id & 0xF;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700226 }
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000227 return 16 + (id & 0xF);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700228}
229
230static unsigned long route4_get(struct tcf_proto *tp, u32 handle)
231{
John Fastabend1109c002014-09-12 20:07:50 -0700232 struct route4_head *head = rtnl_dereference(tp->root);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700233 struct route4_bucket *b;
234 struct route4_filter *f;
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000235 unsigned int h1, h2;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700236
237 if (!head)
238 return 0;
239
240 h1 = to_hash(handle);
241 if (h1 > 256)
242 return 0;
243
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000244 h2 = from_hash(handle >> 16);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700245 if (h2 > 32)
246 return 0;
247
John Fastabend1109c002014-09-12 20:07:50 -0700248 b = rtnl_dereference(head->table[h1]);
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000249 if (b) {
John Fastabend1109c002014-09-12 20:07:50 -0700250 for (f = rtnl_dereference(b->ht[h2]);
251 f;
252 f = rtnl_dereference(f->next))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700253 if (f->handle == handle)
254 return (unsigned long)f;
255 }
256 return 0;
257}
258
Linus Torvalds1da177e2005-04-16 15:20:36 -0700259static int route4_init(struct tcf_proto *tp)
260{
WANG Conga05c2d12015-03-04 20:11:43 -0800261 struct route4_head *head;
262
263 head = kzalloc(sizeof(struct route4_head), GFP_KERNEL);
264 if (head == NULL)
265 return -ENOBUFS;
266
267 rcu_assign_pointer(tp->root, head);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700268 return 0;
269}
270
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -0400271static void route4_delete_filter(struct rcu_head *head)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700272{
John Fastabend1109c002014-09-12 20:07:50 -0700273 struct route4_filter *f = container_of(head, struct route4_filter, rcu);
John Fastabend1109c002014-09-12 20:07:50 -0700274
WANG Cong18d02642014-09-25 10:26:37 -0700275 tcf_exts_destroy(&f->exts);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700276 kfree(f);
277}
278
Cong Wang1e052be2015-03-06 11:47:59 -0800279static bool route4_destroy(struct tcf_proto *tp, bool force)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700280{
John Fastabend1109c002014-09-12 20:07:50 -0700281 struct route4_head *head = rtnl_dereference(tp->root);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700282 int h1, h2;
283
284 if (head == NULL)
Cong Wang1e052be2015-03-06 11:47:59 -0800285 return true;
286
287 if (!force) {
288 for (h1 = 0; h1 <= 256; h1++) {
289 if (rcu_access_pointer(head->table[h1]))
290 return false;
291 }
292 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700293
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000294 for (h1 = 0; h1 <= 256; h1++) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700295 struct route4_bucket *b;
296
John Fastabend1109c002014-09-12 20:07:50 -0700297 b = rtnl_dereference(head->table[h1]);
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000298 if (b) {
299 for (h2 = 0; h2 <= 32; h2++) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700300 struct route4_filter *f;
301
John Fastabend1109c002014-09-12 20:07:50 -0700302 while ((f = rtnl_dereference(b->ht[h2])) != NULL) {
303 struct route4_filter *next;
304
305 next = rtnl_dereference(f->next);
306 RCU_INIT_POINTER(b->ht[h2], next);
John Fastabend18cdb372014-10-05 21:28:52 -0700307 tcf_unbind_filter(tp, &f->res);
John Fastabend1109c002014-09-12 20:07:50 -0700308 call_rcu(&f->rcu, route4_delete_filter);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700309 }
310 }
John Fastabend1109c002014-09-12 20:07:50 -0700311 RCU_INIT_POINTER(head->table[h1], NULL);
312 kfree_rcu(b, rcu);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700313 }
314 }
John Fastabend1109c002014-09-12 20:07:50 -0700315 RCU_INIT_POINTER(tp->root, NULL);
316 kfree_rcu(head, rcu);
Cong Wang1e052be2015-03-06 11:47:59 -0800317 return true;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700318}
319
320static int route4_delete(struct tcf_proto *tp, unsigned long arg)
321{
John Fastabend1109c002014-09-12 20:07:50 -0700322 struct route4_head *head = rtnl_dereference(tp->root);
323 struct route4_filter *f = (struct route4_filter *)arg;
324 struct route4_filter __rcu **fp;
325 struct route4_filter *nf;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700326 struct route4_bucket *b;
John Fastabend1109c002014-09-12 20:07:50 -0700327 unsigned int h = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700328 int i;
329
330 if (!head || !f)
331 return -EINVAL;
332
333 h = f->handle;
334 b = f->bkt;
335
John Fastabend1109c002014-09-12 20:07:50 -0700336 fp = &b->ht[from_hash(h >> 16)];
337 for (nf = rtnl_dereference(*fp); nf;
338 fp = &nf->next, nf = rtnl_dereference(*fp)) {
339 if (nf == f) {
340 /* unlink it */
341 RCU_INIT_POINTER(*fp, rtnl_dereference(f->next));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700342
John Fastabend1109c002014-09-12 20:07:50 -0700343 /* Remove any fastmap lookups that might ref filter
344 * notice we unlink'd the filter so we can't get it
345 * back in the fastmap.
346 */
347 route4_reset_fastmap(head);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700348
John Fastabend1109c002014-09-12 20:07:50 -0700349 /* Delete it */
John Fastabend18cdb372014-10-05 21:28:52 -0700350 tcf_unbind_filter(tp, &f->res);
John Fastabend1109c002014-09-12 20:07:50 -0700351 call_rcu(&f->rcu, route4_delete_filter);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700352
John Fastabend1109c002014-09-12 20:07:50 -0700353 /* Strip RTNL protected tree */
354 for (i = 0; i <= 32; i++) {
355 struct route4_filter *rt;
356
357 rt = rtnl_dereference(b->ht[i]);
358 if (rt)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700359 return 0;
John Fastabend1109c002014-09-12 20:07:50 -0700360 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700361
362 /* OK, session has no flows */
John Fastabend1109c002014-09-12 20:07:50 -0700363 RCU_INIT_POINTER(head->table[to_hash(h)], NULL);
364 kfree_rcu(b, rcu);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700365
Linus Torvalds1da177e2005-04-16 15:20:36 -0700366 return 0;
367 }
368 }
369 return 0;
370}
371
Patrick McHardy6fa8c012008-01-23 20:36:12 -0800372static const struct nla_policy route4_policy[TCA_ROUTE4_MAX + 1] = {
373 [TCA_ROUTE4_CLASSID] = { .type = NLA_U32 },
374 [TCA_ROUTE4_TO] = { .type = NLA_U32 },
375 [TCA_ROUTE4_FROM] = { .type = NLA_U32 },
376 [TCA_ROUTE4_IIF] = { .type = NLA_U32 },
377};
378
Benjamin LaHaisec1b52732013-01-14 05:15:39 +0000379static int route4_set_parms(struct net *net, struct tcf_proto *tp,
380 unsigned long base, struct route4_filter *f,
381 u32 handle, struct route4_head *head,
Cong Wang2f7ef2f2014-04-25 13:54:06 -0700382 struct nlattr **tb, struct nlattr *est, int new,
383 bool ovr)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700384{
Linus Torvalds1da177e2005-04-16 15:20:36 -0700385 u32 id = 0, to = 0, nhandle = 0x8000;
386 struct route4_filter *fp;
387 unsigned int h1;
388 struct route4_bucket *b;
389 struct tcf_exts e;
WANG Congb9a24bb2016-08-19 12:36:54 -0700390 int err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700391
WANG Congb9a24bb2016-08-19 12:36:54 -0700392 err = tcf_exts_init(&e, TCA_ROUTE4_ACT, TCA_ROUTE4_POLICE);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700393 if (err < 0)
394 return err;
WANG Congb9a24bb2016-08-19 12:36:54 -0700395 err = tcf_exts_validate(net, tp, tb, est, &e, ovr);
396 if (err < 0)
397 goto errout;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700398
399 err = -EINVAL;
Patrick McHardyadd93b62008-01-22 22:11:33 -0800400 if (tb[TCA_ROUTE4_TO]) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700401 if (new && handle & 0x8000)
402 goto errout;
Patrick McHardy1587bac2008-01-23 20:35:03 -0800403 to = nla_get_u32(tb[TCA_ROUTE4_TO]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700404 if (to > 0xFF)
405 goto errout;
406 nhandle = to;
407 }
408
Patrick McHardyadd93b62008-01-22 22:11:33 -0800409 if (tb[TCA_ROUTE4_FROM]) {
410 if (tb[TCA_ROUTE4_IIF])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700411 goto errout;
Patrick McHardy1587bac2008-01-23 20:35:03 -0800412 id = nla_get_u32(tb[TCA_ROUTE4_FROM]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700413 if (id > 0xFF)
414 goto errout;
415 nhandle |= id << 16;
Patrick McHardyadd93b62008-01-22 22:11:33 -0800416 } else if (tb[TCA_ROUTE4_IIF]) {
Patrick McHardy1587bac2008-01-23 20:35:03 -0800417 id = nla_get_u32(tb[TCA_ROUTE4_IIF]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700418 if (id > 0x7FFF)
419 goto errout;
420 nhandle |= (id | 0x8000) << 16;
421 } else
422 nhandle |= 0xFFFF << 16;
423
424 if (handle && new) {
425 nhandle |= handle & 0x7F00;
426 if (nhandle != handle)
427 goto errout;
428 }
429
430 h1 = to_hash(nhandle);
John Fastabend1109c002014-09-12 20:07:50 -0700431 b = rtnl_dereference(head->table[h1]);
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000432 if (!b) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700433 err = -ENOBUFS;
Panagiotis Issaris0da974f2006-07-21 14:51:30 -0700434 b = kzalloc(sizeof(struct route4_bucket), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700435 if (b == NULL)
436 goto errout;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700437
John Fastabend1109c002014-09-12 20:07:50 -0700438 rcu_assign_pointer(head->table[h1], b);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700439 } else {
440 unsigned int h2 = from_hash(nhandle >> 16);
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000441
Linus Torvalds1da177e2005-04-16 15:20:36 -0700442 err = -EEXIST;
John Fastabend1109c002014-09-12 20:07:50 -0700443 for (fp = rtnl_dereference(b->ht[h2]);
444 fp;
445 fp = rtnl_dereference(fp->next))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700446 if (fp->handle == f->handle)
447 goto errout;
448 }
449
Patrick McHardyadd93b62008-01-22 22:11:33 -0800450 if (tb[TCA_ROUTE4_TO])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700451 f->id = to;
452
Patrick McHardyadd93b62008-01-22 22:11:33 -0800453 if (tb[TCA_ROUTE4_FROM])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700454 f->id = to | id<<16;
Patrick McHardyadd93b62008-01-22 22:11:33 -0800455 else if (tb[TCA_ROUTE4_IIF])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700456 f->iif = id;
457
458 f->handle = nhandle;
459 f->bkt = b;
John Fastabend1109c002014-09-12 20:07:50 -0700460 f->tp = tp;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700461
Patrick McHardyadd93b62008-01-22 22:11:33 -0800462 if (tb[TCA_ROUTE4_CLASSID]) {
Patrick McHardy1587bac2008-01-23 20:35:03 -0800463 f->res.classid = nla_get_u32(tb[TCA_ROUTE4_CLASSID]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700464 tcf_bind_filter(tp, &f->res, base);
465 }
466
467 tcf_exts_change(tp, &f->exts, &e);
468
469 return 0;
470errout:
WANG Cong18d02642014-09-25 10:26:37 -0700471 tcf_exts_destroy(&e);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700472 return err;
473}
474
Benjamin LaHaisec1b52732013-01-14 05:15:39 +0000475static int route4_change(struct net *net, struct sk_buff *in_skb,
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -0400476 struct tcf_proto *tp, unsigned long base, u32 handle,
477 struct nlattr **tca, unsigned long *arg, bool ovr)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700478{
John Fastabend1109c002014-09-12 20:07:50 -0700479 struct route4_head *head = rtnl_dereference(tp->root);
480 struct route4_filter __rcu **fp;
481 struct route4_filter *fold, *f1, *pfp, *f = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700482 struct route4_bucket *b;
Patrick McHardyadd93b62008-01-22 22:11:33 -0800483 struct nlattr *opt = tca[TCA_OPTIONS];
484 struct nlattr *tb[TCA_ROUTE4_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700485 unsigned int h, th;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700486 int err;
John Fastabend1109c002014-09-12 20:07:50 -0700487 bool new = true;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700488
489 if (opt == NULL)
490 return handle ? -EINVAL : 0;
491
Patrick McHardy6fa8c012008-01-23 20:36:12 -0800492 err = nla_parse_nested(tb, TCA_ROUTE4_MAX, opt, route4_policy);
Patrick McHardycee63722008-01-23 20:33:32 -0800493 if (err < 0)
494 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700495
John Fastabend1109c002014-09-12 20:07:50 -0700496 fold = (struct route4_filter *)*arg;
497 if (fold && handle && fold->handle != handle)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700498 return -EINVAL;
499
Linus Torvalds1da177e2005-04-16 15:20:36 -0700500 err = -ENOBUFS;
Panagiotis Issaris0da974f2006-07-21 14:51:30 -0700501 f = kzalloc(sizeof(struct route4_filter), GFP_KERNEL);
John Fastabend1109c002014-09-12 20:07:50 -0700502 if (!f)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700503 goto errout;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700504
WANG Congb9a24bb2016-08-19 12:36:54 -0700505 err = tcf_exts_init(&f->exts, TCA_ROUTE4_ACT, TCA_ROUTE4_POLICE);
506 if (err < 0)
507 goto errout;
508
John Fastabend1109c002014-09-12 20:07:50 -0700509 if (fold) {
510 f->id = fold->id;
511 f->iif = fold->iif;
512 f->res = fold->res;
513 f->handle = fold->handle;
514
515 f->tp = fold->tp;
516 f->bkt = fold->bkt;
517 new = false;
518 }
519
Benjamin LaHaisec1b52732013-01-14 05:15:39 +0000520 err = route4_set_parms(net, tp, base, f, handle, head, tb,
John Fastabend1109c002014-09-12 20:07:50 -0700521 tca[TCA_RATE], new, ovr);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700522 if (err < 0)
523 goto errout;
524
Linus Torvalds1da177e2005-04-16 15:20:36 -0700525 h = from_hash(f->handle >> 16);
John Fastabend1109c002014-09-12 20:07:50 -0700526 fp = &f->bkt->ht[h];
527 for (pfp = rtnl_dereference(*fp);
528 (f1 = rtnl_dereference(*fp)) != NULL;
529 fp = &f1->next)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700530 if (f->handle < f1->handle)
531 break;
532
Eric Dumazet02875872014-10-05 18:38:35 -0700533 netif_keep_dst(qdisc_dev(tp->q));
John Fastabend1109c002014-09-12 20:07:50 -0700534 rcu_assign_pointer(f->next, f1);
535 rcu_assign_pointer(*fp, f);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700536
John Fastabend1109c002014-09-12 20:07:50 -0700537 if (fold && fold->handle && f->handle != fold->handle) {
538 th = to_hash(fold->handle);
539 h = from_hash(fold->handle >> 16);
540 b = rtnl_dereference(head->table[th]);
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000541 if (b) {
John Fastabend1109c002014-09-12 20:07:50 -0700542 fp = &b->ht[h];
543 for (pfp = rtnl_dereference(*fp); pfp;
544 fp = &pfp->next, pfp = rtnl_dereference(*fp)) {
545 if (pfp == f) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700546 *fp = f->next;
547 break;
548 }
549 }
550 }
551 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700552
John Fastabend1109c002014-09-12 20:07:50 -0700553 route4_reset_fastmap(head);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700554 *arg = (unsigned long)f;
John Fastabend18cdb372014-10-05 21:28:52 -0700555 if (fold) {
556 tcf_unbind_filter(tp, &fold->res);
John Fastabend1109c002014-09-12 20:07:50 -0700557 call_rcu(&fold->rcu, route4_delete_filter);
John Fastabend18cdb372014-10-05 21:28:52 -0700558 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700559 return 0;
560
561errout:
WANG Cong21641c22016-09-18 15:52:20 -0700562 if (f)
563 tcf_exts_destroy(&f->exts);
Jesper Juhla51482b2005-11-08 09:41:34 -0800564 kfree(f);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700565 return err;
566}
567
568static void route4_walk(struct tcf_proto *tp, struct tcf_walker *arg)
569{
John Fastabend1109c002014-09-12 20:07:50 -0700570 struct route4_head *head = rtnl_dereference(tp->root);
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000571 unsigned int h, h1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700572
573 if (head == NULL)
574 arg->stop = 1;
575
576 if (arg->stop)
577 return;
578
579 for (h = 0; h <= 256; h++) {
John Fastabend1109c002014-09-12 20:07:50 -0700580 struct route4_bucket *b = rtnl_dereference(head->table[h]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700581
582 if (b) {
583 for (h1 = 0; h1 <= 32; h1++) {
584 struct route4_filter *f;
585
John Fastabend1109c002014-09-12 20:07:50 -0700586 for (f = rtnl_dereference(b->ht[h1]);
587 f;
588 f = rtnl_dereference(f->next)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700589 if (arg->count < arg->skip) {
590 arg->count++;
591 continue;
592 }
593 if (arg->fn(tp, (unsigned long)f, arg) < 0) {
594 arg->stop = 1;
595 return;
596 }
597 arg->count++;
598 }
599 }
600 }
601 }
602}
603
WANG Cong832d1d52014-01-09 16:14:01 -0800604static int route4_dump(struct net *net, struct tcf_proto *tp, unsigned long fh,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700605 struct sk_buff *skb, struct tcmsg *t)
606{
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000607 struct route4_filter *f = (struct route4_filter *)fh;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800608 struct nlattr *nest;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700609 u32 id;
610
611 if (f == NULL)
612 return skb->len;
613
614 t->tcm_handle = f->handle;
615
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800616 nest = nla_nest_start(skb, TCA_OPTIONS);
617 if (nest == NULL)
618 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700619
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000620 if (!(f->handle & 0x8000)) {
621 id = f->id & 0xFF;
David S. Miller1b34ec42012-03-29 05:11:39 -0400622 if (nla_put_u32(skb, TCA_ROUTE4_TO, id))
623 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700624 }
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000625 if (f->handle & 0x80000000) {
David S. Miller1b34ec42012-03-29 05:11:39 -0400626 if ((f->handle >> 16) != 0xFFFF &&
627 nla_put_u32(skb, TCA_ROUTE4_IIF, f->iif))
628 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700629 } else {
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000630 id = f->id >> 16;
David S. Miller1b34ec42012-03-29 05:11:39 -0400631 if (nla_put_u32(skb, TCA_ROUTE4_FROM, id))
632 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700633 }
David S. Miller1b34ec42012-03-29 05:11:39 -0400634 if (f->res.classid &&
635 nla_put_u32(skb, TCA_ROUTE4_CLASSID, f->res.classid))
636 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700637
WANG Cong5da57f42013-12-15 20:15:07 -0800638 if (tcf_exts_dump(skb, &f->exts) < 0)
Patrick McHardyadd93b62008-01-22 22:11:33 -0800639 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700640
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800641 nla_nest_end(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700642
WANG Cong5da57f42013-12-15 20:15:07 -0800643 if (tcf_exts_dump_stats(skb, &f->exts) < 0)
Patrick McHardyadd93b62008-01-22 22:11:33 -0800644 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700645
646 return skb->len;
647
Patrick McHardyadd93b62008-01-22 22:11:33 -0800648nla_put_failure:
Jiri Pirko6ea3b442014-12-09 22:23:29 +0100649 nla_nest_cancel(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700650 return -1;
651}
652
Patrick McHardy2eb9d752008-01-22 22:10:42 -0800653static struct tcf_proto_ops cls_route4_ops __read_mostly = {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700654 .kind = "route",
655 .classify = route4_classify,
656 .init = route4_init,
657 .destroy = route4_destroy,
658 .get = route4_get,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700659 .change = route4_change,
660 .delete = route4_delete,
661 .walk = route4_walk,
662 .dump = route4_dump,
663 .owner = THIS_MODULE,
664};
665
666static int __init init_route4(void)
667{
668 return register_tcf_proto_ops(&cls_route4_ops);
669}
670
671static void __exit exit_route4(void)
672{
673 unregister_tcf_proto_ops(&cls_route4_ops);
674}
675
676module_init(init_route4)
677module_exit(exit_route4)
678MODULE_LICENSE("GPL");