1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 Copyright 2010 Lennart Poettering
6 Permission is hereby granted, free of charge, to any person
7 obtaining a copy of this software and associated documentation files
8 (the "Software"), to deal in the Software without restriction,
9 including without limitation the rights to use, copy, modify, merge,
10 publish, distribute, sublicense, and/or sell copies of the Software,
11 and to permit persons to whom the Software is furnished to do so,
12 subject to the following conditions:
14 The above copyright notice and this permission notice shall be
15 included in all copies or substantial portions of the Software.
17 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
18 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
19 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
20 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
21 BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
22 ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
23 CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
31 #include <sys/types.h>
33 #include <sys/socket.h>
35 #include <sys/fcntl.h>
36 #include <netinet/in.h>
46 #if defined(__linux__)
50 #include "sd-daemon.h"
52 #if (__GNUC__ >= 4) && !defined(SD_EXPORT_SYMBOLS)
53 #define _sd_hidden_ __attribute__ ((visibility("hidden")))
58 _sd_hidden_ int sd_listen_fds(int unset_environment) {
60 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
68 if (!(e = getenv("LISTEN_PID"))) {
74 l = strtoul(e, &p, 10);
81 if (!p || *p || l <= 0) {
87 if (getpid() != (pid_t) l) {
92 if (!(e = getenv("LISTEN_FDS"))) {
98 l = strtoul(e, &p, 10);
110 for (fd = SD_LISTEN_FDS_START; fd < SD_LISTEN_FDS_START + (int) l; fd ++) {
113 if ((flags = fcntl(fd, F_GETFD)) < 0) {
118 if (flags & FD_CLOEXEC)
121 if (fcntl(fd, F_SETFD, flags | FD_CLOEXEC) < 0) {
130 if (unset_environment) {
131 unsetenv("LISTEN_PID");
132 unsetenv("LISTEN_FDS");
139 _sd_hidden_ int sd_is_fifo(int fd, const char *path) {
145 memset(&st_fd, 0, sizeof(st_fd));
146 if (fstat(fd, &st_fd) < 0)
149 if (!S_ISFIFO(st_fd.st_mode))
155 memset(&st_path, 0, sizeof(st_path));
156 if (stat(path, &st_path) < 0) {
158 if (errno == ENOENT || errno == ENOTDIR)
165 st_path.st_dev == st_fd.st_dev &&
166 st_path.st_ino == st_fd.st_ino;
172 static int sd_is_socket_internal(int fd, int type, int listening) {
175 if (fd < 0 || type < 0)
178 if (fstat(fd, &st_fd) < 0)
181 if (!S_ISSOCK(st_fd.st_mode))
186 socklen_t l = sizeof(other_type);
188 if (getsockopt(fd, SOL_SOCKET, SO_TYPE, &other_type, &l) < 0)
191 if (l != sizeof(other_type))
194 if (other_type != type)
198 if (listening >= 0) {
200 socklen_t l = sizeof(accepting);
202 if (getsockopt(fd, SOL_SOCKET, SO_ACCEPTCONN, &accepting, &l) < 0)
205 if (l != sizeof(accepting))
208 if (!accepting != !listening)
215 union sockaddr_union {
217 struct sockaddr_in in4;
218 struct sockaddr_in6 in6;
219 struct sockaddr_un un;
220 struct sockaddr_storage storage;
223 _sd_hidden_ int sd_is_socket(int fd, int family, int type, int listening) {
229 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
233 union sockaddr_union sockaddr;
236 memset(&sockaddr, 0, sizeof(sockaddr));
237 l = sizeof(sockaddr);
239 if (getsockname(fd, &sockaddr.sa, &l) < 0)
242 if (l < sizeof(sa_family_t))
245 return sockaddr.sa.sa_family == family;
251 _sd_hidden_ int sd_is_socket_inet(int fd, int family, int type, int listening, uint16_t port) {
252 union sockaddr_union sockaddr;
256 if (family != 0 && family != AF_INET && family != AF_INET6)
259 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
262 memset(&sockaddr, 0, sizeof(sockaddr));
263 l = sizeof(sockaddr);
265 if (getsockname(fd, &sockaddr.sa, &l) < 0)
268 if (l < sizeof(sa_family_t))
271 if (sockaddr.sa.sa_family != AF_INET &&
272 sockaddr.sa.sa_family != AF_INET6)
276 if (sockaddr.sa.sa_family != family)
280 if (sockaddr.sa.sa_family == AF_INET) {
281 if (l < sizeof(struct sockaddr_in))
284 return htons(port) == sockaddr.in4.sin_port;
286 if (l < sizeof(struct sockaddr_in6))
289 return htons(port) == sockaddr.in6.sin6_port;
296 _sd_hidden_ int sd_is_socket_unix(int fd, int type, int listening, const char *path, size_t length) {
297 union sockaddr_union sockaddr;
301 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
304 memset(&sockaddr, 0, sizeof(sockaddr));
305 l = sizeof(sockaddr);
307 if (getsockname(fd, &sockaddr.sa, &l) < 0)
310 if (l < sizeof(sa_family_t))
313 if (sockaddr.sa.sa_family != AF_UNIX)
318 length = strlen(path);
322 return l == offsetof(struct sockaddr_un, sun_path);
325 /* Normal path socket */
327 (l >= offsetof(struct sockaddr_un, sun_path) + length + 1) &&
328 memcmp(path, sockaddr.un.sun_path, length+1) == 0;
330 /* Abstract namespace socket */
332 (l == offsetof(struct sockaddr_un, sun_path) + length) &&
333 memcmp(path, sockaddr.un.sun_path, length) == 0;
339 _sd_hidden_ int sd_is_mq(int fd, const char *path) {
340 #if !defined(__linux__)
348 if (mq_getattr(fd, &attr) < 0)
352 char fpath[PATH_MAX];
358 if (fstat(fd, &a) < 0)
361 strncpy(stpcpy(fpath, "/dev/mqueue"), path, sizeof(fpath) - 12);
362 fpath[sizeof(fpath)-1] = 0;
364 if (stat(fpath, &b) < 0)
367 if (a.st_dev != b.st_dev ||
368 a.st_ino != b.st_ino)
376 _sd_hidden_ int sd_notify(int unset_environment, const char *state) {
377 #if defined(DISABLE_SYSTEMD) || !defined(__linux__) || !defined(SOCK_CLOEXEC)
381 struct msghdr msghdr;
383 union sockaddr_union sockaddr;
391 if (!(e = getenv("NOTIFY_SOCKET")))
394 /* Must be an abstract socket, or an absolute path */
395 if ((e[0] != '@' && e[0] != '/') || e[1] == 0) {
400 if ((fd = socket(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC, 0)) < 0) {
405 memset(&sockaddr, 0, sizeof(sockaddr));
406 sockaddr.sa.sa_family = AF_UNIX;
407 strncpy(sockaddr.un.sun_path, e, sizeof(sockaddr.un.sun_path));
409 if (sockaddr.un.sun_path[0] == '@')
410 sockaddr.un.sun_path[0] = 0;
412 memset(&iovec, 0, sizeof(iovec));
413 iovec.iov_base = (char*) state;
414 iovec.iov_len = strlen(state);
416 memset(&msghdr, 0, sizeof(msghdr));
417 msghdr.msg_name = &sockaddr;
418 msghdr.msg_namelen = offsetof(struct sockaddr_un, sun_path) + strlen(e);
420 if (msghdr.msg_namelen > sizeof(struct sockaddr_un))
421 msghdr.msg_namelen = sizeof(struct sockaddr_un);
423 msghdr.msg_iov = &iovec;
424 msghdr.msg_iovlen = 1;
426 if (sendmsg(fd, &msghdr, MSG_NOSIGNAL) < 0) {
434 if (unset_environment)
435 unsetenv("NOTIFY_SOCKET");
444 _sd_hidden_ int sd_notifyf(int unset_environment, const char *format, ...) {
445 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
452 va_start(ap, format);
453 r = vasprintf(&p, format, ap);
459 r = sd_notify(unset_environment, p);
466 _sd_hidden_ int sd_booted(void) {
467 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
473 /* We simply test whether the systemd cgroup hierarchy is
476 if (lstat("/sys/fs/cgroup", &a) < 0)
479 if (lstat("/sys/fs/cgroup/systemd", &b) < 0)
482 return a.st_dev != b.st_dev;