断开套接字连接-----网络编程(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。