http_get: preliminary *BSD sendfile() support (untested)

We can optimize this later to use SF_NODISKIO, but for
now it looks like the Linux version.

1 files changed, 29 insertions, 1 deletions

diff --git a/http_get.c b/http_get.c
index c58da34..cd55612 100644
--- a/http_get.c
+++ b/http_get.c
@@ -2,9 +2,37 @@
  * Copyright (C) 2012, Eric Wong <normalperson@yhbt.net>
  * License: GPLv3 or later (see COPYING for details)
  */
+
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/uio.h>
+
 #include "cmogstored.h"
 #include "http.h"
-#include <sys/sendfile.h>
+#ifdef __linux__
+#  include <sys/sendfile.h>
+#else
+/*
+ * make BSD sendfile look like Linux for now...
+ * we can support SF_NODISKIO later
+ */
+static ssize_t my_sendfile(int sockfd, int filefd, off_t *off, size_t count)
+{
+        int flags = 0;
+        off_t sbytes = 0;
+        int rc;
+
+        rc = sendfile(filefd, sockfd, *off, count, NULL, &sbytes, flags);
+        if (sbytes > 0) {
+                *off += sbytes;
+                return (ssize_t)sbytes;
+        }
+
+        return (ssize_t)rc;
+}
+#  define sendfile(out_fd, in_fd, offset, count) \
+          my_sendfile((out_fd),(in_fd),(offset),(count))
+#endif /* ! __linux__ */
 
 void mog_http_get_open(struct mog_http *http, char *buf, bool head_only)
 {