X-Git-Url: http://git.archive.openwrt.org/?p=project%2Fuhttpd.git;a=blobdiff_plain;f=proc.c;h=d24649f78d1313321b73fbd45f3429854aa0d151;hp=e37176d8f1fb39c31ddad349c0e802a171ee2b46;hb=0590256080feb5e8f8742938efa539f2aec90601;hpb=27e60fb7bbea043a4d5972a05bda1146fac84da0

diff --git a/proc.c b/proc.c
index e37176d..d24649f 100644
--- a/proc.c
+++ b/proc.c
@@ -147,7 +147,9 @@ struct env_var *uh_get_process_vars(struct client *cl, struct path_info *pi)
 	snprintf(redirect_status, sizeof(redirect_status),
 		 "%d", req->redirect_status);
 	inet_ntop(cl->srv_addr.family, &cl->srv_addr.in, local_addr, sizeof(local_addr));
+	snprintf(local_port, sizeof(local_port), "%d", cl->srv_addr.port);
 	inet_ntop(cl->peer_addr.family, &cl->peer_addr.in, remote_addr, sizeof(remote_addr));
+	snprintf(remote_port, sizeof(remote_port), "%d", cl->peer_addr.port);
 
 	blobmsg_parse(hdr_policy, __HDR_MAX, tb, blob_data(data), blob_len(data));
 	for (i = 0; i < ARRAY_SIZE(proc_header_env); i++) {
@@ -168,7 +170,11 @@ struct env_var *uh_get_process_vars(struct client *cl, struct path_info *pi)
 
 static void proc_close_fds(struct client *cl)
 {
-	close(cl->dispatch.proc.r.sfd.fd.fd);
+	struct dispatch_proc *p = &cl->dispatch.proc;
+
+	close(p->r.sfd.fd.fd);
+	if (p->wrfd.fd >= 0)
+		close(p->wrfd.fd);
 }
 
 static void proc_handle_close(struct relay *r, int ret)
@@ -245,6 +251,16 @@ static void proc_write_cb(struct uloop_fd *fd, unsigned int events)
 	client_poll_post_data(cl);
 }
 
+static void proc_relay_write_cb(struct client *cl)
+{
+	struct dispatch_proc *p = &cl->dispatch.proc;
+
+	if (ustream_pending_data(cl->us, true))
+		return;
+
+	ustream_set_read_blocked(&p->r.sfd.stream, false);
+}
+
 static int proc_data_send(struct client *cl, const char *data, int len)
 {
 	struct dispatch_proc *p = &cl->dispatch.proc;
@@ -282,8 +298,8 @@ static int proc_data_send(struct client *cl, const char *data, int len)
 	return retlen;
 }
 
-bool uh_create_process(struct client *cl, struct path_info *pi,
-		       void (*cb)(struct client *cl, struct path_info *pi))
+bool uh_create_process(struct client *cl, struct path_info *pi, char *url,
+		       void (*cb)(struct client *cl, struct path_info *pi, char *url))
 {
 	struct dispatch *d = &cl->dispatch;
 	struct dispatch_proc *proc = &d->proc;
@@ -317,7 +333,7 @@ bool uh_create_process(struct client *cl, struct path_info *pi,
 		close(wfd[1]);
 
 		uh_close_fds();
-		cb(cl, pi);
+		cb(cl, pi, url);
 		exit(0);
 	}
 
@@ -331,6 +347,7 @@ bool uh_create_process(struct client *cl, struct path_info *pi,
 	d->close_fds = proc_close_fds;
 	d->data_send = proc_data_send;
 	d->data_done = proc_write_close;
+	d->write_cb = proc_relay_write_cb;
 	proc->r.header_cb = proc_handle_header;
 	proc->r.header_end = proc_handle_header_end;
 	proc->r.close = proc_handle_close;