int main(int argc, char **argv)
{
int facility = LOG_DEST;
int daemon = 0;
int lockfd = -1;
const char *userarg = 0;
const char *grouparg = 0;
int droproot=0;
if (argc == 2 && argv[1][0] != '-')
{
/* backwards-compatibility mode */
close(1);
close(2);
checkfd(open("/dev/null", O_WRONLY), 1);
checkfd(open("/dev/null", O_WRONLY), 2);
do_logger(argv[1], facility, stdin);
exit(0);
}
if (argc <= 1)
{
fprintf(stderr,
"Usage: courierlogger [-name=name] [-pid=pidfile] [-facility=type]\n"
" [-start|-stop|-restart] [cmd [args...]]\n"
);
exit(1);
}
argv++, argc--;
while (argc > 0 && argv[0][0]=='-')
{
if (strncmp(argv[0],"-pid=",5) == 0 && argv[0][5])
{
pidarg=&argv[0][5];
lockfilename=malloc(strlen(pidarg)+sizeof(".lock"));
if (!lockfilename)
{
perror("malloc");
exit(1);
}
strcat(strcpy(lockfilename, pidarg), ".lock");
}
else
if (strncmp(argv[0],"-name=",6) == 0 && argv[0][6])
namearg=&argv[0][6];
else if (strncmp(argv[0],"-user="******"-group=",7) == 0)
grouparg=argv[0]+7;
else if (strcmp(argv[0], "-droproot") == 0)
droproot=1;
else
if (strncmp(argv[0],"-facility=",10) == 0)
{
struct lognames *p = facilitynames;
while (p->name && strcmp(p->name, &argv[0][10]))
p++;
if (p->name == 0)
{
fprintf(stderr, "Unknown facility name '%s'\n",
&argv[0][10]);
exit(1);
}
facility = p->value;
}
else
if (strcmp(argv[0],"-start") == 0)
daemon = 1;
else
if (strcmp(argv[0],"-stop") == 0)
daemon = 2;
else
if (strcmp(argv[0],"-restart") == 0)
daemon = 3;
else
if (strcmp(argv[0],"-respawn") == 0)
respawn = 1;
else
{
fprintf(stderr, "Unknown option '%s'\n", argv[0]);
exit(1);
}
argv++, argc--;
}
if (daemon && !pidarg)
{
fprintf(stderr, "-pid argument required\n");
exit(1);
}
if (!daemon && pidarg)
daemon = 1; /* -start implied */
if (!namearg && daemon != 2 && daemon != 3)
{
/* choose a default name based on the program we're running */
if (argc <= 0 || !argv[0] || !argv[0][0])
{
fprintf(stderr, "-name option required for standalone logger\n");
exit(1);
}
namearg = strrchr(argv[0],'/');
namearg = namearg ? namearg+1 : argv[0];
}
switch (daemon)
{
case 1: /* start */
if (argc <= 0 || !argv[0] || !argv[0][0])
{
fprintf(stderr, "-start must be followed by a command to execute\n");
exit(1);
}
lockfd=ll_daemon_start(lockfilename);
if (lockfd < 0)
{
perror("ll_daemon_start");
exit(1);
}
startchild(argv, droproot ? userarg:NULL,
droproot ? grouparg:NULL);
ll_daemon_started(pidarg, lockfd);
break;
case 2: /* stop */
exit(ll_daemon_stop(lockfilename, pidarg));
case 3: /* restart */
exit(ll_daemon_restart(lockfilename, pidarg));
default: /* run in foreground, with or without a child process */
if (argc > 0)
startchild(argv, droproot ? userarg:NULL,
droproot ? grouparg:NULL);
}
setuidgid(userarg, grouparg);
while (1)
{
int waitstat;
pid_t p2;
FILE *f = fdopen(0, "r");
do_logger(namearg, facility, f);
fclose(f);
if (child_pid < 0) break;
while ((p2=wait(&waitstat)) != child_pid &&
(p2 != -1 || errno != ECHILD))
;
if (hup_restart)
hup_restart = 0;
else if (respawn)
sleep (5);
else
break;
startchild(argv, NULL, NULL);
}
exit(0);
}
static int restart()
{
return (ll_daemon_restart(LOCKFILE, PIDFILE));
}