X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=stdin.c;h=b52123d388a22617e66de7512413c3f7fc9d0a00;hp=5fc91c9de757c51e42746765de2620d4ee9cf0e6;hb=1995ceed8d70a5b50dbf84ca4132a8c318afa51b;hpb=e90440367f744a7723b201a95888e66f070ffa92 diff --git a/stdin.c b/stdin.c index 5fc91c9d..b52123d3 100644 --- a/stdin.c +++ b/stdin.c @@ -1,14 +1,12 @@ /* - * Copyright (C) 2006-2010 Andre Noll + * Copyright (C) 2006-2013 Andre Noll * * Licensed under the GPL v2. For licencing details see COPYING. */ /** \file stdin.c Functions that deal with reading from stdin. */ -#include /* readdir() */ #include -#include #include #include "para.h" @@ -35,7 +33,6 @@ static void stdin_pre_select(struct sched *s, struct task *t) struct stdin_task *sit = container_of(t, struct stdin_task, task); int ret; - t->error = 0; ret = btr_node_status(sit->btrn, 0, BTR_NT_ROOT); if (ret < 0) sched_min_delay(s); @@ -57,43 +54,36 @@ static void stdin_pre_select(struct sched *s, struct task *t) * appears to be readable, data is read from stdin and fed into the buffer * tree. */ -static void stdin_post_select(struct sched *s, struct task *t) +static int stdin_post_select(struct sched *s, struct task *t) { struct stdin_task *sit = container_of(t, struct stdin_task, task); ssize_t ret; - size_t sz; + size_t sz, n; char *buf = NULL; - t->error = 0; ret = btr_node_status(sit->btrn, 0, BTR_NT_ROOT); if (ret < 0) goto err; if (ret == 0) - return; - if (!FD_ISSET(STDIN_FILENO, &s->rfds)) - return; + return 0; sz = btr_pool_get_buffer(sit->btrp, &buf); if (sz == 0) - return; + return 0; /* * Do not use the maximal size to avoid having only a single buffer * reference for the whole pool. This is bad because if that single * reference can not be freed, we're stuck. */ sz = PARA_MIN(sz, btr_pool_size(sit->btrp) / 2); - ret = read(STDIN_FILENO, buf, sz); - if (ret < 0) - ret = -ERRNO_TO_PARA_ERROR(errno); - if (ret == 0) - ret = -E_STDIN_EOF; - if (ret < 0) - goto err; - btr_add_output_pool(sit->btrp, ret, sit->btrn); - return; + ret = read_nonblock(STDIN_FILENO, buf, sz, &s->rfds, &n); + if (n > 0) + btr_add_output_pool(sit->btrp, n, sit->btrn); + if (ret >= 0) + return 0; err: - btr_remove_node(sit->btrn); + btr_remove_node(&sit->btrn); //btr_pool_free(sit->btrp); - t->error = ret; + return ret; } /** @@ -110,8 +100,9 @@ void stdin_set_defaults(struct stdin_task *sit) int ret; sit->task.pre_select = stdin_pre_select; - sit->task.post_select = stdin_post_select; - sit->btrp = btr_pool_new("stdin", 64 * 1024); + sit->task.new_post_select = stdin_post_select; + sit->task.post_select = NULL; + sit->btrp = btr_pool_new("stdin", 128 * 1024); sprintf(sit->task.status, "stdin reader"); ret = mark_fd_nonblocking(STDIN_FILENO); if (ret >= 0)