Correct handling of reaper thread in forking binaries

This commit is contained in:
Thulinma 2021-03-19 22:36:27 +01:00
parent 5f8518fe23
commit dbafdf6744
5 changed files with 36 additions and 7 deletions

View file

@ -157,6 +157,28 @@ void Util::Procs::exit_handler() {
FAIL_MSG("Giving up with %d children left.", (int)listcopy.size());
}
// Joins the reaper thread, if any, before a fork
void Util::Procs::fork_prepare(){
tthread::lock_guard<tthread::mutex> guard(plistMutex);
if (handler_set){
thread_handler = false;
if (reaper_thread){
reaper_thread->join();
delete reaper_thread;
reaper_thread = 0;
}
}
}
/// Restarts reaper thread if it was joined
void Util::Procs::fork_complete(){
tthread::lock_guard<tthread::mutex> guard(plistMutex);
if (handler_set){
thread_handler = true;
reaper_thread = new tthread::thread(grim_reaper, 0);
}
}
/// Sets up exit and childsig handlers.
/// Spawns grim_reaper. exit handler despawns grim_reaper
/// Called by every Start* function.

View file

@ -26,6 +26,8 @@ namespace Util {
static bool childRunning(pid_t p);
static tthread::thread * reaper_thread;
static bool handler_set; ///< If true, the sigchld handler has been setup.
static void fork_prepare();
static void fork_complete();
static void setHandler();
static std::string getOutputOf(char * const * argv);
static std::string getOutputOf(std::deque<std::string> & argDeq);

View file

@ -245,8 +245,10 @@ namespace Util{
int pipeErr[2];
if (pipe(pipeErr) >= 0){
//Start reading log messages from the unnamed pipe
Util::Procs::fork_prepare();
pid_t pid = fork();
if (pid == 0) { //child
Util::Procs::fork_complete();
close(pipeErr[1]); // close the unneeded pipe file descriptor
//Close all sockets in the socketList
for (std::set<int>::iterator it = Util::Procs::socketList.begin(); it != Util::Procs::socketList.end(); ++it){
@ -264,6 +266,7 @@ namespace Util{
Util::logParser(pipeErr[0], true_stderr, isatty(true_stderr));
exit(0);
}
Util::Procs::fork_complete();
if (pid == -1){
FAIL_MSG("Failed to fork child process for log handling!");
}else{