From ba1270008021470b2e451bce8e75053e0a00f662 Mon Sep 17 00:00:00 2001 From: David Oberhollenzer Date: Thu, 28 Mar 2019 14:15:39 +0100 Subject: Add start/stop commands to init socket Signed-off-by: David Oberhollenzer --- initd/init.h | 5 +++++ initd/main.c | 8 +++++++ initd/supervisor.c | 42 +++++++++++++++++++++++++++++++++++++ lib/include/initsock.h | 6 ++++++ lib/include/service.h | 1 + lib/init/init_socket_send_request.c | 5 +++++ 6 files changed, 67 insertions(+) diff --git a/initd/init.h b/initd/init.h index 1d292d3..b98da98 100644 --- a/initd/init.h +++ b/initd/init.h @@ -9,6 +9,7 @@ #include #include #include +#include #include #include #include @@ -70,6 +71,10 @@ bool supervisor_process_queues(void); void supervisor_answer_status_request(int fd, const void *dest_addr, size_t addrlen, E_SERVICE_STATE filter); +void supervisor_start(int id); + +void supervisor_stop(int id); + /********** signal_.c **********/ /* diff --git a/initd/main.c b/initd/main.c index 4eb4832..2b1a1cc 100644 --- a/initd/main.c +++ b/initd/main.c @@ -64,6 +64,14 @@ retry: supervisor_answer_status_request(sockfd, &addr, addrlen, rq.arg.status.filter); break; + case EIR_START: + rq.arg.startstop.id = be32toh(rq.arg.startstop.id); + supervisor_start(rq.arg.startstop.id); + break; + case EIR_STOP: + rq.arg.startstop.id = be32toh(rq.arg.startstop.id); + supervisor_stop(rq.arg.startstop.id); + break; } } diff --git a/initd/supervisor.c b/initd/supervisor.c index 871618a..53a2aec 100644 --- a/initd/supervisor.c +++ b/initd/supervisor.c @@ -38,6 +38,9 @@ static void handle_terminated_service(service_t *svc) if (target == TGT_REBOOT || target == TGT_SHUTDOWN) break; + if (svc->flags & SVC_FLAG_ADMIN_STOPPED) + break; + if (svc->rspwn_limit > 0) { svc->rspwn_limit -= 1; @@ -221,3 +224,42 @@ void supervisor_answer_status_request(int fd, const void *dst, size_t addrlen, return; init_socket_send_status(fd, dst, addrlen, ESS_NONE, NULL); } + +void supervisor_start(int id) +{ + service_t *svc; + + for (svc = completed; svc != NULL; svc = svc->next) { + if (svc->id == id) + goto found; + } + + for (svc = failed; svc != NULL; svc = svc->next) { + if (svc->id == id) + goto found; + } + return; +found: + if (svc->type == SVC_RESPAWN) + svc->rspwn_limit = 0; + + svc->flags &= ~SVC_FLAG_ADMIN_STOPPED; + svc->next = queue; + queue = svc; +} + +void supervisor_stop(int id) +{ + service_t *svc; + + for (svc = running; svc != NULL; svc = svc->next) { + if (svc->id == id) + break; + } + + if (svc != NULL) { + /* TODO: something more sophisticated? */ + svc->flags |= SVC_FLAG_ADMIN_STOPPED; + kill(svc->pid, SIGTERM); + } +} diff --git a/lib/include/initsock.h b/lib/include/initsock.h index 10abe08..a687481 100644 --- a/lib/include/initsock.h +++ b/lib/include/initsock.h @@ -11,6 +11,8 @@ typedef enum { EIR_STATUS = 0x00, + EIR_START = 0x01, + EIR_STOP = 0x02, } E_INIT_REQUEST; typedef enum { @@ -30,6 +32,10 @@ typedef struct { uint8_t filter; uint8_t padd[3]; } status; + + struct { + uint32_t id; + } startstop; } arg; } init_request_t; diff --git a/lib/include/service.h b/lib/include/service.h index 1ff5204..4e8f0b3 100644 --- a/lib/include/service.h +++ b/lib/include/service.h @@ -43,6 +43,7 @@ enum { SVC_FLAG_TRUNCATE_OUT = 0x01, SVC_FLAG_HAS_EXEC = 0x10, + SVC_FLAG_ADMIN_STOPPED = 0x20, }; typedef struct service_t { diff --git a/lib/init/init_socket_send_request.c b/lib/init/init_socket_send_request.c index e9c7cb9..53b31f6 100644 --- a/lib/init/init_socket_send_request.c +++ b/lib/init/init_socket_send_request.c @@ -2,6 +2,7 @@ #include #include #include +#include #include #include @@ -21,6 +22,10 @@ int init_socket_send_request(int fd, E_INIT_REQUEST rq, ...) case EIR_STATUS: request.arg.status.filter = va_arg(ap, E_SERVICE_STATE); break; + case EIR_START: + case EIR_STOP: + request.arg.startstop.id = htobe32(va_arg(ap, int)); + break; default: break; } -- cgit v1.2.3