Added a really basic sendfile implementation.

diff --git a/src/.gitignore b/src/.gitignore
index bc7940f..8c14912 100644 (file)
--- a/src/.gitignore
+++ b/src/.gitignore
@@ -1,3 +1,4 @@
 /htparser
 /debugsink
 /dirplex
+/sendfile

diff --git a/src/Makefile.am b/src/Makefile.am
index e8e6c70..7b5457e 100644 (file)
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -1,4 +1,4 @@
-bin_PROGRAMS = htparser dirplex
+bin_PROGRAMS = htparser dirplex sendfile
 noinst_PROGRAMS = debugsink
 
 htparser_SOURCES = htparser.c

diff --git a/src/sendfile.c b/src/sendfile.c
new file mode 100644 (file)
index 0000000..76c0ad4
--- /dev/null
+++ b/src/sendfile.c
@@ -0,0 +1,109 @@
+/*
+    ashd - A Sane HTTP Daemon
+    Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <stdlib.h>
+#include <string.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <sys/stat.h>
+#include <stdint.h>
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+#include <utils.h>
+#include <log.h>
+#include <resp.h>
+
+static void passdata(int in, int out)
+{
+    int ret, len, off;
+    char *buf;
+    
+    buf = smalloc(65536);
+    while(1) {
+       len = read(in, buf, 65536);
+       if(len < 0) {
+           flog(LOG_ERR, "sendfile: could not read input: %s", strerror(errno));
+           break;
+       }
+       if(len == 0)
+           break;
+       for(off = 0; off < len; off += ret) {
+           ret = write(out, buf + off, len - off);
+           if(ret < 0) {
+               flog(LOG_ERR, "sendfile: could not write output: %s", strerror(errno));
+               break;
+           }
+       }
+    }
+    free(buf);
+}
+
+static int strrcmp(char *str, char *end)
+{
+    return(strcmp(str + strlen(str) - strlen(end), end));
+}
+
+static char *getmimetype(char *file, struct stat *sb)
+{
+    /* Rewrite with libmagic. */
+    if(!strrcmp(file, ".html"))
+       return("text/html");
+    if(!strrcmp(file, ".xhtml"))
+       return("application/xhtml+xml");
+    if(!strrcmp(file, ".txt"))
+       return("text/plain");
+    return("application/octet-stream");
+}
+
+int main(int argc, char **argv)
+{
+    char *file;
+    struct stat sb;
+    int fd;
+    
+    if(argc < 4) {
+       flog(LOG_ERR, "usage: sendfile METHOD URL REST");
+       exit(1);
+    }
+    if((file = getenv("REQ_X_ASH_FILE")) == NULL) {
+       flog(LOG_ERR, "sendfile: needs to be called with the X-Ash-File header");
+       exit(1);
+    }
+    if(*argv[3]) {
+       simpleerror(1, 404, "Not Found", "The requested URL has no corresponding resource.");
+       exit(0);
+    }
+    if(stat(file, &sb) || ((fd = open(file, O_RDONLY)) < 0)) {
+       flog(LOG_ERR, "sendfile: could not stat input file %s: %s", file, strerror(errno));
+       simpleerror(1, 500, "Internal Error", "The server could not access its own data.");
+       exit(1);
+    }
+    
+    printf("HTTP/1.1 200 OK\r\n");
+    printf("Content-Type: %s\r\n", getmimetype(file, &sb));
+    printf("Content-Length: %ji\r\n", (intmax_t)sb.st_size);
+    printf("\r\n");
+    fflush(stdout);
+    if(strcasecmp(argv[1], "head"))
+       passdata(fd, 1);
+    return(0);
+}