X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=dccp_recv.c;h=0b62529f417352715d14304685565f1efeae581d;hp=85842248a97999b60824a267ed1180e0a8dafe0c;hb=48cf708771f7e3d43357ee2717b6052728518006;hpb=65d6f677cb6f79c69b8b35c5b3fe6aa337dbfb23 diff --git a/dccp_recv.c b/dccp_recv.c index 85842248..0b62529f 100644 --- a/dccp_recv.c +++ b/dccp_recv.c @@ -28,9 +28,6 @@ #include "dccp_recv.cmdline.h" -/** the size of the output buffer */ -#define DCCP_BUFSIZE 40960 - /** * data specific to the dccp receiver * @@ -50,8 +47,6 @@ static void dccp_recv_close(struct receiver_node *rn) if (pdd && pdd->fd > 0) close(pdd->fd); - free(rn->buf); - rn->buf = NULL; free(rn->private_data); rn->private_data = NULL; } @@ -80,7 +75,6 @@ static int dccp_recv_open(struct receiver_node *rn) ret = mark_fd_nonblocking(fd); if (ret < 0) goto err; - rn->buf = para_calloc(DCCP_BUFSIZE); rn->private_data = pdd = para_calloc(sizeof(struct private_dccp_recv_data)); pdd->btrp = btr_pool_new("dccp_recv", 320 * 1024); pdd->fd = fd; @@ -105,10 +99,9 @@ static void dccp_recv_pre_select(struct sched *s, struct task *t) struct receiver_node *rn = container_of(t, struct receiver_node, task); struct private_dccp_recv_data *pdd = rn->private_data; - if (rn->btrn) - if (generic_recv_pre_select(s, t) <= 0) - return; t->error = 0; + if (generic_recv_pre_select(s, t) <= 0) + return; para_fd_set(pdd->fd, &s->rfds, &s->max_fileno); } @@ -118,55 +111,29 @@ static void dccp_recv_post_select(struct sched *s, struct task *t) struct private_dccp_recv_data *pdd = rn->private_data; struct btr_node *btrn = rn->btrn; int ret; + char *buf; + size_t sz; - if (btrn) { - ret = btr_node_status(btrn, 0, BTR_NT_ROOT); - if (ret < 0) - goto err; - if (ret == 0) - return; - } else { - if (rn->output_error && *rn->output_error < 0) { - t->error = *rn->output_error; - return; - } - } + ret = btr_node_status(btrn, 0, BTR_NT_ROOT); + if (ret < 0) + goto err; + if (ret == 0) + return; if (!FD_ISSET(pdd->fd, &s->rfds)) return; /* nothing to do */ - if (rn->loaded >= DCCP_BUFSIZE) { - t->error = -E_DCCP_OVERRUN; - return; - } - if (btrn) { - char *buf; - size_t sz; - - sz = btr_pool_get_buffer(pdd->btrp, &buf); - ret = -E_DCCP_OVERRUN; - if (sz == 0) - goto err; - //buf = para_malloc(HTTP_RECV_READ_BUF_SIZE); - //sz = HTTP_RECV_READ_BUF_SIZE; - ret = recv_bin_buffer(pdd->fd, buf, sz); - if (ret == 0) - ret = -E_RECV_EOF; - if (ret < 0) - goto err; - btr_add_output_pool(pdd->btrp, ret, btrn); - return; - } - t->error = recv_bin_buffer(pdd->fd, rn->buf + rn->loaded, - DCCP_BUFSIZE - rn->loaded); - if (t->error > 0) { - rn->loaded += t->error; - return; - } - if (!t->error) - t->error = -E_RECV_EOF; + ret = -E_DCCP_OVERRUN; + sz = btr_pool_get_buffer(pdd->btrp, &buf); + if (sz == 0) + goto err; + ret = recv_bin_buffer(pdd->fd, buf, sz); + if (ret == 0) + ret = -E_RECV_EOF; + if (ret < 0) + goto err; + btr_add_output_pool(pdd->btrp, ret, btrn); return; err: - if (btrn) - btr_remove_node(rn->btrn); + btr_remove_node(rn->btrn); t->error = ret; }