如何让在 recv() 上阻塞的线程优雅地退出？

Question

有这样一个话题：

{  
    ......    
    while (1)
    {
        recv(socket, buffer, sizeof(buffer), 0);
        ......
    }
    close(socket);           
}  

由于线程在 recv() 调用时被阻塞，我怎样才能让线程优雅地退出？

Answer 1

您可以致电：-

shutdown(sock, SHUT_RDWR)    //on the remote end

签出this。

Answer 2

不要在 recv 上阻塞。而是在 'select' 上实现一个块并测试输入 fdset 以确定是否首先要读取任何内容。

如果有数据要读取，调用recv。如果不是，则循环，检查启动关闭时由另一个线程设置的“正在运行”易失性布尔变量。

Answer 3

要么：

1. 使用setsockopt() 和SO_RCVTIMEO 设置读取超时，并在触发时检查状态变量，看看您是否告诉自己停止读取。
2. 如果您想永远停止读取套接字，请使用shutdown(sd, SHUT_RD) 将其关闭以进行输入。这将导致 recv() 从现在开始返回零。
3. 将套接字设置为非阻塞模式并使用select() 和超时来告诉您何时读取，采用与上述（1）中的状态变量相同的策略。但是，非阻塞模式会给发送操作带来相当大的复杂性，因此您应该更喜欢上面的 (1) 或 (2)。

您的伪代码缺少 EOS 和错误检查。我希望它不是真的那样。

Answer 4

声明一些“停止”布尔值，在每次 recv() 返回后检查它，如果设置则终止。要关闭，请设置布尔值并从另一个线程关闭套接字。阻塞的 recv() 将“立即”返回一个错误，但这并不重要，因为无论如何你都将要终止:)

Answer 5

我可能会使用 @alk 的罚款 answer（也讨论过 here）中的信号。

或者，您可以使用多路复用 I/O。

在初始化时，创建一个全局管道(2)。当需要结束程序时，关闭管道的写端——现在读端将立即 select(2)/poll(2) 可读（对于 EOF）。同时，让你的recv-blocked线程包含这个管道的读取端以及它们的套接字，例如，无限期阻塞的select(2)调用。如果管道的读取端返回可读，则 I/O 线程知道是时候优雅地终止了。

这种技术的主要警告是确保写入端只关闭一次，因为随后的天真的 close(2) 可能会 nip 一些无辜的文件，这些文件恰好被赋予与管道的旧写入相同的描述符-结束。

Answer 6

声明一个全局退出标志：

int bExit = 0;

让关键部分测试一下：

 while(1)
 {
   ssize_t result = recv(socket, buffer, sizeof(buffer), 0);
   if ((-1 == result) && (EINTR == error) && bExit) 
   {
     break;
   }

   ...
 }

要破坏你的阅读器，首先设置退出标志

bExit = 1;

然后向阅读器线程发送信号

pthread_kill(pthreadReader, SIGUSR1);

---

注意： 我在这个例子中遗漏了保护bExit 防止并发访问。

这可以通过使用互斥体或适当的声明来实现。

Answer 7

另外，将您的套接字设置为非阻塞：

int listen_sd = socket(PF_INET6, SOCK_STREAM, IPPROTO_TCP);
if (listen_sd < 0) {
    perror("socket() failed");
}

int on = 1;

//set socket to be non-blocking
int rc = ioctl(listen_sd, FIONBIO,(char *)&on);
if (rc < 0) {
    perror("ioctl() failed");
    close(listen_sd);
}

然后，您可以在套接字上调用 recv()，它不会阻塞。如果没有可读取的内容，则将 ERRNO 全局变量设置为常量 EWOULDBLOCK

int rc = recv(listen_sd, buffer, sizeof(buffer), 0);
if (rc < 0) {
    if (errno != EWOULDBLOCK) {
        perror("recv() failed");
    }
}