src/util/atomic_io.c - master - githubmirror/sssd

src/util/atomic_io.c
102 строки3 KB
Justin Stephenson
util: Realloc buffer size for atomic safe read
27 сен 2023, 20:39
27 сен 2023, 20:391f4fffd
/*
    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);
}