Optimize some uses of stpcpy away.
[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 0;
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       if (result == NULL)
208         {
209           if (tablename_val == NULL)
210             if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
211               return NSS_STATUS_UNAVAIL;
212
213           result = nis_first_entry (tablename_val);
214           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
215             return niserr2nss (result->status);
216         }
217       else
218         {
219           nis_result *res;
220
221           res = nis_next_entry (tablename_val, &result->cookie);
222           nis_freeresult (result);
223           result = res;
224           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
225             return niserr2nss (result->status);
226         }
227
228       parse_res = _nss_nisplus_parse_servent (result, serv, buffer, buflen);
229     }
230   while (!parse_res);
231
232   return NSS_STATUS_SUCCESS;
233 }
234
235 enum nss_status
236 _nss_nisplus_getservent_r (struct servent *result, char *buffer,
237                            size_t buflen)
238 {
239   int status;
240
241   __libc_lock_lock (lock);
242
243   status = internal_nisplus_getservent_r (result, buffer, buflen);
244
245   __libc_lock_unlock (lock);
246
247   return status;
248 }
249
250 enum nss_status
251 _nss_nisplus_getservbyname_r (const char *name, const char *protocol,
252                               struct servent *serv,
253                               char *buffer, size_t buflen)
254 {
255   int parse_res;
256
257   if (tablename_val == NULL)
258     if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
259       return NSS_STATUS_UNAVAIL;
260
261   if (name == NULL || protocol == NULL)
262     {
263       __set_errno (EINVAL);
264       return NSS_STATUS_NOTFOUND;
265     }
266   else
267     {
268       nis_result *result;
269       char buf[strlen (name) + 255 + tablename_len];
270
271       /* Search at first in the alias list, and use the correct name
272          for the next search */
273       sprintf (buf, "[name=%s,proto=%s],%s", name, protocol,
274                tablename_val);
275       result = nis_list (buf, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
276
277       /* If we do not find it, try it as original name. But if the
278          database is correct, we should find it in the first case, too */
279       if ((result->status != NIS_SUCCESS &&
280            result->status != NIS_S_SUCCESS) ||
281           __type_of (result->objects.objects_val) != ENTRY_OBJ ||
282           strcmp (result->objects.objects_val->EN_data.en_type,
283                   "services_tbl") != 0 ||
284           result->objects.objects_val->EN_data.en_cols.en_cols_len < 4)
285         sprintf (buf, "[cname=%s,proto=%s],%s", name, protocol,
286                  tablename_val);
287       else
288         sprintf (buf, "[cname=%s,proto=%s],%s",
289                  NISENTRYVAL (0, 0, result), protocol, tablename_val);
290
291       nis_freeresult (result);
292       result = nis_list (buf, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
293
294       if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
295         {
296           enum nss_status status = niserr2nss (result->status);
297
298           nis_freeresult (result);
299           return status;
300         }
301
302       parse_res = _nss_nisplus_parse_servent (result, serv, buffer, buflen);
303
304       nis_freeresult (result);
305
306       if (parse_res)
307         return NSS_STATUS_SUCCESS;
308
309       if (!parse_res && errno == ERANGE)
310         return NSS_STATUS_TRYAGAIN;
311       else
312         return NSS_STATUS_NOTFOUND;
313     }
314 }
315
316 enum nss_status
317 _nss_nisplus_getservbynumber_r (const int number, const char *protocol,
318                                 struct servent *serv,
319                                 char *buffer, size_t buflen)
320 {
321   if (tablename_val == NULL)
322     if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
323       return NSS_STATUS_UNAVAIL;
324
325   if (protocol == NULL)
326     {
327       __set_errno (EINVAL);
328       return NSS_STATUS_NOTFOUND;
329     }
330   else
331     {
332       int parse_res;
333       nis_result *result;
334       char buf[60 + strlen (protocol) + tablename_len];
335
336       snprintf (buf, sizeof (buf), "[number=%d,proto=%s],%s",
337                 number, protocol, tablename_val);
338
339       result = nis_list (buf, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
340
341       if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
342         {
343           enum nss_status status = niserr2nss (result->status);
344
345           nis_freeresult (result);
346           return status;
347         }
348
349       parse_res = _nss_nisplus_parse_servent (result, serv, buffer, buflen);
350
351       nis_freeresult (result);
352
353       if (parse_res)
354         return NSS_STATUS_SUCCESS;
355
356       if (!parse_res && errno == ERANGE)
357         return NSS_STATUS_TRYAGAIN;
358       else
359         return NSS_STATUS_NOTFOUND;
360     }
361 }