cleanup and add fallback to read/write if no sendfile support.

This commit is contained in:
Isaac Connor 2022-02-16 14:16:25 -05:00 committed by Ben Dailey
parent 00b4ed9c53
commit e9b3269278
1 changed files with 29 additions and 22 deletions

View File

@ -3,37 +3,44 @@
#ifdef HAVE_SENDFILE4_SUPPORT
#include <sys/sendfile.h>
ssize_t zm_sendfile(int out_fd, int in_fd, off_t *offset, size_t size) {
size_t remaining = size;
while (remaining) {
ssize_t err = sendfile(out_fd, in_fd, offset, remaining);
if (err < 0) {
return -errno;
}
remaining -= err;
offset += err;
}
return size-remaining;
}
#elif HAVE_SENDFILE7_SUPPORT
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/uio.h>
ssize_t zm_sendfile(int out_fd, int in_fd, off_t *offset, off_t size) {
#else
#include <unistd.h>
#endif
/* Function to send the contents of a file. Will use sendfile or fall back to reading/writing */
ssize_t zm_sendfile(int out_fd, int in_fd, off_t *offset, size_t size) {
#ifdef HAVE_SENDFILE4_SUPPORT
ssize_t err = sendfile(out_fd, in_fd, offset, size);
if (err < 0) {
return -errno;
}
return err;
#elif HAVE_SENDFILE7_SUPPORT
ssize_t err = sendfile(in_fd, out_fd, *offset, size, nullptr, &size, 0);
if (err && errno != EAGAIN)
return -errno;
if (size) {
*offset += size;
return size;
#else
uint8_t buffer[size];
ssize_t err = read(in_fd, buffer, size);
if (err < 0) {
Error("Unable to read %zu bytes: %s", size, strerror(errno));
return -errno;
}
return -EAGAIN;
}
#else
#error "Your platform does not support sendfile. Sorry."
err = fwrite(out_fd, buffer, size);
if (err < 0) {
Error("Unable to write %zu bytes: %s", size, strerror(errno));
return -errno;
}
return err;
#endif
}
#endif // ZM_SENDFILE_H