问关于Godbach兄的通信代码的问题

kidexp

我按照Godbach兄的代码，写了一个关于TCP包计数的程序，只是单纯记录TCP包的数量，
在下载小容量文件<=0.5k左右时可以实现很快的内核用户态交互通信，
但略微大一点的文件1.5k左右就不可以了...或者说效率太低，文件下载不了，想请教下原因
还有在linux里面对文件进行些操作用write()，它所需要的头文件是fctnl.h?
如果是，为什么我编译的时候说 write未定义?
谢谢
代码如下 ：

1. #include <stdio.h>
   
2. #include <stdlib.h>
   
3. #include <string.h>
   
4. #include <signal.h>
   
5. #include <netinet/tcp.h>
   
6. #include <netinet/ip.h>
   
7. #include <arpa/inet.h>
   
8. #include "libipq.h"
   
9. #include <fcntl.h>
   
10. #include </home/kide/clamav/libclamav/clamav.h>
    
11. 
    
12. #define ETH_HDRLEN 14
    
13. #define NF_ACCEPT         1
    
14. #define NF_DROP          0
    
15. 
    
16. 
    
17. struct ipq_handle *h = NULL;
    
18. 
    
19. static void sig_int(int signo)
    
20. {
    
21.         ipq_destroy_handle(h);
    
22.         printf("Exit: %s\n", ipq_errstr());
    
23.         exit(0);
    
24. }
    
25. 
    
26. int main(void)
    
27. {
    
28.         unsigned char buf[1024];
    
29.         /* creat handle*/
    
30.         h = ipq_create_handle(0, PF_INET);
    
31.         if(h == NULL){
    
32.                 printf("%s\n", ipq_errstr());
    
33.                 return 0;
    
34.         }
    
35.         printf("ipq_creat_handle success!\n");
    
36.         /*set mode*/
    
37.         unsigned char mode = IPQ_COPY_PACKET;
    
38.         int range = sizeof(buf);
    
39.         int ret = ipq_set_mode(h, mode, range);
    
40.         printf("ipq_set_mode: send bytes =%d, range=%d\n", ret, range);
    
41.        
    
42.         /*register signal handler*/
    
43.         signal(SIGINT, sig_int);
    
44. 
    
45.         /*read packet from kernel*/
    
46.         int status;
    
47.         struct nlmsghdr *nlh;
    
48.         ipq_packet_msg_t *ipq_packet;
    
49.         int count = 0;
    
50.        
    
51.         while(1){
    
52.                 status = ipq_read(h, buf, sizeof(buf));
    
53.                 if(status > sizeof(struct nlmsghdr))
    
54.                 {
    
55.                         nlh = (struct nlmsghdr *)buf;
    
56.                         ipq_packet = ipq_get_packet(buf);
    
57.                         printf("recv bytes =%d, nlmsg_len=%d, indev=%s, datalen=%d, packet_id=%x\n", status, nlh->nlmsg_len,
    
58.                                         ipq_packet->indev_name,  ipq_packet->data_len, ipq_packet->packet_id);
    
59.                         unsigned char payload[65532];
    
60.                         memset(payload, 0x00, sizeof(payload));
    
61.                         memcpy(payload + ETH_HDRLEN, ipq_packet->payload, ipq_packet->data_len);
    
62.                         /*display packet data in hex including 14 bytes of eth hdr(set by 0x00)*/
    
63.                         int i;
    
64.                         for(i = 0; i < ipq_packet->data_len + ETH_HDRLEN; i++){
    
65.                                 if(i%16 == 0)
    
66.                                         printf("00%.2x:  ", i);
    
67.                                 printf("%.2x ", payload[i]);
    
68.                                 if(i % 16 == 15)
    
69.                                         printf("\n");
    
70.                         }
    
71.                         printf("\n");
    
72.                 }
    
73.                 struct iphdr * iph = (struct iphdr *) ipq_packet->payload;
    
74.                 if( iph->protocol == 6)
    
75.                 {
    
76.                         //int fd = open("/home/kide/temp",O_WRONLY);
    
77.                         unsigned char *data= ipq_packet->payload + sizeof(struct iphdr) + sizeof(struct tcphdr);
    
78.                         //long int num=write(fd,data,ipq_packet->data_len-sizeof(struct iphdr) - sizeof(struct tcphdr));
    
79.                         printf("%dbytes has been written \n",num);
    
80.                         if(ipq_set_verdict(h, ipq_packet->packet_id, NF_ACCEPT,ipq_packet->data_len,ipq_packet->payload) !=-1)
    
81.                         count++;
    
82.                         printf("%dTCP success\n",count);
    
83.                 }
    
84.                 else
    
85.                 {
    
86.                         if(ipq_set_verdict(h, ipq_packet->packet_id, NF_ACCEPT,ipq_packet->data_len,ipq_packet->payload) !=-1)
    
87.                         printf("OTHERS        success\n");
    
88.                 }
    
89.                
    
90.         }
    
91.         return 0;
    
92. }

复制代码

[ 本帖最后由 Godbach 于 2009-8-18 16:38 编辑 ]

Godbach

                        for(i = 0; i < ipq_packet->data_len + ETH_HDRLEN; i++){
                                if(i%16 == 0)
                                        printf("00%.2x:  ", i);
                                printf("%.2x ", payload);
                                if(i % 16 == 15)
                                        printf("\n");
                        }

这段代码是显示报文内容的，每个包都打印到终端上，肯定效率不会搞了。你把这段代码屏蔽掉，然后试一下。

Godbach

我的示例中只是为了演示，所以把报文的全部内容都打出来了，你需要根据自己的实际情况，来选择打印哪些内容。

kidexp

改了之后还是不能下载... 我下载300多字节的文件才传了4个TCP包，但是下这个1.7KB的文件却收了500多数据包，还没有下载成功，浏览器未跳出下载窗口，这是什么原因？

Godbach

你这里对报文做了计数之后，就告诉内核ACCEPT了吗？

Godbach

帮你编辑了一下代码，这样看起来舒服些。

kidexp

从代码上看，应该是先告诉内核ACCEPT计数
怎么作出那个CODE的效果？

Godbach

增加统计计数，然后告诉内核ACCEPT啊。你指的CODE效果是什么意思

kidexp

就是你编辑我帖子中代码后代码会在框中显示
为什么要先计数？

kidexp

感觉数据包从核心态到用户态，返回核心态，效率太低下了吧？
还有每次用open打开文件,用write进行写的时候是不是从文件最后开始写？
我的temp文件只能达到987字节，不知道为什么，这样的大小显然不能满足我的要求....

[ 本帖最后由 kidexp 于 2009-8-18 18:57 编辑 ]