http_recv/dccp_recv: simplify post_select()

[paraslash.git] / write_common.c

diff --git a/write_common.c b/write_common.c
index 50784cec6b8b7815c6969eb85b8088e35dd83dd1..a413d47e2dd0d3c75d4ef26f7a630325685c5d14 100644 (file)
--- a/write_common.c
+++ b/write_common.c
@@ -66,7 +66,7 @@ int wng_open(struct writer_node_group *g)
                wn->wng = g;
                ret = wn->writer->open(wn);
                if (ret < 0)
-                       goto out;
+                       goto err_out;
                wn->chunk_bytes = ret;
                g->max_chunk_bytes = PARA_MAX(g->max_chunk_bytes, ret);
                wn->task.pre_select = wn->writer->pre_select;
@@ -75,9 +75,17 @@ int wng_open(struct writer_node_group *g)
                register_task(&wn->task);
        }
        sprintf(g->task.status, "%s", "writer node group");
-       g->eof = 0;
        register_task(&g->task);
-out:
+       g->eof = 0;
+       return 1;
+err_out:
+       while (i > 0) {
+               struct writer_node *wn = &g->writer_nodes[--i];
+               unregister_task(&wn->task);
+               wn->writer->close(wn);
+       }
+       g->num_writers = 0;
+       g->eof = 1;
        return ret;
 }
 
@@ -90,6 +98,7 @@ void wng_unregister(struct writer_node_group *g)
                unregister_task(&wn->task);
        }
        unregister_task(&g->task);
+       g->eof = 1;
 }
 
 void wng_close(struct writer_node_group *g)