深入理解TCP协议及其源代码

// 根据alloc_slab来决定是否为其分配告诉缓存
int proto_register(struct proto *prot, int alloc_slab)
{
    if (alloc_slab) {  // 表明注册协议时要分配slab缓存
        prot->slab = kmem_cache_create(prot->name, prot->obj_size, 0,     // 由kmem_cache_create()完成TCP协议的slab缓存建立；
                    SLAB_HWCACHE_ALIGN | prot->slab_flags,
                    NULL);

        if (prot->slab == NULL) { 
            pr_crit("%s: Can‘t create sock SLAB cache!\n",
                prot->name);
            goto out;
        }

        if (prot->rsk_prot != NULL) {
            prot->rsk_prot->slab_name = kasprintf(GFP_KERNEL, "request_sock_%s", prot->name);  // 申请内存空间
            if (prot->rsk_prot->slab_name == NULL)
                goto out_free_sock_slab;

            prot->rsk_prot->slab = kmem_cache_create(prot->rsk_prot->slab_name,  // 创建缓存
                                 prot->rsk_prot->obj_size, 0,
                                 SLAB_HWCACHE_ALIGN, NULL);

            if (prot->rsk_prot->slab == NULL) {
                pr_crit("%s: Can‘t create request sock SLAB cache!\n",
                    prot->name);
                goto out_free_request_sock_slab_name;
            }
        }

        if (prot->twsk_prot != NULL) {
            prot->twsk_prot->twsk_slab_name = kasprintf(GFP_KERNEL, "tw_sock_%s", prot->name);

            if (prot->twsk_prot->twsk_slab_name == NULL)
                goto out_free_request_sock_slab;

            prot->twsk_prot->twsk_slab =
                kmem_cache_create(prot->twsk_prot->twsk_slab_name,
                          prot->twsk_prot->twsk_obj_size,
                          0,
                          SLAB_HWCACHE_ALIGN |
                            prot->slab_flags,
                          NULL);
            if (prot->twsk_prot->twsk_slab == NULL)
                goto out_free_timewait_sock_slab_name;
        }
    }

    mutex_lock(&proto_list_mutex);
    // proto_list在/proc/net/protocols文件中输出当前系统支持的协议
    list_add(&prot->node, &proto_list);   // 将tcp_prot添加到proto_list链表上，等下一次; proto_list是一個全局静态链表，inet域支持的所有协议都在这里链表中
    assign_proto_idx(prot);　　　　　　　　  
    mutex_unlock(&proto_list_mutex);
    return 0;

out_free_timewait_sock_slab_name:
    kfree(prot->twsk_prot->twsk_slab_name);
out_free_request_sock_slab:
    if (prot->rsk_prot && prot->rsk_prot->slab) {
        kmem_cache_destroy(prot->rsk_prot->slab);
        prot->rsk_prot->slab = NULL;
    }
out_free_request_sock_slab_name:
    if (prot->rsk_prot)
        kfree(prot->rsk_prot->slab_name);
out_free_sock_slab:
    kmem_cache_destroy(prot->slab);
    prot->slab = NULL;
out:
    return -ENOBUFS;
}
EXPORT_SYMBOL(proto_register);

struct proto tcp_prot = {
    .name            = "TCP",           
    .owner            = THIS_MODULE,
    .close            = tcp_close,
    .connect        = tcp_v4_connect,
    .disconnect        = tcp_disconnect,
    .accept            = inet_csk_accept,
    .ioctl            = tcp_ioctl,
    .init            = tcp_v4_init_sock,    
    .destroy        = tcp_v4_destroy_sock,
    .shutdown        = tcp_shutdown,
    .setsockopt        = tcp_setsockopt,
    .getsockopt        = tcp_getsockopt,
    .recvmsg        = tcp_recvmsg,
    .sendmsg        = tcp_sendmsg,
    .sendpage        = tcp_sendpage,
    .backlog_rcv        = tcp_v4_do_rcv,
    .release_cb        = tcp_release_cb,
    .hash            = inet_hash,
    .unhash            = inet_unhash,
    .get_port        = inet_csk_get_port,
    .enter_memory_pressure    = tcp_enter_memory_pressure,
    .stream_memory_free    = tcp_stream_memory_free,
    .sockets_allocated    = &tcp_sockets_allocated,
    .orphan_count        = &tcp_orphan_count,
    .memory_allocated    = &tcp_memory_allocated,
    .memory_pressure    = &tcp_memory_pressure,
    .sysctl_mem        = sysctl_tcp_mem,
    .sysctl_wmem        = sysctl_tcp_wmem,
    .sysctl_rmem        = sysctl_tcp_rmem,
    .max_header        = MAX_TCP_HEADER,
    .obj_size        = sizeof(struct tcp_sock),
    .slab_flags        = SLAB_DESTROY_BY_RCU,
    .twsk_prot        = &tcp_timewait_sock_ops,
    .rsk_prot        = &tcp_request_sock_ops,
    .h.hashinfo        = &tcp_hashinfo,
    .no_autobind        = true,
#ifdef CONFIG_COMPAT
    .compat_setsockopt    = compat_tcp_setsockopt,
    .compat_getsockopt    = compat_tcp_getsockopt,
#endif
#ifdef CONFIG_MEMCG_KMEM
    .init_cgroup        = tcp_init_cgroup,
    .destroy_cgroup        = tcp_destroy_cgroup,
    .proto_cgroup        = tcp_proto_cgroup,
#endif
};
EXPORT_SYMBOL(tcp_prot);

/* thinking of making this const? Don‘t.
 * early_demux can change based on sysctl.
 */
static struct net_protocol tcp_protocol = {
    .early_demux    =    tcp_v4_early_demux,
    .early_demux_handler =  tcp_v4_early_demux,
    .handler    =    tcp_v4_rcv,
    .err_handler    =    tcp_v4_err,
    .no_policy    =    1,
    .netns_ok    =    1,
    .icmp_strict_tag_validation = 1,
};