TCP/IP网络编程学习笔记（五）断开套接字连接

一、基于TCP的半关闭

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

Linux 的 close 函数和 Windows 的 closesocket 函数意味着完全断开连接。完全断开连接指：同时断开输入流和输出流。如果主机A不再需要向主机B传输数据，主机A还有来自主机B的数据必须接收时，完全断开会导致数据丢失。半关闭流（只关闭一部分数据交换中使用的流，可以传输数据但无法接收，或者可以接收数据但无法传输）可以解决这个问题。

2.相关函数

shutdown 函数：文件传输结束时，close 函数既关闭 I/O 流，又向对方发送 EOF 表示文件传输结束，对方通过函数返回值接收 EOF .而shutdown 函数只关闭部分流，同时也会向对方发送 EOF

#include<sys/socket.h>
// 功能：半关闭流
// 参数：
//    sock--需要断开的套接字文件描述符
//    howto--传递断开方式信息
// 返回值：成功时返回 0，失败时返回 -1
int shutdown(int sock,int howto);

howto参数的可选值：

• SHUT_RD：断开输入流
• SHUT_WR：断开输出流
• SHTU_RDWR：同时断开I/O流

3.验证程序

服务器端向客户端发送文件，客户端收到文件后回送“Thank you”

• 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) {
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

int main(int argc, char* argv[]) {
    int serv_sock, clnt_sock;
    struct sockaddr_in serv_addr, clnt_addr;
    char buf[BUF_SIZE];
    FILE* fp;

    if (argc != 2) {
        printf("Usage: %s <port>\n", argv[0]);
        exit(1);
    }

    fp = fopen("file_server.c", "rb");

    serv_sock = socket(PF_INET, SOCK_STREAM, 0);
    if (serv_sock == -1)
        error_handling("socket() error");

    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_addr.sin_port = htons(atoi(argv[1]));

    if (bind(serv_sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) == -1)
        error_handling("bind() error");

    if (listen(serv_sock, 5) == -1)
        error_handling("listen() error");

    socklen_t clnt_addr_size = sizeof(clnt_addr);
    clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_addr, &clnt_addr_size);
    if (clnt_sock == -1)
        error_handling("accept() error");

    while (1) {
        // fread 函数成功时返回同第三个参数一样的数值即 BUF_SIZE，如果返回值大于 0 小于 BUF_SIZE，
        // 则说明达到文件末尾，如果返回值为 0，说明读取的字节数小于第二个参数指定的字节数，读取失败
        int read_cnt = fread((void*)buf, 1, BUF_SIZE, fp);
        if (read_cnt < BUF_SIZE) {
            write(clnt_sock, buf, read_cnt);
            break;
        }
        write(clnt_sock, buf, BUF_SIZE);
    }

    shutdown(clnt_sock, SHUT_WR);
    read(clnt_sock, buf, BUF_SIZE);
    printf("Message form client: %s \n", buf);

    fclose(fp);
    close(clnt_sock);
    close(serv_sock);
    return 0;
}

• file_client.c

#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<string.h>
#include<arpa/inet.h>
#include<sys/socket.h>

#define BUF_SIZE 30

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

int main(int argc, char* argv[]) {
    int sock;
    struct sockaddr_in serv_addr;
    char buf[BUF_SIZE];
    FILE* fp;

    if (argc != 3) {
        printf("Usage: %s <IP> <port>\n", argv[0]);
        exit(1);
    }

    fp = fopen("receive.dat", "wb");

    sock = socket(PF_INET, SOCK_STREAM, 0);

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

    connect(sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr));

    int read_cnt;
    while ((read_cnt = read(sock, buf, BUF_SIZE)) != 0)
        fwrite((void*)buf, 1, read_cnt, fp);

    puts("Received file data");
    write(sock, "Thank you", 10);

    fclose(fp);
    close(sock);
    return 0;
}

• 运行结果

参考书籍：《TCP/IP网络编程》尹圣雨 著，金果哲 译