about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--include/unistd.h3
-rw-r--r--src/misc/getusershell.c33
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;
+}