Remove signalThread
and move reaping to catchSignals
This commit is contained in:
78
src/main.cpp
78
src/main.cpp
@ -14,64 +14,6 @@ std::mutex reap_mtx;
|
|||||||
std::list<pid_t> reap;
|
std::list<pid_t> reap;
|
||||||
volatile bool reload;
|
volatile bool reload;
|
||||||
|
|
||||||
void* signalThread(void* args) {
|
|
||||||
int err;
|
|
||||||
int signum;
|
|
||||||
sigset_t mask;
|
|
||||||
sigemptyset(&mask);
|
|
||||||
sigaddset(&mask, SIGCHLD);
|
|
||||||
|
|
||||||
while (true) {
|
|
||||||
err = sigwait(&mask, &signum);
|
|
||||||
if (err != 0) {
|
|
||||||
spdlog::error("sigwait failed: {}", strerror(errno));
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (signum) {
|
|
||||||
case SIGCHLD:
|
|
||||||
spdlog::debug("Received SIGCHLD in signalThread");
|
|
||||||
if (!reap.empty()) {
|
|
||||||
reap_mtx.lock();
|
|
||||||
for (auto it = reap.begin(); it != reap.end(); ++it) {
|
|
||||||
if (waitpid(*it, nullptr, WNOHANG) == *it) {
|
|
||||||
spdlog::debug("Reaped child with PID: {}", *it);
|
|
||||||
it = reap.erase(it);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
reap_mtx.unlock();
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
spdlog::debug("Received signal with number {}, but not handling", signum);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void startSignalThread() {
|
|
||||||
int err;
|
|
||||||
sigset_t mask;
|
|
||||||
sigemptyset(&mask);
|
|
||||||
sigaddset(&mask, SIGCHLD);
|
|
||||||
|
|
||||||
// Block SIGCHLD so it can be handled by the signal thread
|
|
||||||
// Any threads created by this one (the main thread) should not
|
|
||||||
// modify their signal mask to unblock SIGCHLD
|
|
||||||
err = pthread_sigmask(SIG_BLOCK, &mask, nullptr);
|
|
||||||
if (err != 0) {
|
|
||||||
spdlog::error("pthread_sigmask failed in startSignalThread: {}", strerror(err));
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
pthread_t thread_id;
|
|
||||||
err = pthread_create(&thread_id, nullptr, signalThread, nullptr);
|
|
||||||
if (err != 0) {
|
|
||||||
spdlog::error("pthread_create failed in startSignalThread: {}", strerror(err));
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static int signal_pipe_write_fd;
|
static int signal_pipe_write_fd;
|
||||||
|
|
||||||
// Write a single signal to `signal_pipe_write_fd`.
|
// Write a single signal to `signal_pipe_write_fd`.
|
||||||
@ -87,8 +29,8 @@ static void writeSignalToPipe(int signum) {
|
|||||||
// This initializes `signal_pipe_write_fd`, and sets up signal handlers.
|
// This initializes `signal_pipe_write_fd`, and sets up signal handlers.
|
||||||
//
|
//
|
||||||
// This function will run forever, emitting every `SIGUSR1`, `SIGUSR2`,
|
// This function will run forever, emitting every `SIGUSR1`, `SIGUSR2`,
|
||||||
// `SIGINT`, and `SIGRTMIN + 1`...`SIGRTMAX` signal received to
|
// `SIGINT`, `SIGCHLD`, and `SIGRTMIN + 1`...`SIGRTMAX` signal received
|
||||||
// `signal_handler`.
|
// to `signal_handler`.
|
||||||
static void catchSignals(waybar::SafeSignal<int> &signal_handler) {
|
static void catchSignals(waybar::SafeSignal<int> &signal_handler) {
|
||||||
int fd[2];
|
int fd[2];
|
||||||
pipe(fd);
|
pipe(fd);
|
||||||
@ -108,6 +50,7 @@ static void catchSignals(waybar::SafeSignal<int> &signal_handler) {
|
|||||||
std::signal(SIGUSR1, writeSignalToPipe);
|
std::signal(SIGUSR1, writeSignalToPipe);
|
||||||
std::signal(SIGUSR2, writeSignalToPipe);
|
std::signal(SIGUSR2, writeSignalToPipe);
|
||||||
std::signal(SIGINT, writeSignalToPipe);
|
std::signal(SIGINT, writeSignalToPipe);
|
||||||
|
std::signal(SIGCHLD, writeSignalToPipe);
|
||||||
|
|
||||||
for (int sig = SIGRTMIN + 1; sig <= SIGRTMAX; ++sig) {
|
for (int sig = SIGRTMIN + 1; sig <= SIGRTMAX; ++sig) {
|
||||||
std::signal(sig, writeSignalToPipe);
|
std::signal(sig, writeSignalToPipe);
|
||||||
@ -156,6 +99,19 @@ static void handleSignalMainThread(int signum) {
|
|||||||
reload = false;
|
reload = false;
|
||||||
waybar::Client::inst()->reset();
|
waybar::Client::inst()->reset();
|
||||||
break;
|
break;
|
||||||
|
case SIGCHLD:
|
||||||
|
spdlog::debug("Received SIGCHLD in signalThread");
|
||||||
|
if (!reap.empty()) {
|
||||||
|
reap_mtx.lock();
|
||||||
|
for (auto it = reap.begin(); it != reap.end(); ++it) {
|
||||||
|
if (waitpid(*it, nullptr, WNOHANG) == *it) {
|
||||||
|
spdlog::debug("Reaped child with PID: {}", *it);
|
||||||
|
it = reap.erase(it);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
reap_mtx.unlock();
|
||||||
|
}
|
||||||
|
break;
|
||||||
default:
|
default:
|
||||||
spdlog::debug("Received signal with number {}, but not handling", signum);
|
spdlog::debug("Received signal with number {}, but not handling", signum);
|
||||||
break;
|
break;
|
||||||
@ -166,8 +122,6 @@ int main(int argc, char* argv[]) {
|
|||||||
try {
|
try {
|
||||||
auto* client = waybar::Client::inst();
|
auto* client = waybar::Client::inst();
|
||||||
|
|
||||||
startSignalThread();
|
|
||||||
|
|
||||||
waybar::SafeSignal<int> posix_signal_received;
|
waybar::SafeSignal<int> posix_signal_received;
|
||||||
posix_signal_received.connect([&](int signum) {
|
posix_signal_received.connect([&](int signum) {
|
||||||
handleSignalMainThread(signum);
|
handleSignalMainThread(signum);
|
||||||
|
Reference in New Issue
Block a user