usershell.c 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  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. #define __fsetlocking __fsetlocking_internal
  33. #define fileno __fileno
  34. #define fgets_unlocked __fgets_unlocked
  35. #define _GNU_SOURCE
  36. #include <sys/param.h>
  37. #include <sys/file.h>
  38. #include <sys/stat.h>
  39. #include <stdio.h>
  40. #include <stdio_ext.h>
  41. #include <ctype.h>
  42. #include <stdlib.h>
  43. #include <unistd.h>
  44. #include <paths.h>
  45. /*
  46. * Local shells should NOT be added here. They should be added in
  47. * /etc/shells.
  48. */
  49. static const char * const validsh[] = { _PATH_BSHELL, _PATH_CSHELL, NULL };
  50. static char **curshell, **shells, *strings;
  51. static char **initshells __P((void));
  52. /*
  53. * Get a list of shells from _PATH_SHELLS, if it exists.
  54. */
  55. char * getusershell(void)
  56. {
  57. char *ret;
  58. if (curshell == NULL)
  59. curshell = initshells();
  60. ret = *curshell;
  61. if (ret != NULL)
  62. curshell++;
  63. return (ret);
  64. }
  65. static void __free_initshell_memory(void)
  66. {
  67. if (shells != NULL) {
  68. free(shells);
  69. }
  70. shells = NULL;
  71. if (strings != NULL) {
  72. free(strings);
  73. }
  74. strings = NULL;
  75. }
  76. void endusershell(void)
  77. {
  78. __free_initshell_memory();
  79. curshell = NULL;
  80. }
  81. void setusershell(void)
  82. {
  83. curshell = initshells();
  84. }
  85. static char ** initshells(void)
  86. {
  87. register char **sp, *cp;
  88. register FILE *fp;
  89. struct stat statb;
  90. int flen;
  91. __free_initshell_memory();
  92. if ((fp = fopen(_PATH_SHELLS, "r")) == NULL)
  93. return (char **) validsh;
  94. if (__fstat(fileno(fp), &statb) == -1) {
  95. goto cleanup;
  96. }
  97. if ((strings = malloc((unsigned)statb.st_size + 1)) == NULL) {
  98. goto cleanup;
  99. }
  100. if ((shells = calloc((unsigned)statb.st_size / 3, sizeof (char *))) == NULL) {
  101. goto cleanup;
  102. }
  103. /* No threads using this stream. */
  104. #ifdef __UCLIBC_HAS_THREADS__
  105. __fsetlocking (fp, FSETLOCKING_BYCALLER);
  106. #endif
  107. sp = shells;
  108. cp = strings;
  109. flen = statb.st_size;
  110. while (fgets_unlocked(cp, flen - (cp - strings), fp) != NULL) {
  111. while (*cp != '#' && *cp != '/' && *cp != '\0')
  112. cp++;
  113. if (*cp == '#' || *cp == '\0')
  114. continue;
  115. *sp++ = cp;
  116. while (!isspace(*cp) && *cp != '#' && *cp != '\0')
  117. cp++;
  118. *cp++ = '\0';
  119. }
  120. *sp = NULL;
  121. fclose(fp);
  122. return (shells);
  123. cleanup:
  124. __free_initshell_memory();
  125. fclose(fp);
  126. return (char **) validsh;
  127. }