Back to home page

OSCL-LXR

 
 

    


0001 /* Linux multicast routing support
0002  * Common logic shared by IPv4 [ipmr] and IPv6 [ip6mr] implementation
0003  */
0004 
0005 #include <linux/rhashtable.h>
0006 #include <linux/mroute_base.h>
0007 
0008 /* Sets everything common except 'dev', since that is done under locking */
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         /* It's ok if the vifi is part of the static tree */
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     /* exhausted cache_array, show unresolved */
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     /* If cache is unresolved, don't try to parse IIF and OIF */
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     /* multicast does not track protocol or have route type other
0368      * than RTN_MULTICAST
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         /* Notifiy on table VIF entries */
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         /* Notify on table MFC entries */
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);