Fix a bug in the mood line parser.
[paraslash.git] / write.c
1 /*
2 * Copyright (C) 2005-2007 Andre Noll <maan@systemlinux.org>
3 *
4 * Licensed under the GPL v2. For licencing details see COPYING.
5 */
6
7 /** \file write.c Paraslash's standalone wav/raw player */
8
9 #include <sys/types.h>
10 #include <dirent.h>
11
12 #include "para.h"
13 #include "string.h"
14 #include "write.cmdline.h"
15 #include "list.h"
16 #include "sched.h"
17 #include "stdin.h"
18 #include "write.h"
19 #include "write_common.h"
20 #include "fd.h"
21 #include "error.h"
22
23 /** \cond */
24 INIT_WRITE_ERRLISTS;
25 /** \endcond */
26
27 /**
28 * check if given buffer contains a valid wave header
29 */
30 struct check_wav_task {
31 /** the buffer to check */
32 char *buf;
33 /** number of bytes loaded in \a buf */
34 size_t *loaded;
35 /** non-zero if end of file was reached */
36 int *eof;
37 /** number of channels specified in wav header given by \a buf */
38 unsigned channels;
39 /** samplerate specified in wav header given by \a buf */
40 unsigned samplerate;
41 /** the task structure for this task */
42 struct task task;
43 };
44
45 /**
46 * delay writing until given time
47 */
48 struct initial_delay_task {
49 /** the time the first data should be written out */
50 struct timeval start_time;
51 /** the task structure for this task */
52 struct task task;
53 };
54
55 static struct write_args_info conf;
56 static struct stdin_task sit;
57 static struct check_wav_task cwt;
58 static struct initial_delay_task idt;
59 static struct writer_node_group *wng;
60
61 /** length of a standard wav header */
62 #define WAV_HEADER_LEN 44
63
64 /**
65 * test if audio buffer contains a valid wave header
66 *
67 * \return If not, return -E_NO_WAV_HEADER, otherwise, return zero. If
68 * there is less than WAV_HEADER_LEN bytes awailable, return one.
69 */
70 static void check_wav_pre_select(__a_unused struct sched *s, struct task *t)
71 {
72 struct check_wav_task *wt = t->private_data;
73 unsigned char *a;
74
75 if (*wt->loaded < WAV_HEADER_LEN) {
76 t->ret = *wt->eof? -E_PREMATURE_END : 1;
77 return;
78 }
79 wt->channels = 2;
80 wt->samplerate = 44100;
81 a = (unsigned char*)wt->buf;
82 t->ret = -E_NO_WAV_HEADER;
83 if (a[0] != 'R' || a[1] != 'I' || a[2] != 'F' || a[3] != 'F')
84 return;
85 wt->channels = (unsigned) a[22];
86 wt->samplerate = a[24] + (a[25] << 8) + (a[26] << 16) + (a[27] << 24);
87 *wt->loaded -= WAV_HEADER_LEN;
88 memmove(wt->buf, wt->buf + WAV_HEADER_LEN, *wt->loaded);
89 t->ret = -E_WAV_HEADER_SUCCESS;
90 PARA_INFO_LOG("channels: %d, sample rate: %d\n", wt->channels, wt->samplerate);
91 }
92
93 static void initial_delay_pre_select(struct sched *s, struct task *t)
94 {
95 struct initial_delay_task *dt = t->private_data;
96 struct timeval diff;
97
98 t->ret = -E_NO_DELAY;
99 if (!dt->start_time.tv_sec && !dt->start_time.tv_usec)
100 return;
101 t->ret = -E_DELAY_TIMEOUT;
102 if (tv_diff(now, &dt->start_time, &diff) > 0)
103 return;
104 t->ret = 1;
105 if (tv_diff(&s->timeout , &diff, NULL) > 0)
106 s->timeout = diff;
107 }
108
109 INIT_STDERR_LOGGING(conf.loglevel_arg)
110
111 static struct writer_node_group *check_args(void)
112 {
113 int i, ret = -E_WRITE_SYNTAX;
114 struct writer_node_group *g = NULL;
115
116 if (conf.list_writers_given) {
117 char *msg = NULL;
118 FOR_EACH_WRITER(i) {
119 char *tmp = make_message("%s%s%s",
120 i? msg : "",
121 i? " " : "",
122 writer_names[i]);
123 free(msg);
124 msg = tmp;
125 }
126 fprintf(stderr, "%s\n", msg);
127 free(msg);
128 exit(EXIT_SUCCESS);
129 }
130 if (conf.start_time_given) {
131 long unsigned sec, usec;
132 if (sscanf(conf.start_time_arg, "%lu:%lu",
133 &sec, &usec) != 2)
134 goto out;
135 idt.start_time.tv_sec = sec;
136 idt.start_time.tv_usec = usec;
137 }
138 if (!conf.writer_given) {
139 g = setup_default_wng();
140 ret = 1;
141 goto out;
142 }
143 g = wng_new(conf.writer_given);
144 ret = -E_WRITE_SYNTAX;
145 for (i = 0; i < conf.writer_given; i++) {
146 int writer_num;
147 g->writer_nodes[i].conf = check_writer_arg(
148 conf.writer_arg[i], &writer_num);
149 if (!g->writer_nodes[i].conf)
150 goto out;
151 g->writer_nodes[i].writer = &writers[writer_num];
152 }
153 ret = 1;
154 out:
155 if (ret > 0)
156 return g;
157 free(g);
158 return NULL;
159 }
160
161 static void wng_event_handler(struct task *t)
162 {
163 struct writer_node_group *g = t->private_data;
164
165 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
166 unregister_task(t);
167 wng_close(g);
168 }
169
170
171 static void idt_event_handler(struct task *t)
172 {
173 int ret;
174
175 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
176 unregister_task(t);
177 wng->buf = sit.buf;
178 wng->loaded = &sit.loaded;
179 wng->input_eof = &sit.eof;
180 wng->task.event_handler = wng_event_handler;
181 wng->channels = &cwt.channels;
182 wng->samplerate = &cwt.samplerate;
183 ret = wng_open(wng);
184 if (ret < 0) {
185 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
186 exit(EXIT_FAILURE);
187 }
188 }
189
190 static void cwt_event_handler(struct task *t)
191 {
192 if (t->ret != -E_NO_WAV_HEADER && t->ret != -E_WAV_HEADER_SUCCESS) {
193 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t->ret));
194 exit(EXIT_FAILURE);
195 }
196 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
197 unregister_task(t);
198 // if (t->ret == -E_WAV_HEADER_SUCCESS) {
199 // conf.channels_arg = cwt.channels;
200 // conf.sample_rate_arg = cwt.sample_rate;
201 // }
202 idt.task.pre_select = initial_delay_pre_select;
203 idt.task.private_data = &idt;
204 idt.task.event_handler = idt_event_handler;
205 sprintf(idt.task.status, "initial_delay");
206 register_task(&idt.task);
207 }
208
209 /**
210 * para_write's main function
211 *
212 * \param argc the usual argument counter
213 * \param argv the usual argument vector
214 *
215 * It registers the stdin task, the check_wav_task, the task for initial delay
216 * and all tasks for actually writing out the stream.
217 *
218 * \return \p EXIT_SUCCESS or EXIT_FAILURE
219 */
220 int main(int argc, char *argv[])
221 {
222 int ret = -E_WRITE_SYNTAX;
223 struct sched s;
224
225 write_cmdline_parser(argc, argv, &conf);
226 HANDLE_VERSION_FLAG("write", conf);
227 init_supported_writers();
228
229 wng = check_args();
230 if (!wng)
231 goto out;
232 stdin_set_defaults(&sit);
233 if (conf.bufsize_given)
234 sit.bufsize = conf.bufsize_arg;
235 sit.buf = para_malloc(sit.bufsize),
236 register_task(&sit.task);
237
238 cwt.task.pre_select = check_wav_pre_select;
239 cwt.task.private_data = &cwt;
240 cwt.task.event_handler = cwt_event_handler;
241 cwt.buf = sit.buf;
242 cwt.loaded = &sit.loaded;
243 cwt.eof = &sit.eof;
244 sprintf(cwt.task.status, "check wav");
245 register_task(&cwt.task);
246
247 s.default_timeout.tv_sec = 1;
248 s.default_timeout.tv_usec = 0;
249 ret = sched(&s);
250
251 out:
252 if (ret < 0) {
253 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
254 ret = EXIT_FAILURE;
255 } else
256 ret = EXIT_SUCCESS;
257 return ret;
258 }