ubase/su.c

185 lines
3.9 KiB
C

/* See LICENSE file for copyright and license details. */
#define _XOPEN_SOURCE
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>
#include <pwd.h>
#include <grp.h>
#include <shadow.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>
#include "config.h"
#include "util.h"
extern char **environ;
static const char *randreply(void);
static int dologin(struct passwd *);
static void
usage(void)
{
eprintf("usage: %s [-lp] [user]\n", argv0);
}
static int lflag = 0;
static int pflag = 0;
int
main(int argc, char *argv[])
{
char *usr = "root", *pass, *cryptpass;
char * const *newargv;
struct spwd *spw;
struct passwd *pw;
uid_t uid;
ARGBEGIN {
case 'l':
lflag = 1;
break;
case 'p':
pflag = 1;
break;
default:
usage();
} ARGEND;
if (argc < 1)
;
else if (argc == 1)
usr = argv[0];
else
usage();
srand(time(NULL));
errno = 0;
pw = getpwnam(usr);
if (errno)
eprintf("getpwnam: %s:", usr);
else if (!pw)
eprintf("who are you?\n");
switch (pw->pw_passwd[0]) {
case '!':
case '*':
eprintf("denied\n");
}
/* Empty password? Su now */
if (pw->pw_passwd[0] == '\0')
goto dosu;
uid = getuid();
if (uid) {
pass = getpass("Password: ");
if (!pass)
eprintf("getpass:");
}
if (pw->pw_passwd[0] == 'x' && pw->pw_passwd[1] == '\0') {
errno = 0;
spw = getspnam(usr);
if (errno)
eprintf("getspnam: %s:", usr);
else if (!spw)
eprintf("who are you?\n");
switch (spw->sp_pwdp[0]) {
case '!':
case '*':
eprintf("denied\n");
}
if (uid) {
cryptpass = crypt(pass, spw->sp_pwdp);
if (!cryptpass)
eprintf("crypt:");
if (strcmp(cryptpass, spw->sp_pwdp) != 0)
eprintf(randreply());
}
} else {
if (uid) {
cryptpass = crypt(pass, pw->pw_passwd);
if (!cryptpass)
eprintf("crypt:");
if (strcmp(cryptpass, pw->pw_passwd) != 0)
eprintf("login failed\n");
}
}
dosu:
if (initgroups(usr, pw->pw_gid) < 0)
eprintf("initgroups:");
if (setgid(pw->pw_gid) < 0)
eprintf("setgid:");
if (setuid(pw->pw_uid) < 0)
eprintf("setuid:");
if (lflag) {
return dologin(pw);
} else {
newargv = (char *const[]){pw->pw_shell, NULL};
if (!pflag) {
setenv("HOME", pw->pw_dir, 1);
setenv("SHELL", pw->pw_shell, 1);
if (strcmp(pw->pw_name, "root") != 0) {
setenv("USER", pw->pw_name, 1);
setenv("LOGNAME", pw->pw_name, 1);
}
}
if (strcmp(pw->pw_name, "root") == 0)
setenv("PATH", ENV_SUPATH, 1);
else
setenv("PATH", ENV_PATH, 1);
execve(pflag ? getenv("SHELL") : pw->pw_shell,
newargv, environ);
weprintf("execve %s:", pw->pw_shell);
return (errno == ENOENT) ? 127 : 126;
}
return EXIT_SUCCESS;
}
static const char *
randreply(void)
{
static const char *replies[] = {
"Time flies like an arrow, fruit flies like a banana.\n",
"Denied.\n",
"You type like a dairy farmer.\n",
"CChheecckk yyoouurr dduupplleexx sswwiittcchh..\n",
"I met a girl with 12 nipples, it sounds weird dozen tit?\n",
"Here I am, brain the size of a planet and they ask me to keep hashing rubbish.\n",
"Clones are people two.\n",
"Your mom is an interesting su response.\n",
"no.\n",
"Your mom forgot to null-terminate???B?33??Abort (core dumped)\n",
"A fool-proof method for sculpting an elephant: first, get a huge block of marble; then you chip away everything that doesn't look like an elephant.\n",
"Bloating .data for fun and profit.\n",
};
return replies[rand() % LEN(replies)];
}
static int
dologin(struct passwd *pw)
{
char *term = getenv("TERM");
clearenv();
setenv("HOME", pw->pw_dir, 1);
setenv("SHELL", pw->pw_shell, 1);
setenv("USER", pw->pw_name, 1);
setenv("LOGNAME", pw->pw_name, 1);
setenv("TERM", term ? term : "vt100", 1);
if (strcmp(pw->pw_name, "root") == 0)
setenv("PATH", ENV_SUPATH, 1);
else
setenv("PATH", ENV_PATH, 1);
if (chdir(pw->pw_dir) < 0)
eprintf("chdir %s:", pw->pw_dir);
execlp(pw->pw_shell, pw->pw_shell, "-l", NULL);
weprintf("execlp %s:", pw->pw_shell);
return (errno == ENOENT) ? 127 : 126;
}