Linux：TCP编程流程

1.TCP数据读写

#include<sys/types.h>
#include<sys/socket.h>

ssize_t recv(int sockfd, void *buf, size_t len, int flags);

ssize_t send(int sockfd, const void *buf, size_t len, int flags);

• recv()读取 sockfd 上的数据，buff 和 len 参数分别指定读缓冲区的位置和大小
• send()往 socket 上写入数据，buff 和 len 参数分别指定写缓冲区的位置和数据长度
• flags 参数为数据收发提供了额外的控制

2.TCP编程流程

TCP 提供的是面向连接的、可靠的、字节流服务。TCP 的服务器端和客户端编程流程为：

• socket()方法是用来创建一个套接字，有了套接字就可以通过网络进行数据的收发。这也是为什么进行网络通信的程序首先要创建一个套接字。创建套接字时要指定使用的服务类型，使用 TCP 协议选择流式服务（SOCK_STREAM）。
• bind()方法是用来指定套接字使用的 IP 地址和端口。IP 地址就是自己主机的地址，如果主机没有接入网络，测试程序时可以使用回环地址“127.0.0.1”。端口是一个 16 位的整形值，一般 0-1024 为知名端口，如 HTTP 使用的 80 号端口。这类端口一般用户不能随便使用。其次，1024-4096 为保留端口，用户一般也不使用。4096 以上为临时端口，用户可以使用。在Linux 上，1024 以内的端口号，只有 root 用户可以使用。
• listen()方法是用来创建监听队列。监听队列有两种，一个是存放未完成三次握手的连接，一种是存放已完成三次握手的连接。listen()第二个参数就是指定已完成三次握手队列的长度。
• accept()处理存放在 listen 创建的已完成三次握手的队列中的连接。每处理一个连接，则accept()返回该连接对应的套接字描述符。如果该队列为空，则 accept 阻塞。
• connect()方法一般由客户端程序执行，需要指定连接的服务器端的 IP 地址和端口。该方法执行后，会进行三次握手， 建立连接。
• send()方法用来向 TCP 连接的对端发送数据。send()执行成功，只能说明将数据成功写入到发送端的发送缓冲区中，并不能说明数据已经发送到了对端。send()的返回值为实际写入到发送缓冲区中的数据长度。
• recv()方法用来接收 TCP 连接的对端发送来的数据。recv()从本端的接收缓冲区中读取数据，如果接收缓冲区中没有数据，则 recv()方法会阻塞。返回值是实际读到的字节数，如果recv()返回值为 0， 说明对方已经关闭了 TCP 连接。
• close()方法用来关闭 TCP 连接。此时，会进行四次挥手。

TCP服务端示例

// TCP Server.c

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

int main(void)
{
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    assert(sockfd != -1);

    struct sockaddr_in saddr;
    memset(&saddr, 0, sizeof(saddr));
    saddr.sin_family = AF_INET;
    saddr.sin_port = htons(6000);
    saddr.sin_addr.s_addr = inet_addr("127.0.0.1");

    int res = bind(sockfd, (struct sockaddr*)&saddr, sizeof(saddr));
    assert(res != -1);

    res = listen(sockfd, 5);
    assert(res != -1);

    while (1)
    {
        struct sockaddr_in caddr;
        socklen_t len = sizeof(caddr);
        int c = accept(sockfd, (struct sockaddr*)&caddr, &len);
        if (c == -1)
        {
            perror("accept error");
            continue;
        }

        printf("accept c = %d\n", c);
        char data[128] = {0};
        int n = recv (c, data, 127, 0);

        printf("n = %d, buff = %s\n", n, data);

        send(c, "OK", 2, 0);

        close(c);
    }
    close(sockfd);

    exit(0);
}

TCP客户端示例

//TCP Client.c

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

int main(void)
{
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    assert(sockfd != -1);

    struct sockaddr_in saddr;
    memset(&saddr, 0, sizeof(saddr));
    saddr.sin_family = AF_INET;
    saddr.sin_port = htons(6000);
    saddr.sin_addr.s_addr = inet_addr("127.0.0.1");

    int res = connect(sockfd, (struct sockaddr*)&saddr, sizeof(saddr));
    assert(res != -1);

    printf("Please input data:\n");
    char buff[128] = {0};
    fgets(buff, 128, stdin);

    send(sockfd, buff, strlen(buff) - 1, 0);

    char data[128] = {0};
    int n = recv(sockfd, data, 127, 0);

    printf("%s\n", data);

    close(sockfd);

    exit(0);
}

运行示例