X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Fcore%2Fsocket.c;h=7eac0eb66de22a7489dd76dc9a05fefde774581d;hb=cb8ccb2271727fc114ca43104d3333ee4635cc79;hp=2442221355356a1e4f485d4485dcf93b0a25cb3f;hpb=aec8de63b14a93b91b85dc15bf879604352fbbe1;p=elogind.git diff --git a/src/core/socket.c b/src/core/socket.c index 244222135..7eac0eb66 100644 --- a/src/core/socket.c +++ b/src/core/socket.c @@ -179,7 +179,8 @@ static int socket_arm_timer(Socket *s) { } static int socket_instantiate_service(Socket *s) { - char *prefix, *name; + _cleanup_free_ char *prefix = NULL; + _cleanup_free_ char *name = NULL; int r; Unit *u; @@ -199,13 +200,11 @@ static int socket_instantiate_service(Socket *s) { return -ENOMEM; r = asprintf(&name, "%s@%u.service", prefix, s->n_accepted); - free(prefix); if (r < 0) return -ENOMEM; r = manager_load_unit(UNIT(s)->manager, name, NULL, NULL, &u); - free(name); if (r < 0) return r; @@ -671,10 +670,11 @@ static int instance_from_socket(int fd, unsigned nr, char **instance) { case AF_UNIX: { struct ucred ucred; + int k; - l = sizeof(ucred); - if (getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &ucred, &l) < 0) - return -errno; + k = getpeercred(fd, &ucred); + if (k < 0) + return k; if (asprintf(&r, "%u-%lu-%lu", @@ -1254,6 +1254,7 @@ static int socket_spawn(Socket *s, ExecCommand *c, pid_t *_pid) { UNIT(s)->manager->cgroup_supported, UNIT(s)->cgroup_path, UNIT(s)->id, + 0, NULL, s->exec_runtime, &pid); @@ -2343,6 +2344,20 @@ static int socket_kill(Unit *u, KillWho who, int signo, sd_bus_error *error) { return unit_kill_common(u, who, signo, -1, SOCKET(u)->control_pid, error); } +static int socket_get_timeout(Unit *u, uint64_t *timeout) { + Socket *s = SOCKET(u); + int r; + + if (!s->timer_event_source) + return 0; + + r = sd_event_source_get_time(s->timer_event_source, timeout); + if (r < 0) + return r; + + return 1; +} + static const char* const socket_state_table[_SOCKET_STATE_MAX] = { [SOCKET_DEAD] = "dead", [SOCKET_START_PRE] = "start-pre", @@ -2407,6 +2422,8 @@ const UnitVTable socket_vtable = { .kill = socket_kill, + .get_timeout = socket_get_timeout, + .serialize = socket_serialize, .deserialize_item = socket_deserialize_item, .distribute_fds = socket_distribute_fds,