summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--uhttpd-cgi.c12
-rw-r--r--uhttpd-lua.c12
2 files changed, 14 insertions, 10 deletions
diff --git a/uhttpd-cgi.c b/uhttpd-cgi.c
index 4a30f2f..8bd2250 100644
--- a/uhttpd-cgi.c
+++ b/uhttpd-cgi.c
@@ -360,6 +360,9 @@ void uh_cgi_request(struct client *cl, struct http_request *req, struct path_inf
memset(hdr, 0, sizeof(hdr));
+ timeout.tv_sec = cl->server->conf->script_timeout;
+ timeout.tv_usec = 0;
+
#define ensure(x) \
do { if( x < 0 ) goto out; } while(0)
@@ -372,12 +375,11 @@ void uh_cgi_request(struct client *cl, struct http_request *req, struct path_inf
FD_SET(rfd[0], &reader);
FD_SET(wfd[1], &writer);
- timeout.tv_sec = cl->server->conf->script_timeout;
- timeout.tv_usec = 0;
-
/* wait until we can read or write or both */
- if( select(fd_max, &reader, (content_length > -1) ? &writer : NULL, NULL, &timeout) > 0 )
- {
+ if( select(fd_max, &reader,
+ (content_length > -1) ? &writer : NULL, NULL,
+ (header_sent < 1) ? &timeout : NULL) > 0
+ ) {
/* ready to write to cgi program */
if( FD_ISSET(wfd[1], &writer) )
{
diff --git a/uhttpd-lua.c b/uhttpd-lua.c
index c1e7490..fcbdc64 100644
--- a/uhttpd-lua.c
+++ b/uhttpd-lua.c
@@ -439,6 +439,9 @@ void uh_lua_request(struct client *cl, struct http_request *req, lua_State *L)
data_sent = 0;
+ timeout.tv_sec = cl->server->conf->script_timeout;
+ timeout.tv_usec = 0;
+
/* I/O loop, watch our pipe ends and dispatch child reads/writes from/to socket */
while( 1 )
{
@@ -448,12 +451,11 @@ void uh_lua_request(struct client *cl, struct http_request *req, lua_State *L)
FD_SET(rfd[0], &reader);
FD_SET(wfd[1], &writer);
- timeout.tv_sec = cl->server->conf->script_timeout;
- timeout.tv_usec = 0;
-
/* wait until we can read or write or both */
- if( select(fd_max, &reader, (content_length > -1) ? &writer : NULL, NULL, &timeout) > 0 )
- {
+ if( select(fd_max, &reader,
+ (content_length > -1) ? &writer : NULL, NULL,
+ (data_sent < 1) ? &timeout : NULL) > 0
+ ) {
/* ready to write to Lua child */
if( FD_ISSET(wfd[1], &writer) )
{