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

时间:2022-04-15 10:22:49

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

1、基于TCP的半关闭

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

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

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

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

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

(2)套接字和流(Stream)

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

(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流。
基于半关闭的文件传输程序

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

服务器端: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。