Part 1 start network programming:chapter SEVEN：优雅地断开套接字连接

什么是优雅地断开套接字连接呢？
之前我们直接调用 close 或者在win 下调用 closesock函数单方面断开了套接字的连接

7.1 基于TCP的半关闭

掌握TCP的关闭过程是非常重要的，因为在建立连接的过程中是很稳定的，断开时是容易发生预想不到的情况的。

7.1.1 单方面断开带来的问题

单方面的切断连接会让动作发出者无法传输数据也无法接收数据。
下图展示的是A方单方切断了连接带来的效果。

从图中可以看出，不仅仅A无法再继续发送数据，存储在A的输入缓冲的数据也会被销毁。

为了解决这样的问题，“直关闭一部分数据交换中使用的流”的方法诞生了（Half-close）
断开一半是指，只断开输出流 或者直关闭输入流。

7.1.2 套接字和流（stream）

两台计算机通过套接字建立连接后进入数据可交换的状态，又称为“流形成的状态”

为了进行双向通信，每个套接字都拥有两个流。如下图所示：

一旦建立了套接字连接，每个主机就拥有单独的输入流和输出流。一个主机的输入流只能和另一个主机的输出流相连接。

本章所说的就是只断开一台主机的一个流的行为。

7.1.3 优雅地断开函数 shutdown

这个函数用来关闭其中一个流。也叫半关闭函数。

#include <sys/socket.h>
int shutdown(int sock,int howto);
-> 成功时返回0，失败时返回-1
sock：需要断开的套接字文件描述符
howto：传递断开方式信息

上面的第二个参数的取值如下：

1. SHUT_RD:断开输入流 ： 只能输出，不能再接收数据，输入缓冲数据被销毁
2. SHUT_WR:断开输出流：只能接收数据，不能再向输出缓冲中写入数据，原本在输出缓冲中的数据可以继续传输。
3. SHUT_RDWR:同时断开I/O流：啥都不能干了。只有原来输出缓冲中的数据还能继续传输。

7.1.3为啥需要半关闭？

假设服务器端已经向客户端传输了数据，没有要继续传的东西了，只关闭服务器端的输出流（半关闭），这样服务器的仍然可以接受客户端发来的数据。

7.1.4 基于半关闭的文件传输程序

基于上面的过程，我们编写一段基于半关闭的文件传输程序。

file_client.c

//客户端的思路，接收完我就全断了。直接close。
#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 sock;
	FILE* fp;

	char buf[BUF_SIZE];
	int read_count;

	struct sockaddr_in serv_addr;

	if(argc != 3){
		printf("Usage :%s <IP> <port>\n",argv[0]);
		exit(1);
	}

	fp = fopen("receive.dat","wb");
	sock = socket(PF_INET,SOCK_STREAM,0);

	memset(&serv_addr,0,sizeof(serv_addr));
	serv_addr.sin_family = AF_INET;
	serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
	serv_addr.sin_port = htons(atoi(argv[2]));

	if(connect(sock,(struct sockaddr*)&serv_addr,sizeof(serv_addr)) == -1){
		error_handling("connect error");
	}

	while((read_count = read(sock,buf,BUF_SIZE)) != 0)
	{
		fwrite((void*)buf,1,read_count,fp);
	}

	puts("Receive file data");
	write(sock,"Thank you",10);

	fclose(fp);
	close(sock);

	return 0;
}

void error_handling(char* message)
{
	fputs(message,stderr);
	fputc('\n',stderr);
	exit(1);
}

接下里是服务器端，
服务器端的思路，给你发完我不能全断了，我还要接收你的感谢呢，断了输出流就行了叭

file_server.c
#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_sock,client_sock;

	FILE* fp;
	char buf[BUF_SIZE];
	int read_count;

	struct sockaddr_in serv_addr,client_addr;

	socklen_t client_addr_size;

	if(argc != 2){
		printf("Usage : %s <port> \n",argv[0]);
		exit(1);
	}

	fp = fopen("file_server.c","rb");

	serv_sock = socket(PF_INET,SOCK_STREAM,0);
	memset(&serv_addr,0,sizeof(serv_addr));
	serv_addr.sin_family = AF_INET;
	serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	serv_addr.sin_port = htons(atoi(argv[1]));

	bind(serv_sock,(struct sockaddr*)&serv_addr,sizeof(serv_addr));

	listen(serv_sock,5);

	client_sock = accept(serv_sock,(struct sockaddr*)&client_addr,&client_addr_size);

	while(1){	// 从fp文件中读取数据并保存到，buf中
		read_count = fread((void*)buf,1,BUF_SIZE,fp);
		if(read_count < BUF_SIZE){ // 向buf中写入
			write(client_sock,buf,read_count);
			break;
		}
		write(client_sock,buf,BUF_SIZE);
	}

	shutdown(client_sock,SHUT_WR);
	read(client_sock,buf,BUF_SIZE);
	printf("message from client :%s\n",buf);

	fclose(fp);
	close(client_sock);
	close(serv_sock);
	return 0;
}

void error_handling(char* message)
{
	fputs(message,stderr);
	fputc('\n',stderr);
	exit(1);
}

下面是测试结果，可以看到，在服务器端能够正确接收服务器端最后传输的消息。