/*
- * Copyright (C) 2005-2007 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2005-2009 Andre Noll <maan@systemlinux.org>
*
* Licensed under the GPL v2. For licencing details see COPYING.
*/
-/** \file write.c Paraslash's standalone wav/raw player */
+/** \file write.c Paraslash's standalone wav/raw player. */
+#include <regex.h>
#include <sys/types.h>
#include <dirent.h>
+#include <stdbool.h>
#include "para.h"
#include "string.h"
#include "write.cmdline.h"
#include "list.h"
#include "sched.h"
+#include "ggo.h"
#include "stdin.h"
#include "write.h"
#include "write_common.h"
#include "fd.h"
#include "error.h"
+#include "buffer_tree.h"
-/** \cond */
INIT_WRITE_ERRLISTS;
-/** \endcond */
-/**
- * check if given buffer contains a valid wave header
- */
+/** Check if given buffer contains a valid wave header. */
struct check_wav_task {
- /** the buffer to check */
+ /** The buffer to check. */
char *buf;
- /** number of bytes loaded in \a buf */
+ /** Number of bytes loaded in \a buf. */
size_t *loaded;
- /** non-zero if end of file was reached */
- int *eof;
- /** number of channels specified in wav header given by \a buf */
+ /** Non-zero if an error occurred or end of file was reached. */
+ int *input_error;
+ /** Number of channels specified in wav header given by \a buf. */
unsigned channels;
- /** samplerate specified in wav header given by \a buf */
+ /** Sample rate specified in wav header given by \a buf. */
unsigned samplerate;
- /** the task structure for this task */
+ /** The task structure used by the scheduler. */
struct task task;
};
-/**
- * delay writing until given time
- */
+enum check_wav_state {
+ CWS_NEED_HEADER,
+ CWS_HAVE_HEADER,
+ CWS_NO_HEADER,
+};
+
+struct check_wav_task_btr {
+ int state;
+ /** Number of channels specified in wav header given by \a buf. */
+ unsigned channels;
+ /** Sample rate specified in wav header given by \a buf. */
+ unsigned samplerate;
+ /** The task structure used by the scheduler. */
+ struct task task;
+ struct btr_node *btrn;
+};
+
+/** Delay writing until given time. */
struct initial_delay_task {
- /** the time the first data should be written out */
+ /** The time the first data should be written out. */
struct timeval start_time;
- /** the task structure for this task */
+ /** The task structure for this task. */
struct task task;
};
static struct write_args_info conf;
+
static struct stdin_task sit;
-static struct check_wav_task cwt;
-static struct initial_delay_task idt;
+
+static struct check_wav_task the_check_wav_task;
+static struct initial_delay_task the_initial_delay_task;
+
static struct writer_node_group *wng;
-/** length of a standard wav header */
+/** Length of a standard wav header. */
#define WAV_HEADER_LEN 44
/**
- * test if audio buffer contains a valid wave header
+ * Test if audio buffer contains a valid wave header.
*
* \return If not, return -E_NO_WAV_HEADER, otherwise, return zero. If
- * there is less than WAV_HEADER_LEN bytes awailable, return one.
+ * there is less than WAV_HEADER_LEN bytes available, return one.
*/
static void check_wav_pre_select(__a_unused struct sched *s, struct task *t)
{
- struct check_wav_task *wt = t->private_data;
+ struct check_wav_task *cwt = container_of(t, struct check_wav_task, task);
unsigned char *a;
+ int ret;
- if (*wt->loaded < WAV_HEADER_LEN) {
- t->ret = *wt->eof? -E_PREMATURE_END : 1;
+ if (*cwt->loaded < WAV_HEADER_LEN) {
+ if (*cwt->input_error < 0)
+ t->error = *cwt->input_error;
return;
}
- wt->channels = 2;
- wt->samplerate = 44100;
- a = (unsigned char*)wt->buf;
- t->ret = -E_NO_WAV_HEADER;
- if (a[0] != 'R' || a[1] != 'I' || a[2] != 'F' || a[3] != 'F')
+ cwt->channels = 2;
+ cwt->samplerate = 44100;
+ a = (unsigned char*)cwt->buf;
+ if (a[0] != 'R' || a[1] != 'I' || a[2] != 'F' || a[3] != 'F') {
+ PARA_NOTICE_LOG("wav header not found\n");
+ t->error = -E_NO_WAV_HEADER;
+ goto out;
+ }
+ cwt->channels = (unsigned) a[22];
+ cwt->samplerate = a[24] + (a[25] << 8) + (a[26] << 16) + (a[27] << 24);
+ *cwt->loaded -= WAV_HEADER_LEN;
+ memmove(cwt->buf, cwt->buf + WAV_HEADER_LEN, *cwt->loaded);
+ t->error = -E_WAV_HEADER_SUCCESS;
+ PARA_INFO_LOG("channels: %d, sample rate: %d\n", cwt->channels, cwt->samplerate);
+out:
+ wng->channels = &cwt->channels;
+ wng->samplerate = &cwt->samplerate;
+ ret = wng_open(wng);
+ if (ret < 0)
+ t->error = ret;
+ s->timeout.tv_sec = 0;
+ s->timeout.tv_usec = 1;
+}
+
+static void check_wav_pre_select_btr(__a_unused struct sched *s, struct task *t)
+{
+ struct check_wav_task_btr *cwt = container_of(t, struct check_wav_task_btr, task);
+
+ if (btr_get_input_queue_size(cwt->btrn) < WAV_HEADER_LEN)
return;
- wt->channels = (unsigned) a[22];
- wt->samplerate = a[24] + (a[25] << 8) + (a[26] << 16) + (a[27] << 24);
- *wt->loaded -= WAV_HEADER_LEN;
- memmove(wt->buf, wt->buf + WAV_HEADER_LEN, *wt->loaded);
- t->ret = -E_WAV_HEADER_SUCCESS;
- PARA_INFO_LOG("channels: %d, sample rate: %d\n", wt->channels, wt->samplerate);
+ s->timeout.tv_sec = 0;
+ s->timeout.tv_usec = 1;
+}
+
+static int check_wav_exec(struct btr_node *btrn, const char *cmd, char **result)
+{
+ struct check_wav_task_btr *cwt = btr_context(btrn);
+
+ if (!strcmp(cmd, "samplerate")) {
+ if (cwt->state != CWS_HAVE_HEADER)
+ return -ERRNO_TO_PARA_ERROR(ENAVAIL);
+ *result = make_message("%d", cwt->samplerate);
+ return 1;
+ }
+ if (!strcmp(cmd, "channels")) {
+ if (cwt->state != CWS_HAVE_HEADER)
+ return -ERRNO_TO_PARA_ERROR(ENAVAIL);
+ *result = make_message("%d", cwt->samplerate);
+ return 1;
+ }
+ return -ERRNO_TO_PARA_ERROR(ENOTSUP);
+}
+
+static void check_wav_post_select_btr(__a_unused struct sched *s, struct task *t)
+{
+ struct check_wav_task_btr *cwt = container_of(t, struct check_wav_task_btr, task);
+ unsigned char *a;
+ size_t sz = btr_get_input_queue_size(cwt->btrn);
+
+ t->error = 0;
+ if (cwt->state != CWS_NEED_HEADER)
+ goto out;
+ if (sz < WAV_HEADER_LEN) {
+ if (!btr_no_parent(cwt->btrn))
+ return;
+ if (sz != 0) {
+ cwt->state = CWS_NO_HEADER;
+ goto out;
+ }
+ t->error = -E_WRITE_EOF;
+ goto err;
+ }
+ cwt->channels = 2;
+ cwt->samplerate = 44100;
+ btr_next_buffer(cwt->btrn, (char **)&a);
+ if (a[0] != 'R' || a[1] != 'I' || a[2] != 'F' || a[3] != 'F') {
+ PARA_NOTICE_LOG("wav header not found\n");
+ cwt->state = CWS_NO_HEADER;
+ sprintf(t->status, "check wav: no header");
+ goto consume;
+ }
+ PARA_INFO_LOG("found wav header\n");
+ cwt->state = CWS_HAVE_HEADER;
+ sprintf(t->status, "check wav: have header");
+ cwt->channels = (unsigned) a[22];
+ cwt->samplerate = a[24] + (a[25] << 8) + (a[26] << 16) + (a[27] << 24);
+consume:
+ PARA_INFO_LOG("channels: %d, sample rate: %d\n", cwt->channels, cwt->samplerate);
+ btr_consume(cwt->btrn, WAV_HEADER_LEN);
+out:
+ if (sz) {
+ btr_pushdown(cwt->btrn);
+ s->timeout.tv_sec = 0;
+ s->timeout.tv_usec = 1;
+ } else {
+ if (btr_no_parent(cwt->btrn))
+ t->error = -E_WRITE_EOF;
+ }
+err:
+ if (t->error < 0)
+ btr_del_node(cwt->btrn);
}
static void initial_delay_pre_select(struct sched *s, struct task *t)
{
- struct initial_delay_task *dt = t->private_data;
+ struct initial_delay_task *idt = container_of(t, struct initial_delay_task, task);
struct timeval diff;
- t->ret = -E_NO_DELAY;
- if (!dt->start_time.tv_sec && !dt->start_time.tv_usec)
- return;
- t->ret = -E_DELAY_TIMEOUT;
- if (tv_diff(now, &dt->start_time, &diff) > 0)
- return;
- t->ret = 1;
+ if (!idt->start_time.tv_sec && !idt->start_time.tv_usec) {
+ t->error = -E_NO_DELAY;
+ goto register_check_wav;
+ }
+ if (tv_diff(now, &idt->start_time, &diff) > 0) {
+ t->error = -E_DELAY_TIMEOUT;
+ goto register_check_wav;
+ }
if (tv_diff(&s->timeout , &diff, NULL) > 0)
s->timeout = diff;
+ return;
+register_check_wav:
+ register_task(&the_check_wav_task.task);
+ s->timeout.tv_sec = 0;
+ s->timeout.tv_usec = 1;
}
-INIT_STDERR_LOGGING(conf.loglevel_arg)
+static int loglevel;
+INIT_STDERR_LOGGING(loglevel)
static struct writer_node_group *check_args(void)
{
int i, ret = -E_WRITE_SYNTAX;
struct writer_node_group *g = NULL;
+ struct initial_delay_task *idt = &the_initial_delay_task;
- if (conf.list_writers_given) {
- char *msg = NULL;
- FOR_EACH_WRITER(i) {
- char *tmp = make_message("%s%s%s",
- i? msg : "",
- i? " " : "",
- writer_names[i]);
- free(msg);
- msg = tmp;
- }
- fprintf(stderr, "%s\n", msg);
- free(msg);
- exit(EXIT_SUCCESS);
- }
+ loglevel = get_loglevel_by_name(conf.loglevel_arg);
if (conf.start_time_given) {
long unsigned sec, usec;
if (sscanf(conf.start_time_arg, "%lu:%lu",
&sec, &usec) != 2)
goto out;
- idt.start_time.tv_sec = sec;
- idt.start_time.tv_usec = usec;
+ idt->start_time.tv_sec = sec;
+ idt->start_time.tv_usec = usec;
}
if (!conf.writer_given) {
g = setup_default_wng();
conf.writer_arg[i], &writer_num);
if (!g->writer_nodes[i].conf)
goto out;
- g->writer_nodes[i].writer = &writers[writer_num];
+ g->writer_nodes[i].writer_num = writer_num;
}
ret = 1;
out:
return NULL;
}
-static void wng_event_handler(struct task *t)
+__noreturn static void print_help_and_die(void)
{
- struct writer_node_group *g = t->private_data;
+ int d = conf.detailed_help_given;
+ const char **p = d? write_args_info_detailed_help
+ : write_args_info_help;
- PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
- unregister_task(t);
- wng_close(g);
+ printf_or_die("%s\n\n", WRITE_CMDLINE_PARSER_PACKAGE "-"
+ WRITE_CMDLINE_PARSER_VERSION);
+ printf_or_die("%s\n\n", write_args_info_usage);
+ for (; *p; p++)
+ printf_or_die("%s\n", *p);
+ print_writer_helps(d);
+ exit(0);
}
-
-static void idt_event_handler(struct task *t)
+/*
+ TODO: check wav, initial delay, multiple writers, non-default writers
+ */
+static int main_btr(struct sched *s)
{
+ struct writer_node *wn = para_malloc(sizeof(*wn));
+ struct writer *w = writers + DEFAULT_WRITER;
int ret;
+ struct check_wav_task_btr _cwt, *cwt = &_cwt;
- PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
- unregister_task(t);
- wng->buf = sit.buf;
- wng->loaded = &sit.loaded;
- wng->input_eof = &sit.eof;
- wng->task.event_handler = wng_event_handler;
- wng->channels = &cwt.channels;
- wng->samplerate = &cwt.samplerate;
- ret = wng_open(wng);
- if (ret < 0) {
- PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
- exit(EXIT_FAILURE);
- }
-}
+ sit.btrn = btr_new_node("stdin", NULL /* stdin has no parent */, NULL, NULL);
+ stdin_set_defaults(&sit);
+ register_task(&sit.task);
-static void cwt_event_handler(struct task *t)
-{
- if (t->ret != -E_NO_WAV_HEADER && t->ret != -E_WAV_HEADER_SUCCESS) {
- PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t->ret));
- exit(EXIT_FAILURE);
- }
- PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
- unregister_task(t);
-// if (t->ret == -E_WAV_HEADER_SUCCESS) {
-// conf.channels_arg = cwt.channels;
-// conf.sample_rate_arg = cwt.sample_rate;
-// }
- idt.task.pre_select = initial_delay_pre_select;
- idt.task.private_data = &idt;
- idt.task.event_handler = idt_event_handler;
- sprintf(idt.task.status, "initial_delay");
- register_task(&idt.task);
+ cwt->state = CWS_NEED_HEADER;
+ cwt->btrn = btr_new_node("check wav", sit.btrn, check_wav_exec, cwt);
+ sprintf(cwt->task.status, "check wav");
+ cwt->task.pre_select = check_wav_pre_select_btr;
+ cwt->task.post_select = check_wav_post_select_btr;
+ register_task(&cwt->task);
+
+ wn->writer_num = DEFAULT_WRITER;
+ wn->conf = writers[DEFAULT_WRITER].parse_config("-B");
+ wn->btrn = btr_new_node("writer", cwt->btrn, NULL, NULL);
+ sprintf(wn->task.status, "some writer");
+ w->open(wn);
+ wn->task.post_select = w->post_select_btr;
+ wn->task.pre_select = w->pre_select_btr;
+ register_task(&wn->task);
+
+
+
+ s->default_timeout.tv_sec = 10;
+ s->default_timeout.tv_usec = 50000;
+ ret = schedule(s);
+ w->close(wn);
+ return ret;
}
/**
- * para_write's main function
+ * Para_write's main function.
*
- * \param argc the usual argument counter
- * \param argv the usual argument vector
+ * \param argc The usual argument counter.
+ * \param argv The usual argument vector.
*
* It registers the stdin task, the check_wav_task, the task for initial delay
* and all tasks for actually writing out the stream.
int main(int argc, char *argv[])
{
int ret = -E_WRITE_SYNTAX;
- struct sched s;
+ static struct sched s;
+ struct check_wav_task *cwt = &the_check_wav_task;
+ struct initial_delay_task *idt = &the_initial_delay_task;
+ writer_init();
write_cmdline_parser(argc, argv, &conf);
HANDLE_VERSION_FLAG("write", conf);
- init_supported_writers();
+ if (conf.help_given || conf.detailed_help_given)
+ print_help_and_die();
+ if (conf.buffer_tree_given) {
+ ret = main_btr(&s);
+ goto out;
+ }
wng = check_args();
if (!wng)
goto out;
stdin_set_defaults(&sit);
- if (conf.bufsize_given)
- sit.bufsize = conf.bufsize_arg;
- sit.buf = para_malloc(sit.bufsize),
+ ret = -ERRNO_TO_PARA_ERROR(EINVAL);
+ if (conf.bufsize_arg < 0)
+ goto out;
+ if (conf.bufsize_arg >= INT_MAX / 1024)
+ goto out;
+ sit.bufsize = conf.bufsize_arg * 1024;
+ sit.buf = para_malloc(sit.bufsize);
+
+ wng->bufp = &sit.buf;
+ wng->loaded = &sit.loaded;
+ wng->input_error = &sit.task.error;
+
register_task(&sit.task);
- cwt.task.pre_select = check_wav_pre_select;
- cwt.task.private_data = &cwt;
- cwt.task.event_handler = cwt_event_handler;
- cwt.buf = sit.buf;
- cwt.loaded = &sit.loaded;
- cwt.eof = &sit.eof;
- sprintf(cwt.task.status, "check wav");
- register_task(&cwt.task);
+ cwt->buf = sit.buf;
+ cwt->loaded = &sit.loaded;
+ cwt->input_error = &sit.task.error;
+ sprintf(cwt->task.status, "check wav");
+ cwt->task.pre_select = check_wav_pre_select;
- s.default_timeout.tv_sec = 1;
- s.default_timeout.tv_usec = 0;
- ret = sched(&s);
+ idt->task.pre_select = initial_delay_pre_select;
+ sprintf(idt->task.status, "initial_delay");
+ register_task(&idt->task);
+ s.default_timeout.tv_sec = 10;
+ s.default_timeout.tv_usec = 0;
+ ret = schedule(&s);
+ wng_close(wng);
out:
if (ret < 0) {
- PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
- ret = EXIT_FAILURE;
- } else
- ret = EXIT_SUCCESS;
- return ret;
+ PARA_ERROR_LOG("%s\n", para_strerror(-ret));
+ exit(EXIT_FAILURE);
+ }
+ exit(EXIT_SUCCESS);
}