If NSS function reports too small buffer return
[kopensolaris-gnu/glibc.git] / nss / getXXbyYY_r.c
1 /* Copyright (C) 1996, 1997, 1998 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
4
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Library General Public License as
7    published by the Free Software Foundation; either version 2 of the
8    License, or (at your option) any later version.
9
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    Library General Public License for more details.
14
15    You should have received a copy of the GNU Library General Public
16    License along with the GNU C Library; see the file COPYING.LIB.  If not,
17    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18    Boston, MA 02111-1307, USA.  */
19
20 #include <errno.h>
21 #include "nsswitch.h"
22 #include <nscd/nscd_proto.h>
23
24 /*******************************************************************\
25 |* Here we assume several symbols to be defined:                   *|
26 |*                                                                 *|
27 |* LOOKUP_TYPE   - the return type of the function                 *|
28 |*                                                                 *|
29 |* FUNCTION_NAME - name of the non-reentrant function              *|
30 |*                                                                 *|
31 |* DATABASE_NAME - name of the database the function accesses      *|
32 |*                 (e.g., host, services, ...)                     *|
33 |*                                                                 *|
34 |* ADD_PARAMS    - additional parameter, can vary in number        *|
35 |*                                                                 *|
36 |* ADD_VARIABLES - names of additional parameter                   *|
37 |*                                                                 *|
38 |* Optionally the following vars can be defined:                   *|
39 |*                                                                 *|
40 |* NEED_H_ERRNO  - an extra parameter will be passed to point to   *|
41 |*                 the global `h_errno' variable.                  *|
42 |*                                                                 *|
43 |* NEED__RES     - the global _res variable might be used so we    *|
44 |*                 will have to initialize it if necessary         *|
45 |*                                                                 *|
46 \*******************************************************************/
47
48 /* To make the real sources a bit prettier.  */
49 #define REENTRANT_NAME APPEND_R (FUNCTION_NAME)
50 #define APPEND_R(name) APPEND_R1 (name)
51 #define APPEND_R1(name) name##_r
52 #define INTERNAL(name) INTERNAL1 (name)
53 #define INTERNAL1(name) __##name
54
55 #ifdef USE_NSCD
56 # define NSCD_NAME ADD_NSCD (REENTRANT_NAME)
57 # define ADD_NSCD(name) ADD_NSCD1 (name)
58 # define ADD_NSCD1(name) __nscd_##name
59 #endif
60
61 #define FUNCTION_NAME_STRING STRINGIZE (FUNCTION_NAME)
62 #define REENTRANT_NAME_STRING STRINGIZE (REENTRANT_NAME)
63 #define DATABASE_NAME_STRING STRINGIZE (DATABASE_NAME)
64 #define STRINGIZE(name) STRINGIZE1 (name)
65 #define STRINGIZE1(name) #name
66
67 #define DB_LOOKUP_FCT CONCAT3_1 (__nss_, DATABASE_NAME, _lookup)
68 #define CONCAT3_1(Pre, Name, Post) CONCAT3_2 (Pre, Name, Post)
69 #define CONCAT3_2(Pre, Name, Post) Pre##Name##Post
70
71 /* Sometimes we need to store error codes in the `h_errno' variable.  */
72 #ifdef NEED_H_ERRNO
73 # define H_ERRNO_PARM , int *h_errnop
74 # define H_ERRNO_VAR , h_errnop
75 #else
76 # define H_ERRNO_PARM
77 # define H_ERRNO_VAR
78 #endif
79
80
81 /* Type of the lookup function we need here.  */
82 typedef int (*lookup_function) (ADD_PARAMS, LOOKUP_TYPE *, char *, size_t,
83                                 int * H_ERRNO_PARM);
84
85 /* Some usages of this file might use this variable.  */
86 extern struct __res_state _res;
87
88 /* The lookup function for the first entry of this service.  */
89 extern int DB_LOOKUP_FCT (service_user **nip, const char *name, void **fctp);
90
91
92
93 int
94 INTERNAL (REENTRANT_NAME) (ADD_PARAMS, LOOKUP_TYPE *resbuf, char *buffer,
95                            size_t buflen, LOOKUP_TYPE **result H_ERRNO_PARM)
96 {
97   static service_user *startp = NULL;
98   static lookup_function start_fct;
99   service_user *nip;
100   lookup_function fct;
101   int no_more;
102   enum nss_status status = NSS_STATUS_UNAVAIL;
103 #ifdef USE_NSCD
104   int nscd_status;
105 #endif
106
107 #ifdef HANDLE_DIGITS_DOTS
108 # define resbuf (*resbuf)
109 # include "digits_dots.c"
110 # undef resbuf
111 #endif
112
113 #ifdef USE_NSCD
114   nscd_status = NSCD_NAME (ADD_VARIABLES, resbuf, buffer, buflen H_ERRNO_VAR);
115   if (nscd_status < 1)
116     {
117       *result = nscd_status == 0 ? resbuf : NULL;
118       return nscd_status;
119     }
120 #endif
121
122   if (startp == NULL)
123     {
124       no_more = DB_LOOKUP_FCT (&nip, REENTRANT_NAME_STRING, (void **) &fct);
125       if (no_more)
126         startp = (service_user *) -1l;
127       else
128         {
129           startp = nip;
130           start_fct = fct;
131
132 #ifdef NEED__RES
133           /* The resolver code will really be used so we have to
134              initialize it.  */
135           if ((_res.options & RES_INIT) == 0 && res_init () == -1)
136             {
137               *h_errnop = NETDB_INTERNAL;
138               *result = NULL;
139               return -1;
140             }
141 #endif /* need _res */
142         }
143     }
144   else
145     {
146       fct = start_fct;
147       no_more = (nip = startp) == (service_user *) -1l;
148     }
149
150   while (no_more == 0)
151     {
152       status = (*fct) (ADD_VARIABLES, resbuf, buffer, buflen,
153                        __errno_location () H_ERRNO_VAR);
154
155       /* The the status is NSS_STATUS_TRYAGAIN and errno is ERANGE the
156          provided buffer is too small.  In this case we should give
157          the user the possibility to enlarge the buffer and we should
158          not simply go on with the next service (even if the TRYAGAIN
159          action tells us so).  */
160       if (status == NSS_STATUS_TRYAGAIN
161 #ifdef NEED_H_ERRNO
162           && *h_errnop == NETDB_INTERNAL
163 #endif
164           && errno == ERANGE)
165         break;
166
167       no_more = __nss_next (&nip, REENTRANT_NAME_STRING,
168                             (void **) &fct, status, 0);
169     }
170
171 #ifdef HANDLE_DIGITS_DOTS
172 done:
173 #endif
174   *result = status == NSS_STATUS_SUCCESS ? resbuf : NULL;
175   return status == NSS_STATUS_SUCCESS ? 0 : -1;
176 }
177
178 #define do_weak_alias(n1, n2) weak_alias (n1, (n2))
179 do_weak_alias (INTERNAL (REENTRANT_NAME), REENTRANT_NAME)