diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 21:11:59 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 21:11:59 +0000 |
commit | 3cd01b932e1c85394272ae64fae67ebeda92fb00 (patch) | |
tree | c5a3115d710afc1879ddea5349362a2bc651733c /threadname.cc | |
parent | Initial commit. (diff) | |
download | dnsdist-3cd01b932e1c85394272ae64fae67ebeda92fb00.tar.xz dnsdist-3cd01b932e1c85394272ae64fae67ebeda92fb00.zip |
Adding upstream version 1.8.3.upstream/1.8.3
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | threadname.cc | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/threadname.cc b/threadname.cc new file mode 100644 index 0000000..39e904c --- /dev/null +++ b/threadname.cc @@ -0,0 +1,81 @@ +/* + * 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 <string.h> + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include <pthread.h> + +#if HAVE_PTHREAD_NP_H +#include <pthread_np.h> +#endif + +#ifdef DNSDIST +#include "dolog.hh" +#else +#include "logger.hh" +#include "logging.hh" +#endif + +#include "threadname.hh" + +static int trySetThreadName(const std::string& threadName) { + int retval = 0; + +#ifdef HAVE_PTHREAD_SETNAME_NP_2 + retval = pthread_setname_np(pthread_self(), threadName.c_str()); +#endif +#ifdef HAVE_PTHREAD_SET_NAME_NP_2 + retval = pthread_set_name_np(pthread_self(), threadName.c_str()); +#endif +#ifdef HAVE_PTHREAD_SET_NAME_NP_2_VOID + pthread_set_name_np(pthread_self(), threadName.c_str()); +#endif +#ifdef HAVE_PTHREAD_SETNAME_NP_1 + retval = pthread_setname_np(threadName.c_str()); +#endif +#ifdef HAVE_PTHREAD_SETNAME_NP_3 + retval = pthread_setname_np(pthread_self(), threadName.c_str(), nullptr); +#endif + + return retval; +} + +void setThreadName(const std::string& threadName) { + int retval = trySetThreadName(threadName); + if (retval == ERANGE) { + const std::string shortThreadName(threadName.substr(0, 15)); + retval = trySetThreadName(shortThreadName); + } + + if (retval != 0) { +#ifdef DNSDIST + warnlog("Could not set thread name %s for thread: %s", threadName, strerror(retval)); +#else + SLOG(g_log<<Logger::Warning<<"Could not set thread name "<<threadName<<" for thread: "<<strerror(retval)<<endl, + g_slog->withName("runtime")->error(Logr::Warning, retval, "Could not set thread name", "name", Logging::Loggable(threadName))); +#endif + } +} + |