1 files changed, 5 insertions, 1 deletions
diff --git a/proc.c b/proc.c
index e360897..edfcc8f 100644
--- a/proc.c
+++ b/proc.c
@@ -265,6 +265,7 @@ static void proc_write_cb(struct uloop_fd *fd, unsigned int events)
 	struct client *cl = container_of(fd, struct client, dispatch.proc.wrfd);
 
 	client_poll_post_data(cl);
+	cl->dispatch.data_blocked = false;
 }
 
 static void proc_relay_write_cb(struct client *cl)
@@ -291,8 +292,10 @@ static int proc_data_send(struct client *cl, const char *data, int len)
 			if (errno == EINTR)
 				continue;
 
-			if (errno == EAGAIN || errno == EWOULDBLOCK)
+			if (errno == EAGAIN || errno == EWOULDBLOCK) {
+				cl->dispatch.data_blocked = true;
 				break;
+			}
 
 			/* consume all data */
 			ret = len;
@@ -366,6 +369,7 @@ bool uh_create_process(struct client *cl, struct path_info *pi, char *url,
 
 	proc->wrfd.fd = wfd[1];
 	uh_relay_open(cl, &proc->r, rfd[0], pid);
+	uloop_fd_add(&proc->wrfd, ULOOP_WRITE);
 
 	d->free = proc_free;
 	d->close_fds = proc_close_fds;