From 3dad2125b9048fdc3790d3e7c4770f7174be889c Mon Sep 17 00:00:00 2001 From: "Markus F.X.J. Oberhumer" Date: Thu, 15 Apr 2021 18:37:32 +0200 Subject: util: add safe_read_all() and safe_write_all() functions. --- util.c | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) (limited to 'util.c') diff --git a/util.c b/util.c index 611d4b5..00f5b78 100644 --- a/util.c +++ b/util.c @@ -379,6 +379,23 @@ ssize_t safe_read(int fd, void *buf, size_t nbytes) { } } +ssize_t safe_read_all(int fd, void *buf, size_t nbytes) { + size_t count = 0; + for (;;) { + ssize_t ret = read(fd, (char *)buf + count, nbytes - count); + if (ret < 0 && errno == EINTR) { + continue; + } + if (ret < 0) { + return ret; // always return error < 0 + } + count += ret; + if (ret == 0 || count == nbytes) { // EOF or success + return count; + } + } +} + ssize_t safe_write(int fd, const void *buf, size_t nbytes) { for (;;) { ssize_t ret = write(fd, buf, nbytes); @@ -388,3 +405,21 @@ ssize_t safe_write(int fd, const void *buf, size_t nbytes) { return ret; } } + +ssize_t safe_write_all(int fd, const void *buf, size_t nbytes) +{ + size_t count = 0; + for (;;) { + ssize_t ret = write(fd, (const char *)buf + count, nbytes - count); + if (ret < 0 && errno == EINTR) { + continue; + } + if (ret < 0) { + return ret; // always return error < 0 + } + count += ret; + if (ret == 0 || count == nbytes) { // EOF (should never happen with write) or success + return count; + } + } +} -- cgit v1.2.3