Fix "buffer to small" problems and memory leaks.
[kopensolaris-gnu/glibc.git] / nis / nss_nisplus / nisplus-proto.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 <netdb.h>
24 #include <string.h>
25 #include <bits/libc-lock.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_protoent (nis_result * result, struct protoent *proto,
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   if (result == NULL)
53     return 0;
54
55   if ((result->status != NIS_SUCCESS && result->status != NIS_S_SUCCESS) ||
56       __type_of (NIS_RES_OBJECT (result)) != ENTRY_OBJ ||
57       strcmp (NIS_RES_OBJECT (result)->EN_data.en_type, "protocols_tbl") != 0
58       || NIS_RES_OBJECT (result)->EN_data.en_cols.en_cols_len < 3)
59     return 0;
60
61   /* Generate the protocols entry format and use the normal parser */
62   if (NISENTRYLEN (0, 0, result) + 1 > room_left)
63     {
64     no_more_room:
65       __set_errno (ERANGE);
66       return -1;
67     }
68   strncpy (first_unused, NISENTRYVAL (0, 0, result),
69            NISENTRYLEN (0, 0, result));
70   first_unused[NISENTRYLEN (0, 0, result)] = '\0';
71   proto->p_name = first_unused;
72   room_left -= (strlen (first_unused) +1);
73   first_unused += strlen (first_unused) +1;
74
75
76   if (NISENTRYLEN (0, 2, result) + 1 > room_left)
77     goto no_more_room;
78   proto->p_proto = atoi (NISENTRYVAL (0, 2, result));
79   p = first_unused;
80
81   line = p;
82   for (i = 0; i < result->objects.objects_len; i++)
83     {
84       if (strcmp (NISENTRYVAL (i, 1, result), proto->p_name) != 0)
85         {
86           if (NISENTRYLEN (i, 1, result) + 2 > room_left)
87             goto no_more_room;
88           *p++ = ' ';
89           p = stpncpy (p, NISENTRYVAL (i, 1, result),
90                        NISENTRYLEN (i, 1, result));
91           *p = '\0';
92           room_left -= (NISENTRYLEN (i, 1, result) + 1);
93         }
94     }
95   ++p;
96   first_unused = p;
97
98   /* Adjust the pointer so it is aligned for
99      storing pointers.  */
100   first_unused += __alignof__ (char *) - 1;
101   first_unused -= ((first_unused - (char *) 0) % __alignof__ (char *));
102   proto->p_aliases = (char **) first_unused;
103   if (room_left < sizeof (char *))
104     goto no_more_room;
105   room_left -= (sizeof (char *));
106   proto->p_aliases[0] = NULL;
107
108   i = 0;
109   while (*line != '\0')
110     {
111       /* Skip leading blanks.  */
112       while (isspace (*line))
113         line++;
114       if (*line == '\0')
115         break;
116
117       if (room_left < sizeof (char *))
118         goto no_more_room;
119
120       room_left -= sizeof (char *);
121       proto->p_aliases[i] = line;
122
123       while (*line != '\0' && *line != ' ')
124         ++line;
125
126       if (*line == ' ')
127         {
128           *line = '\0';
129           ++line;
130           ++i;
131         }
132       else
133         proto->p_aliases[i+1] = NULL;
134     }
135
136   return 1;
137 }
138
139 static enum nss_status
140 _nss_create_tablename (void)
141 {
142   if (tablename_val == NULL)
143     {
144       char buf [40 + strlen (nis_local_directory ())];
145       char *p;
146
147       p = stpcpy (buf, "protocols.org_dir.");
148       p = stpcpy (p, nis_local_directory ());
149       tablename_val = strdup (buf);
150       if (tablename_val == NULL)
151         return NSS_STATUS_TRYAGAIN;
152       tablename_len = strlen (tablename_val);
153     }
154   return NSS_STATUS_SUCCESS;
155 }
156
157 enum nss_status
158 _nss_nisplus_setprotoent (void)
159 {
160   enum nss_status status = NSS_STATUS_SUCCESS;
161
162   __libc_lock_lock (lock);
163
164   if (result)
165     nis_freeresult (result);
166   result = NULL;
167
168   if (tablename_val == NULL)
169     status = _nss_create_tablename ();
170
171   __libc_lock_unlock (lock);
172
173   return status;
174 }
175
176 enum nss_status
177 _nss_nisplus_endprotoent (void)
178 {
179   __libc_lock_lock (lock);
180
181   if (result)
182     nis_freeresult (result);
183   result = NULL;
184
185   __libc_lock_unlock (lock);
186
187   return NSS_STATUS_SUCCESS;
188 }
189
190 static enum nss_status
191 internal_nisplus_getprotoent_r (struct protoent *proto, char *buffer,
192                                 size_t buflen)
193 {
194   int parse_res;
195
196   /* Get the next entry until we found a correct one. */
197   do
198     {
199       nis_result *saved_res;
200
201       if (result == NULL)
202         {
203           saved_res = NULL;
204           if (tablename_val == NULL)
205             if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
206               return NSS_STATUS_UNAVAIL;
207
208           result = nis_first_entry (tablename_val);
209           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
210             return niserr2nss (result->status);
211         }
212       else
213         {
214           nis_result *res;
215           
216           saved_res = result;
217           res = nis_next_entry (tablename_val, &result->cookie);
218           result = res;
219
220           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
221             {
222               nis_freeresult (saved_res);
223               return niserr2nss (result->status);
224             }
225         }
226
227       if ((parse_res = _nss_nisplus_parse_protoent (result, proto, buffer, 
228                                                     buflen)) == -1)
229         {
230           nis_freeresult (result);
231           result = saved_res;
232           return NSS_STATUS_TRYAGAIN;
233         }
234       else
235         {
236           if (saved_res)
237             nis_freeresult (saved_res);
238         }
239     }
240   while (!parse_res);
241
242   return NSS_STATUS_SUCCESS;
243 }
244
245 enum nss_status
246 _nss_nisplus_getprotoent_r (struct protoent *result, char *buffer,
247                             size_t buflen)
248 {
249   int status;
250
251   __libc_lock_lock (lock);
252
253   status = internal_nisplus_getprotoent_r (result, buffer, buflen);
254
255   __libc_lock_unlock (lock);
256
257   return status;
258 }
259
260 enum nss_status
261 _nss_nisplus_getprotobyname_r (const char *name, struct protoent *proto,
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                   "protocols_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_protoent (result, proto, buffer, buflen);
307
308       nis_freeresult (result);
309
310       if (parse_res == -1)
311         return NSS_STATUS_TRYAGAIN;
312       if (parse_res)
313         return NSS_STATUS_SUCCESS;
314
315       return NSS_STATUS_NOTFOUND;
316     }
317 }
318
319 enum nss_status
320 _nss_nisplus_getprotobynumber_r (const int number, struct protoent *proto,
321                                  char *buffer, size_t buflen)
322 {
323   if (tablename_val == NULL)
324     if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
325       return NSS_STATUS_UNAVAIL;
326   {
327     int parse_res;
328     nis_result *result;
329     char buf[46 + tablename_len];
330
331     snprintf (buf, sizeof (buf), "[number=%d],%s", number, tablename_val);
332
333     result = nis_list (buf, FOLLOW_LINKS | FOLLOW_PATH, NULL, NULL);
334
335     if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
336       {
337         enum nss_status status = niserr2nss (result->status);
338
339         nis_freeresult (result);
340         return status;
341       }
342
343     parse_res = _nss_nisplus_parse_protoent (result, proto, buffer, buflen);
344
345     nis_freeresult (result);
346
347     if (parse_res == -1)
348       return NSS_STATUS_TRYAGAIN;
349     
350     if (parse_res)
351       return NSS_STATUS_SUCCESS;
352     
353     return NSS_STATUS_NOTFOUND;
354   }
355 }