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