断开套接字连接-----网络编程(Linux----C)

1、基于TCP的半关闭

(1)单方面断开连接带来的问题

Linux的close函数和Windows的closesocket函数意味着完全断开连接,完全断开不仅指无法传输数据,而且也不能接收数据。在某些情况下,通信一方调用close或closesocket函数断开连接就显得不太优雅。


2台主机正在进行双向通信,主机A发送完最后的数据后,调用close函数断开了连接,之后主机A无法再接收主机B传输的数据。实际上,是完全无法调用与接收数据相关的函数。最终,由主机B传输的、主机A必须接收的数据也销毁了。

为了解决这类问题,“只关闭一部分数据交换中使用的流的方法应运而生。断开一部分连接是指,可以传输数据但无法接收,或可以接收数据但无法传输。即只关闭流的一半。

(2)套接字和流(Stream)


(3)针对优雅断开的shutdown函数

用来关闭其中1个流。

函数:#include <sys/socket.h>

int  shutdown(int  sock,int  howto);

成功返回0,失败返回-1。

参数:

  • sock:需要断开的套接字文件描述符。
  • howto:传递断开方式信息。
断开连接方式的可能值:

  • SHUT_RD:断开输入流。
  • SHUT_WR:断开输出流。
  • SHUT_RDWR:同时断开I/O流。
基于半关闭的文件传输程序


服务器端: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_sd,clnt_sd;
	FILE*fp;
	char buf[BUF_SIZE];
	int read_cnt;
	
	struct sockaddr_in serv_adr,clnt_adr;
	socklen_t clnt_adr_sz;
	
	if(argc!=2)
	{
		printf("Usage:%s<port>\n",argv[0]);
		exit(1);
	}
	
	fp=fopen("file_server.c","rb");
	serv_sd=socket(PF_INET,SOCK_STREAM,0);
	
	memset(&serv_adr,0,sizeof(serv_adr));
	serv_adr.sin_family=AF_INET;
	serv_adr.sin_addr.s_addr=htonl(INADDR_ANY);
	serv_adr.sin_port=htons(atoi(argv[1]));
	
	bind(serv_sd,(struct sockaddr*)&serv_adr,sizeof(serv_adr));
	listen(serv_sd,5);
	
	clnt_adr_sz=sizeof(clnt_adr);
	clnt_sd=accept(serv_sd,(struct sockaddr*)&clnt_adr,&clnt_adr_sz);
	while(1)
	{
		read_cnt=fread((void*)buf,1,BUF_SIZE,fp);
		if(read_cnt<BUF_SIZE)
		{
			write(clnt_sd,buf,BUF_SIZE);
			break;
		}
		write(clnt_sd,buf,BUF_SIZE);
	}
	shutdown(clnt_sd,SHUT_WR);
	read(clnt_sd,buf,BUF_SIZE);
	printf("Message from client:%s\n",buf);
	
	fclose(fp);
	close(clnt_sd);close(serv_sd);
	return 0;
}
void error_handling(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_handling(char *message);

int main(int argc,char *argv[])
{
	int sd;
	FILE *fp;
	
	char buf[BUF_SIZE];
	int read_cnt;
	struct sockaddr_in serv_adr;
	if(argc!=3)
	{
		printf("Usage:%s<port>\n",argv[0]);
		exit(1);
	}
	fp=fopen("receive.dat","wb");
	sd=socket(PF_INET,SOCK_STREAM,0);
	
	memset(&serv_adr,0,sizeof(serv_adr));
	serv_adr.sin_family=AF_INET;
	serv_adr.sin_addr.s_addr=inet_addr(argv[1]);
	serv_adr.sin_port=htons(atoi(argv[2]));
	
	connect(sd,(struct sockaddr*)&serv_adr,sizeof(serv_adr));
	
	while((read_cnt=read(sd,buf,BUF_SIZE))!=0)
		fwrite((void*)buf,1,read_cnt,fp);
	
	puts("Received file data");
	write(sd,"Thank you",10);
	fclose(fp);
	close(sd);
	return 0;
}
void error_handling(char *message)
{
	fputs(message,stderr);
	fputc('\n',stderr);
	exit(1);
}



可以看到服务器端向客户端传输服务器端的源文件file_server.c。
Logo

更多推荐