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