1bd428526SJulian Pullen /*
2bd428526SJulian Pullen * CDDL HEADER START
3bd428526SJulian Pullen *
4bd428526SJulian Pullen * The contents of this file are subject to the terms of the
5bd428526SJulian Pullen * Common Development and Distribution License (the "License").
6bd428526SJulian Pullen * You may not use this file except in compliance with the License.
7bd428526SJulian Pullen *
8bd428526SJulian Pullen * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9bd428526SJulian Pullen * or http://www.opensolaris.org/os/licensing.
10bd428526SJulian Pullen * See the License for the specific language governing permissions
11bd428526SJulian Pullen * and limitations under the License.
12bd428526SJulian Pullen *
13bd428526SJulian Pullen * When distributing Covered Code, include this CDDL HEADER in each
14bd428526SJulian Pullen * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15bd428526SJulian Pullen * If applicable, add the following below this CDDL HEADER, with the
16bd428526SJulian Pullen * fields enclosed by brackets "[]" replaced with your own identifying
17bd428526SJulian Pullen * information: Portions Copyright [yyyy] [name of copyright owner]
18bd428526SJulian Pullen *
19bd428526SJulian Pullen * CDDL HEADER END
20bd428526SJulian Pullen */
21bd428526SJulian Pullen
22bd428526SJulian Pullen /*
23bd428526SJulian Pullen * Copyright 2009 Sun Microsystems, Inc. All rights reserved.
24bd428526SJulian Pullen * Use is subject to license terms.
25bd428526SJulian Pullen */
26bd428526SJulian Pullen
27bd428526SJulian Pullen /*
28bd428526SJulian Pullen * Functions for managing thread-local storage for LDAP, and in particular
29bd428526SJulian Pullen * for managing storage of the LDAP error state.
30bd428526SJulian Pullen */
31bd428526SJulian Pullen
32bd428526SJulian Pullen #include <ldap.h>
33bd428526SJulian Pullen #include <pthread.h>
34bd428526SJulian Pullen #include <errno.h>
35bd428526SJulian Pullen #include <note.h>
36bd428526SJulian Pullen #include <syslog.h>
37bd428526SJulian Pullen #include <string.h>
38bd428526SJulian Pullen #include "solaris-int.h" /* This is a libladp5 private include file */
39bd428526SJulian Pullen /* which has the defintion for */
40bd428526SJulian Pullen /* struct ldap_extra_thread_fns */
41bd428526SJulian Pullen #include "adutils_impl.h"
42bd428526SJulian Pullen
43bd428526SJulian Pullen struct adutils_lderrno {
44bd428526SJulian Pullen int le_errno;
45bd428526SJulian Pullen char *le_matched;
46bd428526SJulian Pullen char *le_errmsg;
47bd428526SJulian Pullen };
48bd428526SJulian Pullen
49*2c961aefSToomas Soome static void *adutils_threadid(void);
50bd428526SJulian Pullen static void *adutils_mutex_alloc(void);
51bd428526SJulian Pullen static void adutils_mutex_free(void *mutexp);
52bd428526SJulian Pullen static int adutils_get_errno(void);
53bd428526SJulian Pullen static void adutils_set_errno(int err);
54bd428526SJulian Pullen static void adutils_set_lderrno(int err, char *matched, char *errmsg,
55bd428526SJulian Pullen void *dummy);
56bd428526SJulian Pullen static int adutils_get_lderrno(char **matched, char **errmsg, void *dummy);
57bd428526SJulian Pullen static void adutils_lderrno_destructor(void *tsd);
58bd428526SJulian Pullen
59bd428526SJulian Pullen static pthread_key_t adutils_lderrno_key = PTHREAD_ONCE_KEY_NP;
60bd428526SJulian Pullen
61bd428526SJulian Pullen static struct ldap_thread_fns thread_fns = {
62bd428526SJulian Pullen .ltf_mutex_alloc = adutils_mutex_alloc,
63bd428526SJulian Pullen .ltf_mutex_free = adutils_mutex_free,
64bd428526SJulian Pullen .ltf_mutex_lock = (int (*)(void *)) pthread_mutex_lock,
65bd428526SJulian Pullen .ltf_mutex_unlock = (int (*)(void *)) pthread_mutex_unlock,
66bd428526SJulian Pullen .ltf_get_errno = adutils_get_errno,
67bd428526SJulian Pullen .ltf_set_errno = adutils_set_errno,
68bd428526SJulian Pullen .ltf_get_lderrno = adutils_get_lderrno,
69bd428526SJulian Pullen .ltf_set_lderrno = adutils_set_lderrno,
70bd428526SJulian Pullen .ltf_lderrno_arg = NULL
71bd428526SJulian Pullen };
72bd428526SJulian Pullen
73bd428526SJulian Pullen struct ldap_extra_thread_fns extra_thread_fns = {
74*2c961aefSToomas Soome .ltf_threadid_fn = adutils_threadid
75bd428526SJulian Pullen };
76bd428526SJulian Pullen
77bd428526SJulian Pullen /*
78bd428526SJulian Pullen * Set up thread management functions for the specified LDAP session.
79bd428526SJulian Pullen * Returns either LDAP_SUCCESS or -1.
80bd428526SJulian Pullen */
81bd428526SJulian Pullen int
adutils_set_thread_functions(LDAP * ld)82bd428526SJulian Pullen adutils_set_thread_functions(LDAP *ld)
83bd428526SJulian Pullen {
84bd428526SJulian Pullen int rc;
85bd428526SJulian Pullen
86bd428526SJulian Pullen if (adutils_lderrno_key == PTHREAD_ONCE_KEY_NP) {
87bd428526SJulian Pullen if ((rc = pthread_key_create_once_np(&adutils_lderrno_key,
88bd428526SJulian Pullen adutils_lderrno_destructor)) != 0) {
89bd428526SJulian Pullen logger(LOG_ERR, "adutils_set_thread_functions() "
90bd428526SJulian Pullen "pthread_key_create_once_np failed (%s)",
91bd428526SJulian Pullen strerror(rc));
92bd428526SJulian Pullen rc = -1;
93bd428526SJulian Pullen return (rc);
94bd428526SJulian Pullen }
95bd428526SJulian Pullen }
96bd428526SJulian Pullen
97bd428526SJulian Pullen rc = ldap_set_option(ld, LDAP_OPT_THREAD_FN_PTRS,
98bd428526SJulian Pullen &thread_fns);
99bd428526SJulian Pullen if (rc != LDAP_SUCCESS) {
100bd428526SJulian Pullen logger(LOG_ERR,
101bd428526SJulian Pullen "ldap_set_option LDAP_OPT_THREAD_FN_PTRS failed");
102bd428526SJulian Pullen return (rc);
103bd428526SJulian Pullen }
104bd428526SJulian Pullen
105bd428526SJulian Pullen rc = ldap_set_option(ld, LDAP_OPT_EXTRA_THREAD_FN_PTRS,
106bd428526SJulian Pullen &extra_thread_fns);
107bd428526SJulian Pullen if (rc != LDAP_SUCCESS) {
108bd428526SJulian Pullen logger(LOG_ERR,
109bd428526SJulian Pullen "ldap_set_option LDAP_OPT_EXTRA_THREAD_FN_PTRS failed");
110bd428526SJulian Pullen return (rc);
111bd428526SJulian Pullen }
112bd428526SJulian Pullen return (rc);
113bd428526SJulian Pullen }
114bd428526SJulian Pullen
115*2c961aefSToomas Soome static void *
adutils_threadid(void)116*2c961aefSToomas Soome adutils_threadid(void)
117*2c961aefSToomas Soome {
118*2c961aefSToomas Soome return ((void *)(uintptr_t)pthread_self());
119*2c961aefSToomas Soome }
120*2c961aefSToomas Soome
121bd428526SJulian Pullen /*
122bd428526SJulian Pullen * Allocate a mutex.
123bd428526SJulian Pullen */
124bd428526SJulian Pullen static
125bd428526SJulian Pullen void *
adutils_mutex_alloc(void)126bd428526SJulian Pullen adutils_mutex_alloc(void)
127bd428526SJulian Pullen {
128bd428526SJulian Pullen pthread_mutex_t *mutexp;
129bd428526SJulian Pullen int rc;
130bd428526SJulian Pullen
131bd428526SJulian Pullen mutexp = malloc(sizeof (pthread_mutex_t));
132bd428526SJulian Pullen if (mutexp == NULL) {
133bd428526SJulian Pullen logger(LOG_ERR,
134bd428526SJulian Pullen "adutils_mutex_alloc: malloc failed (%s)",
135bd428526SJulian Pullen strerror(errno));
136bd428526SJulian Pullen return (NULL);
137bd428526SJulian Pullen }
138bd428526SJulian Pullen
139bd428526SJulian Pullen rc = pthread_mutex_init(mutexp, NULL);
140bd428526SJulian Pullen if (rc != 0) {
141bd428526SJulian Pullen logger(LOG_ERR,
142bd428526SJulian Pullen "adutils_mutex_alloc: "
143bd428526SJulian Pullen "pthread_mutex_init failed (%s)",
144bd428526SJulian Pullen strerror(rc));
145bd428526SJulian Pullen free(mutexp);
146bd428526SJulian Pullen return (NULL);
147bd428526SJulian Pullen }
148bd428526SJulian Pullen return (mutexp);
149bd428526SJulian Pullen }
150bd428526SJulian Pullen
151bd428526SJulian Pullen /*
152bd428526SJulian Pullen * Free a mutex.
153bd428526SJulian Pullen */
154bd428526SJulian Pullen static
155bd428526SJulian Pullen void
adutils_mutex_free(void * mutexp)156bd428526SJulian Pullen adutils_mutex_free(void *mutexp)
157bd428526SJulian Pullen {
158bd428526SJulian Pullen (void) pthread_mutex_destroy((pthread_mutex_t *)mutexp);
159bd428526SJulian Pullen free(mutexp);
160bd428526SJulian Pullen }
161bd428526SJulian Pullen
162bd428526SJulian Pullen /*
163bd428526SJulian Pullen * Get the thread's local errno.
164bd428526SJulian Pullen */
165bd428526SJulian Pullen static
166bd428526SJulian Pullen int
adutils_get_errno(void)167bd428526SJulian Pullen adutils_get_errno(void)
168bd428526SJulian Pullen {
169bd428526SJulian Pullen return (errno);
170bd428526SJulian Pullen }
171bd428526SJulian Pullen
172bd428526SJulian Pullen /*
173bd428526SJulian Pullen * Set the thread's local errno.
174bd428526SJulian Pullen */
175bd428526SJulian Pullen static
176bd428526SJulian Pullen void
adutils_set_errno(int err)177bd428526SJulian Pullen adutils_set_errno(int err)
178bd428526SJulian Pullen {
179bd428526SJulian Pullen errno = err;
180bd428526SJulian Pullen }
181bd428526SJulian Pullen
182bd428526SJulian Pullen /*
183bd428526SJulian Pullen * Get a pointer to the thread's local LDAP error state structure.
184bd428526SJulian Pullen * Lazily allocate the thread-local storage, so that we don't need
185bd428526SJulian Pullen * initialization when each thread starts.
186bd428526SJulian Pullen */
187bd428526SJulian Pullen static
188bd428526SJulian Pullen struct adutils_lderrno *
adutils_get_lderrno_struct(void)189bd428526SJulian Pullen adutils_get_lderrno_struct(void)
190bd428526SJulian Pullen {
191bd428526SJulian Pullen struct adutils_lderrno *le;
192bd428526SJulian Pullen int rc;
193bd428526SJulian Pullen
194bd428526SJulian Pullen le = pthread_getspecific(adutils_lderrno_key);
195bd428526SJulian Pullen if (le == NULL) {
196bd428526SJulian Pullen le = calloc(1, sizeof (*le));
197bd428526SJulian Pullen if (le == NULL) {
198bd428526SJulian Pullen logger(LOG_ERR,
199bd428526SJulian Pullen "adutils_get_lderrno_struct: calloc failed (%s)",
200bd428526SJulian Pullen strerror(errno));
201bd428526SJulian Pullen return (NULL);
202bd428526SJulian Pullen }
203bd428526SJulian Pullen rc = pthread_setspecific(adutils_lderrno_key, le);
204bd428526SJulian Pullen if (rc != 0) {
205bd428526SJulian Pullen logger(LOG_ERR,
206bd428526SJulian Pullen "adutils_get_lderrno_struct: "
207bd428526SJulian Pullen "pthread_setspecific failed (%s)",
208bd428526SJulian Pullen strerror(rc));
209bd428526SJulian Pullen free(le);
210bd428526SJulian Pullen return (NULL);
211bd428526SJulian Pullen }
212bd428526SJulian Pullen }
213bd428526SJulian Pullen
214bd428526SJulian Pullen return (le);
215bd428526SJulian Pullen }
216bd428526SJulian Pullen
217bd428526SJulian Pullen /*
218bd428526SJulian Pullen * Store an error report in the thread's local LDAP error state structure.
219bd428526SJulian Pullen */
220bd428526SJulian Pullen static
221bd428526SJulian Pullen void
adutils_set_lderrno(int err,char * matched,char * errmsg,void * dummy)222bd428526SJulian Pullen adutils_set_lderrno(int err, char *matched, char *errmsg, void *dummy)
223bd428526SJulian Pullen {
224bd428526SJulian Pullen NOTE(ARGUNUSED(dummy))
225bd428526SJulian Pullen struct adutils_lderrno *le;
226bd428526SJulian Pullen
227bd428526SJulian Pullen le = adutils_get_lderrno_struct();
228bd428526SJulian Pullen if (le != NULL) {
229bd428526SJulian Pullen le->le_errno = err;
230bd428526SJulian Pullen if (le->le_matched != NULL)
231bd428526SJulian Pullen ldap_memfree(le->le_matched);
232bd428526SJulian Pullen le->le_matched = matched;
233bd428526SJulian Pullen if (le->le_errmsg != NULL)
234bd428526SJulian Pullen ldap_memfree(le->le_errmsg);
235bd428526SJulian Pullen le->le_errmsg = errmsg;
236bd428526SJulian Pullen }
237bd428526SJulian Pullen }
238bd428526SJulian Pullen
239bd428526SJulian Pullen /*
240bd428526SJulian Pullen * Retrieve an error report from the thread's local LDAP error state structure.
241bd428526SJulian Pullen */
242bd428526SJulian Pullen static
243bd428526SJulian Pullen int
adutils_get_lderrno(char ** matched,char ** errmsg,void * dummy)244bd428526SJulian Pullen adutils_get_lderrno(char **matched, char **errmsg, void *dummy)
245bd428526SJulian Pullen {
246bd428526SJulian Pullen NOTE(ARGUNUSED(dummy))
247bd428526SJulian Pullen struct adutils_lderrno *le;
248bd428526SJulian Pullen static struct adutils_lderrno empty = { LDAP_SUCCESS, NULL, NULL };
249bd428526SJulian Pullen
250bd428526SJulian Pullen le = adutils_get_lderrno_struct();
251bd428526SJulian Pullen if (le == NULL)
252bd428526SJulian Pullen le = ∅
253bd428526SJulian Pullen
254bd428526SJulian Pullen if (matched != NULL)
255bd428526SJulian Pullen *matched = le->le_matched;
256bd428526SJulian Pullen if (errmsg != NULL)
257bd428526SJulian Pullen *errmsg = le->le_errmsg;
258bd428526SJulian Pullen return (le->le_errno);
259bd428526SJulian Pullen }
260bd428526SJulian Pullen
261bd428526SJulian Pullen /*
262bd428526SJulian Pullen * Free the thread's local LDAP error state structure.
263bd428526SJulian Pullen */
264bd428526SJulian Pullen static
265bd428526SJulian Pullen void
adutils_lderrno_destructor(void * tsd)266bd428526SJulian Pullen adutils_lderrno_destructor(void *tsd)
267bd428526SJulian Pullen {
268bd428526SJulian Pullen struct adutils_lderrno *le = tsd;
269bd428526SJulian Pullen
270bd428526SJulian Pullen if (le == NULL)
271bd428526SJulian Pullen return;
272bd428526SJulian Pullen
273bd428526SJulian Pullen if (le->le_matched != NULL) {
274bd428526SJulian Pullen ldap_memfree(le->le_matched);
275bd428526SJulian Pullen le->le_matched = NULL;
276bd428526SJulian Pullen }
277bd428526SJulian Pullen if (le->le_errmsg != NULL) {
278bd428526SJulian Pullen ldap_memfree(le->le_errmsg);
279bd428526SJulian Pullen le->le_errmsg = NULL;
280bd428526SJulian Pullen }
281bd428526SJulian Pullen free(le);
282bd428526SJulian Pullen }
283