1/*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1995-1998 John Birrell <jb@cimlogic.com.au>
5 * Copyright (c) 2018 The FreeBSD Foundation
6 * All rights reserved.
7 *
8 * Portions of this software were developed by Konstantin Belousov
9 * under sponsorship from the FreeBSD Foundation.
10 *
11 * Redistribution and use in source and binary forms, with or without
12 * modification, are permitted provided that the following conditions
13 * are met:
14 * 1. Redistributions of source code must retain the above copyright
15 *    notice, this list of conditions and the following disclaimer.
16 * 2. Redistributions in binary form must reproduce the above copyright
17 *    notice, this list of conditions and the following disclaimer in the
18 *    documentation and/or other materials provided with the distribution.
19 * 3. Neither the name of the author nor the names of any co-contributors
20 *    may be used to endorse or promote products derived from this software
21 *    without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY JOHN BIRRELL AND CONTRIBUTORS ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * SUCH DAMAGE.
34 */
35
36#include <sys/cdefs.h>
37__FBSDID("$FreeBSD$");
38
39#include "namespace.h"
40#include <sys/errno.h>
41#include <stdlib.h>
42#include <string.h>
43#include <pthread.h>
44#include <pthread_np.h>
45#include "un-namespace.h"
46
47#include "thr_private.h"
48
49static void
50thr_set_name_np(struct pthread *thread, char **tmp_name)
51{
52
53	free(thread->name);
54	thread->name = *tmp_name;
55	*tmp_name = NULL;
56}
57
58/* Set the thread name. */
59__weak_reference(_pthread_setname_np, pthread_setname_np);
60int
61_pthread_setname_np(pthread_t thread, const char *name)
62{
63	struct pthread *curthread;
64	char *tmp_name;
65	int res;
66
67	if (name != NULL) {
68		tmp_name = strdup(name);
69		if (tmp_name == NULL)
70			return (ENOMEM);
71	} else {
72		tmp_name = NULL;
73	}
74	curthread = _get_curthread();
75	if (curthread == thread) {
76		res = 0;
77		THR_THREAD_LOCK(curthread, thread);
78		if (thr_set_name(thread->tid, name) == -1)
79			res = errno;
80		else
81			thr_set_name_np(thread, &tmp_name);
82		THR_THREAD_UNLOCK(curthread, thread);
83	} else {
84		res = ESRCH;
85		if (_thr_find_thread(curthread, thread, 0) == 0) {
86			if (thread->state != PS_DEAD) {
87				if (thr_set_name(thread->tid, name) == -1) {
88					res = errno;
89				} else {
90					thr_set_name_np(thread, &tmp_name);
91					res = 0;
92				}
93			}
94			THR_THREAD_UNLOCK(curthread, thread);
95		}
96	}
97	free(tmp_name);
98	return (res);
99}
100
101/* Set the thread name for debug. */
102__weak_reference(_pthread_set_name_np, pthread_set_name_np);
103void
104_pthread_set_name_np(pthread_t thread, const char *name)
105{
106	(void)_pthread_setname_np(thread, name);
107}
108
109static void
110thr_get_name_np(struct pthread *thread, char *buf, size_t len)
111{
112
113	if (thread->name != NULL)
114		strlcpy(buf, thread->name, len);
115	else if (len > 0)
116		buf[0] = '\0';
117}
118
119__weak_reference(_pthread_getname_np, pthread_getname_np);
120int
121_pthread_getname_np(pthread_t thread, char *buf, size_t len)
122{
123	struct pthread *curthread;
124	int res;
125
126	res = 0;
127	curthread = _get_curthread();
128	if (curthread == thread) {
129		THR_THREAD_LOCK(curthread, thread);
130		thr_get_name_np(thread, buf, len);
131		THR_THREAD_UNLOCK(curthread, thread);
132	} else {
133		if (_thr_find_thread(curthread, thread, 0) == 0) {
134			if (thread->state != PS_DEAD)
135				thr_get_name_np(thread, buf, len);
136			else
137				res = ESRCH;
138			THR_THREAD_UNLOCK(curthread, thread);
139		} else {
140			res = ESRCH;
141			if (len > 0)
142				buf[0] = '\0';
143		}
144	}
145	return (res);
146}
147
148__weak_reference(_pthread_get_name_np, pthread_get_name_np);
149void
150_pthread_get_name_np(pthread_t thread, char *buf, size_t len)
151{
152	(void)_pthread_getname_np(thread, buf, len);
153}
154