Wednesday, January 26, 2011

ip_queue的实现分析

ip_queue的实现分析

文章分类:互联网
本文档的Copyleft归yfydz所有,使用GPL发布,可以自由拷贝,转载,转载时请保持文档的完整性,严禁用于任何商业用途。
msn: yfydz_no1@hotmail.com
来源:http://yfydz.cublog.cn

Java代码
  1. 1. 前言  
  2.    
  3. ip_queue 是netfilter提供的将网络数据包从内核传递到用户空间的方法,内核中要提供ip_queue支持,在用户层空间打开一个netlink的 socket后就可以接受内核通过ip_queue所传递来的网络数据包,具体数据包类型可由iptables命令来确定,只要将规则动作设置为 “-j QUEUE”即可。  
  4.    
  5. 之所以要命名为ip_queue,是因为这是一个队列处理过程,iptables规则把指定的包发给QUEUE是一个数据进入队列的过程,而用户空间程序通过netlink socket获取数据包进行裁定,结果返回内核,进行出队列的操作。  
  6.    
  7. 在iptables代码中,提供了libipq库,封装了对ipq的一些操作,用户层程序可以直接使用libipq库函数处理数据。  
  8.    
  9. 2. 用户层接口:libipq  
  10.    
  11. libipq主要是在iptables-<version>/libipq/libipq.c中实现,提供了以下函数:  
  12.    
  13. //建立ipq的handle:  
  14. struct ipq_handle *ipq_create_handle(u_int32_t flags, u_int32_t protocol);  
  15.    
  16. // 释放ipq handle  
  17. int ipq_destroy_handle(struct ipq_handle *h);  
  18.    
  19. // 读取数据到buf中  
  20. ssize_t ipq_read(const struct ipq_handle *h,  
  21.                 unsigned char *buf, size_t len, int timeout);  
  22.    
  23. // 设置ipq拷贝模式  
  24. int ipq_set_mode(const struct ipq_handle *h, u_int8_t mode, size_t len);  
  25.    
  26. // 从buf中解析数据包结构  
  27. ipq_packet_msg_t *ipq_get_packet(const unsigned char *buf);  
  28.    
  29. // 返回包的类型  
  30. int ipq_message_type(const unsigned char *buf);  
  31.    
  32. // 设置对数据包的裁决  
  33. int ipq_set_verdict(const struct ipq_handle *h,  
  34.                     ipq_id_t id,  
  35.                     unsigned int verdict,  
  36.                     size_t data_len,  
  37.                     unsigned char *buf);  
  38.    
  39. 有了libipq,用户层程序就很简单了,libipq.3中提供了一个实例,比较简单,只列出,不再赘述。  
  40. /* 
  41.  * This code is GPL. 
  42.  */  
  43. #include <linux/netfilter.h>  
  44. #include <libipq.h>  
  45. #include <stdio.h>  
  46. #define BUFSIZE 2048  
  47. static void die(struct ipq_handle *h)  
  48. {  
  49.  ipq_perror("passer");  
  50.  ipq_destroy_handle(h);  
  51.  exit(1);  
  52. }  
  53. int main(int argc, char **argv)  
  54. {  
  55.  int status;  
  56.  unsigned char buf[BUFSIZE];  
  57.  struct ipq_handle *h;  
  58.    
  59.  h = ipq_create_handle(0, PF_INET);  
  60.  if (!h)  
  61.   die(h);  
  62.     
  63.  status = ipq_set_mode(h, IPQ_COPY_PACKET, BUFSIZE);  
  64.  if (status < 0)  
  65.   die(h);  
  66.     
  67.  do{  
  68.   status = ipq_read(h, buf, BUFSIZE, 0);  
  69.   if (status < 0)  
  70.    die(h);  
  71.      
  72.   switch (ipq_message_type(buf)) {  
  73.    case NLMSG_ERROR:  
  74.     fprintf(stderr, "Received error message %d\\n",  
  75.             ipq_get_msgerr(buf));  
  76.     break;  
  77.       
  78.    case IPQM_PACKET: {  
  79.     ipq_packet_msg_t *m = ipq_get_packet(buf);  
  80.       
  81.     status = ipq_set_verdict(h, m->packet_id,  
  82.                              NF_ACCEPT, 0, NULL);  
  83.     if (status < 0)  
  84.      die(h);  
  85.     break;  
  86.    }  
  87.      
  88.    default:  
  89.     fprintf(stderr, "Unknown message type!\\n");  
  90.     break;  
  91.   }  
  92.  } while (1);  
  93.    
  94.  ipq_destroy_handle(h);  
  95.  return 0;  
  96. }  
  97.   
  98. 3. 内核:数据包进入队列  
  99.    
  100. 以下内核代码版本为2.4.26。  
  101.    
  102. 在net/core/netfilter.c中的对于要进行动作NF_QUEUE的数据处理流程为:  
  103.    
  104. nf_hook_slow()->nf_queue->queue_handler[pf].outfn  
  105.    
  106. 如果ip_queue模块有效,这个queue_handler[pf].outfn函数实际上是对应ipq_enqueue_packet()函数(net/ipv4/netfilter/ip_queue.c),这是通过下面的函数进行登记的:  
  107.   
  108. /* net/ipv4/netfilter/ip_queue.c */  
  109. ...  
  110.  status = nf_register_queue_handler(PF_INET, ipq_enqueue_packet, NULL);  
  111. ...  
  112.    
  113. ipq_enqueue_packet()函数:  
  114.    
  115. static int  
  116. ipq_enqueue_packet(struct sk_buff *skb, struct nf_info *info, void *data)  
  117. {  
  118.  int status = -EINVAL;  
  119.  struct sk_buff *nskb;  
  120.  struct ipq_queue_entry *entry;  
  121. // copy_mode是一个全局变量,IPQ_COPY_NONE表示还没初始化,数据包会被丢弃  
  122. // 通常要初始化为IPQ_COPY_META(只拷贝META信息到用户层)或  
  123. // IPQ_COPY_PACKET(拷贝全部信息到用户层)  
  124.  if (copy_mode == IPQ_COPY_NONE)  
  125.   return -EAGAIN;  
  126.    
  127. // 记录数据包的相关信息,包括其路由信息  
  128.  entry = kmalloc(sizeof(*entry), GFP_ATOMIC);  
  129.  if (entry == NULL) {  
  130.   printk(KERN_ERR "ip_queue: OOM in ipq_enqueue_packet()\n");  
  131.   return -ENOMEM;  
  132.  }  
  133.  entry->info = info;  
  134.  entry->skb = skb;  
  135.    
  136.  if (entry->info->hook == NF_IP_LOCAL_OUT) {  
  137. // 在OUTPUT点进行QUEUE时记录相关路由信息:TOS,源、目的IP  
  138.   struct iphdr *iph = skb->nh.iph;  
  139.   entry->rt_info.tos = iph->tos;  
  140.   entry->rt_info.daddr = iph->daddr;  
  141.   entry->rt_info.saddr = iph->saddr;  
  142.  }  
  143.    
  144. // 生成一个新的skb包,该包中保存关于entry的信息,其数据部分是准备传递给用户  
  145. // 空间的数据结构也就是libipq所读取的数据内容,如果拷贝模式是IPQ_COPY_META,  
  146. // 只包含ipq数据头信息;如果是IPQ_COPY_PACKET,在ipq数据头后拷贝整个skb包  
  147. // IP数据信息  
  148.  nskb = ipq_build_packet_message(entry, &status);  
  149.  if (nskb == NULL)  
  150.   goto err_out_free;  
  151.     
  152.  write_lock_bh(&queue_lock);  
  153.    
  154.  if (!peer_pid)  
  155.   goto err_out_free_nskb;  
  156.   /* netlink_unicast will either free the nskb or attach it to a socket */  
  157. // 将该skb附加到用户层打开的netlink socket上,放到其等待队列中,如果不成功这丢弃该包  
  158. // ipqnl是ip_queue对应的netlink sock  
  159. // peer_pid用户空间程序的pid  
  160.  status = netlink_unicast(ipqnl, nskb, peer_pid, MSG_DONTWAIT);  
  161.  if (status < 0)  
  162.   goto err_out_unlock;  
  163.    
  164. // 将entry信息入QUEUE队列,等待用户层的处理结果,如果队列满则丢弃该包  
  165.  status = __ipq_enqueue_entry(entry);  
  166.  if (status < 0)  
  167.   goto err_out_unlock;  
  168.  write_unlock_bh(&queue_lock);  
  169.  return status;  
  170.    
  171. err_out_free_nskb:  
  172.  kfree_skb(nskb);  
  173.    
  174. err_out_unlock:  
  175.  write_unlock_bh(&queue_lock);  
  176. err_out_free:  
  177.  kfree(entry);  
  178.  return status;  
  179. }  
  180.    
  181. 所附加的META数据是这样一个结构:  
  182.    
  183. /* include/linux/netlink.h */  
  184. struct nlmsghdr  
  185. {  
  186.  __u32  nlmsg_len; /* Length of message including header */  
  187.  __u16  nlmsg_type; /* Message content */  
  188.  __u16  nlmsg_flags; /* Additional flags */  
  189.  __u32  nlmsg_seq; /* Sequence number */  
  190.  __u32  nlmsg_pid; /* Sending process PID */  
  191. };  
  192.    
  193. 一 旦数据进入了netlink sock的输入队列中,用户层对数据的读取就由netlink sock来处理了,ip_queue就不再管 了,ip_queue只需要处理从用户层发来的数据,从用户层看是对netlink socket的写,从内核的ip_queue看是用户层数据的数据读 取过程。  
  194.   
  195. 4. 内核:读取用户层数据  
  196.    
  197. ip_queue要读取netlink socket中返回的处理数据结果,函数流程为:  
  198.    
  199. ipq_rcv_sk()  
  200.    |  
  201.    V  
  202. ipq_rcv_skb()  
  203.    |  
  204.    V  
  205. ipq_receive_peer()  
  206.    |  
  207.    |------------------------------+  
  208.    V                              V  
  209. ipq_set_verdict()             ipq_set_mode()  
  210.    |                              |  
  211.    V                              V  
  212. ipq_find_dequeue_entry()      __ipq_set_mode()  
  213. ipq_issue_verdict()    
  214.    |  
  215.    V  
  216. nf_reinject()  
  217.   
  218. 在模块初始化时建立netlink sock:  
  219.    
  220.  ipqnl = netlink_kernel_create(NETLINK_FIREWALL, ipq_rcv_sk);  
  221.    
  222. 其接收数据函数为ipq_rcv_sk():  
  223.    
  224. static void  
  225. ipq_rcv_sk(struct sock *sk, int len)  
  226. {  
  227.  do {  
  228.   struct sk_buff *skb;  
  229.   if (down_trylock(&ipqnl_sem))  
  230.    return;  
  231. // 从sock的等待队列中取出skb  
  232.   while ((skb = skb_dequeue(&sk->receive_queue)) != NULL) {  
  233. // 接收skb内容,skb中的数据格式和发送skb到ipq的格式是一样的,前面是ipq的  
  234. // 控制头,即META部分, 后面才是真正的skb中的数据  
  235.    ipq_rcv_skb(skb);  
  236. // 丢弃skb包,这个skb本来就不是正常的网络skb,而是ipq通信的skb  
  237.    kfree_skb(skb);  
  238.   }  
  239.     
  240.   up(&ipqnl_sem);  
  241.  } while (ipqnl && ipqnl->receive_queue.qlen);  
  242. }  
  243.    
  244. ipq_rcv_skb()函数本身都是再为ipq_receive_peer()函数作准备,忽略;  
  245.   
  246. ipq_receive_peer函数:  
  247.    
  248. static int  
  249. ipq_receive_peer(struct ipq_peer_msg *pmsg,  
  250.                  unsigned char type, unsigned int len)  
  251. {  
  252.  int status = 0;  
  253.  if (len < sizeof(*pmsg))  
  254.   return -EINVAL;  
  255.  switch (type) {  
  256. // 设置IPQ的拷贝模式:IPQ_COPY_META or IPQ_COPY_PACKET  
  257.  case IPQM_MODE:  
  258.   status = ipq_set_mode(pmsg->msg.mode.value,  
  259.                         pmsg->msg.mode.range);  
  260.   break;  
  261. // 处理数据包的裁决    
  262.  case IPQM_VERDICT:  
  263.   if (pmsg->msg.verdict.value > NF_MAX_VERDICT)  
  264.    status = -EINVAL;  
  265.   else  
  266.    status = ipq_set_verdict(&pmsg->msg.verdict,  
  267.                             len - sizeof(*pmsg));  
  268.    break;  
  269.  default:  
  270.   status = -EINVAL;  
  271.  }  
  272.  return status;  
  273. }  
  274.    
  275. ipq_set_verdict()函数:  
  276.    
  277. static int  
  278. ipq_set_verdict(struct ipq_verdict_msg *vmsg, unsigned int len)  
  279. {  
  280.  struct ipq_queue_entry *entry;  
  281.  if (vmsg->value > NF_MAX_VERDICT)  
  282.   return -EINVAL;  
  283. // 根据包的ID找出以前放入QUEUE队列中的ipq_queue_entry结构,该结构保存  
  284. // 最初的skb包的地址  
  285.  entry = ipq_find_dequeue_entry(id_cmp, vmsg->id);  
  286.  if (entry == NULL)  
  287.   return -ENOENT;  
  288.  else {  
  289.   int verdict = vmsg->value;  
  290.     
  291.   if (vmsg->data_len && vmsg->data_len == len)  
  292. // 如果数据被用户层修改,将修改后的信息替换skb中原来的信息  
  293.    if (ipq_mangle_ipv4(vmsg, entry) < 0)  
  294.     verdict = NF_DROP;  
  295. // 最终进行裁定  
  296.   ipq_issue_verdict(entry, verdict);  
  297.   return 0;  
  298.  }  
  299. }  
  300.    
  301. ipq_issue_verdict()函数:  
  302.    
  303. static void  
  304. ipq_issue_verdict(struct ipq_queue_entry *entry, int verdict)  
  305. {  
  306. // 所有QUEUE的包都要由该函数返回netfilter,在net/core/netfilter.c中定义  
  307.  nf_reinject(entry->skb, entry->info, verdict);  
  308.  kfree(entry);  
  309. }  
  310.    
  311. 5. 结论  
  312.    
  313. ip_queue工具的提供使得很多在内核里不太容易实现的功能可以放到用户层空间内实现,处理安全性高,毕竟内核中的错误会导致系统崩溃,而用户层程序的出错不影响系统的整体运行,当然这是以性能降低为代价的。  
  314.    
  315. ip_queue队列实现是使用queue_handler的,queue_handler对于每个协议族只支持一个队列,所以如果有两个需要使用queue功能的应用就会发生冲突,如实现QoS的IMQ也使用这个队列,因此两者在内核中是不能共存的。 

No comments:

Post a Comment