diff options
-rw-r--r-- | include/unistd.h | 3 | ||||
-rw-r--r-- | src/misc/getusershell.c | 33 |
2 files changed, 36 insertions, 0 deletions
diff --git a/include/unistd.h b/include/unistd.h index 17f024fd..93098728 100644 --- a/include/unistd.h +++ b/include/unistd.h @@ -163,6 +163,9 @@ char *get_current_dir_name(void); int daemon(int, int); int getdomainname(char *, size_t); int getdtablesize(void); +void setusershell(void); +void endusershell(void); +char *getusershell(void); #endif #define _XOPEN_VERSION 700 diff --git a/src/misc/getusershell.c b/src/misc/getusershell.c new file mode 100644 index 00000000..11e697b6 --- /dev/null +++ b/src/misc/getusershell.c @@ -0,0 +1,33 @@ +#define _GNU_SOURCE +#include <stdio.h> +#include <string.h> +#include <stdlib.h> + +static const char defshells[] = "/bin/sh\n/bin/csh\n"; + +static char *line; +size_t linesize; +static FILE *f; + +void endusershell(void) +{ + if (f) fclose(f); + f = 0; +} + +void setusershell(void) +{ + if (!f) f = fopen("/etc/shells", "rb"); + if (!f) f = fmemopen((void *)defshells, sizeof defshells - 1, "rb"); +} + +char *getusershell(void) +{ + ssize_t l; + if (!f) setusershell(); + if (!f) return 0; + l = getline(&line, &linesize, f); + if (l <= 0) return 0; + if (line[l-1]=='\n') line[l-1]=0; + return line; +} |