第八节、基于半关闭的文件传输

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
#define SERVER_IP "127.0.0.1"

void error_handing(const char *message);

int main(int argc, char const *argv[])
{
	if(argc != 2)
	{
		printf("Usage <port>\n",argv[0]);
		exit(1);
	}

	int serv_fd = socket(PF_INET,SOCK_STREAM,0);
	if (serv_fd == -1)
	{
		error_handing("socket() error");
	}

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

	if(bind(serv_fd,(struct sockaddr*)&serv_addr,sizeof(serv_addr)) == -1)
	{
		error_handing("bind() error");
	}

	if(listen(serv_fd,5) == -1)
	{
		error_handing("listen() error");
	}

	struct sockaddr_in clie_addr;
	memset(&clie_addr,0,sizeof(clie_addr));
	socklen_t clie_len = sizeof(clie_addr);

	int clie_fd = accept(serv_fd,(struct sockaddr*)&clie_addr,&clie_len);
	if (clie_fd == -1)
	{
		error_handing("accept() error");
	}

	char buf[BUF_SIZE];
	FILE *fp;
	fp = fopen("file_server.c","rb");

	int read_cnt = 0;
	while(1) //循环写操作
	{
		memset(&buf,0,sizeof(buf));
		read_cnt = fread((void*)buf,1,BUF_SIZE,fp);
		if(read_cnt < BUF_SIZE) //直至写到文件末尾
		{
			write(clie_fd,buf,read_cnt);
			break;
		}
		write(clie_fd,buf,BUF_SIZE);
	}

	//发送文件后针对输出流进行半关闭,这样就向客户端传输了EOF,而客户端也知道文件传输已完成
	shutdown(clie_fd,SHUT_WR);
	//只关闭了输出流,依然可以通过输入流接收数据 发送EOF
	read(clie_fd,buf,BUF_SIZE);
	printf("Message from client: %s \n",buf );
	fclose(fp);
	close(clie_fd);
	close(serv_fd);

	return 0;
}


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

file_client.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_handing(const char *message);

int main(int argc, char const *argv[])
{
	if (argc != 3)
	{
		printf("Usage %s <port>,<ip>\n",argv[0]);
		exit(1);
	}

	int clie_fd = socket(PF_INET,SOCK_STREAM,0);
	if(clie_fd == -1)
	{
		error_handing("socket() error");
	}

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

	if(connect(clie_fd,(struct sockaddr *)&clie_addr,sizeof(clie_addr)) == -1)
	{
		error_handing("connect() error");
	}

	FILE *fp = fopen("receive.dat","wb");
	char buf[BUF_SIZE];
	memset(&buf,0,sizeof(buf));

	int read_cnt;

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

	puts("Received file data");
	write(clie_fd,"Thank you",10);
	fclose(fp);
	close(clie_fd);

	return 0;
}


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