yay, we can start socket units

This commit is contained in:
Lennart Poettering 2010-01-27 04:31:52 +01:00
parent b858b600e4
commit acbb02252a
16 changed files with 289 additions and 181 deletions

View File

@ -11,6 +11,7 @@
#include "strv.h" #include "strv.h"
#include "macro.h" #include "macro.h"
#include "util.h" #include "util.h"
#include "log.h"
static int close_fds(int except[], unsigned n_except) { static int close_fds(int except[], unsigned n_except) {
DIR *d; DIR *d;
@ -111,6 +112,8 @@ int exec_spawn(const ExecCommand *command, const ExecContext *context, int *fds,
assert(ret); assert(ret);
assert(fds || n_fds <= 0); assert(fds || n_fds <= 0);
log_debug("About to execute %s", command->path);
if ((pid = fork()) < 0) if ((pid = fork()) < 0)
return -errno; return -errno;

2
fixme
View File

@ -40,3 +40,5 @@
- verify fragment data after loading: refuse cycles on yourself, service units contradicting, more than one Start executable, ... - verify fragment data after loading: refuse cycles on yourself, service units contradicting, more than one Start executable, ...
- rate limit startups - rate limit startups
- automatically delete stale unix sockets

View File

@ -450,6 +450,35 @@ static int config_parse_service_restart(
return 0; return 0;
} }
int config_parse_bindtodevice(
const char *filename,
unsigned line,
const char *section,
const char *lvalue,
const char *rvalue,
void *data,
void *userdata) {
Socket *s = data;
char *n;
assert(filename);
assert(lvalue);
assert(rvalue);
assert(data);
if (rvalue[0] && !streq(rvalue, "*")) {
if (!(n = strdup(rvalue)))
return -ENOMEM;
} else
n = NULL;
free(s->bind_to_device);
s->bind_to_device = n;
return 0;
}
#define FOLLOW_MAX 8 #define FOLLOW_MAX 8
static char *build_path(const char *path, const char *filename) { static char *build_path(const char *path, const char *filename) {
@ -599,6 +628,7 @@ static int load_from_path(Unit *u, const char *path) {
{ "ListenFIFO", config_parse_listen, &u->socket, "Socket" }, { "ListenFIFO", config_parse_listen, &u->socket, "Socket" },
{ "BindIPv6Only", config_parse_socket_bind, &u->socket, "Socket" }, { "BindIPv6Only", config_parse_socket_bind, &u->socket, "Socket" },
{ "Backlog", config_parse_unsigned, &u->socket.backlog, "Socket" }, { "Backlog", config_parse_unsigned, &u->socket.backlog, "Socket" },
{ "BindToDevice", config_parse_bindtodevice, &u->socket, "Socket" },
{ "ExecStartPre", config_parse_exec, u->socket.exec_command+SOCKET_EXEC_START_PRE, "Socket" }, { "ExecStartPre", config_parse_exec, u->socket.exec_command+SOCKET_EXEC_START_PRE, "Socket" },
{ "ExecStartPost", config_parse_exec, u->socket.exec_command+SOCKET_EXEC_START_POST, "Socket" }, { "ExecStartPost", config_parse_exec, u->socket.exec_command+SOCKET_EXEC_START_POST, "Socket" },
{ "ExecStopPre", config_parse_exec, u->socket.exec_command+SOCKET_EXEC_STOP_PRE, "Socket" }, { "ExecStopPre", config_parse_exec, u->socket.exec_command+SOCKET_EXEC_STOP_PRE, "Socket" },

5
main.c
View File

@ -37,7 +37,10 @@ int main(int argc, char *argv[]) {
printf("→ By jobs:\n"); printf("→ By jobs:\n");
manager_dump_jobs(m, stdout, "\t"); manager_dump_jobs(m, stdout, "\t");
/* manager_loop(m); */ if ((r = manager_loop(m)) < 0) {
log_error("Failed to run mainloop: %s", strerror(-r));
goto finish;
}
retval = 0; retval = 0;

View File

@ -24,7 +24,7 @@ Manager* manager_new(void) {
if (!(m = new0(Manager, 1))) if (!(m = new0(Manager, 1)))
return NULL; return NULL;
m->signal_fd = m->epoll_fd = -1; m->signal_watch.fd = m->epoll_fd = -1;
if (!(m->units = hashmap_new(string_hash_func, string_compare_func))) if (!(m->units = hashmap_new(string_hash_func, string_compare_func)))
goto fail; goto fail;
@ -45,14 +45,15 @@ Manager* manager_new(void) {
assert_se(sigaddset(&mask, SIGCHLD) == 0); assert_se(sigaddset(&mask, SIGCHLD) == 0);
assert_se(sigprocmask(SIG_SETMASK, &mask, NULL) == 0); assert_se(sigprocmask(SIG_SETMASK, &mask, NULL) == 0);
if ((m->signal_fd = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC)) < 0) m->signal_watch.type = WATCH_SIGNAL_FD;
if ((m->signal_watch.fd = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC)) < 0)
goto fail; goto fail;
zero(ev); zero(ev);
ev.events = EPOLLIN; ev.events = EPOLLIN;
ev.data.fd = m->signal_fd; ev.data.ptr = &m->signal_watch;
if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, m->signal_fd, &ev) < 0) if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, m->signal_watch.fd, &ev) < 0)
goto fail; goto fail;
return m; return m;
@ -81,8 +82,8 @@ void manager_free(Manager *m) {
if (m->epoll_fd >= 0) if (m->epoll_fd >= 0)
close_nointr(m->epoll_fd); close_nointr(m->epoll_fd);
if (m->signal_fd >= 0) if (m->signal_watch.fd >= 0)
close_nointr(m->signal_fd); close_nointr(m->signal_watch.fd);
free(m); free(m);
} }
@ -555,6 +556,7 @@ static int transaction_apply(Manager *m, JobMode mode) {
assert(!j->transaction_next); assert(!j->transaction_next);
assert(!j->transaction_prev); assert(!j->transaction_prev);
job_schedule_run(j);
} }
/* As last step, kill all remaining job dependencies. */ /* As last step, kill all remaining job dependencies. */
@ -943,13 +945,20 @@ void manager_dispatch_run_queue(Manager *m) {
static int manager_dispatch_sigchld(Manager *m) { static int manager_dispatch_sigchld(Manager *m) {
assert(m); assert(m);
log_debug("dispatching SIGCHLD");
for (;;) { for (;;) {
siginfo_t si; siginfo_t si;
Unit *u; Unit *u;
zero(si); zero(si);
if (waitid(P_ALL, 0, &si, WNOHANG) < 0) if (waitid(P_ALL, 0, &si, WEXITED|WNOHANG) < 0) {
if (errno == ECHILD)
break;
return -errno; return -errno;
}
if (si.si_pid == 0) if (si.si_pid == 0)
break; break;
@ -957,6 +966,8 @@ static int manager_dispatch_sigchld(Manager *m) {
if (si.si_code != CLD_EXITED && si.si_code != CLD_KILLED && si.si_code != CLD_DUMPED) if (si.si_code != CLD_EXITED && si.si_code != CLD_KILLED && si.si_code != CLD_DUMPED)
continue; continue;
log_debug("child %llu died (code=%s, status=%i)", (long long unsigned) si.si_pid, sigchld_code(si.si_code), si.si_status);
if (!(u = hashmap_remove(m->watch_pids, UINT32_TO_PTR(si.si_pid)))) if (!(u = hashmap_remove(m->watch_pids, UINT32_TO_PTR(si.si_pid))))
continue; continue;
@ -974,13 +985,13 @@ static int manager_process_signal_fd(Manager *m) {
assert(m); assert(m);
for (;;) { for (;;) {
if ((n = read(m->signal_fd, &sfsi, sizeof(sfsi))) != sizeof(sfsi)) { if ((n = read(m->signal_watch.fd, &sfsi, sizeof(sfsi))) != sizeof(sfsi)) {
if (n >= 0) if (n >= 0)
return -EIO; return -EIO;
if (errno == EAGAIN) if (errno == EAGAIN)
return 0; break;
return -errno; return -errno;
} }
@ -997,54 +1008,51 @@ static int manager_process_signal_fd(Manager *m) {
static int process_event(Manager *m, struct epoll_event *ev) { static int process_event(Manager *m, struct epoll_event *ev) {
int r; int r;
Watch *w;
assert(m); assert(m);
assert(ev); assert(ev);
switch (ev->data.u32) { assert(w = ev->data.ptr);
case MANAGER_SIGNAL: switch (w->type) {
assert(ev->data.fd == m->signal_fd);
/* An incoming signal? */ case WATCH_SIGNAL_FD:
if (ev->events != POLLIN)
return -EINVAL;
if ((r = manager_process_signal_fd(m)) < 0) /* An incoming signal? */
return -r; if (ev->events != POLLIN)
return -EINVAL;
break; if ((r = manager_process_signal_fd(m)) < 0)
return r;
case MANAGER_FD: { break;
Unit *u;
/* Some fd event, to be dispatched to the units */ case WATCH_FD:
assert_se(u = ev->data.ptr);
UNIT_VTABLE(u)->fd_event(u, ev->data.fd, ev->events); /* Some fd event, to be dispatched to the units */
break; UNIT_VTABLE(w->unit)->fd_event(w->unit, w->fd, ev->events, w);
break;
case WATCH_TIMER: {
uint64_t v;
ssize_t k;
/* Some timer event, to be dispatched to the units */
if ((k = read(ev->data.fd, &v, sizeof(v))) != sizeof(v)) {
if (k < 0 && (errno == EINTR || errno == EAGAIN))
break;
return k < 0 ? -errno : -EIO;
} }
case MANAGER_TIMER: { UNIT_VTABLE(w->unit)->timer_event(w->unit, v, w);
Unit *u; break;
uint64_t v; }
ssize_t k;
/* Some timer event, to be dispatched to the units */ default:
if ((k = read(ev->data.fd, &v, sizeof(v))) != sizeof(v)) { assert_not_reached("Unknown epoll event type.");
if (k < 0 && (errno == EINTR || errno == EAGAIN))
break;
return k < 0 ? -errno : -EIO;
}
assert_se(u = ev->data.ptr);
UNIT_VTABLE(u)->timer_event(u, ev->data.fd, v);
break;
}
default:
assert_not_reached("Unknown epoll event type.");
} }
return 0; return 0;

View File

@ -8,7 +8,21 @@
#include <stdio.h> #include <stdio.h>
typedef struct Manager Manager; typedef struct Manager Manager;
typedef enum ManagerEventType ManagerEventType; typedef enum WatchType WatchType;
typedef struct Watch Watch;
enum WatchType {
WATCH_INVALID,
WATCH_SIGNAL_FD,
WATCH_FD,
WATCH_TIMER
};
struct Watch {
int fd;
WatchType type;
union Unit *unit;
};
#include "unit.h" #include "unit.h"
#include "job.h" #include "job.h"
@ -16,12 +30,6 @@ typedef enum ManagerEventType ManagerEventType;
#include "list.h" #include "list.h"
#include "set.h" #include "set.h"
enum ManagerEventType {
MANAGER_SIGNAL,
MANAGER_FD,
MANAGER_TIMER
};
struct Manager { struct Manager {
uint32_t current_job_id; uint32_t current_job_id;
@ -49,7 +57,8 @@ struct Manager {
Hashmap *watch_pids; /* pid => Unit object n:1 */ Hashmap *watch_pids; /* pid => Unit object n:1 */
int epoll_fd; int epoll_fd;
int signal_fd;
Watch signal_watch;
}; };
Manager* manager_new(void); Manager* manager_new(void);

View File

@ -9,7 +9,7 @@
#include "load-dropin.h" #include "load-dropin.h"
#include "log.h" #include "log.h"
static const UnitActiveState state_table[_SERVICE_STATE_MAX] = { static const UnitActiveState state_translation_table[_SERVICE_STATE_MAX] = {
[SERVICE_DEAD] = UNIT_INACTIVE, [SERVICE_DEAD] = UNIT_INACTIVE,
[SERVICE_START_PRE] = UNIT_ACTIVATING, [SERVICE_START_PRE] = UNIT_ACTIVATING,
[SERVICE_START] = UNIT_ACTIVATING, [SERVICE_START] = UNIT_ACTIVATING,
@ -26,6 +26,23 @@ static const UnitActiveState state_table[_SERVICE_STATE_MAX] = {
[SERVICE_AUTO_RESTART] = UNIT_ACTIVATING, [SERVICE_AUTO_RESTART] = UNIT_ACTIVATING,
}; };
static const char* const state_string_table[_SERVICE_STATE_MAX] = {
[SERVICE_DEAD] = "dead",
[SERVICE_START_PRE] = "start-pre",
[SERVICE_START] = "start",
[SERVICE_START_POST] = "post",
[SERVICE_RUNNING] = "running",
[SERVICE_RELOAD] = "reload",
[SERVICE_STOP] = "stop",
[SERVICE_STOP_SIGTERM] = "stop-sigterm",
[SERVICE_STOP_SIGKILL] = "stop-sigkill",
[SERVICE_STOP_POST] = "stop-post",
[SERVICE_FINAL_SIGTERM] = "final-sigterm",
[SERVICE_FINAL_SIGKILL] = "final-sigkill",
[SERVICE_MAINTAINANCE] = "maintainance",
[SERVICE_AUTO_RESTART] = "auto-restart",
};
static void service_done(Unit *u) { static void service_done(Unit *u) {
Service *s = SERVICE(u); Service *s = SERVICE(u);
@ -50,7 +67,7 @@ static void service_done(Unit *u) {
s->control_pid = 0; s->control_pid = 0;
} }
unit_unwatch_timer(u, &s->timer_id); unit_unwatch_timer(u, &s->timer_watch);
} }
static int service_load_sysv(Service *s) { static int service_load_sysv(Service *s) {
@ -79,7 +96,7 @@ static int service_init(Unit *u) {
exec_context_init(&s->exec_context); exec_context_init(&s->exec_context);
s->timer_id = -1; s->timer_watch.type = WATCH_INVALID;
s->state = SERVICE_DEAD; s->state = SERVICE_DEAD;
@ -106,23 +123,6 @@ static int service_init(Unit *u) {
static void service_dump(Unit *u, FILE *f, const char *prefix) { static void service_dump(Unit *u, FILE *f, const char *prefix) {
static const char* const state_table[_SERVICE_STATE_MAX] = {
[SERVICE_DEAD] = "dead",
[SERVICE_START_PRE] = "start-pre",
[SERVICE_START] = "start",
[SERVICE_START_POST] = "post",
[SERVICE_RUNNING] = "running",
[SERVICE_RELOAD] = "reload",
[SERVICE_STOP] = "stop",
[SERVICE_STOP_SIGTERM] = "stop-sigterm",
[SERVICE_STOP_SIGKILL] = "stop-sigkill",
[SERVICE_STOP_POST] = "stop-post",
[SERVICE_FINAL_SIGTERM] = "final-sigterm",
[SERVICE_FINAL_SIGKILL] = "final-sigkill",
[SERVICE_MAINTAINANCE] = "maintainance",
[SERVICE_AUTO_RESTART] = "auto-restart",
};
static const char* const command_table[_SERVICE_EXEC_MAX] = { static const char* const command_table[_SERVICE_EXEC_MAX] = {
[SERVICE_EXEC_START_PRE] = "ExecStartPre", [SERVICE_EXEC_START_PRE] = "ExecStartPre",
[SERVICE_EXEC_START] = "ExecStart", [SERVICE_EXEC_START] = "ExecStart",
@ -144,7 +144,7 @@ static void service_dump(Unit *u, FILE *f, const char *prefix) {
fprintf(f, fprintf(f,
"%sService State: %s\n", "%sService State: %s\n",
prefix, state_table[s->state]); prefix, state_string_table[s->state]);
if (s->pid_file) if (s->pid_file)
fprintf(f, fprintf(f,
@ -216,7 +216,7 @@ static void service_set_state(Service *s, ServiceState state) {
state != SERVICE_FINAL_SIGTERM && state != SERVICE_FINAL_SIGTERM &&
state != SERVICE_FINAL_SIGKILL && state != SERVICE_FINAL_SIGKILL &&
state != SERVICE_AUTO_RESTART) state != SERVICE_AUTO_RESTART)
unit_unwatch_timer(UNIT(s), &s->timer_id); unit_unwatch_timer(UNIT(s), &s->timer_watch);
if (state != SERVICE_START_POST && if (state != SERVICE_START_POST &&
state != SERVICE_RUNNING && state != SERVICE_RUNNING &&
@ -224,7 +224,7 @@ static void service_set_state(Service *s, ServiceState state) {
state != SERVICE_STOP && state != SERVICE_STOP &&
state != SERVICE_STOP_SIGTERM && state != SERVICE_STOP_SIGTERM &&
state != SERVICE_STOP_SIGKILL) state != SERVICE_STOP_SIGKILL)
if (s->main_pid >= 0) { if (s->main_pid > 0) {
unit_unwatch_pid(UNIT(s), s->main_pid); unit_unwatch_pid(UNIT(s), s->main_pid);
s->main_pid = 0; s->main_pid = 0;
} }
@ -239,7 +239,7 @@ static void service_set_state(Service *s, ServiceState state) {
state != SERVICE_STOP_POST && state != SERVICE_STOP_POST &&
state != SERVICE_FINAL_SIGTERM && state != SERVICE_FINAL_SIGTERM &&
state != SERVICE_FINAL_SIGKILL) state != SERVICE_FINAL_SIGKILL)
if (s->control_pid >= 0) { if (s->control_pid > 0) {
unit_unwatch_pid(UNIT(s), s->control_pid); unit_unwatch_pid(UNIT(s), s->control_pid);
s->control_pid = 0; s->control_pid = 0;
} }
@ -252,7 +252,9 @@ static void service_set_state(Service *s, ServiceState state) {
state != SERVICE_STOP_POST) state != SERVICE_STOP_POST)
s->control_command = NULL; s->control_command = NULL;
unit_notify(UNIT(s), state_table[old_state], state_table[s->state]); log_debug("%s changing %s → %s", unit_id(UNIT(s)), state_string_table[old_state], state_string_table[state]);
unit_notify(UNIT(s), state_translation_table[old_state], state_translation_table[state]);
} }
static int service_collect_fds(Service *s, int **fds, unsigned *n_fds) { static int service_collect_fds(Service *s, int **fds, unsigned *n_fds) {
@ -335,10 +337,10 @@ static int service_spawn(Service *s, ExecCommand *c, bool timeout, bool pass_fds
goto fail; goto fail;
if (timeout) { if (timeout) {
if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_id)) < 0) if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
goto fail; goto fail;
} else } else
unit_unwatch_timer(UNIT(s), &s->timer_id); unit_unwatch_timer(UNIT(s), &s->timer_watch);
if ((r = exec_spawn(c, &s->exec_context, fds, n_fds, &pid)) < 0) if ((r = exec_spawn(c, &s->exec_context, fds, n_fds, &pid)) < 0)
goto fail; goto fail;
@ -356,7 +358,7 @@ fail:
free(fds); free(fds);
if (timeout) if (timeout)
unit_unwatch_timer(UNIT(s), &s->timer_id); unit_unwatch_timer(UNIT(s), &s->timer_watch);
return r; return r;
} }
@ -372,7 +374,7 @@ static void service_enter_dead(Service *s, bool success, bool allow_restart) {
(s->restart == SERVICE_RESTART_ALWAYS || (s->restart == SERVICE_RESTART_ALWAYS ||
(s->restart == SERVICE_RESTART_ON_SUCCESS && !s->failure))) { (s->restart == SERVICE_RESTART_ON_SUCCESS && !s->failure))) {
if ((r = unit_watch_timer(UNIT(s), s->restart_usec, &s->timer_id)) < 0) if ((r = unit_watch_timer(UNIT(s), s->restart_usec, &s->timer_watch)) < 0)
goto fail; goto fail;
service_set_state(s, SERVICE_AUTO_RESTART); service_set_state(s, SERVICE_AUTO_RESTART);
@ -703,7 +705,7 @@ static bool service_can_reload(Unit *u) {
static UnitActiveState service_active_state(Unit *u) { static UnitActiveState service_active_state(Unit *u) {
assert(u); assert(u);
return state_table[SERVICE(u)->state]; return state_translation_table[SERVICE(u)->state];
} }
static int main_pid_good(Service *s) { static int main_pid_good(Service *s) {
@ -888,13 +890,13 @@ static void service_sigchld_event(Unit *u, pid_t pid, int code, int status) {
assert_not_reached("Got SIGCHLD for unkown PID"); assert_not_reached("Got SIGCHLD for unkown PID");
} }
static void service_timer_event(Unit *u, int id, uint64_t elapsed) { static void service_timer_event(Unit *u, uint64_t elapsed, Watch* w) {
Service *s = SERVICE(u); Service *s = SERVICE(u);
assert(s); assert(s);
assert(elapsed == 1); assert(elapsed == 1);
assert(s->timer_id == id); assert(w == &s->timer_watch);
switch (s->state) { switch (s->state) {

View File

@ -70,7 +70,7 @@ struct Service {
bool main_pid_known:1; bool main_pid_known:1;
bool failure:1; /* if we shut down, remember why */ bool failure:1; /* if we shut down, remember why */
int timer_id; Watch timer_watch;
}; };
const UnitVTable service_vtable; const UnitVTable service_vtable;

View File

@ -111,6 +111,11 @@ int socket_address_parse(SocketAddress *a, const char *s) {
} else { } else {
unsigned idx; unsigned idx;
if (strlen(n) > IF_NAMESIZE-1) {
free(n);
return -EINVAL;
}
/* Uh, our last resort, an interface name */ /* Uh, our last resort, an interface name */
idx = if_nametoindex(n); idx = if_nametoindex(n);
free(n); free(n);
@ -123,6 +128,7 @@ int socket_address_parse(SocketAddress *a, const char *s) {
a->sockaddr.in6.sin6_scope_id = idx; a->sockaddr.in6.sin6_scope_id = idx;
a->sockaddr.in6.sin6_addr = in6addr_any; a->sockaddr.in6.sin6_addr = in6addr_any;
a->size = sizeof(struct sockaddr_in6); a->size = sizeof(struct sockaddr_in6);
} }
} else { } else {
@ -274,8 +280,8 @@ int socket_address_print(const SocketAddress *a, char **p) {
} }
} }
int socket_address_listen(const SocketAddress *a, int backlog, SocketAddressBindIPv6Only only, int *ret) { int socket_address_listen(const SocketAddress *a, int backlog, SocketAddressBindIPv6Only only, const char *bind_to_device, int *ret) {
int r, fd; int r, fd, one;
assert(a); assert(a);
assert(ret); assert(ret);
@ -288,23 +294,30 @@ int socket_address_listen(const SocketAddress *a, int backlog, SocketAddressBind
if (socket_address_family(a) == AF_INET6 && only != SOCKET_ADDRESS_DEFAULT) { if (socket_address_family(a) == AF_INET6 && only != SOCKET_ADDRESS_DEFAULT) {
int flag = only == SOCKET_ADDRESS_IPV6_ONLY; int flag = only == SOCKET_ADDRESS_IPV6_ONLY;
if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &flag, sizeof(flag)) < 0) { if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &flag, sizeof(flag)) < 0)
close_nointr(fd); goto fail;
return -errno;
}
} }
if (bind(fd, &a->sockaddr.sa, a->size) < 0) { if (bind_to_device)
close_nointr(fd); if (setsockopt(fd, SOL_SOCKET, SO_BINDTODEVICE, bind_to_device, strlen(bind_to_device)+1) < 0)
return -errno; goto fail;
}
one = 1;
if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one)) < 0)
goto fail;
if (bind(fd, &a->sockaddr.sa, a->size) < 0)
goto fail;
if (a->type == SOCK_STREAM) if (a->type == SOCK_STREAM)
if (listen(fd, backlog) < 0) { if (listen(fd, backlog) < 0)
close_nointr(fd); goto fail;
return -errno;
}
*ret = fd; *ret = fd;
return 0; return 0;
fail:
r = -errno;
close_nointr(fd);
return r;
} }

View File

@ -6,6 +6,7 @@
#include <sys/socket.h> #include <sys/socket.h>
#include <netinet/in.h> #include <netinet/in.h>
#include <sys/un.h> #include <sys/un.h>
#include <net/if.h>
#include "macro.h" #include "macro.h"
#include "util.h" #include "util.h"
@ -38,6 +39,6 @@ typedef enum SocketAddressBindIPv6Only {
int socket_address_parse(SocketAddress *a, const char *s); int socket_address_parse(SocketAddress *a, const char *s);
int socket_address_print(const SocketAddress *a, char **p); int socket_address_print(const SocketAddress *a, char **p);
int socket_address_verify(const SocketAddress *a); int socket_address_verify(const SocketAddress *a);
int socket_address_listen(const SocketAddress *a, int backlog, SocketAddressBindIPv6Only only, int *ret); int socket_address_listen(const SocketAddress *a, int backlog, SocketAddressBindIPv6Only only, const char *bind_to_device, int *ret);
#endif #endif

View File

@ -12,7 +12,7 @@
#include "socket.h" #include "socket.h"
#include "log.h" #include "log.h"
static const UnitActiveState state_table[_SOCKET_STATE_MAX] = { static const UnitActiveState state_translation_table[_SOCKET_STATE_MAX] = {
[SOCKET_DEAD] = UNIT_INACTIVE, [SOCKET_DEAD] = UNIT_INACTIVE,
[SOCKET_START_PRE] = UNIT_ACTIVATING, [SOCKET_START_PRE] = UNIT_ACTIVATING,
[SOCKET_START_POST] = UNIT_ACTIVATING, [SOCKET_START_POST] = UNIT_ACTIVATING,
@ -27,6 +27,21 @@ static const UnitActiveState state_table[_SOCKET_STATE_MAX] = {
[SOCKET_MAINTAINANCE] = UNIT_INACTIVE, [SOCKET_MAINTAINANCE] = UNIT_INACTIVE,
}; };
static const char* const state_string_table[_SOCKET_STATE_MAX] = {
[SOCKET_DEAD] = "dead",
[SOCKET_START_PRE] = "start-pre",
[SOCKET_START_POST] = "start-post",
[SOCKET_LISTENING] = "listening",
[SOCKET_RUNNING] = "running",
[SOCKET_STOP_PRE] = "stop-pre",
[SOCKET_STOP_PRE_SIGTERM] = "stop-pre-sigterm",
[SOCKET_STOP_PRE_SIGKILL] = "stop-pre-sigkill",
[SOCKET_STOP_POST] = "stop-post",
[SOCKET_STOP_POST_SIGTERM] = "stop-post-sigterm",
[SOCKET_STOP_POST_SIGKILL] = "stop-post-sigkill",
[SOCKET_MAINTAINANCE] = "maintainance"
};
static void socket_done(Unit *u) { static void socket_done(Unit *u) {
Socket *s = SOCKET(u); Socket *s = SOCKET(u);
SocketPort *p; SocketPort *p;
@ -53,7 +68,9 @@ static void socket_done(Unit *u) {
s->service = NULL; s->service = NULL;
unit_unwatch_timer(u, &s->timer_id); free(s->bind_to_device);
unit_unwatch_timer(u, &s->timer_watch);
} }
static int socket_init(Unit *u) { static int socket_init(Unit *u) {
@ -65,7 +82,7 @@ static int socket_init(Unit *u) {
* reload */ * reload */
s->state = 0; s->state = 0;
s->timer_id = -1; s->timer_watch.type = WATCH_INVALID;
s->bind_ipv6_only = false; s->bind_ipv6_only = false;
s->backlog = SOMAXCONN; s->backlog = SOMAXCONN;
s->timeout_usec = DEFAULT_TIMEOUT_USEC; s->timeout_usec = DEFAULT_TIMEOUT_USEC;
@ -110,21 +127,6 @@ static const char* listen_lookup(int type) {
static void socket_dump(Unit *u, FILE *f, const char *prefix) { static void socket_dump(Unit *u, FILE *f, const char *prefix) {
static const char* const state_table[_SOCKET_STATE_MAX] = {
[SOCKET_DEAD] = "dead",
[SOCKET_START_PRE] = "start-pre",
[SOCKET_START_POST] = "start-post",
[SOCKET_LISTENING] = "listening",
[SOCKET_RUNNING] = "running",
[SOCKET_STOP_PRE] = "stop-pre",
[SOCKET_STOP_PRE_SIGTERM] = "stop-pre-sigterm",
[SOCKET_STOP_PRE_SIGKILL] = "stop-pre-sigkill",
[SOCKET_STOP_POST] = "stop-post",
[SOCKET_STOP_POST_SIGTERM] = "stop-post-sigterm",
[SOCKET_STOP_POST_SIGKILL] = "stop-post-sigkill",
[SOCKET_MAINTAINANCE] = "maintainance"
};
static const char* const command_table[_SOCKET_EXEC_MAX] = { static const char* const command_table[_SOCKET_EXEC_MAX] = {
[SOCKET_EXEC_START_PRE] = "StartPre", [SOCKET_EXEC_START_PRE] = "StartPre",
[SOCKET_EXEC_START_POST] = "StartPost", [SOCKET_EXEC_START_POST] = "StartPost",
@ -147,10 +149,15 @@ static void socket_dump(Unit *u, FILE *f, const char *prefix) {
"%sSocket State: %s\n" "%sSocket State: %s\n"
"%sBindIPv6Only: %s\n" "%sBindIPv6Only: %s\n"
"%sBacklog: %u\n", "%sBacklog: %u\n",
prefix, state_table[s->state], prefix, state_string_table[s->state],
prefix, yes_no(s->bind_ipv6_only), prefix, yes_no(s->bind_ipv6_only),
prefix, s->backlog); prefix, s->backlog);
if (s->bind_to_device)
fprintf(f,
"%sBindToDevice: %s\n",
prefix, s->bind_to_device);
LIST_FOREACH(port, p, s->ports) { LIST_FOREACH(port, p, s->ports) {
if (p->type == SOCKET_SOCKET) { if (p->type == SOCKET_SOCKET) {
@ -193,7 +200,7 @@ static void socket_close_fds(Socket *s) {
if (p->fd < 0) if (p->fd < 0)
continue; continue;
unit_unwatch_fd(UNIT(s), p->fd); unit_unwatch_fd(UNIT(s), &p->fd_watch);
assert_se(close_nointr(p->fd) >= 0); assert_se(close_nointr(p->fd) >= 0);
p->fd = -1; p->fd = -1;
@ -213,7 +220,7 @@ static int socket_open_fds(Socket *s) {
if (p->type == SOCKET_SOCKET) { if (p->type == SOCKET_SOCKET) {
if ((r = socket_address_listen(&p->address, s->backlog, s->bind_ipv6_only, &p->fd)) < 0) if ((r = socket_address_listen(&p->address, s->backlog, s->bind_ipv6_only, s->bind_to_device, &p->fd)) < 0)
goto rollback; goto rollback;
} else { } else {
@ -260,7 +267,7 @@ static void socket_unwatch_fds(Socket *s) {
if (p->fd < 0) if (p->fd < 0)
continue; continue;
unit_unwatch_fd(UNIT(s), p->fd); unit_unwatch_fd(UNIT(s), &p->fd_watch);
} }
} }
@ -274,7 +281,7 @@ static int socket_watch_fds(Socket *s) {
if (p->fd < 0) if (p->fd < 0)
continue; continue;
if ((r = unit_watch_fd(UNIT(s), p->fd, POLLIN)) < 0) if ((r = unit_watch_fd(UNIT(s), p->fd, POLLIN, &p->fd_watch)) < 0)
goto fail; goto fail;
} }
@ -300,7 +307,7 @@ static void socket_set_state(Socket *s, SocketState state) {
state != SOCKET_STOP_POST && state != SOCKET_STOP_POST &&
state != SOCKET_STOP_POST_SIGTERM && state != SOCKET_STOP_POST_SIGTERM &&
state != SOCKET_STOP_POST_SIGKILL) state != SOCKET_STOP_POST_SIGKILL)
unit_unwatch_timer(UNIT(s), &s->timer_id); unit_unwatch_timer(UNIT(s), &s->timer_watch);
if (state != SOCKET_START_PRE && if (state != SOCKET_START_PRE &&
state != SOCKET_START_POST && state != SOCKET_START_POST &&
@ -310,7 +317,7 @@ static void socket_set_state(Socket *s, SocketState state) {
state != SOCKET_STOP_POST && state != SOCKET_STOP_POST &&
state != SOCKET_STOP_POST_SIGTERM && state != SOCKET_STOP_POST_SIGTERM &&
state != SOCKET_STOP_POST_SIGKILL) state != SOCKET_STOP_POST_SIGKILL)
if (s->control_pid >= 0) { if (s->control_pid > 0) {
unit_unwatch_pid(UNIT(s), s->control_pid); unit_unwatch_pid(UNIT(s), s->control_pid);
s->control_pid = 0; s->control_pid = 0;
} }
@ -332,7 +339,9 @@ static void socket_set_state(Socket *s, SocketState state) {
if (state != SOCKET_LISTENING) if (state != SOCKET_LISTENING)
socket_unwatch_fds(s); socket_unwatch_fds(s);
unit_notify(UNIT(s), state_table[old_state], state_table[s->state]); log_debug("%s changing %s → %s", unit_id(UNIT(s)), state_string_table[old_state], state_string_table[state]);
unit_notify(UNIT(s), state_translation_table[old_state], state_translation_table[state]);
} }
static int socket_spawn(Socket *s, ExecCommand *c, bool timeout, pid_t *_pid) { static int socket_spawn(Socket *s, ExecCommand *c, bool timeout, pid_t *_pid) {
@ -344,10 +353,10 @@ static int socket_spawn(Socket *s, ExecCommand *c, bool timeout, pid_t *_pid) {
assert(_pid); assert(_pid);
if (timeout) { if (timeout) {
if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_id)) < 0) if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
goto fail; goto fail;
} else } else
unit_unwatch_timer(UNIT(s), &s->timer_id); unit_unwatch_timer(UNIT(s), &s->timer_watch);
if ((r = exec_spawn(c, &s->exec_context, NULL, 0, &pid)) < 0) if ((r = exec_spawn(c, &s->exec_context, NULL, 0, &pid)) < 0)
goto fail; goto fail;
@ -362,7 +371,7 @@ static int socket_spawn(Socket *s, ExecCommand *c, bool timeout, pid_t *_pid) {
fail: fail:
if (timeout) if (timeout)
unit_unwatch_timer(UNIT(s), &s->timer_id); unit_unwatch_timer(UNIT(s), &s->timer_watch);
return r; return r;
} }
@ -594,10 +603,10 @@ static int socket_stop(Unit *u) {
static UnitActiveState socket_active_state(Unit *u) { static UnitActiveState socket_active_state(Unit *u) {
assert(u); assert(u);
return state_table[SOCKET(u)->state]; return state_translation_table[SOCKET(u)->state];
} }
static void socket_fd_event(Unit *u, int fd, uint32_t events) { static void socket_fd_event(Unit *u, int fd, uint32_t events, Watch *w) {
Socket *s = SOCKET(u); Socket *s = SOCKET(u);
assert(s); assert(s);
@ -626,20 +635,23 @@ static void socket_sigchld_event(Unit *u, pid_t pid, int code, int status) {
exec_status_fill(&s->control_command->exec_status, pid, code, status); exec_status_fill(&s->control_command->exec_status, pid, code, status);
s->control_pid = 0; s->control_pid = 0;
log_debug("%s: control process exited, code=%s status=%i", unit_id(u), sigchld_code(code), status); log_debug("%s control process exited, code=%s status=%i", unit_id(u), sigchld_code(code), status);
if (s->control_command->command_next && if (s->control_command->command_next &&
(success || (s->state == SOCKET_EXEC_STOP_PRE || s->state == SOCKET_EXEC_STOP_POST))) (success || (s->state == SOCKET_EXEC_STOP_PRE || s->state == SOCKET_EXEC_STOP_POST))) {
log_debug("%s running next command for the state %s", unit_id(u), state_string_table[s->state]);
socket_run_next(s, success); socket_run_next(s, success);
else { } else {
/* No further commands for this step, so let's figure /* No further commands for this step, so let's figure
* out what to do next */ * out what to do next */
log_debug("%s finished with state %s", unit_id(u), state_string_table[s->state]);
switch (s->state) { switch (s->state) {
case SOCKET_START_PRE: case SOCKET_START_PRE:
if (success) if (success)
socket_enter_start_pre(s); socket_enter_start_post(s);
else else
socket_enter_stop_pre(s, false); socket_enter_stop_pre(s, false);
break; break;
@ -669,13 +681,13 @@ static void socket_sigchld_event(Unit *u, pid_t pid, int code, int status) {
} }
} }
static void socket_timer_event(Unit *u, int id, uint64_t elapsed) { static void socket_timer_event(Unit *u, uint64_t elapsed, Watch *w) {
Socket *s = SOCKET(u); Socket *s = SOCKET(u);
assert(s); assert(s);
assert(elapsed == 1); assert(elapsed == 1);
assert(s->timer_id == id); assert(w == &s->timer_watch);
switch (s->state) { switch (s->state) {

View File

@ -5,6 +5,7 @@
typedef struct Socket Socket; typedef struct Socket Socket;
#include "manager.h"
#include "unit.h" #include "unit.h"
#include "socket-util.h" #include "socket-util.h"
@ -46,6 +47,7 @@ struct SocketPort {
char *path; char *path;
int fd; int fd;
Watch fd_watch;
LIST_FIELDS(SocketPort, port); LIST_FIELDS(SocketPort, port);
}; };
@ -71,8 +73,10 @@ struct Socket {
ExecCommand* control_command; ExecCommand* control_command;
pid_t control_pid; pid_t control_pid;
char *bind_to_device;
bool failure; bool failure;
int timer_id; Watch timer_watch;
}; };
/* Called from the service code when collecting fds */ /* Called from the service code when collecting fds */

View File

@ -10,3 +10,4 @@ ExecStartPost=/bin/echo "Created sockets..."
ExecStopPre=/bin/echo "About to delete sockets..." ExecStopPre=/bin/echo "About to delete sockets..."
ExecStopPost=/bin/echo "Deleted sockets..." ExecStopPost=/bin/echo "Deleted sockets..."
ExecStopPost=/bin/echo "Deleted sockets... 2" ExecStopPost=/bin/echo "Deleted sockets... 2"
#BindToDevice=eth0

View File

@ -4,3 +4,5 @@ Description=Syslog Socket
[Socket] [Socket]
ListenDatagram=/tmp/systemd-syslog-socket ListenDatagram=/tmp/systemd-syslog-socket
ListenStream=eth0:3456 ListenStream=eth0:3456
ExecStartPre=/bin/rm -f /tmp/systemd-syslog-socket
ExecStopPost=/bin/rm -f /tmp/systemd-syslog-socket

90
unit.c
View File

@ -627,33 +627,44 @@ void unit_notify(Unit *u, UnitActiveState os, UnitActiveState ns) {
retroactively_stop_dependencies(u); retroactively_stop_dependencies(u);
} }
int unit_watch_fd(Unit *u, int fd, uint32_t events) { int unit_watch_fd(Unit *u, int fd, uint32_t events, Watch *w) {
struct epoll_event ev; struct epoll_event ev;
assert(u); assert(u);
assert(fd >= 0); assert(fd >= 0);
assert(w);
assert(w->type == WATCH_INVALID || (w->type == WATCH_FD && w->fd == fd && w->unit == u));
zero(ev); zero(ev);
ev.data.fd = fd; ev.data.ptr = w;
ev.data.ptr = u;
ev.data.u32 = MANAGER_FD;
ev.events = events; ev.events = events;
if (epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_ADD, fd, &ev) >= 0) if (epoll_ctl(u->meta.manager->epoll_fd,
return 0; w->type == WATCH_INVALID ? EPOLL_CTL_ADD : EPOLL_CTL_MOD,
fd,
&ev) < 0)
return -errno;
if (errno == EEXIST) w->fd = fd;
if (epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_MOD, fd, &ev) >= 0) w->type = WATCH_FD;
return 0; w->unit = u;
return -errno; return 0;
} }
void unit_unwatch_fd(Unit *u, int fd) { void unit_unwatch_fd(Unit *u, Watch *w) {
assert(u); assert(u);
assert(fd >= 0); assert(w);
assert_se(epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_DEL, fd, NULL) >= 0 || errno == ENOENT); if (w->type == WATCH_INVALID)
return;
assert(w->type == WATCH_FD && w->unit == u);
assert_se(epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_DEL, w->fd, NULL) >= 0);
w->fd = -1;
w->type = WATCH_INVALID;
w->unit = NULL;
} }
int unit_watch_pid(Unit *u, pid_t pid) { int unit_watch_pid(Unit *u, pid_t pid) {
@ -670,25 +681,22 @@ void unit_unwatch_pid(Unit *u, pid_t pid) {
hashmap_remove(u->meta.manager->watch_pids, UINT32_TO_PTR(pid)); hashmap_remove(u->meta.manager->watch_pids, UINT32_TO_PTR(pid));
} }
int unit_watch_timer(Unit *u, usec_t delay, int *id) { int unit_watch_timer(Unit *u, usec_t delay, Watch *w) {
struct epoll_event ev;
int fd;
struct itimerspec its; struct itimerspec its;
int flags; int flags, fd;
bool ours; bool ours;
assert(u); assert(u);
assert(id); assert(w);
assert(w->type == WATCH_INVALID || (w->type == WATCH_TIMER && w->unit == u));
/* This will try to reuse the old timer if there is one */ /* This will try to reuse the old timer if there is one */
if (*id >= 0) { if (w->type == WATCH_TIMER) {
ours = false; ours = false;
fd = *id; fd = w->fd;
} else { } else {
ours = true; ours = true;
if ((fd = timerfd_create(CLOCK_MONOTONIC, TFD_NONBLOCK|TFD_CLOEXEC)) < 0) if ((fd = timerfd_create(CLOCK_MONOTONIC, TFD_NONBLOCK|TFD_CLOEXEC)) < 0)
return -errno; return -errno;
} }
@ -711,16 +719,21 @@ int unit_watch_timer(Unit *u, usec_t delay, int *id) {
if (timerfd_settime(fd, flags, &its, NULL) < 0) if (timerfd_settime(fd, flags, &its, NULL) < 0)
goto fail; goto fail;
zero(ev); if (w->type == WATCH_INVALID) {
ev.data.fd = fd; struct epoll_event ev;
ev.data.ptr = u;
ev.data.u32 = MANAGER_TIMER;
ev.events = POLLIN;
if (epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_ADD, fd, &ev) < 0) zero(ev);
goto fail; ev.data.ptr = w;
ev.events = POLLIN;
if (epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_ADD, fd, &ev) < 0)
goto fail;
}
w->fd = fd;
w->type = WATCH_TIMER;
w->unit = u;
*id = fd;
return 0; return 0;
fail: fail:
@ -730,16 +743,21 @@ fail:
return -errno; return -errno;
} }
void unit_unwatch_timer(Unit *u, int *id) { void unit_unwatch_timer(Unit *u, Watch *w) {
assert(u); assert(u);
assert(id); assert(w);
if (*id < 0) if (w->type == WATCH_INVALID)
return; return;
assert_se(epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_DEL, *id, NULL) >= 0); assert(w->type == WATCH_TIMER && w->unit == u);
assert_se(close_nointr(*id) == 0);
*id = -1; assert_se(epoll_ctl(u->meta.manager->epoll_fd, EPOLL_CTL_DEL, w->fd, NULL) >= 0);
assert_se(close_nointr(w->fd) == 0);
w->fd = -1;
w->type = WATCH_INVALID;
w->unit = NULL;
} }
bool unit_job_is_applicable(Unit *u, JobType j) { bool unit_job_is_applicable(Unit *u, JobType j) {

12
unit.h
View File

@ -157,9 +157,9 @@ struct UnitVTable {
* a simpler one that the engine can understand */ * a simpler one that the engine can understand */
UnitActiveState (*active_state)(Unit *u); UnitActiveState (*active_state)(Unit *u);
void (*fd_event)(Unit *u, int fd, uint32_t events); void (*fd_event)(Unit *u, int fd, uint32_t events, Watch *w);
void (*sigchld_event)(Unit *u, pid_t pid, int code, int status); void (*sigchld_event)(Unit *u, pid_t pid, int code, int status);
void (*timer_event)(Unit *u, int id, uint64_t n_elapsed); void (*timer_event)(Unit *u, uint64_t n_elapsed, Watch *w);
void (*retry)(Unit *u); void (*retry)(Unit *u);
}; };
@ -222,14 +222,14 @@ int unit_reload(Unit *u);
void unit_notify(Unit *u, UnitActiveState os, UnitActiveState ns); void unit_notify(Unit *u, UnitActiveState os, UnitActiveState ns);
int unit_watch_fd(Unit *u, int fd, uint32_t events); int unit_watch_fd(Unit *u, int fd, uint32_t events, Watch *w);
void unit_unwatch_fd(Unit *u, int fd); void unit_unwatch_fd(Unit *u, Watch *w);
int unit_watch_pid(Unit *u, pid_t pid); int unit_watch_pid(Unit *u, pid_t pid);
void unit_unwatch_pid(Unit *u, pid_t pid); void unit_unwatch_pid(Unit *u, pid_t pid);
int unit_watch_timer(Unit *u, usec_t delay, int *id); int unit_watch_timer(Unit *u, usec_t delay, Watch *w);
void unit_unwatch_timer(Unit *u, int *id); void unit_unwatch_timer(Unit *u, Watch *w);
bool unit_job_is_applicable(Unit *u, JobType j); bool unit_job_is_applicable(Unit *u, JobType j);