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 #if 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 #endif
  39
  40 #include "threadname.hh"
  41
  42 void setThreadName(const std::string& threadName) {
  43   int retval = 0;
  44
  45 #ifdef HAVE_PTHREAD_SETNAME_NP_2
  46   retval = pthread_setname_np(pthread_self(), threadName.c_str());
  47 #endif
  48 #ifdef HAVE_PTHREAD_SET_NAME_NP_2
  49   retval = pthread_set_name_np(pthread_self(), threadName.c_str());
  50 #endif
  51 #ifdef HAVE_PTHREAD_SET_NAME_NP_2_VOID
  52   pthread_set_name_np(pthread_self(), threadName.c_str());
  53 #endif
  54 #ifdef HAVE_PTHREAD_SETNAME_NP_1
  55   retval = pthread_setname_np(threadName.c_str());
  56 #endif
  57 #ifdef HAVE_PTHREAD_SETNAME_NP_3
  58   retval = pthread_setname_np(pthread_self(), threadName.c_str(), nullptr);
  59 #endif
  60
  61   if (retval != 0) {
  62 #ifdef DNSDIST
  63     warnlog("Could not set thread name %s for thread: %s", threadName, strerror(retval));
  64 #else
  65     g_log<<Logger::Warning<<"Could not set thread name "<<threadName<<" for thread: "<<strerror(retval)<<endl;
  66 #endif
  67   }
  68 }
  69