sigwait 处理 SIGPIPE

Question

当一个进程的输出通过管道传输到另一个进程时，我正在尝试实现一个进程的正常关闭。我正在通过管道输出来测试下面的代码：./a.out | less 并在出现提示时按q。我看到的是信号处理程序的调用，而不是sigwait() 的预期完成（这里添加它只是为了显示正在发生的事情）。

#include <csignal>
#include <chrono>
#include <iostream>
#include <thread>

#include <signal.h>

int handlerSig {0};

void signalHandler(int s)
{
    handlerSig = s;
    std::cerr << "handlerSig: "  << handlerSig << std::endl;
}

int main()
{
    for (int i = 1; i < 32; ++i)
    {
        std::signal(i, signalHandler);
    }

    bool run {true};
    std::thread thread {[&]
        {
            while (run)
            {
                std::cout << "ping" << std::endl;
                std::this_thread::sleep_for(std::chrono::milliseconds {500});        
            }
        }};

    sigset_t waitSet;
    sigemptyset(&waitSet);
    sigaddset(&waitSet, SIGINT);
    sigaddset(&waitSet, SIGPIPE);
    sigaddset(&waitSet, SIGTERM);

    pthread_sigmask(SIG_BLOCK, &waitSet, nullptr);

    int waitSig {0};
    sigwait(&waitSet, &waitSig);
    run = false;
    thread.join();

    std::cerr << "waitSig: "  << waitSig << std::endl;
}

我在 WSL2 和 CentOS 机器上得到了一致的结果，我更愿意专注于在那里解决这个问题。在 WSL1 下运行时，除非我删除 pthread_sigmask(SIG_BLOCK...)，否则 SIGINT 和 SIGTERM 都不会导致 sigwait() 完成，但这似乎与我对应该如何使用 sigwait() 的理解相矛盾。

Answer 1

您需要设计其他方法来发现写入失败，例如，忽略 SIGPIPE 但设置 std::cout.exceptions(ios::badbit)，或在写入线程中处理信号。

重要的是，将始终为您的写作线程生成 SIGPIPE，尽管您的 sigwait()ing 线程也如此。线程活动产生的某些信号专门为该线程生成，这意味着它们将仅被传递给该线程或由该线程接受。 (POSIX.1-2008 System Interfaces 2.4.1) 通常，“自然发生”的 SIGPIPE、SIGFPE 和 SIGSEGV 是这样工作的。

Answer 2

这是将 SIGPIPE 转发到主线程的示例 - 在我的情况下可能就足够了：

#include <csignal>
#include <chrono>
#include <iostream>
#include <thread>

#include <signal.h>

pthread_t mainThread {pthread_self()};

void forwardSig(int sig)
{
    if (not pthread_equal(pthread_self(), mainThread))
    {
        pthread_kill(mainThread, sig);
    }
}

int main()
{
    struct sigaction newAction {};
    sigemptyset(&newAction.sa_mask);
    newAction.sa_handler = forwardSig;
    sigaction(SIGPIPE, &newAction, nullptr);

    bool run {true};
    std::thread thread {[&]
        {
            while (run)
            {
                std::cout << "ping" << std::endl;
                std::this_thread::sleep_for(std::chrono::milliseconds {500});        
            }
        }};

    sigset_t waitSet;
    sigemptyset(&waitSet);
    sigaddset(&waitSet, SIGINT);
    sigaddset(&waitSet, SIGPIPE);
    sigaddset(&waitSet, SIGTERM);

    pthread_sigmask(SIG_BLOCK, &waitSet, nullptr);

    int waitSig {0};
    sigwait(&waitSet, &waitSig);
    run = false;
    thread.join();

    std::cerr << "waitSig: "  << waitSig << std::endl;
}