usershell.c 3.7 KB

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