mirror of
https://git.haproxy.org/git/haproxy.git/
synced 2025-08-07 07:37:02 +02:00
f0548302bb
systemd.c doesn't build on Solaris / Illumos because it uses "sun" as
the field name in a structure, while "sun" is the name of the macro
used to detect Solaris:
src/systemd.c: In function 'sd_notify':
src/systemd.c:43:22: error: expected identifier or '(' before numeric constant
struct sockaddr_un sun;
^
src/systemd.c:44:2: warning: no semicolon at end of struct or union
} socket_addr = {
^
Admittedly, the OS could have instead defined "sun" to itself to avoid
this. Any other name will work, let's just use "ux" for the short form
of "unix".
The problem appeared in 3.0-dev with commit aa3632962f ("MEDIUM:
mworker: get rid of libsystemd"), though by then this file was only
built when USE_SYSTEMD was set, which was not the case for non-linux
platforms. However since 3.1-dev14 with commit 15845247db ("MEDIUM:
mworker: remove USE_SYSTEMD requirement for -Ws"), all platforms
now build this file.
No backport is needed even though it will not hurt to have it in 3.0
for completeness.
141 lines
2.9 KiB
C
141 lines
2.9 KiB
C
/* SPDX-License-Identifier: MIT-0 */
|
|
|
|
/* Implement the systemd notify protocol without external dependencies.
|
|
* Supports both readiness notification on startup and on reloading,
|
|
* according to the protocol defined at:
|
|
* https://www.freedesktop.org/software/systemd/man/latest/sd_notify.html
|
|
* This protocol is guaranteed to be stable as per:
|
|
* https://systemd.io/PORTABILITY_AND_STABILITY/
|
|
*
|
|
*/
|
|
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <inttypes.h>
|
|
#include <signal.h>
|
|
#include <stdbool.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
#include <stdarg.h>
|
|
|
|
#include <haproxy/tools.h>
|
|
|
|
/*
|
|
* standalone reimplementation of sd_notify from the libsystemd
|
|
* Return:
|
|
* -errno in case of error
|
|
* 0 when ignored
|
|
* >0 when succeeded
|
|
*
|
|
* Will send <message> over the NOTIFY_SOCKET.
|
|
* When unset_environement is set, unsetenv NOTIFY_SOCKET.
|
|
*/
|
|
int sd_notify(int unset_environment, const char *message)
|
|
{
|
|
union sockaddr_union {
|
|
struct sockaddr sa;
|
|
struct sockaddr_un ux;
|
|
} socket_addr = {
|
|
.ux.sun_family = AF_UNIX,
|
|
};
|
|
int ret = 1;
|
|
int fd = -1;
|
|
size_t path_length, message_length;
|
|
const char *socket_path;
|
|
ssize_t written;
|
|
|
|
socket_path = getenv("NOTIFY_SOCKET");
|
|
if (!socket_path) {
|
|
ret = 0; /* Not running under systemd? Nothing to do */
|
|
goto end;
|
|
}
|
|
|
|
if (unset_environment)
|
|
unsetenv("NOTIFY_SOCKET");
|
|
|
|
if (!message) {
|
|
ret = -EINVAL;
|
|
goto end;
|
|
}
|
|
|
|
message_length = strlen(message);
|
|
if (message_length == 0) {
|
|
ret = -EINVAL;
|
|
goto end;
|
|
}
|
|
|
|
/* Only AF_UNIX is supported, with path or abstract sockets */
|
|
if (socket_path[0] != '/' && socket_path[0] != '@') {
|
|
ret = -EAFNOSUPPORT;
|
|
goto end;
|
|
}
|
|
|
|
path_length = strlen(socket_path);
|
|
/* Ensure there is room for NUL byte */
|
|
if (path_length >= sizeof(socket_addr.ux.sun_path)) {
|
|
ret = -E2BIG;
|
|
goto end;
|
|
}
|
|
|
|
memcpy(socket_addr.ux.sun_path, socket_path, path_length);
|
|
|
|
/* Support for abstract socket */
|
|
if (socket_addr.ux.sun_path[0] == '@')
|
|
socket_addr.ux.sun_path[0] = 0;
|
|
|
|
fd = socket(AF_UNIX, SOCK_DGRAM, 0);
|
|
if (fd < 0) {
|
|
ret = -errno;
|
|
goto end;
|
|
}
|
|
|
|
if (fcntl(fd, F_SETFD, FD_CLOEXEC) != 0) {
|
|
ret = -errno;
|
|
goto end;
|
|
}
|
|
|
|
if (connect(fd, &socket_addr.sa, offsetof(struct sockaddr_un, sun_path) + path_length) != 0) {
|
|
ret = -errno;
|
|
goto end;
|
|
}
|
|
|
|
written = write(fd, message, message_length);
|
|
if (written != (ssize_t) message_length) {
|
|
ret = written < 0 ? -errno : -EPROTO;
|
|
goto end;
|
|
}
|
|
|
|
end:
|
|
if (fd > -1)
|
|
close(fd);
|
|
return ret; /* Notified! */
|
|
}
|
|
|
|
/* va_args variant of sd_notify */
|
|
int sd_notifyf(int unset_environment, const char *format, ...)
|
|
{
|
|
int r;
|
|
va_list args;
|
|
char *strp = NULL;
|
|
|
|
va_start(args, format);
|
|
strp = memvprintf(&strp, format, args);
|
|
va_end(args);
|
|
|
|
if (strp == NULL) {
|
|
r = -ENOMEM;
|
|
goto end;
|
|
}
|
|
|
|
r = sd_notify(unset_environment, strp);
|
|
free(strp);
|
|
end:
|
|
return r;
|
|
}
|
|
|