Make the wng->eof a pointer.
[paraslash.git] / write.c
1 /*
2 * Copyright (C) 2005-2006 Andre Noll <maan@systemlinux.org>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
17 */
18
19 #include "para.h"
20 #include "string.h"
21 #include "write.cmdline.h"
22 #include "list.h"
23 #include "sched.h"
24 #include "stdin.h"
25 #include "write.h"
26 #include "write_common.h"
27 #include "fd.h"
28 #include "error.h"
29
30 INIT_WRITE_ERRLISTS;
31
32 struct check_wav_task {
33 char *buf;
34 size_t *loaded;
35 unsigned channels;
36 unsigned sample_rate;
37 struct task task;
38 };
39
40 struct initial_delay_task {
41 struct timeval start_time;
42 struct task task;
43 };
44
45 struct gengetopt_args_info conf;
46 struct stdin_task sit;
47 struct check_wav_task cwt;
48 struct initial_delay_task idt;
49 static struct writer_node_group *wng;
50
51 #define WAV_HEADER_LEN 44
52
53 /**
54 * test if audio buffer contains a valid wave header
55 *
56 * \return If not, return 0, otherwise, store number of channels and sample rate
57 * in struct conf and return WAV_HEADER_LEN.
58 */
59 static void check_wav_pre_select(struct sched *s, struct task *t)
60 {
61 struct check_wav_task *cwt = t->private_data;
62 unsigned char *a;
63
64 if (*cwt->loaded < WAV_HEADER_LEN) {
65 t->ret = 1;
66 return;
67 }
68 a = (unsigned char*)cwt->buf;
69 t->ret = -E_NO_WAV_HEADER;
70 if (a[0] != 'R' || a[1] != 'I' || a[2] != 'F' || a[3] != 'F')
71 return;
72 cwt->channels = (unsigned) a[22];
73 cwt->sample_rate = a[24] + (a[25] << 8) + (a[26] << 16) + (a[27] << 24);
74 *cwt->loaded -= WAV_HEADER_LEN;
75 memmove(cwt->buf, cwt->buf + WAV_HEADER_LEN, *cwt->loaded);
76 t->ret = 0;
77 PARA_INFO_LOG("channels: %d, sample_rate: %d\n", cwt->channels, cwt->sample_rate);
78 }
79
80 static void initial_delay_pre_select(struct sched *s, struct task *t)
81 {
82 struct initial_delay_task *idt = t->private_data;
83 struct timeval diff;
84
85 PARA_ERROR_LOG("task %p, ret: %d\n", t, t->ret);
86 t->ret = -E_NO_DELAY;
87 if (!idt->start_time.tv_sec && !idt->start_time.tv_usec)
88 return;
89 t->ret = 0; /* timeout */
90 if (tv_diff(&s->now, &idt->start_time, &diff) > 0)
91 return;
92 t->ret = 1;
93 if (tv_diff(&s->timeout , &diff, NULL) > 0)
94 s->timeout = diff;
95 }
96
97 void para_log(int ll, const char* fmt,...)
98 {
99 va_list argp;
100
101 if (ll < conf.loglevel_arg)
102 return;
103 va_start(argp, fmt);
104 vfprintf(stderr, fmt, argp);
105 va_end(argp);
106 }
107
108 static struct writer_node_group *check_args(void)
109 {
110 int i, ret = -E_WRITE_SYNTAX;
111 struct writer_node_group *wng = NULL;
112
113 if (conf.list_writers_given) {
114 char *msg = NULL;
115 FOR_EACH_WRITER(i) {
116 char *tmp = make_message("%s%s%s",
117 i? msg : "",
118 i? " " : "",
119 writer_names[i]);
120 free(msg);
121 msg = tmp;
122 }
123 fprintf(stderr, "%s\n", msg);
124 free(msg);
125 exit(EXIT_SUCCESS);
126 }
127 if (conf.prebuffer_arg < 0 || conf.prebuffer_arg > 100)
128 goto out;
129 if (conf.start_time_given) {
130 long unsigned sec, usec;
131 if (sscanf(conf.start_time_arg, "%lu:%lu",
132 &sec, &usec) != 2)
133 goto out;
134 idt.start_time.tv_sec = sec;
135 idt.start_time.tv_usec = usec;
136 }
137 if (!conf.writer_given) {
138 wng = setup_default_wng();
139 ret = 1;
140 goto out;
141 }
142 wng = wng_new(conf.writer_given);
143 for (i = 0; i < conf.writer_given; i++) {
144 ret = check_writer_arg(conf.writer_arg[i]);
145 if (ret < 0)
146 goto out;
147 wng->writer_nodes[i].writer = &writers[ret];
148 }
149 ret = 1;
150 out:
151 if (ret > 0) {
152 wng->eof = &sit.eof;
153 return wng;
154 }
155 free(wng);
156 return NULL;
157 }
158
159 static void idt_error_handler(struct task *t)
160 {
161 PARA_ERROR_LOG("task %p, ret: %d\n", t, t->ret);
162 int ret;
163 unregister_task(t);
164 wng->buf = sit.buf;
165 wng->loaded = &sit.loaded;
166 ret = wng_open(wng);
167 if (ret < 0) {
168 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
169 exit(EXIT_FAILURE);
170 }
171 }
172
173 static void cwt_error_handler(struct task *t)
174 {
175 PARA_ERROR_LOG("task %p, ret: %d\n", t, t->ret);
176 if (t->ret < 0) {
177 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t->ret));
178 if (t->ret != -E_NO_WAV_HEADER && t->ret != -E_PRE_EOF)
179 exit(EXIT_FAILURE);
180 if (t->ret == -E_PRE_EOF) {
181 conf.channels_arg = cwt.channels;
182 conf.sample_rate_arg = cwt.sample_rate;
183 }
184 }
185 unregister_task(t);
186 idt.task.pre_select = initial_delay_pre_select;
187 idt.task.private_data = &idt;
188 idt.task.error_handler = idt_error_handler;
189 idt.task.flags = PRE_EOF_IS_ERROR;
190 register_task(&idt.task);
191 }
192
193 static void stdin_error_handler(struct task *t)
194 {
195 unregister_task(t);
196 PARA_INFO_LOG("task %p, ret: %d\n", t, t->ret);
197 if (t->ret < 0)
198 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t->ret));
199 }
200
201 int main(int argc, char *argv[])
202 {
203 int ret = -E_WRITE_SYNTAX;
204 struct sched s;
205
206 cmdline_parser(argc, argv, &conf);
207 wng = check_args();
208 if (!wng)
209 goto out;
210 init_supported_writers();
211 init_sched();
212
213 sit.bufsize = 16 * 1024,
214 sit.buf = para_malloc(16 * 1024),
215 sit.loaded = 0,
216 sit.task.pre_select = stdin_pre_select;
217 sit.task.post_select = stdin_post_select;
218 sit.task.error_handler = stdin_error_handler;
219 sit.task.flags = POST_EOF_IS_ERROR;
220 sit.task.private_data = &sit;
221 register_task(&sit.task);
222
223 cwt.task.pre_select = check_wav_pre_select;
224 cwt.task.private_data = &cwt;
225 cwt.task.error_handler = cwt_error_handler;
226 cwt.buf = sit.buf;
227 cwt.loaded = &sit.loaded;
228 cwt.task.flags = PRE_EOF_IS_ERROR;
229 register_task(&cwt.task);
230
231 s.default_timeout.tv_sec = 1;
232 s.default_timeout.tv_usec = 0;
233 ret = sched(&s);
234
235 out:
236 if (ret < 0) {
237 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
238 ret = EXIT_FAILURE;
239 } else
240 ret = EXIT_SUCCESS;
241 return ret;
242 }