port: FreeBSD has no ENODATA, alias it to ENOATTR

[lttng-tools.git] / src / common / sessiond-comm / inet.c

diff --git a/src/common/sessiond-comm/inet.c b/src/common/sessiond-comm/inet.c
index ac450dbc8b31b9e394eac7062d326edefcc26748..827b97598f0810a0a8ce229318be60be67c3e1e3 100644 (file)
--- a/src/common/sessiond-comm/inet.c
+++ b/src/common/sessiond-comm/inet.c
@@ -1,18 +1,8 @@
 /*
- * Copyright (C) 2012 - David Goulet <dgoulet@efficios.com>
+ * Copyright (C) 2012 David Goulet <dgoulet@efficios.com>
  *
- * This program is free software; you can redistribute it and/or modify it
- * under the terms of the GNU General Public License, version 2 only, as
- * published by the Free Software Foundation.
+ * SPDX-License-Identifier: GPL-2.0-only
  *
- * This program is distributed in the hope that it will be useful, but WITHOUT
- * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
- * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
- * more details.
- *
- * You should have received a copy of the GNU General Public License along with
- * this program; if not, write to the Free Software Foundation, Inc., 51
- * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
 #define _LGPL_SOURCE
@@ -24,13 +14,13 @@
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <unistd.h>
-#include <errno.h>
 #include <fcntl.h>
 #include <common/compat/time.h>
 #include <poll.h>
 
 #include <common/common.h>
 #include <common/time.h>
+#include <common/compat/errno.h>
 
 #include "inet.h"
 
@@ -101,14 +91,17 @@ LTTNG_HIDDEN
 int lttcomm_bind_inet_sock(struct lttcomm_sock *sock)
 {
        return bind(sock->fd,
-                       (const struct sockaddr *) &sock->sockaddr.addr.sin,
+                       (const struct sockaddr *) ALIGNED_CONST_PTR(
+                                       sock->sockaddr.addr.sin),
                        sizeof(sock->sockaddr.addr.sin));
 }
 
 static
 int connect_no_timeout(struct lttcomm_sock *sock)
 {
-       return connect(sock->fd, (struct sockaddr *) &sock->sockaddr.addr.sin,
+       return connect(sock->fd,
+                       (const struct sockaddr *) ALIGNED_CONST_PTR(
+                                       sock->sockaddr.addr.sin),
                        sizeof(sock->sockaddr.addr.sin));
 }
 
@@ -141,11 +134,11 @@ int connect_with_timeout(struct lttcomm_sock *sock)
        }
 
        connect_ret = connect(sock->fd,
-               (struct sockaddr *) &sock->sockaddr.addr.sin,
-               sizeof(sock->sockaddr.addr.sin));
-       if (connect_ret == -1 && errno != EAGAIN
-                       && errno != EWOULDBLOCK
-                       && errno != EINPROGRESS) {
+                       (const struct sockaddr *) ALIGNED_CONST_PTR(
+                                       sock->sockaddr.addr.sin),
+                       sizeof(sock->sockaddr.addr.sin));
+       if (connect_ret == -1 && errno != EAGAIN && errno != EWOULDBLOCK &&
+                       errno != EINPROGRESS) {
                goto error;
        } else if (!connect_ret) {
                /* Connect succeeded */
@@ -261,6 +254,7 @@ struct lttcomm_sock *lttcomm_accept_inet_sock(struct lttcomm_sock *sock)
        socklen_t len;
        struct lttcomm_sock *new_sock;
        unsigned long timeout;
+       struct sockaddr_in new_addr = {};
 
        if (sock->proto == LTTCOMM_SOCK_UDP) {
                /*
@@ -275,15 +269,15 @@ struct lttcomm_sock *lttcomm_accept_inet_sock(struct lttcomm_sock *sock)
                goto error;
        }
 
-       len = sizeof(new_sock->sockaddr.addr.sin);
+       len = sizeof(new_addr);
 
        /* Blocking call */
-       new_fd = accept(sock->fd, (struct sockaddr *) &new_sock->sockaddr.addr.sin,
-                       &len);
+       new_fd = accept(sock->fd, (struct sockaddr *) &new_addr, &len);
        if (new_fd < 0) {
                PERROR("accept inet");
                goto error;
        }
+       new_sock->sockaddr.addr.sin = new_addr;
        timeout = lttcomm_get_network_timeout();
        if (timeout) {
                int ret;
@@ -356,6 +350,7 @@ ssize_t lttcomm_recvmsg_inet_sock(struct lttcomm_sock *sock, void *buf,
        struct iovec iov[1];
        ssize_t ret = -1;
        size_t len_last;
+       struct sockaddr_in addr = sock->sockaddr.addr.sin;
 
        memset(&msg, 0, sizeof(msg));
 
@@ -364,7 +359,7 @@ ssize_t lttcomm_recvmsg_inet_sock(struct lttcomm_sock *sock, void *buf,
        msg.msg_iov = iov;
        msg.msg_iovlen = 1;
 
-       msg.msg_name = (struct sockaddr *) &sock->sockaddr.addr.sin;
+       msg.msg_name = (struct sockaddr *) &addr;
        msg.msg_namelen = sizeof(sock->sockaddr.addr.sin);
 
        do {
@@ -381,6 +376,15 @@ ssize_t lttcomm_recvmsg_inet_sock(struct lttcomm_sock *sock, void *buf,
        } while ((ret > 0 && ret < len_last) || (ret < 0 && errno == EINTR));
 
        if (ret < 0) {
+               if (errno == EAGAIN && flags & MSG_DONTWAIT) {
+                       /*
+                        * EAGAIN is expected in non-blocking mode and should
+                        * not be reported as an error. Moreover, if no data
+                        * was read, 0 must not be returned as it would be
+                        * interpreted as an orderly shutdown of the socket.
+                        */
+                       goto end;
+               }
                PERROR("recvmsg inet");
        } else if (ret > 0) {
                ret = len;
@@ -412,9 +416,13 @@ ssize_t lttcomm_sendmsg_inet_sock(struct lttcomm_sock *sock, const void *buf,
 
        switch (sock->proto) {
        case LTTCOMM_SOCK_UDP:
-               msg.msg_name = (struct sockaddr *) &sock->sockaddr.addr.sin;
+       {
+               struct sockaddr_in addr = sock->sockaddr.addr.sin;
+
+               msg.msg_name = (struct sockaddr *) &addr;
                msg.msg_namelen = sizeof(sock->sockaddr.addr.sin);
                break;
+       }
        default:
                break;
        }