getusershell.c revision 1.21 1 /* $NetBSD: getusershell.c,v 1.21 2000/07/07 08:03:37 itohy Exp $ */
2
3 /*
4 * Copyright (c) 1985, 1993
5 * The Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. All advertising materials mentioning features or use of this software
16 * must display the following acknowledgement:
17 * This product includes software developed by the University of
18 * California, Berkeley and its contributors.
19 * 4. Neither the name of the University nor the names of its contributors
20 * may be used to endorse or promote products derived from this software
21 * without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * SUCH DAMAGE.
34 */
35
36 #include <sys/cdefs.h>
37 #if defined(LIBC_SCCS) && !defined(lint)
38 #if 0
39 static char sccsid[] = "@(#)getusershell.c 8.1 (Berkeley) 6/4/93";
40 #else
41 __RCSID("$NetBSD: getusershell.c,v 1.21 2000/07/07 08:03:37 itohy Exp $");
42 #endif
43 #endif /* LIBC_SCCS and not lint */
44
45 #include "namespace.h"
46 #include <sys/param.h>
47 #include <sys/file.h>
48
49 #include <ctype.h>
50 #include <errno.h>
51 #include <nsswitch.h>
52 #include <paths.h>
53 #include <stdio.h>
54 #include <stdlib.h>
55 #include <string.h>
56 #include <stringlist.h>
57 #include <unistd.h>
58
59 #ifdef __STDC__
60 #include <stdarg.h>
61 #else
62 #include <varargs.h>
63 #endif
64
65 #ifdef HESIOD
66 #include <hesiod.h>
67 #endif
68 #ifdef YP
69 #include <rpc/rpc.h>
70 #include <rpcsvc/ypclnt.h>
71 #include <rpcsvc/yp_prot.h>
72 #endif
73
74 #ifdef __weak_alias
75 __weak_alias(endusershell,_endusershell)
76 __weak_alias(getusershell,_getusershell)
77 __weak_alias(setusershell,_setusershell)
78 #endif
79
80 /*
81 * Local shells should NOT be added here. They should be added in
82 * /etc/shells.
83 */
84
85 static const char *const okshells[] = { _PATH_BSHELL, _PATH_CSHELL, NULL };
86 static const char *const *curshell;
87 static StringList *sl;
88
89 static const char *const *initshells __P((void));
90
91 /*
92 * Get a list of shells from "shells" nsswitch database
93 */
94 __aconst char *
95 getusershell()
96 {
97 __aconst char *ret;
98
99 if (curshell == NULL)
100 curshell = initshells();
101 /*LINTED*/
102 ret = (__aconst char *)*curshell;
103 if (ret != NULL)
104 curshell++;
105 return (ret);
106 }
107
108 void
109 endusershell()
110 {
111 if (sl)
112 sl_free(sl, 1);
113 sl = NULL;
114 curshell = NULL;
115 }
116
117 void
118 setusershell()
119 {
120
121 curshell = initshells();
122 }
123
124
125 static int _local_initshells __P((void *, void *, va_list));
126
127 /*ARGSUSED*/
128 static int
129 _local_initshells(rv, cb_data, ap)
130 void *rv;
131 void *cb_data;
132 va_list ap;
133 {
134 char *sp, *cp;
135 FILE *fp;
136 char line[MAXPATHLEN + 2];
137
138 if (sl)
139 sl_free(sl, 1);
140 sl = sl_init();
141 if (!sl)
142 return (NS_UNAVAIL);
143
144 if ((fp = fopen(_PATH_SHELLS, "r")) == NULL)
145 return (NS_UNAVAIL);
146
147 sp = cp = line;
148 while (fgets(cp, MAXPATHLEN + 1, fp) != NULL) {
149 while (*cp != '#' && *cp != '/' && *cp != '\0')
150 cp++;
151 if (*cp == '#' || *cp == '\0')
152 continue;
153 sp = cp;
154 while (!isspace((unsigned char) *cp) && *cp != '#'
155 && *cp != '\0')
156 cp++;
157 *cp++ = '\0';
158 if (sl_add(sl, strdup(sp)) == -1)
159 return (NS_UNAVAIL);
160 }
161 (void)fclose(fp);
162 return (NS_SUCCESS);
163 }
164
165 #ifdef HESIOD
166 static int _dns_initshells __P((void *, void *, va_list));
167
168 /*ARGSUSED*/
169 static int
170 _dns_initshells(rv, cb_data, ap)
171 void *rv;
172 void *cb_data;
173 va_list ap;
174 {
175 char shellname[] = "shells-XXXXX";
176 int hsindex, hpi, r;
177 char **hp;
178 void *context;
179
180 if (sl)
181 sl_free(sl, 1);
182 sl = sl_init();
183 if (!sl)
184 return (NS_UNAVAIL);
185
186 r = NS_UNAVAIL;
187 if (hesiod_init(&context) == -1)
188 return (r);
189
190 for (hsindex = 0; ; hsindex++) {
191 snprintf(shellname, sizeof(shellname)-1, "shells-%d", hsindex);
192 hp = hesiod_resolve(context, shellname, "shells");
193 if (hp == NULL) {
194 if (errno == ENOENT) {
195 if (hsindex == 0)
196 r = NS_NOTFOUND;
197 else
198 r = NS_SUCCESS;
199 }
200 break;
201 } else {
202 int bad = 0;
203
204 for (hpi = 0; hp[hpi]; hpi++)
205 if (sl_add(sl, hp[hpi]) == -1) {
206 bad = 1;
207 break;
208 }
209 free(hp);
210 if (bad)
211 break;
212 }
213 }
214 hesiod_end(context);
215 return (r);
216 }
217 #endif /* HESIOD */
218
219 #ifdef YP
220 static int _nis_initshells __P((void *, void *, va_list));
221
222 /*ARGSUSED*/
223 static int
224 _nis_initshells(rv, cb_data, ap)
225 void *rv;
226 void *cb_data;
227 va_list ap;
228 {
229 static char *ypdomain;
230
231 if (sl)
232 sl_free(sl, 1);
233 sl = sl_init();
234 if (!sl)
235 return (NS_UNAVAIL);
236
237 if (ypdomain == NULL) {
238 switch (yp_get_default_domain(&ypdomain)) {
239 case 0:
240 break;
241 case YPERR_RESRC:
242 return (NS_TRYAGAIN);
243 default:
244 return (NS_UNAVAIL);
245 }
246 }
247
248 for (;;) {
249 char *ypcur = NULL;
250 int ypcurlen = 0; /* XXX: GCC */
251 char *key, *data;
252 int keylen, datalen;
253 int r;
254
255 key = data = NULL;
256 if (ypcur) {
257 r = yp_next(ypdomain, "shells", ypcur, ypcurlen,
258 &key, &keylen, &data, &datalen);
259 free(ypcur);
260 switch (r) {
261 case 0:
262 break;
263 case YPERR_NOMORE:
264 free(key);
265 free(data);
266 return (NS_SUCCESS);
267 default:
268 free(key);
269 free(data);
270 return (NS_UNAVAIL);
271 }
272 ypcur = key;
273 ypcurlen = keylen;
274 } else {
275 if (yp_first(ypdomain, "shells", &ypcur,
276 &ypcurlen, &data, &datalen)) {
277 free(data);
278 return (NS_UNAVAIL);
279 }
280 }
281 data[datalen] = '\0'; /* clear trailing \n */
282 if (sl_add(sl, data) == -1)
283 return (NS_UNAVAIL);
284 }
285 }
286 #endif /* YP */
287
288 static const char *const *
289 initshells()
290 {
291 static const ns_dtab dtab[] = {
292 NS_FILES_CB(_local_initshells, NULL)
293 NS_DNS_CB(_dns_initshells, NULL)
294 NS_NIS_CB(_nis_initshells, NULL)
295 { 0 }
296 };
297 if (sl)
298 sl_free(sl, 1);
299 sl = sl_init();
300 if (!sl)
301 goto badinitshells;
302
303 if (nsdispatch(NULL, dtab, NSDB_SHELLS, "initshells", __nsdefaultsrc)
304 != NS_SUCCESS) {
305 badinitshells:
306 if (sl)
307 sl_free(sl, 1);
308 sl = NULL;
309 return (okshells);
310 }
311 if (sl_add(sl, NULL) == -1)
312 goto badinitshells;
313
314 return (const char *const *)(sl->sl_str);
315 }
316