playground/suckless-quark
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

added basic HEAD support, some bug fixes

Browse source
This commit is contained in:
Anselm R Garbe 2010-03-28 01:57:44 +00:00
parent 07db66c273
commit a20243ea7c
2 changed files with 84 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
config.def.h
View file

@ -6,7 +6,8 @@ static const char docroot[] = ".";
static const char docindex[] = "index.html"; static const char docindex[] = "index.html";
static const char user[] = "nobody"; static const char user[] = "nobody";
static const char group[] = "nobody"; static const char group[] = "nobody";
static const char cgi_script[] = "/var/www/werc-dev/bin/werc.rc"; static const char cgi_dir[] = "/var/www/werc-dev/bin";
static const char cgi_script[] = "./werc.rc";
static const int cgi_mode = 0; static const int cgi_mode = 0;
static const MimeType servermimes[] = { static const MimeType servermimes[] = {

103
quark.c
View file

@ -20,6 +20,11 @@
#define LENGTH(x) (sizeof x / sizeof x[0]) #define LENGTH(x) (sizeof x / sizeof x[0])
#define MAXREQLEN 255 #define MAXREQLEN 255
enum {
GET = 4,
HEAD = 5,
};
typedef struct { typedef struct {
const char *extension; const char *extension;
const char *mimetype; const char *mimetype;
@ -31,7 +36,8 @@ static const char HttpUnauthorized[] = "401 Unauthorized";
static const char HttpNotFound[] = "404 Not Found"; static const char HttpNotFound[] = "404 Not Found";
static const char texthtml[] = "text/html"; static const char texthtml[] = "text/html";
static ssize_t writedata(const char *buf); static ssize_t writetext(const char *buf);
static ssize_t writedata(const char *buf, size_t buflen);
static void die(const char *errstr, ...); static void die(const char *errstr, ...);
static void response(void); static void response(void);
static void responsecgi(void); static void responsecgi(void);
@ -51,15 +57,14 @@ static int running = 1;
static char name[128]; static char name[128];
static char reqbuf[MAXREQLEN+1]; static char reqbuf[MAXREQLEN+1];
static char respbuf[1024]; static char respbuf[1024];
static int fd, cfd; static int fd, cfd, reqtype;
ssize_t ssize_t
writedata(const char *buf) { writedata(const char *buf, size_t buf_len) {
ssize_t r, offset = 0; ssize_t r, offset = 0;
size_t len = strlen(buf);
while(offset < len) { while(offset < buf_len) {
if((r = write(cfd, buf + offset, len - offset)) == -1) { if((r = write(cfd, buf + offset, buf_len - offset)) == -1) {
fprintf(stderr, "%s: client %s closed connection\n", tstamp(), name); fprintf(stderr, "%s: client %s closed connection\n", tstamp(), name);
return -1; return -1;
} }
@ -68,6 +73,11 @@ writedata(const char *buf) {
return offset; return offset;
} }
ssize_t
writetext(const char *buf) {
return writedata(buf, strlen(buf));
}
void void
logmsg(const char *errstr, ...) { logmsg(const char *errstr, ...) {
va_list ap; va_list ap;
@ -111,7 +121,7 @@ responsehdr(const char *status) {
logerrmsg("snprintf failed, buffer size exceeded"); logerrmsg("snprintf failed, buffer size exceeded");
return -1; return -1;
} }
return writedata(respbuf); return writetext(respbuf);
} }
int int
@ -123,7 +133,7 @@ responsecontentlen(off_t size) {
logerrmsg("snprintf failed, buffer sizeof exceeded"); logerrmsg("snprintf failed, buffer sizeof exceeded");
return -1; return -1;
} }
return writedata(respbuf); return writetext(respbuf);
} }
int int
@ -135,7 +145,7 @@ responselocation(const char *location, const char *pathinfo) {
logerrmsg("snprintf failed, buffer sizeof exceeded"); logerrmsg("snprintf failed, buffer sizeof exceeded");
return -1; return -1;
} }
return writedata(respbuf); return writetext(respbuf);
} }
int int
@ -147,7 +157,7 @@ responsecontenttype(const char *mimetype) {
logerrmsg("snprintf failed, buffer sizeof exceeded"); logerrmsg("snprintf failed, buffer sizeof exceeded");
return -1; return -1;
} }
return writedata(respbuf); return writetext(respbuf);
} }
void void
@ -172,8 +182,12 @@ responsefile(void) {
if((ffd = open(reqbuf, O_RDONLY)) == -1) { if((ffd = open(reqbuf, O_RDONLY)) == -1) {
fprintf(stderr, "%s: %s requests unknown path %s\n", tstamp(), name, reqbuf); fprintf(stderr, "%s: %s requests unknown path %s\n", tstamp(), name, reqbuf);
if(responsehdr(HttpNotFound) != -1 if(responsehdr(HttpNotFound) != -1
&& responsecontenttype(texthtml) != -1 && responsecontenttype(texthtml) != -1)
&& writedata("\r\n<html><body>404 Not Found</body></html>\r\n") != -1); ;
else
return;
if(reqtype == GET)
writetext("\r\n<html><body>404 Not Found</body></html>\r\n");
} }
else { else {
if((p = strrchr(reqbuf, '.'))) { if((p = strrchr(reqbuf, '.'))) {
@ -186,8 +200,11 @@ responsefile(void) {
} }
if(responsehdr(HttpOk) != -1 if(responsehdr(HttpOk) != -1
&& responsecontentlen(st.st_size) != -1 && responsecontentlen(st.st_size) != -1
&& responsecontenttype(mimetype) != -1 && responsecontenttype(mimetype) != -1)
&& writedata("\r\n") != -1) ;
else
return;
if(reqtype == GET && writetext("\r\n") != -1)
responsefiledata(ffd, st.st_size); responsefiledata(ffd, st.st_size);
close(ffd); close(ffd);
} }
@ -198,10 +215,13 @@ responsedirdata(DIR *d) {
struct dirent *e; struct dirent *e;
if(responsehdr(HttpOk) != -1 if(responsehdr(HttpOk) != -1
&& responsecontenttype(texthtml) != -1 && responsecontenttype(texthtml) != -1)
&& writedata("\r\n<html><body><a href='..'>..</a><br>\r\n") != -1); ;
else else
return; return;
if(reqtype == GET) {
if(writetext("\r\n<html><body><a href='..'>..</a><br>\r\n") == -1)
return;
while((e = readdir(d))) { while((e = readdir(d))) {
if(e->d_name[0] == '.') /* ignore hidden files, ., .. */ if(e->d_name[0] == '.') /* ignore hidden files, ., .. */
continue; continue;
@ -211,10 +231,11 @@ responsedirdata(DIR *d) {
logerrmsg("snprintf failed, buffer sizeof exceeded"); logerrmsg("snprintf failed, buffer sizeof exceeded");
return; return;
} }
if(writedata(respbuf) == -1) if(writetext(respbuf) == -1)
return; return;
} }
writedata("</body></html>\r\n"); writetext("</body></html>\r\n");
}
} }
void void
@ -229,8 +250,12 @@ responsedir(void) {
fprintf(stdout, "%s: redirecting %s to %s%s\n", tstamp(), name, location, reqbuf); fprintf(stdout, "%s: redirecting %s to %s%s\n", tstamp(), name, location, reqbuf);
if(responsehdr(HttpMoved) != -1 if(responsehdr(HttpMoved) != -1
&& responselocation(location, reqbuf) != -1 && responselocation(location, reqbuf) != -1
&& responsecontenttype(texthtml) != -1 && responsecontenttype(texthtml) != -1)
&& writedata("\r\n<html><body>301 Moved Permanently</a></body></html>\r\n") != -1); ;
else
return;
if(reqtype == GET)
writetext("\r\n<html><body>301 Moved Permanently</a></body></html>\r\n");
return; return;
} }
if(len + strlen(docindex) + 1 < MAXREQLEN) if(len + strlen(docindex) + 1 < MAXREQLEN)
@ -251,15 +276,21 @@ responsecgi(void) {
FILE *cgi; FILE *cgi;
int r; int r;
if(reqtype == GET)
setenv("REQUEST_METHOD", "GET", 1); setenv("REQUEST_METHOD", "GET", 1);
else if(reqtype == HEAD)
setenv("REQUEST_METHOD", "HEAD", 1);
else
return;
setenv("SERVER_NAME", servername, 1); setenv("SERVER_NAME", servername, 1);
setenv("SCRIPT_NAME", cgi_script, 1); setenv("SCRIPT_NAME", cgi_script, 1);
setenv("REQUEST_URI", reqbuf, 1);
chdir(cgi_dir);
if((cgi = popen(cgi_script, "r"))) { if((cgi = popen(cgi_script, "r"))) {
if(responsehdr(HttpOk) == -1) if(responsehdr(HttpOk) == -1)
return; return;
while((r = fread(respbuf, 1, sizeof respbuf - 1, cgi) > 0)) { while((r = fread(respbuf, 1, sizeof respbuf - 1, cgi)) > 0) {
respbuf[r] = 0; if(writedata(respbuf, r) == -1) {
if(writedata(respbuf) == -1) {
pclose(cgi); pclose(cgi);
return; return;
} }
@ -269,8 +300,12 @@ responsecgi(void) {
else { else {
fprintf(stderr, "%s: %s requests %s, but cannot run cgi script %s\n", tstamp(), name, cgi_script, reqbuf); fprintf(stderr, "%s: %s requests %s, but cannot run cgi script %s\n", tstamp(), name, cgi_script, reqbuf);
if(responsehdr(HttpNotFound) != -1 if(responsehdr(HttpNotFound) != -1
&& responsecontenttype(texthtml) != -1 && responsecontenttype(texthtml) != -1)
&& writedata("\r\n<html><body>404 Not Found</body></html>\r\n") != -1); ;
else
return;
if(reqtype == GET)
writetext("\r\n<html><body>404 Not Found</body></html>\r\n");
} }
} }
@ -283,8 +318,12 @@ response(void) {
if(*p == '\\' || *p == '?' || *p == '%' || *p == '&' || (*p == '/' && *(p + 1) == '.')) { /* don't serve bogus or hidden files */ if(*p == '\\' || *p == '?' || *p == '%' || *p == '&' || (*p == '/' && *(p + 1) == '.')) { /* don't serve bogus or hidden files */
fprintf(stderr, "%s: %s requests bogus or hidden file %s\n", tstamp(), name, reqbuf); fprintf(stderr, "%s: %s requests bogus or hidden file %s\n", tstamp(), name, reqbuf);
if(responsehdr(HttpUnauthorized) != -1 if(responsehdr(HttpUnauthorized) != -1
&& responsecontenttype(texthtml) != -1 && responsecontenttype(texthtml) != -1)
&& writedata("\r\n<html><body>401 Unauthorized</body></html>\r\n") != -1); ;
else
return;
if(reqtype == GET)
writetext("\r\n<html><body>401 Unauthorized</body></html>\r\n");
return; return;
} }
fprintf(stdout, "%s: %s requests: %s\n", tstamp(), name, reqbuf); fprintf(stdout, "%s: %s requests: %s\n", tstamp(), name, reqbuf);
@ -313,20 +352,24 @@ request(void) {
offset += r; offset += r;
reqbuf[offset] = 0; reqbuf[offset] = 0;
} }
while(offset < MAXREQLEN && (!strstr(reqbuf, "\r\n\r\n") || !strstr(reqbuf, "\n\n"))); while(offset < MAXREQLEN && !strstr(reqbuf, "\r\n\r\n") && !strstr(reqbuf, "\n\n"));
for(p = reqbuf; *p && *p != '\r' && *p != '\n'; p++); for(p = reqbuf; *p && *p != '\r' && *p != '\n'; p++);
if(p >= reqbuf + MAXREQLEN) if(p >= reqbuf + MAXREQLEN)
goto invalid_request; goto invalid_request;
if(*p == '\r' || *p == '\n') { if(*p == '\r' || *p == '\n') {
*p = 0; *p = 0;
/* check command */ /* check command */
if(strncmp(reqbuf, "GET ", 4) || reqbuf[4] != '/') if(!strncmp(reqbuf, "GET ", 4) && reqbuf[4] == '/')
reqtype = GET;
else if(!strncmp(reqbuf, "HEAD ", 5) && reqbuf[5] != '/')
reqtype = HEAD;
else
goto invalid_request; goto invalid_request;
} }
else else
goto invalid_request; goto invalid_request;
/* determine path */ /* determine path */
for(res = reqbuf + 4; *res && *(res + 1) == '/'; res++); /* strip '/' */ for(res = reqbuf + reqtype; *res && *(res + 1) == '/'; res++); /* strip '/' */
if(res >= reqbuf + MAXREQLEN) if(res >= reqbuf + MAXREQLEN)
goto invalid_request; goto invalid_request;
for(p = res; *p && *p != ' ' && *p != '\t'; p++); for(p = res; *p && *p != ' ' && *p != '\t'; p++);