From 99709ab03387ca623e3fc1cac69d242ed44da45c Mon Sep 17 00:00:00 2001 From: Bernhard Reutner-Fischer Date: Wed, 24 Feb 2010 16:10:09 +0100 Subject: crontab: use setup_environment function old new delta setup_environment 184 198 +14 .rodata 131770 131747 -23 ------------------------------------------------------------------------------ (add/remove: 0/0 grow/shrink: 1/1 up/down: 14/-23) Total: -9 bytes Signed-off-by: Bernhard Reutner-Fischer --- include/libbb.h | 5 ++++- libbb/setup_environment.c | 7 +++---- loginutils/login.c | 2 +- miscutils/crontab.c | 21 ++++----------------- 4 files changed, 12 insertions(+), 23 deletions(-) diff --git a/include/libbb.h b/include/libbb.h index 9e3c18407..9d99b0d1a 100644 --- a/include/libbb.h +++ b/include/libbb.h @@ -1142,6 +1142,7 @@ extern void selinux_or_die(void) FAST_FUNC; extern int restricted_shell(const char *shell) FAST_FUNC; /* setup_environment: + * if chdir pw->pw_dir: ok: else if to_tmp == 1: goto /tmp else: goto / or die * if clear_env = 1: cd(pw->pw_dir), clear environment, then set * TERM=(old value) * USER=pw->pw_name, LOGNAME=pw->pw_name @@ -1155,7 +1156,9 @@ extern int restricted_shell(const char *shell) FAST_FUNC; * SHELL=shell * else does nothing */ -extern void setup_environment(const char *shell, int clear_env, int change_env, const struct passwd *pw) FAST_FUNC; +#define SETUP_ENV_CHANGEENV (1<<0) +#define SETUP_ENV_TO_TMP (1<<1) +extern void setup_environment(const char *shell, int clear_env, int flags, const struct passwd *pw) FAST_FUNC; extern int correct_password(const struct passwd *pw) FAST_FUNC; /* Returns a malloced string */ #if !ENABLE_USE_BB_CRYPT diff --git a/libbb/setup_environment.c b/libbb/setup_environment.c index 78318ce62..f0802f0e5 100644 --- a/libbb/setup_environment.c +++ b/libbb/setup_environment.c @@ -30,12 +30,12 @@ #include "libbb.h" -void FAST_FUNC setup_environment(const char *shell, int clear_env, int change_env, const struct passwd *pw) +void FAST_FUNC setup_environment(const char *shell, int clear_env, int flags, const struct passwd *pw) { /* Change the current working directory to be the home directory * of the user */ if (chdir(pw->pw_dir)) { - xchdir("/"); + xchdir((flags & SETUP_ENV_TO_TMP) ? "/tmp" : "/"); bb_error_msg("can't chdir to home directory '%s'", pw->pw_dir); } @@ -55,8 +55,7 @@ void FAST_FUNC setup_environment(const char *shell, int clear_env, int change_en //xsetenv("LOGNAME", pw->pw_name); //xsetenv("HOME", pw->pw_dir); //xsetenv("SHELL", shell); - } - else if (change_env) { + } else if (flags & SETUP_ENV_CHANGEENV) { /* Set HOME, SHELL, and if not becoming a super-user, USER and LOGNAME. */ if (pw->pw_uid) { diff --git a/loginutils/login.c b/loginutils/login.c index 7a6f246ad..974125d88 100644 --- a/loginutils/login.c +++ b/loginutils/login.c @@ -478,7 +478,7 @@ int login_main(int argc UNUSED_PARAM, char **argv) if (!tmp || !*tmp) tmp = DEFAULT_SHELL; /* setup_environment params: shell, clear_env, change_env, pw */ - setup_environment(tmp, !(opt & LOGIN_OPT_p), 1, pw); + setup_environment(tmp, !(opt & LOGIN_OPT_p), SETUP_ENV_CHANGEENV, pw); motd(); diff --git a/miscutils/crontab.c b/miscutils/crontab.c index 044440435..7d5709521 100644 --- a/miscutils/crontab.c +++ b/miscutils/crontab.c @@ -17,22 +17,6 @@ #define CRONUPDATE "cron.update" #endif -static void change_user(const struct passwd *pas) -{ - xsetenv("USER", pas->pw_name); - xsetenv("HOME", pas->pw_dir); - xsetenv("SHELL", DEFAULT_SHELL); - - /* initgroups, setgid, setuid */ - change_identity(pas); - - if (chdir(pas->pw_dir) < 0) { - bb_perror_msg("chdir(%s) by %s failed", - pas->pw_dir, pas->pw_name); - xchdir("/tmp"); - } -} - static void edit_file(const struct passwd *pas, const char *file) { const char *ptr; @@ -46,7 +30,10 @@ static void edit_file(const struct passwd *pas, const char *file) } /* CHILD - change user and run editor */ - change_user(pas); + /* initgroups, setgid, setuid */ + change_identity(pas); + setup_environment(DEFAULT_SHELL, 0, + SETUP_ENV_CHANGEENV | SETUP_ENV_TO_TMP, pas); ptr = getenv("VISUAL"); if (!ptr) { ptr = getenv("EDITOR"); -- cgit v1.2.3