pdns/threadname.cc

   1 /*
   2  * This file is part of PowerDNS or dnsdist.
   3  * Copyright -- PowerDNS.COM B.V. and its contributors
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of version 2 of the GNU General Public License as
   7  * published by the Free Software Foundation.
   8  *
   9  * In addition, for the avoidance of any doubt, permission is granted to
  10  * link this program with OpenSSL and to (re)distribute the binaries
  11  * produced as the result of such linking.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21  */
  22 #include <string.h>
  23
  24 #ifdef HAVE_CONFIG_H
  25 #include "config.h"
  26 #endif
  27
  28 #include <pthread.h>
  29
  30 #ifdef HAVE_PTHREAD_NP_H
  31 #include <pthread_np.h>
  32 #endif
  33
  34 #ifdef DNSDIST
  35 #include "dolog.hh"
  36 #else
  37 #include "logger.hh"
  38 #include "logging.hh"
  39 #endif
  40
  41 #include "threadname.hh"
  42
  43 static int trySetThreadName(const std::string& threadName) {
  44   int retval = 0;
  45
  46 #ifdef HAVE_PTHREAD_SETNAME_NP_2
  47   retval = pthread_setname_np(pthread_self(), threadName.c_str());
  48 #endif
  49 #ifdef HAVE_PTHREAD_SET_NAME_NP_2
  50   retval = pthread_set_name_np(pthread_self(), threadName.c_str());
  51 #endif
  52 #ifdef HAVE_PTHREAD_SET_NAME_NP_2_VOID
  53   pthread_set_name_np(pthread_self(), threadName.c_str());
  54 #endif
  55 #ifdef HAVE_PTHREAD_SETNAME_NP_1
  56   retval = pthread_setname_np(threadName.c_str());
  57 #endif
  58 #ifdef HAVE_PTHREAD_SETNAME_NP_3
  59   retval = pthread_setname_np(pthread_self(), threadName.c_str(), nullptr);
  60 #endif
  61
  62   return retval;
  63 }
  64
  65 void setThreadName(const std::string& threadName) {
  66   int retval = trySetThreadName(threadName);
  67   if (retval == ERANGE) {
  68     const std::string shortThreadName(threadName.substr(0, 15));
  69     retval = trySetThreadName(shortThreadName);
  70   }
  71
  72   if (retval != 0) {
  73 #ifdef DNSDIST
  74     warnlog("Could not set thread name %s for thread: %s", threadName, strerror(retval));
  75 #else
  76     SLOG(g_log<<Logger::Warning<<"Could not set thread name "<<threadName<<" for thread: "<<strerror(retval)<<endl,
  77          g_slog->withName("runtime")->error(Logr::Warning, retval, "Could not set thread name", "name", Logging::Loggable(threadName)));
  78 #endif
  79   }
  80 }
  81