util: introduce safe versions of read & write that handle interrupted

systems calls.

1 files changed, 20 insertions, 0 deletions

diff --git a/util.c b/util.c
index 42e3d98..611d4b5 100644
--- a/util.c
+++ b/util.c
@@ -368,3 +368,23 @@ int bfs_minor(dev_t dev) {
 	return dev & 0xFF;
 #endif
 }
+
+ssize_t safe_read(int fd, void *buf, size_t nbytes) {
+	for (;;) {
+		ssize_t ret = read(fd, buf, nbytes);
+		if (ret < 0 && errno == EINTR) {
+			continue;
+		}
+		return ret;
+	}
+}
+
+ssize_t safe_write(int fd, const void *buf, size_t nbytes) {
+	for (;;) {
+		ssize_t ret = write(fd, buf, nbytes);
+		if (ret < 0 && errno == EINTR) {
+			continue;
+		}
+		return ret;
+	}
+}