网络编程（7）—— 使用shutdown半关闭服务器和客户端的连接

        之前在socket服务器和客户端中，在linux平台上一直使用close，windows平台使用closesocket断开连接，无论是close和closesocket，都是输入和输出的链接同时断开。试想下面一种应用场景，搭建一个文件传输的服务器和客户端，客户端从服务器处下载文件，当下载的文件完成时，客户端需要向服务器传递一个字符串，比如“ThankYou”。处理这种情况，再使用之前的close或者closesocket显示不合适，因为在服务器把文件全部传输完成之后，输入和输出连接全部断开，显然无法再收到客户端发送的字符串。此时我们就需要在文件传输后，能够优雅的断开服务器的输出连接，而保留输入连接，使用shutdown函数可以满足我们的需求。

shutdown(socket sock,int flag);

        shutdown的两个参数：sock是需要断开连接的socket文件操作符，flag可选三个数值：

SHUT_RD - 断开输入流
SHUT_WR - 断开输出流
SHUT_RDWR - 同时断开输出流和输入流

        下面的例子，利用socket创建了一个简单的文件传输服务器和客户端。服务器不断读取一个名叫file_server.c（就是服务端的代码所在文件）的文件，发给客户端，待读取完成后，接受客户端发过来的信息。服务端代码如下：

file_server.c：

#include<string.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#include<stdio.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>",argv[0]);
        exit(1);
    }

    fp=fopen("file_server.c","rb");
    serv_sd=socket(AF_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,read_cnt);
            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(serv_sd);
    close(clnt_sd);
    return 0;
}


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

客户端的代码比较简单，就是接受服务端的文件，接受完成后给服务端发送一个“Thank you”的字符串。

file_clent.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;
    FILE* fp;
    int read_cnt;
    char buf[BUF_SIZE];

    struct sockaddr_in serv_adr;
    if(argc!=3)
    {
        printf("Usage %s <addr> <port>",argv[0]);
        exit(1);
    }
    fp=fopen("recieve.dat","wb");
    serv_sock=socket(AF_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(serv_sock,(struct sockaddr*)&serv_adr,sizeof(serv_adr));
    
    while((read_cnt=read(serv_sock,buf,BUF_SIZE))!=0)
    {
        fwrite((void*)buf,1,read_cnt,fp);
        printf("read_cnt= %d\n",read_cnt);
    }

    puts("recieved all data");
    write(serv_sock,"Thank you",10);
    fclose(fp);
    close(serv_sock);
}


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

运行结果如下：

[Hyman@Hyman-PC csdn]$ ./serv 9190
message from client: Thank you 

[Hyman@Hyman-PC csdn]$ ./clnt 127.0.0.1 9190
recieved all data

Github位置：

https://github.com/HymanLiuTS/NetDevelopment

克隆本项目：

git clone git@github.com:HymanLiuTS/NetDevelopment.git

获取本文源代码：

git checkout NL07