Linux 内核网络之 Shutdown 的实现

系统 Linux
shutdown 系统调用关闭连接的读通道、写通道或者读写通道。

shutdown 系统调用关闭连接的读通道、写通道或者读写通道。

对于读通道,shutdown 丢弃所有进程还没有读取的数据以及调用 shutdown 之后到达的数据。

对于写通道,shutdown 使用协议作相应的处理。对于 TCP,所有剩余的数据将被发送,返送完后发送 fin 报文。

以上就是 TCP 的半关闭特点。

asmlinkage long sys_shutdown(int fd, int how)
{
int err, fput_needed;
struct socket *sock;
sock = sockfd_lookup_light(fd, &err, &fput_needed);
if (sock != NULL) {
err = security_socket_shutdown(sock, how);
if (!err)
err = sock->ops->shutdown(sock, how); // inet_shutdown
fput_light(sock->file, fput_needed);
}
return err;
}
int inet_shutdown(struct socket *sock, int how)
{
struct sock sk = sock->sk;
int err = 0;
/ This should really check to make sure
the socket is a TCP socket. (WHY AC...)
/
//how 增1 是为了利用how变量进行为操作
how++; / maps 0->1 has the advantage of making bit 1 rcvs and
1->2 bit 2 snds.
2->3 /
if ((how & ~SHUTDOWN_MASK) || !how) / MAXINT->0 /
return -EINVAL;
lock_sock(sk);
//根据传输控制块的状态重新设置套接口状态,使套接口状态在完成关闭前只有2种
if (sock->state == SS_CONNECTING) {
if ((1 << sk->sk_state) &
(TCPF_SYN_SENT | TCPF_SYN_RECV | TCPF_CLOSE))
sock->state = SS_DISCONNECTING;
else
sock->state = SS_CONNECTED;
}
//若传输控制块处于其他状态,则设置shutdown关闭方式后,调用传输层接口shutdown,进行具体传输层关闭
switch (sk->sk_state) {
case TCP_CLOSE:
err = -ENOTCONN;
/ Hack to wake up other listeners, who can poll for
POLLHUP, even on eg. unconnected UDP sockets -- RR /
default:
sk->sk_shutdown |= how; // 把关闭信息设置到sk_shutdown 中
if (sk->sk_prot->shutdown)
sk->sk_prot->shutdown(sk, how); // tcp_shutdown
break;
/ Remaining two branches are temporary solution for missing
close() in multithreaded environment. It is not a good idea,
but we have no choice until close() is repaired at VFS level.
/
//若处于TCP_LISTEN,则需要判断关闭方式,若有接收方向的关闭操作,则和TCP_SYN_SENT处理一样
case TCP_LISTEN:
if (!(how & RCV_SHUTDOWN))
break;
/ Fall through /
//若处于连接状态过程中(LISTEN或者SYN_SENT状态),则不允许在继续连接,调用disconnect断开连接操作
case TCP_SYN_SENT:
err = sk->sk_prot->disconnect(sk, O_NONBLOCK); // tcp_disconnect
sock->state = err ? SS_DISCONNECTING : SS_UNCONNECTED;
break;
}
/ Wake up anyone sleeping in poll. */
// 调用sk_state_change,唤醒在传输控制块的等待队列上的进程。sk_state_change 在 sock_init_data 中被初始化
sk->sk_state_change(sk);
release_sock(sk);
return err;
}

把关闭方式 设置到套接口 sk_shutdown 中。

若传输控制块处于其他状态,调用 tcp_shutdown 。

若处于正在连接过程中,则调用 tcp_disconnect 断开连接。
void tcp_shutdown(struct sock sk, int how)
{
/ We need to grab some memory, and put together a FIN,
and then put it into the queue to be sent.
Tim MacKenzie(​​tym@dibbler.cs.monash.edu.au​​) 4 Dec '92.
/
/ 不含有SEND_SHUTDOWN,返回,接收方关闭,不发fin */
if (!(how & SEND_SHUTDOWN))
return;
/* If we've already sent a FIN, or it's a closed state, skip this. /
/ 以下这几个状态发fin /
if ((1 << sk->sk_state) &
(TCPF_ESTABLISHED | TCPF_SYN_SENT |
TCPF_SYN_RECV | TCPF_CLOSE_WAIT)) {
/ Clear out any half completed packets. FIN if needed. /
if (tcp_close_state(sk)) / 设置新状态,发送fin */
tcp_send_fin(sk);
}
}
对于关闭读通道,不需要发送 FIN 报文,直接返回。

