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

时间:2019-04-15
本文章向大家介绍TCP/IP网络编程学习笔记(五)断开套接字连接,主要包括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网络编程》尹圣雨 著,金果哲 译

上一页 下一页

原文地址:http://www.manongjc.com/article/103550.html