mirror of
https://github.com/sheumann/hush.git
synced 2024-08-19 13:28:57 +00:00
613 lines
12 KiB
C
613 lines
12 KiB
C
/*
|
|
* 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 <time.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <limits.h>
|
|
#include <errno.h>
|
|
#include <sys/fcntl.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/poll.h>
|
|
#include <sys/reboot.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <sys/un.h>
|
|
#include <sys/wait.h>
|
|
#include <linux/kd.h>
|
|
|
|
#include "busybox.h"
|
|
|
|
#define MINITROOT "/etc/minit"
|
|
|
|
static int i_am_init;
|
|
static int infd, outfd;
|
|
|
|
extern char **environ;
|
|
|
|
static struct process {
|
|
char *name;
|
|
pid_t pid;
|
|
char respawn;
|
|
char circular;
|
|
time_t startedat;
|
|
int __stdin, __stdout;
|
|
int logservice;
|
|
} *root;
|
|
|
|
static int maxprocess = -1;
|
|
|
|
static int processalloc = 0;
|
|
|
|
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;
|
|
}
|
|
|
|
/* split buf into n strings that are separated by c. return n as *len.
|
|
* Allocate plus more slots and leave the first ofs of them alone. */
|
|
static char **split(char *buf, int c, int *len, int plus, int ofs)
|
|
{
|
|
int n = 1;
|
|
char **v = 0;
|
|
char **w;
|
|
|
|
/* step 1: count tokens */
|
|
char *s;
|
|
|
|
for (s = buf; *s; s++)
|
|
if (*s == c)
|
|
n++;
|
|
/* step 2: allocate space for pointers */
|
|
v = (char **) malloc((n + plus) * sizeof(char *));
|
|
if (!v)
|
|
return 0;
|
|
w = v + ofs;
|
|
*w++ = buf;
|
|
for (s = buf;; s++) {
|
|
while (*s && *s != c)
|
|
s++;
|
|
if (*s == 0)
|
|
break;
|
|
if (*s == c) {
|
|
*s = 0;
|
|
*w++ = s + 1;
|
|
}
|
|
}
|
|
*len = w - v;
|
|
return v;
|
|
}
|
|
|
|
static int openreadclose(char *fn, char **buf, unsigned long *len)
|
|
{
|
|
int fd = open(fn, O_RDONLY);
|
|
|
|
if (fd < 0)
|
|
return -1;
|
|
if (!*buf) {
|
|
*len = lseek(fd, 0, SEEK_END);
|
|
lseek(fd, 0, SEEK_SET);
|
|
*buf = (char *) malloc(*len + 1);
|
|
if (!*buf) {
|
|
close(fd);
|
|
return -1;
|
|
}
|
|
}
|
|
*len = read(fd, *buf, *len);
|
|
if (*len != (unsigned long) -1)
|
|
(*buf)[*len] = 0;
|
|
return close(fd);
|
|
}
|
|
|
|
/* return index of service in process data structure or -1 if not found */
|
|
static int findservice(char *service)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i <= maxprocess; i++) {
|
|
if (!strcmp(root[i].name, service))
|
|
return i;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
/* look up process index in data structure by PID */
|
|
static int findbypid(pid_t pid)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i <= maxprocess; i++) {
|
|
if (root[i].pid == pid)
|
|
return i;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
/* clear circular dependency detection flags */
|
|
static void circsweep(void)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i <= maxprocess; i++)
|
|
root[i].circular = 0;
|
|
}
|
|
|
|
/* add process to data structure, return index or -1 */
|
|
static int addprocess(struct process *p)
|
|
{
|
|
if (maxprocess + 1 >= processalloc) {
|
|
struct process *fump;
|
|
|
|
processalloc += 8;
|
|
if ((fump =
|
|
(struct process *) xrealloc(root,
|
|
processalloc *
|
|
sizeof(struct process))) == 0)
|
|
return -1;
|
|
root = fump;
|
|
}
|
|
memmove(&root[++maxprocess], p, sizeof(struct process));
|
|
return maxprocess;
|
|
}
|
|
|
|
/* load a service into the process data structure and return index or -1
|
|
* if failed */
|
|
static int loadservice(char *service)
|
|
{
|
|
struct process tmp;
|
|
int fd;
|
|
|
|
if (*service == 0)
|
|
return -1;
|
|
fd = findservice(service);
|
|
if (fd >= 0)
|
|
return fd;
|
|
if (chdir(MINITROOT) || chdir(service))
|
|
return -1;
|
|
if (!(tmp.name = strdup(service)))
|
|
return -1;
|
|
tmp.pid = 0;
|
|
fd = open("respawn", O_RDONLY);
|
|
if (fd >= 0) {
|
|
tmp.respawn = 1;
|
|
close(fd);
|
|
} else
|
|
tmp.respawn = 0;
|
|
tmp.startedat = 0;
|
|
tmp.circular = 0;
|
|
tmp.__stdin = 0;
|
|
tmp.__stdout = 1;
|
|
{
|
|
char *logservice = alloca(strlen(service) + 5);
|
|
|
|
strcpy(logservice, service);
|
|
strcat(logservice, "/log");
|
|
tmp.logservice = loadservice(logservice);
|
|
if (tmp.logservice >= 0) {
|
|
int pipefd[2];
|
|
|
|
if (pipe(pipefd))
|
|
return -1;
|
|
root[tmp.logservice].__stdin = pipefd[0];
|
|
tmp.__stdout = pipefd[1];
|
|
}
|
|
}
|
|
return (addprocess(&tmp));
|
|
}
|
|
|
|
/* usage: isup(findservice("sshd")).
|
|
* returns nonzero if process is up */
|
|
static int isup(int service)
|
|
{
|
|
if (service < 0)
|
|
return 0;
|
|
return (root[service].pid != 0);
|
|
}
|
|
|
|
static void opendevconsole(void)
|
|
{
|
|
int fd;
|
|
|
|
if ((fd = open("/dev/console", O_RDWR | O_NOCTTY)) >= 0) {
|
|
dup2(fd, 0);
|
|
dup2(fd, 1);
|
|
dup2(fd, 2);
|
|
if (fd > 2)
|
|
close(fd);
|
|
}
|
|
}
|
|
|
|
/* called from inside the service directory, return the PID or 0 on error */
|
|
static pid_t forkandexec(int pause_flag, int service)
|
|
{
|
|
char **argv = 0;
|
|
int count = 0;
|
|
pid_t p;
|
|
int fd;
|
|
unsigned long len;
|
|
char *s = 0;
|
|
int argc;
|
|
char *argv0 = 0;
|
|
|
|
again:
|
|
switch (p = fork()) {
|
|
case (pid_t) - 1:
|
|
if (count > 3)
|
|
return 0;
|
|
sleep(++count * 2);
|
|
goto again;
|
|
case 0:
|
|
/* child */
|
|
|
|
if (i_am_init) {
|
|
ioctl(0, TIOCNOTTY, 0);
|
|
setsid();
|
|
opendevconsole();
|
|
tcsetpgrp(0, getpgrp());
|
|
}
|
|
close(infd);
|
|
close(outfd);
|
|
if (pause_flag) {
|
|
struct timespec req;
|
|
|
|
req.tv_sec = 0;
|
|
req.tv_nsec = 500000000;
|
|
nanosleep(&req, 0);
|
|
}
|
|
if (!openreadclose("params", &s, &len)) {
|
|
argv = split(s, '\n', &argc, 2, 1);
|
|
if (argv[argc - 1])
|
|
argv[argc - 1] = 0;
|
|
else
|
|
argv[argc] = 0;
|
|
} else {
|
|
argv = (char **) xmalloc(2 * sizeof(char *));
|
|
argv[1] = 0;
|
|
}
|
|
argv0 = (char *) xmalloc(PATH_MAX + 1);
|
|
if (!argv || !argv0)
|
|
goto abort;
|
|
if (readlink("run", argv0, PATH_MAX) < 0) {
|
|
if (errno != EINVAL)
|
|
goto abort; /* not a symbolic link */
|
|
argv0 = strdup("./run");
|
|
}
|
|
argv[0] = strrchr(argv0, '/');
|
|
if (argv[0])
|
|
argv[0]++;
|
|
else
|
|
argv[0] = argv0;
|
|
if (root[service].__stdin != 0)
|
|
dup2(root[service].__stdin, 0);
|
|
if (root[service].__stdout != 1) {
|
|
dup2(root[service].__stdout, 1);
|
|
dup2(root[service].__stdout, 2);
|
|
}
|
|
{
|
|
int i;
|
|
|
|
for (i = 3; i < 1024; ++i)
|
|
close(i);
|
|
}
|
|
execve(argv0, argv, environ);
|
|
_exit(0);
|
|
abort:
|
|
free(argv0);
|
|
free(argv);
|
|
_exit(0);
|
|
default:
|
|
fd = open("sync", O_RDONLY);
|
|
if (fd >= 0) {
|
|
pid_t p2;
|
|
|
|
close(fd);
|
|
p2 = waitpid(p, 0, 0);
|
|
return 1;
|
|
}
|
|
return p;
|
|
}
|
|
}
|
|
|
|
/* start a service, return nonzero on error */
|
|
static int startnodep(int service, int pause_flag)
|
|
{
|
|
/* step 1: see if the process is already up */
|
|
if (isup(service))
|
|
return 0;
|
|
|
|
/* step 2: fork and exec service, put PID in data structure */
|
|
if (chdir(MINITROOT) || chdir(root[service].name))
|
|
return -1;
|
|
root[service].startedat = time(0);
|
|
root[service].pid = forkandexec(pause_flag, service);
|
|
return root[service].pid;
|
|
}
|
|
|
|
static int startservice(int service, int pause_flag)
|
|
{
|
|
int dir = -1;
|
|
unsigned long len;
|
|
char *s = 0;
|
|
pid_t pid;
|
|
|
|
if (service < 0)
|
|
return 0;
|
|
if (root[service].circular)
|
|
return 0;
|
|
root[service].circular = 1;
|
|
if (root[service].logservice >= 0)
|
|
startservice(root[service].logservice, pause_flag);
|
|
if (chdir(MINITROOT) || chdir(root[service].name))
|
|
return -1;
|
|
if ((dir = open(".", O_RDONLY)) >= 0) {
|
|
if (!openreadclose("depends", &s, &len)) {
|
|
char **deps;
|
|
int depc, i;
|
|
|
|
deps = split(s, '\n', &depc, 0, 0);
|
|
for (i = 0; i < depc; i++) {
|
|
int service_index;
|
|
|
|
if (deps[i][0] == '#')
|
|
continue;
|
|
service_index = loadservice(deps[i]);
|
|
if (service_index >= 0 && root[service_index].pid != 1)
|
|
startservice(service_index, 0);
|
|
}
|
|
fchdir(dir);
|
|
}
|
|
pid = startnodep(service, pause_flag);
|
|
close(dir);
|
|
dir = -1;
|
|
return pid;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static void sulogin(void)
|
|
{
|
|
/* exiting on an initialization failure is not a good idea for init */
|
|
char *argv[] = { "sulogin", 0 };
|
|
execve("/sbin/sulogin", argv, environ);
|
|
exit(1);
|
|
}
|
|
|
|
static void handlekilled(pid_t killed)
|
|
{
|
|
int i;
|
|
|
|
if (killed == (pid_t) - 1) {
|
|
write(2, "all services exited.\n", 21);
|
|
exit(0);
|
|
}
|
|
if (killed == 0)
|
|
return;
|
|
i = findbypid(killed);
|
|
if (i >= 0) {
|
|
root[i].pid = 0;
|
|
if (root[i].respawn) {
|
|
circsweep();
|
|
startservice(i, time(0) - root[i].startedat < 1);
|
|
} else {
|
|
root[i].startedat = time(0);
|
|
root[i].pid = 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void childhandler(void)
|
|
{
|
|
int status;
|
|
pid_t killed;
|
|
|
|
do {
|
|
killed = waitpid(-1, &status, WNOHANG);
|
|
handlekilled(killed);
|
|
} while (killed && killed != (pid_t) - 1);
|
|
}
|
|
|
|
static volatile int dowinch = 0;
|
|
static volatile int doint = 0;
|
|
|
|
static void sigchild(int whatever)
|
|
{
|
|
}
|
|
static void sigwinch(int sig)
|
|
{
|
|
dowinch = 1;
|
|
}
|
|
static void sigint(int sig)
|
|
{
|
|
doint = 1;
|
|
}
|
|
|
|
extern int minit_main(int argc, char *argv[])
|
|
{
|
|
/* Schritt 1: argv[1] als Service nehmen und starten */
|
|
struct pollfd pfd;
|
|
time_t last = time(0);
|
|
int nfds = 1;
|
|
int count = 0;
|
|
int i;
|
|
|
|
infd = open("/etc/minit/in", O_RDWR);
|
|
outfd = open("/etc/minit/out", O_RDWR | O_NONBLOCK);
|
|
if (getpid() == 1) {
|
|
int fd;
|
|
|
|
i_am_init = 1;
|
|
reboot(0);
|
|
if ((fd = open("/dev/console", O_RDWR | O_NOCTTY))) {
|
|
ioctl(fd, KDSIGACCEPT, SIGWINCH);
|
|
close(fd);
|
|
} else
|
|
ioctl(0, KDSIGACCEPT, SIGWINCH);
|
|
}
|
|
/* signal(SIGPWR,sighandler); don't know what to do about it */
|
|
/* signal(SIGHUP,sighandler); ??? */
|
|
{
|
|
struct sigaction sa;
|
|
|
|
sigemptyset(&sa.sa_mask);
|
|
sa.sa_sigaction = 0;
|
|
sa.sa_flags = SA_RESTART | SA_NOCLDSTOP;
|
|
sa.sa_handler = sigchild;
|
|
sigaction(SIGCHLD, &sa, 0);
|
|
sa.sa_handler = sigint;
|
|
sigaction(SIGINT, &sa, 0); /* ctrl-alt-del */
|
|
sa.sa_handler = sigwinch;
|
|
sigaction(SIGWINCH, &sa, 0); /* keyboard request */
|
|
}
|
|
if (infd < 0 || outfd < 0) {
|
|
puts("minit: could not open /etc/minit/in or /etc/minit/out\n");
|
|
sulogin();
|
|
nfds = 0;
|
|
} else
|
|
pfd.fd = infd;
|
|
pfd.events = POLLIN;
|
|
|
|
for (i = 1; i < argc; i++) {
|
|
circsweep();
|
|
if (startservice(loadservice(argv[i]), 0))
|
|
count++;
|
|
}
|
|
circsweep();
|
|
if (!count)
|
|
startservice(loadservice("default"), 0);
|
|
for (;;) {
|
|
char buf[1501];
|
|
time_t now;
|
|
|
|
if (doint) {
|
|
doint = 0;
|
|
startservice(loadservice("ctrlaltdel"), 0);
|
|
}
|
|
if (dowinch) {
|
|
dowinch = 0;
|
|
startservice(loadservice("kbreq"), 0);
|
|
}
|
|
childhandler();
|
|
now = time(0);
|
|
if (now < last || now - last > 30) {
|
|
/* The system clock was reset. Compensate. */
|
|
long diff = last - now;
|
|
int j;
|
|
|
|
for (j = 0; j <= maxprocess; ++j) {
|
|
root[j].startedat -= diff;
|
|
}
|
|
}
|
|
last = now;
|
|
switch (poll(&pfd, nfds, 5000)) {
|
|
case -1:
|
|
if (errno == EINTR) {
|
|
childhandler();
|
|
break;
|
|
}
|
|
opendevconsole();
|
|
puts("poll failed!\n");
|
|
sulogin();
|
|
/* what should we do if poll fails?! */
|
|
break;
|
|
case 1:
|
|
i = read(infd, buf, 1500);
|
|
if (i > 1) {
|
|
pid_t pid;
|
|
int idx = 0;
|
|
int tmp;
|
|
|
|
buf[i] = 0;
|
|
|
|
if (buf[0] != 's' && ((idx = findservice(buf + 1)) < 0))
|
|
error:
|
|
write(outfd, "0", 1);
|
|
else {
|
|
switch (buf[0]) {
|
|
case 'p':
|
|
write(outfd, buf, fmt_ulong(buf, root[idx].pid));
|
|
break;
|
|
case 'r':
|
|
root[idx].respawn = 0;
|
|
goto ok;
|
|
case 'R':
|
|
root[idx].respawn = 1;
|
|
goto ok;
|
|
case 'C':
|
|
if (kill(root[idx].pid, 0)) { /* check if still active */
|
|
handlekilled(root[idx].pid); /* no!?! remove form active list */
|
|
goto error;
|
|
}
|
|
goto ok;
|
|
break;
|
|
case 'P':
|
|
{
|
|
unsigned char *x = buf + strlen(buf) + 1;
|
|
unsigned char c;
|
|
|
|
tmp = 0;
|
|
while ((c = *x++ - '0') < 10)
|
|
tmp = tmp * 10 + c;
|
|
}
|
|
if (tmp > 0) {
|
|
if (kill(tmp, 0))
|
|
goto error;
|
|
pid = tmp;
|
|
}
|
|
root[idx].pid = tmp;
|
|
goto ok;
|
|
case 's':
|
|
idx = loadservice(buf + 1);
|
|
if (idx < 0)
|
|
goto error;
|
|
if (root[idx].pid < 2) {
|
|
root[idx].pid = 0;
|
|
circsweep();
|
|
idx = startservice(idx, 0);
|
|
if (idx == 0) {
|
|
write(outfd, "0", 1);
|
|
break;
|
|
}
|
|
}
|
|
ok:
|
|
write(outfd, "1", 1);
|
|
break;
|
|
case 'u':
|
|
write(outfd, buf,
|
|
fmt_ulong(buf, time(0) - root[idx].startedat));
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
}
|