之前的方法不够优雅是因为,我们直接调用了close或closesocket函数单方面断开连接。
7.1 基于TCP的半关闭
TCP中的断开连接过程比建立连接过程更重要,因为连接过程一般不会出现变数,但断开过程有可能发生预想不到的结果。
单方面断开连接带来的问题
假如两台主机正在双向通信。当主机A发送完最后的数据后,调用close函数断开了连接,之后主机A再也无法接收主机B的传输数据。
断开一部分连接指的是可以传输数据但无法接收,或者可以接收数据但无法传输。
套接字和流
把套接字建立后可进行数据交换的状态看做是一种流。
为了进行双向通讯,需要2个流。
其中一个主机的输入流与另一个主机的输出流相连,而输出流则与另一主机的输入流相连。
close函数将同时断开这两个流。
针对优雅断开的shutdown函数
shutdown函数用来关闭其中的一个流
- SHUT_RD:断开输入流,套接字无法接收数据。即使输入缓冲收到数据也会抹去,而且无法调用输入相关的函数
- SHUT_WR:断开输出流,套接字无法传输数据。但如果输出缓冲还留有未传输的数据,则将传递至目标主机。
- SHUT_RDWR:同时中断IO流。
为何需要半关闭
1)意见1
是否只需要留出足够长的时间,保证完成数据交换即可?这样就没必要使用半关闭了呀。
但是要考虑这样的情况。一旦客户端连接到服务器端,服务器端将文件传递给客户端,客户端收到之后发送字符串“thanks”给服务器端。这说明客户端断开连接前还有数据需要传递。
此时程序实现的难度就大了,因为传输文件的服务器端只需要连续传输文件数据即可,而客户端无法知道需要接收数据到什么时候。客户端也没办法无休止地调用输入函数,因为这有可能导致程序阻塞(调用的函数未返回)
2)意见2
是否可以让服务器端和客户端约定一个代表文件尾的字符?
也有问题。这意味着文件中不能有与约定字符相同的内容。
3)意见3
服务器端可以最后向客户端传递EOF表示文件传输结束。客户端通过函数返回值接收EOF,这样可以避免与文件内容冲突。
问题是,服务器如何传递EOF?
断开输出流时向对方主机传输EOF。
调用close函数的同时关闭IO流,这样也会向对方发送EOF,但此时无法再接收对方传输的数据。
也就是说,如果服务器端调用close关闭IO流,就无法接收客户端在接收文件结束以后发送的“thanks”
所以解决的方法是:调用shutdown函数,只关闭服务器的输出流(半关闭)。这样既可以发送EOF,同时又保留了输入流,可以接收对方的数据
基于半关闭的文件传输程序
1)服务器端
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#define BUF_SIZE 30
void error_handling(char *message);
int main(int argc, char *argv[]){
int serv_sd,clnt_sd;
FILE *fp;
char buf[BUF_SIZE];
int read_cnt