From: Otto Moerbeek <otto.moerbeek@open-xchange.com>
Date: Fri, 8 Nov 2019 13:34:39 +0000 (+0100)
Subject: Make threads run until asked to stop.
X-Git-Tag: dnsdist-1.4.0~12^2~3
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=31a8c1b3f8663df9a265e79f0bdf132f245b3417;p=thirdparty%2Fpdns.git

Make threads run until asked to stop.

This is safer since the atexit handler is not ran while threads are
still active. Also, when using valgrind we get more clean leak reports.
---

diff --git a/pdns/pdns_recursor.cc b/pdns/pdns_recursor.cc
index 5e1d794324..bb2f617639 100644
--- a/pdns/pdns_recursor.cc
+++ b/pdns/pdns_recursor.cc
@@ -4313,7 +4313,9 @@ static int serviceMain(int argc, char*argv[])
     infos.isHandler = true;
     infos.thread = std::thread(recursorThread, 0, "web+stat");
 
-    s_threadInfos.at(0).thread.join();
+    for (auto & ti : s_threadInfos) {
+      ti.thread.join();
+    }
   }
   return 0;
 }
@@ -4448,7 +4450,8 @@ try
   time_t carbonInterval=::arg().asNum("carbon-interval");
   time_t luaMaintenanceInterval=::arg().asNum("lua-maintenance-interval");
   counter.store(0); // used to periodically execute certain tasks
-  for(;;) {
+
+  while (!RecursorControlChannel::stop) {
     while(MT->schedule(&g_now)); // MTasker letting the mthreads do their thing
 
     if(!(counter%500)) {
@@ -4516,6 +4519,7 @@ try
       }
     }
   }
+  return 0;
 }
 catch(PDNSException &ae) {
   g_log<<Logger::Error<<"Exception: "<<ae.reason<<endl;
diff --git a/pdns/rec_channel.cc b/pdns/rec_channel.cc
index 0884afaa63..65afbe092d 100644
--- a/pdns/rec_channel.cc
+++ b/pdns/rec_channel.cc
@@ -17,6 +17,8 @@
 
 #include "namespaces.hh"
 
+volatile sig_atomic_t RecursorControlChannel::stop;
+
 RecursorControlChannel::RecursorControlChannel()
 {
   d_fd=-1;
diff --git a/pdns/rec_channel.hh b/pdns/rec_channel.hh
index ee653752ba..a391190d87 100644
--- a/pdns/rec_channel.hh
+++ b/pdns/rec_channel.hh
@@ -53,6 +53,7 @@ public:
   std::string recv(std::string* remote=0, unsigned int timeout=5);
 
   int d_fd;
+  static volatile sig_atomic_t stop;
 private:
   struct sockaddr_un d_local;
 };
diff --git a/pdns/rec_channel_rec.cc b/pdns/rec_channel_rec.cc
index a86500ba12..a294e62ac7 100644
--- a/pdns/rec_channel_rec.cc
+++ b/pdns/rec_channel_rec.cc
@@ -1165,15 +1165,16 @@ void doExitGeneric(bool nicely)
 {
   g_log<<Logger::Error<<"Exiting on user request"<<endl;
   extern RecursorControlChannel s_rcc;
-  s_rcc.~RecursorControlChannel(); 
+  s_rcc.~RecursorControlChannel();
 
   extern string s_pidfname;
-  if(!s_pidfname.empty()) 
+  if(!s_pidfname.empty())
     unlink(s_pidfname.c_str()); // we can at least try..
-  if(nicely)
-    exit(1);
-  else
+  if(nicely) {
+    RecursorControlChannel::stop = 1;
+  } else {
     _exit(1);
+  }
 }
 
 void doExit()