TCP选择性应答源代码

以下是一个简单的TCP选择性应答的源代码示例，基于Linux平台和C语言：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <netinet/in.h>
#include <netinet/tcp.h>

#define SERVER_PORT 8888
#define SERVER_ADDR "127.0.0.1"
#define BUFFER_SIZE 1024

int main(int argc, char *argv[]) {
    int sockfd;
    struct sockaddr_in addr;
    char send_buf[BUFFER_SIZE];
    char recv_buf[BUFFER_SIZE];
    int read_len, send_len;
    int i, j, k;
    int seq_num = 0, ack_num = 0;
    int window_size = 1;
    int flag = 0;
    struct tcp_info info;
    socklen_t info_len = sizeof(info);

    // 创建套接字
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        perror("socket");
        exit(1);
    }

    memset(&addr, 0, sizeof(addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons(SERVER_PORT);
    addr.sin_addr.s_addr = inet_addr(SERVER_ADDR);

    // 连接服务器
    if (connect(sockfd, (struct sockaddr *)&addr, sizeof(addr)) == -1) {
        perror("connect");
        exit(1);
    }

    // 开启TCP选择性应答
    if (setsockopt(sockfd, IPPROTO_TCP, TCP_SACK, &flag, sizeof(flag)) != 0) {
        perror("setsockopt");
        exit(1);
    }

    while (1) {
        // 读取用户输入
        printf("请输入要发送的数据：");
        fgets(send_buf, BUFFER_SIZE, stdin);
        send_len = strlen(send_buf);

        // 发送数据
        if (send(sockfd, send_buf, send_len, 0) != send_len) {
            perror("send");
            exit(1);
        }

        // 接收数据
        read_len = recv(sockfd, recv_buf, BUFFER_SIZE, 0);
        if (read_len == -1) {
            perror("recv");
            exit(1);
        } else if (read_len == 0) {
            printf("服务器关闭连接\n");
            break;
        }

        // 读取TCP信息
        if (getsockopt(sockfd, IPPROTO_TCP, TCP_INFO, &info, &info_len) == 0) {
            seq_num = info.tcpi_snd_nxt;
            ack_num = info.tcpi_rcv_nxt;
            window_size = info.tcpi_snd_cwnd;
        }

        // 输出收到的数据、序列号、确认号和窗口大小
        printf("收到数据：%.*s，序列号：%d，确认号：%d，窗口大小：%d\n",
               read_len, recv_buf, seq_num, ack_num, window_size);

        // 模拟丢包，只发送部分数据
        if (send_len > 5) {
            send_len = 5;
        }

        // 发送部分数据
        for (i = 0; i < send_len; i += window_size) {
            k = (send_len - i) > window_size ? window_size : (send_len - i);
            if (send(sockfd, send_buf + i, k, 0) != k) {
                perror("send");
                exit(1);
            }

            // 接收确认
            read_len = recv(sockfd, recv_buf, BUFFER_SIZE, 0);
            if (read_len == -1) {
                perror("recv");
                exit(1);
            } else if (read_len == 0) {
                printf("服务器关闭连接\n");
                break;
            }

            // 输出收到的确认号和窗口大小
            if (getsockopt(sockfd, IPPROTO_TCP, TCP_INFO, &info, &info_len) == 0) {
                ack_num = info.tcpi_rcv_nxt;
                window_size = info.tcpi_snd_cwnd;
            }
            printf("收到确认，确认号：%d，窗口大小：%d\n", ack_num, window_size);
        }
    }

    // 关闭套接字
    close(sockfd);
    return 0;
}

这个示例程序实现了一个简单的TCP客户端，可以与服务器进行双向通信。在程序中开启了TCP选择性应答，并且模拟了丢包的情况，以测试TCP选择性应答的效果。程序中的注释可以帮助理解代码。