Fix invalid free of memory allocated during rtld init
[glibc.git] / nss / getXXbyYY_r.c
blob1067744d63378741a8f2f28437820f41168ecb2f
1 /* Copyright (C) 1996-2013 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3 Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, see
17 <http://www.gnu.org/licenses/>. */
19 #include <assert.h>
20 #include <atomic.h>
21 #include <errno.h>
22 #include <stdbool.h>
23 #include "nsswitch.h"
24 #include "sysdep.h"
25 #ifdef USE_NSCD
26 # include <nscd/nscd_proto.h>
27 #endif
28 #ifdef NEED__RES_HCONF
29 # include <resolv/res_hconf.h>
30 #endif
31 #ifdef NEED__RES
32 # include <resolv.h>
33 #endif
34 /*******************************************************************\
35 |* Here we assume several symbols to be defined: *|
36 |* *|
37 |* LOOKUP_TYPE - the return type of the function *|
38 |* *|
39 |* FUNCTION_NAME - name of the non-reentrant function *|
40 |* *|
41 |* DATABASE_NAME - name of the database the function accesses *|
42 |* (e.g., host, services, ...) *|
43 |* *|
44 |* ADD_PARAMS - additional parameters, can vary in number *|
45 |* *|
46 |* ADD_VARIABLES - names of additional parameters *|
47 |* *|
48 |* Optionally the following vars can be defined: *|
49 |* *|
50 |* EXTRA_PARAMS - optional parameters, can vary in number *|
51 |* *|
52 |* EXTRA_VARIABLES - names of optional parameter *|
53 |* *|
54 |* FUNCTION2_NAME - alternative name of the non-reentrant function *|
55 |* *|
56 |* NEED_H_ERRNO - an extra parameter will be passed to point to *|
57 |* the global `h_errno' variable. *|
58 |* *|
59 |* NEED__RES - the global _res variable might be used so we *|
60 |* will have to initialize it if necessary *|
61 |* *|
62 |* PREPROCESS - code run before anything else *|
63 |* *|
64 |* POSTPROCESS - code run after the lookup *|
65 |* *|
66 \*******************************************************************/
68 /* To make the real sources a bit prettier. */
69 #define REENTRANT_NAME APPEND_R (FUNCTION_NAME)
70 #ifdef FUNCTION2_NAME
71 # define REENTRANT2_NAME APPEND_R (FUNCTION2_NAME)
72 #else
73 # define REENTRANT2_NAME NULL
74 #endif
75 #define APPEND_R(name) APPEND_R1 (name)
76 #define APPEND_R1(name) name##_r
77 #define INTERNAL(name) INTERNAL1 (name)
78 #define INTERNAL1(name) __##name
79 #define NEW(name) NEW1 (name)
80 #define NEW1(name) __new_##name
82 #ifdef USE_NSCD
83 # define NSCD_NAME ADD_NSCD (REENTRANT_NAME)
84 # define ADD_NSCD(name) ADD_NSCD1 (name)
85 # define ADD_NSCD1(name) __nscd_##name
86 # define NOT_USENSCD_NAME ADD_NOT_NSCDUSE (DATABASE_NAME)
87 # define ADD_NOT_NSCDUSE(name) ADD_NOT_NSCDUSE1 (name)
88 # define ADD_NOT_NSCDUSE1(name) __nss_not_use_nscd_##name
89 # define CONCAT2(arg1, arg2) CONCAT2_2 (arg1, arg2)
90 # define CONCAT2_2(arg1, arg2) arg1##arg2
91 #endif
93 #define FUNCTION_NAME_STRING STRINGIZE (FUNCTION_NAME)
94 #define REENTRANT_NAME_STRING STRINGIZE (REENTRANT_NAME)
95 #ifdef FUNCTION2_NAME
96 # define REENTRANT2_NAME_STRING STRINGIZE (REENTRANT2_NAME)
97 #else
98 # define REENTRANT2_NAME_STRING NULL
99 #endif
100 #define DATABASE_NAME_STRING STRINGIZE (DATABASE_NAME)
101 #define STRINGIZE(name) STRINGIZE1 (name)
102 #define STRINGIZE1(name) #name
104 #ifndef DB_LOOKUP_FCT
105 # define DB_LOOKUP_FCT CONCAT3_1 (__nss_, DATABASE_NAME, _lookup2)
106 # define CONCAT3_1(Pre, Name, Post) CONCAT3_2 (Pre, Name, Post)
107 # define CONCAT3_2(Pre, Name, Post) Pre##Name##Post
108 #endif
110 /* Sometimes we need to store error codes in the `h_errno' variable. */
111 #ifdef NEED_H_ERRNO
112 # define H_ERRNO_PARM , int *h_errnop
113 # define H_ERRNO_VAR , h_errnop
114 # define H_ERRNO_VAR_P h_errnop
115 #else
116 # define H_ERRNO_PARM
117 # define H_ERRNO_VAR
118 # define H_ERRNO_VAR_P NULL
119 #endif
121 #ifndef EXTRA_PARAMS
122 # define EXTRA_PARAMS
123 #endif
124 #ifndef EXTRA_VARIABLES
125 # define EXTRA_VARIABLES
126 #endif
128 #ifdef HAVE_AF
129 # define AF_VAL af
130 #else
131 # define AF_VAL AF_INET
132 #endif
134 /* Type of the lookup function we need here. */
135 typedef enum nss_status (*lookup_function) (ADD_PARAMS, LOOKUP_TYPE *, char *,
136 size_t, int * H_ERRNO_PARM
137 EXTRA_PARAMS);
139 /* The lookup function for the first entry of this service. */
140 extern int DB_LOOKUP_FCT (service_user **nip, const char *name,
141 const char *name2, void **fctp)
142 internal_function;
143 libc_hidden_proto (DB_LOOKUP_FCT)
147 INTERNAL (REENTRANT_NAME) (ADD_PARAMS, LOOKUP_TYPE *resbuf, char *buffer,
148 size_t buflen, LOOKUP_TYPE **result H_ERRNO_PARM
149 EXTRA_PARAMS)
151 static bool startp_initialized;
152 static service_user *startp;
153 static lookup_function start_fct;
154 service_user *nip;
155 union
157 lookup_function l;
158 void *ptr;
159 } fct;
161 int no_more;
162 enum nss_status status = NSS_STATUS_UNAVAIL;
163 #ifdef USE_NSCD
164 int nscd_status;
165 #endif
166 #ifdef NEED_H_ERRNO
167 bool any_service = false;
168 #endif
170 #ifdef PREPROCESS
171 PREPROCESS;
172 #endif
174 #ifdef HANDLE_DIGITS_DOTS
175 switch (__nss_hostname_digits_dots (name, resbuf, &buffer, NULL,
176 buflen, result, &status, AF_VAL,
177 H_ERRNO_VAR_P))
179 case -1:
180 return errno;
181 case 1:
182 goto done;
184 #endif
186 #ifdef USE_NSCD
187 if (NOT_USENSCD_NAME > 0 && ++NOT_USENSCD_NAME > NSS_NSCD_RETRY)
188 NOT_USENSCD_NAME = 0;
190 if (!NOT_USENSCD_NAME
191 && !__nss_database_custom[CONCAT2 (NSS_DBSIDX_, DATABASE_NAME)])
193 nscd_status = NSCD_NAME (ADD_VARIABLES, resbuf, buffer, buflen, result
194 H_ERRNO_VAR);
195 if (nscd_status >= 0)
196 return nscd_status;
198 #endif
200 if (! startp_initialized)
202 no_more = DB_LOOKUP_FCT (&nip, REENTRANT_NAME_STRING,
203 REENTRANT2_NAME_STRING, &fct.ptr);
204 if (no_more)
206 void *tmp_ptr = (service_user *) -1l;
207 #ifdef PTR_MANGLE
208 PTR_MANGLE (tmp_ptr);
209 #endif
210 startp = tmp_ptr;
212 else
214 #ifdef NEED__RES
215 /* The resolver code will really be used so we have to
216 initialize it. */
217 if (__res_maybe_init (&_res, 0) == -1)
219 *h_errnop = NETDB_INTERNAL;
220 *result = NULL;
221 return errno;
223 #endif /* need _res */
224 #ifdef NEED__RES_HCONF
225 if (!_res_hconf.initialized)
226 _res_hconf_init ();
227 #endif /* need _res_hconf */
229 void *tmp_ptr = fct.l;
230 #ifdef PTR_MANGLE
231 PTR_MANGLE (tmp_ptr);
232 #endif
233 start_fct = tmp_ptr;
234 tmp_ptr = nip;
235 #ifdef PTR_MANGLE
236 PTR_MANGLE (tmp_ptr);
237 #endif
238 startp = tmp_ptr;
241 /* Make sure start_fct and startp are written before
242 startp_initialized. */
243 atomic_write_barrier ();
244 startp_initialized = true;
246 else
248 fct.l = start_fct;
249 nip = startp;
250 #ifdef PTR_DEMANGLE
251 PTR_DEMANGLE (fct.l);
252 PTR_DEMANGLE (nip);
253 #endif
254 no_more = nip == (service_user *) -1l;
257 while (no_more == 0)
259 #ifdef NEED_H_ERRNO
260 any_service = true;
261 #endif
263 status = DL_CALL_FCT (fct.l, (ADD_VARIABLES, resbuf, buffer, buflen,
264 &errno H_ERRNO_VAR EXTRA_VARIABLES));
266 /* The status is NSS_STATUS_TRYAGAIN and errno is ERANGE the
267 provided buffer is too small. In this case we should give
268 the user the possibility to enlarge the buffer and we should
269 not simply go on with the next service (even if the TRYAGAIN
270 action tells us so). */
271 if (status == NSS_STATUS_TRYAGAIN
272 #ifdef NEED_H_ERRNO
273 && *h_errnop == NETDB_INTERNAL
274 #endif
275 && errno == ERANGE)
276 break;
278 no_more = __nss_next2 (&nip, REENTRANT_NAME_STRING,
279 REENTRANT2_NAME_STRING, &fct.ptr, status, 0);
282 #ifdef HANDLE_DIGITS_DOTS
283 done:
284 #endif
285 *result = status == NSS_STATUS_SUCCESS ? resbuf : NULL;
286 #ifdef NEED_H_ERRNO
287 if (status == NSS_STATUS_UNAVAIL)
288 /* Either we failed to lookup the functions or the functions themselves
289 had a system error. Set NETDB_INTERNAL here to let the caller know
290 that the errno may have the real reason for failure. */
291 *h_errnop = NETDB_INTERNAL;
292 else if (status != NSS_STATUS_SUCCESS && !any_service)
293 /* We were not able to use any service. */
294 *h_errnop = NO_RECOVERY;
295 #endif
296 #ifdef POSTPROCESS
297 POSTPROCESS;
298 #endif
300 int res;
301 if (status == NSS_STATUS_SUCCESS || status == NSS_STATUS_NOTFOUND)
302 res = 0;
303 /* Don't pass back ERANGE if this is not for a too-small buffer. */
304 else if (errno == ERANGE && status != NSS_STATUS_TRYAGAIN)
305 res = EINVAL;
306 #ifdef NEED_H_ERRNO
307 /* These functions only set errno if h_errno is NETDB_INTERNAL. */
308 else if (status == NSS_STATUS_TRYAGAIN && *h_errnop != NETDB_INTERNAL)
309 res = EAGAIN;
310 #endif
311 else
312 return errno;
314 __set_errno (res);
315 return res;
319 #ifdef NO_COMPAT_NEEDED
320 strong_alias (INTERNAL (REENTRANT_NAME), REENTRANT_NAME);
321 #elif !defined FUNCTION2_NAME
322 # include <shlib-compat.h>
323 # if SHLIB_COMPAT (libc, GLIBC_2_0, GLIBC_2_1_2)
324 # define OLD(name) OLD1 (name)
325 # define OLD1(name) __old_##name
328 attribute_compat_text_section
329 OLD (REENTRANT_NAME) (ADD_PARAMS, LOOKUP_TYPE *resbuf, char *buffer,
330 size_t buflen, LOOKUP_TYPE **result H_ERRNO_PARM)
332 int ret = INTERNAL (REENTRANT_NAME) (ADD_VARIABLES, resbuf, buffer,
333 buflen, result H_ERRNO_VAR);
335 if (ret != 0 || result == NULL)
336 ret = -1;
338 return ret;
341 # define do_symbol_version(real, name, version) \
342 compat_symbol (libc, real, name, version)
343 do_symbol_version (OLD (REENTRANT_NAME), REENTRANT_NAME, GLIBC_2_0);
344 # endif
346 /* As INTERNAL (REENTRANT_NAME) may be hidden, we need an alias
347 in between so that the REENTRANT_NAME@@GLIBC_2.1.2 is not
348 hidden too. */
349 strong_alias (INTERNAL (REENTRANT_NAME), NEW (REENTRANT_NAME));
351 # define do_default_symbol_version(real, name, version) \
352 versioned_symbol (libc, real, name, version)
353 do_default_symbol_version (NEW (REENTRANT_NAME),
354 REENTRANT_NAME, GLIBC_2_1_2);
355 #endif
357 nss_interface_function (REENTRANT_NAME)