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>
45 #include "sd-daemon.h"
47 int sd_listen_fds(int unset_environment) {
48 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
56 if (!(e = getenv("LISTEN_PID"))) {
62 l = strtoul(e, &p, 10);
69 if (!p || *p || l <= 0) {
75 if (getpid() != (pid_t)l) {
80 if (!(e = getenv("LISTEN_FDS"))) {
86 l = strtoul(e, &p, 10);
98 for (fd = SD_LISTEN_FDS_START; fd < SD_LISTEN_FDS_START + (int)l; fd++) {
101 if ((flags = fcntl(fd, F_GETFD)) < 0) {
106 if (flags & FD_CLOEXEC)
109 if (fcntl(fd, F_SETFD, flags | FD_CLOEXEC) < 0) {
118 if (unset_environment) {
119 unsetenv("LISTEN_PID");
120 unsetenv("LISTEN_FDS");
127 int sd_is_fifo(int fd, const char *path) {
133 memset(&st_fd, 0, sizeof(st_fd));
134 if (fstat(fd, &st_fd) < 0)
137 if (!S_ISFIFO(st_fd.st_mode))
143 memset(&st_path, 0, sizeof(st_path));
144 if (stat(path, &st_path) < 0) {
145 if (errno == ENOENT || errno == ENOTDIR)
151 return st_path.st_dev == st_fd.st_dev &&
152 st_path.st_ino == st_fd.st_ino;
158 static int sd_is_socket_internal(int fd, int type, int listening) {
161 if (fd < 0 || type < 0)
164 if (fstat(fd, &st_fd) < 0)
167 if (!S_ISSOCK(st_fd.st_mode))
172 socklen_t l = sizeof(other_type);
174 if (getsockopt(fd, SOL_SOCKET, SO_TYPE, &other_type, &l) < 0)
177 if (l != sizeof(other_type))
180 if (other_type != type)
184 if (listening >= 0) {
186 socklen_t l = sizeof(accepting);
188 if (getsockopt(fd, SOL_SOCKET, SO_ACCEPTCONN, &accepting, &l) < 0)
191 if (l != sizeof(accepting))
194 if (!accepting != !listening)
201 union sockaddr_union {
203 struct sockaddr_in in4;
204 struct sockaddr_in6 in6;
205 struct sockaddr_un un;
206 struct sockaddr_storage storage;
209 int sd_is_socket(int fd, int family, int type, int listening) {
215 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
219 union sockaddr_union sockaddr;
222 memset(&sockaddr, 0, sizeof(sockaddr));
223 l = sizeof(sockaddr);
225 if (getsockname(fd, &sockaddr.sa, &l) < 0)
228 if (l < sizeof(sa_family_t))
231 return sockaddr.sa.sa_family == family;
237 int sd_is_socket_inet(int fd, int family, int type, int listening, uint16_t port) {
238 union sockaddr_union sockaddr;
242 if (family != 0 && family != AF_INET && family != AF_INET6)
245 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
248 memset(&sockaddr, 0, sizeof(sockaddr));
249 l = sizeof(sockaddr);
251 if (getsockname(fd, &sockaddr.sa, &l) < 0)
254 if (l < sizeof(sa_family_t))
257 if (sockaddr.sa.sa_family != AF_INET &&
258 sockaddr.sa.sa_family != AF_INET6)
262 if (sockaddr.sa.sa_family != family)
266 if (sockaddr.sa.sa_family == AF_INET) {
267 if (l < sizeof(struct sockaddr_in))
270 return htons(port) == sockaddr.in4.sin_port;
272 if (l < sizeof(struct sockaddr_in6))
275 return htons(port) == sockaddr.in6.sin6_port;
282 int sd_is_socket_unix(int fd, int type, int listening, const char *path, size_t length) {
283 union sockaddr_union sockaddr;
287 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
290 memset(&sockaddr, 0, sizeof(sockaddr));
291 l = sizeof(sockaddr);
293 if (getsockname(fd, &sockaddr.sa, &l) < 0)
296 if (l < sizeof(sa_family_t))
299 if (sockaddr.sa.sa_family != AF_UNIX)
304 length = strlen(path);
308 return l == offsetof(struct sockaddr_un, sun_path);
311 /* Normal path socket */
312 return (l >= offsetof(struct sockaddr_un, sun_path) + length + 1) &&
313 memcmp(path, sockaddr.un.sun_path, length + 1) == 0;
315 /* Abstract namespace socket */
316 return (l == offsetof(struct sockaddr_un, sun_path) + length) &&
317 memcmp(path, sockaddr.un.sun_path, length) == 0;
323 int sd_notify(int unset_environment, const char *state) {
324 #if defined(DISABLE_SYSTEMD) || !defined(__linux__) || !defined(SOCK_CLOEXEC)
328 struct msghdr msghdr;
330 union sockaddr_union sockaddr;
338 if (!(e = getenv("NOTIFY_SOCKET")))
341 /* Must be an abstract socket, or an absolute path */
342 if ((e[0] != '@' && e[0] != '/') || e[1] == 0) {
347 if ((fd = socket(AF_UNIX, SOCK_DGRAM | SOCK_CLOEXEC, 0)) < 0) {
352 memset(&sockaddr, 0, sizeof(sockaddr));
353 sockaddr.sa.sa_family = AF_UNIX;
354 strncpy(sockaddr.un.sun_path, e, sizeof(sockaddr.un.sun_path));
356 if (sockaddr.un.sun_path[0] == '@')
357 sockaddr.un.sun_path[0] = 0;
359 memset(&iovec, 0, sizeof(iovec));
360 iovec.iov_base = (char *)state;
361 iovec.iov_len = strlen(state);
363 memset(&msghdr, 0, sizeof(msghdr));
364 msghdr.msg_name = &sockaddr;
365 msghdr.msg_namelen = offsetof(struct sockaddr_un, sun_path) + strlen(e);
367 if (msghdr.msg_namelen > sizeof(struct sockaddr_un))
368 msghdr.msg_namelen = sizeof(struct sockaddr_un);
370 msghdr.msg_iov = &iovec;
371 msghdr.msg_iovlen = 1;
373 if (sendmsg(fd, &msghdr, MSG_NOSIGNAL) < 0) {
381 if (unset_environment)
382 unsetenv("NOTIFY_SOCKET");
391 int sd_notifyf(int unset_environment, const char *format, ...) {
392 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
399 va_start(ap, format);
400 r = vasprintf(&p, format, ap);
406 r = sd_notify(unset_environment, p);
413 int sd_booted(void) {
414 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
420 /* We simply test whether the systemd cgroup hierarchy is
423 if (lstat("/sys/fs/cgroup", &a) < 0)
426 if (lstat("/sys/fs/cgroup/systemd", &b) < 0)
429 return a.st_dev != b.st_dev;