xref: /dflybsd-src/lib/libc/include/nss_tls.h (revision 86d7f5d305c6adaa56ff4582ece9859d73106103)
186d7f5d3SJohn Marino /*-
286d7f5d3SJohn Marino  * Copyright (c) 2003 Networks Associates Technology, Inc.
386d7f5d3SJohn Marino  * All rights reserved.
486d7f5d3SJohn Marino  *
586d7f5d3SJohn Marino  * This software was developed for the FreeBSD Project by
686d7f5d3SJohn Marino  * Jacques A. Vidrine, Safeport Network Services, and Network
786d7f5d3SJohn Marino  * Associates Laboratories, the Security Research Division of Network
886d7f5d3SJohn Marino  * Associates, Inc. under DARPA/SPAWAR contract N66001-01-C-8035
986d7f5d3SJohn Marino  * ("CBOSS"), as part of the DARPA CHATS research program.
1086d7f5d3SJohn Marino  *
1186d7f5d3SJohn Marino  * Redistribution and use in source and binary forms, with or without
1286d7f5d3SJohn Marino  * modification, are permitted provided that the following conditions
1386d7f5d3SJohn Marino  * are met:
1486d7f5d3SJohn Marino  * 1. Redistributions of source code must retain the above copyright
1586d7f5d3SJohn Marino  *    notice, this list of conditions and the following disclaimer.
1686d7f5d3SJohn Marino  * 2. Redistributions in binary form must reproduce the above copyright
1786d7f5d3SJohn Marino  *    notice, this list of conditions and the following disclaimer in the
1886d7f5d3SJohn Marino  *    documentation and/or other materials provided with the distribution.
1986d7f5d3SJohn Marino  *
2086d7f5d3SJohn Marino  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
2186d7f5d3SJohn Marino  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
2286d7f5d3SJohn Marino  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2386d7f5d3SJohn Marino  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
2486d7f5d3SJohn Marino  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
2586d7f5d3SJohn Marino  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
2686d7f5d3SJohn Marino  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
2786d7f5d3SJohn Marino  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
2886d7f5d3SJohn Marino  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
2986d7f5d3SJohn Marino  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
3086d7f5d3SJohn Marino  * SUCH DAMAGE.
3186d7f5d3SJohn Marino  *
3286d7f5d3SJohn Marino  * $FreeBSD: src/lib/libc/include/nss_tls.h,v 1.3 2004/03/30 15:56:15 nectar Exp $
3386d7f5d3SJohn Marino  *
3486d7f5d3SJohn Marino  * Macros which generate thread local storage handling code in NSS modules.
3586d7f5d3SJohn Marino  */
3686d7f5d3SJohn Marino #ifndef _NSS_TLS_H_
3786d7f5d3SJohn Marino #define _NSS_TLS_H_
3886d7f5d3SJohn Marino 
3986d7f5d3SJohn Marino #define NSS_TLS_HANDLING(name)					\
4086d7f5d3SJohn Marino static pthread_key_t name##_state_key;				\
4186d7f5d3SJohn Marino static	void	 name##_keyinit(void);				\
4286d7f5d3SJohn Marino static	int	 name##_getstate(struct name##_state **);	\
4386d7f5d3SJohn Marino \
4486d7f5d3SJohn Marino static void								\
4586d7f5d3SJohn Marino name##_keyinit(void)							\
4686d7f5d3SJohn Marino {									\
4786d7f5d3SJohn Marino 	_pthread_key_create(&name##_state_key, name##_endstate);	\
4886d7f5d3SJohn Marino }									\
4986d7f5d3SJohn Marino \
5086d7f5d3SJohn Marino static int							\
5186d7f5d3SJohn Marino name##_getstate(struct name##_state **p)			\
5286d7f5d3SJohn Marino {								\
5386d7f5d3SJohn Marino 	static struct name##_state st;				\
5486d7f5d3SJohn Marino 	static pthread_once_t	keyinit = PTHREAD_ONCE_INIT;	\
5586d7f5d3SJohn Marino 	int			rv;				\
5686d7f5d3SJohn Marino 								\
5786d7f5d3SJohn Marino 	if (!__isthreaded || _pthread_main_np() != 0) {		\
5886d7f5d3SJohn Marino 		*p = &st;					\
5986d7f5d3SJohn Marino 		return (0);					\
6086d7f5d3SJohn Marino 	}							\
6186d7f5d3SJohn Marino 	rv = _pthread_once(&keyinit, name##_keyinit);		\
6286d7f5d3SJohn Marino 	if (rv != 0)						\
6386d7f5d3SJohn Marino 		return (rv);					\
6486d7f5d3SJohn Marino 	*p = _pthread_getspecific(name##_state_key);		\
6586d7f5d3SJohn Marino 	if (*p != NULL)						\
6686d7f5d3SJohn Marino 		return (0);					\
6786d7f5d3SJohn Marino 	*p = calloc(1, sizeof(**p));				\
6886d7f5d3SJohn Marino 	if (*p == NULL)						\
6986d7f5d3SJohn Marino 		return (ENOMEM);				\
7086d7f5d3SJohn Marino 	rv = _pthread_setspecific(name##_state_key, *p);	\
7186d7f5d3SJohn Marino 	if (rv != 0) {						\
7286d7f5d3SJohn Marino 		free(*p);					\
7386d7f5d3SJohn Marino 		*p = NULL;					\
7486d7f5d3SJohn Marino 	}							\
7586d7f5d3SJohn Marino 	return (rv);						\
7686d7f5d3SJohn Marino }								\
7786d7f5d3SJohn Marino /* allow the macro invocation to end with a semicolon */	\
7886d7f5d3SJohn Marino struct _clashproof_bmVjdGFy
7986d7f5d3SJohn Marino 
8086d7f5d3SJohn Marino #endif /* _NSS_TLS_H_ */
81