0001
0002
0003
0004
0005 #include <linux/rhashtable.h>
0006 #include <linux/mroute_base.h>
0007
0008
0009 void vif_device_init(struct vif_device *v,
0010 struct net_device *dev,
0011 unsigned long rate_limit,
0012 unsigned char threshold,
0013 unsigned short flags,
0014 unsigned short get_iflink_mask)
0015 {
0016 RCU_INIT_POINTER(v->dev, NULL);
0017 v->bytes_in = 0;
0018 v->bytes_out = 0;
0019 v->pkt_in = 0;
0020 v->pkt_out = 0;
0021 v->rate_limit = rate_limit;
0022 v->flags = flags;
0023 v->threshold = threshold;
0024 if (v->flags & get_iflink_mask)
0025 v->link = dev_get_iflink(dev);
0026 else
0027 v->link = dev->ifindex;
0028 }
0029 EXPORT_SYMBOL(vif_device_init);
0030
0031 struct mr_table *
0032 mr_table_alloc(struct net *net, u32 id,
0033 struct mr_table_ops *ops,
0034 void (*expire_func)(struct timer_list *t),
0035 void (*table_set)(struct mr_table *mrt,
0036 struct net *net))
0037 {
0038 struct mr_table *mrt;
0039 int err;
0040
0041 mrt = kzalloc(sizeof(*mrt), GFP_KERNEL);
0042 if (!mrt)
0043 return ERR_PTR(-ENOMEM);
0044 mrt->id = id;
0045 write_pnet(&mrt->net, net);
0046
0047 mrt->ops = *ops;
0048 err = rhltable_init(&mrt->mfc_hash, mrt->ops.rht_params);
0049 if (err) {
0050 kfree(mrt);
0051 return ERR_PTR(err);
0052 }
0053 INIT_LIST_HEAD(&mrt->mfc_cache_list);
0054 INIT_LIST_HEAD(&mrt->mfc_unres_queue);
0055
0056 timer_setup(&mrt->ipmr_expire_timer, expire_func, 0);
0057
0058 mrt->mroute_reg_vif_num = -1;
0059 table_set(mrt, net);
0060 return mrt;
0061 }
0062 EXPORT_SYMBOL(mr_table_alloc);
0063
0064 void *mr_mfc_find_parent(struct mr_table *mrt, void *hasharg, int parent)
0065 {
0066 struct rhlist_head *tmp, *list;
0067 struct mr_mfc *c;
0068
0069 list = rhltable_lookup(&mrt->mfc_hash, hasharg, *mrt->ops.rht_params);
0070 rhl_for_each_entry_rcu(c, tmp, list, mnode)
0071 if (parent == -1 || parent == c->mfc_parent)
0072 return c;
0073
0074 return NULL;
0075 }
0076 EXPORT_SYMBOL(mr_mfc_find_parent);
0077
0078 void *mr_mfc_find_any_parent(struct mr_table *mrt, int vifi)
0079 {
0080 struct rhlist_head *tmp, *list;
0081 struct mr_mfc *c;
0082
0083 list = rhltable_lookup(&mrt->mfc_hash, mrt->ops.cmparg_any,
0084 *mrt->ops.rht_params);
0085 rhl_for_each_entry_rcu(c, tmp, list, mnode)
0086 if (c->mfc_un.res.ttls[vifi] < 255)
0087 return c;
0088
0089 return NULL;
0090 }
0091 EXPORT_SYMBOL(mr_mfc_find_any_parent);
0092
0093 void *mr_mfc_find_any(struct mr_table *mrt, int vifi, void *hasharg)
0094 {
0095 struct rhlist_head *tmp, *list;
0096 struct mr_mfc *c, *proxy;
0097
0098 list = rhltable_lookup(&mrt->mfc_hash, hasharg, *mrt->ops.rht_params);
0099 rhl_for_each_entry_rcu(c, tmp, list, mnode) {
0100 if (c->mfc_un.res.ttls[vifi] < 255)
0101 return c;
0102
0103
0104 proxy = mr_mfc_find_any_parent(mrt, c->mfc_parent);
0105 if (proxy && proxy->mfc_un.res.ttls[vifi] < 255)
0106 return c;
0107 }
0108
0109 return mr_mfc_find_any_parent(mrt, vifi);
0110 }
0111 EXPORT_SYMBOL(mr_mfc_find_any);
0112
0113 #ifdef CONFIG_PROC_FS
0114 void *mr_vif_seq_idx(struct net *net, struct mr_vif_iter *iter, loff_t pos)
0115 {
0116 struct mr_table *mrt = iter->mrt;
0117
0118 for (iter->ct = 0; iter->ct < mrt->maxvif; ++iter->ct) {
0119 if (!VIF_EXISTS(mrt, iter->ct))
0120 continue;
0121 if (pos-- == 0)
0122 return &mrt->vif_table[iter->ct];
0123 }
0124 return NULL;
0125 }
0126 EXPORT_SYMBOL(mr_vif_seq_idx);
0127
0128 void *mr_vif_seq_next(struct seq_file *seq, void *v, loff_t *pos)
0129 {
0130 struct mr_vif_iter *iter = seq->private;
0131 struct net *net = seq_file_net(seq);
0132 struct mr_table *mrt = iter->mrt;
0133
0134 ++*pos;
0135 if (v == SEQ_START_TOKEN)
0136 return mr_vif_seq_idx(net, iter, 0);
0137
0138 while (++iter->ct < mrt->maxvif) {
0139 if (!VIF_EXISTS(mrt, iter->ct))
0140 continue;
0141 return &mrt->vif_table[iter->ct];
0142 }
0143 return NULL;
0144 }
0145 EXPORT_SYMBOL(mr_vif_seq_next);
0146
0147 void *mr_mfc_seq_idx(struct net *net,
0148 struct mr_mfc_iter *it, loff_t pos)
0149 {
0150 struct mr_table *mrt = it->mrt;
0151 struct mr_mfc *mfc;
0152
0153 rcu_read_lock();
0154 it->cache = &mrt->mfc_cache_list;
0155 list_for_each_entry_rcu(mfc, &mrt->mfc_cache_list, list)
0156 if (pos-- == 0)
0157 return mfc;
0158 rcu_read_unlock();
0159
0160 spin_lock_bh(it->lock);
0161 it->cache = &mrt->mfc_unres_queue;
0162 list_for_each_entry(mfc, it->cache, list)
0163 if (pos-- == 0)
0164 return mfc;
0165 spin_unlock_bh(it->lock);
0166
0167 it->cache = NULL;
0168 return NULL;
0169 }
0170 EXPORT_SYMBOL(mr_mfc_seq_idx);
0171
0172 void *mr_mfc_seq_next(struct seq_file *seq, void *v,
0173 loff_t *pos)
0174 {
0175 struct mr_mfc_iter *it = seq->private;
0176 struct net *net = seq_file_net(seq);
0177 struct mr_table *mrt = it->mrt;
0178 struct mr_mfc *c = v;
0179
0180 ++*pos;
0181
0182 if (v == SEQ_START_TOKEN)
0183 return mr_mfc_seq_idx(net, seq->private, 0);
0184
0185 if (c->list.next != it->cache)
0186 return list_entry(c->list.next, struct mr_mfc, list);
0187
0188 if (it->cache == &mrt->mfc_unres_queue)
0189 goto end_of_list;
0190
0191
0192 rcu_read_unlock();
0193 it->cache = &mrt->mfc_unres_queue;
0194
0195 spin_lock_bh(it->lock);
0196 if (!list_empty(it->cache))
0197 return list_first_entry(it->cache, struct mr_mfc, list);
0198
0199 end_of_list:
0200 spin_unlock_bh(it->lock);
0201 it->cache = NULL;
0202
0203 return NULL;
0204 }
0205 EXPORT_SYMBOL(mr_mfc_seq_next);
0206 #endif
0207
0208 int mr_fill_mroute(struct mr_table *mrt, struct sk_buff *skb,
0209 struct mr_mfc *c, struct rtmsg *rtm)
0210 {
0211 struct net_device *vif_dev;
0212 struct rta_mfc_stats mfcs;
0213 struct nlattr *mp_attr;
0214 struct rtnexthop *nhp;
0215 unsigned long lastuse;
0216 int ct;
0217
0218
0219 if (c->mfc_parent >= MAXVIFS) {
0220 rtm->rtm_flags |= RTNH_F_UNRESOLVED;
0221 return -ENOENT;
0222 }
0223
0224 rcu_read_lock();
0225 vif_dev = rcu_dereference(mrt->vif_table[c->mfc_parent].dev);
0226 if (vif_dev && nla_put_u32(skb, RTA_IIF, vif_dev->ifindex) < 0) {
0227 rcu_read_unlock();
0228 return -EMSGSIZE;
0229 }
0230 rcu_read_unlock();
0231
0232 if (c->mfc_flags & MFC_OFFLOAD)
0233 rtm->rtm_flags |= RTNH_F_OFFLOAD;
0234
0235 mp_attr = nla_nest_start_noflag(skb, RTA_MULTIPATH);
0236 if (!mp_attr)
0237 return -EMSGSIZE;
0238
0239 rcu_read_lock();
0240 for (ct = c->mfc_un.res.minvif; ct < c->mfc_un.res.maxvif; ct++) {
0241 struct vif_device *vif = &mrt->vif_table[ct];
0242
0243 vif_dev = rcu_dereference(vif->dev);
0244 if (vif_dev && c->mfc_un.res.ttls[ct] < 255) {
0245
0246 nhp = nla_reserve_nohdr(skb, sizeof(*nhp));
0247 if (!nhp) {
0248 rcu_read_unlock();
0249 nla_nest_cancel(skb, mp_attr);
0250 return -EMSGSIZE;
0251 }
0252
0253 nhp->rtnh_flags = 0;
0254 nhp->rtnh_hops = c->mfc_un.res.ttls[ct];
0255 nhp->rtnh_ifindex = vif_dev->ifindex;
0256 nhp->rtnh_len = sizeof(*nhp);
0257 }
0258 }
0259 rcu_read_unlock();
0260
0261 nla_nest_end(skb, mp_attr);
0262
0263 lastuse = READ_ONCE(c->mfc_un.res.lastuse);
0264 lastuse = time_after_eq(jiffies, lastuse) ? jiffies - lastuse : 0;
0265
0266 mfcs.mfcs_packets = c->mfc_un.res.pkt;
0267 mfcs.mfcs_bytes = c->mfc_un.res.bytes;
0268 mfcs.mfcs_wrong_if = c->mfc_un.res.wrong_if;
0269 if (nla_put_64bit(skb, RTA_MFC_STATS, sizeof(mfcs), &mfcs, RTA_PAD) ||
0270 nla_put_u64_64bit(skb, RTA_EXPIRES, jiffies_to_clock_t(lastuse),
0271 RTA_PAD))
0272 return -EMSGSIZE;
0273
0274 rtm->rtm_type = RTN_MULTICAST;
0275 return 1;
0276 }
0277 EXPORT_SYMBOL(mr_fill_mroute);
0278
0279 static bool mr_mfc_uses_dev(const struct mr_table *mrt,
0280 const struct mr_mfc *c,
0281 const struct net_device *dev)
0282 {
0283 int ct;
0284
0285 for (ct = c->mfc_un.res.minvif; ct < c->mfc_un.res.maxvif; ct++) {
0286 const struct net_device *vif_dev;
0287 const struct vif_device *vif;
0288
0289 vif = &mrt->vif_table[ct];
0290 vif_dev = rcu_access_pointer(vif->dev);
0291 if (vif_dev && c->mfc_un.res.ttls[ct] < 255 &&
0292 vif_dev == dev)
0293 return true;
0294 }
0295 return false;
0296 }
0297
0298 int mr_table_dump(struct mr_table *mrt, struct sk_buff *skb,
0299 struct netlink_callback *cb,
0300 int (*fill)(struct mr_table *mrt, struct sk_buff *skb,
0301 u32 portid, u32 seq, struct mr_mfc *c,
0302 int cmd, int flags),
0303 spinlock_t *lock, struct fib_dump_filter *filter)
0304 {
0305 unsigned int e = 0, s_e = cb->args[1];
0306 unsigned int flags = NLM_F_MULTI;
0307 struct mr_mfc *mfc;
0308 int err;
0309
0310 if (filter->filter_set)
0311 flags |= NLM_F_DUMP_FILTERED;
0312
0313 list_for_each_entry_rcu(mfc, &mrt->mfc_cache_list, list) {
0314 if (e < s_e)
0315 goto next_entry;
0316 if (filter->dev &&
0317 !mr_mfc_uses_dev(mrt, mfc, filter->dev))
0318 goto next_entry;
0319
0320 err = fill(mrt, skb, NETLINK_CB(cb->skb).portid,
0321 cb->nlh->nlmsg_seq, mfc, RTM_NEWROUTE, flags);
0322 if (err < 0)
0323 goto out;
0324 next_entry:
0325 e++;
0326 }
0327
0328 spin_lock_bh(lock);
0329 list_for_each_entry(mfc, &mrt->mfc_unres_queue, list) {
0330 if (e < s_e)
0331 goto next_entry2;
0332 if (filter->dev &&
0333 !mr_mfc_uses_dev(mrt, mfc, filter->dev))
0334 goto next_entry2;
0335
0336 err = fill(mrt, skb, NETLINK_CB(cb->skb).portid,
0337 cb->nlh->nlmsg_seq, mfc, RTM_NEWROUTE, flags);
0338 if (err < 0) {
0339 spin_unlock_bh(lock);
0340 goto out;
0341 }
0342 next_entry2:
0343 e++;
0344 }
0345 spin_unlock_bh(lock);
0346 err = 0;
0347 out:
0348 cb->args[1] = e;
0349 return err;
0350 }
0351 EXPORT_SYMBOL(mr_table_dump);
0352
0353 int mr_rtm_dumproute(struct sk_buff *skb, struct netlink_callback *cb,
0354 struct mr_table *(*iter)(struct net *net,
0355 struct mr_table *mrt),
0356 int (*fill)(struct mr_table *mrt,
0357 struct sk_buff *skb,
0358 u32 portid, u32 seq, struct mr_mfc *c,
0359 int cmd, int flags),
0360 spinlock_t *lock, struct fib_dump_filter *filter)
0361 {
0362 unsigned int t = 0, s_t = cb->args[0];
0363 struct net *net = sock_net(skb->sk);
0364 struct mr_table *mrt;
0365 int err;
0366
0367
0368
0369
0370 if (filter->filter_set) {
0371 if (filter->protocol || filter->flags ||
0372 (filter->rt_type && filter->rt_type != RTN_MULTICAST))
0373 return skb->len;
0374 }
0375
0376 rcu_read_lock();
0377 for (mrt = iter(net, NULL); mrt; mrt = iter(net, mrt)) {
0378 if (t < s_t)
0379 goto next_table;
0380
0381 err = mr_table_dump(mrt, skb, cb, fill, lock, filter);
0382 if (err < 0)
0383 break;
0384 cb->args[1] = 0;
0385 next_table:
0386 t++;
0387 }
0388 rcu_read_unlock();
0389
0390 cb->args[0] = t;
0391
0392 return skb->len;
0393 }
0394 EXPORT_SYMBOL(mr_rtm_dumproute);
0395
0396 int mr_dump(struct net *net, struct notifier_block *nb, unsigned short family,
0397 int (*rules_dump)(struct net *net,
0398 struct notifier_block *nb,
0399 struct netlink_ext_ack *extack),
0400 struct mr_table *(*mr_iter)(struct net *net,
0401 struct mr_table *mrt),
0402 struct netlink_ext_ack *extack)
0403 {
0404 struct mr_table *mrt;
0405 int err;
0406
0407 err = rules_dump(net, nb, extack);
0408 if (err)
0409 return err;
0410
0411 for (mrt = mr_iter(net, NULL); mrt; mrt = mr_iter(net, mrt)) {
0412 struct vif_device *v = &mrt->vif_table[0];
0413 struct net_device *vif_dev;
0414 struct mr_mfc *mfc;
0415 int vifi;
0416
0417
0418 rcu_read_lock();
0419 for (vifi = 0; vifi < mrt->maxvif; vifi++, v++) {
0420 vif_dev = rcu_dereference(v->dev);
0421 if (!vif_dev)
0422 continue;
0423
0424 err = mr_call_vif_notifier(nb, family,
0425 FIB_EVENT_VIF_ADD, v,
0426 vif_dev, vifi,
0427 mrt->id, extack);
0428 if (err)
0429 break;
0430 }
0431 rcu_read_unlock();
0432
0433 if (err)
0434 return err;
0435
0436
0437 list_for_each_entry_rcu(mfc, &mrt->mfc_cache_list, list) {
0438 err = mr_call_mfc_notifier(nb, family,
0439 FIB_EVENT_ENTRY_ADD,
0440 mfc, mrt->id, extack);
0441 if (err)
0442 return err;
0443 }
0444 }
0445
0446 return 0;
0447 }
0448 EXPORT_SYMBOL(mr_dump);