Fix "buffer to small" problems and memory leaks.
[kopensolaris-gnu/glibc.git] / nis / nss_nisplus / nisplus-service.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_servent (nis_result *result, struct servent *serv,
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 (result->objects.objects_val) != ENTRY_OBJ ||
57       strcmp (result->objects.objects_val->EN_data.en_type,
58               "services_tbl") != 0 ||
59       result->objects.objects_val->EN_data.en_cols.en_cols_len < 4)
60     return 0;
61
62   if (NISENTRYLEN (0, 0, result) >= 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   serv->s_name = first_unused;
72   room_left -= (strlen (first_unused) +1);
73   first_unused += strlen (first_unused) +1;
74
75   if (NISENTRYLEN (0, 2, result) >= room_left)
76     goto no_more_room;
77   strncpy (first_unused, NISENTRYVAL (0, 2, result),
78            NISENTRYLEN (0, 2, result));
79   first_unused[NISENTRYLEN (0, 2, result)] = '\0';
80   serv->s_proto = first_unused;
81   room_left -= (strlen (first_unused) +1);
82   first_unused += strlen (first_unused) +1;
83
84   serv->s_port = atoi (NISENTRYVAL (0, 3, result));
85   p = first_unused;
86
87   line = p;
88   for (i = 0; i < result->objects.objects_len; i++)
89     {
90       if (strcmp (NISENTRYVAL (i, 1, result), serv->s_name) != 0)
91         {
92           if (NISENTRYLEN (i, 1, result) + 2 > room_left)
93             goto no_more_room;
94           *p++ = ' ';
95           p = stpncpy (p, NISENTRYVAL (i, 1, result),
96                        NISENTRYLEN (i, 1, result));
97           *p = '\0';
98           room_left -= (NISENTRYLEN (i, 1, result) + 1);
99         }
100     }
101   ++p;
102   first_unused = p;
103
104   /* Adjust the pointer so it is aligned for
105      storing pointers.  */
106   first_unused += __alignof__ (char *) - 1;
107   first_unused -= ((first_unused - (char *) 0) % __alignof__ (char *));
108   serv->s_aliases = (char **) first_unused;
109   if (room_left < sizeof (char *))
110     goto no_more_room;
111   room_left -= (sizeof (char *));
112   serv->s_aliases[0] = NULL;
113
114   i = 0;
115   while (*line != '\0')
116     {
117       /* Skip leading blanks.  */
118       while (isspace (*line))
119         line++;
120
121       if (*line == '\0')
122         break;
123
124       if (room_left < sizeof (char *))
125         goto no_more_room;
126
127       room_left -= sizeof (char *);
128       serv->s_aliases[i] = line;
129
130       while (*line != '\0' && *line != ' ')
131         ++line;
132
133       if (*line == ' ')
134         {
135           *line = '\0';
136           ++line;
137           ++i;
138         }
139       else
140         serv->s_aliases[i+1] = NULL;
141     }
142
143   return 1;
144 }
145
146 static enum nss_status
147 _nss_create_tablename (void)
148 {
149   if (tablename_val == NULL)
150     {
151       char buf [40 + strlen (nis_local_directory ())];
152       char *p;
153
154       p = stpcpy (buf, "services.org_dir.");
155       p = stpcpy (p, nis_local_directory ());
156       tablename_val = strdup (buf);
157       if (tablename_val == NULL)
158         return NSS_STATUS_TRYAGAIN;
159       tablename_len = strlen (tablename_val);
160     }
161   return NSS_STATUS_SUCCESS;
162 }
163
164
165 enum nss_status
166 _nss_nisplus_setservent (void)
167 {
168   enum nss_status status = NSS_STATUS_SUCCESS;
169
170   __libc_lock_lock (lock);
171
172   if (result)
173     nis_freeresult (result);
174   result = NULL;
175
176   if (tablename_val == NULL)
177     status = _nss_create_tablename ();
178
179   __libc_lock_unlock (lock);
180
181   return status;
182 }
183
184 enum nss_status
185 _nss_nisplus_endservent (void)
186 {
187   __libc_lock_lock (lock);
188
189   if (result)
190     nis_freeresult (result);
191   result = NULL;
192
193   __libc_lock_unlock (lock);
194
195   return NSS_STATUS_SUCCESS;
196 }
197
198 static enum nss_status
199 internal_nisplus_getservent_r (struct servent *serv, char *buffer,
200                                size_t buflen)
201 {
202   int parse_res;
203
204   /* Get the next entry until we found a correct one. */
205   do
206     {
207       nis_result *saved_res;
208       
209       if (result == NULL)
210         {
211           saved_res = NULL;
212           if (tablename_val == NULL)
213             if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
214               return NSS_STATUS_UNAVAIL;
215
216           result = nis_first_entry (tablename_val);
217           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
218             return niserr2nss (result->status);
219         }
220       else
221         {
222           nis_result *res;
223
224           saved_res = result;
225           res = nis_next_entry (tablename_val, &result->cookie);
226           result = res;
227           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
228             {
229               nis_freeresult (saved_res);
230               return niserr2nss (result->status);
231             }
232         }
233
234       if ((parse_res = _nss_nisplus_parse_servent (result, serv, buffer,
235                                                    buflen)) == -1)
236         {
237           nis_freeresult (result);
238           result = saved_res;
239           return NSS_STATUS_TRYAGAIN;
240         }
241       else
242         {
243           if (saved_res)
244             nis_freeresult (saved_res);
245         }
246     }
247   while (!parse_res);
248
249   return NSS_STATUS_SUCCESS;
250 }
251
252 enum nss_status
253 _nss_nisplus_getservent_r (struct servent *result, char *buffer,
254                            size_t buflen)
255 {
256   int status;
257
258   __libc_lock_lock (lock);
259
260   status = internal_nisplus_getservent_r (result, buffer, buflen);
261
262   __libc_lock_unlock (lock);
263
264   return status;
265 }
266
267 enum nss_status
268 _nss_nisplus_getservbyname_r (const char *name, const char *protocol,
269                               struct servent *serv,
270                               char *buffer, size_t buflen)
271 {
272   int parse_res;
273
274   if (tablename_val == NULL)
275     if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
276       return NSS_STATUS_UNAVAIL;
277
278   if (name == NULL || protocol == NULL)
279     {
280       __set_errno (EINVAL);
281       return NSS_STATUS_NOTFOUND;
282     }
283   else
284     {
285       nis_result *result;
286       char buf[strlen (name) + 255 + tablename_len];
287
288       /* Search at first in the alias list, and use the correct name
289          for the next search */
290       sprintf (buf, "[name=%s,proto=%s],%s", name, protocol,
291                tablename_val);
292       result = nis_list (buf, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
293
294       /* If we do not find it, try it as original name. But if the
295          database is correct, we should find it in the first case, too */
296       if ((result->status != NIS_SUCCESS &&
297            result->status != NIS_S_SUCCESS) ||
298           __type_of (result->objects.objects_val) != ENTRY_OBJ ||
299           strcmp (result->objects.objects_val->EN_data.en_type,
300                   "services_tbl") != 0 ||
301           result->objects.objects_val->EN_data.en_cols.en_cols_len < 4)
302         sprintf (buf, "[cname=%s,proto=%s],%s", name, protocol,
303                  tablename_val);
304       else
305         sprintf (buf, "[cname=%s,proto=%s],%s",
306                  NISENTRYVAL (0, 0, result), protocol, tablename_val);
307
308       nis_freeresult (result);
309       result = nis_list (buf, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
310
311       if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
312         {
313           enum nss_status status = niserr2nss (result->status);
314
315           nis_freeresult (result);
316           return status;
317         }
318
319       parse_res = _nss_nisplus_parse_servent (result, serv, buffer, buflen);
320       nis_freeresult (result);
321
322       if (parse_res == -1)
323         return NSS_STATUS_TRYAGAIN;
324
325       if (parse_res)
326         return NSS_STATUS_SUCCESS;
327
328       return NSS_STATUS_NOTFOUND;
329     }
330 }
331
332 enum nss_status
333 _nss_nisplus_getservbynumber_r (const int number, const char *protocol,
334                                 struct servent *serv,
335                                 char *buffer, size_t buflen)
336 {
337   if (tablename_val == NULL)
338     if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
339       return NSS_STATUS_UNAVAIL;
340
341   if (protocol == NULL)
342     {
343       __set_errno (EINVAL);
344       return NSS_STATUS_NOTFOUND;
345     }
346   else
347     {
348       int parse_res;
349       nis_result *result;
350       char buf[60 + strlen (protocol) + tablename_len];
351
352       snprintf (buf, sizeof (buf), "[number=%d,proto=%s],%s",
353                 number, protocol, tablename_val);
354
355       result = nis_list (buf, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
356
357       if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
358         {
359           enum nss_status status = niserr2nss (result->status);
360
361           nis_freeresult (result);
362           return status;
363         }
364
365       parse_res = _nss_nisplus_parse_servent (result, serv, buffer, buflen);
366       nis_freeresult (result);
367
368       if (parse_res == -1)
369         return NSS_STATUS_TRYAGAIN;
370
371       if (parse_res)
372         return NSS_STATUS_SUCCESS;
373
374       return NSS_STATUS_NOTFOUND;
375     }
376 }