X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=filter.c;h=608bb8b3bf5eedd8c1bac69d6454f857cb6efaeb;hp=c84b4823c745474f5ba92d59053f824cf946ca06;hb=95491e280363ddaed05599445138fd8191110dc1;hpb=607ac63646d7c31aa1792dcf1d9a60e498654376 diff --git a/filter.c b/filter.c index c84b4823..608bb8b3 100644 --- a/filter.c +++ b/filter.c @@ -21,16 +21,21 @@ #include "filter.cmdline.h" #include "list.h" +#include "sched.h" #include "filter.h" -#include "error.h" #include "string.h" +#include "stdin.h" +#include "stdout.h" +#include "error.h" INIT_FILTER_ERRLISTS; -#define INBUF_SIZE 32 * 1024 - +static struct stdin_task stdin_task_struct; +static struct stdin_task *sit = &stdin_task_struct; static struct filter_chain filter_chain_struct; static struct filter_chain *fc = &filter_chain_struct; +static struct stdout_task stdout_task_struct; +static struct stdout_task *sot = &stdout_task_struct; struct gengetopt_args_info conf; @@ -46,23 +51,44 @@ __printf_2_3 void para_log(int ll, const char* fmt,...) va_end(argp); } -static char *inbuf; -static size_t loaded; -static int eof; +void filter_event_handler(struct task *t) +{ + PARA_NOTICE_LOG("%s\n", PARA_STRERROR(-t->ret)); + unregister_task(t); +} -static int init_active_filter_list(void) +static void open_filters(void) +{ + struct filter_node *fn; + + list_for_each_entry(fn, &fc->filters, node) { + fn->filter->open(fn); + PARA_INFO_LOG("opened %s filter\n", fn->filter->name); + fc->outbuf = fn->buf; + fc->out_loaded = &fn->loaded; + } +} + + +static int init_filter_chain(void) { int i, filter_num; struct filter_node *fn; INIT_LIST_HEAD(&fc->filters); - fc->inbuf = inbuf; - fc->in_loaded = &loaded; - fc->eof = &eof; + fc->inbuf = sit->buf; + fc->in_loaded = &sit->loaded; + fc->input_eof = &sit->eof; + fc->eof = 0; + fc->output_eof = &sot->eof; + fc->task.private_data = fc; + fc->task.pre_select = filter_pre_select; + fc->task.event_handler = filter_event_handler; + sprintf(fc->task.status, "filter chain"); for (i = 0; i < conf.filter_given; i++) { - char *fa = para_strdup(conf.filter_arg[i]); + char *fa = conf.filter_arg[i]; fn = para_calloc(sizeof(struct filter_node)); filter_num = check_filter_arg(fa, &fn->conf); if (filter_num < 0) { @@ -77,21 +103,10 @@ static int init_active_filter_list(void) } if (list_empty(&fc->filters)) return -E_NO_FILTERS; + open_filters(); return 1; } -static void open_filters(void) -{ - struct filter_node *fn; - - list_for_each_entry(fn, &fc->filters, node) { - fn->filter->open(fn); - PARA_INFO_LOG("opened %s filter\n", fn->filter->name); - fc->outbuf = fn->buf; - fc->out_loaded = &fn->loaded; - } -} - static int parse_config(int argc, char *argv[]) { static char *cf; /* config file */ @@ -122,55 +137,36 @@ static int parse_config(int argc, char *argv[]) int main(int argc, char *argv[]) { - int converted, ret; - char *ib, *ob; /* input/output buffer */ - size_t *il, *ol; /* number of loaded bytes in input/output buffer */ + int ret; + struct sched s; + + init_sched(); + stdin_set_defaults(sit); + sit->buf = para_malloc(sit->bufsize), filter_init(filters); ret = parse_config(argc, argv); if (ret < 0) goto out; - inbuf = para_malloc(INBUF_SIZE); - ret = init_active_filter_list(); - if (ret < 0) - goto out; - open_filters(); - ib = fc->inbuf; - ob = fc->outbuf; - il = fc->in_loaded; - ol = fc->out_loaded; - PARA_DEBUG_LOG("ib %p in, ob: %p\n", ib, ob); -again: - if (*il < INBUF_SIZE && !eof) { - ret = read(STDIN_FILENO, ib + *il, INBUF_SIZE - *il); - PARA_DEBUG_LOG("read %d/%zd\n", ret, INBUF_SIZE - *il); - if (ret < 0) - goto out; - if (!ret) - eof = 1; - *il += ret; - } - ret = filter_io(fc); + ret = init_filter_chain(); if (ret < 0) goto out; - converted = ret; - if (*ol) { - ret = write(STDOUT_FILENO, ob, *ol); - PARA_DEBUG_LOG("wrote %d/%zd\n", ret, *ol); - if (ret <= 0) - goto out; - *ol -= ret; - if (*ol) { - PARA_NOTICE_LOG("short write: %zd bytes left\n", *ol); - memmove(ob, ob + ret, *ol); - } - } - if (!eof || converted) - goto again; - ret = 0; + + stdout_set_defaults(sot); + sot->buf = fc->outbuf; + sot->loaded = fc->out_loaded; + sot->input_eof = &fc->eof; + + register_task(&sot->task); + register_task(&fc->task); + register_task(&sit->task); + s.default_timeout.tv_sec = 1; + s.default_timeout.tv_usec = 0; + ret = sched(&s); out: + free(sit->buf); + close_filters(fc); if (ret < 0) PARA_EMERG_LOG("%s\n", PARA_STRERROR(-ret)); - close_filters(fc); - return ret; + return ret < 0? EXIT_FAILURE : EXIT_SUCCESS; }