请教一个问题，Netfilter过滤URL，遇到视频类网站内核崩溃

smillers

如题，用Netfilter写了一段代码来打印HOST和URI信息，编译后可正常打印出URL，当浏览器打开腾讯视频等一些内容的时候，内核就崩溃了，麻烦各位大牛帮忙分析一下，谢谢！

1. #ifndef __KERNEL__
   
2. #define __KERNEL__
   
3. #endif
   
4. #ifndef MODULE
   
5. #define MODULE
   
6. #endif
   
7. 
   
8. #include   <linux/module.h>
   
9. #include   <linux/kernel.h>
   
10. #include   <linux/skbuff.h>
    
11. #include   <linux/in.h>
    
12. #include   <linux/ip.h>
    
13. #include   <linux/tcp.h>
    
14. #include   <linux/icmp.h>
    
15. #include   <linux/netdevice.h>
    
16. #include   <linux/netfilter.h>
    
17. #include   <linux/netfilter_ipv4.h>
    
18. #include   <linux/if_arp.h>
    
19. #include   <linux/if_ether.h>
    
20. #include   <linux/if_packet.h>
    
21. #include   <linux/mm.h>
    
22. #include   <linux/fs.h>
    
23. #include   <linux/uaccess.h>
    
24. #include   <asm/uaccess.h>
    
25. 
    
26. //#define    DEBUG
    
27. 
    
28. MODULE_LICENSE("GPL");
    
29. 
    
30. char* h_tmp ;           //HOST
    
31. char* u_tmp ;           //URI
    
32. 
    
33. 
    
34. static struct nf_hook_ops nfho;
    
35. 
    
36. 
    
37. //获取HOST
    
38. void print_host(const unsigned char* start,unsigned int len)
    
39. {
    
40.         char* host_tmp ;
    
41.         char* url_tmp ;
    
42.         int i = 0;
    
43.         int j = 0;
    
44.         if (len > 4)
    
45.         {
    
46.            if (0 == strncmp(start, "GET", 3) || 0 == strncmp(start, "POST", 4))
    
47.            {
    
48.                 /*获取HOST的值*/
    
49.                    host_tmp = strstr(start,"Host:");
    
50.                    if(memcmp(host_tmp,"Host:",5) == 0)
    
51.                    {
    
52.                            host_tmp = host_tmp + 6;               // 跳过HOST： 这6个字符
    
53.                            while(*host_tmp!='\r')
    
54.                            {
    
55.                                    host_tmp++;
    
56.                                    i++;
    
57.                            }
    
58.                                 host_tmp -= i;
    
59.                                 h_tmp = kmalloc((i + 1)*sizeof(char),GFP_KERNEL);
    
60.                                 memcpy(h_tmp,host_tmp,i);
    
61.                                 *(h_tmp+i) ='\0';
    
62.                                 printk(">>>Host:%s",h_tmp);
    
63.                                 kfree(h_tmp);
    
64.                    }
    
65.                 /*获取GET后面的参数*/
    
66.                            url_tmp = strstr(start,"/");            /*从/开始获取GET后面的URI部分*/
    
67.                            while(*url_tmp!=' ')                                /*到空格的地方结束*/
    
68.                                 {
    
69.                                    url_tmp++;
    
70.                                    j++;
    
71.                                 }
    
72.                            url_tmp -=j;
    
73.                            u_tmp = kmalloc((j + 1)*sizeof(char),GFP_KERNEL);
    
74.                            memcpy(u_tmp,url_tmp,j);
    
75.                            *(u_tmp+j) ='\0';
    
76.                            printk("%s\n",u_tmp);
    
77.                            kfree(u_tmp);
    
78.                    }
    
79.          }
    
80. }
    
81. 
    
82. /* HOOK函数 */
    
83. unsigned int hook_func(unsigned int hooknum,
    
84.                                            struct sk_buff *skb,
    
85.                                            const struct net_device *in,
    
86.                                            const struct net_device *out,
    
87.                                            int (*okfn)(struct sk_buff *))
    
