Fix "buffer to small" problems and memory leaks.
[kopensolaris-gnu/glibc.git] / nis / nss_nisplus / nisplus-rpc.c
1 /* Copyright (C) 1997 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Thorsten Kukuk <kukuk@vt.uni-paderborn.de>, 1997.
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 <nss.h>
21 #include <errno.h>
22 #include <ctype.h>
23 #include <string.h>
24 #include <bits/libc-lock.h>
25 #include <rpc/netdb.h>
26 #include <rpcsvc/nis.h>
27 #include <rpcsvc/nislib.h>
28
29 #include "nss-nisplus.h"
30
31 __libc_lock_define_initialized (static, lock)
32
33 static nis_result *result = NULL;
34 static nis_name tablename_val = NULL;
35 static u_long tablename_len = 0;
36
37 #define NISENTRYVAL(idx,col,res) \
38         ((res)->objects.objects_val[(idx)].EN_data.en_cols.en_cols_val[(col)].ec_value.ec_value_val)
39
40 #define NISENTRYLEN(idx,col,res) \
41         ((res)->objects.objects_val[(idx)].EN_data.en_cols.en_cols_val[(col)].ec_value.ec_value_len)
42
43 static int
44 _nss_nisplus_parse_rpcent (nis_result *result, struct rpcent *rpc,
45                            char *buffer, size_t buflen)
46 {
47   char *first_unused = buffer;
48   size_t room_left = buflen;
49   unsigned int i;
50   char *p, *line;
51
52
53   if (result == NULL)
54     return 0;
55
56   if ((result->status != NIS_SUCCESS && result->status != NIS_S_SUCCESS) ||
57       __type_of (result->objects.objects_val) != ENTRY_OBJ ||
58       strcmp(result->objects.objects_val[0].EN_data.en_type,
59              "rpc_tbl") != 0 ||
60       result->objects.objects_val[0].EN_data.en_cols.en_cols_len < 3)
61     return 0;
62
63   if (NISENTRYLEN (0, 0, result) >= room_left)
64     {
65     no_more_room:
66       __set_errno (ERANGE);
67       return -1;
68     }
69   strncpy (first_unused, NISENTRYVAL (0, 0, result),
70            NISENTRYLEN (0, 0, result));
71   first_unused[NISENTRYLEN (0, 0, result)] = '\0';
72   rpc->r_name = first_unused;
73   room_left -= (strlen (first_unused) +1);
74   first_unused += strlen (first_unused) +1;
75   rpc->r_number = atoi (NISENTRYVAL (0, 2, result));
76   p = first_unused;
77
78   line = p;
79   for (i = 0; i < result->objects.objects_len; i++)
80     {
81       if (strcmp (NISENTRYVAL (i, 1, result), rpc->r_name) != 0)
82         {
83           if (NISENTRYLEN (i, 1, result) + 2 > room_left)
84             goto no_more_room;
85           *p++ = ' ';
86           p = stpncpy (p, NISENTRYVAL (i, 1, result),
87                        NISENTRYLEN (i, 1, result));
88           *p = '\0';
89           room_left -= (NISENTRYLEN (i, 1, result) + 1);
90         }
91     }
92   ++p;
93   first_unused = p;
94
95   /* Adjust the pointer so it is aligned for
96      storing pointers.  */
97   first_unused += __alignof__ (char *) - 1;
98   first_unused -= ((first_unused - (char *) 0) % __alignof__ (char *));
99   rpc->r_aliases = (char **) first_unused;
100   if (room_left < sizeof (char *))
101     goto no_more_room;
102   room_left -= (sizeof (char *));
103   rpc->r_aliases[0] = NULL;
104
105   i = 0;
106   while (*line != '\0')
107     {
108       /* Skip leading blanks.  */
109       while (isspace (*line))
110         line++;
111
112       if (*line == '\0')
113         break;
114
115       if (room_left < sizeof (char *))
116         goto no_more_room;
117
118       room_left -= sizeof (char *);
119       rpc->r_aliases[i] = line;
120
121       while (*line != '\0' && *line != ' ')
122         ++line;
123
124       if (line != rpc->r_aliases[i])
125         {
126           if (*line != '\0')
127             {
128               *line = '\0';
129               ++line;
130             }
131           ++i;
132         }
133       else
134         rpc->r_aliases[i] = NULL;
135     }
136
137   return 1;
138 }
139
140 static enum nss_status
141 _nss_create_tablename (void)
142 {
143   if (tablename_val == NULL)
144     {
145       char buf [40 + strlen (nis_local_directory ())];
146       char *p;
147
148       p = stpcpy (buf, "rpc.org_dir.");
149       p = stpcpy (p, nis_local_directory ());
150       tablename_val = strdup (buf);
151       if (tablename_val == NULL)
152         return NSS_STATUS_TRYAGAIN;
153       tablename_len = strlen (tablename_val);
154     }
155   return NSS_STATUS_SUCCESS;
156 }
157
158
159 enum nss_status
160 _nss_nisplus_setrpcent (void)
161 {
162   enum nss_status status = NSS_STATUS_SUCCESS;
163
164   __libc_lock_lock (lock);
165
166   if (result)
167     nis_freeresult (result);
168   result = NULL;
169
170   if (tablename_val == NULL)
171     status = _nss_create_tablename ();
172
173   __libc_lock_unlock (lock);
174
175   return status;
176 }
177
178 enum nss_status
179 _nss_nisplus_endrpcent (void)
180 {
181   __libc_lock_lock (lock);
182
183   if (result)
184     nis_freeresult (result);
185   result = NULL;
186
187   __libc_lock_unlock (lock);
188
189   return NSS_STATUS_SUCCESS;
190 }
191
192 static enum nss_status
193 internal_nisplus_getrpcent_r (struct rpcent *rpc, char *buffer,
194                               size_t buflen)
195 {
196   int parse_res;
197
198   /* Get the next entry until we found a correct one. */
199   do
200     {
201       nis_result *saved_res;
202
203       if (result == NULL)
204         {
205           saved_res = NULL;
206           if (tablename_val == NULL)
207             if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
208               return NSS_STATUS_UNAVAIL;
209
210           result = nis_first_entry(tablename_val);
211           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
212             return niserr2nss (result->status);
213         }
214       else
215         {
216           nis_result *res;
217
218           saved_res = result;
219           res = nis_next_entry (tablename_val, &result->cookie);
220           result = res;
221           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
222             {
223               nis_freeresult (saved_res);
224               return niserr2nss (result->status);
225             }
226         }
227
228       if ((parse_res = _nss_nisplus_parse_rpcent (result, rpc, buffer, 
229                                                   buflen)) == -1)
230         {
231           nis_freeresult (result);
232           result = saved_res;
233           return NSS_STATUS_TRYAGAIN;
234         }
235       else
236         {
237           if (saved_res)
238             nis_freeresult (saved_res);
239         }
240     } while (!parse_res);
241
242   return NSS_STATUS_SUCCESS;
243 }
244
245 enum nss_status
246 _nss_nisplus_getrpcent_r (struct rpcent *result, char *buffer,
247                           size_t buflen)
248 {
249   int status;
250
251   __libc_lock_lock (lock);
252
253   status = internal_nisplus_getrpcent_r (result, buffer, buflen);
254
255   __libc_lock_unlock (lock);
256
257   return status;
258 }
259
260 enum nss_status
261 _nss_nisplus_getrpcbyname_r (const char *name, struct rpcent *rpc,
262                              char *buffer, size_t buflen)
263 {
264   int parse_res;
265
266   if (tablename_val == NULL)
267     if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
268       return NSS_STATUS_UNAVAIL;
269
270   if (name == NULL)
271     return NSS_STATUS_NOTFOUND;
272   else
273     {
274       nis_result *result;
275       char buf[strlen (name) + 255 + tablename_len];
276
277       /* Search at first in the alias list, and use the correct name
278          for the next search */
279       sprintf (buf, "[name=%s],%s", name, tablename_val);
280       result = nis_list (buf, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
281
282       /* If we do not find it, try it as original name. But if the
283          database is correct, we should find it in the first case, too */
284       if ((result->status != NIS_SUCCESS &&
285            result->status != NIS_S_SUCCESS) ||
286           __type_of (result->objects.objects_val) != ENTRY_OBJ ||
287           strcmp (result->objects.objects_val->EN_data.en_type,
288                   "rpc_tbl") != 0 ||
289           result->objects.objects_val->EN_data.en_cols.en_cols_len < 3)
290         sprintf (buf, "[cname=%s],%s", name, tablename_val);
291       else
292         sprintf (buf, "[cname=%s],%s", NISENTRYVAL(0, 0, result),
293                  tablename_val);
294
295       nis_freeresult (result);
296       result = nis_list(buf, FOLLOW_PATH | FOLLOW_LINKS , NULL, NULL);
297
298       if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
299         {
300           enum nss_status status = niserr2nss (result->status);
301
302           nis_freeresult (result);
303           return status;
304         }
305
306       parse_res = _nss_nisplus_parse_rpcent (result, rpc, buffer, buflen);
307
308       nis_freeresult (result);
309
310       if (parse_res == -1)
311         return NSS_STATUS_TRYAGAIN;
312
313       if (parse_res)
314         return NSS_STATUS_SUCCESS;
315
316       return NSS_STATUS_NOTFOUND;
317     }
318 }
319
320 enum nss_status
321 _nss_nisplus_getrpcbynumber_r (const int number, struct rpcent *rpc,
322                                 char *buffer, size_t buflen)
323 {
324   if (tablename_val == NULL)
325     if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
326       return NSS_STATUS_UNAVAIL;
327
328   {
329     int parse_res;
330     nis_result *result;
331     char buf[100 + tablename_len];
332
333     snprintf (buf, sizeof (buf), "[number=%d],%s", number, tablename_val);
334
335     result = nis_list(buf, FOLLOW_LINKS | FOLLOW_PATH, NULL, NULL);
336
337     if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
338       {
339         enum nss_status status = niserr2nss (result->status);
340
341         nis_freeresult (result);
342         return status;
343       }
344
345     parse_res = _nss_nisplus_parse_rpcent (result, rpc, buffer, buflen);
346
347     nis_freeresult (result);
348
349     if (parse_res == -1)
350       return NSS_STATUS_TRYAGAIN;
351
352     if (parse_res)
353       return NSS_STATUS_SUCCESS;
354
355     return NSS_STATUS_NOTFOUND;
356   }
357 }