busybox/init/msvc.c

301 lines
6.6 KiB
C
Raw Normal View History

2003-05-11 20:22:39 +05:30
/*
* minit version 0.9.1 by Felix von Leitner
* ported to busybox by Glenn McGrath <bug1@optushome.com.au>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
#include <sys/fcntl.h>
#include <sys/file.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "busybox.h"
static int infd, outfd;
static char buf[1500];
static unsigned int fmt_ulong(char *dest, unsigned long i)
{
register unsigned long len, tmp, len2;
/* first count the number of bytes needed */
for (len = 1, tmp = i; tmp > 9; ++len)
tmp /= 10;
if (dest)
for (tmp = i, dest += len, len2 = len + 1; --len2; tmp /= 10)
*--dest = (tmp % 10) + '0';
return len;
}
static void addservice(char *service)
{
char *service_ptr;
if (strncmp(service, "/etc/minit/", 11) == 0) {
service += 11;
}
while ((service_ptr = last_char_is(service, '/')) != NULL) {
*service_ptr = 0;
}
strncpy(buf + 1, service, 1400);
buf[1400] = 0;
}
static int addreadwrite(char *service)
{
addservice(service);
write(infd, buf, strlen(buf));
return read(outfd, buf, 1500);
}
/* return PID, 0 if error */
static pid_t __readpid(char *service)
{
int len;
buf[0] = 'p';
len = addreadwrite(service);
if (len < 0)
return 0;
buf[len] = 0;
return atoi(buf);
}
/* return nonzero if error */
static int respawn(char *service, int yesno)
{
int len;
buf[0] = yesno ? 'R' : 'r';
len = addreadwrite(service);
return (len != 1 || buf[0] == '0');
}
/* return nonzero if error */
static int startservice(char *service)
{
int len;
buf[0] = 's';
len = addreadwrite(service);
return (len != 1 || buf[0] == '0');
}
extern int msvc_main(int argc, char **argv)
{
if (argc < 2) {
bb_show_usage();
}
infd = bb_xopen("/etc/minit/in", O_WRONLY);
outfd = bb_xopen("/etc/minit/out", O_RDONLY);
while (lockf(infd, F_LOCK, 1)) {
bb_perror_msg("could not aquire lock!\n");
sleep(1);
}
if (argc == 2) {
pid_t pid = __readpid(argv[1]);
if (buf[0] != '0') {
int len;
int up_time;
printf("%s: ", argv[1]);
if (pid == 0)
printf("down ");
else if (pid == 1)
printf("finished ");
else {
printf("up (pid %d) ", pid);
}
buf[0] = 'u';
len = addreadwrite(argv[1]);
if (len < 0) {
up_time = 0;
} else {
buf[len] = 0;
up_time = atoi(buf);
}
printf("%d seconds\n", up_time);
if (pid == 0)
return 2;
else if (pid == 1)
return 3;
else
return 0;
} else {
bb_error_msg_and_die("no such service");
}
} else {
int i;
int ret = 0;
int sig = 0;
pid_t pid;
if (argv[1][0] == '-') {
switch (argv[1][1]) {
case 'g':
for (i = 2; i < argc; ++i) {
pid = __readpid(argv[i]);
if (pid < 2) {
if (pid == 1) {
bb_error_msg("%s, service termination", argv[i]);
} else {
bb_error_msg("%s, no such service", argv[i]);
}
ret = 1;
}
printf("%d\n", pid);
}
break;
case 'p':
sig = SIGSTOP;
goto dokill;
break;
case 'c':
sig = SIGCONT;
goto dokill;
break;
case 'h':
sig = SIGHUP;
goto dokill;
break;
case 'a':
sig = SIGALRM;
goto dokill;
break;
case 'i':
sig = SIGINT;
goto dokill;
break;
case 't':
sig = SIGTERM;
goto dokill;
break;
case 'k':
sig = SIGKILL;
goto dokill;
break;
case 'o':
for (i = 2; i < argc; ++i)
if (startservice(argv[i]) || respawn(argv[i], 0)) {
bb_error_msg("Couldnt not start %s\n", argv[i]);
ret = 1;
}
break;
case 'd':
for (i = 2; i < argc; ++i) {
pid = __readpid(argv[i]);
if (pid == 0) {
bb_error_msg("%s, no such service\n", argv[i]);
ret = 1;
} else if (pid == 1)
continue;
if (respawn(argv[i], 0) || kill(pid, SIGTERM)
|| kill(pid, SIGCONT));
}
break;
case 'u':
for (i = 2; i < argc; ++i) {
if (startservice(argv[i]) || respawn(argv[i], 1)) {
bb_error_msg("Couldnt not start %s\n", argv[i]);
ret = 1;
}
break;
}
case 'C':
for (i = 2; i < argc; ++i) {
int len;
buf[0] = 'C';
len = addreadwrite(argv[i]);
if (len != 1 || buf[0] == '0') {
bb_error_msg("%s has terminated or was killed\n",
argv[i]);
ret = 1;
}
}
break;
case 'P':
pid = atoi(argv[1] + 2);
if (pid > 1) {
char *tmp;
int len;
buf[0] = 'P';
addservice(argv[2]);
tmp = buf + strlen(buf) + 1;
tmp[fmt_ulong(tmp, pid)] = 0;
write(infd, buf, strlen(buf) + strlen(tmp) + 2);
len = read(outfd, buf, 1500);
if (len != 1 || buf[0] == '0') {
bb_error_msg_and_die("Couldnt not set pid of service %s\n", argv[2]);
}
}
break;
default:
bb_show_usage();
}
} else {
bb_show_usage();
}
return ret;
dokill:
for (i = 2; i < argc; i++) {
pid = __readpid(argv[i]);
if (!pid) {
bb_error_msg("%s no such service\n", argv[i]);
ret = 1;
}
if (kill(pid, sig)) {
bb_error_msg("%s, could not send signal %d to PID %d\n",
argv[i], sig, pid);
ret = 1;
}
}
return ret;
}
}
/*
-u Up. If the service is not running, start it. If the service stops,
restart it.
-d Down. If the service is running, send it a TERM signal and then a CONT
signal. After it stops, do not restart it.
-o Once. If the service is not running, start it. Do not restart it if it
stops.
-r Tell supervise that the service is normally running; this affects status
messages.
-s Tell supervise that the service is normally stopped; this affects status
messages.
-p Pause. Send the service a STOP signal.
-c Continue. Send the service a CONT signal.
-h Hangup. Send the service a HUP signal.
-a Alarm. Send the service an ALRM signal.
-i Interrupt. Send the service an INT signal.
-t Terminate. Send the service a TERM signal.
-k Kill. Send the service a KILL signal.
-x Exit. supervise will quit as soon as the service is down.
*/