static bool uh_cgi_socket_cb(struct client *cl)
{
- int i, len, hdroff;
+ int i, len, blen, hdroff;
char buf[UH_LIMIT_MSGHEAD];
struct uh_cgi_state *state = (struct uh_cgi_state *)cl->priv;
}
/* try to read data from child */
- while ((len = uh_raw_recv(cl->rpipe.fd, buf, sizeof(buf), -1)) > 0)
+ while ((len = uh_raw_recv(cl->rpipe.fd, buf, state->header_sent
+ ? sizeof(buf) : state->httpbuf.len, -1)) > 0)
{
/* we have not pushed out headers yet, parse input */
if (!state->header_sent)
{
/* try to parse header ... */
- memcpy(state->httpbuf, buf, len);
+ memcpy(state->httpbuf.ptr, buf, len);
+ state->httpbuf.len -= len;
+ state->httpbuf.ptr += len;
- if (uh_cgi_header_parse(res, state->httpbuf, len, &hdroff))
+ blen = state->httpbuf.ptr - state->httpbuf.buf;
+
+ if (uh_cgi_header_parse(res, state->httpbuf.buf, blen, &hdroff))
{
/* write status */
ensure_out(uh_http_sendf(cl, NULL,
state->header_sent = true;
/* push out remaining head buffer */
- if (hdroff < len)
+ if (hdroff < blen)
{
D("CGI: Child(%d) relaying %d rest bytes\n",
- cl->proc.pid, len - hdroff);
+ cl->proc.pid, blen - hdroff);
ensure_out(uh_http_send(cl, req,
- &buf[hdroff], len - hdroff));
+ state->httpbuf.buf + hdroff,
+ blen - hdroff));
}
}
/* ... failed and head buffer exceeded */
- else
+ else if (!state->httpbuf.len)
{
/* I would do this ...
*
D("CGI: Child(%d) created: rfd(%d) wfd(%d)\n", child, rfd[0], wfd[1]);
+ state->httpbuf.ptr = state->httpbuf.buf;
+ state->httpbuf.len = sizeof(state->httpbuf.buf);
+
state->content_length = cl->httpbuf.len;
/* find content length */