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 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");
166 if(config_read(1, NULL))
167 disorder_fatal(0, "cannot read configuration");
169 openlog(progname, LOG_PID, LOG_DAEMON);
170 log_default = &log_syslog;
172 memset(&latest_format, 0, sizeof latest_format);
174 /* Read one header */
176 while((size_t)n < sizeof header) {
177 int r = read(0, (char *)&header + n, sizeof header - n);
181 disorder_fatal(errno, "error reading header");
184 disorder_fatal(0, "EOF reading header");
191 /* Sanity check the header */
192 if(header.rate < 100 || header.rate > 1000000)
193 disorder_fatal(0, "implausible rate %"PRId32"Hz (%#"PRIx32")",
194 header.rate, header.rate);
195 if(header.channels < 1 || header.channels > 2)
196 disorder_fatal(0, "unsupported channel count %d", header.channels);
197 if(header.bits % 8 || !header.bits || header.bits > 64)
198 disorder_fatal(0, "unsupported sample size %d bits", header.bits);
199 if(header.endian != ENDIAN_BIG && header.endian != ENDIAN_LITTLE)
200 disorder_fatal(0, "unsupported byte order %d", header.endian);
201 /* Skip empty chunks regardless of their alleged format */
202 if(header.nbytes == 0)
204 /* If the format has changed we stop/start the converter */
205 #if HAVE_SAMPLERATE_H
206 /* We have libsamplerate */
207 if(formats_equal(&header, &config->sample_format))
208 /* If the format is already correct then we just write out the data */
209 copy(0, 1, header.nbytes);
211 /* If we have a resampler active already check it is suitable and destroy
213 if(!formats_equal(&header, &latest_format) && rs_in_use) {
217 /*syslog(LOG_INFO, "%d/%d/%d/%d/%d -> %d/%d/%d/%d/%d",
223 config->sample_format.bits,
224 config->sample_format.channels,
225 config->sample_format.rate,
227 config->sample_format.endian);*/
229 /* Create a suitable resampler. */
236 config->sample_format.bits,
237 config->sample_format.channels,
238 config->sample_format.rate,
240 config->sample_format.endian);
241 latest_format = header;
243 /* TODO speaker protocol does not record signedness of samples. It's
244 * assumed that they are always signed. This should be fixed in the
245 * future (and the sample format syntax extended in a compatible
248 /* Feed data through the resampler */
249 size_t used = 0, left = header.nbytes;
250 while(used || left) {
252 size_t limit = (sizeof buffer) - used;
255 ssize_t r = read(0, buffer + used, limit);
257 disorder_fatal(errno, "reading from stdin");
259 disorder_fatal(0, "unexpected EOF");
262 //syslog(LOG_INFO, "read %zd bytes", r);
264 /*syslog(LOG_INFO, " in: %02x %02x %02x %02x",
268 (uint8_t)buffer[3]);*/
269 const size_t consumed = resample_convert(rs,
270 (uint8_t *)buffer, used,
273 //syslog(LOG_INFO, "used=%zu consumed=%zu", used, consumed);
274 memmove(buffer, buffer + consumed, used - consumed);
279 /* We do not have libsamplerate. We will use sox instead. */
280 if(!formats_equal(&header, &latest_format)) {
282 /* There's a running converter, stop it */
284 if(waitpid(pid, &n, 0) < 0)
285 disorder_fatal(errno, "error calling waitpid");
287 disorder_fatal(0, "sox failed: %#x", n);
291 if(!formats_equal(&header, &config->sample_format)) {
292 const char *av[32], **pp = av;
293 char argbuf[1024], *q = argbuf;
295 /* Input format doesn't match target, need to start a converter */
297 soxargs(&pp, &q, &header);
298 *pp++ = "-"; /* stdin */
299 soxargs(&pp, &q, &config->sample_format);
300 *pp++ = "-"; /* stdout */
302 /* This pipe will be sox's stdin */
305 if(!(pid = xfork())) {
310 execvp(av[0], (char **)av);
311 disorder_fatal(errno, "sox");
316 /* Input format matches output, can just copy bytes */
318 /* Remember current format for next iteration */
319 latest_format = header;
321 /* Convert or copy this chunk */
322 copy(0, outfd, header.nbytes);
328 /* There's still a converter running */
329 if(waitpid(pid, &n, 0) < 0)
330 disorder_fatal(errno, "error calling waitpid");
332 disorder_fatal(0, "sox failed: %#x", n);