内核中的UDP socket流程（4）——sock_create

内核中的UDP socket流程（4）——sock_create

作者:[e-mail]

原文：http://blog.chinaunix.net/space.php?uid=23629988&do=blog&id=85910

又懒了2天，继续sock_create

    /* Compatibility.

     This uglymoron is moved from INET layer to here to avoid
     deadlock in module load.
     */
    if (family == PF_INET && type == SOCK_PACKET) {
        static int warned;
        if (!warned) {
            warned = 1;
            printk(KERN_INFO "%s uses obsolete (PF_INET,SOCK_PACKET)\n",
             current->comm);
        }
        family = PF_PACKET;
    }

    err = security_socket_create(family, type, protocol, kern);
    if (err)
        return err;

首先是对PF_INET，SOCK_PACKET的兼容性处理。security_socket_create函数是一个空函数，省略。

下面是去申请一个socket node，我们需要进入sock_alloc中看如何分配的节点。

    /*
     *    Allocate the socket and allow the family to set things up. if
     *    the protocol is 0, the family is instructed to select an appropriate
     *    default.
     */
    sock = sock_alloc();
    if (!sock) {
        if (net_ratelimit())
            printk(KERN_WARNING "socket: no more sockets\n");
        return -ENFILE;    /* Not exactly a match, but its the
                 closest posix thing */
    }

下面是sock_alloc的代码

static struct socket *sock_alloc(void)
{
    struct inode *inode;
    struct socket *sock;

    inode = new_inode(sock_mnt->mnt_sb);
    if (!inode)
        return NULL;

    sock = SOCKET_I(inode);

    kmemcheck_annotate_bitfield(sock, type);
    inode->i_mode = S_IFSOCK | S_IRWXUGO;
    inode->i_uid = current_fsuid();
    inode->i_gid = current_fsgid();

    percpu_add(sockets_in_use, 1);
    return sock;
}

其中sock_mnt是一个全局变量，它在sock_init中被初始化的。并挂载到VFS层上。在sock_alloc函数中，首先是从sock_mnt->mnt_sb即socket的super block中申请一个节点。然后通过SOCKET_I宏，取得inode对应的socket的地址。

这里之所以这样做的，原因是因为，inode的分配是统一的，但是每个inode的用途是不一样的。

在socket中，申请的inode的节点，实际上对应的结构应该为下面这个结构。

struct socket_alloc {
    struct socket socket;
    struct inode vfs_inode;
};

new_inode中实际上是申请了一个struct socket_alloc的地址，但是返回的确是socket_all->vs_inode的地址。所以在这里需要使用SOCKET_I宏，来从inode地址中，得到socket的地址。

后面的宏kmemcheck_annotate_bitfield是一个内存检查，略过。然后，设置inode的mode，uid，gid。

percpu_add(sockets_in_use, 1)是增加sockets_in_use的统计计数——这个计数变量是每个cpu的私有变量。

然后，我们就回到__sock_create。

    rcu_read_lock();
    pf = rcu_dereference(net_families[family]);
    err = -EAFNOSUPPORT;
    if (!pf)
        goto out_release;

    /*
     * We will call the ->create function, that possibly is in a loadable
     * module, so we have to bump that loadable module refcnt first.
     */
    if (!try_module_get(pf->owner))
        goto out_release;

    /* Now protected by module ref count */
    rcu_read_unlock();

通过RCU机制，获得pf（family）对应的net_families中的指针。

下面通过调用函数指针调用用户指定family的函数去创建socket。

    err = pf->create(net, sock, protocol, kern);
    if (err < 0)
        goto out_module_put;

对于TCP/IP来说，family是PF_INET。

现在又需要跳到文件linux/net/ipv4/af_inet.c，下面是PF_INET对应的协议域定义

static const struct net_proto_family inet_family_ops = {
    .family = PF_INET,
    .create = inet_create,
    .owner    = THIS_MODULE,
};

所以对于对于TCP/IP的socket来说，sock_create中这里的create函数实际上指向的是inet_create——这个函数明天再说，继续sock_create的代码。

    /*
     * Now to bump the refcnt of the [loadable] module that owns this
     * socket at sock_release time we decrement its refcnt.
     */
    if (!try_module_get(sock->ops->owner))
        goto out_module_busy;

    /*
     * Now that we're done with the ->create function, the [loadable]
     * module can have its refcnt decremented
     */
    module_put(pf->owner);
    err = security_socket_post_create(sock, family, type, protocol, kern);
    if (err)
        goto out_sock_release;
    *res = sock;

    return 0;

这几行代码的注释已经很清楚，到此sock_create已经完结。