2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <errno.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <limits.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <mqueue.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <netinet/in.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <stdarg.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <stddef.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <stdio.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <stdlib.h>

73f860db9893deab6aebceb53dd7d0deb662e832Zbigniew Jędrzejewski-Szmek#include <string.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <sys/socket.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <sys/stat.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <sys/un.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include <unistd.h>

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include "path-util.h"

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include "socket-util.h"

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include "strv.h"

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek#include "util.h"

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek#include "sd-daemon.h"

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmekstatic void unsetenv_all(bool unset_environment) {

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering if (!unset_environment)

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering return;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek unsetenv("LISTEN_PID");

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek unsetenv("LISTEN_FDS");

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek unsetenv("LISTEN_FDNAMES");

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek}

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek_public_ int sd_listen_fds(int unset_environment) {

603938e0a9b5379984d7397e3cf81683c0037a53Lennart Poettering const char *e;

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt unsigned n;

603938e0a9b5379984d7397e3cf81683c0037a53Lennart Poettering int r, fd;

603938e0a9b5379984d7397e3cf81683c0037a53Lennart Poettering pid_t pid;

603938e0a9b5379984d7397e3cf81683c0037a53Lennart Poettering

603938e0a9b5379984d7397e3cf81683c0037a53Lennart Poettering e = getenv("LISTEN_PID");

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (!e) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek r = 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek goto finish;

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering }

29a5ca9baa58e55c4d9e1d008cdd014aa9c3c3e1Lennart Poettering

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek r = parse_pid(e, &pid);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (r < 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek goto finish;

eb56eb9b40950f1edcffdb7313f8de4f8572a6d5Michal Schmidt

eb56eb9b40950f1edcffdb7313f8de4f8572a6d5Michal Schmidt /* Is this for us? */

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering if (getpid() != pid) {

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering r = 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek goto finish;

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering }

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering e = getenv("LISTEN_FDS");

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering if (!e) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek r = 0;

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering goto finish;

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek r = safe_atou(e, &n);

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek if (r < 0)

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek goto finish;

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek for (fd = SD_LISTEN_FDS_START; fd < SD_LISTEN_FDS_START + (int) n; fd ++) {

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek r = fd_cloexec(fd, true);

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek if (r < 0)

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek goto finish;

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek }

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek r = (int) n;

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek

fff40a51ccbb02e8dec4ff2ee505bc84f75e445cZbigniew Jędrzejewski-Szmekfinish:

fff40a51ccbb02e8dec4ff2ee505bc84f75e445cZbigniew Jędrzejewski-Szmek unsetenv_all(unset_environment);

fff40a51ccbb02e8dec4ff2ee505bc84f75e445cZbigniew Jędrzejewski-Szmek return r;

fff40a51ccbb02e8dec4ff2ee505bc84f75e445cZbigniew Jędrzejewski-Szmek}

fff40a51ccbb02e8dec4ff2ee505bc84f75e445cZbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek_public_ int sd_listen_fds_with_names(int unset_environment, char ***names) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek _cleanup_strv_free_ char **l = NULL;

e0aa3726103448097e5ad7cc6f427e142103a321Zbigniew Jędrzejewski-Szmek bool have_names;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek int n_names = 0, n_fds;

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek const char *e;

23bbb0de4e3f85d9704a5c12a5afa2dfa0159e41Michal Schmidt int r;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (!names)

