16、socket缓冲区以及阻塞模式详解

最新推荐文章于 2024-07-25 21:29:45 发布

守望178

最新推荐文章于 2024-07-25 21:29:45 发布

阅读量410

点赞数

分类专栏：网络通信文章标签： Socket 缓冲区 Linux 非阻塞

原文链接：http://c.biancheng.net/view/2349.html

版权

网络通信专栏收录该内容

27 篇文章 15 订阅

订阅专栏

在《发送数据和接收数据》一节中讲到，可以使用 write()/send() 函数发送数据，使用 read()/recv() 函数接收数据，本节就来看看数据是如何传递的。

socket缓冲区

每个 socket 被创建后，都会分配两个缓冲区，输入缓冲区和输出缓冲区。

write()/send() 并不立即向网络中传输数据，而是先将数据写入缓冲区中，再由TCP协议将数据从缓冲区发送到目标机器。一旦将数据写入到缓冲区，函数就可以成功返回，不管它们有没有到达目标机器，也不管它们何时被发送到网络，这些都是TCP协议负责的事情。

TCP协议独立于 write()/send() 函数，数据有可能刚被写入缓冲区就发送到网络，也可能在缓冲区中不断积压，多次写入的数据被一次性发送到网络，这取决于当时的网络情况、当前线程是否空闲等诸多因素，不由程序员控制。

read()/recv() 函数也是如此，也从输入缓冲区中读取数据，而不是直接从网络中读取。

图：TCP套接字的I/O缓冲区示意图

这些I/O缓冲区特性可整理如下：

I/O缓冲区在每个TCP套接字中单独存在；
I/O缓冲区在创建套接字时自动生成；
即使关闭套接字也会继续传送输出缓冲区中遗留的数据；
关闭套接字将丢失输入缓冲区中的数据。

输入输出缓冲区的默认大小一般都是 8K，可以通过 getsockopt() 函数获取：

unsigned optVal;
int optLen = sizeof(int);
getsockopt(servSock, SOL_SOCKET, SO_SNDBUF, (char*)&optVal, &optLen);
printf("Buffer length: %d\n", optVal);

运行结果：
Buffer length: 8192

这里仅给出示例，后面会详细讲解。

阻塞模式

对于TCP套接字（默认情况下），当使用 write()/send() 发送数据时：
1) 首先会检查缓冲区，如果缓冲区的可用空间长度小于要发送的数据，那么 write()/send() 会被阻塞（暂停执行），直到缓冲区中的数据被发送到目标机器，腾出足够的空间，才唤醒 write()/send() 函数继续写入数据。

2) 如果TCP协议正在向网络发送数据，那么输出缓冲区会被锁定，不允许写入，write()/send() 也会被阻塞，直到数据发送完毕缓冲区解锁，write()/send() 才会被唤醒。

3) 如果要写入的数据大于缓冲区的最大长度，那么将分批写入。

4) 直到所有数据被写入缓冲区 write()/send() 才能返回。

当使用 read()/recv() 读取数据时：
1) 首先会检查缓冲区，如果缓冲区中有数据，那么就读取，否则函数会被阻塞，直到网络上有数据到来。

2) 如果要读取的数据长度小于缓冲区中的数据长度，那么就不能一次性将缓冲区中的所有数据读出，剩余数据将不断积压，直到有 read()/recv() 函数再次读取。

3) 直到读取到数据后 read()/recv() 函数才会返回，否则就一直被阻塞。

这就是TCP套接字的阻塞模式。所谓阻塞，就是上一步动作没有完成，下一步动作将暂停，直到上一步动作完成后才能继续，以保持同步性。

TCP套接字默认情况下是阻塞模式，也是最常用的。当然你也可以更改为非阻塞模式。

使用阻塞模式将会导致线程等待，无法处理其他事物，效率低下。在实际项目，通常会使用非阻塞模式。调用如下函数，设置非阻塞模式：

fcntl(serverSock, F_SETFL, O_NONBLOCK);

这样使用原本blocking的各种函数，可以立即获得返回结果。通过判断返回的errno了解状态：

accept():

在non-blocking模式下，如果返回值为-1，且errno == EAGAIN或errno == EWOULDBLOCK表示no connections没有新连接请求；

recv()/recvfrom():

在non-blocking模式下，如果返回值为-1，且errno == EAGAIN表示没有可接受的数据或很在接受尚未完成；

send()/sendto():

在non-blocking模式下，如果返回值为-1，且errno == EAGAIN或errno == EWOULDBLOCK表示没有可发送数据或数据发送正在进行没有完成。

read/write:

在non-blocking模式下，如果返回-1，且errno == EAGAIN表示没有可读写数据或可读写正在进行尚未完成。

connect():

在non-bloking模式下，如果返回-1，且errno = EINPROGRESS表示正在连接。

使用如上方法，可以创建一个non-blocking的server的程序，类似如下代码：

    int main(int argc, char *argv[]) {
 
        int sock;
        if ( (sock = socket(PF_INET, SOCK_STREAM, 0)) == -1 ) {
            perror("socket failed");
            return 1;
        }
 
        //set socket to be non-blocking
        int flags = fcntl(sock, F_GETFL, 0);
        fcntl(sock, F_SETFL, flags | O_NONBLOCK);
 
        //create socket address to bind
        struct sockaddr_in bind_addr
        ...
 
        //bind
        bind(...)
        ...
 
        //listen
        listen(...)
        ...
 
        //loop 
        int new_sock;
        while (1) {
            new_sock = accept(sock, NULL, NULL);
            if (new_sock == -1 && errno == EAGAIN) {
                fprintf(stderr, "no client connections yet\n");
                continue;
            } else if (new_sock == -1) {
                perror("accept failed");
                return 1;
            }
 
            //read and write
            ...
 
        }   
 
        ...
    }

纯non-blocking程序缺点：函数调用会立即返回，而不会等操作结束，无法通过返回值获取操作结果，因此需要循环多次调用函数，直到获取最终的操作结果，但这样会耗费大量的CPU time，实际中并不会这样使用。实际中将non-blocking和select结合使用。