uhttpd: - abort file serving if client connection is lost (#7742) - don't send bad request headers twice

SVN-Revision: 22602
This commit is contained in:
Jo-Philipp Wich 2010-08-11 23:44:30 +00:00
parent b36a1274bb
commit 2d175dcdce
3 changed files with 50 additions and 44 deletions

View file

@ -8,7 +8,7 @@
include $(TOPDIR)/rules.mk include $(TOPDIR)/rules.mk
PKG_NAME:=uhttpd PKG_NAME:=uhttpd
PKG_RELEASE:=13 PKG_RELEASE:=14
PKG_BUILD_DIR := $(BUILD_DIR)/$(PKG_NAME) PKG_BUILD_DIR := $(BUILD_DIR)/$(PKG_NAME)
PKG_BUILD_DEPENDS := libcyassl liblua PKG_BUILD_DEPENDS := libcyassl liblua

View file

@ -97,36 +97,39 @@ static char * uh_file_header_lookup(struct http_request *req, const char *name)
return NULL; return NULL;
} }
#define ensure_ret(x) \
do { if( x < 0 ) return; } while(0)
static void uh_file_response_ok_hdrs(struct client *cl, struct http_request *req, struct stat *s) static void uh_file_response_ok_hdrs(struct client *cl, struct http_request *req, struct stat *s)
{ {
uh_http_sendf(cl, NULL, "Connection: close\r\n"); ensure_ret(uh_http_sendf(cl, NULL, "Connection: close\r\n"));
if( s ) if( s )
{ {
uh_http_sendf(cl, NULL, "ETag: %s\r\n", uh_file_mktag(s)); ensure_ret(uh_http_sendf(cl, NULL, "ETag: %s\r\n", uh_file_mktag(s)));
uh_http_sendf(cl, NULL, "Last-Modified: %s\r\n", uh_file_unix2date(s->st_mtime)); ensure_ret(uh_http_sendf(cl, NULL, "Last-Modified: %s\r\n", uh_file_unix2date(s->st_mtime)));
} }
uh_http_sendf(cl, NULL, "Date: %s\r\n", uh_file_unix2date(time(NULL))); ensure_ret(uh_http_sendf(cl, NULL, "Date: %s\r\n", uh_file_unix2date(time(NULL))));
} }
static void uh_file_response_200(struct client *cl, struct http_request *req, struct stat *s) static void uh_file_response_200(struct client *cl, struct http_request *req, struct stat *s)
{ {
uh_http_sendf(cl, NULL, "HTTP/%.1f 200 OK\r\n", req->version); ensure_ret(uh_http_sendf(cl, NULL, "HTTP/%.1f 200 OK\r\n", req->version));
uh_file_response_ok_hdrs(cl, req, s); uh_file_response_ok_hdrs(cl, req, s);
} }
static void uh_file_response_304(struct client *cl, struct http_request *req, struct stat *s) static void uh_file_response_304(struct client *cl, struct http_request *req, struct stat *s)
{ {
uh_http_sendf(cl, NULL, "HTTP/%.1f 304 Not Modified\r\n", req->version); ensure_ret(uh_http_sendf(cl, NULL, "HTTP/%.1f 304 Not Modified\r\n", req->version));
uh_file_response_ok_hdrs(cl, req, s); uh_file_response_ok_hdrs(cl, req, s);
} }
static void uh_file_response_412(struct client *cl, struct http_request *req) static void uh_file_response_412(struct client *cl, struct http_request *req)
{ {
uh_http_sendf(cl, NULL, ensure_ret(uh_http_sendf(cl, NULL,
"HTTP/%.1f 412 Precondition Failed\r\n" "HTTP/%.1f 412 Precondition Failed\r\n"
"Connection: close\r\n", req->version); "Connection: close\r\n", req->version));
} }
static int uh_file_if_match(struct client *cl, struct http_request *req, struct stat *s) static int uh_file_if_match(struct client *cl, struct http_request *req, struct stat *s)
@ -244,6 +247,9 @@ static int uh_file_if_unmodified_since(struct client *cl, struct http_request *r
} }
#define ensure_out(x) \
do { if( x < 0 ) goto out; } while(0)
static int uh_file_scandir_filter_dir(const struct dirent *e) static int uh_file_scandir_filter_dir(const struct dirent *e)
{ {
return strcmp(e->d_name, ".") ? 1 : 0; return strcmp(e->d_name, ".") ? 1 : 0;
@ -251,17 +257,18 @@ static int uh_file_scandir_filter_dir(const struct dirent *e)
static void uh_file_dirlist(struct client *cl, struct http_request *req, struct path_info *pi) static void uh_file_dirlist(struct client *cl, struct http_request *req, struct path_info *pi)
{ {
int i, count; int i;
int count = 0;
char filename[PATH_MAX]; char filename[PATH_MAX];
char *pathptr; char *pathptr;
struct dirent **files = NULL; struct dirent **files = NULL;
struct stat s; struct stat s;
uh_http_sendf(cl, req, ensure_out(uh_http_sendf(cl, req,
"<html><head><title>Index of %s</title></head>" "<html><head><title>Index of %s</title></head>"
"<body><h1>Index of %s</h1><hr /><ol>", "<body><h1>Index of %s</h1><hr /><ol>",
pi->name, pi->name pi->name, pi->name
); ));
if( (count = scandir(pi->phys, &files, uh_file_scandir_filter_dir, alphasort)) > 0 ) if( (count = scandir(pi->phys, &files, uh_file_scandir_filter_dir, alphasort)) > 0 )
{ {
@ -278,13 +285,13 @@ static void uh_file_dirlist(struct client *cl, struct http_request *req, struct
if( !stat(filename, &s) && if( !stat(filename, &s) &&
(s.st_mode & S_IFDIR) && (s.st_mode & S_IXOTH) (s.st_mode & S_IFDIR) && (s.st_mode & S_IXOTH)
) )
uh_http_sendf(cl, req, ensure_out(uh_http_sendf(cl, req,
"<li><strong><a href='%s%s'>%s</a>/</strong><br />" "<li><strong><a href='%s%s'>%s</a>/</strong><br />"
"<small>modified: %s<br />directory - %.02f kbyte" "<small>modified: %s<br />directory - %.02f kbyte"
"<br /><br /></small></li>", "<br /><br /></small></li>",
pi->name, files[i]->d_name, files[i]->d_name, pi->name, files[i]->d_name, files[i]->d_name,
uh_file_unix2date(s.st_mtime), s.st_size / 1024.0 uh_file_unix2date(s.st_mtime), s.st_size / 1024.0
); ));
*pathptr = 0; *pathptr = 0;
} }
@ -298,30 +305,37 @@ static void uh_file_dirlist(struct client *cl, struct http_request *req, struct
if( !stat(filename, &s) && if( !stat(filename, &s) &&
!(s.st_mode & S_IFDIR) && (s.st_mode & S_IROTH) !(s.st_mode & S_IFDIR) && (s.st_mode & S_IROTH)
) )
uh_http_sendf(cl, req, ensure_out(uh_http_sendf(cl, req,
"<li><strong><a href='%s%s'>%s</a></strong><br />" "<li><strong><a href='%s%s'>%s</a></strong><br />"
"<small>modified: %s<br />%s - %.02f kbyte<br />" "<small>modified: %s<br />%s - %.02f kbyte<br />"
"<br /></small></li>", "<br /></small></li>",
pi->name, files[i]->d_name, files[i]->d_name, pi->name, files[i]->d_name, files[i]->d_name,
uh_file_unix2date(s.st_mtime), uh_file_unix2date(s.st_mtime),
uh_file_mime_lookup(filename), s.st_size / 1024.0 uh_file_mime_lookup(filename), s.st_size / 1024.0
); ));
*pathptr = 0; *pathptr = 0;
free(files[i]);
} }
} }
free(files); ensure_out(uh_http_sendf(cl, req, "</ol><hr /></body></html>"));
ensure_out(uh_http_sendf(cl, req, ""));
uh_http_sendf(cl, req, "</ol><hr /></body></html>"); out:
uh_http_sendf(cl, req, ""); if( files )
{
for( i = 0; i < count; i++ )
free(files[i]);
free(files);
}
} }
void uh_file_request(struct client *cl, struct http_request *req, struct path_info *pi) void uh_file_request(struct client *cl, struct http_request *req, struct path_info *pi)
{ {
int fd, rlen; int rlen;
int fd = -1;
char buf[UH_LIMIT_MSGHEAD]; char buf[UH_LIMIT_MSGHEAD];
/* we have a file */ /* we have a file */
@ -338,38 +352,33 @@ void uh_file_request(struct client *cl, struct http_request *req, struct path_in
/* write status */ /* write status */
uh_file_response_200(cl, req, &pi->stat); uh_file_response_200(cl, req, &pi->stat);
uh_http_sendf(cl, NULL, "Content-Type: %s\r\n", uh_file_mime_lookup(pi->name)); ensure_out(uh_http_sendf(cl, NULL, "Content-Type: %s\r\n", uh_file_mime_lookup(pi->name)));
uh_http_sendf(cl, NULL, "Content-Length: %i\r\n", pi->stat.st_size); ensure_out(uh_http_sendf(cl, NULL, "Content-Length: %i\r\n", pi->stat.st_size));
/* if request was HTTP 1.1 we'll respond chunked */ /* if request was HTTP 1.1 we'll respond chunked */
if( (req->version > 1.0) && (req->method != UH_HTTP_MSG_HEAD) ) if( (req->version > 1.0) && (req->method != UH_HTTP_MSG_HEAD) )
uh_http_send(cl, NULL, "Transfer-Encoding: chunked\r\n", -1); ensure_out(uh_http_send(cl, NULL, "Transfer-Encoding: chunked\r\n", -1));
/* close header */ /* close header */
uh_http_send(cl, NULL, "\r\n", -1); ensure_out(uh_http_send(cl, NULL, "\r\n", -1));
/* send body */ /* send body */
if( req->method != UH_HTTP_MSG_HEAD ) if( req->method != UH_HTTP_MSG_HEAD )
{ {
/* pump file data */ /* pump file data */
while( (rlen = read(fd, buf, sizeof(buf))) > 0 ) while( (rlen = read(fd, buf, sizeof(buf))) > 0 )
{ ensure_out(uh_http_send(cl, req, buf, rlen));
if( uh_http_send(cl, req, buf, rlen) < 0 )
break;
}
/* send trailer in chunked mode */ /* send trailer in chunked mode */
uh_http_send(cl, req, "", 0); ensure_out(uh_http_send(cl, req, "", 0));
} }
} }
/* one of the preconditions failed, terminate opened header and exit */ /* one of the preconditions failed, terminate opened header and exit */
else else
{ {
uh_http_send(cl, NULL, "\r\n", -1); ensure_out(uh_http_send(cl, NULL, "\r\n", -1));
} }
close(fd);
} }
/* directory */ /* directory */
@ -379,9 +388,9 @@ void uh_file_request(struct client *cl, struct http_request *req, struct path_in
uh_file_response_200(cl, req, NULL); uh_file_response_200(cl, req, NULL);
if( req->version > 1.0 ) if( req->version > 1.0 )
uh_http_send(cl, NULL, "Transfer-Encoding: chunked\r\n", -1); ensure_out(uh_http_send(cl, NULL, "Transfer-Encoding: chunked\r\n", -1));
uh_http_send(cl, NULL, "Content-Type: text/html\r\n\r\n", -1); ensure_out(uh_http_send(cl, NULL, "Content-Type: text/html\r\n\r\n", -1));
/* content */ /* content */
uh_file_dirlist(cl, req, pi); uh_file_dirlist(cl, req, pi);
@ -390,8 +399,12 @@ void uh_file_request(struct client *cl, struct http_request *req, struct path_in
/* 403 */ /* 403 */
else else
{ {
uh_http_sendhf(cl, 403, "Forbidden", ensure_out(uh_http_sendhf(cl, 403, "Forbidden",
"Access to this resource is forbidden"); "Access to this resource is forbidden"));
} }
out:
if( fd > -1 )
close(fd);
} }

View file

@ -1001,13 +1001,6 @@ int main (int argc, char **argv)
} }
} }
/* 400 */
else
{
uh_http_sendhf(cl, 400, "Bad Request",
"Malformed request received");
}
#ifdef HAVE_TLS #ifdef HAVE_TLS
/* free client tls context */ /* free client tls context */
if( conf.tls ) if( conf.tls )