linux kernel socket实例分析

 代码转自：http://blog.chinaunix.net/uid-26748923-id-3548615.html

#include <asm/atomic.h>
#include <asm/checksum.h>
#include <asm/unaligned.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/net.h>
#include <linux/in.h>
#include <linux/types.h>
#include <linux/kthread.h>
#include <linux/wait.h>
#include <linux/skbuff.h>
#include <linux/string.h>
#include <linux/sysctl.h>
#include <linux/netfilter.h>
#include <linux/netfilter_ipv4.h>
#include <linux/ip.h>
#include <linux/workqueue.h>
#include <linux/jiffies.h>
#include <linux/route.h>
#include <linux/stddef.h>
#include <linux/mutex.h>
#include <linux/icmp.h>
#include <net/net_namespace.h>
#include <net/sock.h>
#include <net/route.h>
#include <net/inet_connection_sock.h>
#include <net/request_sock.h>

#define err(msg) printk(KERN_ERR "%s failed.\n", msg)
#define SA struct sockaddr

static struct task_struct *task;

static int threadfn(void *data)
{
    struct socket *sock;
    struct sockaddr_in addr;

    //linux 4.0以前创建socket的函数，4.0之后多了一个参数&init_net
    if (sock_create_kern(PF_INET, SOCK_STREAM, IPPROTO_TCP, &sock) < 0) { 
        err("sock_create_kern");
        goto out;
    }

    sock->sk->sk_reuse = 1;

    memset(&addr, '\0', sizeof(addr));    
    addr.sin_family     = AF_INET;
    addr.sin_addr.s_addr     = htonl(INADDR_ANY);
    addr.sin_port        = htons(8888);

    if (kernel_bind(sock, (SA *)&addr, sizeof(addr)) < 0) {
        err("kernel_bind");
        goto err;
    }

    if (kernel_listen(sock, 1024) < 0) {
        err("kernel_listen");
        goto err;
    }

    while (!kthread_should_stop()) {

        struct socket *newsock;
        struct inet_connection_sock *icsk = inet_csk(sock->sk);
    
        wait_event_interruptible(*sock->sk->sk_sleep, 
                !reqsk_queue_empty(&icsk->icsk_accept_queue) || kthread_should_stop());

        if (!reqsk_queue_empty(&icsk->icsk_accept_queue)) {
            if (kernel_accept(sock, &newsock, 0) != 0) {
                err("kernel_accept");
                goto err;
            }

            printk(KERN_INFO "accept a new connection request.\n");
            sock_release(newsock);
        }

    }

err:
    sock_release(sock);
out:
    return 0;
}

static int __init main_init(void)
{
    task = kthread_run(threadfn, NULL, "listen thread");

    return 0;
}

static void __exit main_exit(void)
{
    kthread_stop(task);
}

module_init(main_init);
module_exit(main_exit);
MODULE_LICENSE("GPL");

wait_event_interruptible(queue, condition)

1 queue 的中断队列里面有对应的中断产生

在linux kernel 里面 , wait.h 里面有个宏定义 ： DECLARE_WAIT_QUEUE_HEAD  就是一个wait 等待的队列结构体

DECLARE_WAIT_QUEUE_HEAD queue 就是定义了一个等待的队列

2 condition设定条件符合

accept()实际要做的事件并不多，它的作用是返回一个已经建立连接的socket(即经过了三次握手)，这个过程是异步的，accept()并不亲自去处理三次握手过程，而只是监听icsk_accept_queue队列，当有socket经过了三次握手，它就会被加到icsk_accept_queue中，所以accept要做的就是等待队列中插入socket，然后被唤醒并返回这个socket。而三次握手的过程完全是协议栈本身去完成的。换句话说，协议栈相当于写者，将socket写入队列，accept()相当于读者，将socket从队列读出。这个过程从listen就已开始，所以即使不调用accept()，客户仍可以和服务器建立连接，但由于没有处理，队列很快会被占满。

if (reqsk_queue_empty(&icsk->icsk_accept_queue)) {  

        ......

}

协议栈向队列中加入socket的过程就是完成三次握手的过程，客户端通过向已知的listen fd发起连接请求，对于到来的每个连接，都会创建一个新的sock，当它经历了TCP_SYN_RCV -> TCP_ESTABLISHED后，就会被添加到icsk_accept_queue中，而监听的socket状态始终为TCP_LISTEN，保证连接的建立不会影响socket的接收。