X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=stdin.c;h=438883a76f5925d32a2e04810049902386019fce;hp=d33be3588b9211c8efa4a98991c9a6bd1b1ce2ca;hb=e6940831fbcf21dd0e6243ca89ea0c7f8b195d9e;hpb=704605b466aca3ae6ca6d1e03b6af55e3d245502 diff --git a/stdin.c b/stdin.c index d33be358..438883a7 100644 --- a/stdin.c +++ b/stdin.c @@ -1,3 +1,15 @@ +/* + * Copyright (C) 2006-2008 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 "para.h" #include "string.h" #include "list.h" @@ -6,50 +18,79 @@ #include "error.h" #include "stdin.h" -void stdin_pre_select(struct sched *s, struct task *t) -{ - struct stdin_task *sit = t->private_data; - if (sit->loaded < sit->bufsize) - para_fd_set(STDIN_FILENO, &s->rfds, &s->max_fileno); - t->ret = 1; /* success */ -} - -static void stdin_default_event_handler(struct task *t) +/** + * The pre_select function of the stdin task. + * + * \param s The scheduler this task was registered to. + * \param t The task structure of the stdin task. + * + * This function is always successful. If there is space left in the + * buffer of the stdin task, it adds \p STDIN_FILENO to the read fd set + * of \a s. + */ +static void stdin_pre_select(struct sched *s, struct task *t) { - PARA_NOTICE_LOG("%p: %s\n", t, PARA_STRERROR(-t->ret)); - unregister_task(t); + struct stdin_task *sit = container_of(t, struct stdin_task, task); + t->error = 0; + sit->check_fd = 0; + if (sit->loaded >= sit->bufsize) + return; + sit->check_fd = 1; + para_fd_set(STDIN_FILENO, &s->rfds, &s->max_fileno); } -void stdin_post_select(struct sched *s, struct task *t) +/** + * The post select function of the stdin task. + * + * \param s The scheduler this task was registered to. + * \param t The task structure of the stdin task. + * + * This function checks if \p STDIN_FILENO was included by in the read fd set + * of \a s during the previous pre_select call. If yes, and \p STDIN_FILENO + * appeears to be readable, data is read from stdin into the buffer of the + * stdin task. + */ +static void stdin_post_select(struct sched *s, struct task *t) { - struct stdin_task *sit = t->private_data; + struct stdin_task *sit = container_of(t, struct stdin_task, task); ssize_t ret; - t->ret = 1; - if (sit->loaded >= sit->bufsize) + t->error = 0; + if (!sit->check_fd) return; if (!FD_ISSET(STDIN_FILENO, &s->rfds)) return; ret = read(STDIN_FILENO, sit->buf + sit->loaded, sit->bufsize - sit->loaded); if (ret < 0) - t->ret = -E_STDIN_READ; - else if (ret > 0) { + t->error = ERRNO_TO_PARA_ERROR(errno); + else if (ret > 0) sit->loaded += ret; - t->ret = ret; - } else - t->ret = -E_STDIN_EOF; - if (t->ret < 0) - sit->eof = 1; + else + t->error = -E_STDIN_EOF; } +/** + * Initialize a stdin task structure with default values. + * + * \param sit The stdin task structure. + * + * This fills in the pre/post select function poinzters of the task structure + * given by \a sit. It also sets up a default error handler which unregisters + * the task. Moreover, \a loaded and \a eof are set to zero and \a bufsize is + * initialized to 16 KB (but no buffer is allocated). + */ void stdin_set_defaults(struct stdin_task *sit) { + int ret; + sit->bufsize = 16 * 1024, sit->loaded = 0, - sit->eof = 0, sit->task.pre_select = stdin_pre_select; sit->task.post_select = stdin_post_select; - sit->task.event_handler = stdin_default_event_handler; - sit->task.private_data = sit; sprintf(sit->task.status, "stdin reader"); + ret = mark_fd_nonblocking(STDIN_FILENO); + if (ret >= 0) + return; + PARA_EMERG_LOG("%s\n", para_strerror(-ret)); + exit(EXIT_FAILURE); }