Add free_mem function which disposes all statically allocated memory
[kopensolaris-gnu/glibc.git] / nss / getXXbyYY.c
1 /* Copyright (C) 1996, 1997, 1998, 1999 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3
4    The GNU C Library is free software; you can redistribute it and/or
5    modify it under the terms of the GNU Library General Public License as
6    published by the Free Software Foundation; either version 2 of the
7    License, or (at your option) any later version.
8
9    The GNU C Library is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12    Library General Public License for more details.
13
14    You should have received a copy of the GNU Library General Public
15    License along with the GNU C Library; see the file COPYING.LIB.  If not,
16    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17    Boston, MA 02111-1307, USA.  */
18
19 #include <assert.h>
20 #include <errno.h>
21 #include <bits/libc-lock.h>
22 #include <stdlib.h>
23
24 #include "nsswitch.h"
25
26 /*******************************************************************\
27 |* Here we assume several symbols to be defined:                   *|
28 |*                                                                 *|
29 |* LOOKUP_TYPE   - the return type of the function                 *|
30 |*                                                                 *|
31 |* FUNCTION_NAME - name of the non-reentrant function              *|
32 |*                                                                 *|
33 |* DATABASE_NAME - name of the database the function accesses      *|
34 |*                 (e.g., host, services, ...)                     *|
35 |*                                                                 *|
36 |* ADD_PARAMS    - additional parameter, can vary in number        *|
37 |*                                                                 *|
38 |* ADD_VARIABLES - names of additional parameter                   *|
39 |*                                                                 *|
40 |* BUFLEN        - length of buffer allocated for the non          *|
41 |*                 reentrant version                               *|
42 |*                                                                 *|
43 |* Optionally the following vars can be defined:                   *|
44 |*                                                                 *|
45 |* NEED_H_ERRNO  - an extra parameter will be passed to point to   *|
46 |*                 the global `h_errno' variable.                  *|
47 |*                                                                 *|
48 \*******************************************************************/
49
50 /* To make the real sources a bit prettier.  */
51 #define REENTRANT_NAME APPEND_R (FUNCTION_NAME)
52 #define APPEND_R(name) APPEND_R1 (name)
53 #define APPEND_R1(name) name##_r
54 #define INTERNAL(name) INTERNAL1 (name)
55 #define INTERNAL1(name) __##name
56
57 /* Sometimes we need to store error codes in the `h_errno' variable.  */
58 #ifdef NEED_H_ERRNO
59 # define H_ERRNO_PARM , int *h_errnop
60 # define H_ERRNO_VAR , &h_errno_tmp
61 #else
62 # define H_ERRNO_PARM
63 # define H_ERRNO_VAR
64 #endif
65
66
67 /* Prototype for reentrant version we use here.  */
68 extern int INTERNAL (REENTRANT_NAME) (ADD_PARAMS, LOOKUP_TYPE *resbuf,
69                                       char *buffer, size_t buflen,
70                                       LOOKUP_TYPE **result H_ERRNO_PARM);
71
72 /* We need to protect the dynamic buffer handling.  */
73 __libc_lock_define_initialized (static, lock);
74
75 /* This points to the static buffer used.  */
76 static char *buffer;
77
78
79 LOOKUP_TYPE *
80 FUNCTION_NAME (ADD_PARAMS)
81 {
82   static size_t buffer_size;
83   static LOOKUP_TYPE resbuf;
84   LOOKUP_TYPE *result;
85   int save;
86 #ifdef NEED_H_ERRNO
87   int h_errno_tmp = 0;
88 #endif
89
90   /* Get lock.  */
91   __libc_lock_lock (lock);
92
93   if (buffer == NULL)
94     {
95       buffer_size = BUFLEN;
96       buffer = malloc (buffer_size);
97     }
98
99   if (buffer != NULL)
100     {
101 #ifdef HANDLE_DIGITS_DOTS
102       /* We have to test for the use of IPv6 which can only be done by
103          examining `_res'.  */
104       if ((_res.options & RES_INIT) == 0 && res_init () == -1)
105         {
106 # ifdef NEED_H_ERRNO
107           h_errno_tmp = NETDB_INTERNAL;
108 # endif
109           result = NULL;
110           goto done;
111         }
112 # include "digits_dots.c"
113 #endif
114     }
115
116   while (buffer != NULL
117          && INTERNAL (REENTRANT_NAME) (ADD_VARIABLES, &resbuf, buffer,
118                                        buffer_size, &result H_ERRNO_VAR) != 0
119 #ifdef NEED_H_ERRNO
120          && h_errno_tmp == NETDB_INTERNAL
121 #endif
122          && errno == ERANGE)
123     {
124       char *new_buf;
125       buffer_size += BUFLEN;
126       new_buf = realloc (buffer, buffer_size);
127       if (new_buf == NULL)
128         {
129           /* We are out of memory.  Free the current buffer so that the
130              process gets a chance for a normal termination.  */
131           save = errno;
132           free (buffer);
133           __set_errno (save);
134         }
135       buffer = new_buf;
136     }
137
138   if (buffer == NULL)
139     result = NULL;
140
141 #ifdef HANDLE_DIGITS_DOTS
142 done:
143 #endif
144   /* Release lock.  Preserve error value.  */
145   save = errno;
146   __libc_lock_unlock (lock);
147   __set_errno (save);
148
149 #ifdef NEED_H_ERRNO
150   if (h_errno_tmp != 0)
151     __set_h_errno (h_errno_tmp);
152 #endif
153
154   return result;
155 }
156
157
158 /* Free all resources if necessary.  */
159 static void __attribute__ ((unused))
160 free_mem (void)
161 {
162   if (buffer != NULL)
163     free (buffer);
164 }
165
166 text_set_element (__libc_subfreeres, free_mem);