2 * This file is part of DisOrder
3 * Copyright (C) 2007-2009 Richard Kettlewell
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 /** @file server/normalize.c
19 * @brief Convert "raw" format output to the configured format
21 * If libsamplerate is available then resample_convert() is used to do all
22 * conversions. If not then we invoke sox (even for trivial conversions such
23 * as byte-swapping). The sox support might be removed in a future version.
26 #include "disorder-server.h"
29 static char buffer
[1024 * 1024];
31 static const struct option options
[] = {
32 { "help", no_argument
, 0, 'h' },
33 { "version", no_argument
, 0, 'V' },
34 { "config", required_argument
, 0, 'c' },
35 { "debug", no_argument
, 0, 'd' },
36 { "no-debug", no_argument
, 0, 'D' },
37 { "syslog", no_argument
, 0, 's' },
38 { "no-syslog", no_argument
, 0, 'S' },
42 /* display usage message and terminate */
43 static void attribute((noreturn
)) help(void) {
45 " disorder-normalize [OPTIONS]\n"
47 " --help, -h Display usage message\n"
48 " --version, -V Display version number\n"
49 " --config PATH, -c PATH Set configuration file\n"
50 " --debug, -d Turn on debugging\n"
51 " --[no-]syslog Force logging\n"
53 "Audio format normalizer for DisOrder. Not intended to be run\n"
59 /** @brief Copy bytes from one file descriptor to another
60 * @param infd File descriptor read from
61 * @param outfd File descriptor to write to
62 * @param n Number of bytes to copy
64 static void copy(int infd
, int outfd
, size_t n
) {
68 const ssize_t readden
= read(infd
, buffer
,
69 n
> sizeof buffer ?
sizeof buffer
: n
);
74 disorder_fatal(errno
, "read error");
77 disorder_fatal(0, "unexpected EOF");
80 while(written
< readden
) {
81 const ssize_t w
= write(outfd
, buffer
+ written
, readden
- written
);
83 disorder_fatal(errno
, "write error");
89 #if !HAVE_SAMPLERATE_H
90 static void soxargs(const char ***pp
, char **qq
,
91 const struct stream_header
*header
) {
94 *qq
+= sprintf((char *)(*(*pp
)++ = *qq
), "-r%d", header
->rate
) + 1;
95 *qq
+= sprintf((char *)(*(*pp
)++ = *qq
), "-c%d", header
->channels
) + 1;
96 /* sox 12.17.9 insists on -b etc; CVS sox insists on -<n> etc; both are
98 switch(config
->sox_generation
) {
101 && header
->endian
!= ENDIAN_NATIVE
)
103 switch(header
->bits
) {
104 case 8: *(*pp
)++ = "-b"; break;
105 case 16: *(*pp
)++ = "-w"; break;
106 case 32: *(*pp
)++ = "-l"; break;
107 case 64: *(*pp
)++ = "-d"; break;
108 default: disorder_fatal(0, "cannot handle sample size %d", header
->bits
);
113 && header
->endian
!= ENDIAN_NATIVE
)
114 switch(header
->endian
) {
115 case ENDIAN_BIG
: *(*pp
)++ = "-B"; break;
116 case ENDIAN_LITTLE
: *(*pp
)++ = "-L"; break;
119 disorder_fatal(0, "cannot handle sample size %d", header
->bits
);
120 *qq
+= sprintf((char *)(*(*pp
)++ = *qq
), "-%d", header
->bits
/ 8) + 1;
123 disorder_fatal(0, "unknown sox_generation %ld", config
->sox_generation
);
127 static void converted(uint8_t *bytes
,
129 void attribute((unused
)) *cd
) {
130 /*syslog(LOG_INFO, "out: %02x %02x %02x %02x",
136 ssize_t n
= write(1, bytes
, nbytes
);
138 disorder_fatal(errno
, "writing to stdout");
145 int main(int argc
, char attribute((unused
)) **argv
) {
146 struct stream_header header
, latest_format
;
147 int n
, outfd
= -1, logsyslog
= !isatty(2), rs_in_use
= 0;
149 struct resampler rs
[1];
152 if(!setlocale(LC_CTYPE
, ""))
153 disorder_fatal(errno
, "error calling setlocale");
154 while((n
= getopt_long(argc
, argv
, "hVc:dDSs", options
, 0)) >= 0) {
157 case 'V': version("disorder-normalize");
158 case 'c': configfile
= optarg
; break;
159 case 'd': debugging
= 1; break;
160 case 'D': debugging
= 0; break;
161 case 'S': logsyslog
= 0; break;
162 case 's': logsyslog
= 1; break;
163 default: disorder_fatal(0, "invalid option");
167 if(config_read(1, NULL
))
168 disorder_fatal(0, "cannot read configuration");
170 openlog(progname
, LOG_PID
, LOG_DAEMON
);
171 log_default
= &log_syslog
;
173 memset(&latest_format
, 0, sizeof latest_format
);
175 /* Read one header */
177 while((size_t)n
< sizeof header
) {
178 int r
= read(0, (char *)&header
+ n
, sizeof header
- n
);
182 disorder_fatal(errno
, "error reading header");
185 disorder_fatal(0, "EOF reading header");
192 D(("NEW HEADER: %"PRIu32
" bytes %"PRIu32
"Hz %"PRIu8
" channels %"PRIu8
" bits %"PRIu8
" endian",
193 header
.nbytes
, header
.rate
, header
.channels
, header
.bits
, header
.endian
));
194 /* Sanity check the header */
195 if(header
.rate
< 100 || header
.rate
> 1000000)
196 disorder_fatal(0, "implausible rate %"PRId32
"Hz (%#"PRIx32
")",
197 header
.rate
, header
.rate
);
198 if(header
.channels
< 1 || header
.channels
> 2)
199 disorder_fatal(0, "unsupported channel count %d", header
.channels
);
200 if(header
.bits
% 8 || !header
.bits
|| header
.bits
> 64)
201 disorder_fatal(0, "unsupported sample size %d bits", header
.bits
);
202 if(header
.endian
!= ENDIAN_BIG
&& header
.endian
!= ENDIAN_LITTLE
)
203 disorder_fatal(0, "unsupported byte order %d", header
.endian
);
204 /* Skip empty chunks regardless of their alleged format */
205 if(header
.nbytes
== 0)
207 /* If the format has changed we stop/start the converter */
208 #if HAVE_SAMPLERATE_H
209 /* We have libsamplerate */
210 if(formats_equal(&header
, &config
->sample_format
))
211 /* If the format is already correct then we just write out the data */
212 copy(0, 1, header
.nbytes
);
214 /* If we have a resampler active already check it is suitable and destroy
217 D(("call resample_close"));
221 /*syslog(LOG_INFO, "%d/%d/%d/%d/%d -> %d/%d/%d/%d/%d",
227 config->sample_format.bits,
228 config->sample_format.channels,
229 config->sample_format.rate,
231 config->sample_format.endian);*/
233 /* Create a suitable resampler. */
234 D(("call resample_init"));
241 config
->sample_format
.bits
,
242 config
->sample_format
.channels
,
243 config
->sample_format
.rate
,
245 config
->sample_format
.endian
);
246 latest_format
= header
;
248 /* TODO speaker protocol does not record signedness of samples. It's
249 * assumed that they are always signed. This should be fixed in the
250 * future (and the sample format syntax extended in a compatible
253 /* Feed data through the resampler */
254 size_t used
= 0, left
= header
.nbytes
;
255 while(used
|| left
) {
257 size_t limit
= (sizeof buffer
) - used
;
260 ssize_t r
= read(0, buffer
+ used
, limit
);
262 disorder_fatal(errno
, "reading from stdin");
264 disorder_fatal(0, "unexpected EOF");
267 //syslog(LOG_INFO, "read %zd bytes", r);
268 D(("read %zd bytes", r
));
270 /*syslog(LOG_INFO, " in: %02x %02x %02x %02x",
274 (uint8_t)buffer[3]);*/
275 D(("calling resample_convert used=%zu !left=%d", used
, !left
));
276 const size_t consumed
= resample_convert(rs
,
277 (uint8_t *)buffer
, used
,
280 //syslog(LOG_INFO, "used=%zu consumed=%zu", used, consumed);
281 D(("consumed=%zu", consumed
));
282 memmove(buffer
, buffer
+ consumed
, used
- consumed
);
287 /* We do not have libsamplerate. We will use sox instead. */
288 if(!formats_equal(&header
, &latest_format
)) {
290 /* There's a running converter, stop it */
292 if(waitpid(pid
, &n
, 0) < 0)
293 disorder_fatal(errno
, "error calling waitpid");
295 disorder_fatal(0, "sox failed: %#x", n
);
299 if(!formats_equal(&header
, &config
->sample_format
)) {
300 const char *av
[32], **pp
= av
;
301 char argbuf
[1024], *q
= argbuf
;
303 /* Input format doesn't match target, need to start a converter */
305 soxargs(&pp
, &q
, &header
);
306 *pp
++ = "-"; /* stdin */
307 soxargs(&pp
, &q
, &config
->sample_format
);
308 *pp
++ = "-"; /* stdout */
310 /* This pipe will be sox's stdin */
313 if(!(pid
= xfork())) {
318 execvp(av
[0], (char **)av
);
319 disorder_fatal(errno
, "sox");
324 /* Input format matches output, can just copy bytes */
326 /* Remember current format for next iteration */
327 latest_format
= header
;
329 /* Convert or copy this chunk */
330 copy(0, outfd
, header
.nbytes
);
336 /* There's still a converter running */
337 if(waitpid(pid
, &n
, 0) < 0)
338 disorder_fatal(errno
, "error calling waitpid");
340 disorder_fatal(0, "sox failed: %#x", n
);