fib_frontend.c
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:15k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * INET An implementation of the TCP/IP protocol suite for the LINUX
  3.  * operating system.  INET is implemented using the  BSD Socket
  4.  * interface as the means of communication with the user level.
  5.  *
  6.  * IPv4 Forwarding Information Base: FIB frontend.
  7.  *
  8.  * Version: $Id: fib_frontend.c,v 1.26 2001/10/31 21:55:54 davem Exp $
  9.  *
  10.  * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
  11.  *
  12.  * This program is free software; you can redistribute it and/or
  13.  * modify it under the terms of the GNU General Public License
  14.  * as published by the Free Software Foundation; either version
  15.  * 2 of the License, or (at your option) any later version.
  16.  */
  17. #include <linux/config.h>
  18. #include <asm/uaccess.h>
  19. #include <asm/system.h>
  20. #include <asm/bitops.h>
  21. #include <linux/types.h>
  22. #include <linux/kernel.h>
  23. #include <linux/sched.h>
  24. #include <linux/mm.h>
  25. #include <linux/string.h>
  26. #include <linux/socket.h>
  27. #include <linux/sockios.h>
  28. #include <linux/errno.h>
  29. #include <linux/in.h>
  30. #include <linux/inet.h>
  31. #include <linux/netdevice.h>
  32. #include <linux/if_arp.h>
  33. #include <linux/proc_fs.h>
  34. #include <linux/skbuff.h>
  35. #include <linux/netlink.h>
  36. #include <linux/init.h>
  37. #include <net/ip.h>
  38. #include <net/protocol.h>
  39. #include <net/route.h>
  40. #include <net/tcp.h>
  41. #include <net/sock.h>
  42. #include <net/icmp.h>
  43. #include <net/arp.h>
  44. #include <net/ip_fib.h>
  45. #define FFprint(a...) printk(KERN_DEBUG a)
  46. #ifndef CONFIG_IP_MULTIPLE_TABLES
  47. #define RT_TABLE_MIN RT_TABLE_MAIN
  48. struct fib_table *local_table;
  49. struct fib_table *main_table;
  50. #else
  51. #define RT_TABLE_MIN 1
  52. struct fib_table *fib_tables[RT_TABLE_MAX+1];
  53. struct fib_table *__fib_new_table(int id)
  54. {
  55. struct fib_table *tb;
  56. tb = fib_hash_init(id);
  57. if (!tb)
  58. return NULL;
  59. fib_tables[id] = tb;
  60. return tb;
  61. }
  62. #endif /* CONFIG_IP_MULTIPLE_TABLES */
  63. void fib_flush(void)
  64. {
  65. int flushed = 0;
  66. #ifdef CONFIG_IP_MULTIPLE_TABLES
  67. struct fib_table *tb;
  68. int id;
  69. for (id = RT_TABLE_MAX; id>0; id--) {
  70. if ((tb = fib_get_table(id))==NULL)
  71. continue;
  72. flushed += tb->tb_flush(tb);
  73. }
  74. #else /* CONFIG_IP_MULTIPLE_TABLES */
  75. flushed += main_table->tb_flush(main_table);
  76. flushed += local_table->tb_flush(local_table);
  77. #endif /* CONFIG_IP_MULTIPLE_TABLES */
  78. if (flushed)
  79. rt_cache_flush(-1);
  80. }
  81. #ifdef CONFIG_PROC_FS
  82. /* 
  83.  * Called from the PROCfs module. This outputs /proc/net/route.
  84.  *
  85.  * It always works in backward compatibility mode.
  86.  * The format of the file is not supposed to be changed.
  87.  */
  88.  
  89. static int
  90. fib_get_procinfo(char *buffer, char **start, off_t offset, int length)
  91. {
  92. int first = offset/128;
  93. char *ptr = buffer;
  94. int count = (length+127)/128;
  95. int len;
  96. *start = buffer + offset%128;
  97. if (--first < 0) {
  98. sprintf(buffer, "%-127sn", "IfacetDestinationtGateway tFlagstRefCnttUsetMetrictMaskttMTUtWindowtIRTT");
  99. --count;
  100. ptr += 128;
  101. first = 0;
  102.    }
  103. if (main_table && count > 0) {
  104. int n = main_table->tb_get_info(main_table, ptr, first, count);
  105. count -= n;
  106. ptr += n*128;
  107. }
  108. len = ptr - *start;
  109. if (len >= length)
  110. return length;
  111. if (len >= 0)
  112. return len;
  113. return 0;
  114. }
  115. #endif /* CONFIG_PROC_FS */
  116. /*
  117.  * Find the first device with a given source address.
  118.  */
  119. struct net_device * ip_dev_find(u32 addr)
  120. {
  121. struct rt_key key;
  122. struct fib_result res;
  123. struct net_device *dev = NULL;
  124. memset(&key, 0, sizeof(key));
  125. key.dst = addr;
  126. #ifdef CONFIG_IP_MULTIPLE_TABLES
  127. res.r = NULL;
  128. #endif
  129. if (!local_table || local_table->tb_lookup(local_table, &key, &res)) {
  130. return NULL;
  131. }
  132. if (res.type != RTN_LOCAL)
  133. goto out;
  134. dev = FIB_RES_DEV(res);
  135. if (dev)
  136. atomic_inc(&dev->refcnt);
  137. out:
  138. fib_res_put(&res);
  139. return dev;
  140. }
  141. unsigned inet_addr_type(u32 addr)
  142. {
  143. struct rt_key key;
  144. struct fib_result res;
  145. unsigned ret = RTN_BROADCAST;
  146. if (ZERONET(addr) || BADCLASS(addr))
  147. return RTN_BROADCAST;
  148. if (MULTICAST(addr))
  149. return RTN_MULTICAST;
  150. memset(&key, 0, sizeof(key));
  151. key.dst = addr;
  152. #ifdef CONFIG_IP_MULTIPLE_TABLES
  153. res.r = NULL;
  154. #endif
  155. if (local_table) {
  156. ret = RTN_UNICAST;
  157. if (local_table->tb_lookup(local_table, &key, &res) == 0) {
  158. ret = res.type;
  159. fib_res_put(&res);
  160. }
  161. }
  162. return ret;
  163. }
  164. /* Given (packet source, input interface) and optional (dst, oif, tos):
  165.    - (main) check, that source is valid i.e. not broadcast or our local
  166.      address.
  167.    - figure out what "logical" interface this packet arrived
  168.      and calculate "specific destination" address.
  169.    - check, that packet arrived from expected physical interface.
  170.  */
  171. int fib_validate_source(u32 src, u32 dst, u8 tos, int oif,
  172. struct net_device *dev, u32 *spec_dst, u32 *itag)
  173. {
  174. struct in_device *in_dev;
  175. struct rt_key key;
  176. struct fib_result res;
  177. int no_addr, rpf;
  178. int ret;
  179. key.dst = src;
  180. key.src = dst;
  181. key.tos = tos;
  182. key.oif = 0;
  183. key.iif = oif;
  184. key.scope = RT_SCOPE_UNIVERSE;
  185. no_addr = rpf = 0;
  186. read_lock(&inetdev_lock);
  187. in_dev = __in_dev_get(dev);
  188. if (in_dev) {
  189. no_addr = in_dev->ifa_list == NULL;
  190. rpf = IN_DEV_RPFILTER(in_dev);
  191. }
  192. read_unlock(&inetdev_lock);
  193. if (in_dev == NULL)
  194. goto e_inval;
  195. if (fib_lookup(&key, &res))
  196. goto last_resort;
  197. if (res.type != RTN_UNICAST)
  198. goto e_inval_res;
  199. *spec_dst = FIB_RES_PREFSRC(res);
  200. fib_combine_itag(itag, &res);
  201. #ifdef CONFIG_IP_ROUTE_MULTIPATH
  202. if (FIB_RES_DEV(res) == dev || res.fi->fib_nhs > 1)
  203. #else
  204. if (FIB_RES_DEV(res) == dev)
  205. #endif
  206. {
  207. ret = FIB_RES_NH(res).nh_scope >= RT_SCOPE_HOST;
  208. fib_res_put(&res);
  209. return ret;
  210. }
  211. fib_res_put(&res);
  212. if (no_addr)
  213. goto last_resort;
  214. if (rpf)
  215. goto e_inval;
  216. key.oif = dev->ifindex;
  217. ret = 0;
  218. if (fib_lookup(&key, &res) == 0) {
  219. if (res.type == RTN_UNICAST) {
  220. *spec_dst = FIB_RES_PREFSRC(res);
  221. ret = FIB_RES_NH(res).nh_scope >= RT_SCOPE_HOST;
  222. }
  223. fib_res_put(&res);
  224. }
  225. return ret;
  226. last_resort:
  227. if (rpf)
  228. goto e_inval;
  229. *spec_dst = inet_select_addr(dev, 0, RT_SCOPE_UNIVERSE);
  230. *itag = 0;
  231. return 0;
  232. e_inval_res:
  233. fib_res_put(&res);
  234. e_inval:
  235. return -EINVAL;
  236. }
  237. #ifndef CONFIG_IP_NOSIOCRT
  238. /*
  239.  * Handle IP routing ioctl calls. These are used to manipulate the routing tables
  240.  */
  241.  
  242. int ip_rt_ioctl(unsigned int cmd, void *arg)
  243. {
  244. int err;
  245. struct kern_rta rta;
  246. struct rtentry  r;
  247. struct {
  248. struct nlmsghdr nlh;
  249. struct rtmsg rtm;
  250. } req;
  251. switch (cmd) {
  252. case SIOCADDRT: /* Add a route */
  253. case SIOCDELRT: /* Delete a route */
  254. if (!capable(CAP_NET_ADMIN))
  255. return -EPERM;
  256. if (copy_from_user(&r, arg, sizeof(struct rtentry)))
  257. return -EFAULT;
  258. rtnl_lock();
  259. err = fib_convert_rtentry(cmd, &req.nlh, &req.rtm, &rta, &r);
  260. if (err == 0) {
  261. if (cmd == SIOCDELRT) {
  262. struct fib_table *tb = fib_get_table(req.rtm.rtm_table);
  263. err = -ESRCH;
  264. if (tb)
  265. err = tb->tb_delete(tb, &req.rtm, &rta, &req.nlh, NULL);
  266. } else {
  267. struct fib_table *tb = fib_new_table(req.rtm.rtm_table);
  268. err = -ENOBUFS;
  269. if (tb)
  270. err = tb->tb_insert(tb, &req.rtm, &rta, &req.nlh, NULL);
  271. }
  272. if (rta.rta_mx)
  273. kfree(rta.rta_mx);
  274. }
  275. rtnl_unlock();
  276. return err;
  277. }
  278. return -EINVAL;
  279. }
  280. #else
  281. int ip_rt_ioctl(unsigned int cmd, void *arg)
  282. {
  283. return -EINVAL;
  284. }
  285. #endif
  286. static int inet_check_attr(struct rtmsg *r, struct rtattr **rta)
  287. {
  288. int i;
  289. for (i=1; i<=RTA_MAX; i++) {
  290. struct rtattr *attr = rta[i-1];
  291. if (attr) {
  292. if (RTA_PAYLOAD(attr) < 4)
  293. return -EINVAL;
  294. if (i != RTA_MULTIPATH && i != RTA_METRICS)
  295. rta[i-1] = (struct rtattr*)RTA_DATA(attr);
  296. }
  297. }
  298. return 0;
  299. }
  300. int inet_rtm_delroute(struct sk_buff *skb, struct nlmsghdr* nlh, void *arg)
  301. {
  302. struct fib_table * tb;
  303. struct rtattr **rta = arg;
  304. struct rtmsg *r = NLMSG_DATA(nlh);
  305. if (inet_check_attr(r, rta))
  306. return -EINVAL;
  307. tb = fib_get_table(r->rtm_table);
  308. if (tb)
  309. return tb->tb_delete(tb, r, (struct kern_rta*)rta, nlh, &NETLINK_CB(skb));
  310. return -ESRCH;
  311. }
  312. int inet_rtm_newroute(struct sk_buff *skb, struct nlmsghdr* nlh, void *arg)
  313. {
  314. struct fib_table * tb;
  315. struct rtattr **rta = arg;
  316. struct rtmsg *r = NLMSG_DATA(nlh);
  317. if (inet_check_attr(r, rta))
  318. return -EINVAL;
  319. tb = fib_new_table(r->rtm_table);
  320. if (tb)
  321. return tb->tb_insert(tb, r, (struct kern_rta*)rta, nlh, &NETLINK_CB(skb));
  322. return -ENOBUFS;
  323. }
  324. int inet_dump_fib(struct sk_buff *skb, struct netlink_callback *cb)
  325. {
  326. int t;
  327. int s_t;
  328. struct fib_table *tb;
  329. if (NLMSG_PAYLOAD(cb->nlh, 0) >= sizeof(struct rtmsg) &&
  330.     ((struct rtmsg*)NLMSG_DATA(cb->nlh))->rtm_flags&RTM_F_CLONED)
  331. return ip_rt_dump(skb, cb);
  332. s_t = cb->args[0];
  333. if (s_t == 0)
  334. s_t = cb->args[0] = RT_TABLE_MIN;
  335. for (t=s_t; t<=RT_TABLE_MAX; t++) {
  336. if (t < s_t) continue;
  337. if (t > s_t)
  338. memset(&cb->args[1], 0, sizeof(cb->args)-sizeof(cb->args[0]));
  339. if ((tb = fib_get_table(t))==NULL)
  340. continue;
  341. if (tb->tb_dump(tb, skb, cb) < 0) 
  342. break;
  343. }
  344. cb->args[0] = t;
  345. return skb->len;
  346. }
  347. /* Prepare and feed intra-kernel routing request.
  348.    Really, it should be netlink message, but :-( netlink
  349.    can be not configured, so that we feed it directly
  350.    to fib engine. It is legal, because all events occur
  351.    only when netlink is already locked.
  352.  */
  353. static void fib_magic(int cmd, int type, u32 dst, int dst_len, struct in_ifaddr *ifa)
  354. {
  355. struct fib_table * tb;
  356. struct {
  357. struct nlmsghdr nlh;
  358. struct rtmsg rtm;
  359. } req;
  360. struct kern_rta rta;
  361. memset(&req.rtm, 0, sizeof(req.rtm));
  362. memset(&rta, 0, sizeof(rta));
  363. if (type == RTN_UNICAST)
  364. tb = fib_new_table(RT_TABLE_MAIN);
  365. else
  366. tb = fib_new_table(RT_TABLE_LOCAL);
  367. if (tb == NULL)
  368. return;
  369. req.nlh.nlmsg_len = sizeof(req);
  370. req.nlh.nlmsg_type = cmd;
  371. req.nlh.nlmsg_flags = NLM_F_REQUEST|NLM_F_CREATE|NLM_F_APPEND;
  372. req.nlh.nlmsg_pid = 0;
  373. req.nlh.nlmsg_seq = 0;
  374. req.rtm.rtm_dst_len = dst_len;
  375. req.rtm.rtm_table = tb->tb_id;
  376. req.rtm.rtm_protocol = RTPROT_KERNEL;
  377. req.rtm.rtm_scope = (type != RTN_LOCAL ? RT_SCOPE_LINK : RT_SCOPE_HOST);
  378. req.rtm.rtm_type = type;
  379. rta.rta_dst = &dst;
  380. rta.rta_prefsrc = &ifa->ifa_local;
  381. rta.rta_oif = &ifa->ifa_dev->dev->ifindex;
  382. if (cmd == RTM_NEWROUTE)
  383. tb->tb_insert(tb, &req.rtm, &rta, &req.nlh, NULL);
  384. else
  385. tb->tb_delete(tb, &req.rtm, &rta, &req.nlh, NULL);
  386. }
  387. static void fib_add_ifaddr(struct in_ifaddr *ifa)
  388. {
  389. struct in_device *in_dev = ifa->ifa_dev;
  390. struct net_device *dev = in_dev->dev;
  391. struct in_ifaddr *prim = ifa;
  392. u32 mask = ifa->ifa_mask;
  393. u32 addr = ifa->ifa_local;
  394. u32 prefix = ifa->ifa_address&mask;
  395. if (ifa->ifa_flags&IFA_F_SECONDARY) {
  396. prim = inet_ifa_byprefix(in_dev, prefix, mask);
  397. if (prim == NULL) {
  398. printk(KERN_DEBUG "fib_add_ifaddr: bug: prim == NULLn");
  399. return;
  400. }
  401. }
  402. fib_magic(RTM_NEWROUTE, RTN_LOCAL, addr, 32, prim);
  403. if (!(dev->flags&IFF_UP))
  404. return;
  405. /* Add broadcast address, if it is explicitly assigned. */
  406. if (ifa->ifa_broadcast && ifa->ifa_broadcast != 0xFFFFFFFF)
  407. fib_magic(RTM_NEWROUTE, RTN_BROADCAST, ifa->ifa_broadcast, 32, prim);
  408. if (!ZERONET(prefix) && !(ifa->ifa_flags&IFA_F_SECONDARY) &&
  409.     (prefix != addr || ifa->ifa_prefixlen < 32)) {
  410. fib_magic(RTM_NEWROUTE, dev->flags&IFF_LOOPBACK ? RTN_LOCAL :
  411.   RTN_UNICAST, prefix, ifa->ifa_prefixlen, prim);
  412. /* Add network specific broadcasts, when it takes a sense */
  413. if (ifa->ifa_prefixlen < 31) {
  414. fib_magic(RTM_NEWROUTE, RTN_BROADCAST, prefix, 32, prim);
  415. fib_magic(RTM_NEWROUTE, RTN_BROADCAST, prefix|~mask, 32, prim);
  416. }
  417. }
  418. }
  419. static void fib_del_ifaddr(struct in_ifaddr *ifa)
  420. {
  421. struct in_device *in_dev = ifa->ifa_dev;
  422. struct net_device *dev = in_dev->dev;
  423. struct in_ifaddr *ifa1;
  424. struct in_ifaddr *prim = ifa;
  425. u32 brd = ifa->ifa_address|~ifa->ifa_mask;
  426. u32 any = ifa->ifa_address&ifa->ifa_mask;
  427. #define LOCAL_OK 1
  428. #define BRD_OK 2
  429. #define BRD0_OK 4
  430. #define BRD1_OK 8
  431. unsigned ok = 0;
  432. if (!(ifa->ifa_flags&IFA_F_SECONDARY))
  433. fib_magic(RTM_DELROUTE, dev->flags&IFF_LOOPBACK ? RTN_LOCAL :
  434.   RTN_UNICAST, any, ifa->ifa_prefixlen, prim);
  435. else {
  436. prim = inet_ifa_byprefix(in_dev, any, ifa->ifa_mask);
  437. if (prim == NULL) {
  438. printk(KERN_DEBUG "fib_del_ifaddr: bug: prim == NULLn");
  439. return;
  440. }
  441. }
  442. /* Deletion is more complicated than add.
  443.    We should take care of not to delete too much :-)
  444.    Scan address list to be sure that addresses are really gone.
  445.  */
  446. for (ifa1 = in_dev->ifa_list; ifa1; ifa1 = ifa1->ifa_next) {
  447. if (ifa->ifa_local == ifa1->ifa_local)
  448. ok |= LOCAL_OK;
  449. if (ifa->ifa_broadcast == ifa1->ifa_broadcast)
  450. ok |= BRD_OK;
  451. if (brd == ifa1->ifa_broadcast)
  452. ok |= BRD1_OK;
  453. if (any == ifa1->ifa_broadcast)
  454. ok |= BRD0_OK;
  455. }
  456. if (!(ok&BRD_OK))
  457. fib_magic(RTM_DELROUTE, RTN_BROADCAST, ifa->ifa_broadcast, 32, prim);
  458. if (!(ok&BRD1_OK))
  459. fib_magic(RTM_DELROUTE, RTN_BROADCAST, brd, 32, prim);
  460. if (!(ok&BRD0_OK))
  461. fib_magic(RTM_DELROUTE, RTN_BROADCAST, any, 32, prim);
  462. if (!(ok&LOCAL_OK)) {
  463. fib_magic(RTM_DELROUTE, RTN_LOCAL, ifa->ifa_local, 32, prim);
  464. /* Check, that this local address finally disappeared. */
  465. if (inet_addr_type(ifa->ifa_local) != RTN_LOCAL) {
  466. /* And the last, but not the least thing.
  467.    We must flush stray FIB entries.
  468.    First of all, we scan fib_info list searching
  469.    for stray nexthop entries, then ignite fib_flush.
  470. */
  471. if (fib_sync_down(ifa->ifa_local, NULL, 0))
  472. fib_flush();
  473. }
  474. }
  475. #undef LOCAL_OK
  476. #undef BRD_OK
  477. #undef BRD0_OK
  478. #undef BRD1_OK
  479. }
  480. static void fib_disable_ip(struct net_device *dev, int force)
  481. {
  482. if (fib_sync_down(0, dev, force))
  483. fib_flush();
  484. rt_cache_flush(0);
  485. arp_ifdown(dev);
  486. }
  487. static int fib_inetaddr_event(struct notifier_block *this, unsigned long event, void *ptr)
  488. {
  489. struct in_ifaddr *ifa = (struct in_ifaddr*)ptr;
  490. switch (event) {
  491. case NETDEV_UP:
  492. fib_add_ifaddr(ifa);
  493. rt_cache_flush(-1);
  494. break;
  495. case NETDEV_DOWN:
  496. fib_del_ifaddr(ifa);
  497. if (ifa->ifa_dev && ifa->ifa_dev->ifa_list == NULL) {
  498. /* Last address was deleted from this interface.
  499.    Disable IP.
  500.  */
  501. fib_disable_ip(ifa->ifa_dev->dev, 1);
  502. } else {
  503. rt_cache_flush(-1);
  504. }
  505. break;
  506. }
  507. return NOTIFY_DONE;
  508. }
  509. static int fib_netdev_event(struct notifier_block *this, unsigned long event, void *ptr)
  510. {
  511. struct net_device *dev = ptr;
  512. struct in_device *in_dev = __in_dev_get(dev);
  513. if (!in_dev)
  514. return NOTIFY_DONE;
  515. switch (event) {
  516. case NETDEV_UP:
  517. for_ifa(in_dev) {
  518. fib_add_ifaddr(ifa);
  519. } endfor_ifa(in_dev);
  520. #ifdef CONFIG_IP_ROUTE_MULTIPATH
  521. fib_sync_up(dev);
  522. #endif
  523. rt_cache_flush(-1);
  524. break;
  525. case NETDEV_DOWN:
  526. fib_disable_ip(dev, 0);
  527. break;
  528. case NETDEV_UNREGISTER:
  529. fib_disable_ip(dev, 1);
  530. break;
  531. case NETDEV_CHANGEMTU:
  532. case NETDEV_CHANGE:
  533. rt_cache_flush(0);
  534. break;
  535. }
  536. return NOTIFY_DONE;
  537. }
  538. struct notifier_block fib_inetaddr_notifier = {
  539. notifier_call: fib_inetaddr_event,
  540. };
  541. struct notifier_block fib_netdev_notifier = {
  542. notifier_call: fib_netdev_event,
  543. };
  544. void __init ip_fib_init(void)
  545. {
  546. #ifdef CONFIG_PROC_FS
  547. proc_net_create("route",0,fib_get_procinfo);
  548. #endif /* CONFIG_PROC_FS */
  549. #ifndef CONFIG_IP_MULTIPLE_TABLES
  550. local_table = fib_hash_init(RT_TABLE_LOCAL);
  551. main_table = fib_hash_init(RT_TABLE_MAIN);
  552. #else
  553. fib_rules_init();
  554. #endif
  555. register_netdevice_notifier(&fib_netdev_notifier);
  556. register_inetaddr_notifier(&fib_inetaddr_notifier);
  557. }