Commit | Line | Data |
---|---|---|
14c0b97d TG |
1 | /* |
2 | * net/core/fib_rules.c Generic Routing Rules | |
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 as | |
6 | * published by the Free Software Foundation, version 2. | |
7 | * | |
8 | * Authors: Thomas Graf <tgraf@suug.ch> | |
9 | */ | |
10 | ||
14c0b97d TG |
11 | #include <linux/types.h> |
12 | #include <linux/kernel.h> | |
5a0e3ad6 | 13 | #include <linux/slab.h> |
14c0b97d | 14 | #include <linux/list.h> |
3a9a231d | 15 | #include <linux/module.h> |
e9dc8653 | 16 | #include <net/net_namespace.h> |
881d966b | 17 | #include <net/sock.h> |
14c0b97d TG |
18 | #include <net/fib_rules.h> |
19 | ||
2994c638 DL |
20 | int fib_default_rule_add(struct fib_rules_ops *ops, |
21 | u32 pref, u32 table, u32 flags) | |
22 | { | |
23 | struct fib_rule *r; | |
24 | ||
25 | r = kzalloc(ops->rule_size, GFP_KERNEL); | |
26 | if (r == NULL) | |
27 | return -ENOMEM; | |
28 | ||
29 | atomic_set(&r->refcnt, 1); | |
30 | r->action = FR_ACT_TO_TBL; | |
31 | r->pref = pref; | |
32 | r->table = table; | |
33 | r->flags = flags; | |
3661a910 | 34 | r->fr_net = hold_net(ops->fro_net); |
2994c638 | 35 | |
73f5698e ST |
36 | r->suppress_prefixlen = -1; |
37 | r->suppress_ifgroup = -1; | |
38 | ||
2994c638 DL |
39 | /* The lock is not required here, the list in unreacheable |
40 | * at the moment this function is called */ | |
41 | list_add_tail(&r->list, &ops->rules_list); | |
42 | return 0; | |
43 | } | |
44 | EXPORT_SYMBOL(fib_default_rule_add); | |
45 | ||
d8a566be PM |
46 | u32 fib_default_rule_pref(struct fib_rules_ops *ops) |
47 | { | |
48 | struct list_head *pos; | |
49 | struct fib_rule *rule; | |
50 | ||
51 | if (!list_empty(&ops->rules_list)) { | |
52 | pos = ops->rules_list.next; | |
53 | if (pos->next != &ops->rules_list) { | |
54 | rule = list_entry(pos->next, struct fib_rule, list); | |
55 | if (rule->pref) | |
56 | return rule->pref - 1; | |
57 | } | |
58 | } | |
59 | ||
60 | return 0; | |
61 | } | |
62 | EXPORT_SYMBOL(fib_default_rule_pref); | |
63 | ||
9e3a5487 | 64 | static void notify_rule_change(int event, struct fib_rule *rule, |
c17084d2 TG |
65 | struct fib_rules_ops *ops, struct nlmsghdr *nlh, |
66 | u32 pid); | |
14c0b97d | 67 | |
5fd30ee7 | 68 | static struct fib_rules_ops *lookup_rules_ops(struct net *net, int family) |
14c0b97d TG |
69 | { |
70 | struct fib_rules_ops *ops; | |
71 | ||
72 | rcu_read_lock(); | |
5fd30ee7 | 73 | list_for_each_entry_rcu(ops, &net->rules_ops, list) { |
14c0b97d TG |
74 | if (ops->family == family) { |
75 | if (!try_module_get(ops->owner)) | |
76 | ops = NULL; | |
77 | rcu_read_unlock(); | |
78 | return ops; | |
79 | } | |
80 | } | |
81 | rcu_read_unlock(); | |
82 | ||
83 | return NULL; | |
84 | } | |
85 | ||
86 | static void rules_ops_put(struct fib_rules_ops *ops) | |
87 | { | |
88 | if (ops) | |
89 | module_put(ops->owner); | |
90 | } | |
91 | ||
73417f61 TG |
92 | static void flush_route_cache(struct fib_rules_ops *ops) |
93 | { | |
94 | if (ops->flush_cache) | |
ae299fc0 | 95 | ops->flush_cache(ops); |
73417f61 TG |
96 | } |
97 | ||
e9c5158a | 98 | static int __fib_rules_register(struct fib_rules_ops *ops) |
14c0b97d TG |
99 | { |
100 | int err = -EEXIST; | |
101 | struct fib_rules_ops *o; | |
9e3a5487 DL |
102 | struct net *net; |
103 | ||
104 | net = ops->fro_net; | |
14c0b97d TG |
105 | |
106 | if (ops->rule_size < sizeof(struct fib_rule)) | |
107 | return -EINVAL; | |
108 | ||
109 | if (ops->match == NULL || ops->configure == NULL || | |
110 | ops->compare == NULL || ops->fill == NULL || | |
111 | ops->action == NULL) | |
112 | return -EINVAL; | |
113 | ||
5fd30ee7 DL |
114 | spin_lock(&net->rules_mod_lock); |
115 | list_for_each_entry(o, &net->rules_ops, list) | |
14c0b97d TG |
116 | if (ops->family == o->family) |
117 | goto errout; | |
118 | ||
5fd30ee7 DL |
119 | hold_net(net); |
120 | list_add_tail_rcu(&ops->list, &net->rules_ops); | |
14c0b97d TG |
121 | err = 0; |
122 | errout: | |
5fd30ee7 | 123 | spin_unlock(&net->rules_mod_lock); |
14c0b97d TG |
124 | |
125 | return err; | |
126 | } | |
127 | ||
e9c5158a | 128 | struct fib_rules_ops * |
3d0c9c4e | 129 | fib_rules_register(const struct fib_rules_ops *tmpl, struct net *net) |
e9c5158a EB |
130 | { |
131 | struct fib_rules_ops *ops; | |
132 | int err; | |
133 | ||
2fb3573d | 134 | ops = kmemdup(tmpl, sizeof(*ops), GFP_KERNEL); |
e9c5158a EB |
135 | if (ops == NULL) |
136 | return ERR_PTR(-ENOMEM); | |
137 | ||
138 | INIT_LIST_HEAD(&ops->rules_list); | |
139 | ops->fro_net = net; | |
140 | ||
141 | err = __fib_rules_register(ops); | |
142 | if (err) { | |
143 | kfree(ops); | |
144 | ops = ERR_PTR(err); | |
145 | } | |
146 | ||
147 | return ops; | |
148 | } | |
14c0b97d TG |
149 | EXPORT_SYMBOL_GPL(fib_rules_register); |
150 | ||
1df9916e | 151 | static void fib_rules_cleanup_ops(struct fib_rules_ops *ops) |
14c0b97d TG |
152 | { |
153 | struct fib_rule *rule, *tmp; | |
154 | ||
76c72d4f | 155 | list_for_each_entry_safe(rule, tmp, &ops->rules_list, list) { |
14c0b97d | 156 | list_del_rcu(&rule->list); |
7a9bc9b8 DM |
157 | if (ops->delete) |
158 | ops->delete(rule); | |
14c0b97d TG |
159 | fib_rule_put(rule); |
160 | } | |
161 | } | |
162 | ||
e9c5158a EB |
163 | static void fib_rules_put_rcu(struct rcu_head *head) |
164 | { | |
165 | struct fib_rules_ops *ops = container_of(head, struct fib_rules_ops, rcu); | |
166 | struct net *net = ops->fro_net; | |
167 | ||
168 | release_net(net); | |
169 | kfree(ops); | |
170 | } | |
171 | ||
9e3a5487 | 172 | void fib_rules_unregister(struct fib_rules_ops *ops) |
14c0b97d | 173 | { |
9e3a5487 | 174 | struct net *net = ops->fro_net; |
14c0b97d | 175 | |
5fd30ee7 | 176 | spin_lock(&net->rules_mod_lock); |
72132c1b DL |
177 | list_del_rcu(&ops->list); |
178 | fib_rules_cleanup_ops(ops); | |
5fd30ee7 | 179 | spin_unlock(&net->rules_mod_lock); |
14c0b97d | 180 | |
e9c5158a | 181 | call_rcu(&ops->rcu, fib_rules_put_rcu); |
14c0b97d | 182 | } |
14c0b97d TG |
183 | EXPORT_SYMBOL_GPL(fib_rules_unregister); |
184 | ||
3dfbcc41 TG |
185 | static int fib_rule_match(struct fib_rule *rule, struct fib_rules_ops *ops, |
186 | struct flowi *fl, int flags) | |
187 | { | |
188 | int ret = 0; | |
189 | ||
1d28f42c | 190 | if (rule->iifindex && (rule->iifindex != fl->flowi_iif)) |
3dfbcc41 TG |
191 | goto out; |
192 | ||
1d28f42c | 193 | if (rule->oifindex && (rule->oifindex != fl->flowi_oif)) |
1b038a5e PM |
194 | goto out; |
195 | ||
1d28f42c | 196 | if ((rule->mark ^ fl->flowi_mark) & rule->mark_mask) |
3dfbcc41 TG |
197 | goto out; |
198 | ||
199 | ret = ops->match(rule, fl, flags); | |
200 | out: | |
201 | return (rule->flags & FIB_RULE_INVERT) ? !ret : ret; | |
202 | } | |
203 | ||
14c0b97d TG |
204 | int fib_rules_lookup(struct fib_rules_ops *ops, struct flowi *fl, |
205 | int flags, struct fib_lookup_arg *arg) | |
206 | { | |
207 | struct fib_rule *rule; | |
208 | int err; | |
209 | ||
210 | rcu_read_lock(); | |
211 | ||
76c72d4f | 212 | list_for_each_entry_rcu(rule, &ops->rules_list, list) { |
0947c9fe | 213 | jumped: |
3dfbcc41 | 214 | if (!fib_rule_match(rule, ops, fl, flags)) |
14c0b97d TG |
215 | continue; |
216 | ||
0947c9fe TG |
217 | if (rule->action == FR_ACT_GOTO) { |
218 | struct fib_rule *target; | |
219 | ||
220 | target = rcu_dereference(rule->ctarget); | |
221 | if (target == NULL) { | |
222 | continue; | |
223 | } else { | |
224 | rule = target; | |
225 | goto jumped; | |
226 | } | |
fa0b2d1d TG |
227 | } else if (rule->action == FR_ACT_NOP) |
228 | continue; | |
229 | else | |
0947c9fe TG |
230 | err = ops->action(rule, fl, flags, arg); |
231 | ||
7764a45a ST |
232 | if (!err && ops->suppress && ops->suppress(rule, arg)) |
233 | continue; | |
234 | ||
14c0b97d | 235 | if (err != -EAGAIN) { |
ebc0ffae ED |
236 | if ((arg->flags & FIB_LOOKUP_NOREF) || |
237 | likely(atomic_inc_not_zero(&rule->refcnt))) { | |
7fa7cb71 ED |
238 | arg->rule = rule; |
239 | goto out; | |
240 | } | |
241 | break; | |
14c0b97d TG |
242 | } |
243 | } | |
244 | ||
83886b6b | 245 | err = -ESRCH; |
14c0b97d TG |
246 | out: |
247 | rcu_read_unlock(); | |
248 | ||
249 | return err; | |
250 | } | |
14c0b97d TG |
251 | EXPORT_SYMBOL_GPL(fib_rules_lookup); |
252 | ||
e1701c68 TG |
253 | static int validate_rulemsg(struct fib_rule_hdr *frh, struct nlattr **tb, |
254 | struct fib_rules_ops *ops) | |
255 | { | |
256 | int err = -EINVAL; | |
257 | ||
258 | if (frh->src_len) | |
259 | if (tb[FRA_SRC] == NULL || | |
260 | frh->src_len > (ops->addr_size * 8) || | |
261 | nla_len(tb[FRA_SRC]) != ops->addr_size) | |
262 | goto errout; | |
263 | ||
264 | if (frh->dst_len) | |
265 | if (tb[FRA_DST] == NULL || | |
266 | frh->dst_len > (ops->addr_size * 8) || | |
267 | nla_len(tb[FRA_DST]) != ops->addr_size) | |
268 | goto errout; | |
269 | ||
270 | err = 0; | |
271 | errout: | |
272 | return err; | |
273 | } | |
274 | ||
661d2967 | 275 | static int fib_nl_newrule(struct sk_buff *skb, struct nlmsghdr* nlh) |
14c0b97d | 276 | { |
3b1e0a65 | 277 | struct net *net = sock_net(skb->sk); |
14c0b97d TG |
278 | struct fib_rule_hdr *frh = nlmsg_data(nlh); |
279 | struct fib_rules_ops *ops = NULL; | |
280 | struct fib_rule *rule, *r, *last = NULL; | |
281 | struct nlattr *tb[FRA_MAX+1]; | |
0947c9fe | 282 | int err = -EINVAL, unresolved = 0; |
14c0b97d TG |
283 | |
284 | if (nlh->nlmsg_len < nlmsg_msg_size(sizeof(*frh))) | |
285 | goto errout; | |
286 | ||
5fd30ee7 | 287 | ops = lookup_rules_ops(net, frh->family); |
14c0b97d | 288 | if (ops == NULL) { |
2fe195cf | 289 | err = -EAFNOSUPPORT; |
14c0b97d TG |
290 | goto errout; |
291 | } | |
292 | ||
293 | err = nlmsg_parse(nlh, sizeof(*frh), tb, FRA_MAX, ops->policy); | |
294 | if (err < 0) | |
295 | goto errout; | |
296 | ||
e1701c68 TG |
297 | err = validate_rulemsg(frh, tb, ops); |
298 | if (err < 0) | |
299 | goto errout; | |
300 | ||
14c0b97d TG |
301 | rule = kzalloc(ops->rule_size, GFP_KERNEL); |
302 | if (rule == NULL) { | |
303 | err = -ENOMEM; | |
304 | goto errout; | |
305 | } | |
3661a910 | 306 | rule->fr_net = hold_net(net); |
14c0b97d TG |
307 | |
308 | if (tb[FRA_PRIORITY]) | |
309 | rule->pref = nla_get_u32(tb[FRA_PRIORITY]); | |
310 | ||
491deb24 | 311 | if (tb[FRA_IIFNAME]) { |
14c0b97d TG |
312 | struct net_device *dev; |
313 | ||
491deb24 PM |
314 | rule->iifindex = -1; |
315 | nla_strlcpy(rule->iifname, tb[FRA_IIFNAME], IFNAMSIZ); | |
316 | dev = __dev_get_by_name(net, rule->iifname); | |
14c0b97d | 317 | if (dev) |
491deb24 | 318 | rule->iifindex = dev->ifindex; |
14c0b97d TG |
319 | } |
320 | ||
1b038a5e PM |
321 | if (tb[FRA_OIFNAME]) { |
322 | struct net_device *dev; | |
323 | ||
324 | rule->oifindex = -1; | |
325 | nla_strlcpy(rule->oifname, tb[FRA_OIFNAME], IFNAMSIZ); | |
326 | dev = __dev_get_by_name(net, rule->oifname); | |
327 | if (dev) | |
328 | rule->oifindex = dev->ifindex; | |
329 | } | |
330 | ||
b8964ed9 TG |
331 | if (tb[FRA_FWMARK]) { |
332 | rule->mark = nla_get_u32(tb[FRA_FWMARK]); | |
333 | if (rule->mark) | |
334 | /* compatibility: if the mark value is non-zero all bits | |
335 | * are compared unless a mask is explicitly specified. | |
336 | */ | |
337 | rule->mark_mask = 0xFFFFFFFF; | |
338 | } | |
339 | ||
340 | if (tb[FRA_FWMASK]) | |
341 | rule->mark_mask = nla_get_u32(tb[FRA_FWMASK]); | |
342 | ||
14c0b97d TG |
343 | rule->action = frh->action; |
344 | rule->flags = frh->flags; | |
9e762a4a | 345 | rule->table = frh_get_table(frh, tb); |
73f5698e ST |
346 | if (tb[FRA_SUPPRESS_PREFIXLEN]) |
347 | rule->suppress_prefixlen = nla_get_u32(tb[FRA_SUPPRESS_PREFIXLEN]); | |
348 | else | |
349 | rule->suppress_prefixlen = -1; | |
14c0b97d | 350 | |
6ef94cfa ST |
351 | if (tb[FRA_SUPPRESS_IFGROUP]) |
352 | rule->suppress_ifgroup = nla_get_u32(tb[FRA_SUPPRESS_IFGROUP]); | |
73f5698e ST |
353 | else |
354 | rule->suppress_ifgroup = -1; | |
6ef94cfa | 355 | |
5adef180 | 356 | if (!tb[FRA_PRIORITY] && ops->default_pref) |
868d13ac | 357 | rule->pref = ops->default_pref(ops); |
14c0b97d | 358 | |
0947c9fe TG |
359 | err = -EINVAL; |
360 | if (tb[FRA_GOTO]) { | |
361 | if (rule->action != FR_ACT_GOTO) | |
362 | goto errout_free; | |
363 | ||
364 | rule->target = nla_get_u32(tb[FRA_GOTO]); | |
365 | /* Backward jumps are prohibited to avoid endless loops */ | |
366 | if (rule->target <= rule->pref) | |
367 | goto errout_free; | |
368 | ||
76c72d4f | 369 | list_for_each_entry(r, &ops->rules_list, list) { |
0947c9fe | 370 | if (r->pref == rule->target) { |
7a2b03c5 | 371 | RCU_INIT_POINTER(rule->ctarget, r); |
0947c9fe TG |
372 | break; |
373 | } | |
374 | } | |
375 | ||
7a2b03c5 | 376 | if (rcu_dereference_protected(rule->ctarget, 1) == NULL) |
0947c9fe TG |
377 | unresolved = 1; |
378 | } else if (rule->action == FR_ACT_GOTO) | |
379 | goto errout_free; | |
380 | ||
8b3521ee | 381 | err = ops->configure(rule, skb, frh, tb); |
14c0b97d TG |
382 | if (err < 0) |
383 | goto errout_free; | |
384 | ||
76c72d4f | 385 | list_for_each_entry(r, &ops->rules_list, list) { |
14c0b97d TG |
386 | if (r->pref > rule->pref) |
387 | break; | |
388 | last = r; | |
389 | } | |
390 | ||
391 | fib_rule_get(rule); | |
392 | ||
ebb9fed2 ED |
393 | if (last) |
394 | list_add_rcu(&rule->list, &last->list); | |
395 | else | |
396 | list_add_rcu(&rule->list, &ops->rules_list); | |
397 | ||
0947c9fe TG |
398 | if (ops->unresolved_rules) { |
399 | /* | |
400 | * There are unresolved goto rules in the list, check if | |
401 | * any of them are pointing to this new rule. | |
402 | */ | |
76c72d4f | 403 | list_for_each_entry(r, &ops->rules_list, list) { |
0947c9fe | 404 | if (r->action == FR_ACT_GOTO && |
561dac2d G |
405 | r->target == rule->pref && |
406 | rtnl_dereference(r->ctarget) == NULL) { | |
0947c9fe TG |
407 | rcu_assign_pointer(r->ctarget, rule); |
408 | if (--ops->unresolved_rules == 0) | |
409 | break; | |
410 | } | |
411 | } | |
412 | } | |
413 | ||
414 | if (rule->action == FR_ACT_GOTO) | |
415 | ops->nr_goto_rules++; | |
416 | ||
417 | if (unresolved) | |
418 | ops->unresolved_rules++; | |
419 | ||
15e47304 | 420 | notify_rule_change(RTM_NEWRULE, rule, ops, nlh, NETLINK_CB(skb).portid); |
73417f61 | 421 | flush_route_cache(ops); |
14c0b97d TG |
422 | rules_ops_put(ops); |
423 | return 0; | |
424 | ||
425 | errout_free: | |
3661a910 | 426 | release_net(rule->fr_net); |
14c0b97d TG |
427 | kfree(rule); |
428 | errout: | |
429 | rules_ops_put(ops); | |
430 | return err; | |
431 | } | |
432 | ||
661d2967 | 433 | static int fib_nl_delrule(struct sk_buff *skb, struct nlmsghdr* nlh) |
14c0b97d | 434 | { |
3b1e0a65 | 435 | struct net *net = sock_net(skb->sk); |
14c0b97d TG |
436 | struct fib_rule_hdr *frh = nlmsg_data(nlh); |
437 | struct fib_rules_ops *ops = NULL; | |
0947c9fe | 438 | struct fib_rule *rule, *tmp; |
14c0b97d TG |
439 | struct nlattr *tb[FRA_MAX+1]; |
440 | int err = -EINVAL; | |
441 | ||
442 | if (nlh->nlmsg_len < nlmsg_msg_size(sizeof(*frh))) | |
443 | goto errout; | |
444 | ||
5fd30ee7 | 445 | ops = lookup_rules_ops(net, frh->family); |
14c0b97d | 446 | if (ops == NULL) { |
2fe195cf | 447 | err = -EAFNOSUPPORT; |
14c0b97d TG |
448 | goto errout; |
449 | } | |
450 | ||
451 | err = nlmsg_parse(nlh, sizeof(*frh), tb, FRA_MAX, ops->policy); | |
452 | if (err < 0) | |
453 | goto errout; | |
454 | ||
e1701c68 TG |
455 | err = validate_rulemsg(frh, tb, ops); |
456 | if (err < 0) | |
457 | goto errout; | |
458 | ||
76c72d4f | 459 | list_for_each_entry(rule, &ops->rules_list, list) { |
14c0b97d TG |
460 | if (frh->action && (frh->action != rule->action)) |
461 | continue; | |
462 | ||
13eb2ab2 AH |
463 | if (frh_get_table(frh, tb) && |
464 | (frh_get_table(frh, tb) != rule->table)) | |
14c0b97d TG |
465 | continue; |
466 | ||
467 | if (tb[FRA_PRIORITY] && | |
468 | (rule->pref != nla_get_u32(tb[FRA_PRIORITY]))) | |
469 | continue; | |
470 | ||
491deb24 PM |
471 | if (tb[FRA_IIFNAME] && |
472 | nla_strcmp(tb[FRA_IIFNAME], rule->iifname)) | |
14c0b97d TG |
473 | continue; |
474 | ||
1b038a5e PM |
475 | if (tb[FRA_OIFNAME] && |
476 | nla_strcmp(tb[FRA_OIFNAME], rule->oifname)) | |
477 | continue; | |
478 | ||
b8964ed9 TG |
479 | if (tb[FRA_FWMARK] && |
480 | (rule->mark != nla_get_u32(tb[FRA_FWMARK]))) | |
481 | continue; | |
482 | ||
483 | if (tb[FRA_FWMASK] && | |
484 | (rule->mark_mask != nla_get_u32(tb[FRA_FWMASK]))) | |
485 | continue; | |
486 | ||
14c0b97d TG |
487 | if (!ops->compare(rule, frh, tb)) |
488 | continue; | |
489 | ||
490 | if (rule->flags & FIB_RULE_PERMANENT) { | |
491 | err = -EPERM; | |
492 | goto errout; | |
493 | } | |
494 | ||
495 | list_del_rcu(&rule->list); | |
0947c9fe | 496 | |
afaef734 | 497 | if (rule->action == FR_ACT_GOTO) { |
0947c9fe | 498 | ops->nr_goto_rules--; |
afaef734 YZ |
499 | if (rtnl_dereference(rule->ctarget) == NULL) |
500 | ops->unresolved_rules--; | |
501 | } | |
0947c9fe TG |
502 | |
503 | /* | |
504 | * Check if this rule is a target to any of them. If so, | |
505 | * disable them. As this operation is eventually very | |
506 | * expensive, it is only performed if goto rules have | |
507 | * actually been added. | |
508 | */ | |
509 | if (ops->nr_goto_rules > 0) { | |
76c72d4f | 510 | list_for_each_entry(tmp, &ops->rules_list, list) { |
7a2b03c5 | 511 | if (rtnl_dereference(tmp->ctarget) == rule) { |
a9b3cd7f | 512 | RCU_INIT_POINTER(tmp->ctarget, NULL); |
0947c9fe TG |
513 | ops->unresolved_rules++; |
514 | } | |
515 | } | |
516 | } | |
517 | ||
9e3a5487 | 518 | notify_rule_change(RTM_DELRULE, rule, ops, nlh, |
15e47304 | 519 | NETLINK_CB(skb).portid); |
7a9bc9b8 DM |
520 | if (ops->delete) |
521 | ops->delete(rule); | |
14c0b97d | 522 | fib_rule_put(rule); |
73417f61 | 523 | flush_route_cache(ops); |
14c0b97d TG |
524 | rules_ops_put(ops); |
525 | return 0; | |
526 | } | |
527 | ||
528 | err = -ENOENT; | |
529 | errout: | |
530 | rules_ops_put(ops); | |
531 | return err; | |
532 | } | |
533 | ||
339bf98f TG |
534 | static inline size_t fib_rule_nlmsg_size(struct fib_rules_ops *ops, |
535 | struct fib_rule *rule) | |
536 | { | |
537 | size_t payload = NLMSG_ALIGN(sizeof(struct fib_rule_hdr)) | |
491deb24 | 538 | + nla_total_size(IFNAMSIZ) /* FRA_IIFNAME */ |
1b038a5e | 539 | + nla_total_size(IFNAMSIZ) /* FRA_OIFNAME */ |
339bf98f TG |
540 | + nla_total_size(4) /* FRA_PRIORITY */ |
541 | + nla_total_size(4) /* FRA_TABLE */ | |
73f5698e | 542 | + nla_total_size(4) /* FRA_SUPPRESS_PREFIXLEN */ |
6ef94cfa | 543 | + nla_total_size(4) /* FRA_SUPPRESS_IFGROUP */ |
339bf98f TG |
544 | + nla_total_size(4) /* FRA_FWMARK */ |
545 | + nla_total_size(4); /* FRA_FWMASK */ | |
546 | ||
547 | if (ops->nlmsg_payload) | |
548 | payload += ops->nlmsg_payload(rule); | |
549 | ||
550 | return payload; | |
551 | } | |
552 | ||
14c0b97d TG |
553 | static int fib_nl_fill_rule(struct sk_buff *skb, struct fib_rule *rule, |
554 | u32 pid, u32 seq, int type, int flags, | |
555 | struct fib_rules_ops *ops) | |
556 | { | |
557 | struct nlmsghdr *nlh; | |
558 | struct fib_rule_hdr *frh; | |
559 | ||
560 | nlh = nlmsg_put(skb, pid, seq, type, sizeof(*frh), flags); | |
561 | if (nlh == NULL) | |
26932566 | 562 | return -EMSGSIZE; |
14c0b97d TG |
563 | |
564 | frh = nlmsg_data(nlh); | |
28bb1726 | 565 | frh->family = ops->family; |
14c0b97d | 566 | frh->table = rule->table; |
0e3cea7b DM |
567 | if (nla_put_u32(skb, FRA_TABLE, rule->table)) |
568 | goto nla_put_failure; | |
73f5698e | 569 | if (nla_put_u32(skb, FRA_SUPPRESS_PREFIXLEN, rule->suppress_prefixlen)) |
7764a45a | 570 | goto nla_put_failure; |
14c0b97d TG |
571 | frh->res1 = 0; |
572 | frh->res2 = 0; | |
573 | frh->action = rule->action; | |
574 | frh->flags = rule->flags; | |
575 | ||
7a2b03c5 | 576 | if (rule->action == FR_ACT_GOTO && |
33d480ce | 577 | rcu_access_pointer(rule->ctarget) == NULL) |
0947c9fe TG |
578 | frh->flags |= FIB_RULE_UNRESOLVED; |
579 | ||
491deb24 | 580 | if (rule->iifname[0]) { |
0e3cea7b DM |
581 | if (nla_put_string(skb, FRA_IIFNAME, rule->iifname)) |
582 | goto nla_put_failure; | |
491deb24 PM |
583 | if (rule->iifindex == -1) |
584 | frh->flags |= FIB_RULE_IIF_DETACHED; | |
2b443683 TG |
585 | } |
586 | ||
1b038a5e | 587 | if (rule->oifname[0]) { |
0e3cea7b DM |
588 | if (nla_put_string(skb, FRA_OIFNAME, rule->oifname)) |
589 | goto nla_put_failure; | |
1b038a5e PM |
590 | if (rule->oifindex == -1) |
591 | frh->flags |= FIB_RULE_OIF_DETACHED; | |
592 | } | |
593 | ||
0e3cea7b DM |
594 | if ((rule->pref && |
595 | nla_put_u32(skb, FRA_PRIORITY, rule->pref)) || | |
596 | (rule->mark && | |
597 | nla_put_u32(skb, FRA_FWMARK, rule->mark)) || | |
598 | ((rule->mark_mask || rule->mark) && | |
599 | nla_put_u32(skb, FRA_FWMASK, rule->mark_mask)) || | |
600 | (rule->target && | |
601 | nla_put_u32(skb, FRA_GOTO, rule->target))) | |
602 | goto nla_put_failure; | |
6ef94cfa ST |
603 | |
604 | if (rule->suppress_ifgroup != -1) { | |
605 | if (nla_put_u32(skb, FRA_SUPPRESS_IFGROUP, rule->suppress_ifgroup)) | |
606 | goto nla_put_failure; | |
607 | } | |
608 | ||
04af8cf6 | 609 | if (ops->fill(rule, skb, frh) < 0) |
14c0b97d TG |
610 | goto nla_put_failure; |
611 | ||
612 | return nlmsg_end(skb, nlh); | |
613 | ||
614 | nla_put_failure: | |
26932566 PM |
615 | nlmsg_cancel(skb, nlh); |
616 | return -EMSGSIZE; | |
14c0b97d TG |
617 | } |
618 | ||
c454673d TG |
619 | static int dump_rules(struct sk_buff *skb, struct netlink_callback *cb, |
620 | struct fib_rules_ops *ops) | |
14c0b97d TG |
621 | { |
622 | int idx = 0; | |
623 | struct fib_rule *rule; | |
14c0b97d | 624 | |
e67f88dd ED |
625 | rcu_read_lock(); |
626 | list_for_each_entry_rcu(rule, &ops->rules_list, list) { | |
c454673d | 627 | if (idx < cb->args[1]) |
14c0b97d TG |
628 | goto skip; |
629 | ||
15e47304 | 630 | if (fib_nl_fill_rule(skb, rule, NETLINK_CB(cb->skb).portid, |
14c0b97d TG |
631 | cb->nlh->nlmsg_seq, RTM_NEWRULE, |
632 | NLM_F_MULTI, ops) < 0) | |
633 | break; | |
634 | skip: | |
635 | idx++; | |
636 | } | |
2907c35f | 637 | rcu_read_unlock(); |
c454673d | 638 | cb->args[1] = idx; |
14c0b97d TG |
639 | rules_ops_put(ops); |
640 | ||
641 | return skb->len; | |
642 | } | |
643 | ||
c454673d TG |
644 | static int fib_nl_dumprule(struct sk_buff *skb, struct netlink_callback *cb) |
645 | { | |
3b1e0a65 | 646 | struct net *net = sock_net(skb->sk); |
c454673d TG |
647 | struct fib_rules_ops *ops; |
648 | int idx = 0, family; | |
649 | ||
650 | family = rtnl_msg_family(cb->nlh); | |
651 | if (family != AF_UNSPEC) { | |
652 | /* Protocol specific dump request */ | |
5fd30ee7 | 653 | ops = lookup_rules_ops(net, family); |
c454673d TG |
654 | if (ops == NULL) |
655 | return -EAFNOSUPPORT; | |
656 | ||
657 | return dump_rules(skb, cb, ops); | |
658 | } | |
659 | ||
660 | rcu_read_lock(); | |
5fd30ee7 | 661 | list_for_each_entry_rcu(ops, &net->rules_ops, list) { |
c454673d TG |
662 | if (idx < cb->args[0] || !try_module_get(ops->owner)) |
663 | goto skip; | |
664 | ||
665 | if (dump_rules(skb, cb, ops) < 0) | |
666 | break; | |
667 | ||
668 | cb->args[1] = 0; | |
2fb3573d | 669 | skip: |
c454673d TG |
670 | idx++; |
671 | } | |
672 | rcu_read_unlock(); | |
673 | cb->args[0] = idx; | |
674 | ||
675 | return skb->len; | |
676 | } | |
14c0b97d | 677 | |
9e3a5487 | 678 | static void notify_rule_change(int event, struct fib_rule *rule, |
c17084d2 TG |
679 | struct fib_rules_ops *ops, struct nlmsghdr *nlh, |
680 | u32 pid) | |
14c0b97d | 681 | { |
9e3a5487 | 682 | struct net *net; |
c17084d2 TG |
683 | struct sk_buff *skb; |
684 | int err = -ENOBUFS; | |
14c0b97d | 685 | |
9e3a5487 | 686 | net = ops->fro_net; |
339bf98f | 687 | skb = nlmsg_new(fib_rule_nlmsg_size(ops, rule), GFP_KERNEL); |
14c0b97d | 688 | if (skb == NULL) |
c17084d2 TG |
689 | goto errout; |
690 | ||
691 | err = fib_nl_fill_rule(skb, rule, pid, nlh->nlmsg_seq, event, 0, ops); | |
26932566 PM |
692 | if (err < 0) { |
693 | /* -EMSGSIZE implies BUG in fib_rule_nlmsg_size() */ | |
694 | WARN_ON(err == -EMSGSIZE); | |
695 | kfree_skb(skb); | |
696 | goto errout; | |
697 | } | |
9e3a5487 | 698 | |
1ce85fe4 PNA |
699 | rtnl_notify(skb, net, pid, ops->nlgroup, nlh, GFP_KERNEL); |
700 | return; | |
c17084d2 TG |
701 | errout: |
702 | if (err < 0) | |
5fd30ee7 | 703 | rtnl_set_sk_err(net, ops->nlgroup, err); |
14c0b97d TG |
704 | } |
705 | ||
706 | static void attach_rules(struct list_head *rules, struct net_device *dev) | |
707 | { | |
708 | struct fib_rule *rule; | |
709 | ||
710 | list_for_each_entry(rule, rules, list) { | |
491deb24 PM |
711 | if (rule->iifindex == -1 && |
712 | strcmp(dev->name, rule->iifname) == 0) | |
713 | rule->iifindex = dev->ifindex; | |
1b038a5e PM |
714 | if (rule->oifindex == -1 && |
715 | strcmp(dev->name, rule->oifname) == 0) | |
716 | rule->oifindex = dev->ifindex; | |
14c0b97d TG |
717 | } |
718 | } | |
719 | ||
720 | static void detach_rules(struct list_head *rules, struct net_device *dev) | |
721 | { | |
722 | struct fib_rule *rule; | |
723 | ||
1b038a5e | 724 | list_for_each_entry(rule, rules, list) { |
491deb24 PM |
725 | if (rule->iifindex == dev->ifindex) |
726 | rule->iifindex = -1; | |
1b038a5e PM |
727 | if (rule->oifindex == dev->ifindex) |
728 | rule->oifindex = -1; | |
729 | } | |
14c0b97d TG |
730 | } |
731 | ||
732 | ||
733 | static int fib_rules_event(struct notifier_block *this, unsigned long event, | |
351638e7 | 734 | void *ptr) |
14c0b97d | 735 | { |
351638e7 | 736 | struct net_device *dev = netdev_notifier_info_to_dev(ptr); |
c346dca1 | 737 | struct net *net = dev_net(dev); |
14c0b97d TG |
738 | struct fib_rules_ops *ops; |
739 | ||
748e2d93 | 740 | ASSERT_RTNL(); |
14c0b97d TG |
741 | |
742 | switch (event) { | |
743 | case NETDEV_REGISTER: | |
5fd30ee7 | 744 | list_for_each_entry(ops, &net->rules_ops, list) |
76c72d4f | 745 | attach_rules(&ops->rules_list, dev); |
14c0b97d TG |
746 | break; |
747 | ||
946c032e MŻ |
748 | case NETDEV_CHANGENAME: |
749 | list_for_each_entry(ops, &net->rules_ops, list) { | |
750 | detach_rules(&ops->rules_list, dev); | |
751 | attach_rules(&ops->rules_list, dev); | |
752 | } | |
753 | break; | |
754 | ||
14c0b97d | 755 | case NETDEV_UNREGISTER: |
5fd30ee7 | 756 | list_for_each_entry(ops, &net->rules_ops, list) |
76c72d4f | 757 | detach_rules(&ops->rules_list, dev); |
14c0b97d TG |
758 | break; |
759 | } | |
760 | ||
14c0b97d TG |
761 | return NOTIFY_DONE; |
762 | } | |
763 | ||
764 | static struct notifier_block fib_rules_notifier = { | |
765 | .notifier_call = fib_rules_event, | |
766 | }; | |
767 | ||
2c8c1e72 | 768 | static int __net_init fib_rules_net_init(struct net *net) |
5fd30ee7 DL |
769 | { |
770 | INIT_LIST_HEAD(&net->rules_ops); | |
771 | spin_lock_init(&net->rules_mod_lock); | |
772 | return 0; | |
773 | } | |
774 | ||
775 | static struct pernet_operations fib_rules_net_ops = { | |
776 | .init = fib_rules_net_init, | |
777 | }; | |
778 | ||
14c0b97d TG |
779 | static int __init fib_rules_init(void) |
780 | { | |
5fd30ee7 | 781 | int err; |
c7ac8679 GR |
782 | rtnl_register(PF_UNSPEC, RTM_NEWRULE, fib_nl_newrule, NULL, NULL); |
783 | rtnl_register(PF_UNSPEC, RTM_DELRULE, fib_nl_delrule, NULL, NULL); | |
784 | rtnl_register(PF_UNSPEC, RTM_GETRULE, NULL, fib_nl_dumprule, NULL); | |
9d9e6a58 | 785 | |
5d6d4809 | 786 | err = register_pernet_subsys(&fib_rules_net_ops); |
5fd30ee7 DL |
787 | if (err < 0) |
788 | goto fail; | |
789 | ||
5d6d4809 | 790 | err = register_netdevice_notifier(&fib_rules_notifier); |
5fd30ee7 DL |
791 | if (err < 0) |
792 | goto fail_unregister; | |
5d6d4809 | 793 | |
5fd30ee7 DL |
794 | return 0; |
795 | ||
796 | fail_unregister: | |
5d6d4809 | 797 | unregister_pernet_subsys(&fib_rules_net_ops); |
5fd30ee7 DL |
798 | fail: |
799 | rtnl_unregister(PF_UNSPEC, RTM_NEWRULE); | |
800 | rtnl_unregister(PF_UNSPEC, RTM_DELRULE); | |
801 | rtnl_unregister(PF_UNSPEC, RTM_GETRULE); | |
802 | return err; | |
14c0b97d TG |
803 | } |
804 | ||
805 | subsys_initcall(fib_rules_init); |