mirror of
https://git.busybox.net/busybox.git
synced 2024-12-01 01:23:32 +08:00
crontab: we indent using tabs
This commit is contained in:
Denis Vlasenko
parent
956a569360
commit
94d5d82bd8
@ -22,315 +22,320 @@
|
|||||||
#define CRONUPDATE "cron.update"
|
#define CRONUPDATE "cron.update"
|
||||||
#endif
|
#endif
|
||||||
#ifndef PATH_VI
|
#ifndef PATH_VI
|
||||||
#define PATH_VI "/bin/vi" /* location of vi */
|
#define PATH_VI "/bin/vi" /* location of vi */
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
static const char *CDir = CRONTABS;
|
static const char *CDir = CRONTABS;
|
||||||
|
|
||||||
static void EditFile(const char *user, const char *file);
|
static void EditFile(const char *user, const char *file);
|
||||||
static int GetReplaceStream(const char *user, const char *file);
|
static int GetReplaceStream(const char *user, const char *file);
|
||||||
static int ChangeUser(const char *user, short dochdir);
|
static int ChangeUser(const char *user, short dochdir);
|
||||||
|
|
||||||
int crontab_main(int ac, char **av)
|
int crontab_main(int ac, char **av)
|
||||||
{
|
{
|
||||||
enum { NONE, EDIT, LIST, REPLACE, DELETE } option = NONE;
|
enum { NONE, EDIT, LIST, REPLACE, DELETE } option = NONE;
|
||||||
const struct passwd *pas;
|
const struct passwd *pas;
|
||||||
const char *repFile = NULL;
|
const char *repFile = NULL;
|
||||||
int repFd = 0;
|
int repFd = 0;
|
||||||
int i;
|
int i;
|
||||||
char caller[256]; /* user that ran program */
|
char caller[256]; /* user that ran program */
|
||||||
int UserId;
|
char buf[1024];
|
||||||
|
int UserId;
|
||||||
|
|
||||||
UserId = getuid();
|
UserId = getuid();
|
||||||
if ((pas = getpwuid(UserId)) == NULL)
|
pas = getpwuid(UserId);
|
||||||
bb_perror_msg_and_die("getpwuid");
|
if (pas == NULL)
|
||||||
|
bb_perror_msg_and_die("getpwuid");
|
||||||
|
|
||||||
safe_strncpy(caller, pas->pw_name, sizeof(caller));
|
safe_strncpy(caller, pas->pw_name, sizeof(caller));
|
||||||
|
|
||||||
i = 1;
|
i = 1;
|
||||||
if (ac > 1) {
|
if (ac > 1) {
|
||||||
if (av[1][0] == '-' && av[1][1] == 0) {
|
if (av[1][0] == '-' && av[1][1] == 0) {
|
||||||
option = REPLACE;
|
option = REPLACE;
|
||||||
++i;
|
++i;
|
||||||
} else if (av[1][0] != '-') {
|
} else if (av[1][0] != '-') {
|
||||||
option = REPLACE;
|
option = REPLACE;
|
||||||
++i;
|
++i;
|
||||||
repFile = av[1];
|
repFile = av[1];
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (; i < ac; ++i) {
|
|
||||||
char *ptr = av[i];
|
|
||||||
|
|
||||||
if (*ptr != '-')
|
|
||||||
break;
|
|
||||||
ptr += 2;
|
|
||||||
|
|
||||||
switch(ptr[-1]) {
|
|
||||||
case 'l':
|
|
||||||
if (ptr[-1] == 'l')
|
|
||||||
option = LIST;
|
|
||||||
/* fall through */
|
|
||||||
case 'e':
|
|
||||||
if (ptr[-1] == 'e')
|
|
||||||
option = EDIT;
|
|
||||||
/* fall through */
|
|
||||||
case 'd':
|
|
||||||
if (ptr[-1] == 'd')
|
|
||||||
option = DELETE;
|
|
||||||
/* fall through */
|
|
||||||
case 'u':
|
|
||||||
if (i + 1 < ac && av[i+1][0] != '-') {
|
|
||||||
++i;
|
|
||||||
if (getuid() == geteuid()) {
|
|
||||||
pas = getpwnam(av[i]);
|
|
||||||
if (pas) {
|
|
||||||
UserId = pas->pw_uid;
|
|
||||||
} else {
|
|
||||||
bb_error_msg_and_die("user %s unknown", av[i]);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
bb_error_msg_and_die("only the superuser may specify a user");
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
break;
|
|
||||||
case 'c':
|
|
||||||
if (getuid() == geteuid()) {
|
|
||||||
CDir = (*ptr) ? ptr : av[++i];
|
|
||||||
} else {
|
|
||||||
bb_error_msg_and_die("-c option: superuser only");
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
i = ac;
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
if (i != ac || option == NONE)
|
|
||||||
bb_show_usage();
|
|
||||||
|
|
||||||
/*
|
for (; i < ac; ++i) {
|
||||||
* Get password entry
|
char *ptr = av[i];
|
||||||
*/
|
|
||||||
|
|
||||||
if ((pas = getpwuid(UserId)) == NULL)
|
if (*ptr != '-')
|
||||||
bb_perror_msg_and_die("getpwuid");
|
break;
|
||||||
|
ptr += 2;
|
||||||
|
|
||||||
/*
|
switch (ptr[-1]) {
|
||||||
* If there is a replacement file, obtain a secure descriptor to it.
|
case 'l':
|
||||||
*/
|
if (ptr[-1] == 'l')
|
||||||
|
option = LIST;
|
||||||
if (repFile) {
|
/* fall through */
|
||||||
repFd = GetReplaceStream(caller, repFile);
|
case 'e':
|
||||||
if (repFd < 0)
|
if (ptr[-1] == 'e')
|
||||||
bb_error_msg_and_die("unable to read replacement file");
|
option = EDIT;
|
||||||
}
|
/* fall through */
|
||||||
|
case 'd':
|
||||||
/*
|
if (ptr[-1] == 'd')
|
||||||
* Change directory to our crontab directory
|
option = DELETE;
|
||||||
*/
|
/* fall through */
|
||||||
|
case 'u':
|
||||||
xchdir(CDir);
|
if (i + 1 < ac && av[i+1][0] != '-') {
|
||||||
|
++i;
|
||||||
/*
|
if (getuid() == geteuid()) {
|
||||||
* Handle options as appropriate
|
pas = getpwnam(av[i]);
|
||||||
*/
|
if (pas) {
|
||||||
|
UserId = pas->pw_uid;
|
||||||
switch(option) {
|
} else {
|
||||||
case LIST:
|
bb_error_msg_and_die("user %s unknown", av[i]);
|
||||||
{
|
}
|
||||||
FILE *fi;
|
} else {
|
||||||
char buf[1024];
|
bb_error_msg_and_die("only the superuser may specify a user");
|
||||||
|
}
|
||||||
if ((fi = fopen(pas->pw_name, "r"))) {
|
}
|
||||||
while (fgets(buf, sizeof(buf), fi) != NULL)
|
break;
|
||||||
fputs(buf, stdout);
|
case 'c':
|
||||||
fclose(fi);
|
if (getuid() == geteuid()) {
|
||||||
} else {
|
CDir = (*ptr) ? ptr : av[++i];
|
||||||
bb_error_msg("no crontab for %s", pas->pw_name);
|
} else {
|
||||||
}
|
bb_error_msg_and_die("-c option: superuser only");
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
case EDIT:
|
default:
|
||||||
{
|
i = ac;
|
||||||
FILE *fi;
|
break;
|
||||||
int fd;
|
|
||||||
int n;
|
|
||||||
char tmp[128];
|
|
||||||
char buf[1024];
|
|
||||||
|
|
||||||
snprintf(tmp, sizeof(tmp), TMPDIR "/crontab.%d", getpid());
|
|
||||||
fd = xopen3(tmp, O_RDWR|O_CREAT|O_TRUNC|O_EXCL, 0600);
|
|
||||||
chown(tmp, getuid(), getgid());
|
|
||||||
if ((fi = fopen(pas->pw_name, "r"))) {
|
|
||||||
while ((n = fread(buf, 1, sizeof(buf), fi)) > 0)
|
|
||||||
write(fd, buf, n);
|
|
||||||
}
|
|
||||||
EditFile(caller, tmp);
|
|
||||||
remove(tmp);
|
|
||||||
lseek(fd, 0L, 0);
|
|
||||||
repFd = fd;
|
|
||||||
}
|
|
||||||
option = REPLACE;
|
|
||||||
/* fall through */
|
|
||||||
case REPLACE:
|
|
||||||
{
|
|
||||||
char buf[1024];
|
|
||||||
char path[1024];
|
|
||||||
int fd;
|
|
||||||
int n;
|
|
||||||
|
|
||||||
snprintf(path, sizeof(path), "%s.new", pas->pw_name);
|
|
||||||
if ((fd = open(path, O_CREAT|O_TRUNC|O_APPEND|O_WRONLY, 0600)) >= 0) {
|
|
||||||
while ((n = read(repFd, buf, sizeof(buf))) > 0) {
|
|
||||||
write(fd, buf, n);
|
|
||||||
}
|
}
|
||||||
close(fd);
|
|
||||||
rename(path, pas->pw_name);
|
|
||||||
} else {
|
|
||||||
bb_error_msg("unable to create %s/%s", CDir, path);
|
|
||||||
}
|
|
||||||
close(repFd);
|
|
||||||
}
|
}
|
||||||
break;
|
if (i != ac || option == NONE)
|
||||||
case DELETE:
|
bb_show_usage();
|
||||||
remove(pas->pw_name);
|
|
||||||
break;
|
|
||||||
case NONE:
|
|
||||||
default:
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Bump notification file. Handle window where crond picks file up
|
* Get password entry
|
||||||
* before we can write our entry out.
|
*/
|
||||||
*/
|
|
||||||
|
|
||||||
if (option == REPLACE || option == DELETE) {
|
pas = getpwuid(UserId);
|
||||||
FILE *fo;
|
if (pas == NULL)
|
||||||
struct stat st;
|
bb_perror_msg_and_die("getpwuid");
|
||||||
|
|
||||||
while ((fo = fopen(CRONUPDATE, "a"))) {
|
/*
|
||||||
fprintf(fo, "%s\n", pas->pw_name);
|
* If there is a replacement file, obtain a secure descriptor to it.
|
||||||
fflush(fo);
|
*/
|
||||||
if (fstat(fileno(fo), &st) != 0 || st.st_nlink != 0) {
|
|
||||||
fclose(fo);
|
if (repFile) {
|
||||||
|
repFd = GetReplaceStream(caller, repFile);
|
||||||
|
if (repFd < 0)
|
||||||
|
bb_error_msg_and_die("unable to read replacement file");
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Change directory to our crontab directory
|
||||||
|
*/
|
||||||
|
|
||||||
|
xchdir(CDir);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Handle options as appropriate
|
||||||
|
*/
|
||||||
|
|
||||||
|
switch (option) {
|
||||||
|
case LIST:
|
||||||
|
{
|
||||||
|
FILE *fi;
|
||||||
|
|
||||||
|
fi = fopen(pas->pw_name, "r");
|
||||||
|
if (fi) {
|
||||||
|
while (fgets(buf, sizeof(buf), fi) != NULL)
|
||||||
|
fputs(buf, stdout);
|
||||||
|
fclose(fi);
|
||||||
|
} else {
|
||||||
|
bb_error_msg("no crontab for %s", pas->pw_name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case EDIT:
|
||||||
|
{
|
||||||
|
FILE *fi;
|
||||||
|
int fd;
|
||||||
|
int n;
|
||||||
|
char tmp[128];
|
||||||
|
|
||||||
|
snprintf(tmp, sizeof(tmp), TMPDIR "/crontab.%d", getpid());
|
||||||
|
fd = xopen3(tmp, O_RDWR|O_CREAT|O_TRUNC|O_EXCL, 0600);
|
||||||
|
chown(tmp, getuid(), getgid());
|
||||||
|
fi = fopen(pas->pw_name, "r");
|
||||||
|
if (fi) {
|
||||||
|
while ((n = fread(buf, 1, sizeof(buf), fi)) > 0)
|
||||||
|
write(fd, buf, n);
|
||||||
|
}
|
||||||
|
EditFile(caller, tmp);
|
||||||
|
remove(tmp);
|
||||||
|
lseek(fd, 0L, 0);
|
||||||
|
repFd = fd;
|
||||||
|
}
|
||||||
|
option = REPLACE;
|
||||||
|
/* fall through */
|
||||||
|
case REPLACE:
|
||||||
|
{
|
||||||
|
char path[1024];
|
||||||
|
int fd;
|
||||||
|
int n;
|
||||||
|
|
||||||
|
snprintf(path, sizeof(path), "%s.new", pas->pw_name);
|
||||||
|
fd = open(path, O_CREAT|O_TRUNC|O_APPEND|O_WRONLY, 0600);
|
||||||
|
if (fd >= 0) {
|
||||||
|
while ((n = read(repFd, buf, sizeof(buf))) > 0) {
|
||||||
|
write(fd, buf, n);
|
||||||
|
}
|
||||||
|
close(fd);
|
||||||
|
rename(path, pas->pw_name);
|
||||||
|
} else {
|
||||||
|
bb_error_msg("unable to create %s/%s", CDir, path);
|
||||||
|
}
|
||||||
|
close(repFd);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case DELETE:
|
||||||
|
remove(pas->pw_name);
|
||||||
|
break;
|
||||||
|
case NONE:
|
||||||
|
default:
|
||||||
break;
|
break;
|
||||||
}
|
|
||||||
fclose(fo);
|
|
||||||
/* loop */
|
|
||||||
}
|
}
|
||||||
if (fo == NULL) {
|
|
||||||
bb_error_msg("unable to append to %s/%s", CDir, CRONUPDATE);
|
/*
|
||||||
|
* Bump notification file. Handle window where crond picks file up
|
||||||
|
* before we can write our entry out.
|
||||||
|
*/
|
||||||
|
|
||||||
|
if (option == REPLACE || option == DELETE) {
|
||||||
|
FILE *fo;
|
||||||
|
struct stat st;
|
||||||
|
|
||||||
|
while ((fo = fopen(CRONUPDATE, "a"))) {
|
||||||
|
fprintf(fo, "%s\n", pas->pw_name);
|
||||||
|
fflush(fo);
|
||||||
|
if (fstat(fileno(fo), &st) != 0 || st.st_nlink != 0) {
|
||||||
|
fclose(fo);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
fclose(fo);
|
||||||
|
/* loop */
|
||||||
|
}
|
||||||
|
if (fo == NULL) {
|
||||||
|
bb_error_msg("unable to append to %s/%s", CDir, CRONUPDATE);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
return 0;
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static int GetReplaceStream(const char *user, const char *file)
|
static int GetReplaceStream(const char *user, const char *file)
|
||||||
{
|
{
|
||||||
int filedes[2];
|
int filedes[2];
|
||||||
int pid;
|
int pid;
|
||||||
int fd;
|
int fd;
|
||||||
int n;
|
int n;
|
||||||
char buf[1024];
|
char buf[1024];
|
||||||
|
|
||||||
|
if (pipe(filedes) < 0) {
|
||||||
|
perror("pipe");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
pid = fork();
|
||||||
|
if (pid < 0) {
|
||||||
|
perror("fork");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if (pid > 0) {
|
||||||
|
/*
|
||||||
|
* PARENT
|
||||||
|
*/
|
||||||
|
|
||||||
|
close(filedes[1]);
|
||||||
|
if (read(filedes[0], buf, 1) != 1) {
|
||||||
|
close(filedes[0]);
|
||||||
|
filedes[0] = -1;
|
||||||
|
}
|
||||||
|
return filedes[0];
|
||||||
|
}
|
||||||
|
|
||||||
if (pipe(filedes) < 0) {
|
|
||||||
perror("pipe");
|
|
||||||
return(-1);
|
|
||||||
}
|
|
||||||
if ((pid = fork()) < 0) {
|
|
||||||
perror("fork");
|
|
||||||
return(-1);
|
|
||||||
}
|
|
||||||
if (pid > 0) {
|
|
||||||
/*
|
/*
|
||||||
* PARENT
|
* CHILD
|
||||||
*/
|
*/
|
||||||
|
|
||||||
close(filedes[1]);
|
close(filedes[0]);
|
||||||
if (read(filedes[0], buf, 1) != 1) {
|
|
||||||
close(filedes[0]);
|
if (ChangeUser(user, 0) < 0)
|
||||||
filedes[0] = -1;
|
exit(0);
|
||||||
|
|
||||||
|
bb_default_error_retval = 0;
|
||||||
|
fd = xopen(file, O_RDONLY);
|
||||||
|
buf[0] = 0;
|
||||||
|
write(filedes[1], buf, 1);
|
||||||
|
while ((n = read(fd, buf, sizeof(buf))) > 0) {
|
||||||
|
write(filedes[1], buf, n);
|
||||||
}
|
}
|
||||||
return(filedes[0]);
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* CHILD
|
|
||||||
*/
|
|
||||||
|
|
||||||
close(filedes[0]);
|
|
||||||
|
|
||||||
if (ChangeUser(user, 0) < 0)
|
|
||||||
exit(0);
|
exit(0);
|
||||||
|
|
||||||
bb_default_error_retval = 0;
|
|
||||||
fd = xopen(file, O_RDONLY);
|
|
||||||
buf[0] = 0;
|
|
||||||
write(filedes[1], buf, 1);
|
|
||||||
while ((n = read(fd, buf, sizeof(buf))) > 0) {
|
|
||||||
write(filedes[1], buf, n);
|
|
||||||
}
|
|
||||||
exit(0);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void EditFile(const char *user, const char *file)
|
static void EditFile(const char *user, const char *file)
|
||||||
{
|
{
|
||||||
int pid;
|
int pid = fork();
|
||||||
|
|
||||||
if ((pid = fork()) == 0) {
|
if (pid == 0) {
|
||||||
/*
|
/*
|
||||||
* CHILD - change user and run editor
|
* CHILD - change user and run editor
|
||||||
*/
|
*/
|
||||||
char *ptr;
|
char *ptr;
|
||||||
char visual[1024];
|
char visual[1024];
|
||||||
|
|
||||||
if (ChangeUser(user, 1) < 0)
|
if (ChangeUser(user, 1) < 0)
|
||||||
exit(0);
|
exit(0);
|
||||||
if ((ptr = getenv("VISUAL")) == NULL || strlen(ptr) > 256)
|
ptr = getenv("VISUAL");
|
||||||
ptr = PATH_VI;
|
if (ptr == NULL || strlen(ptr) > 256)
|
||||||
|
ptr = PATH_VI;
|
||||||
|
|
||||||
snprintf(visual, sizeof(visual), "%s %s", ptr, file);
|
snprintf(visual, sizeof(visual), "%s %s", ptr, file);
|
||||||
execl(DEFAULT_SHELL, DEFAULT_SHELL, "-c", visual, NULL);
|
execl(DEFAULT_SHELL, DEFAULT_SHELL, "-c", visual, NULL);
|
||||||
perror("exec");
|
perror("exec");
|
||||||
exit(0);
|
exit(0);
|
||||||
}
|
}
|
||||||
if (pid < 0) {
|
if (pid < 0) {
|
||||||
/*
|
/*
|
||||||
* PARENT - failure
|
* PARENT - failure
|
||||||
*/
|
*/
|
||||||
bb_perror_msg_and_die("fork");
|
bb_perror_msg_and_die("fork");
|
||||||
}
|
}
|
||||||
wait4(pid, NULL, 0, NULL);
|
wait4(pid, NULL, 0, NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int ChangeUser(const char *user, short dochdir)
|
static int ChangeUser(const char *user, short dochdir)
|
||||||
{
|
{
|
||||||
struct passwd *pas;
|
struct passwd *pas;
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Obtain password entry and change privileges
|
* Obtain password entry and change privileges
|
||||||
*/
|
*/
|
||||||
|
|
||||||
if ((pas = getpwnam(user)) == NULL) {
|
pas = getpwnam(user);
|
||||||
bb_perror_msg_and_die("failed to get uid for %s", user);
|
if (pas == NULL) {
|
||||||
return(-1);
|
bb_perror_msg_and_die("failed to get uid for %s", user);
|
||||||
}
|
|
||||||
setenv("USER", pas->pw_name, 1);
|
|
||||||
setenv("HOME", pas->pw_dir, 1);
|
|
||||||
setenv("SHELL", DEFAULT_SHELL, 1);
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Change running state to the user in question
|
|
||||||
*/
|
|
||||||
change_identity(pas);
|
|
||||||
|
|
||||||
if (dochdir) {
|
|
||||||
if (chdir(pas->pw_dir) < 0) {
|
|
||||||
bb_perror_msg("chdir failed: %s %s", user, pas->pw_dir);
|
|
||||||
xchdir(TMPDIR);
|
|
||||||
}
|
}
|
||||||
}
|
setenv("USER", pas->pw_name, 1);
|
||||||
return(pas->pw_uid);
|
setenv("HOME", pas->pw_dir, 1);
|
||||||
|
setenv("SHELL", DEFAULT_SHELL, 1);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Change running state to the user in question
|
||||||
|
*/
|
||||||
|
change_identity(pas);
|
||||||
|
|
||||||
|
if (dochdir) {
|
||||||
|
if (chdir(pas->pw_dir) < 0) {
|
||||||
|
bb_perror_msg("chdir(%s) by %s failed", pas->pw_dir, user);
|
||||||
|
xchdir(TMPDIR);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return pas->pw_uid;
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user