linux内核中怎样根据端口号获得与之对应sock

求大神帮帮忙

刚刚接触linux内核编程，在练习写一个netfilter的模块：通过local_in和local_out两个点来截取数据包，之后从数据包里面获取struct sock的指针。
在local_out里面可以在sk_buff直接获取。但是，local_in就有麻烦了，sk_buff里面的sk成员因为还在ip层，没有填写（为NULL）。
试过：分协议，利用IP层以上的函数来查找。以TCP为例用inet_lookup函数获取，但是得到的还是null。
求解：有什么办法可以解决？

求大神帮帮忙

二楼附上代码：

1. #include <linux/kernel.h>
   
2. #include <linux/module.h>
   
3. #include <linux/netfilter.h>
   
4. #include <linux/netfilter_ipv4.h>
   
5. #include <linux/skbuff.h>
   
6. #include <linux/tcp.h>
   
7. #include <linux/udp.h>
   
8. #include <net/tcp.h>
   
9. #include <net/inet_hashtables.h>
   
10. #include <linux/ip.h>
    
11. #include <linux/net.h>
    
12. #include <net/sock.h>
    
13. 
    
14. static struct nf_hook_ops nfno;
    
15. static struct nf_hook_ops nfno2;
    
16. 
    
17. static unsigned int my_hook_fun(unsigned int hooknum,
    
18.                                         struct sk_buff **skb,
    
19.                                         const struct net_device *in,
    
20.                                         const struct net_device *out,
    
21.                                         int (*okfn)(struct sk_buff *))
    
22. {
    
23.         struct iphdr * iph = (*skb)->nh.iph;
    
24.         struct tcphdr * tcph = (struct tcphdr *)((__u32*)iph+iph->ihl);
    
25.         struct udphdr * udph = (struct udphdr *)((__u32*)iph+iph->ihl);
    
26.         struct sock *sk;
    
27.         if(hooknum == NF_IP_LOCAL_OUT)
    
28.         {
    
29.                 printk("send a pack\n");
    
30.         }
    
31.         else
    
32.         {
    
33.                 printk("recv a pack\n");
    
34.         }
    
35.         sk = (*skb)->sk;
    
36.         if(iph->protocol == 6 )
    
37.         {
    
38.                 printk("tcp,source port %d , dst port %d\n",ntohs(tcph->source),ntohs(tcph->dest));
    
39.                 if(sk==NULL)
    
40.                 {
    
41.                         sk = inet_lookup(&tcp_hashinfo,iph->saddr,tcph->source,iph->daddr,ntohs(tcph->dest),inet_iif(*skb));
    
42.                        
    
43.                 }
    
44.        
    
45.         }
    
46.         else if(iph->protocol == 17)
    
47.         {
    
48.                 printk("udp,source prot %d, dst prot %d\n",ntohs(udph->source),ntohs(udph->dest));
    
49.         }
    
50.        
    
51.         if(sk!=NULL && sk->sk_socket!=NULL)
    
52.         printk("socket:%x \n",(unsigned int)(sk->sk_socket) );
    
53.         else
    
54.         {
    
55.                 if(sk==NULL)
    
56.                 {
    
57.                         printk("sk\n");
    
58.                 }
    
59.                 else
    
60.                 {
    
61.                         printk("socket\n");
    
62.                 }
    
63.         }
    
64.         return NF_ACCEPT;
    
65. }
    
66. 
    
67. static int __init drop_pack_ini(void)
    
68. {
    
69.         nfno.hook = my_hook_fun;
    
70.         nfno.hooknum = NF_IP_LOCAL_OUT;
    
71.         nfno.pf = PF_INET;
    
72.         nfno.priority = NF_IP_PRI_LAST;
    
73.         nf_register_hook(&nfno);
    
74.         nfno2.hook = my_hook_fun;
    
75.         nfno2.hooknum = NF_IP_LOCAL_IN;
    
76.         nfno2.pf = PF_INET;
    
77.         nfno2.priority = NF_IP_PRI_LAST;
    
78.         nf_register_hook(&nfno2);
    
79.         return 0;
    
80. }
    
81. 
    
82. static void  drop_pack_exe(void)
    
83. {
    
84.         nf_unregister_hook(&nfno);
    
85.         nf_unregister_hook(&nfno2);
    
86. }
    
87. 
    
88. module_init(drop_pack_ini);
    
89. module_exit(drop_pack_exe);
    
90. MODULE_LICENSE("GPL");
    
91. 
    
92. 
    
93. 
    

复制代码

求大神帮帮忙

另外，我用的是2.6.18版本的linux内核源代码

求大神帮帮忙

问题已解决：
sk = inet_lookup(&tcp_hashinfo,iph->saddr,tcph->source,iph->daddr,ntohs(tcph->dest),inet_iif(*skb));
应改为：
sk = inet_lookup(&tcp_hashinfo,iph->saddr,tcph->source,iph->daddr,tcph->dest,inet_iif(*skb));