网络编程0x01 一个简单的回射服务器

0x00 Socket通讯流程

服务器调用socket()、bind()、listen()完成初始化后，调用accept()阻塞等待，处于监听端口的状态，客户端调用socket()初始化后，调用connect()发出SYN段并阻塞等待服务器应答，服务器应答一个SYN-ACK段，客户端收到后从connect()返回，同时应答一个ACK段，服务器收到后从accept()返回。

数据传输的过程:

建立连接后，TCP协议提供全双工的通信服务，但是一般的客户端/服务器程序的流程是由客户端主动发起请求，服务器被动处理请求，一问一答的方式。因此，服务器从accept()返回后立刻调用read()，读socket就像读管道一样，如果没有数据到达就阻塞等待，这时客户端调用write()发送请求给服务器，服务器收到后从read()返回，对客户端的请求进行处理，在此期间客户端调用read()阻塞等待服务器的应答，服务器调用write()将处理结果发回给客户端，再次调用read()阻塞等待下一条请求，客户端收到后从read()返回，发送下一条请求，如此循环下去。

如果客户端没有更多的请求了，就调用close()关闭连接，就像写端关闭的管道一样，服务器的read()返回0，这样服务器就知道客户端关闭了连接，也调用close()关闭连接。注意，任何一方调用close()后，连接的两个传输方向都关闭，不能再发送数据了。如果一方调用shutdown()则连接处于半关闭状态，仍可接收对方发来的数据。

0x01. 服务器端

• socket(创建socket)
• bind(绑定socket和端口号)
• listen(监听该端口号)
• accept(等待接受客户端连接请求)
• accept(接收来自客户端的连接请求)
• read,write(读取数据和返回数据) -
• close(关闭socket)

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

#define ERR_EXIT(m) \
        do  \
        {   \
            perror(m);  \
            exit(EXIT_FAILURE); \
        } while(0);
#define SOCK_PORT 8899
int main(int argc, char **argv)
{
    int cliSockFd;
    int servSockFd;
    socklen_t clientLen;
    struct sockaddr_in servAddr;
    struct sockaddr_in clientAddr;

    // 1. 创建套接字
    servSockFd = socket(AF_INET, SOCK_STREAM, 0);
    if (-1 == servSockFd)
    {
        ERR_EXIT("create socket err");
    }

    // 2. 分配套接字地址
    servAddr.sin_family      = AF_INET;
    servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    //servAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    servAddr.sin_port        = htons(SOCK_PORT);
    bzero(&servAddr.sin_zero, 8);

    int on = 1;
    // 确保time_wait状态下同一端口仍可使用
    if (setsockopt(servSockFd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof on) < 0)
    {
        ERR_EXIT("setsockopt");
    }

    // 3. 绑定套接字地址
    if (bind(servSockFd, (struct sockaddr*)&servAddr, sizeof(servAddr)) == -1)
    {
        ERR_EXIT("bind err");
    }

    // 4. 等待连接请求状态
    if (listen(servSockFd, SOMAXCONN) == -1) 
    {
        ERR_EXIT("listen err");
    }

    // 5. 允许连接
    memset(&clientAddr, 0, sizeof(clientAddr));
    cliSockFd = accept(servSockFd, (struct sockaddr*)&clientAddr, &clientLen);
    if (cliSockFd == -1) 
    {
        ERR_EXIT("accept err");
    }

    // 6. 数据交换
    char recvbuf[1024];
    while (1)
    {
        memset(recvbuf, 0, sizeof recvbuf);
        int ret = read(cliSockFd, recvbuf, sizeof(recvbuf));
        if (ret > 0)
        {
            fputs(recvbuf, stdout);
        }

        write(cliSockFd, recvbuf, ret);
    }

    // 7. 断开连接
    close(cliSockFd);
    close(servSockFd);
    
    return 0;
}

0x02. 客户端

• socket(创建socket)
• connect(连接指定的端口)
• read,write(读取数据和返回数据)
• close(关闭socket)

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

#define ERR_EXIT(m) \
        do  \
        {   \
            perror(m);  \
            exit(EXIT_FAILURE); \
        } while(0);

#define SOCK_PORT 8899
int main(int argc, char **argv)
{
    int cliSockFd;
    struct sockaddr_in servAddr;

    // 1. 创建套接字
    cliSockFd = socket(AF_INET, SOCK_STREAM, 0);
    if (-1 == cliSockFd)
    {
        ERR_EXIT("create socket err");
    }

    // 2. 分配套接字地址
    servAddr.sin_family      = AF_INET;
    servAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    servAddr.sin_port        = htons(SOCK_PORT);
    bzero(&servAddr.sin_zero, 8);

    // 3. 请求链接
    if (connect(cliSockFd, (struct sockaddr*)&servAddr, sizeof(servAddr)) == -1)
    {
        ERR_EXIT("connect err");
    }

    // 4. 数据交换
    char recvbuf[1024];
    char sendbuf[1024];
    while (fgets(sendbuf, sizeof(sendbuf), stdin) != NULL)   // 键盘输入获取
    {
        write(cliSockFd, sendbuf, sizeof(sendbuf)); // 写入服务器
        int ret = read(cliSockFd, recvbuf, sizeof(recvbuf));    // 服务器读取
        if (ret == 0)
        {
            printf("server close\n");
            break;
        } else if (ret == -1)
        {
            ERR_EXIT("read");
        }
        fputs(recvbuf, stdout); // 服务器返回数据输出

        // 清空
        memset(recvbuf, 0, sizeof recvbuf);
        memset(sendbuf, 0, sizeof sendbuf);
    }

    // 5. 断开连接
    close(cliSockFd);

    return 0;
}

0x03 实验结果

➜  network ./server 
hello world
ni
5769829
32423432
dfadasjfdsf
我们都一样

---

➜  network ls
client  echocli.c  echosrv.c  server
➜  network ./client 
hello world
hello world
ni
ni
32423432
32423432
dfadasjfdsf
dfadasjfdsf
我们都一样
我们都一样