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