From: Hristo Venev <hristo@venev.name>
To: cgit@lists.zx2c4.com
Cc: Hristo Venev <hristo@venev.name>
Subject: [PATCH] cache: Tolerate short writes in print_slot
Date: Fri, 10 Sep 2021 17:18:41 +0300 [thread overview]
Message-ID: <20210910141841.2092532-1-hristo@venev.name> (raw)
sendfile() can return after a short read/write, so we may need to call
it more than once. Furthermore, not all files support sendfile(), so we
may need to fall back to read/write.
On the read/write path, use write_in_full which deals with short writes.
Signed-off-by: Hristo Venev <hristo@venev.name>
---
cache.c | 46 ++++++++++++++++++++++++----------------------
1 file changed, 24 insertions(+), 22 deletions(-)
diff --git a/cache.c b/cache.c
index 55199e8..85cfbd9 100644
--- a/cache.c
+++ b/cache.c
@@ -85,40 +85,42 @@ static int close_slot(struct cache_slot *slot)
/* Print the content of the active cache slot (but skip the key). */
static int print_slot(struct cache_slot *slot)
{
-#ifdef HAVE_LINUX_SENDFILE
- off_t start_off;
- int ret;
+ off_t off;
+ ssize_t i;
+
+ off = slot->keylen + 1;
- start_off = slot->keylen + 1;
+#ifdef HAVE_LINUX_SENDFILE
+ off_t size;
+ size = slot->cache_st.st_size;
do {
- ret = sendfile(STDOUT_FILENO, slot->cache_fd, &start_off,
- slot->cache_st.st_size - start_off);
- if (ret < 0) {
+ i = sendfile(STDOUT_FILENO, slot->cache_fd, &off, size - off);
+ if (i < 0) {
if (errno == EAGAIN || errno == EINTR)
continue;
+ /* Fall back to read/write on EINVAL */
+ if (errno == EINVAL)
+ break;
return errno;
}
- return 0;
+ if (off == size)
+ return 0;
} while (1);
-#else
- ssize_t i, j;
+#endif
- i = lseek(slot->cache_fd, slot->keylen + 1, SEEK_SET);
- if (i != slot->keylen + 1)
+ if (lseek(slot->cache_fd, off, SEEK_SET) != off)
return errno;
do {
- i = j = xread(slot->cache_fd, slot->buf, sizeof(slot->buf));
- if (i > 0)
- j = xwrite(STDOUT_FILENO, slot->buf, i);
- } while (i > 0 && j == i);
-
- if (i < 0 || j != i)
- return errno;
- else
- return 0;
-#endif
+ i = xread(slot->cache_fd, slot->buf, sizeof(slot->buf));
+ if (i < 0)
+ return errno;
+ if (i == 0)
+ return 0;
+ if (write_in_full(STDOUT_FILENO, slot->buf, i) < 0)
+ return errno;
+ } while (1);
}
/* Check if the slot has expired */
--
2.31.1
next reply other threads:[~2021-09-10 14:19 UTC|newest]
Thread overview: 6+ messages / expand[flat|nested] mbox.gz Atom feed top
2021-09-10 14:18 Hristo Venev [this message]
2021-10-07 9:35 ` John Keeping
2021-10-07 15:44 ` Hristo Venev
2022-05-07 21:28 ` Jason A. Donenfeld
2022-05-07 21:32 ` Hristo Venev
2022-05-07 21:38 ` Jason A. Donenfeld
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=20210910141841.2092532-1-hristo@venev.name \
--to=hristo@venev.name \
--cc=cgit@lists.zx2c4.com \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for NNTP newsgroup(s).