内核模块中创建tcp连接

asert

在网上到处搜不到如何在内核中创建tcp连接，后来自己参考内核源代码中的linux/net/sunrpc/xprtsock.c写了一个可用的测试程序，与大家分享,代码有可以改进的地方还请高手都指教呀^_^

2. /* code is referred to linux/net/sunrpc/xprtsock.c */
4. #include <linux/module.h>
5. #include <linux/kernel.h>
6. #include <linux/init.h>
7. #include <linux/types.h>
8. #include <linux/errno.h>
9. #include <linux/socket.h>
10. #include <linux/in.h>
11. #include <linux/inet.h>
12. #include <linux/net.h>
13. #include <linux/tcp.h>
14. #include <net/sock.h>
15. #include <net/checksum.h>
16. #include <net/tcp.h>
18. static char *peer_addr = "192.168.0.100";
19. static unsigned short peer_port = 21;
20. static char *msg = "USER anonymous\r\n";
22. static struct socket *sock = NULL;
23. static struct work_struct work;
24. static int recvd = 0;
26. static int tcp_sendto(struct socket *sock, void * buff, size_t len,
27.                 unsigned flags, struct sockaddr *addr, int addr_len) {
28.   struct kvec vec;
29.   struct msghdr msg;
31.   vec.iov_base=buff;
32.   vec.iov_len=len;
34.   memset(&msg, 0x00, sizeof(msg));
35.   msg.msg_name=addr;
36.   msg.msg_namelen=addr_len;
37.   msg.msg_flags = flags | MSG_DONTWAIT | MSG_NOSIGNAL;
39.   return kernel_sendmsg(sock, &msg, &vec, 1, len);
40. }
42. static void kernsock_send_msg(void) {
43.         int n;
44.   struct sockaddr_in addr;
45.   
46.   memset(&addr, 0x00, sizeof(addr));
47.   addr.sin_family = AF_INET;
48.   addr.sin_port = htons(peer_port);
49.   addr.sin_addr.s_addr = in_aton(peer_addr);
50.   n = tcp_sendto(sock, msg, strlen(msg), 0, (struct sockaddr *)&addr, sizeof(addr));
51.   printk("send %d bytes\n", n);
52. }
54. static int kernsock_tcp_data_recv(read_descriptor_t *rd_desc, struct sk_buff *skb, unsigned int offset, size_t len) {
55.         struct iphdr *ip;
56.         struct tcphdr *tcp;
57.         char *tcp_data;
58.         int        payload,doff;
59.                
60.         printk("kernsock_tcp_data_recv started:offset:%d,len:%d\n",offset,len);
61.        
62.         if (!skb) return 0;
63.        
64.         if (!(skb->nh.iph)) return 0;
65.        
66.         ip = skb->nh.iph;
67.         if(ip->protocol == IPPROTO_TCP) {
68.                 tcp = (struct tcphdr *)(skb->h.th);
69.                 printk("src=%u.%u.%u.%u:%u,dest=%u.%u.%u.%u:%u\n",NIPQUAD(ip->saddr),ntohs(tcp->source),NIPQUAD(ip->daddr),ntohs(tcp->dest));
70.                
71.                 payload = ntohs(ip->tot_len) - (ip->ihl << 2);
72.                 doff = tcp->doff << 2;
73.                 tcp_data = (char *)((int)tcp + doff);
74.                        
75.                 if(tcp->psh) {
76.                         printk("peer sent psh:%.*s\n",payload - doff,tcp_data);
77.                         recvd = 1;
78.                 }
79.         }
80.         printk("kernsock_tcp_data_recv done\n");
81.         return len;
82. }
84. static void kernsock_tcp_data_ready(struct sock *sk, int bytes) {
85.         read_descriptor_t rd_desc;
87.         read_lock(&sk->sk_callback_lock);
88.         printk("kernsock_tcp_data_ready\n");
90.         rd_desc.arg.data = NULL;
91.         rd_desc.count = 65536;
92.         tcp_read_sock(sk, &rd_desc, kernsock_tcp_data_recv);
94.         read_unlock(&sk->sk_callback_lock);
95. }
97. static void kernsock_tcp_state_change(struct sock *sk) {
98.         read_lock(&sk->sk_callback_lock);
100.         switch (sk->sk_state) {
101.         case TCP_ESTABLISHED:
102.                 printk("TCP_ESTABLISHED\n");
103.                 break;
104.         case TCP_SYN_SENT:
105.                 printk("TCP_SYN_SENT\n");
106.                 break;
107.         case TCP_SYN_RECV:
108.                 printk("TCP_SYN_RECV\n");
109.                 break;
110.         case TCP_CLOSE_WAIT:
111.                 printk("TCP_CLOSE_WAIT\n");
112.                 break;
113.         }
114.         read_unlock(&sk->sk_callback_lock);
115. }
117. static void kernsock_tcp_write_space(struct sock *sk) {
118.         read_lock(&sk->sk_callback_lock);
119.         printk("kernsock_tcp_write_space\n");
120.         read_unlock(&sk->sk_callback_lock);
121. }
123. static void tcp_connect_worker(struct sockaddr_in *addr) {
124.         int err, status = -EIO;
125.         struct sock *sk;
126.        
127.         /* start from scratch */
128.         if ((err = sock_create_kern(PF_INET, SOCK_STREAM, IPPROTO_TCP, &sock)) < 0) {
129.                 printk("can't create TCP transport socket (%d).\n", -err);
130.                 return;
131.         }
133.         sk = sock->sk;
134.         write_lock_bh(&sk->sk_callback_lock);
135.         sk->sk_data_ready = kernsock_tcp_data_ready;
136.         sk->sk_state_change = kernsock_tcp_state_change;
137.         sk->sk_write_space = kernsock_tcp_write_space;
138.         sk->sk_allocation = GFP_ATOMIC;
140.         /* socket options */
141.         sk->sk_userlocks |= SOCK_BINDPORT_LOCK;
142.         sock_reset_flag(sk, SOCK_LINGER);
143.         tcp_sk(sk)->linger2 = 0;
144.         tcp_sk(sk)->nonagle |= TCP_NAGLE_OFF;
145.         write_unlock_bh(&sk->sk_callback_lock);
147.         status = sock->ops->connect(sock, (struct sockaddr *) addr, sizeof(struct sockaddr_in), O_NONBLOCK);
149.         if (status < 0) {
150.                 switch (status) {
151.                         case -EINPROGRESS:
152.                                 printk("EINPROGRESS\n");
153.                                 break;
154.                         case -EALREADY:
155.                                 printk("EALREADY\n");
156.                                 break;
157.                         case -ECONNREFUSED:
158.                                 printk("ECONNREFUSED\n");
159.                                 break;
160.                         case -ECONNRESET:
161.                                 printk("ECONNRESET\n");
162.                                 break;
163.                 }
164.         }
165. }
167. static int __init socket_init(void){
168.     struct sockaddr_in addr;
169.    
170.     memset(&addr, 0x00, sizeof(addr));
171.     addr.sin_family = AF_INET;
172.     addr.sin_port = htons(peer_port);
173.     addr.sin_addr.s_addr = in_aton(peer_addr);
174.     tcp_connect_worker(&addr);
175.    
176.     msleep(1000);
177.     if(recvd) {
178.             INIT_WORK(&work, kernsock_send_msg, NULL);
179.                   schedule_work(&work);
180.           } else {
181.                   printk("time out\n");
182.           }
183.     return 0;
184. }
186. static void __exit socket_exit(void) {
187.     sock_release(sock);
188. }
190. module_init(socket_init);
191. module_exit(socket_exit);

复制代码

realtang

没有提供Makefile？？

asert

Post by realtang;1858581
没有提供Makefile？？

2. obj-m := kernsock.o
4. all:make -C /usr/src/linux-2.6-2.6.18/ M=`pwd` modules

复制代码

直接make命令就可以了
然后insmod kernsock.ko后就开始测试了
卸载用rmmod kersock.ko

代码中的这几个变量需要自己修改一下
static char *peer_addr = "192.168.0.100";
static unsigned short peer_port = 21;
static char *msg = "USER anonymous\r\n";

realtang

如果被连接机器不响应，会不会导致内核被挂起，死机？

asert

Post by realtang;1858599
如果被连接机器不响应，会不会导致内核被挂起，死机？

呵呵，安全起见测试内核模块还是用windows + vmware，测试起来感觉很不错，不用再提心吊胆,呵呵

happytianli

楼主强人，我现在也在做内核socket的编程，找了很多例子，不过也没有合适的，楼主的内核socket客户端的代码我试过了，可以向用户的server端发代码，不知楼主是否也做了内核socket 服务器端的代码，如果有，麻烦楼主提供下，多谢了，我的邮箱  happytianli@cse.buaa.edu.cn