套接字之accept系统调用
accept系统调用完成的工作是，从已经完成三次握手的连接队列中取一个连接，进行后续服务；
调用过程中，会重新分配一个socket，以及关联文件和传输控制块，如果应用层需要获取对端信息，则将对端信息拷贝到用户空间；
1 /* 2 * For accept, we attempt to create a new socket, set up the link 3 * with the client, wake up the client, then return the new 4 * connected fd. We collect the address of the connector in kernel 5 * space and move it to user at the very end. This is unclean because 6 * we open the socket then return an error. 7 * 8 * 1003.1g adds the ability to recvmsg() to query connection pending 9 * status to recvmsg. We need to add that support in a way thats 10 * clean when we restucture accept also. 11 */ 12 13 SYSCALL_DEFINE4(accept4, int, fd, struct sockaddr __user *, upeer_sockaddr, 14 int __user *, upeer_addrlen, int, flags) 15 { 16 struct socket *sock, *newsock; 17 struct file *newfile; 18 int err, len, newfd, fput_needed; 19 struct sockaddr_storage address; 20 21 /* 只支持cloexec和nonblock */ 22 if (flags & ~(SOCK_CLOEXEC | SOCK_NONBLOCK)) 23 return -EINVAL; 24 25 /* 如果阻塞标记不等&& 有阻塞标记，则替换为O_NONBLOCK */ 26 if (SOCK_NONBLOCK != O_NONBLOCK && (flags & SOCK_NONBLOCK)) 27 flags = (flags & ~SOCK_NONBLOCK) | O_NONBLOCK; 28 29 /* 找到socket */ 30 sock = sockfd_lookup_light(fd, &err, &fput_needed); 31 if (!sock) 32 goto out; 33 34 err = -ENFILE; 35 36 /* 分配一个新socket */ 37 newsock = sock_alloc(); 38 if (!newsock) 39 goto out_put; 40 41 /* 类型和操作与本端socket相同 */ 42 newsock->type = sock->type; 43 newsock->ops = sock->ops; 44 45 /* 46 * We don't need try_module_get here, as the listening socket (sock) 47 * has the protocol module (sock->ops->owner) held. 48 */ 49 /* 增加模块引用 */ 50 __module_get(newsock->ops->owner); 51 52 /* 获取新的文件描述符 */ 53 newfd = get_unused_fd_flags(flags); 54 if (unlikely(newfd < 0)) { 55 err = newfd; 56 sock_release(newsock); 57 goto out_put; 58 } 59 60 /* 分配新的文件 */ 61 newfile = sock_alloc_file(newsock, flags, sock->sk->sk_prot_creator->name); 62 if (IS_ERR(newfile)) { 63 err = PTR_ERR(newfile); 64 put_unused_fd(newfd); 65 sock_release(newsock); 66 goto out_put; 67 } 68 69 /* accept检查 */ 70 err = security_socket_accept(sock, newsock); 71 if (err) 72 goto out_fd; 73 74 /* 执行具体的accept操作 */ 75 err = sock->ops->accept(sock, newsock, sock->file->f_flags, false); 76 if (err < 0) 77 goto out_fd; 78 79 /* 指定了获取对端信息 */ 80 if (upeer_sockaddr) { 81 if (newsock->ops->getname(newsock, (struct sockaddr *)&address, 82 &len, 2) < 0) { 83 err = -ECONNABORTED; 84 goto out_fd; 85 } 86 87 /* 拷贝对端地址到用户空间结构 */ 88 err = move_addr_to_user(&address, 89 len, upeer_sockaddr, upeer_addrlen); 90 if (err < 0) 91 goto out_fd; 92 } 93 94 /* File flags are not inherited via accept() unlike another OSes. */ 95 96 /* 绑定描述符和文件 */ 97 fd_install(newfd, newfile); 98 err = newfd; 99 100 out_put: 101 fput_light(sock->file, fput_needed); 102 out: 103 return err; 104 out_fd: 105 fput(newfile); 106 put_unused_fd(newfd); 107 goto out_put; 108 } 109 110 SYSCALL_DEFINE3(accept, int, fd, struct sockaddr __user *, upeer_sockaddr, 111 int __user *, upeer_addrlen) 112 { 113 return sys_accept4(fd, upeer_sockaddr, upeer_addrlen, 0); 114 }
inet_accept函数调用传输层的accept操作，并且返回新的连接控制块，新的连接控制块需要与新的socket进行关联，accept完成，将新socket的状态设置为已连接状态；
1 /* 2 * Accept a pending connection. The TCP layer now gives BSD semantics. 3 */ 4 5 int inet_accept(struct socket *sock, struct socket *newsock, int flags, 6 bool kern) 7 { 8 struct sock *sk1 = sock->sk; 9 int err = -EINVAL; 10 11 /* 执行传输层的accept操作 */ 12 struct sock *sk2 = sk1->sk_prot->accept(sk1, flags, &err, kern); 13 14 if (!sk2) 15 goto do_err; 16 17 lock_sock(sk2); 18 19 /* rps处理 */ 20 sock_rps_record_flow(sk2); 21 WARN_ON(!((1 << sk2->sk_state) & 22 (TCPF_ESTABLISHED | TCPF_SYN_RECV | 23 TCPF_CLOSE_WAIT | TCPF_CLOSE))); 24 25 /* 控制块连接到新的socket */ 26 sock_graft(sk2, newsock); 27 28 /* 设置新socket的状态为连接 */ 29 newsock->state = SS_CONNECTED; 30 err = 0; 31 release_sock(sk2); 32 do_err: 33 return err; 34 } 35 EXPORT_SYMBOL(inet_accept);
TCP层对于accept的实现，请移步另外一篇文章：<TCP层accept系统调用的实现分析>；
