libc/unistd/usershell.c

   1 /*
   2  * Copyright (c) 1985, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 4. Neither the name of the University nor the names of its contributors
  14  *    may be used to endorse or promote products derived from this software
  15  *    without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  *
  29  * This version has been hevily modified for use with uClibc
  30  * November 2002, Erik Andersen <andersen@codepoet.org>
  31  */
  32
  33 #define _GNU_SOURCE
  34 #include <sys/param.h>
  35 #include <sys/file.h>
  36 #include <sys/stat.h>
  37 #include <stdio.h>
  38 #include <stdio_ext.h>
  39 #include <ctype.h>
  40 #include <stdlib.h>
  41 #include <unistd.h>
  42 #include <paths.h>
  43
  44 libc_hidden_proto(fstat)
  45 libc_hidden_proto(fopen)
  46 libc_hidden_proto(fclose)
  47 libc_hidden_proto(__fsetlocking)
  48 libc_hidden_proto(fileno)
  49 libc_hidden_proto(fgets_unlocked)
  50
  51 /*
  52  * Local shells should NOT be added here.  They should be added in
  53  * /etc/shells.
  54  */
  55
  56 static const char * const validsh[] = { _PATH_BSHELL, _PATH_CSHELL, NULL };
  57 static char **curshell, **shells, *strings;
  58 static char **initshells __P((void));
  59
  60 /*
  61  * Get a list of shells from _PATH_SHELLS, if it exists.
  62  */
  63 char * getusershell(void)
  64 {
  65     char *ret;
  66
  67     if (curshell == NULL)
  68         curshell = initshells();
  69     ret = *curshell;
  70     if (ret != NULL)
  71         curshell++;
  72     return (ret);
  73 }
  74
  75 static void __free_initshell_memory(void)
  76 {
  77     if (shells != NULL) {
  78         free(shells);
  79     }
  80     shells = NULL;
  81     if (strings != NULL) {
  82         free(strings);
  83     }
  84     strings = NULL;
  85 }
  86
  87 void endusershell(void)
  88 {
  89     __free_initshell_memory();
  90     curshell = NULL;
  91 }
  92
  93 void setusershell(void)
  94 {
  95
  96     curshell = initshells();
  97 }
  98
  99 static char ** initshells(void)
 100 {
 101     register char **sp, *cp;
 102     register FILE *fp;
 103     struct stat statb;
 104     int flen;
 105
 106     __free_initshell_memory();
 107
 108     if ((fp = fopen(_PATH_SHELLS, "r")) == NULL)
 109         return (char **) validsh;
 110     if (fstat(fileno(fp), &statb) == -1) {
 111         goto cleanup;
 112     }
 113     if ((strings = malloc((unsigned)statb.st_size + 1)) == NULL) {
 114         goto cleanup;
 115     }
 116     if ((shells = calloc((unsigned)statb.st_size / 3, sizeof (char *))) == NULL) {
 117         goto cleanup;
 118     }
 119     /* No threads using this stream.  */
 120 #ifdef __UCLIBC_HAS_THREADS__
 121     __fsetlocking (fp, FSETLOCKING_BYCALLER);
 122 #endif
 123     sp = shells;
 124     cp = strings;
 125     flen = statb.st_size;
 126     while (fgets_unlocked(cp, flen - (cp - strings), fp) != NULL) {
 127         while (*cp != '#' && *cp != '/' && *cp != '\0')
 128             cp++;
 129         if (*cp == '#' || *cp == '\0')
 130             continue;
 131         *sp++ = cp;
 132         while (!isspace(*cp) && *cp != '#' && *cp != '\0')
 133             cp++;
 134         *cp++ = '\0';
 135     }
 136     *sp = NULL;
 137     fclose(fp);
 138     return (shells);
 139
 140 cleanup:
 141     __free_initshell_memory();
 142     fclose(fp);
 143     return (char **) validsh;
 144 }