一个基于LINUX的简单防火墙

 

1. // firewall.cpp : 定义控制台应用程序的入口点。
3. //
7. #define _KERNEL_
8. #define MODULE
10. #include <linux/module.h>
11. #include <linux/kernel.h>
12. #include <linux/version.h>
13. #include <linux/netfilter.h>
14. #include <linux/netfilter_ipv4.h>
15. #include <linux/netdevice.h>
16. #include <linux/if_packet.h>
17. #include <linux/skbuff.h>
18. #include <linux/ip.h>
19. #include <linux/icmp.h>
20. #include <linux/in.h>
21. #include <linux/tcp.h>
22. #include <linux/netdevice.h>
25. static struct nf_hook_ops nfho;
27. static unsigned char *drop_ip="/x7f/x00/x00/x01";
30. unsigned int hook_func(unsigned int hooknum,struct sk_buff **skb,const struct net_device *in,const struct net_device *out,int (*okfn)(struct sk_buff *))
32. {
33.     struct sk_buff *sb=*skb;
35.     struct icmphdr *icmp;
36.     if(sb->nh.iph->protocol!=IPPROTO_ICMP)
37.         return NF_ACCEPT;
38.     icmp=(struct icmphdr *)(sb->data + sb->nh.iph->ihl*4);
40.  //       if(icmp->code !=MAGIC_CODE || icmp->type!=ICMP_ECHO||ICMP_PAYLOAD_SIZE<REPLY_SIZE){
41. //      return NF_ACCEPT;
42. //  }
43.     printk("trying!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!/n");
47.         if(icmp->type == ICMP_ECHO)
49.         {
51.             unsigned char *saddr = &(sb->nh.iph->saddr);
53.             printk("<1>---ping from:%d.%d.%d.%d ---/n",*saddr,*(saddr+1),*(saddr+2),*(saddr+3));
55.               sb->h.icmph->checksum+=1;
56.             return NF_DROP;
58.         }
59.              if(icmp->type == ICMP_ECHOREPLY){
60.             unsigned char *saddr = &(sb->nh.iph->saddr);
62.             printk("<1>---ping reply from:%d.%d.%d.%d ---/n",*saddr,*(saddr+1),*(saddr+2),*(saddr+3));
64.             sb->h.icmph->checksum+=1;
65.             return NF_ACCEPT;
67.         }
69. return NF_DROP;
71. }
75. int init_module()
77. {
78. printk("simple firewall by zsh/n");
80. nfho.hook    = hook_func;
82. nfho.hooknum = NF_IP_PRE_ROUTING;
84. nfho.pf      = PF_INET;
86. nfho.priority= NF_IP_PRI_FIRST;
90. nf_register_hook(&nfho);
94. return 0;
100. }
104. void cleanup_module()
106. {
108.     nf_unregister_hook(&nfho);
110. }