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) {
49 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
57 if (!(e = getenv("LISTEN_PID"))) {
63 l = strtoul(e, &p, 10);
70 if (!p || *p || l <= 0) {
76 if (getpid() != (pid_t) l) {
81 if (!(e = getenv("LISTEN_FDS"))) {
87 l = strtoul(e, &p, 10);
99 for (fd = SD_LISTEN_FDS_START; fd < SD_LISTEN_FDS_START + (int) l; fd ++) {
102 if ((flags = fcntl(fd, F_GETFD)) < 0) {
107 if (flags & FD_CLOEXEC)
110 if (fcntl(fd, F_SETFD, flags | FD_CLOEXEC) < 0) {
119 if (unset_environment) {
120 unsetenv("LISTEN_PID");
121 unsetenv("LISTEN_FDS");
128 int sd_is_fifo(int fd, const char *path) {
134 memset(&st_fd, 0, sizeof(st_fd));
135 if (fstat(fd, &st_fd) < 0)
138 if (!S_ISFIFO(st_fd.st_mode))
144 memset(&st_path, 0, sizeof(st_path));
145 if (stat(path, &st_path) < 0) {
147 if (errno == ENOENT || errno == ENOTDIR)
154 st_path.st_dev == st_fd.st_dev &&
155 st_path.st_ino == st_fd.st_ino;
161 static int sd_is_socket_internal(int fd, int type, int listening) {
164 if (fd < 0 || type < 0)
167 if (fstat(fd, &st_fd) < 0)
170 if (!S_ISSOCK(st_fd.st_mode))
175 socklen_t l = sizeof(other_type);
177 if (getsockopt(fd, SOL_SOCKET, SO_TYPE, &other_type, &l) < 0)
180 if (l != sizeof(other_type))
183 if (other_type != type)
187 if (listening >= 0) {
189 socklen_t l = sizeof(accepting);
191 if (getsockopt(fd, SOL_SOCKET, SO_ACCEPTCONN, &accepting, &l) < 0)
194 if (l != sizeof(accepting))
197 if (!accepting != !listening)
204 union sockaddr_union {
206 struct sockaddr_in in4;
207 struct sockaddr_in6 in6;
208 struct sockaddr_un un;
209 struct sockaddr_storage storage;
212 int sd_is_socket(int fd, int family, int type, int listening) {
218 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
222 union sockaddr_union sockaddr;
225 memset(&sockaddr, 0, sizeof(sockaddr));
226 l = sizeof(sockaddr);
228 if (getsockname(fd, &sockaddr.sa, &l) < 0)
231 if (l < sizeof(sa_family_t))
234 return sockaddr.sa.sa_family == family;
240 int sd_is_socket_inet(int fd, int family, int type, int listening, uint16_t port) {
241 union sockaddr_union sockaddr;
245 if (family != 0 && family != AF_INET && family != AF_INET6)
248 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
251 memset(&sockaddr, 0, sizeof(sockaddr));
252 l = sizeof(sockaddr);
254 if (getsockname(fd, &sockaddr.sa, &l) < 0)
257 if (l < sizeof(sa_family_t))
260 if (sockaddr.sa.sa_family != AF_INET &&
261 sockaddr.sa.sa_family != AF_INET6)
265 if (sockaddr.sa.sa_family != family)
269 if (sockaddr.sa.sa_family == AF_INET) {
270 if (l < sizeof(struct sockaddr_in))
273 return htons(port) == sockaddr.in4.sin_port;
275 if (l < sizeof(struct sockaddr_in6))
278 return htons(port) == sockaddr.in6.sin6_port;
285 int sd_is_socket_unix(int fd, int type, int listening, const char *path, size_t length) {
286 union sockaddr_union sockaddr;
290 if ((r = sd_is_socket_internal(fd, type, listening)) <= 0)
293 memset(&sockaddr, 0, sizeof(sockaddr));
294 l = sizeof(sockaddr);
296 if (getsockname(fd, &sockaddr.sa, &l) < 0)
299 if (l < sizeof(sa_family_t))
302 if (sockaddr.sa.sa_family != AF_UNIX)
307 length = strlen(path);
311 return l == offsetof(struct sockaddr_un, sun_path);
314 /* Normal path socket */
316 (l >= offsetof(struct sockaddr_un, sun_path) + length + 1) &&
317 memcmp(path, sockaddr.un.sun_path, length+1) == 0;
319 /* Abstract namespace socket */
321 (l == offsetof(struct sockaddr_un, sun_path) + length) &&
322 memcmp(path, sockaddr.un.sun_path, length) == 0;
328 int sd_notify(int unset_environment, const char *state) {
329 #if defined(DISABLE_SYSTEMD) || !defined(__linux__) || !defined(SOCK_CLOEXEC)
333 struct msghdr msghdr;
335 union sockaddr_union sockaddr;
343 if (!(e = getenv("NOTIFY_SOCKET")))
346 /* Must be an abstract socket, or an absolute path */
347 if ((e[0] != '@' && e[0] != '/') || e[1] == 0) {
352 if ((fd = socket(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC, 0)) < 0) {
357 memset(&sockaddr, 0, sizeof(sockaddr));
358 sockaddr.sa.sa_family = AF_UNIX;
359 strncpy(sockaddr.un.sun_path, e, sizeof(sockaddr.un.sun_path));
361 if (sockaddr.un.sun_path[0] == '@')
362 sockaddr.un.sun_path[0] = 0;
364 memset(&iovec, 0, sizeof(iovec));
365 iovec.iov_base = (char*) state;
366 iovec.iov_len = strlen(state);
368 memset(&msghdr, 0, sizeof(msghdr));
369 msghdr.msg_name = &sockaddr;
370 msghdr.msg_namelen = offsetof(struct sockaddr_un, sun_path) + strlen(e);
372 if (msghdr.msg_namelen > sizeof(struct sockaddr_un))
373 msghdr.msg_namelen = sizeof(struct sockaddr_un);
375 msghdr.msg_iov = &iovec;
376 msghdr.msg_iovlen = 1;
378 if (sendmsg(fd, &msghdr, MSG_NOSIGNAL) < 0) {
386 if (unset_environment)
387 unsetenv("NOTIFY_SOCKET");
396 int sd_notifyf(int unset_environment, const char *format, ...) {
397 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
404 va_start(ap, format);
405 r = vasprintf(&p, format, ap);
411 r = sd_notify(unset_environment, p);
418 int sd_booted(void) {
419 #if defined(DISABLE_SYSTEMD) || !defined(__linux__)
425 /* We simply test whether the systemd cgroup hierarchy is
428 if (lstat("/sys/fs/cgroup", &a) < 0)
431 if (lstat("/sys/fs/cgroup/systemd", &b) < 0)
434 return a.st_dev != b.st_dev;