Adding upstream version 2.9.4.upstream/2.9.4

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 102 insertions, 0 deletions

diff --git a/src/util/atomic_io.c b/src/util/atomic_io.c
new file mode 100644
index 0000000..7cc4304
--- /dev/null
+++ b/src/util/atomic_io.c
@@ -0,0 +1,102 @@
+/*
+    Authors:
+        Jan Cholasta <jcholast@redhat.com>
+
+    Copyright (C) 2012 Red Hat
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 3 of the License, or
+    (at your option) any later version.
+
+    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, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <stdint.h>
+
+#include "util/atomic_io.h"
+
+/* based on code from libssh <http://www.libssh.org> */
+ssize_t sss_atomic_io_s(int fd, void *buf, size_t n, bool do_read)
+{
+    char *b = buf;
+    size_t pos = 0;
+    ssize_t res;
+    struct pollfd pfd;
+
+    pfd.fd = fd;
+    pfd.events = do_read ? POLLIN : POLLOUT;
+
+    while (n > pos) {
+        if (do_read) {
+            res = read(fd, b + pos, n - pos);
+        } else {
+            res = write(fd, b + pos, n - pos);
+        }
+        switch (res) {
+        case -1:
+            if (errno == EINTR) {
+                continue;
+            }
+            if (errno == EAGAIN || errno == EWOULDBLOCK) {
+                (void) poll(&pfd, 1, -1);
+                continue;
+            }
+            return -1;
+        case 0:
+            /* read returns 0 on end-of-file */
+            errno = do_read ? 0 : EPIPE;
+            return pos;
+        default:
+            pos += (size_t) res;
+        }
+    }
+
+    return pos;
+}
+
+ssize_t sss_atomic_write_safe_s(int fd, void *buf, size_t len)
+{
+    uint32_t ulen = len;
+    ssize_t ret;
+
+    ret = sss_atomic_write_s(fd, &ulen, sizeof(uint32_t));
+    if (ret != sizeof(uint32_t)) {
+        errno = errno == 0 ? EIO : errno;
+        return -1;
+    }
+
+    return sss_atomic_write_s(fd, buf, len);
+}
+
+ssize_t sss_atomic_read_safe_s(int fd, void *buf, size_t buf_len, size_t *_len)
+{
+    uint32_t ulen = (uint32_t)-1;
+    ssize_t ret;
+
+    ret = sss_atomic_read_s(fd, &ulen, sizeof(uint32_t));
+    if (ret != sizeof(uint32_t)) {
+        errno = errno == 0 ? EIO : errno;
+        return -1;
+    }
+
+    if (ulen > buf_len) {
+        if (_len != NULL) {
+            *_len = ulen;
+        }
+        errno = ERANGE;
+        return -1;
+    }
+
+    if (_len != NULL) {
+        *_len = ulen;
+    }
+
+    return sss_atomic_read_s(fd, buf, ulen);
+}