175a3d25d0e8596d4ba0759aea3f89ee228e7d6dLennart Poettering return sd_listen_fds(unset_environment);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek e = getenv("LISTEN_FDNAMES");

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (e) {

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek n_names = strv_split_extract(&l, e, ":", EXTRACT_DONT_COALESCE_SEPARATORS);

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek if (n_names < 0) {

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek unsetenv_all(unset_environment);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return n_names;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek have_names = true;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek } else

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek have_names = false;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering n_fds = sd_listen_fds(unset_environment);

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering if (n_fds <= 0)

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering return n_fds;

2c408cb6276e3b8d18fb4e2a81a1128d8bbaa70dLennart Poettering

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering if (have_names) {

29a5ca9baa58e55c4d9e1d008cdd014aa9c3c3e1Lennart Poettering if (n_names != n_fds)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -EINVAL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek } else {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek r = strv_extend_n(&l, "unknown", n_fds);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (r < 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return r;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

23ea3dab15181a3966ec45fa2ccbb81e59d2e0a7Zbigniew Jędrzejewski-Szmek *names = l;

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering l = NULL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering return n_fds;

7fd1b19bc9e9f5574f2877936b8ac267c7706947Harald Hoyer}

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering_public_ int sd_is_fifo(int fd, const char *path) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct stat st_fd;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek assert_return(fd >= 0, -EBADF);

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek if (fstat(fd, &st_fd) < 0)

5b84559a76f30755019d6a0e6b2ccc9bbe0b3c56Lennart Poettering return -errno;

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering if (!S_ISFIFO(st_fd.st_mode))

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek return 0;

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek if (path) {

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek struct stat st_path;

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek

7fd1b19bc9e9f5574f2877936b8ac267c7706947Harald Hoyer if (stat(path, &st_path) < 0) {

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek if (errno == ENOENT || errno == ENOTDIR)

23ea3dab15181a3966ec45fa2ccbb81e59d2e0a7Zbigniew Jędrzejewski-Szmek return 0;

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek return -errno;

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek }

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek return

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek st_path.st_dev == st_fd.st_dev &&

23ea3dab15181a3966ec45fa2ccbb81e59d2e0a7Zbigniew Jędrzejewski-Szmek st_path.st_ino == st_fd.st_ino;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 1;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek}

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek_public_ int sd_is_special(int fd, const char *path) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct stat st_fd;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek assert_return(fd >= 0, -EBADF);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (fstat(fd, &st_fd) < 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (!S_ISREG(st_fd.st_mode) && !S_ISCHR(st_fd.st_mode))

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt return 0;

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (path) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct stat st_path;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

23ea3dab15181a3966ec45fa2ccbb81e59d2e0a7Zbigniew Jędrzejewski-Szmek if (stat(path, &st_path) < 0) {

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (errno == ENOENT || errno == ENOTDIR)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (S_ISREG(st_fd.st_mode) && S_ISREG(st_path.st_mode))

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek st_path.st_dev == st_fd.st_dev &&

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek st_path.st_ino == st_fd.st_ino;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek else if (S_ISCHR(st_fd.st_mode) && S_ISCHR(st_path.st_mode))

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt return st_path.st_rdev == st_fd.st_rdev;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek else

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 1;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek}

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidtstatic int sd_is_socket_internal(int fd, int type, int listening) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct stat st_fd;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek assert_return(fd >= 0, -EBADF);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek assert_return(type >= 0, -EINVAL);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt if (fstat(fd, &st_fd) < 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (!S_ISSOCK(st_fd.st_mode))

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt if (type != 0) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek int other_type = 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek socklen_t l = sizeof(other_type);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (getsockopt(fd, SOL_SOCKET, SO_TYPE, &other_type, &l) < 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (l != sizeof(other_type))

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -EINVAL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (other_type != type)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt if (listening >= 0) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek int accepting = 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek socklen_t l = sizeof(accepting);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (getsockopt(fd, SOL_SOCKET, SO_ACCEPTCONN, &accepting, &l) < 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (l != sizeof(accepting))

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -EINVAL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering if (!accepting != !listening)

aa44499da15a8fa7026463555a7a27e55e4e24a8Philippe De Swert return 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 1;

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt}

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek_public_ int sd_is_socket(int fd, int family, int type, int listening) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek int r;

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering assert_return(fd >= 0, -EBADF);

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering assert_return(family >= 0, -EINVAL);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

30374ebe5e9f0b37e99dcbdc965c00fcf542f89dLennart Poettering r = sd_is_socket_internal(fd, type, listening);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (r <= 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return r;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

e9c1ea9de87d4d508ac38ce87a2fa56e7529a91aJason St. John if (family > 0) {

9d458c0902cdc5d993fd2f78d36fb83a8130c81bLennart Poettering union sockaddr_union sockaddr = {};

9d458c0902cdc5d993fd2f78d36fb83a8130c81bLennart Poettering socklen_t l = sizeof(sockaddr);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

e9c1ea9de87d4d508ac38ce87a2fa56e7529a91aJason St. John if (getsockname(fd, &sockaddr.sa, &l) < 0)

e9c1ea9de87d4d508ac38ce87a2fa56e7529a91aJason St. John return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (l < sizeof(sa_family_t))

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -EINVAL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek return sockaddr.sa.sa_family == family;

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek }

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek

c099716487df4a4f5394e57e7ca14da1d358166aZbigniew Jędrzejewski-Szmek return 1;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek}

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek_public_ int sd_is_socket_inet(int fd, int family, int type, int listening, uint16_t port) {

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt union sockaddr_union sockaddr = {};

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek socklen_t l = sizeof(sockaddr);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek int r;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

601185b43da638b1c74153deae01dbd518680889Zbigniew Jędrzejewski-Szmek assert_return(fd >= 0, -EBADF);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek assert_return(IN_SET(family, 0, AF_INET, AF_INET6), -EINVAL);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek r = sd_is_socket_internal(fd, type, listening);

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek if (r <= 0)

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek return r;

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek if (getsockname(fd, &sockaddr.sa, &l) < 0)

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (l < sizeof(sa_family_t))

601185b43da638b1c74153deae01dbd518680889Zbigniew Jędrzejewski-Szmek return -EINVAL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (sockaddr.sa.sa_family != AF_INET &&

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek sockaddr.sa.sa_family != AF_INET6)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (family != 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (sockaddr.sa.sa_family != family)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek if (port > 0) {

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek if (sockaddr.sa.sa_family == AF_INET) {

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek if (l < sizeof(struct sockaddr_in))

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek return -EINVAL;

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek

eb9da376d76b48585b3b63b4f91903b54f7abd36Lennart Poettering return htons(port) == sockaddr.in.sin_port;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek } else {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (l < sizeof(struct sockaddr_in6))

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -EINVAL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return htons(port) == sockaddr.in6.sin6_port;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

ed6556920c1a6cdfe0bb04e806bc1f54ea191545Mantas Mikulėnas

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 1;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek}

601185b43da638b1c74153deae01dbd518680889Zbigniew Jędrzejewski-Szmek

601185b43da638b1c74153deae01dbd518680889Zbigniew Jędrzejewski-Szmek_public_ int sd_is_socket_unix(int fd, int type, int listening, const char *path, size_t length) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek union sockaddr_union sockaddr = {};

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek socklen_t l = sizeof(sockaddr);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek int r;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek assert_return(fd >= 0, -EBADF);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek r = sd_is_socket_internal(fd, type, listening);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (r <= 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return r;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (getsockname(fd, &sockaddr.sa, &l) < 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (l < sizeof(sa_family_t))

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -EINVAL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (sockaddr.sa.sa_family != AF_UNIX)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek

892213bf1fd23e48d64a407ece6e10b07bef1926Zbigniew Jędrzejewski-Szmek if (path) {

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek if (length == 0)

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek length = strlen(path);

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek if (length == 0)

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek /* Unnamed socket */

5e65c93a433447b15180249166f7b3944c3e6156Zbigniew Jędrzejewski-Szmek return l == offsetof(struct sockaddr_un, sun_path);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (path[0])

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek /* Normal path socket */

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return

eb9da376d76b48585b3b63b4f91903b54f7abd36Lennart Poettering (l >= offsetof(struct sockaddr_un, sun_path) + length + 1) &&

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek memcmp(path, sockaddr.un.sun_path, length+1) == 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek else

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek /* Abstract namespace socket */

601185b43da638b1c74153deae01dbd518680889Zbigniew Jędrzejewski-Szmek return

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek (l == offsetof(struct sockaddr_un, sun_path) + length) &&

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek memcmp(path, sockaddr.un.sun_path, length) == 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 1;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek}

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek_public_ int sd_is_mq(int fd, const char *path) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct mq_attr attr;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek /* Check that the fd is valid */

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek assert_return(fcntl(fd, F_GETFD) >= 0, -errno);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (mq_getattr(fd, &attr) < 0) {

eceb8483e5a02e8e337486b89719a3b99cfcb7ceLennart Poettering if (errno == EBADF)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek /* A non-mq fd (or an invalid one, but we ruled that out above) */

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (path) {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek char fpath[PATH_MAX];

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct stat a, b;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek assert_return(path_is_absolute(path), -EINVAL);

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (fstat(fd, &a) < 0)

2c408cb6276e3b8d18fb4e2a81a1128d8bbaa70dLennart Poettering return -errno;

2c408cb6276e3b8d18fb4e2a81a1128d8bbaa70dLennart Poettering

2c408cb6276e3b8d18fb4e2a81a1128d8bbaa70dLennart Poettering strncpy(stpcpy(fpath, "/dev/mqueue"), path, sizeof(fpath) - 12);

2c408cb6276e3b8d18fb4e2a81a1128d8bbaa70dLennart Poettering fpath[sizeof(fpath)-1] = 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

eceb8483e5a02e8e337486b89719a3b99cfcb7ceLennart Poettering if (stat(fpath, &b) < 0)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return -errno;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek if (a.st_dev != b.st_dev ||

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek a.st_ino != b.st_ino)

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek return 0;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek }

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

56f64d95763a799ba4475daf44d8e9f72a1bd474Michal Schmidt return 1;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek}

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek_public_ int sd_pid_notify_with_fds(pid_t pid, int unset_environment, const char *state, const int *fds, unsigned n_fds) {

2c408cb6276e3b8d18fb4e2a81a1128d8bbaa70dLennart Poettering union sockaddr_union sockaddr = {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek .sa.sa_family = AF_UNIX,

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek };

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct iovec iovec = {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek .iov_base = (char*) state,

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek };

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct msghdr msghdr = {

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek .msg_iov = &iovec,

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek .msg_iovlen = 1,

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek .msg_name = &sockaddr,

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek };

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek _cleanup_close_ int fd = -1;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek struct cmsghdr *cmsg = NULL;

2ca0435be9359bde3020eeb528c2a6d72ac1e0b0Zbigniew Jędrzejewski-Szmek const char *e;

bool have_pid;

int r;

if (!state) {

r = -EINVAL;

goto finish;

}

if (n_fds > 0 && !fds) {

r = -EINVAL;

goto finish;

}

e = getenv("NOTIFY_SOCKET");

if (!e)

return 0;

/* Must be an abstract socket, or an absolute path */

if ((e[0] != '@' && e[0] != '/') || e[1] == 0) {

r = -EINVAL;

goto finish;

}

fd = socket(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC, 0);

if (fd < 0) {

r = -errno;

goto finish;

}

iovec.iov_len = strlen(state);

strncpy(sockaddr.un.sun_path, e, sizeof(sockaddr.un.sun_path));

if (sockaddr.un.sun_path[0] == '@')

sockaddr.un.sun_path[0] = 0;

msghdr.msg_namelen = offsetof(struct sockaddr_un, sun_path) + strlen(e);

if (msghdr.msg_namelen > sizeof(struct sockaddr_un))

msghdr.msg_namelen = sizeof(struct sockaddr_un);

have_pid = pid != 0 && pid != getpid();

if (n_fds > 0 || have_pid) {

/* CMSG_SPACE(0) may return value different then zero, which results in miscalculated controllen. */

msghdr.msg_controllen =

(n_fds > 0 ? CMSG_SPACE(sizeof(int) * n_fds) : 0) +

(have_pid ? CMSG_SPACE(sizeof(struct ucred)) : 0);

msghdr.msg_control = alloca(msghdr.msg_controllen);

cmsg = CMSG_FIRSTHDR(&msghdr);

if (n_fds > 0) {

cmsg->cmsg_level = SOL_SOCKET;

cmsg->cmsg_type = SCM_RIGHTS;

cmsg->cmsg_len = CMSG_LEN(sizeof(int) * n_fds);

memcpy(CMSG_DATA(cmsg), fds, sizeof(int) * n_fds);

if (have_pid)

assert_se(cmsg = CMSG_NXTHDR(&msghdr, cmsg));

}

if (have_pid) {

struct ucred *ucred;

cmsg->cmsg_level = SOL_SOCKET;

cmsg->cmsg_type = SCM_CREDENTIALS;

cmsg->cmsg_len = CMSG_LEN(sizeof(struct ucred));

ucred = (struct ucred*) CMSG_DATA(cmsg);

ucred->pid = pid;

ucred->uid = getuid();

ucred->gid = getgid();

}

}

/* First try with fake ucred data, as requested */

if (sendmsg(fd, &msghdr, MSG_NOSIGNAL) >= 0) {

r = 1;

goto finish;

}

/* If that failed, try with our own ucred instead */

if (have_pid) {

msghdr.msg_controllen -= CMSG_SPACE(sizeof(struct ucred));

if (msghdr.msg_controllen == 0)

msghdr.msg_control = NULL;

if (sendmsg(fd, &msghdr, MSG_NOSIGNAL) >= 0) {

r = 1;

goto finish;

}

}

r = -errno;

finish:

if (unset_environment)

unsetenv("NOTIFY_SOCKET");

return r;

}

_public_ int sd_pid_notify(pid_t pid, int unset_environment, const char *state) {

return sd_pid_notify_with_fds(pid, unset_environment, state, NULL, 0);

}

_public_ int sd_notify(int unset_environment, const char *state) {

return sd_pid_notify_with_fds(0, unset_environment, state, NULL, 0);

}

_public_ int sd_pid_notifyf(pid_t pid, int unset_environment, const char *format, ...) {

_cleanup_free_ char *p = NULL;

int r;

if (format) {

va_list ap;

va_start(ap, format);

r = vasprintf(&p, format, ap);

va_end(ap);

if (r < 0 || !p)

return -ENOMEM;

}

return sd_pid_notify(pid, unset_environment, p);

}

_public_ int sd_notifyf(int unset_environment, const char *format, ...) {

_cleanup_free_ char *p = NULL;

int r;

if (format) {

va_list ap;

va_start(ap, format);

r = vasprintf(&p, format, ap);

va_end(ap);

if (r < 0 || !p)

return -ENOMEM;

}

return sd_pid_notify(0, unset_environment, p);

}

_public_ int sd_booted(void) {

/* We test whether the runtime unit file directory has been

return laccess("/run/systemd/system/", F_OK) >= 0;

}

_public_ int sd_watchdog_enabled(int unset_environment, uint64_t *usec) {

const char *s, *p = ""; /* p is set to dummy value to do unsetting */

uint64_t u;

int r = 0;

s = getenv("WATCHDOG_USEC");

if (!s)

goto finish;

r = safe_atou64(s, &u);

if (r < 0)

goto finish;

if (u <= 0) {

r = -EINVAL;

goto finish;

}

p = getenv("WATCHDOG_PID");

if (p) {

pid_t pid;

r = parse_pid(p, &pid);

if (r < 0)

goto finish;

/* Is this for us? */

if (getpid() != pid) {

r = 0;

goto finish;

}

}

if (usec)

*usec = u;

r = 1;

finish:

if (unset_environment && s)

unsetenv("WATCHDOG_USEC");

if (unset_environment && p)

unsetenv("WATCHDOG_PID");

return r;

}