ipmr.c 55 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477
  1. /*
  2. * IP multicast routing support for mrouted 3.6/3.8
  3. *
  4. * (c) 1995 Alan Cox, <alan@lxorguk.ukuu.org.uk>
  5. * Linux Consultancy and Custom Driver Development
  6. *
  7. * This program is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU General Public License
  9. * as published by the Free Software Foundation; either version
  10. * 2 of the License, or (at your option) any later version.
  11. *
  12. * Fixes:
  13. * Michael Chastain : Incorrect size of copying.
  14. * Alan Cox : Added the cache manager code
  15. * Alan Cox : Fixed the clone/copy bug and device race.
  16. * Mike McLagan : Routing by source
  17. * Malcolm Beattie : Buffer handling fixes.
  18. * Alexey Kuznetsov : Double buffer free and other fixes.
  19. * SVR Anand : Fixed several multicast bugs and problems.
  20. * Alexey Kuznetsov : Status, optimisations and more.
  21. * Brad Parker : Better behaviour on mrouted upcall
  22. * overflow.
  23. * Carlos Picoto : PIMv1 Support
  24. * Pavlin Ivanov Radoslavov: PIMv2 Registers must checksum only PIM header
  25. * Relax this requirement to work with older peers.
  26. *
  27. */
  28. #include <asm/system.h>
  29. #include <asm/uaccess.h>
  30. #include <linux/types.h>
  31. #include <linux/capability.h>
  32. #include <linux/errno.h>
  33. #include <linux/timer.h>
  34. #include <linux/mm.h>
  35. #include <linux/kernel.h>
  36. #include <linux/fcntl.h>
  37. #include <linux/stat.h>
  38. #include <linux/socket.h>
  39. #include <linux/in.h>
  40. #include <linux/inet.h>
  41. #include <linux/netdevice.h>
  42. #include <linux/inetdevice.h>
  43. #include <linux/igmp.h>
  44. #include <linux/proc_fs.h>
  45. #include <linux/seq_file.h>
  46. #include <linux/mroute.h>
  47. #include <linux/init.h>
  48. #include <linux/if_ether.h>
  49. #include <linux/slab.h>
  50. #include <net/net_namespace.h>
  51. #include <net/ip.h>
  52. #include <net/protocol.h>
  53. #include <linux/skbuff.h>
  54. #include <net/route.h>
  55. #include <net/sock.h>
  56. #include <net/icmp.h>
  57. #include <net/udp.h>
  58. #include <net/raw.h>
  59. #include <linux/notifier.h>
  60. #include <linux/if_arp.h>
  61. #include <linux/netfilter_ipv4.h>
  62. #include <net/ipip.h>
  63. #include <net/checksum.h>
  64. #include <net/netlink.h>
  65. #include <net/fib_rules.h>
  66. #if defined(CONFIG_IP_PIMSM_V1) || defined(CONFIG_IP_PIMSM_V2)
  67. #define CONFIG_IP_PIMSM 1
  68. #endif
  69. struct mr_table {
  70. struct list_head list;
  71. #ifdef CONFIG_NET_NS
  72. struct net *net;
  73. #endif
  74. u32 id;
  75. struct sock __rcu *mroute_sk;
  76. struct timer_list ipmr_expire_timer;
  77. struct list_head mfc_unres_queue;
  78. struct list_head mfc_cache_array[MFC_LINES];
  79. struct vif_device vif_table[MAXVIFS];
  80. int maxvif;
  81. atomic_t cache_resolve_queue_len;
  82. int mroute_do_assert;
  83. int mroute_do_pim;
  84. #if defined(CONFIG_IP_PIMSM_V1) || defined(CONFIG_IP_PIMSM_V2)
  85. int mroute_reg_vif_num;
  86. #endif
  87. };
  88. struct ipmr_rule {
  89. struct fib_rule common;
  90. };
  91. struct ipmr_result {
  92. struct mr_table *mrt;
  93. };
  94. /* Big lock, protecting vif table, mrt cache and mroute socket state.
  95. * Note that the changes are semaphored via rtnl_lock.
  96. */
  97. static DEFINE_RWLOCK(mrt_lock);
  98. /*
  99. * Multicast router control variables
  100. */
  101. #define VIF_EXISTS(_mrt, _idx) ((_mrt)->vif_table[_idx].dev != NULL)
  102. /* Special spinlock for queue of unresolved entries */
  103. static DEFINE_SPINLOCK(mfc_unres_lock);
  104. /* We return to original Alan's scheme. Hash table of resolved
  105. * entries is changed only in process context and protected
  106. * with weak lock mrt_lock. Queue of unresolved entries is protected
  107. * with strong spinlock mfc_unres_lock.
  108. *
  109. * In this case data path is free of exclusive locks at all.
  110. */
  111. static struct kmem_cache *mrt_cachep __read_mostly;
  112. static struct mr_table *ipmr_new_table(struct net *net, u32 id);
  113. static int ip_mr_forward(struct net *net, struct mr_table *mrt,
  114. struct sk_buff *skb, struct mfc_cache *cache,
  115. int local);
  116. static int ipmr_cache_report(struct mr_table *mrt,
  117. struct sk_buff *pkt, vifi_t vifi, int assert);
  118. static int __ipmr_fill_mroute(struct mr_table *mrt, struct sk_buff *skb,
  119. struct mfc_cache *c, struct rtmsg *rtm);
  120. static void ipmr_expire_process(unsigned long arg);
  121. #ifdef CONFIG_IP_MROUTE_MULTIPLE_TABLES
  122. #define ipmr_for_each_table(mrt, net) \
  123. list_for_each_entry_rcu(mrt, &net->ipv4.mr_tables, list)
  124. static struct mr_table *ipmr_get_table(struct net *net, u32 id)
  125. {
  126. struct mr_table *mrt;
  127. ipmr_for_each_table(mrt, net) {
  128. if (mrt->id == id)
  129. return mrt;
  130. }
  131. return NULL;
  132. }
  133. static int ipmr_fib_lookup(struct net *net, struct flowi *flp,
  134. struct mr_table **mrt)
  135. {
  136. struct ipmr_result res;
  137. struct fib_lookup_arg arg = { .result = &res, };
  138. int err;
  139. err = fib_rules_lookup(net->ipv4.mr_rules_ops, flp, 0, &arg);
  140. if (err < 0)
  141. return err;
  142. *mrt = res.mrt;
  143. return 0;
  144. }
  145. static int ipmr_rule_action(struct fib_rule *rule, struct flowi *flp,
  146. int flags, struct fib_lookup_arg *arg)
  147. {
  148. struct ipmr_result *res = arg->result;
  149. struct mr_table *mrt;
  150. switch (rule->action) {
  151. case FR_ACT_TO_TBL:
  152. break;
  153. case FR_ACT_UNREACHABLE:
  154. return -ENETUNREACH;
  155. case FR_ACT_PROHIBIT:
  156. return -EACCES;
  157. case FR_ACT_BLACKHOLE:
  158. default:
  159. return -EINVAL;
  160. }
  161. mrt = ipmr_get_table(rule->fr_net, rule->table);
  162. if (mrt == NULL)
  163. return -EAGAIN;
  164. res->mrt = mrt;
  165. return 0;
  166. }
  167. static int ipmr_rule_match(struct fib_rule *rule, struct flowi *fl, int flags)
  168. {
  169. return 1;
  170. }
  171. static const struct nla_policy ipmr_rule_policy[FRA_MAX + 1] = {
  172. FRA_GENERIC_POLICY,
  173. };
  174. static int ipmr_rule_configure(struct fib_rule *rule, struct sk_buff *skb,
  175. struct fib_rule_hdr *frh, struct nlattr **tb)
  176. {
  177. return 0;
  178. }
  179. static int ipmr_rule_compare(struct fib_rule *rule, struct fib_rule_hdr *frh,
  180. struct nlattr **tb)
  181. {
  182. return 1;
  183. }
  184. static int ipmr_rule_fill(struct fib_rule *rule, struct sk_buff *skb,
  185. struct fib_rule_hdr *frh)
  186. {
  187. frh->dst_len = 0;
  188. frh->src_len = 0;
  189. frh->tos = 0;
  190. return 0;
  191. }
  192. static const struct fib_rules_ops __net_initdata ipmr_rules_ops_template = {
  193. .family = RTNL_FAMILY_IPMR,
  194. .rule_size = sizeof(struct ipmr_rule),
  195. .addr_size = sizeof(u32),
  196. .action = ipmr_rule_action,
  197. .match = ipmr_rule_match,
  198. .configure = ipmr_rule_configure,
  199. .compare = ipmr_rule_compare,
  200. .default_pref = fib_default_rule_pref,
  201. .fill = ipmr_rule_fill,
  202. .nlgroup = RTNLGRP_IPV4_RULE,
  203. .policy = ipmr_rule_policy,
  204. .owner = THIS_MODULE,
  205. };
  206. static int __net_init ipmr_rules_init(struct net *net)
  207. {
  208. struct fib_rules_ops *ops;
  209. struct mr_table *mrt;
  210. int err;
  211. ops = fib_rules_register(&ipmr_rules_ops_template, net);
  212. if (IS_ERR(ops))
  213. return PTR_ERR(ops);
  214. INIT_LIST_HEAD(&net->ipv4.mr_tables);
  215. mrt = ipmr_new_table(net, RT_TABLE_DEFAULT);
  216. if (mrt == NULL) {
  217. err = -ENOMEM;
  218. goto err1;
  219. }
  220. err = fib_default_rule_add(ops, 0x7fff, RT_TABLE_DEFAULT, 0);
  221. if (err < 0)
  222. goto err2;
  223. net->ipv4.mr_rules_ops = ops;
  224. return 0;
  225. err2:
  226. kfree(mrt);
  227. err1:
  228. fib_rules_unregister(ops);
  229. return err;
  230. }
  231. static void __net_exit ipmr_rules_exit(struct net *net)
  232. {
  233. struct mr_table *mrt, *next;
  234. list_for_each_entry_safe(mrt, next, &net->ipv4.mr_tables, list) {
  235. list_del(&mrt->list);
  236. kfree(mrt);
  237. }
  238. fib_rules_unregister(net->ipv4.mr_rules_ops);
  239. }
  240. #else
  241. #define ipmr_for_each_table(mrt, net) \
  242. for (mrt = net->ipv4.mrt; mrt; mrt = NULL)
  243. static struct mr_table *ipmr_get_table(struct net *net, u32 id)
  244. {
  245. return net->ipv4.mrt;
  246. }
  247. static int ipmr_fib_lookup(struct net *net, struct flowi *flp,
  248. struct mr_table **mrt)
  249. {
  250. *mrt = net->ipv4.mrt;
  251. return 0;
  252. }
  253. static int __net_init ipmr_rules_init(struct net *net)
  254. {
  255. net->ipv4.mrt = ipmr_new_table(net, RT_TABLE_DEFAULT);
  256. return net->ipv4.mrt ? 0 : -ENOMEM;
  257. }
  258. static void __net_exit ipmr_rules_exit(struct net *net)
  259. {
  260. kfree(net->ipv4.mrt);
  261. }
  262. #endif
  263. static struct mr_table *ipmr_new_table(struct net *net, u32 id)
  264. {
  265. struct mr_table *mrt;
  266. unsigned int i;
  267. mrt = ipmr_get_table(net, id);
  268. if (mrt != NULL)
  269. return mrt;
  270. mrt = kzalloc(sizeof(*mrt), GFP_KERNEL);
  271. if (mrt == NULL)
  272. return NULL;
  273. write_pnet(&mrt->net, net);
  274. mrt->id = id;
  275. /* Forwarding cache */
  276. for (i = 0; i < MFC_LINES; i++)
  277. INIT_LIST_HEAD(&mrt->mfc_cache_array[i]);
  278. INIT_LIST_HEAD(&mrt->mfc_unres_queue);
  279. setup_timer(&mrt->ipmr_expire_timer, ipmr_expire_process,
  280. (unsigned long)mrt);
  281. #ifdef CONFIG_IP_PIMSM
  282. mrt->mroute_reg_vif_num = -1;
  283. #endif
  284. #ifdef CONFIG_IP_MROUTE_MULTIPLE_TABLES
  285. list_add_tail_rcu(&mrt->list, &net->ipv4.mr_tables);
  286. #endif
  287. return mrt;
  288. }
  289. /* Service routines creating virtual interfaces: DVMRP tunnels and PIMREG */
  290. static void ipmr_del_tunnel(struct net_device *dev, struct vifctl *v)
  291. {
  292. struct net *net = dev_net(dev);
  293. dev_close(dev);
  294. dev = __dev_get_by_name(net, "tunl0");
  295. if (dev) {
  296. const struct net_device_ops *ops = dev->netdev_ops;
  297. struct ifreq ifr;
  298. struct ip_tunnel_parm p;
  299. memset(&p, 0, sizeof(p));
  300. p.iph.daddr = v->vifc_rmt_addr.s_addr;
  301. p.iph.saddr = v->vifc_lcl_addr.s_addr;
  302. p.iph.version = 4;
  303. p.iph.ihl = 5;
  304. p.iph.protocol = IPPROTO_IPIP;
  305. sprintf(p.name, "dvmrp%d", v->vifc_vifi);
  306. ifr.ifr_ifru.ifru_data = (__force void __user *)&p;
  307. if (ops->ndo_do_ioctl) {
  308. mm_segment_t oldfs = get_fs();
  309. set_fs(KERNEL_DS);
  310. ops->ndo_do_ioctl(dev, &ifr, SIOCDELTUNNEL);
  311. set_fs(oldfs);
  312. }
  313. }
  314. }
  315. static
  316. struct net_device *ipmr_new_tunnel(struct net *net, struct vifctl *v)
  317. {
  318. struct net_device *dev;
  319. dev = __dev_get_by_name(net, "tunl0");
  320. if (dev) {
  321. const struct net_device_ops *ops = dev->netdev_ops;
  322. int err;
  323. struct ifreq ifr;
  324. struct ip_tunnel_parm p;
  325. struct in_device *in_dev;
  326. memset(&p, 0, sizeof(p));
  327. p.iph.daddr = v->vifc_rmt_addr.s_addr;
  328. p.iph.saddr = v->vifc_lcl_addr.s_addr;
  329. p.iph.version = 4;
  330. p.iph.ihl = 5;
  331. p.iph.protocol = IPPROTO_IPIP;
  332. sprintf(p.name, "dvmrp%d", v->vifc_vifi);
  333. ifr.ifr_ifru.ifru_data = (__force void __user *)&p;
  334. if (ops->ndo_do_ioctl) {
  335. mm_segment_t oldfs = get_fs();
  336. set_fs(KERNEL_DS);
  337. err = ops->ndo_do_ioctl(dev, &ifr, SIOCADDTUNNEL);
  338. set_fs(oldfs);
  339. } else {
  340. err = -EOPNOTSUPP;
  341. }
  342. dev = NULL;
  343. if (err == 0 &&
  344. (dev = __dev_get_by_name(net, p.name)) != NULL) {
  345. dev->flags |= IFF_MULTICAST;
  346. in_dev = __in_dev_get_rtnl(dev);
  347. if (in_dev == NULL)
  348. goto failure;
  349. ipv4_devconf_setall(in_dev);
  350. IPV4_DEVCONF(in_dev->cnf, RP_FILTER) = 0;
  351. if (dev_open(dev))
  352. goto failure;
  353. dev_hold(dev);
  354. }
  355. }
  356. return dev;
  357. failure:
  358. /* allow the register to be completed before unregistering. */
  359. rtnl_unlock();
  360. rtnl_lock();
  361. unregister_netdevice(dev);
  362. return NULL;
  363. }
  364. #ifdef CONFIG_IP_PIMSM
  365. static netdev_tx_t reg_vif_xmit(struct sk_buff *skb, struct net_device *dev)
  366. {
  367. struct net *net = dev_net(dev);
  368. struct mr_table *mrt;
  369. struct flowi fl = {
  370. .oif = dev->ifindex,
  371. .iif = skb->skb_iif,
  372. .mark = skb->mark,
  373. };
  374. int err;
  375. err = ipmr_fib_lookup(net, &fl, &mrt);
  376. if (err < 0) {
  377. kfree_skb(skb);
  378. return err;
  379. }
  380. read_lock(&mrt_lock);
  381. dev->stats.tx_bytes += skb->len;
  382. dev->stats.tx_packets++;
  383. ipmr_cache_report(mrt, skb, mrt->mroute_reg_vif_num, IGMPMSG_WHOLEPKT);
  384. read_unlock(&mrt_lock);
  385. kfree_skb(skb);
  386. return NETDEV_TX_OK;
  387. }
  388. static const struct net_device_ops reg_vif_netdev_ops = {
  389. .ndo_start_xmit = reg_vif_xmit,
  390. };
  391. static void reg_vif_setup(struct net_device *dev)
  392. {
  393. dev->type = ARPHRD_PIMREG;
  394. dev->mtu = ETH_DATA_LEN - sizeof(struct iphdr) - 8;
  395. dev->flags = IFF_NOARP;
  396. dev->netdev_ops = &reg_vif_netdev_ops,
  397. dev->destructor = free_netdev;
  398. dev->features |= NETIF_F_NETNS_LOCAL;
  399. }
  400. static struct net_device *ipmr_reg_vif(struct net *net, struct mr_table *mrt)
  401. {
  402. struct net_device *dev;
  403. struct in_device *in_dev;
  404. char name[IFNAMSIZ];
  405. if (mrt->id == RT_TABLE_DEFAULT)
  406. sprintf(name, "pimreg");
  407. else
  408. sprintf(name, "pimreg%u", mrt->id);
  409. dev = alloc_netdev(0, name, reg_vif_setup);
  410. if (dev == NULL)
  411. return NULL;
  412. dev_net_set(dev, net);
  413. if (register_netdevice(dev)) {
  414. free_netdev(dev);
  415. return NULL;
  416. }
  417. dev->iflink = 0;
  418. rcu_read_lock();
  419. in_dev = __in_dev_get_rcu(dev);
  420. if (!in_dev) {
  421. rcu_read_unlock();
  422. goto failure;
  423. }
  424. ipv4_devconf_setall(in_dev);
  425. IPV4_DEVCONF(in_dev->cnf, RP_FILTER) = 0;
  426. rcu_read_unlock();
  427. if (dev_open(dev))
  428. goto failure;
  429. dev_hold(dev);
  430. return dev;
  431. failure:
  432. /* allow the register to be completed before unregistering. */
  433. rtnl_unlock();
  434. rtnl_lock();
  435. unregister_netdevice(dev);
  436. return NULL;
  437. }
  438. #endif
  439. /*
  440. * Delete a VIF entry
  441. * @notify: Set to 1, if the caller is a notifier_call
  442. */
  443. static int vif_delete(struct mr_table *mrt, int vifi, int notify,
  444. struct list_head *head)
  445. {
  446. struct vif_device *v;
  447. struct net_device *dev;
  448. struct in_device *in_dev;
  449. if (vifi < 0 || vifi >= mrt->maxvif)
  450. return -EADDRNOTAVAIL;
  451. v = &mrt->vif_table[vifi];
  452. write_lock_bh(&mrt_lock);
  453. dev = v->dev;
  454. v->dev = NULL;
  455. if (!dev) {
  456. write_unlock_bh(&mrt_lock);
  457. return -EADDRNOTAVAIL;
  458. }
  459. #ifdef CONFIG_IP_PIMSM
  460. if (vifi == mrt->mroute_reg_vif_num)
  461. mrt->mroute_reg_vif_num = -1;
  462. #endif
  463. if (vifi + 1 == mrt->maxvif) {
  464. int tmp;
  465. for (tmp = vifi - 1; tmp >= 0; tmp--) {
  466. if (VIF_EXISTS(mrt, tmp))
  467. break;
  468. }
  469. mrt->maxvif = tmp+1;
  470. }
  471. write_unlock_bh(&mrt_lock);
  472. dev_set_allmulti(dev, -1);
  473. in_dev = __in_dev_get_rtnl(dev);
  474. if (in_dev) {
  475. IPV4_DEVCONF(in_dev->cnf, MC_FORWARDING)--;
  476. ip_rt_multicast_event(in_dev);
  477. }
  478. if (v->flags & (VIFF_TUNNEL | VIFF_REGISTER) && !notify)
  479. unregister_netdevice_queue(dev, head);
  480. dev_put(dev);
  481. return 0;
  482. }
  483. static void ipmr_cache_free_rcu(struct rcu_head *head)
  484. {
  485. struct mfc_cache *c = container_of(head, struct mfc_cache, rcu);
  486. kmem_cache_free(mrt_cachep, c);
  487. }
  488. static inline void ipmr_cache_free(struct mfc_cache *c)
  489. {
  490. call_rcu(&c->rcu, ipmr_cache_free_rcu);
  491. }
  492. /* Destroy an unresolved cache entry, killing queued skbs
  493. * and reporting error to netlink readers.
  494. */
  495. static void ipmr_destroy_unres(struct mr_table *mrt, struct mfc_cache *c)
  496. {
  497. struct net *net = read_pnet(&mrt->net);
  498. struct sk_buff *skb;
  499. struct nlmsgerr *e;
  500. atomic_dec(&mrt->cache_resolve_queue_len);
  501. while ((skb = skb_dequeue(&c->mfc_un.unres.unresolved))) {
  502. if (ip_hdr(skb)->version == 0) {
  503. struct nlmsghdr *nlh = (struct nlmsghdr *)skb_pull(skb, sizeof(struct iphdr));
  504. nlh->nlmsg_type = NLMSG_ERROR;
  505. nlh->nlmsg_len = NLMSG_LENGTH(sizeof(struct nlmsgerr));
  506. skb_trim(skb, nlh->nlmsg_len);
  507. e = NLMSG_DATA(nlh);
  508. e->error = -ETIMEDOUT;
  509. memset(&e->msg, 0, sizeof(e->msg));
  510. rtnl_unicast(skb, net, NETLINK_CB(skb).pid);
  511. } else {
  512. kfree_skb(skb);
  513. }
  514. }
  515. ipmr_cache_free(c);
  516. }
  517. /* Timer process for the unresolved queue. */
  518. static void ipmr_expire_process(unsigned long arg)
  519. {
  520. struct mr_table *mrt = (struct mr_table *)arg;
  521. unsigned long now;
  522. unsigned long expires;
  523. struct mfc_cache *c, *next;
  524. if (!spin_trylock(&mfc_unres_lock)) {
  525. mod_timer(&mrt->ipmr_expire_timer, jiffies+HZ/10);
  526. return;
  527. }
  528. if (list_empty(&mrt->mfc_unres_queue))
  529. goto out;
  530. now = jiffies;
  531. expires = 10*HZ;
  532. list_for_each_entry_safe(c, next, &mrt->mfc_unres_queue, list) {
  533. if (time_after(c->mfc_un.unres.expires, now)) {
  534. unsigned long interval = c->mfc_un.unres.expires - now;
  535. if (interval < expires)
  536. expires = interval;
  537. continue;
  538. }
  539. list_del(&c->list);
  540. ipmr_destroy_unres(mrt, c);
  541. }
  542. if (!list_empty(&mrt->mfc_unres_queue))
  543. mod_timer(&mrt->ipmr_expire_timer, jiffies + expires);
  544. out:
  545. spin_unlock(&mfc_unres_lock);
  546. }
  547. /* Fill oifs list. It is called under write locked mrt_lock. */
  548. static void ipmr_update_thresholds(struct mr_table *mrt, struct mfc_cache *cache,
  549. unsigned char *ttls)
  550. {
  551. int vifi;
  552. cache->mfc_un.res.minvif = MAXVIFS;
  553. cache->mfc_un.res.maxvif = 0;
  554. memset(cache->mfc_un.res.ttls, 255, MAXVIFS);
  555. for (vifi = 0; vifi < mrt->maxvif; vifi++) {
  556. if (VIF_EXISTS(mrt, vifi) &&
  557. ttls[vifi] && ttls[vifi] < 255) {
  558. cache->mfc_un.res.ttls[vifi] = ttls[vifi];
  559. if (cache->mfc_un.res.minvif > vifi)
  560. cache->mfc_un.res.minvif = vifi;
  561. if (cache->mfc_un.res.maxvif <= vifi)
  562. cache->mfc_un.res.maxvif = vifi + 1;
  563. }
  564. }
  565. }
  566. static int vif_add(struct net *net, struct mr_table *mrt,
  567. struct vifctl *vifc, int mrtsock)
  568. {
  569. int vifi = vifc->vifc_vifi;
  570. struct vif_device *v = &mrt->vif_table[vifi];
  571. struct net_device *dev;
  572. struct in_device *in_dev;
  573. int err;
  574. /* Is vif busy ? */
  575. if (VIF_EXISTS(mrt, vifi))
  576. return -EADDRINUSE;
  577. switch (vifc->vifc_flags) {
  578. #ifdef CONFIG_IP_PIMSM
  579. case VIFF_REGISTER:
  580. /*
  581. * Special Purpose VIF in PIM
  582. * All the packets will be sent to the daemon
  583. */
  584. if (mrt->mroute_reg_vif_num >= 0)
  585. return -EADDRINUSE;
  586. dev = ipmr_reg_vif(net, mrt);
  587. if (!dev)
  588. return -ENOBUFS;
  589. err = dev_set_allmulti(dev, 1);
  590. if (err) {
  591. unregister_netdevice(dev);
  592. dev_put(dev);
  593. return err;
  594. }
  595. break;
  596. #endif
  597. case VIFF_TUNNEL:
  598. dev = ipmr_new_tunnel(net, vifc);
  599. if (!dev)
  600. return -ENOBUFS;
  601. err = dev_set_allmulti(dev, 1);
  602. if (err) {
  603. ipmr_del_tunnel(dev, vifc);
  604. dev_put(dev);
  605. return err;
  606. }
  607. break;
  608. case VIFF_USE_IFINDEX:
  609. case 0:
  610. if (vifc->vifc_flags == VIFF_USE_IFINDEX) {
  611. dev = dev_get_by_index(net, vifc->vifc_lcl_ifindex);
  612. if (dev && __in_dev_get_rtnl(dev) == NULL) {
  613. dev_put(dev);
  614. return -EADDRNOTAVAIL;
  615. }
  616. } else {
  617. dev = ip_dev_find(net, vifc->vifc_lcl_addr.s_addr);
  618. }
  619. if (!dev)
  620. return -EADDRNOTAVAIL;
  621. err = dev_set_allmulti(dev, 1);
  622. if (err) {
  623. dev_put(dev);
  624. return err;
  625. }
  626. break;
  627. default:
  628. return -EINVAL;
  629. }
  630. in_dev = __in_dev_get_rtnl(dev);
  631. if (!in_dev) {
  632. dev_put(dev);
  633. return -EADDRNOTAVAIL;
  634. }
  635. IPV4_DEVCONF(in_dev->cnf, MC_FORWARDING)++;
  636. ip_rt_multicast_event(in_dev);
  637. /* Fill in the VIF structures */
  638. v->rate_limit = vifc->vifc_rate_limit;
  639. v->local = vifc->vifc_lcl_addr.s_addr;
  640. v->remote = vifc->vifc_rmt_addr.s_addr;
  641. v->flags = vifc->vifc_flags;
  642. if (!mrtsock)
  643. v->flags |= VIFF_STATIC;
  644. v->threshold = vifc->vifc_threshold;
  645. v->bytes_in = 0;
  646. v->bytes_out = 0;
  647. v->pkt_in = 0;
  648. v->pkt_out = 0;
  649. v->link = dev->ifindex;
  650. if (v->flags & (VIFF_TUNNEL | VIFF_REGISTER))
  651. v->link = dev->iflink;
  652. /* And finish update writing critical data */
  653. write_lock_bh(&mrt_lock);
  654. v->dev = dev;
  655. #ifdef CONFIG_IP_PIMSM
  656. if (v->flags & VIFF_REGISTER)
  657. mrt->mroute_reg_vif_num = vifi;
  658. #endif
  659. if (vifi+1 > mrt->maxvif)
  660. mrt->maxvif = vifi+1;
  661. write_unlock_bh(&mrt_lock);
  662. return 0;
  663. }
  664. /* called with rcu_read_lock() */
  665. static struct mfc_cache *ipmr_cache_find(struct mr_table *mrt,
  666. __be32 origin,
  667. __be32 mcastgrp)
  668. {
  669. int line = MFC_HASH(mcastgrp, origin);
  670. struct mfc_cache *c;
  671. list_for_each_entry_rcu(c, &mrt->mfc_cache_array[line], list) {
  672. if (c->mfc_origin == origin && c->mfc_mcastgrp == mcastgrp)
  673. return c;
  674. }
  675. return NULL;
  676. }
  677. /*
  678. * Allocate a multicast cache entry
  679. */
  680. static struct mfc_cache *ipmr_cache_alloc(void)
  681. {
  682. struct mfc_cache *c = kmem_cache_zalloc(mrt_cachep, GFP_KERNEL);
  683. if (c)
  684. c->mfc_un.res.minvif = MAXVIFS;
  685. return c;
  686. }
  687. static struct mfc_cache *ipmr_cache_alloc_unres(void)
  688. {
  689. struct mfc_cache *c = kmem_cache_zalloc(mrt_cachep, GFP_ATOMIC);
  690. if (c) {
  691. skb_queue_head_init(&c->mfc_un.unres.unresolved);
  692. c->mfc_un.unres.expires = jiffies + 10*HZ;
  693. }
  694. return c;
  695. }
  696. /*
  697. * A cache entry has gone into a resolved state from queued
  698. */
  699. static void ipmr_cache_resolve(struct net *net, struct mr_table *mrt,
  700. struct mfc_cache *uc, struct mfc_cache *c)
  701. {
  702. struct sk_buff *skb;
  703. struct nlmsgerr *e;
  704. /* Play the pending entries through our router */
  705. while ((skb = __skb_dequeue(&uc->mfc_un.unres.unresolved))) {
  706. if (ip_hdr(skb)->version == 0) {
  707. struct nlmsghdr *nlh = (struct nlmsghdr *)skb_pull(skb, sizeof(struct iphdr));
  708. if (__ipmr_fill_mroute(mrt, skb, c, NLMSG_DATA(nlh)) > 0) {
  709. nlh->nlmsg_len = skb_tail_pointer(skb) -
  710. (u8 *)nlh;
  711. } else {
  712. nlh->nlmsg_type = NLMSG_ERROR;
  713. nlh->nlmsg_len = NLMSG_LENGTH(sizeof(struct nlmsgerr));
  714. skb_trim(skb, nlh->nlmsg_len);
  715. e = NLMSG_DATA(nlh);
  716. e->error = -EMSGSIZE;
  717. memset(&e->msg, 0, sizeof(e->msg));
  718. }
  719. rtnl_unicast(skb, net, NETLINK_CB(skb).pid);
  720. } else {
  721. ip_mr_forward(net, mrt, skb, c, 0);
  722. }
  723. }
  724. }
  725. /*
  726. * Bounce a cache query up to mrouted. We could use netlink for this but mrouted
  727. * expects the following bizarre scheme.
  728. *
  729. * Called under mrt_lock.
  730. */
  731. static int ipmr_cache_report(struct mr_table *mrt,
  732. struct sk_buff *pkt, vifi_t vifi, int assert)
  733. {
  734. struct sk_buff *skb;
  735. const int ihl = ip_hdrlen(pkt);
  736. struct igmphdr *igmp;
  737. struct igmpmsg *msg;
  738. struct sock *mroute_sk;
  739. int ret;
  740. #ifdef CONFIG_IP_PIMSM
  741. if (assert == IGMPMSG_WHOLEPKT)
  742. skb = skb_realloc_headroom(pkt, sizeof(struct iphdr));
  743. else
  744. #endif
  745. skb = alloc_skb(128, GFP_ATOMIC);
  746. if (!skb)
  747. return -ENOBUFS;
  748. #ifdef CONFIG_IP_PIMSM
  749. if (assert == IGMPMSG_WHOLEPKT) {
  750. /* Ugly, but we have no choice with this interface.
  751. * Duplicate old header, fix ihl, length etc.
  752. * And all this only to mangle msg->im_msgtype and
  753. * to set msg->im_mbz to "mbz" :-)
  754. */
  755. skb_push(skb, sizeof(struct iphdr));
  756. skb_reset_network_header(skb);
  757. skb_reset_transport_header(skb);
  758. msg = (struct igmpmsg *)skb_network_header(skb);
  759. memcpy(msg, skb_network_header(pkt), sizeof(struct iphdr));
  760. msg->im_msgtype = IGMPMSG_WHOLEPKT;
  761. msg->im_mbz = 0;
  762. msg->im_vif = mrt->mroute_reg_vif_num;
  763. ip_hdr(skb)->ihl = sizeof(struct iphdr) >> 2;
  764. ip_hdr(skb)->tot_len = htons(ntohs(ip_hdr(pkt)->tot_len) +
  765. sizeof(struct iphdr));
  766. } else
  767. #endif
  768. {
  769. /* Copy the IP header */
  770. skb->network_header = skb->tail;
  771. skb_put(skb, ihl);
  772. skb_copy_to_linear_data(skb, pkt->data, ihl);
  773. ip_hdr(skb)->protocol = 0; /* Flag to the kernel this is a route add */
  774. msg = (struct igmpmsg *)skb_network_header(skb);
  775. msg->im_vif = vifi;
  776. skb_dst_set(skb, dst_clone(skb_dst(pkt)));
  777. /* Add our header */
  778. igmp = (struct igmphdr *)skb_put(skb, sizeof(struct igmphdr));
  779. igmp->type =
  780. msg->im_msgtype = assert;
  781. igmp->code = 0;
  782. ip_hdr(skb)->tot_len = htons(skb->len); /* Fix the length */
  783. skb->transport_header = skb->network_header;
  784. }
  785. rcu_read_lock();
  786. mroute_sk = rcu_dereference(mrt->mroute_sk);
  787. if (mroute_sk == NULL) {
  788. rcu_read_unlock();
  789. kfree_skb(skb);
  790. return -EINVAL;
  791. }
  792. /* Deliver to mrouted */
  793. ret = sock_queue_rcv_skb(mroute_sk, skb);
  794. rcu_read_unlock();
  795. if (ret < 0) {
  796. if (net_ratelimit())
  797. printk(KERN_WARNING "mroute: pending queue full, dropping entries.\n");
  798. kfree_skb(skb);
  799. }
  800. return ret;
  801. }
  802. /*
  803. * Queue a packet for resolution. It gets locked cache entry!
  804. */
  805. static int
  806. ipmr_cache_unresolved(struct mr_table *mrt, vifi_t vifi, struct sk_buff *skb)
  807. {
  808. bool found = false;
  809. int err;
  810. struct mfc_cache *c;
  811. const struct iphdr *iph = ip_hdr(skb);
  812. spin_lock_bh(&mfc_unres_lock);
  813. list_for_each_entry(c, &mrt->mfc_unres_queue, list) {
  814. if (c->mfc_mcastgrp == iph->daddr &&
  815. c->mfc_origin == iph->saddr) {
  816. found = true;
  817. break;
  818. }
  819. }
  820. if (!found) {
  821. /* Create a new entry if allowable */
  822. if (atomic_read(&mrt->cache_resolve_queue_len) >= 10 ||
  823. (c = ipmr_cache_alloc_unres()) == NULL) {
  824. spin_unlock_bh(&mfc_unres_lock);
  825. kfree_skb(skb);
  826. return -ENOBUFS;
  827. }
  828. /* Fill in the new cache entry */
  829. c->mfc_parent = -1;
  830. c->mfc_origin = iph->saddr;
  831. c->mfc_mcastgrp = iph->daddr;
  832. /* Reflect first query at mrouted. */
  833. err = ipmr_cache_report(mrt, skb, vifi, IGMPMSG_NOCACHE);
  834. if (err < 0) {
  835. /* If the report failed throw the cache entry
  836. out - Brad Parker
  837. */
  838. spin_unlock_bh(&mfc_unres_lock);
  839. ipmr_cache_free(c);
  840. kfree_skb(skb);
  841. return err;
  842. }
  843. atomic_inc(&mrt->cache_resolve_queue_len);
  844. list_add(&c->list, &mrt->mfc_unres_queue);
  845. if (atomic_read(&mrt->cache_resolve_queue_len) == 1)
  846. mod_timer(&mrt->ipmr_expire_timer, c->mfc_un.unres.expires);
  847. }
  848. /* See if we can append the packet */
  849. if (c->mfc_un.unres.unresolved.qlen > 3) {
  850. kfree_skb(skb);
  851. err = -ENOBUFS;
  852. } else {
  853. skb_queue_tail(&c->mfc_un.unres.unresolved, skb);
  854. err = 0;
  855. }
  856. spin_unlock_bh(&mfc_unres_lock);
  857. return err;
  858. }
  859. /*
  860. * MFC cache manipulation by user space mroute daemon
  861. */
  862. static int ipmr_mfc_delete(struct mr_table *mrt, struct mfcctl *mfc)
  863. {
  864. int line;
  865. struct mfc_cache *c, *next;
  866. line = MFC_HASH(mfc->mfcc_mcastgrp.s_addr, mfc->mfcc_origin.s_addr);
  867. list_for_each_entry_safe(c, next, &mrt->mfc_cache_array[line], list) {
  868. if (c->mfc_origin == mfc->mfcc_origin.s_addr &&
  869. c->mfc_mcastgrp == mfc->mfcc_mcastgrp.s_addr) {
  870. list_del_rcu(&c->list);
  871. ipmr_cache_free(c);
  872. return 0;
  873. }
  874. }
  875. return -ENOENT;
  876. }
  877. static int ipmr_mfc_add(struct net *net, struct mr_table *mrt,
  878. struct mfcctl *mfc, int mrtsock)
  879. {
  880. bool found = false;
  881. int line;
  882. struct mfc_cache *uc, *c;
  883. if (mfc->mfcc_parent >= MAXVIFS)
  884. return -ENFILE;
  885. line = MFC_HASH(mfc->mfcc_mcastgrp.s_addr, mfc->mfcc_origin.s_addr);
  886. list_for_each_entry(c, &mrt->mfc_cache_array[line], list) {
  887. if (c->mfc_origin == mfc->mfcc_origin.s_addr &&
  888. c->mfc_mcastgrp == mfc->mfcc_mcastgrp.s_addr) {
  889. found = true;
  890. break;
  891. }
  892. }
  893. if (found) {
  894. write_lock_bh(&mrt_lock);
  895. c->mfc_parent = mfc->mfcc_parent;
  896. ipmr_update_thresholds(mrt, c, mfc->mfcc_ttls);
  897. if (!mrtsock)
  898. c->mfc_flags |= MFC_STATIC;
  899. write_unlock_bh(&mrt_lock);
  900. return 0;
  901. }
  902. if (!ipv4_is_multicast(mfc->mfcc_mcastgrp.s_addr))
  903. return -EINVAL;
  904. c = ipmr_cache_alloc();
  905. if (c == NULL)
  906. return -ENOMEM;
  907. c->mfc_origin = mfc->mfcc_origin.s_addr;
  908. c->mfc_mcastgrp = mfc->mfcc_mcastgrp.s_addr;
  909. c->mfc_parent = mfc->mfcc_parent;
  910. ipmr_update_thresholds(mrt, c, mfc->mfcc_ttls);
  911. if (!mrtsock)
  912. c->mfc_flags |= MFC_STATIC;
  913. list_add_rcu(&c->list, &mrt->mfc_cache_array[line]);
  914. /*
  915. * Check to see if we resolved a queued list. If so we
  916. * need to send on the frames and tidy up.
  917. */
  918. found = false;
  919. spin_lock_bh(&mfc_unres_lock);
  920. list_for_each_entry(uc, &mrt->mfc_unres_queue, list) {
  921. if (uc->mfc_origin == c->mfc_origin &&
  922. uc->mfc_mcastgrp == c->mfc_mcastgrp) {
  923. list_del(&uc->list);
  924. atomic_dec(&mrt->cache_resolve_queue_len);
  925. found = true;
  926. break;
  927. }
  928. }
  929. if (list_empty(&mrt->mfc_unres_queue))
  930. del_timer(&mrt->ipmr_expire_timer);
  931. spin_unlock_bh(&mfc_unres_lock);
  932. if (found) {
  933. ipmr_cache_resolve(net, mrt, uc, c);
  934. ipmr_cache_free(uc);
  935. }
  936. return 0;
  937. }
  938. /*
  939. * Close the multicast socket, and clear the vif tables etc
  940. */
  941. static void mroute_clean_tables(struct mr_table *mrt)
  942. {
  943. int i;
  944. LIST_HEAD(list);
  945. struct mfc_cache *c, *next;
  946. /* Shut down all active vif entries */
  947. for (i = 0; i < mrt->maxvif; i++) {
  948. if (!(mrt->vif_table[i].flags & VIFF_STATIC))
  949. vif_delete(mrt, i, 0, &list);
  950. }
  951. unregister_netdevice_many(&list);
  952. /* Wipe the cache */
  953. for (i = 0; i < MFC_LINES; i++) {
  954. list_for_each_entry_safe(c, next, &mrt->mfc_cache_array[i], list) {
  955. if (c->mfc_flags & MFC_STATIC)
  956. continue;
  957. list_del_rcu(&c->list);
  958. ipmr_cache_free(c);
  959. }
  960. }
  961. if (atomic_read(&mrt->cache_resolve_queue_len) != 0) {
  962. spin_lock_bh(&mfc_unres_lock);
  963. list_for_each_entry_safe(c, next, &mrt->mfc_unres_queue, list) {
  964. list_del(&c->list);
  965. ipmr_destroy_unres(mrt, c);
  966. }
  967. spin_unlock_bh(&mfc_unres_lock);
  968. }
  969. }
  970. /* called from ip_ra_control(), before an RCU grace period,
  971. * we dont need to call synchronize_rcu() here
  972. */
  973. static void mrtsock_destruct(struct sock *sk)
  974. {
  975. struct net *net = sock_net(sk);
  976. struct mr_table *mrt;
  977. rtnl_lock();
  978. ipmr_for_each_table(mrt, net) {
  979. if (sk == rtnl_dereference(mrt->mroute_sk)) {
  980. IPV4_DEVCONF_ALL(net, MC_FORWARDING)--;
  981. rcu_assign_pointer(mrt->mroute_sk, NULL);
  982. mroute_clean_tables(mrt);
  983. }
  984. }
  985. rtnl_unlock();
  986. }
  987. /*
  988. * Socket options and virtual interface manipulation. The whole
  989. * virtual interface system is a complete heap, but unfortunately
  990. * that's how BSD mrouted happens to think. Maybe one day with a proper
  991. * MOSPF/PIM router set up we can clean this up.
  992. */
  993. int ip_mroute_setsockopt(struct sock *sk, int optname, char __user *optval, unsigned int optlen)
  994. {
  995. int ret;
  996. struct vifctl vif;
  997. struct mfcctl mfc;
  998. struct net *net = sock_net(sk);
  999. struct mr_table *mrt;
  1000. mrt = ipmr_get_table(net, raw_sk(sk)->ipmr_table ? : RT_TABLE_DEFAULT);
  1001. if (mrt == NULL)
  1002. return -ENOENT;
  1003. if (optname != MRT_INIT) {
  1004. if (sk != rcu_dereference_raw(mrt->mroute_sk) &&
  1005. !capable(CAP_NET_ADMIN))
  1006. return -EACCES;
  1007. }
  1008. switch (optname) {
  1009. case MRT_INIT:
  1010. if (sk->sk_type != SOCK_RAW ||
  1011. inet_sk(sk)->inet_num != IPPROTO_IGMP)
  1012. return -EOPNOTSUPP;
  1013. if (optlen != sizeof(int))
  1014. return -ENOPROTOOPT;
  1015. rtnl_lock();
  1016. if (rtnl_dereference(mrt->mroute_sk)) {
  1017. rtnl_unlock();
  1018. return -EADDRINUSE;
  1019. }
  1020. ret = ip_ra_control(sk, 1, mrtsock_destruct);
  1021. if (ret == 0) {
  1022. rcu_assign_pointer(mrt->mroute_sk, sk);
  1023. IPV4_DEVCONF_ALL(net, MC_FORWARDING)++;
  1024. }
  1025. rtnl_unlock();
  1026. return ret;
  1027. case MRT_DONE:
  1028. if (sk != rcu_dereference_raw(mrt->mroute_sk))
  1029. return -EACCES;
  1030. return ip_ra_control(sk, 0, NULL);
  1031. case MRT_ADD_VIF:
  1032. case MRT_DEL_VIF:
  1033. if (optlen != sizeof(vif))
  1034. return -EINVAL;
  1035. if (copy_from_user(&vif, optval, sizeof(vif)))
  1036. return -EFAULT;
  1037. if (vif.vifc_vifi >= MAXVIFS)
  1038. return -ENFILE;
  1039. rtnl_lock();
  1040. if (optname == MRT_ADD_VIF) {
  1041. ret = vif_add(net, mrt, &vif,
  1042. sk == rtnl_dereference(mrt->mroute_sk));
  1043. } else {
  1044. ret = vif_delete(mrt, vif.vifc_vifi, 0, NULL);
  1045. }
  1046. rtnl_unlock();
  1047. return ret;
  1048. /*
  1049. * Manipulate the forwarding caches. These live
  1050. * in a sort of kernel/user symbiosis.
  1051. */
  1052. case MRT_ADD_MFC:
  1053. case MRT_DEL_MFC:
  1054. if (optlen != sizeof(mfc))
  1055. return -EINVAL;
  1056. if (copy_from_user(&mfc, optval, sizeof(mfc)))
  1057. return -EFAULT;
  1058. rtnl_lock();
  1059. if (optname == MRT_DEL_MFC)
  1060. ret = ipmr_mfc_delete(mrt, &mfc);
  1061. else
  1062. ret = ipmr_mfc_add(net, mrt, &mfc,
  1063. sk == rtnl_dereference(mrt->mroute_sk));
  1064. rtnl_unlock();
  1065. return ret;
  1066. /*
  1067. * Control PIM assert.
  1068. */
  1069. case MRT_ASSERT:
  1070. {
  1071. int v;
  1072. if (get_user(v, (int __user *)optval))
  1073. return -EFAULT;
  1074. mrt->mroute_do_assert = (v) ? 1 : 0;
  1075. return 0;
  1076. }
  1077. #ifdef CONFIG_IP_PIMSM
  1078. case MRT_PIM:
  1079. {
  1080. int v;
  1081. if (get_user(v, (int __user *)optval))
  1082. return -EFAULT;
  1083. v = (v) ? 1 : 0;
  1084. rtnl_lock();
  1085. ret = 0;
  1086. if (v != mrt->mroute_do_pim) {
  1087. mrt->mroute_do_pim = v;
  1088. mrt->mroute_do_assert = v;
  1089. }
  1090. rtnl_unlock();
  1091. return ret;
  1092. }
  1093. #endif
  1094. #ifdef CONFIG_IP_MROUTE_MULTIPLE_TABLES
  1095. case MRT_TABLE:
  1096. {
  1097. u32 v;
  1098. if (optlen != sizeof(u32))
  1099. return -EINVAL;
  1100. if (get_user(v, (u32 __user *)optval))
  1101. return -EFAULT;
  1102. rtnl_lock();
  1103. ret = 0;
  1104. if (sk == rtnl_dereference(mrt->mroute_sk)) {
  1105. ret = -EBUSY;
  1106. } else {
  1107. if (!ipmr_new_table(net, v))
  1108. ret = -ENOMEM;
  1109. raw_sk(sk)->ipmr_table = v;
  1110. }
  1111. rtnl_unlock();
  1112. return ret;
  1113. }
  1114. #endif
  1115. /*
  1116. * Spurious command, or MRT_VERSION which you cannot
  1117. * set.
  1118. */
  1119. default:
  1120. return -ENOPROTOOPT;
  1121. }
  1122. }
  1123. /*
  1124. * Getsock opt support for the multicast routing system.
  1125. */
  1126. int ip_mroute_getsockopt(struct sock *sk, int optname, char __user *optval, int __user *optlen)
  1127. {
  1128. int olr;
  1129. int val;
  1130. struct net *net = sock_net(sk);
  1131. struct mr_table *mrt;
  1132. mrt = ipmr_get_table(net, raw_sk(sk)->ipmr_table ? : RT_TABLE_DEFAULT);
  1133. if (mrt == NULL)
  1134. return -ENOENT;
  1135. if (optname != MRT_VERSION &&
  1136. #ifdef CONFIG_IP_PIMSM
  1137. optname != MRT_PIM &&
  1138. #endif
  1139. optname != MRT_ASSERT)
  1140. return -ENOPROTOOPT;
  1141. if (get_user(olr, optlen))
  1142. return -EFAULT;
  1143. olr = min_t(unsigned int, olr, sizeof(int));
  1144. if (olr < 0)
  1145. return -EINVAL;
  1146. if (put_user(olr, optlen))
  1147. return -EFAULT;
  1148. if (optname == MRT_VERSION)
  1149. val = 0x0305;
  1150. #ifdef CONFIG_IP_PIMSM
  1151. else if (optname == MRT_PIM)
  1152. val = mrt->mroute_do_pim;
  1153. #endif
  1154. else
  1155. val = mrt->mroute_do_assert;
  1156. if (copy_to_user(optval, &val, olr))
  1157. return -EFAULT;
  1158. return 0;
  1159. }
  1160. /*
  1161. * The IP multicast ioctl support routines.
  1162. */
  1163. int ipmr_ioctl(struct sock *sk, int cmd, void __user *arg)
  1164. {
  1165. struct sioc_sg_req sr;
  1166. struct sioc_vif_req vr;
  1167. struct vif_device *vif;
  1168. struct mfc_cache *c;
  1169. struct net *net = sock_net(sk);
  1170. struct mr_table *mrt;
  1171. mrt = ipmr_get_table(net, raw_sk(sk)->ipmr_table ? : RT_TABLE_DEFAULT);
  1172. if (mrt == NULL)
  1173. return -ENOENT;
  1174. switch (cmd) {
  1175. case SIOCGETVIFCNT:
  1176. if (copy_from_user(&vr, arg, sizeof(vr)))
  1177. return -EFAULT;
  1178. if (vr.vifi >= mrt->maxvif)
  1179. return -EINVAL;
  1180. read_lock(&mrt_lock);
  1181. vif = &mrt->vif_table[vr.vifi];
  1182. if (VIF_EXISTS(mrt, vr.vifi)) {
  1183. vr.icount = vif->pkt_in;
  1184. vr.ocount = vif->pkt_out;
  1185. vr.ibytes = vif->bytes_in;
  1186. vr.obytes = vif->bytes_out;
  1187. read_unlock(&mrt_lock);
  1188. if (copy_to_user(arg, &vr, sizeof(vr)))
  1189. return -EFAULT;
  1190. return 0;
  1191. }
  1192. read_unlock(&mrt_lock);
  1193. return -EADDRNOTAVAIL;
  1194. case SIOCGETSGCNT:
  1195. if (copy_from_user(&sr, arg, sizeof(sr)))
  1196. return -EFAULT;
  1197. rcu_read_lock();
  1198. c = ipmr_cache_find(mrt, sr.src.s_addr, sr.grp.s_addr);
  1199. if (c) {
  1200. sr.pktcnt = c->mfc_un.res.pkt;
  1201. sr.bytecnt = c->mfc_un.res.bytes;
  1202. sr.wrong_if = c->mfc_un.res.wrong_if;
  1203. rcu_read_unlock();
  1204. if (copy_to_user(arg, &sr, sizeof(sr)))
  1205. return -EFAULT;
  1206. return 0;
  1207. }
  1208. rcu_read_unlock();
  1209. return -EADDRNOTAVAIL;
  1210. default:
  1211. return -ENOIOCTLCMD;
  1212. }
  1213. }
  1214. static int ipmr_device_event(struct notifier_block *this, unsigned long event, void *ptr)
  1215. {
  1216. struct net_device *dev = ptr;
  1217. struct net *net = dev_net(dev);
  1218. struct mr_table *mrt;
  1219. struct vif_device *v;
  1220. int ct;
  1221. LIST_HEAD(list);
  1222. if (event != NETDEV_UNREGISTER)
  1223. return NOTIFY_DONE;
  1224. ipmr_for_each_table(mrt, net) {
  1225. v = &mrt->vif_table[0];
  1226. for (ct = 0; ct < mrt->maxvif; ct++, v++) {
  1227. if (v->dev == dev)
  1228. vif_delete(mrt, ct, 1, &list);
  1229. }
  1230. }
  1231. unregister_netdevice_many(&list);
  1232. return NOTIFY_DONE;
  1233. }
  1234. static struct notifier_block ip_mr_notifier = {
  1235. .notifier_call = ipmr_device_event,
  1236. };
  1237. /*
  1238. * Encapsulate a packet by attaching a valid IPIP header to it.
  1239. * This avoids tunnel drivers and other mess and gives us the speed so
  1240. * important for multicast video.
  1241. */
  1242. static void ip_encap(struct sk_buff *skb, __be32 saddr, __be32 daddr)
  1243. {
  1244. struct iphdr *iph;
  1245. struct iphdr *old_iph = ip_hdr(skb);
  1246. skb_push(skb, sizeof(struct iphdr));
  1247. skb->transport_header = skb->network_header;
  1248. skb_reset_network_header(skb);
  1249. iph = ip_hdr(skb);
  1250. iph->version = 4;
  1251. iph->tos = old_iph->tos;
  1252. iph->ttl = old_iph->ttl;
  1253. iph->frag_off = 0;
  1254. iph->daddr = daddr;
  1255. iph->saddr = saddr;
  1256. iph->protocol = IPPROTO_IPIP;
  1257. iph->ihl = 5;
  1258. iph->tot_len = htons(skb->len);
  1259. ip_select_ident(iph, skb_dst(skb), NULL);
  1260. ip_send_check(iph);
  1261. memset(&(IPCB(skb)->opt), 0, sizeof(IPCB(skb)->opt));
  1262. nf_reset(skb);
  1263. }
  1264. static inline int ipmr_forward_finish(struct sk_buff *skb)
  1265. {
  1266. struct ip_options *opt = &(IPCB(skb)->opt);
  1267. IP_INC_STATS_BH(dev_net(skb_dst(skb)->dev), IPSTATS_MIB_OUTFORWDATAGRAMS);
  1268. if (unlikely(opt->optlen))
  1269. ip_forward_options(skb);
  1270. return dst_output(skb);
  1271. }
  1272. /*
  1273. * Processing handlers for ipmr_forward
  1274. */
  1275. static void ipmr_queue_xmit(struct net *net, struct mr_table *mrt,
  1276. struct sk_buff *skb, struct mfc_cache *c, int vifi)
  1277. {
  1278. const struct iphdr *iph = ip_hdr(skb);
  1279. struct vif_device *vif = &mrt->vif_table[vifi];
  1280. struct net_device *dev;
  1281. struct rtable *rt;
  1282. int encap = 0;
  1283. if (vif->dev == NULL)
  1284. goto out_free;
  1285. #ifdef CONFIG_IP_PIMSM
  1286. if (vif->flags & VIFF_REGISTER) {
  1287. vif->pkt_out++;
  1288. vif->bytes_out += skb->len;
  1289. vif->dev->stats.tx_bytes += skb->len;
  1290. vif->dev->stats.tx_packets++;
  1291. ipmr_cache_report(mrt, skb, vifi, IGMPMSG_WHOLEPKT);
  1292. goto out_free;
  1293. }
  1294. #endif
  1295. if (vif->flags & VIFF_TUNNEL) {
  1296. struct flowi fl = {
  1297. .oif = vif->link,
  1298. .nl_u = {
  1299. .ip4_u = {
  1300. .daddr = vif->remote,
  1301. .saddr = vif->local,
  1302. .tos = RT_TOS(iph->tos)
  1303. }
  1304. },
  1305. .proto = IPPROTO_IPIP
  1306. };
  1307. if (ip_route_output_key(net, &rt, &fl))
  1308. goto out_free;
  1309. encap = sizeof(struct iphdr);
  1310. } else {
  1311. struct flowi fl = {
  1312. .oif = vif->link,
  1313. .nl_u = {
  1314. .ip4_u = {
  1315. .daddr = iph->daddr,
  1316. .tos = RT_TOS(iph->tos)
  1317. }
  1318. },
  1319. .proto = IPPROTO_IPIP
  1320. };
  1321. if (ip_route_output_key(net, &rt, &fl))
  1322. goto out_free;
  1323. }
  1324. dev = rt->dst.dev;
  1325. if (skb->len+encap > dst_mtu(&rt->dst) && (ntohs(iph->frag_off) & IP_DF)) {
  1326. /* Do not fragment multicasts. Alas, IPv4 does not
  1327. * allow to send ICMP, so that packets will disappear
  1328. * to blackhole.
  1329. */
  1330. IP_INC_STATS_BH(dev_net(dev), IPSTATS_MIB_FRAGFAILS);
  1331. ip_rt_put(rt);
  1332. goto out_free;
  1333. }
  1334. encap += LL_RESERVED_SPACE(dev) + rt->dst.header_len;
  1335. if (skb_cow(skb, encap)) {
  1336. ip_rt_put(rt);
  1337. goto out_free;
  1338. }
  1339. vif->pkt_out++;
  1340. vif->bytes_out += skb->len;
  1341. skb_dst_drop(skb);
  1342. skb_dst_set(skb, &rt->dst);
  1343. ip_decrease_ttl(ip_hdr(skb));
  1344. /* FIXME: forward and output firewalls used to be called here.
  1345. * What do we do with netfilter? -- RR
  1346. */
  1347. if (vif->flags & VIFF_TUNNEL) {
  1348. ip_encap(skb, vif->local, vif->remote);
  1349. /* FIXME: extra output firewall step used to be here. --RR */
  1350. vif->dev->stats.tx_packets++;
  1351. vif->dev->stats.tx_bytes += skb->len;
  1352. }
  1353. IPCB(skb)->flags |= IPSKB_FORWARDED;
  1354. /*
  1355. * RFC1584 teaches, that DVMRP/PIM router must deliver packets locally
  1356. * not only before forwarding, but after forwarding on all output
  1357. * interfaces. It is clear, if mrouter runs a multicasting
  1358. * program, it should receive packets not depending to what interface
  1359. * program is joined.
  1360. * If we will not make it, the program will have to join on all
  1361. * interfaces. On the other hand, multihoming host (or router, but
  1362. * not mrouter) cannot join to more than one interface - it will
  1363. * result in receiving multiple packets.
  1364. */
  1365. NF_HOOK(NFPROTO_IPV4, NF_INET_FORWARD, skb, skb->dev, dev,
  1366. ipmr_forward_finish);
  1367. return;
  1368. out_free:
  1369. kfree_skb(skb);
  1370. }
  1371. static int ipmr_find_vif(struct mr_table *mrt, struct net_device *dev)
  1372. {
  1373. int ct;
  1374. for (ct = mrt->maxvif-1; ct >= 0; ct--) {
  1375. if (mrt->vif_table[ct].dev == dev)
  1376. break;
  1377. }
  1378. return ct;
  1379. }
  1380. /* "local" means that we should preserve one skb (for local delivery) */
  1381. static int ip_mr_forward(struct net *net, struct mr_table *mrt,
  1382. struct sk_buff *skb, struct mfc_cache *cache,
  1383. int local)
  1384. {
  1385. int psend = -1;
  1386. int vif, ct;
  1387. vif = cache->mfc_parent;
  1388. cache->mfc_un.res.pkt++;
  1389. cache->mfc_un.res.bytes += skb->len;
  1390. /*
  1391. * Wrong interface: drop packet and (maybe) send PIM assert.
  1392. */
  1393. if (mrt->vif_table[vif].dev != skb->dev) {
  1394. int true_vifi;
  1395. if (skb_rtable(skb)->fl.iif == 0) {
  1396. /* It is our own packet, looped back.
  1397. * Very complicated situation...
  1398. *
  1399. * The best workaround until routing daemons will be
  1400. * fixed is not to redistribute packet, if it was
  1401. * send through wrong interface. It means, that
  1402. * multicast applications WILL NOT work for
  1403. * (S,G), which have default multicast route pointing
  1404. * to wrong oif. In any case, it is not a good
  1405. * idea to use multicasting applications on router.
  1406. */
  1407. goto dont_forward;
  1408. }
  1409. cache->mfc_un.res.wrong_if++;
  1410. true_vifi = ipmr_find_vif(mrt, skb->dev);
  1411. if (true_vifi >= 0 && mrt->mroute_do_assert &&
  1412. /* pimsm uses asserts, when switching from RPT to SPT,
  1413. * so that we cannot check that packet arrived on an oif.
  1414. * It is bad, but otherwise we would need to move pretty
  1415. * large chunk of pimd to kernel. Ough... --ANK
  1416. */
  1417. (mrt->mroute_do_pim ||
  1418. cache->mfc_un.res.ttls[true_vifi] < 255) &&
  1419. time_after(jiffies,
  1420. cache->mfc_un.res.last_assert + MFC_ASSERT_THRESH)) {
  1421. cache->mfc_un.res.last_assert = jiffies;
  1422. ipmr_cache_report(mrt, skb, true_vifi, IGMPMSG_WRONGVIF);
  1423. }
  1424. goto dont_forward;
  1425. }
  1426. mrt->vif_table[vif].pkt_in++;
  1427. mrt->vif_table[vif].bytes_in += skb->len;
  1428. /*
  1429. * Forward the frame
  1430. */
  1431. for (ct = cache->mfc_un.res.maxvif - 1;
  1432. ct >= cache->mfc_un.res.minvif; ct--) {
  1433. if (ip_hdr(skb)->ttl > cache->mfc_un.res.ttls[ct]) {
  1434. if (psend != -1) {
  1435. struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
  1436. if (skb2)
  1437. ipmr_queue_xmit(net, mrt, skb2, cache,
  1438. psend);
  1439. }
  1440. psend = ct;
  1441. }
  1442. }
  1443. if (psend != -1) {
  1444. if (local) {
  1445. struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
  1446. if (skb2)
  1447. ipmr_queue_xmit(net, mrt, skb2, cache, psend);
  1448. } else {
  1449. ipmr_queue_xmit(net, mrt, skb, cache, psend);
  1450. return 0;
  1451. }
  1452. }
  1453. dont_forward:
  1454. if (!local)
  1455. kfree_skb(skb);
  1456. return 0;
  1457. }
  1458. /*
  1459. * Multicast packets for forwarding arrive here
  1460. * Called with rcu_read_lock();
  1461. */
  1462. int ip_mr_input(struct sk_buff *skb)
  1463. {
  1464. struct mfc_cache *cache;
  1465. struct net *net = dev_net(skb->dev);
  1466. int local = skb_rtable(skb)->rt_flags & RTCF_LOCAL;
  1467. struct mr_table *mrt;
  1468. int err;
  1469. /* Packet is looped back after forward, it should not be
  1470. * forwarded second time, but still can be delivered locally.
  1471. */
  1472. if (IPCB(skb)->flags & IPSKB_FORWARDED)
  1473. goto dont_forward;
  1474. err = ipmr_fib_lookup(net, &skb_rtable(skb)->fl, &mrt);
  1475. if (err < 0) {
  1476. kfree_skb(skb);
  1477. return err;
  1478. }
  1479. if (!local) {
  1480. if (IPCB(skb)->opt.router_alert) {
  1481. if (ip_call_ra_chain(skb))
  1482. return 0;
  1483. } else if (ip_hdr(skb)->protocol == IPPROTO_IGMP) {
  1484. /* IGMPv1 (and broken IGMPv2 implementations sort of
  1485. * Cisco IOS <= 11.2(8)) do not put router alert
  1486. * option to IGMP packets destined to routable
  1487. * groups. It is very bad, because it means
  1488. * that we can forward NO IGMP messages.
  1489. */
  1490. struct sock *mroute_sk;
  1491. mroute_sk = rcu_dereference(mrt->mroute_sk);
  1492. if (mroute_sk) {
  1493. nf_reset(skb);
  1494. raw_rcv(mroute_sk, skb);
  1495. return 0;
  1496. }
  1497. }
  1498. }
  1499. /* already under rcu_read_lock() */
  1500. cache = ipmr_cache_find(mrt, ip_hdr(skb)->saddr, ip_hdr(skb)->daddr);
  1501. /*
  1502. * No usable cache entry
  1503. */
  1504. if (cache == NULL) {
  1505. int vif;
  1506. if (local) {
  1507. struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
  1508. ip_local_deliver(skb);
  1509. if (skb2 == NULL)
  1510. return -ENOBUFS;
  1511. skb = skb2;
  1512. }
  1513. read_lock(&mrt_lock);
  1514. vif = ipmr_find_vif(mrt, skb->dev);
  1515. if (vif >= 0) {
  1516. int err2 = ipmr_cache_unresolved(mrt, vif, skb);
  1517. read_unlock(&mrt_lock);
  1518. return err2;
  1519. }
  1520. read_unlock(&mrt_lock);
  1521. kfree_skb(skb);
  1522. return -ENODEV;
  1523. }
  1524. read_lock(&mrt_lock);
  1525. ip_mr_forward(net, mrt, skb, cache, local);
  1526. read_unlock(&mrt_lock);
  1527. if (local)
  1528. return ip_local_deliver(skb);
  1529. return 0;
  1530. dont_forward:
  1531. if (local)
  1532. return ip_local_deliver(skb);
  1533. kfree_skb(skb);
  1534. return 0;
  1535. }
  1536. #ifdef CONFIG_IP_PIMSM
  1537. /* called with rcu_read_lock() */
  1538. static int __pim_rcv(struct mr_table *mrt, struct sk_buff *skb,
  1539. unsigned int pimlen)
  1540. {
  1541. struct net_device *reg_dev = NULL;
  1542. struct iphdr *encap;
  1543. encap = (struct iphdr *)(skb_transport_header(skb) + pimlen);
  1544. /*
  1545. * Check that:
  1546. * a. packet is really sent to a multicast group
  1547. * b. packet is not a NULL-REGISTER
  1548. * c. packet is not truncated
  1549. */
  1550. if (!ipv4_is_multicast(encap->daddr) ||
  1551. encap->tot_len == 0 ||
  1552. ntohs(encap->tot_len) + pimlen > skb->len)
  1553. return 1;
  1554. read_lock(&mrt_lock);
  1555. if (mrt->mroute_reg_vif_num >= 0)
  1556. reg_dev = mrt->vif_table[mrt->mroute_reg_vif_num].dev;
  1557. read_unlock(&mrt_lock);
  1558. if (reg_dev == NULL)
  1559. return 1;
  1560. skb->mac_header = skb->network_header;
  1561. skb_pull(skb, (u8 *)encap - skb->data);
  1562. skb_reset_network_header(skb);
  1563. skb->protocol = htons(ETH_P_IP);
  1564. skb->ip_summed = CHECKSUM_NONE;
  1565. skb->pkt_type = PACKET_HOST;
  1566. skb_tunnel_rx(skb, reg_dev);
  1567. netif_rx(skb);
  1568. return NET_RX_SUCCESS;
  1569. }
  1570. #endif
  1571. #ifdef CONFIG_IP_PIMSM_V1
  1572. /*
  1573. * Handle IGMP messages of PIMv1
  1574. */
  1575. int pim_rcv_v1(struct sk_buff *skb)
  1576. {
  1577. struct igmphdr *pim;
  1578. struct net *net = dev_net(skb->dev);
  1579. struct mr_table *mrt;
  1580. if (!pskb_may_pull(skb, sizeof(*pim) + sizeof(struct iphdr)))
  1581. goto drop;
  1582. pim = igmp_hdr(skb);
  1583. if (ipmr_fib_lookup(net, &skb_rtable(skb)->fl, &mrt) < 0)
  1584. goto drop;
  1585. if (!mrt->mroute_do_pim ||
  1586. pim->group != PIM_V1_VERSION || pim->code != PIM_V1_REGISTER)
  1587. goto drop;
  1588. if (__pim_rcv(mrt, skb, sizeof(*pim))) {
  1589. drop:
  1590. kfree_skb(skb);
  1591. }
  1592. return 0;
  1593. }
  1594. #endif
  1595. #ifdef CONFIG_IP_PIMSM_V2
  1596. static int pim_rcv(struct sk_buff *skb)
  1597. {
  1598. struct pimreghdr *pim;
  1599. struct net *net = dev_net(skb->dev);
  1600. struct mr_table *mrt;
  1601. if (!pskb_may_pull(skb, sizeof(*pim) + sizeof(struct iphdr)))
  1602. goto drop;
  1603. pim = (struct pimreghdr *)skb_transport_header(skb);
  1604. if (pim->type != ((PIM_VERSION << 4) | (PIM_REGISTER)) ||
  1605. (pim->flags & PIM_NULL_REGISTER) ||
  1606. (ip_compute_csum((void *)pim, sizeof(*pim)) != 0 &&
  1607. csum_fold(skb_checksum(skb, 0, skb->len, 0))))
  1608. goto drop;
  1609. if (ipmr_fib_lookup(net, &skb_rtable(skb)->fl, &mrt) < 0)
  1610. goto drop;
  1611. if (__pim_rcv(mrt, skb, sizeof(*pim))) {
  1612. drop:
  1613. kfree_skb(skb);
  1614. }
  1615. return 0;
  1616. }
  1617. #endif
  1618. static int __ipmr_fill_mroute(struct mr_table *mrt, struct sk_buff *skb,
  1619. struct mfc_cache *c, struct rtmsg *rtm)
  1620. {
  1621. int ct;
  1622. struct rtnexthop *nhp;
  1623. u8 *b = skb_tail_pointer(skb);
  1624. struct rtattr *mp_head;
  1625. /* If cache is unresolved, don't try to parse IIF and OIF */
  1626. if (c->mfc_parent >= MAXVIFS)
  1627. return -ENOENT;
  1628. if (VIF_EXISTS(mrt, c->mfc_parent))
  1629. RTA_PUT(skb, RTA_IIF, 4, &mrt->vif_table[c->mfc_parent].dev->ifindex);
  1630. mp_head = (struct rtattr *)skb_put(skb, RTA_LENGTH(0));
  1631. for (ct = c->mfc_un.res.minvif; ct < c->mfc_un.res.maxvif; ct++) {
  1632. if (VIF_EXISTS(mrt, ct) && c->mfc_un.res.ttls[ct] < 255) {
  1633. if (skb_tailroom(skb) < RTA_ALIGN(RTA_ALIGN(sizeof(*nhp)) + 4))
  1634. goto rtattr_failure;
  1635. nhp = (struct rtnexthop *)skb_put(skb, RTA_ALIGN(sizeof(*nhp)));
  1636. nhp->rtnh_flags = 0;
  1637. nhp->rtnh_hops = c->mfc_un.res.ttls[ct];
  1638. nhp->rtnh_ifindex = mrt->vif_table[ct].dev->ifindex;
  1639. nhp->rtnh_len = sizeof(*nhp);
  1640. }
  1641. }
  1642. mp_head->rta_type = RTA_MULTIPATH;
  1643. mp_head->rta_len = skb_tail_pointer(skb) - (u8 *)mp_head;
  1644. rtm->rtm_type = RTN_MULTICAST;
  1645. return 1;
  1646. rtattr_failure:
  1647. nlmsg_trim(skb, b);
  1648. return -EMSGSIZE;
  1649. }
  1650. int ipmr_get_route(struct net *net,
  1651. struct sk_buff *skb, struct rtmsg *rtm, int nowait)
  1652. {
  1653. int err;
  1654. struct mr_table *mrt;
  1655. struct mfc_cache *cache;
  1656. struct rtable *rt = skb_rtable(skb);
  1657. mrt = ipmr_get_table(net, RT_TABLE_DEFAULT);
  1658. if (mrt == NULL)
  1659. return -ENOENT;
  1660. rcu_read_lock();
  1661. cache = ipmr_cache_find(mrt, rt->rt_src, rt->rt_dst);
  1662. if (cache == NULL) {
  1663. struct sk_buff *skb2;
  1664. struct iphdr *iph;
  1665. struct net_device *dev;
  1666. int vif = -1;
  1667. if (nowait) {
  1668. rcu_read_unlock();
  1669. return -EAGAIN;
  1670. }
  1671. dev = skb->dev;
  1672. read_lock(&mrt_lock);
  1673. if (dev)
  1674. vif = ipmr_find_vif(mrt, dev);
  1675. if (vif < 0) {
  1676. read_unlock(&mrt_lock);
  1677. rcu_read_unlock();
  1678. return -ENODEV;
  1679. }
  1680. skb2 = skb_clone(skb, GFP_ATOMIC);
  1681. if (!skb2) {
  1682. read_unlock(&mrt_lock);
  1683. rcu_read_unlock();
  1684. return -ENOMEM;
  1685. }
  1686. skb_push(skb2, sizeof(struct iphdr));
  1687. skb_reset_network_header(skb2);
  1688. iph = ip_hdr(skb2);
  1689. iph->ihl = sizeof(struct iphdr) >> 2;
  1690. iph->saddr = rt->rt_src;
  1691. iph->daddr = rt->rt_dst;
  1692. iph->version = 0;
  1693. err = ipmr_cache_unresolved(mrt, vif, skb2);
  1694. read_unlock(&mrt_lock);
  1695. rcu_read_unlock();
  1696. return err;
  1697. }
  1698. read_lock(&mrt_lock);
  1699. if (!nowait && (rtm->rtm_flags & RTM_F_NOTIFY))
  1700. cache->mfc_flags |= MFC_NOTIFY;
  1701. err = __ipmr_fill_mroute(mrt, skb, cache, rtm);
  1702. read_unlock(&mrt_lock);
  1703. rcu_read_unlock();
  1704. return err;
  1705. }
  1706. static int ipmr_fill_mroute(struct mr_table *mrt, struct sk_buff *skb,
  1707. u32 pid, u32 seq, struct mfc_cache *c)
  1708. {
  1709. struct nlmsghdr *nlh;
  1710. struct rtmsg *rtm;
  1711. nlh = nlmsg_put(skb, pid, seq, RTM_NEWROUTE, sizeof(*rtm), NLM_F_MULTI);
  1712. if (nlh == NULL)
  1713. return -EMSGSIZE;
  1714. rtm = nlmsg_data(nlh);
  1715. rtm->rtm_family = RTNL_FAMILY_IPMR;
  1716. rtm->rtm_dst_len = 32;
  1717. rtm->rtm_src_len = 32;
  1718. rtm->rtm_tos = 0;
  1719. rtm->rtm_table = mrt->id;
  1720. NLA_PUT_U32(skb, RTA_TABLE, mrt->id);
  1721. rtm->rtm_type = RTN_MULTICAST;
  1722. rtm->rtm_scope = RT_SCOPE_UNIVERSE;
  1723. rtm->rtm_protocol = RTPROT_UNSPEC;
  1724. rtm->rtm_flags = 0;
  1725. NLA_PUT_BE32(skb, RTA_SRC, c->mfc_origin);
  1726. NLA_PUT_BE32(skb, RTA_DST, c->mfc_mcastgrp);
  1727. if (__ipmr_fill_mroute(mrt, skb, c, rtm) < 0)
  1728. goto nla_put_failure;
  1729. return nlmsg_end(skb, nlh);
  1730. nla_put_failure:
  1731. nlmsg_cancel(skb, nlh);
  1732. return -EMSGSIZE;
  1733. }
  1734. static int ipmr_rtm_dumproute(struct sk_buff *skb, struct netlink_callback *cb)
  1735. {
  1736. struct net *net = sock_net(skb->sk);
  1737. struct mr_table *mrt;
  1738. struct mfc_cache *mfc;
  1739. unsigned int t = 0, s_t;
  1740. unsigned int h = 0, s_h;
  1741. unsigned int e = 0, s_e;
  1742. s_t = cb->args[0];
  1743. s_h = cb->args[1];
  1744. s_e = cb->args[2];
  1745. rcu_read_lock();
  1746. ipmr_for_each_table(mrt, net) {
  1747. if (t < s_t)
  1748. goto next_table;
  1749. if (t > s_t)
  1750. s_h = 0;
  1751. for (h = s_h; h < MFC_LINES; h++) {
  1752. list_for_each_entry_rcu(mfc, &mrt->mfc_cache_array[h], list) {
  1753. if (e < s_e)
  1754. goto next_entry;
  1755. if (ipmr_fill_mroute(mrt, skb,
  1756. NETLINK_CB(cb->skb).pid,
  1757. cb->nlh->nlmsg_seq,
  1758. mfc) < 0)
  1759. goto done;
  1760. next_entry:
  1761. e++;
  1762. }
  1763. e = s_e = 0;
  1764. }
  1765. s_h = 0;
  1766. next_table:
  1767. t++;
  1768. }
  1769. done:
  1770. rcu_read_unlock();
  1771. cb->args[2] = e;
  1772. cb->args[1] = h;
  1773. cb->args[0] = t;
  1774. return skb->len;
  1775. }
  1776. #ifdef CONFIG_PROC_FS
  1777. /*
  1778. * The /proc interfaces to multicast routing :
  1779. * /proc/net/ip_mr_cache & /proc/net/ip_mr_vif
  1780. */
  1781. struct ipmr_vif_iter {
  1782. struct seq_net_private p;
  1783. struct mr_table *mrt;
  1784. int ct;
  1785. };
  1786. static struct vif_device *ipmr_vif_seq_idx(struct net *net,
  1787. struct ipmr_vif_iter *iter,
  1788. loff_t pos)
  1789. {
  1790. struct mr_table *mrt = iter->mrt;
  1791. for (iter->ct = 0; iter->ct < mrt->maxvif; ++iter->ct) {
  1792. if (!VIF_EXISTS(mrt, iter->ct))
  1793. continue;
  1794. if (pos-- == 0)
  1795. return &mrt->vif_table[iter->ct];
  1796. }
  1797. return NULL;
  1798. }
  1799. static void *ipmr_vif_seq_start(struct seq_file *seq, loff_t *pos)
  1800. __acquires(mrt_lock)
  1801. {
  1802. struct ipmr_vif_iter *iter = seq->private;
  1803. struct net *net = seq_file_net(seq);
  1804. struct mr_table *mrt;
  1805. mrt = ipmr_get_table(net, RT_TABLE_DEFAULT);
  1806. if (mrt == NULL)
  1807. return ERR_PTR(-ENOENT);
  1808. iter->mrt = mrt;
  1809. read_lock(&mrt_lock);
  1810. return *pos ? ipmr_vif_seq_idx(net, seq->private, *pos - 1)
  1811. : SEQ_START_TOKEN;
  1812. }
  1813. static void *ipmr_vif_seq_next(struct seq_file *seq, void *v, loff_t *pos)
  1814. {
  1815. struct ipmr_vif_iter *iter = seq->private;
  1816. struct net *net = seq_file_net(seq);
  1817. struct mr_table *mrt = iter->mrt;
  1818. ++*pos;
  1819. if (v == SEQ_START_TOKEN)
  1820. return ipmr_vif_seq_idx(net, iter, 0);
  1821. while (++iter->ct < mrt->maxvif) {
  1822. if (!VIF_EXISTS(mrt, iter->ct))
  1823. continue;
  1824. return &mrt->vif_table[iter->ct];
  1825. }
  1826. return NULL;
  1827. }
  1828. static void ipmr_vif_seq_stop(struct seq_file *seq, void *v)
  1829. __releases(mrt_lock)
  1830. {
  1831. read_unlock(&mrt_lock);
  1832. }
  1833. static int ipmr_vif_seq_show(struct seq_file *seq, void *v)
  1834. {
  1835. struct ipmr_vif_iter *iter = seq->private;
  1836. struct mr_table *mrt = iter->mrt;
  1837. if (v == SEQ_START_TOKEN) {
  1838. seq_puts(seq,
  1839. "Interface BytesIn PktsIn BytesOut PktsOut Flags Local Remote\n");
  1840. } else {
  1841. const struct vif_device *vif = v;
  1842. const char *name = vif->dev ? vif->dev->name : "none";
  1843. seq_printf(seq,
  1844. "%2Zd %-10s %8ld %7ld %8ld %7ld %05X %08X %08X\n",
  1845. vif - mrt->vif_table,
  1846. name, vif->bytes_in, vif->pkt_in,
  1847. vif->bytes_out, vif->pkt_out,
  1848. vif->flags, vif->local, vif->remote);
  1849. }
  1850. return 0;
  1851. }
  1852. static const struct seq_operations ipmr_vif_seq_ops = {
  1853. .start = ipmr_vif_seq_start,
  1854. .next = ipmr_vif_seq_next,
  1855. .stop = ipmr_vif_seq_stop,
  1856. .show = ipmr_vif_seq_show,
  1857. };
  1858. static int ipmr_vif_open(struct inode *inode, struct file *file)
  1859. {
  1860. return seq_open_net(inode, file, &ipmr_vif_seq_ops,
  1861. sizeof(struct ipmr_vif_iter));
  1862. }
  1863. static const struct file_operations ipmr_vif_fops = {
  1864. .owner = THIS_MODULE,
  1865. .open = ipmr_vif_open,
  1866. .read = seq_read,
  1867. .llseek = seq_lseek,
  1868. .release = seq_release_net,
  1869. };
  1870. struct ipmr_mfc_iter {
  1871. struct seq_net_private p;
  1872. struct mr_table *mrt;
  1873. struct list_head *cache;
  1874. int ct;
  1875. };
  1876. static struct mfc_cache *ipmr_mfc_seq_idx(struct net *net,
  1877. struct ipmr_mfc_iter *it, loff_t pos)
  1878. {
  1879. struct mr_table *mrt = it->mrt;
  1880. struct mfc_cache *mfc;
  1881. rcu_read_lock();
  1882. for (it->ct = 0; it->ct < MFC_LINES; it->ct++) {
  1883. it->cache = &mrt->mfc_cache_array[it->ct];
  1884. list_for_each_entry_rcu(mfc, it->cache, list)
  1885. if (pos-- == 0)
  1886. return mfc;
  1887. }
  1888. rcu_read_unlock();
  1889. spin_lock_bh(&mfc_unres_lock);
  1890. it->cache = &mrt->mfc_unres_queue;
  1891. list_for_each_entry(mfc, it->cache, list)
  1892. if (pos-- == 0)
  1893. return mfc;
  1894. spin_unlock_bh(&mfc_unres_lock);
  1895. it->cache = NULL;
  1896. return NULL;
  1897. }
  1898. static void *ipmr_mfc_seq_start(struct seq_file *seq, loff_t *pos)
  1899. {
  1900. struct ipmr_mfc_iter *it = seq->private;
  1901. struct net *net = seq_file_net(seq);
  1902. struct mr_table *mrt;
  1903. mrt = ipmr_get_table(net, RT_TABLE_DEFAULT);
  1904. if (mrt == NULL)
  1905. return ERR_PTR(-ENOENT);
  1906. it->mrt = mrt;
  1907. it->cache = NULL;
  1908. it->ct = 0;
  1909. return *pos ? ipmr_mfc_seq_idx(net, seq->private, *pos - 1)
  1910. : SEQ_START_TOKEN;
  1911. }
  1912. static void *ipmr_mfc_seq_next(struct seq_file *seq, void *v, loff_t *pos)
  1913. {
  1914. struct mfc_cache *mfc = v;
  1915. struct ipmr_mfc_iter *it = seq->private;
  1916. struct net *net = seq_file_net(seq);
  1917. struct mr_table *mrt = it->mrt;
  1918. ++*pos;
  1919. if (v == SEQ_START_TOKEN)
  1920. return ipmr_mfc_seq_idx(net, seq->private, 0);
  1921. if (mfc->list.next != it->cache)
  1922. return list_entry(mfc->list.next, struct mfc_cache, list);
  1923. if (it->cache == &mrt->mfc_unres_queue)
  1924. goto end_of_list;
  1925. BUG_ON(it->cache != &mrt->mfc_cache_array[it->ct]);
  1926. while (++it->ct < MFC_LINES) {
  1927. it->cache = &mrt->mfc_cache_array[it->ct];
  1928. if (list_empty(it->cache))
  1929. continue;
  1930. return list_first_entry(it->cache, struct mfc_cache, list);
  1931. }
  1932. /* exhausted cache_array, show unresolved */
  1933. rcu_read_unlock();
  1934. it->cache = &mrt->mfc_unres_queue;
  1935. it->ct = 0;
  1936. spin_lock_bh(&mfc_unres_lock);
  1937. if (!list_empty(it->cache))
  1938. return list_first_entry(it->cache, struct mfc_cache, list);
  1939. end_of_list:
  1940. spin_unlock_bh(&mfc_unres_lock);
  1941. it->cache = NULL;
  1942. return NULL;
  1943. }
  1944. static void ipmr_mfc_seq_stop(struct seq_file *seq, void *v)
  1945. {
  1946. struct ipmr_mfc_iter *it = seq->private;
  1947. struct mr_table *mrt = it->mrt;
  1948. if (it->cache == &mrt->mfc_unres_queue)
  1949. spin_unlock_bh(&mfc_unres_lock);
  1950. else if (it->cache == &mrt->mfc_cache_array[it->ct])
  1951. rcu_read_unlock();
  1952. }
  1953. static int ipmr_mfc_seq_show(struct seq_file *seq, void *v)
  1954. {
  1955. int n;
  1956. if (v == SEQ_START_TOKEN) {
  1957. seq_puts(seq,
  1958. "Group Origin Iif Pkts Bytes Wrong Oifs\n");
  1959. } else {
  1960. const struct mfc_cache *mfc = v;
  1961. const struct ipmr_mfc_iter *it = seq->private;
  1962. const struct mr_table *mrt = it->mrt;
  1963. seq_printf(seq, "%08X %08X %-3hd",
  1964. (__force u32) mfc->mfc_mcastgrp,
  1965. (__force u32) mfc->mfc_origin,
  1966. mfc->mfc_parent);
  1967. if (it->cache != &mrt->mfc_unres_queue) {
  1968. seq_printf(seq, " %8lu %8lu %8lu",
  1969. mfc->mfc_un.res.pkt,
  1970. mfc->mfc_un.res.bytes,
  1971. mfc->mfc_un.res.wrong_if);
  1972. for (n = mfc->mfc_un.res.minvif;
  1973. n < mfc->mfc_un.res.maxvif; n++) {
  1974. if (VIF_EXISTS(mrt, n) &&
  1975. mfc->mfc_un.res.ttls[n] < 255)
  1976. seq_printf(seq,
  1977. " %2d:%-3d",
  1978. n, mfc->mfc_un.res.ttls[n]);
  1979. }
  1980. } else {
  1981. /* unresolved mfc_caches don't contain
  1982. * pkt, bytes and wrong_if values
  1983. */
  1984. seq_printf(seq, " %8lu %8lu %8lu", 0ul, 0ul, 0ul);
  1985. }
  1986. seq_putc(seq, '\n');
  1987. }
  1988. return 0;
  1989. }
  1990. static const struct seq_operations ipmr_mfc_seq_ops = {
  1991. .start = ipmr_mfc_seq_start,
  1992. .next = ipmr_mfc_seq_next,
  1993. .stop = ipmr_mfc_seq_stop,
  1994. .show = ipmr_mfc_seq_show,
  1995. };
  1996. static int ipmr_mfc_open(struct inode *inode, struct file *file)
  1997. {
  1998. return seq_open_net(inode, file, &ipmr_mfc_seq_ops,
  1999. sizeof(struct ipmr_mfc_iter));
  2000. }
  2001. static const struct file_operations ipmr_mfc_fops = {
  2002. .owner = THIS_MODULE,
  2003. .open = ipmr_mfc_open,
  2004. .read = seq_read,
  2005. .llseek = seq_lseek,
  2006. .release = seq_release_net,
  2007. };
  2008. #endif
  2009. #ifdef CONFIG_IP_PIMSM_V2
  2010. static const struct net_protocol pim_protocol = {
  2011. .handler = pim_rcv,
  2012. .netns_ok = 1,
  2013. };
  2014. #endif
  2015. /*
  2016. * Setup for IP multicast routing
  2017. */
  2018. static int __net_init ipmr_net_init(struct net *net)
  2019. {
  2020. int err;
  2021. err = ipmr_rules_init(net);
  2022. if (err < 0)
  2023. goto fail;
  2024. #ifdef CONFIG_PROC_FS
  2025. err = -ENOMEM;
  2026. if (!proc_net_fops_create(net, "ip_mr_vif", 0, &ipmr_vif_fops))
  2027. goto proc_vif_fail;
  2028. if (!proc_net_fops_create(net, "ip_mr_cache", 0, &ipmr_mfc_fops))
  2029. goto proc_cache_fail;
  2030. #endif
  2031. return 0;
  2032. #ifdef CONFIG_PROC_FS
  2033. proc_cache_fail:
  2034. proc_net_remove(net, "ip_mr_vif");
  2035. proc_vif_fail:
  2036. ipmr_rules_exit(net);
  2037. #endif
  2038. fail:
  2039. return err;
  2040. }
  2041. static void __net_exit ipmr_net_exit(struct net *net)
  2042. {
  2043. #ifdef CONFIG_PROC_FS
  2044. proc_net_remove(net, "ip_mr_cache");
  2045. proc_net_remove(net, "ip_mr_vif");
  2046. #endif
  2047. ipmr_rules_exit(net);
  2048. }
  2049. static struct pernet_operations ipmr_net_ops = {
  2050. .init = ipmr_net_init,
  2051. .exit = ipmr_net_exit,
  2052. };
  2053. int __init ip_mr_init(void)
  2054. {
  2055. int err;
  2056. mrt_cachep = kmem_cache_create("ip_mrt_cache",
  2057. sizeof(struct mfc_cache),
  2058. 0, SLAB_HWCACHE_ALIGN | SLAB_PANIC,
  2059. NULL);
  2060. if (!mrt_cachep)
  2061. return -ENOMEM;
  2062. err = register_pernet_subsys(&ipmr_net_ops);
  2063. if (err)
  2064. goto reg_pernet_fail;
  2065. err = register_netdevice_notifier(&ip_mr_notifier);
  2066. if (err)
  2067. goto reg_notif_fail;
  2068. #ifdef CONFIG_IP_PIMSM_V2
  2069. if (inet_add_protocol(&pim_protocol, IPPROTO_PIM) < 0) {
  2070. printk(KERN_ERR "ip_mr_init: can't add PIM protocol\n");
  2071. err = -EAGAIN;
  2072. goto add_proto_fail;
  2073. }
  2074. #endif
  2075. rtnl_register(RTNL_FAMILY_IPMR, RTM_GETROUTE, NULL, ipmr_rtm_dumproute);
  2076. return 0;
  2077. #ifdef CONFIG_IP_PIMSM_V2
  2078. add_proto_fail:
  2079. unregister_netdevice_notifier(&ip_mr_notifier);
  2080. #endif
  2081. reg_notif_fail:
  2082. unregister_pernet_subsys(&ipmr_net_ops);
  2083. reg_pernet_fail:
  2084. kmem_cache_destroy(mrt_cachep);
  2085. return err;
  2086. }