对于关闭写通道,把未发送出去的数据发送出去之后,发送 FIN 报文。

int tcp_disconnect(struct sock *sk, int flags)
{
struct inet_sock *inet = inet_sk(sk);
struct inet_connection_sock *icsk = inet_csk(sk);
struct tcp_sock *tp = tcp_sk(sk);
int err = 0;
int old_state = sk->sk_state;
/* 不是close状态则设置为close,从hash中删除控制块 */
if (old_state != TCP_CLOSE)
tcp_set_state(sk, TCP_CLOSE);
/* ABORT function of RFC793 /
/ LISTEN状态,停止监听 /
if (old_state == TCP_LISTEN) {
//删除keepalive定时器,对全连接上的每个socket调用tcp_disconnect
inet_csk_listen_stop(sk);
/ 根据状态确定是否需要发送rst
|| 下一个发送序号并不是最后一个队列数据段序号
&& 是被动关闭的结束状态 /
} else if (tcp_need_reset(old_state) ||
(tp->snd_nxt != tp->write_seq &&
(1 << old_state) & (TCPF_CLOSING | TCPF_LAST_ACK))) {
/ The last check adjusts for discrepancy of Linux wrt. RFC
• states
/
/ 发送rst */
tcp_send_active_reset(sk, gfp_any());
sk->sk_err = ECONNRESET;
} else if (old_state == TCP_SYN_SENT)
sk->sk_err = ECONNRESET;
/* 清除定时器 /
tcp_clear_xmit_timers(sk);
/ 释放接收队列中的skb /
__skb_queue_purge(&sk->sk_receive_queue);
/ 释放发送队列中的skb */
sk_stream_writequeue_purge(sk);
/*释放未按顺序达到的skb */
__skb_queue_purge(&tp->out_of_order_queue);
#ifdef CONFIG_NET_DMA
__skb_queue_purge(&sk->sk_async_wait_queue);
#endif
/* 其他各种清理工作 */
inet->dport = 0;
...
return err;
}

对于监听,全连接上的每个 socket 调用 tcp_disconnect,最终会对每个socket 发送 RST 报文。也即是若 shutdown 的是半打开的连接,则发出 RST 来关闭连接。

若 shutdown 的是正常连接,那么关闭读其实与对端是没有关系的。只要本机把接收掉的消息丢掉,其实就等价于关闭读了,并不一定非要对端关闭写的。实际上,shutdown 正是这么干的。若参数中的标志位含有关闭读,只是标识下,当我们调用 read 等方法时这个标识就起作用了,会使进程读不到任何数据。

若参数中有标志位为关闭写,则发出 FIN 包,告诉对方,本机不会再发消息了。

当关闭写通道后,再次调用 send 发送数据时,直接返回失败。

int tcp_sendmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg,
size_t size)
{
...
err = -EPIPE;
if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))
goto do_error;
...
}

同理,关闭读通道后,再次调用 tcp_recvmsg 时不会读取数据,直接返回。

责任编辑:华轩 来源: 今日头条
相关推荐

2023-03-01 23:56:11

2023-03-10 14:56:37

Linuxconnect系统

2023-03-28 15:51:20

2023-03-06 15:43:56

2021-09-17 11:59:21

tcpdump网络包Linux

2021-09-08 10:21:33

内核网络包Tcpdump

2009-07-16 09:02:38

LINUX 2.4.x网络安全LINUX开发

2023-05-12 07:27:24

Linux内核网络设备驱动

2010-07-19 10:05:52

ibmdwLinux

2021-11-15 04:00:07

Linux 内核动态

2021-11-14 07:29:55

Linux 内核静态追踪Linux 系统

2013-12-18 14:44:10

2021-04-15 05:51:25

Linux

2021-08-23 06:59:38

Linux内核代码

2021-09-30 09:43:11

Linux内核Zstd补丁

2023-11-24 11:24:16

Linux系统

2009-04-09 17:22:23

Linuxshutdown系统

2013-12-20 13:08:12

Linux内核服务器

2013-12-18 11:14:57

Linux内核Linux Kerne

2020-04-13 07:00:00

Fedora防火墙系统运维
点赞
收藏

51CTO技术栈公众号