xref: /dflybsd-src/lib/libthread_xu/thread/thr_rtld.c (revision 16fb9086b36e7c7d81241bbd6502634e40346c7b)
171b3fa15SDavid Xu /*
2d3b15642Szrj  * Copyright (c) 2006, David Xu <davidxu@freebsd.org>
371b3fa15SDavid Xu  * All rights reserved.
471b3fa15SDavid Xu  *
571b3fa15SDavid Xu  * Redistribution and use in source and binary forms, with or without
671b3fa15SDavid Xu  * modification, are permitted provided that the following conditions
771b3fa15SDavid Xu  * are met:
871b3fa15SDavid Xu  * 1. Redistributions of source code must retain the above copyright
9d3b15642Szrj  *    notice unmodified, this list of conditions, and the following
10d3b15642Szrj  *    disclaimer.
1171b3fa15SDavid Xu  * 2. Redistributions in binary form must reproduce the above copyright
1271b3fa15SDavid Xu  *    notice, this list of conditions and the following disclaimer in the
1371b3fa15SDavid Xu  *    documentation and/or other materials provided with the distribution.
1471b3fa15SDavid Xu  *
1571b3fa15SDavid Xu  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
1671b3fa15SDavid Xu  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
1771b3fa15SDavid Xu  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
1871b3fa15SDavid Xu  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
1971b3fa15SDavid Xu  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
2071b3fa15SDavid Xu  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
2171b3fa15SDavid Xu  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
2271b3fa15SDavid Xu  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
2371b3fa15SDavid Xu  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
2471b3fa15SDavid Xu  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2571b3fa15SDavid Xu  *
2671b3fa15SDavid Xu  * $FreeBSD: src/lib/libpthread/thread/thr_rtld.c,v 1.5 2003/11/05 18:19:24 deischen Exp $
2771b3fa15SDavid Xu  */
289e2ee207SJoerg Sonnenberger 
29*16fb9086SImre Vadász #include "namespace.h"
3019451dc5Szrj #include <machine/tls.h>
3171b3fa15SDavid Xu #include <stdlib.h>
3271b3fa15SDavid Xu #include <pthread.h>
3371b3fa15SDavid Xu 
34*16fb9086SImre Vadász #include "un-namespace.h"
3571b3fa15SDavid Xu #include "rtld_lock.h"
3671b3fa15SDavid Xu #include "thr_private.h"
3771b3fa15SDavid Xu 
3871b3fa15SDavid Xu static int	_thr_rtld_clr_flag(int);
3971b3fa15SDavid Xu static void	*_thr_rtld_lock_create(void);
4071b3fa15SDavid Xu static void	_thr_rtld_lock_destroy(void *);
4171b3fa15SDavid Xu static void	_thr_rtld_lock_release(void *);
4271b3fa15SDavid Xu static void	_thr_rtld_rlock_acquire(void *);
4371b3fa15SDavid Xu static int	_thr_rtld_set_flag(int);
4471b3fa15SDavid Xu static void	_thr_rtld_wlock_acquire(void *);
4571b3fa15SDavid Xu 
4671b3fa15SDavid Xu static void *
4771b3fa15SDavid Xu _thr_rtld_lock_create(void)
4871b3fa15SDavid Xu {
4971b3fa15SDavid Xu 	pthread_rwlock_t prwlock;
5071b3fa15SDavid Xu 	if (_pthread_rwlock_init(&prwlock, NULL))
5171b3fa15SDavid Xu 		return (NULL);
5271b3fa15SDavid Xu 	return (prwlock);
5371b3fa15SDavid Xu }
5471b3fa15SDavid Xu 
5571b3fa15SDavid Xu static void
5671b3fa15SDavid Xu _thr_rtld_lock_destroy(void *lock)
5771b3fa15SDavid Xu {
5871b3fa15SDavid Xu 	pthread_rwlock_t prwlock;
5971b3fa15SDavid Xu 
6071b3fa15SDavid Xu 	prwlock = (pthread_rwlock_t)lock;
6171b3fa15SDavid Xu 	if (prwlock != NULL)
6271b3fa15SDavid Xu 		_pthread_rwlock_destroy(&prwlock);
6371b3fa15SDavid Xu }
6471b3fa15SDavid Xu 
6571b3fa15SDavid Xu static void
6671b3fa15SDavid Xu _thr_rtld_rlock_acquire(void *lock)
6771b3fa15SDavid Xu {
6871b3fa15SDavid Xu 	pthread_rwlock_t prwlock;
6971b3fa15SDavid Xu 
7071b3fa15SDavid Xu 	prwlock = (pthread_rwlock_t)lock;
71*16fb9086SImre Vadász 	_pthread_rwlock_rdlock(&prwlock);
7271b3fa15SDavid Xu }
7371b3fa15SDavid Xu 
7471b3fa15SDavid Xu static void
7571b3fa15SDavid Xu _thr_rtld_wlock_acquire(void *lock)
7671b3fa15SDavid Xu {
7771b3fa15SDavid Xu 	pthread_rwlock_t prwlock;
7871b3fa15SDavid Xu 
7971b3fa15SDavid Xu 	prwlock = (pthread_rwlock_t)lock;
80*16fb9086SImre Vadász 	_pthread_rwlock_wrlock(&prwlock);
8171b3fa15SDavid Xu }
8271b3fa15SDavid Xu 
8371b3fa15SDavid Xu static void
8471b3fa15SDavid Xu _thr_rtld_lock_release(void *lock)
8571b3fa15SDavid Xu {
8671b3fa15SDavid Xu 	pthread_rwlock_t prwlock;
8771b3fa15SDavid Xu 
8871b3fa15SDavid Xu 	prwlock = (pthread_rwlock_t)lock;
89*16fb9086SImre Vadász 	_pthread_rwlock_unlock(&prwlock);
9071b3fa15SDavid Xu }
9171b3fa15SDavid Xu 
9271b3fa15SDavid Xu 
9371b3fa15SDavid Xu static int
9471b3fa15SDavid Xu _thr_rtld_set_flag(int mask)
9571b3fa15SDavid Xu {
9671b3fa15SDavid Xu 	struct pthread *curthread;
9771b3fa15SDavid Xu 	int bits;
9871b3fa15SDavid Xu 
999e2ee207SJoerg Sonnenberger 	curthread = tls_get_curthread();
10071b3fa15SDavid Xu 	if (curthread != NULL) {
10171b3fa15SDavid Xu 		bits = curthread->rtld_bits;
10271b3fa15SDavid Xu 		curthread->rtld_bits |= mask;
10371b3fa15SDavid Xu 	} else {
10471b3fa15SDavid Xu 		bits = 0;
10571b3fa15SDavid Xu 		PANIC("No current thread in rtld call");
10671b3fa15SDavid Xu 	}
10771b3fa15SDavid Xu 
10871b3fa15SDavid Xu 	return (bits);
10971b3fa15SDavid Xu }
11071b3fa15SDavid Xu 
11171b3fa15SDavid Xu static int
11271b3fa15SDavid Xu _thr_rtld_clr_flag(int mask)
11371b3fa15SDavid Xu {
11471b3fa15SDavid Xu 	struct pthread *curthread;
11571b3fa15SDavid Xu 	int bits;
11671b3fa15SDavid Xu 
1179e2ee207SJoerg Sonnenberger 	curthread = tls_get_curthread();
11871b3fa15SDavid Xu 	if (curthread != NULL) {
11971b3fa15SDavid Xu 		bits = curthread->rtld_bits;
12071b3fa15SDavid Xu 		curthread->rtld_bits &= ~mask;
12171b3fa15SDavid Xu 	} else {
12271b3fa15SDavid Xu 		bits = 0;
12371b3fa15SDavid Xu 		PANIC("No current thread in rtld call");
12471b3fa15SDavid Xu 	}
12571b3fa15SDavid Xu 	return (bits);
12671b3fa15SDavid Xu }
12771b3fa15SDavid Xu 
12871b3fa15SDavid Xu void
12971b3fa15SDavid Xu _thr_rtld_init(void)
13071b3fa15SDavid Xu {
13171b3fa15SDavid Xu 	struct RtldLockInfo li;
13271b3fa15SDavid Xu 
13371b3fa15SDavid Xu 	li.lock_create  = _thr_rtld_lock_create;
13471b3fa15SDavid Xu 	li.lock_destroy = _thr_rtld_lock_destroy;
13571b3fa15SDavid Xu 	li.rlock_acquire = _thr_rtld_rlock_acquire;
13671b3fa15SDavid Xu 	li.wlock_acquire = _thr_rtld_wlock_acquire;
13771b3fa15SDavid Xu 	li.lock_release  = _thr_rtld_lock_release;
13871b3fa15SDavid Xu 	li.thread_set_flag = _thr_rtld_set_flag;
13971b3fa15SDavid Xu 	li.thread_clr_flag = _thr_rtld_clr_flag;
14071b3fa15SDavid Xu 	li.at_fork = NULL;
14171b3fa15SDavid Xu 	_rtld_thread_init(&li);
14271b3fa15SDavid Xu }
14371b3fa15SDavid Xu 
14471b3fa15SDavid Xu void
14571b3fa15SDavid Xu _thr_rtld_fini(void)
14671b3fa15SDavid Xu {
14771b3fa15SDavid Xu 	_rtld_thread_init(NULL);
14871b3fa15SDavid Xu }
149