auth: Properly record coverage data when exiting

diff --git a/pdns/Makefile.am b/pdns/Makefile.am
index b22bcfdc3cc8d57bc139c330e51d7697ff190382..75e0fa213f22bde20eecd1bb5637b40b8c8cfa81 100644 (file)
--- a/pdns/Makefile.am
+++ b/pdns/Makefile.am
@@ -211,6 +211,7 @@ pdns_server_SOURCES = \
        circular_buffer.hh \
        comment.hh \
        communicator.cc communicator.hh \
+       coverage.cc coverage.hh \
        credentials.cc credentials.hh \
        dbdnsseckeeper.cc \
        digests.hh \

diff --git a/pdns/auth-main.cc b/pdns/auth-main.cc
index e3d4b39b097f59c67afa5ed3b4388b4d55394bf5..559e1f29c358a6656b198e6a5b63bdbf0ce385f9 100644 (file)
--- a/pdns/auth-main.cc
+++ b/pdns/auth-main.cc
@@ -53,6 +53,7 @@
 #endif
 
 #include "auth-main.hh"
+#include "coverage.hh"
 #include "secpoll-auth.hh"
 #include "dynhandler.hh"
 #include "dnsseckeeper.hh"
@@ -1183,6 +1184,14 @@ static void tbhandler(int num)
 }
 #endif
 
+#ifdef COVERAGE
+static void sigTermHandler([[maybe_unused]] int signal)
+{
+  pdns::coverage::dumpCoverageData();
+  _exit(EXIT_SUCCESS);
+}
+#endif /* COVERAGE */
+
 //! The main function of pdns, the pdns process
 int main(int argc, char** argv)
 {
@@ -1282,6 +1291,12 @@ int main(int argc, char** argv)
       cerr << "Um, we did get here!" << endl;
     }
 
+#ifdef COVERAGE
+    if (!::arg().mustDo("guardian") && !::arg().mustDo("daemon")) {
+      signal(SIGTERM, sigTermHandler);
+    }
+#endif
+
     // we really need to do work - either standalone or as an instance
 
 #if defined(__GLIBC__) && !defined(__UCLIBC__)

diff --git a/pdns/coverage.cc b/pdns/coverage.cc
new file mode 100644 (file)
index 0000000..71fcda4
--- /dev/null
+++ b/pdns/coverage.cc
@@ -0,0 +1,48 @@
+/*
+ * This file is part of PowerDNS or dnsdist.
+ * Copyright -- PowerDNS.COM B.V. and its contributors
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of version 2 of the GNU General Public License as
+ * published by the Free Software Foundation.
+ *
+ * In addition, for the avoidance of any doubt, permission is granted to
+ * link this program with OpenSSL and to (re)distribute the binaries
+ * produced as the result of such linking.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+#include "config.h"
+#include "coverage.hh"
+
+#ifdef COVERAGE
+extern "C"
+{
+  //NOLINTNEXTLINE(bugprone-reserved-identifier): not ours
+  void __gcov_dump(void);
+#ifdef CLANG_COVERAGE
+  //NOLINTNEXTLINE(bugprone-reserved-identifier): not ours
+  int __llvm_profile_write_file(void);
+#endif /* CLANG_COVERAGE */
+}
+#endif /* COVERAGE */
+
+namespace pdns::coverage
+{
+void dumpCoverageData()
+{
+#ifdef COVERAGE
+  __gcov_dump();
+#ifdef CLANG_COVERAGE
+  __llvm_profile_write_file();
+#endif /* CLANG_COVERAGE */
+#endif /* COVERAGE */
+}
+}

diff --git a/pdns/coverage.hh b/pdns/coverage.hh
new file mode 100644 (file)
index 0000000..960f28c
--- /dev/null
+++ b/pdns/coverage.hh
@@ -0,0 +1,27 @@
+/*
+ * This file is part of PowerDNS or dnsdist.
+ * Copyright -- PowerDNS.COM B.V. and its contributors
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of version 2 of the GNU General Public License as
+ * published by the Free Software Foundation.
+ *
+ * In addition, for the avoidance of any doubt, permission is granted to
+ * link this program with OpenSSL and to (re)distribute the binaries
+ * produced as the result of such linking.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+#pragma once
+
+namespace pdns::coverage
+{
+void dumpCoverageData();
+}