mirror of
https://github.com/checkpoint-restore/criu
synced 2025-08-22 09:58:09 +00:00
Makes the criu RPC socket socket-activated with systemd [1], meaning that systemd will create and listen to the UNIX socket /var/run/criu-srvice.socket on behalf of criu until a connection comes in, when it will then pass control of the socket, along with the first connection over to a newly spawned criu daemon. This is similar to inetd, but criu stays around after getting started, listening itsself on the socket. [1] http://0pointer.de/blog/projects/socket-activation.html v2: stripped down sd-daemon.[ch] moved units to scripts/sd v3: stripped down further by removing unneeded includes Signed-off-by: Shawn Landden <shawn@churchofgit.com> Signed-off-by: Pavel Emelyanov <xemul@parallels.com>
133 lines
3.3 KiB
C
133 lines
3.3 KiB
C
/* stripped down version */
|
|
/***
|
|
Copyright 2010 Lennart Poettering
|
|
|
|
Permission is hereby granted, free of charge, to any person
|
|
obtaining a copy of this software and associated documentation files
|
|
(the "Software"), to deal in the Software without restriction,
|
|
including without limitation the rights to use, copy, modify, merge,
|
|
publish, distribute, sublicense, and/or sell copies of the Software,
|
|
and to permit persons to whom the Software is furnished to do so,
|
|
subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be
|
|
included in all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
|
BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
|
ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
SOFTWARE.
|
|
***/
|
|
|
|
#ifndef _GNU_SOURCE
|
|
# define _GNU_SOURCE
|
|
#endif
|
|
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
|
|
#include "sd-daemon.h"
|
|
|
|
#if (__GNUC__ >= 4)
|
|
# ifdef SD_EXPORT_SYMBOLS
|
|
/* Export symbols */
|
|
# define _sd_export_ __attribute__ ((visibility("default")))
|
|
# else
|
|
/* Don't export the symbols */
|
|
# define _sd_export_ __attribute__ ((visibility("hidden")))
|
|
# endif
|
|
#else
|
|
# define _sd_export_
|
|
#endif
|
|
|
|
_sd_export_ int sd_listen_fds(int unset_environment) {
|
|
|
|
#if defined(DISABLE_SYSTEMD) || !defined(__linux__)
|
|
return 0;
|
|
#else
|
|
int r, fd;
|
|
const char *e;
|
|
char *p = NULL;
|
|
unsigned long l;
|
|
|
|
e = getenv("LISTEN_PID");
|
|
if (!e) {
|
|
r = 0;
|
|
goto finish;
|
|
}
|
|
|
|
errno = 0;
|
|
l = strtoul(e, &p, 10);
|
|
|
|
if (errno > 0) {
|
|
r = -errno;
|
|
goto finish;
|
|
}
|
|
|
|
if (!p || p == e || *p || l <= 0) {
|
|
r = -EINVAL;
|
|
goto finish;
|
|
}
|
|
|
|
/* Is this for us? */
|
|
if (getpid() != (pid_t) l) {
|
|
r = 0;
|
|
goto finish;
|
|
}
|
|
|
|
e = getenv("LISTEN_FDS");
|
|
if (!e) {
|
|
r = 0;
|
|
goto finish;
|
|
}
|
|
|
|
errno = 0;
|
|
l = strtoul(e, &p, 10);
|
|
|
|
if (errno > 0) {
|
|
r = -errno;
|
|
goto finish;
|
|
}
|
|
|
|
if (!p || p == e || *p) {
|
|
r = -EINVAL;
|
|
goto finish;
|
|
}
|
|
|
|
for (fd = SD_LISTEN_FDS_START; fd < SD_LISTEN_FDS_START + (int) l; fd ++) {
|
|
int flags;
|
|
|
|
flags = fcntl(fd, F_GETFD);
|
|
if (flags < 0) {
|
|
r = -errno;
|
|
goto finish;
|
|
}
|
|
|
|
if (flags & FD_CLOEXEC)
|
|
continue;
|
|
|
|
if (fcntl(fd, F_SETFD, flags | FD_CLOEXEC) < 0) {
|
|
r = -errno;
|
|
goto finish;
|
|
}
|
|
}
|
|
|
|
r = (int) l;
|
|
|
|
finish:
|
|
if (unset_environment) {
|
|
unsetenv("LISTEN_PID");
|
|
unsetenv("LISTEN_FDS");
|
|
}
|
|
|
|
return r;
|
|
#endif
|
|
}
|
|
|