88. {
    
89.         char * payload = NULL;
    
90.         struct sk_buff* sb = skb;                                                                                        //定义一个Linux网络数据包缓存
    
91.         struct iphdr *iph = ip_hdr(sb);                                                                        //定义一个ip报文的数据报头
    
92. 
    
93. /*读取文件结束，过滤掉skb_buf为空以及iph为空的数据包*/
    
94.         if(sb == NULL)
    
95.           return NF_ACCEPT;
    
96.         if(iph == NULL)
    
97.           return NF_ACCEPT;
    
98. 
    
99. /*判断是否为TCP协议，不是TCP协议的数据包直接return NF_ACCEPT*/
    
100.         if (iph && iph->protocol && (iph->protocol == IPPROTO_TCP)) {
     
101.                 //struct tcphdr* tcph = (void *)iph + iph->ihl*4;                           //TCP报文的头部
     
102.                 struct tcphdr* tcph =(void *)iph +(iph->ihl<<2);
     
103.                 unsigned int srcport = tcph->source;                                                 //源端口
     
104.                 unsigned int dstport = tcph->dest;                                                   //目的端口
     
105.                 unsigned int ack_seq_num = tcph->ack_seq;                                                   //ACK-SEQ的值
     
106.                 unsigned int seq_num = tcph->seq;                                                              //SEQ的值
     
107.                 unsigned int ip_saddr = ntohl(iph->saddr);                                                  //源IP
     
108.                 unsigned int ip_daddr = ntohl(iph->daddr);                                                  //目的IP
     
109.                 unsigned int len_tot = ntohs(iph->tot_len);                                                  //包总长度
     
110.                 unsigned int len_tcp = len_tot - iph->ihl*4 - tcph->doff*4;         //TCP包的长度
     
111. 
     
112. /*判断tcph是否为空，为空直接返回NF_ACCEPT*******/
     
113.             if(tcph == NULL)
     
114.                return NF_ACCEPT;
     
115. /*判断目的端口是否为80************************/
     
116.                 if (ntohs(dstport) == 80 )                                                                                   //筛选远端口或者目的
     
117.                 {
     
118.                         if (0 != skb_linearize(sb))
     
119.                            return NF_ACCEPT;
     
120. 
     
121. //                   payload = (char*)iph+(iph->ihl*4) + tcph->doff*4;
     
122.                    payload = (void *)tcph + (tcph->doff<<2);
     
123.                    print_host(payload,len_tcp);                                                                                     //DEBUG模式下打印HOST 和 URL
     
124. 
     
125.                     if (0 == strncmp(payload, "GET", 3) || 0 == strncmp(payload, "POST", 4))
     
126.                          {
     
127. 
     
128.                                          printk("**************Request**************\n");
     
129.                                          printk("In-Device :%s         Out-Device:%s\n",in->name,out->name);
     
130.                                          printk("Src IP    :%d.%d.%d.%d \n", (ip_saddr & 0xff000000) >> 24,
     
131.                                                                                                                   (ip_saddr & 0x00ff0000) >> 16,
     
132.                                                                                                                   (ip_saddr & 0x0000ff00) >> 8,
     
133.                                                                                                                   (ip_saddr & 0x000000ff));
     
134.                                          printk("Dst IP    :%d.%d.%d.%d \n", (ip_daddr & 0xff000000) >> 24,
     
135.                                                                                                                  (ip_daddr & 0x00ff0000) >> 16,
     
136.                                                                                                                 (ip_daddr & 0x0000ff00) >> 8,
     
137.                                                                                                                  (ip_daddr & 0x000000ff));
     
138.                                          printk("ID        :%u\n",ntohs(iph->id));
     
139.                                          printk("Src-Port  :%d         Dst-Port:%d\n", ntohs(srcport), ntohs(dstport));
     
140.                                          printk("Seq       :0x%2x\n", ntohl(seq_num));
     
141.                                          printk("Ack_Seq   :0x%2x\n", ntohl(ack_seq_num));
     
142.                                          printk("Tcp len   :%u\n",len_tcp);
     
143.                                          printk("Next ack  :%u\n",ntohl(seq_num + len_tcp ));
     
144.                                          printk("*******************************\n");
     
145.                                          return NF_ACCEPT;
     
146.                          }
     
147.                          return NF_ACCEPT;
     
148.                 }
     
149.         }
     
150. 
     
151.         return NF_ACCEPT;
     
152. }
     
153. 
     
154. /* 加载模块 */
     
155. int init_module() {
     
156.         nfho.hook = hook_func;                                     //钩子函数指针
     
157.         nfho.hooknum = NF_INET_POST_ROUTING;    //hook类型
     
158.         nfho.pf = PF_INET;                                            //协议簇，对于ipv4而言，是PF_INET
     
159.         nfho.priority = NF_IP_PRI_FIRST;              //优先级
     
160.         nf_register_hook(&nfho);
     
161. 
     
162.         pr_info("Filter add into kernel!\r\n");
     
163.         return 0;
     
164. }
     
165. /* 清除模块 */
     
166. void cleanup_module() {
     
167.         nf_unregister_hook(&nfho);
     
168.         pr_info("Filter removed from kernel!\r\n");
     
169. }
     

复制代码

Godbach

回复 1# smillers
其实如果内核 oops 了，应该有一些有用的新的。

简单看了一下你解析 TCP 报文的问题，首先一个问题没有考虑，就是分片包的问题。如果是 TCP 的分片报文，那么根本就没有所谓的 tcp header 让你解引用了。

此外，一个吹毛求疵的小问题：

        if(iph == NULL)
          return NF_ACCEPT;

/*判断是否为TCP协议，不是TCP协议的数据包直接return NF_ACCEPT*/
        if (iph && iph->protocol && (iph->protocol == IPPROTO_TCP)) {

最后那个条件判断，可以简化了为

if (iph->protocol == IPPROTO_TCP)

你 iph 都判断过了，没必要多两个冗余的条件。

   

humjb_1983

具体可能还得通过kdump，或者串口抓点信息来看。
也可能是printk打印太频繁导致？

kkddkkdd11

Godbach 发表于 2014-06-02 22:52
回复 1# smillers
其实如果内核 oops 了，应该有一些有用的新的。

版主说的这个，可以线性化一下

huangbanban

host_tmp = strstr(start,"Host:");
url_tmp = strstr(start,"/");
strstr函数的使用让我有些不安，没有限制查找的范围，因为是网络数据包
要是能够实现这样一个函数再使用，也许更安心一些吧
strnstr(char* s1, char*s2, int pos)

smillers

好像是strstr()函数的问题，重写了一个函数，解决了，现在已经不崩溃了回复 5# huangbanban


   

smillers

好像是strstr()函数的问题你，重写了一个，解决了，现在已经不崩溃了回复 5# huangbanban


   

huangbanban

恭喜恭喜，呵呵

ysprogram

大哥，strstr是什么问题呀？