mirror of
https://gitlab.com/d3tn/ud3tn.git
synced 2026-08-16 13:01:06 +02:00
a16fe7d69b
This checks for `POLLNVAL` errors for which we would otherwise return `0` (as if the timeout was triggered). It also makes the formatting of checks consistent. Signed-off-by: Felix Walter <felix.walter@d3tn.com>
90 lines
1.7 KiB
C
90 lines
1.7 KiB
C
// SPDX-License-Identifier: BSD-3-Clause OR Apache-2.0
|
|
#include "platform/hal_io.h"
|
|
|
|
#include "ud3tn/common.h"
|
|
|
|
#include <errno.h>
|
|
#include <poll.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <unistd.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
int create_unix_domain_socket(const char *path)
|
|
{
|
|
int sock = socket(AF_UNIX, SOCK_STREAM, 0);
|
|
|
|
if (sock == -1) {
|
|
LOG_ERRNO("Socket Util", "socket(AF_UNIX)", errno);
|
|
return -1;
|
|
}
|
|
|
|
struct sockaddr_un addr = {
|
|
.sun_family = AF_UNIX,
|
|
};
|
|
const int rv1 = snprintf(
|
|
addr.sun_path,
|
|
sizeof(addr.sun_path),
|
|
"%s",
|
|
path
|
|
);
|
|
|
|
if (rv1 <= 0 || (unsigned int)rv1 > sizeof(addr.sun_path)) {
|
|
LOGF_ERROR(
|
|
"Socket Util: Invalid socket path, len = %d, maxlen = %zu",
|
|
rv1,
|
|
sizeof(addr.sun_path)
|
|
);
|
|
return -1;
|
|
}
|
|
|
|
unlink(path);
|
|
int rv2 = bind(
|
|
sock,
|
|
(const struct sockaddr *)&addr,
|
|
sizeof(struct sockaddr_un)
|
|
);
|
|
if (rv2 == -1) {
|
|
LOG_ERRNO("Socket Util", "bind(unix_domain_socket)", errno);
|
|
return -1;
|
|
}
|
|
|
|
return sock;
|
|
}
|
|
|
|
int poll_recv_timeout(const int socket_fd, const int timeout)
|
|
{
|
|
struct pollfd pollfd[1];
|
|
|
|
pollfd[0].events = POLLIN;
|
|
pollfd[0].fd = socket_fd;
|
|
|
|
for (;;) {
|
|
if (poll(pollfd, ARRAY_LENGTH(pollfd), timeout) == -1) {
|
|
const int err = errno;
|
|
|
|
// Try again if interrupted by a signal.
|
|
if (err == EINTR)
|
|
continue;
|
|
|
|
LOG_ERRNO("Socket Util", "poll()", err);
|
|
return -1;
|
|
}
|
|
if (pollfd[0].revents & POLLERR) {
|
|
LOG_WARN("Socket Util: Socket error (e.g. TCP RST) detected.");
|
|
return -1;
|
|
}
|
|
if (pollfd[0].revents & POLLHUP) {
|
|
LOG_INFO("Socket Util: The peer closed the connection.");
|
|
return -1;
|
|
}
|
|
if (pollfd[0].revents & POLLNVAL) {
|
|
LOG_INFO("Socket Util: Connection was not open.");
|
|
return -1;
|
|
}
|
|
if (pollfd[0].revents & POLLIN)
|
|
return 1;
|
|
return 0;
|
|
}
|
|
}
|