+/*
+ * This file is part of DisOrder
+ * Copyright (C) 2007 Richard Kettlewell
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+ * USA
+ */
+/** @file server/disorder-normalize.c
+ * @brief Convert "raw" format output to the configured format
+ *
+ * Currently we invoke sox even for trivial conversions such as byte-swapping.
+ * Ideally we would do all conversion including resampling in this one process
+ * and eliminate the dependency on sox.
+ */
+
+#include <config.h>
+#include "types.h"
+
+#include <locale.h>
+#include <errno.h>
+#include <unistd.h>
+#include <syslog.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+
+#include "syscalls.h"
+#include "log.h"
+#include "configuration.h"
+#include "speaker-protocol.h"
+
+/** @brief Copy bytes from one file descriptor to another
+ * @param infd File descriptor read from
+ * @param outfd File descriptor to write to
+ * @param n Number of bytes to copy
+ */
+static void copy(int infd, int outfd, size_t n) {
+ char buffer[4096], *ptr;
+ int r, w;
+
+ while(n > 0) {
+ r = read(infd, buffer, sizeof buffer);
+ if(r < 0) {
+ if(errno == EINTR)
+ continue;
+ else
+ fatal(errno, "read error");
+ }
+ if(r == 0)
+ fatal(0, "unexpected EOF");
+ n -= r;
+ ptr = buffer;
+ while(r > 0) {
+ w = write(outfd, ptr, r - (ptr - buffer));
+ if(w < 0)
+ fatal(errno, "write error");
+ ptr += w;
+ }
+ }
+}
+
+static void soxargs(const char ***pp, char **qq,
+ const struct stream_header *header) {
+ *(*pp)++ = "-t.raw";
+ *(*pp)++ = "-s";
+ *qq += sprintf((char *)(*(*pp)++ = *qq), "-r%d", header->rate) + 1;
+ *qq += sprintf((char *)(*(*pp)++ = *qq), "-c%d", header->channels) + 1;
+ /* sox 12.17.9 insists on -b etc; CVS sox insists on -<n> etc; both are
+ * deployed! */
+ switch(config->sox_generation) {
+ case 0:
+ if(header->bits != 8
+ && header->endian != ENDIAN_NATIVE)
+ *(*pp)++ = "-x";
+ switch(header->bits) {
+ case 8: *(*pp)++ = "-b"; break;
+ case 16: *(*pp)++ = "-w"; break;
+ case 32: *(*pp)++ = "-l"; break;
+ case 64: *(*pp)++ = "-d"; break;
+ default: fatal(0, "cannot handle sample size %d", header->bits);
+ }
+ break;
+ case 1:
+ if(header->bits != 8
+ && header->endian != ENDIAN_NATIVE)
+ switch(header->endian) {
+ case ENDIAN_BIG: *(*pp)++ = "-B"; break;
+ case ENDIAN_LITTLE: *(*pp)++ = "-L"; break;
+ }
+ if(header->bits % 8)
+ fatal(0, "cannot handle sample size %d", header->bits);
+ *qq += sprintf((char *)(*(*pp)++ = *qq), "-%d", header->bits / 8) + 1;
+ break;
+ default:
+ fatal(0, "unknown sox_generation %ld", config->sox_generation);
+ }
+}
+
+int main(int argc, char attribute((unused)) **argv) {
+ struct stream_header header, latest_format;
+ int n, p[2], outfd = -1;
+ pid_t pid = -1;
+
+ set_progname(argv);
+ if(!setlocale(LC_CTYPE, ""))
+ fatal(errno, "error calling setlocale");
+ if(argc > 1)
+ fatal(0, "not intended to be invoked by users");
+ if(config_read())
+ fatal(0, "cannot read configuration");
+ if(!isatty(2)) {
+ openlog(progname, LOG_PID, LOG_DAEMON);
+ log_default = &log_syslog;
+ }
+ memset(&latest_format, 0, sizeof latest_format);
+ for(;;) {
+ if((n = read(0, &header, sizeof header)) < 0)
+ fatal(errno, "read error");
+ else if(n == 0)
+ exit(0);
+ else if((size_t)n < sizeof header)
+ fatal(0, "short header");
+ /* Sanity check the header */
+ if(header.rate < 100 || header.rate > 1000000)
+ fatal(0, "implausible rate %"PRId32"Hz (%#"PRIx32")",
+ header.rate, header.rate);
+ if(header.channels < 1 || header.channels > 2)
+ fatal(0, "unsupported channel count %d", header.channels);
+ if(header.bits % 8 || !header.bits || header.bits > 64)
+ fatal(0, "unsupported sample size %d bits", header.bits);
+ if(header.endian != ENDIAN_BIG && header.endian != ENDIAN_LITTLE)
+ fatal(0, "unsupported byte order %x", header.bits);
+ /* Skip empty chunks regardless of their alleged format */
+ if(header.nbytes == 0)
+ continue;
+ /* If the format has changed we stop/start the converter */
+ if(!formats_equal(&header, &latest_format)) {
+ if(pid != -1) {
+ /* There's a running converter, stop it */
+ xclose(outfd);
+ if(waitpid(pid, &n, 0) < 0)
+ fatal(errno, "error calling waitpid");
+ if(n)
+ fatal(0, "sox failed: %#x", n);
+ pid = -1;
+ outfd = -1;
+ }
+ if(!formats_equal(&header, &config->sample_format)) {
+ const char *av[32], **pp = av;
+ char argbuf[1024], *q = argbuf;
+
+ /* Input format doesn't match target, need to start a converter */
+ *pp++ = "sox";
+ soxargs(&pp, &q, &header);
+ *pp++ = "-"; /* stdin */
+ soxargs(&pp, &q, &config->sample_format);
+ *pp++ = "-"; /* stdout */
+ *pp = 0;
+ /* This pipe will be sox's stdin */
+ xpipe(p);
+ if(!(pid = xfork())) {
+ exitfn = _exit;
+ xdup2(p[0], 0);
+ xclose(p[0]);
+ xclose(p[1]);
+ execvp(av[0], (char **)av);
+ fatal(errno, "sox");
+ }
+ xclose(p[0]);
+ outfd = p[1];
+ } else
+ /* Input format matches output, can just copy bytes */
+ outfd = 1;
+ /* Remember current format for next iteration */
+ latest_format = header;
+ }
+ /* Convert or copy this chunk */
+ copy(0, outfd, header.nbytes);
+ }
+ if(outfd != -1)
+ xclose(outfd);
+ return 0;
+}
+
+/*
+Local Variables:
+c-basic-offset:2
+comment-column:40
+fill-column:79
+indent-tabs-mode:nil
+End:
+*/
~mdw / disorder / commitdiff