ipt_REDIRECT.c
上传用户:jlfgdled
上传日期:2013-04-10
资源大小:33168k
文件大小:3k
源码类别:

Linux/Unix编程

开发平台:

Unix_Linux

  1. /* Redirect.  Simple mapping which alters dst to a local IP address. */
  2. #include <linux/types.h>
  3. #include <linux/ip.h>
  4. #include <linux/timer.h>
  5. #include <linux/module.h>
  6. #include <linux/netfilter.h>
  7. #include <linux/netdevice.h>
  8. #include <linux/if.h>
  9. #include <linux/inetdevice.h>
  10. #include <net/protocol.h>
  11. #include <net/checksum.h>
  12. #include <linux/netfilter_ipv4.h>
  13. #include <linux/netfilter_ipv4/ip_nat_rule.h>
  14. #if 0
  15. #define DEBUGP printk
  16. #else
  17. #define DEBUGP(format, args...)
  18. #endif
  19. /* FIXME: Take multiple ranges --RR */
  20. static int
  21. redirect_check(const char *tablename,
  22.        const struct ipt_entry *e,
  23.        void *targinfo,
  24.        unsigned int targinfosize,
  25.        unsigned int hook_mask)
  26. {
  27. const struct ip_nat_multi_range *mr = targinfo;
  28. if (strcmp(tablename, "nat") != 0) {
  29. DEBUGP("redirect_check: bad table `%s'.n", table);
  30. return 0;
  31. }
  32. if (targinfosize != IPT_ALIGN(sizeof(*mr))) {
  33. DEBUGP("redirect_check: size %u.n", targinfosize);
  34. return 0;
  35. }
  36. if (hook_mask & ~((1 << NF_IP_PRE_ROUTING) | (1 << NF_IP_LOCAL_OUT))) {
  37. DEBUGP("redirect_check: bad hooks %x.n", hook_mask);
  38. return 0;
  39. }
  40. if (mr->range[0].flags & IP_NAT_RANGE_MAP_IPS) {
  41. DEBUGP("redirect_check: bad MAP_IPS.n");
  42. return 0;
  43. }
  44. if (mr->rangesize != 1) {
  45. DEBUGP("redirect_check: bad rangesize %u.n", mr->rangesize);
  46. return 0;
  47. }
  48. return 1;
  49. }
  50. static unsigned int
  51. redirect_target(struct sk_buff **pskb,
  52. unsigned int hooknum,
  53. const struct net_device *in,
  54. const struct net_device *out,
  55. const void *targinfo,
  56. void *userinfo)
  57. {
  58. struct ip_conntrack *ct;
  59. enum ip_conntrack_info ctinfo;
  60. u_int32_t newdst;
  61. const struct ip_nat_multi_range *mr = targinfo;
  62. struct ip_nat_multi_range newrange;
  63. IP_NF_ASSERT(hooknum == NF_IP_PRE_ROUTING
  64.      || hooknum == NF_IP_LOCAL_OUT);
  65. ct = ip_conntrack_get(*pskb, &ctinfo);
  66. IP_NF_ASSERT(ct && (ctinfo == IP_CT_NEW || ctinfo == IP_CT_RELATED));
  67. /* Local packets: make them go to loopback */
  68. if (hooknum == NF_IP_LOCAL_OUT)
  69. newdst = htonl(0x7F000001);
  70. else {
  71. struct in_device *indev;
  72. /* Device might not have an associated in_device. */
  73. indev = (struct in_device *)(*pskb)->dev->ip_ptr;
  74. if (indev == NULL)
  75. return NF_DROP;
  76. /* Grab first address on interface. */
  77. newdst = indev->ifa_list->ifa_local;
  78. }
  79. /* Transfer from original range. */
  80. newrange = ((struct ip_nat_multi_range)
  81. { 1, { { mr->range[0].flags | IP_NAT_RANGE_MAP_IPS,
  82.  newdst, newdst,
  83.  mr->range[0].min, mr->range[0].max } } });
  84. /* Hand modified range to generic setup. */
  85. return ip_nat_setup_info(ct, &newrange, hooknum);
  86. }
  87. static struct ipt_target redirect_reg
  88. = { { NULL, NULL }, "REDIRECT", redirect_target, redirect_check, NULL,
  89.     THIS_MODULE };
  90. static int __init init(void)
  91. {
  92. return ipt_register_target(&redirect_reg);
  93. }
  94. static void __exit fini(void)
  95. {
  96. ipt_unregister_target(&redirect_reg);
  97. }
  98. module_init(init);
  99. module_exit(fini);
  100. MODULE_LICENSE("GPL");