2010-08-19 15:38:45 -05:00
|
|
|
/*
|
2014-11-15 06:56:12 -06:00
|
|
|
* Copyright (c) 2008-2014 Alexandre Ratchov <alex@caoua.org>
|
2010-08-19 15:38:45 -05:00
|
|
|
*
|
|
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
|
|
* copyright notice and this permission notice appear in all copies.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
*/
|
|
|
|
|
2011-06-03 05:05:47 -05:00
|
|
|
#include <fcntl.h>
|
2010-08-19 15:38:45 -05:00
|
|
|
#include <stdlib.h>
|
|
|
|
#include <stdio.h>
|
2011-06-03 05:05:47 -05:00
|
|
|
#include <string.h>
|
2010-08-19 15:38:45 -05:00
|
|
|
#include <unistd.h>
|
2014-11-15 06:51:02 -06:00
|
|
|
#include "utils.h"
|
2010-08-19 15:38:45 -05:00
|
|
|
#include "wav.h"
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
/*
|
|
|
|
* Max size of a .wav file, format design limitation.
|
|
|
|
*/
|
|
|
|
#define WAV_MAXPOS (0x7fffffff)
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Encoding IDs used in .wav headers.
|
|
|
|
*/
|
2014-12-30 11:45:05 -06:00
|
|
|
#define WAV_FMT_PCM 1
|
|
|
|
#define WAV_FMT_FLOAT 3
|
|
|
|
#define WAV_FMT_ALAW 6
|
|
|
|
#define WAV_FMT_ULAW 7
|
|
|
|
#define WAV_FMT_EXT 0xfffe
|
2014-11-15 06:51:02 -06:00
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
unsigned char ld[4];
|
|
|
|
} le32_t;
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
unsigned char lw[2];
|
|
|
|
} le16_t;
|
|
|
|
|
|
|
|
struct wavriff {
|
|
|
|
char magic[4];
|
|
|
|
le32_t size;
|
|
|
|
char type[4];
|
|
|
|
};
|
|
|
|
|
|
|
|
struct wavchunk {
|
|
|
|
char id[4];
|
|
|
|
le32_t size;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct wavfmt {
|
|
|
|
le16_t fmt;
|
|
|
|
le16_t nch;
|
|
|
|
le32_t rate;
|
|
|
|
le32_t byterate;
|
|
|
|
le16_t blkalign;
|
|
|
|
le16_t bits;
|
|
|
|
#define WAV_FMT_SIZE 16
|
|
|
|
#define WAV_FMT_SIZE2 (16 + 2)
|
|
|
|
#define WAV_FMT_EXT_SIZE (16 + 24)
|
|
|
|
le16_t extsize;
|
|
|
|
le16_t valbits;
|
|
|
|
le32_t chanmask;
|
|
|
|
le16_t extfmt;
|
|
|
|
char guid[14];
|
|
|
|
};
|
|
|
|
|
|
|
|
struct wavhdr {
|
|
|
|
struct wavriff riff; /* 00..11 */
|
|
|
|
struct wavchunk fmt_hdr; /* 12..20 */
|
|
|
|
struct wavfmt fmt;
|
|
|
|
struct wavchunk data_hdr;
|
|
|
|
};
|
|
|
|
|
|
|
|
char wav_id_riff[4] = { 'R', 'I', 'F', 'F' };
|
|
|
|
char wav_id_wave[4] = { 'W', 'A', 'V', 'E' };
|
|
|
|
char wav_id_data[4] = { 'd', 'a', 't', 'a' };
|
|
|
|
char wav_id_fmt[4] = { 'f', 'm', 't', ' ' };
|
|
|
|
char wav_guid[14] = {
|
|
|
|
0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x10, 0x00, 0x80, 0x00,
|
|
|
|
0x00, 0xAA, 0x00, 0x38,
|
|
|
|
0x9B, 0x71
|
|
|
|
};
|
2013-11-18 10:50:04 -06:00
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
static inline unsigned int
|
|
|
|
le16_get(le16_t *p)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
return p->lw[0] | p->lw[1] << 8;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
static inline void
|
|
|
|
le16_set(le16_t *p, unsigned int v)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
p->lw[0] = v;
|
|
|
|
p->lw[1] = v >> 8;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
static inline unsigned int
|
|
|
|
le32_get(le32_t *p)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
return p->ld[0] |
|
|
|
|
p->ld[1] << 8 |
|
|
|
|
p->ld[2] << 16 |
|
|
|
|
p->ld[3] << 24;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
static inline void
|
|
|
|
le32_set(le32_t *p, unsigned int v)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
p->ld[0] = v;
|
|
|
|
p->ld[1] = v >> 8;
|
|
|
|
p->ld[2] = v >> 16;
|
|
|
|
p->ld[3] = v >> 24;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
static int
|
|
|
|
wav_readfmt(struct wav *w, unsigned int csize)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
struct wavfmt fmt;
|
|
|
|
unsigned int nch, rate, bits, bps, enc;
|
2010-08-19 15:38:45 -05:00
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
if (csize < WAV_FMT_SIZE) {
|
|
|
|
log_putu(csize);
|
|
|
|
log_puts(": bugus format chunk size\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (csize > WAV_FMT_EXT_SIZE)
|
|
|
|
csize = WAV_FMT_EXT_SIZE;
|
|
|
|
if (read(w->fd, &fmt, csize) != csize) {
|
|
|
|
log_puts("failed to read .wav format chun\n");
|
|
|
|
return 0;
|
2011-10-05 06:21:01 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
enc = le16_get(&fmt.fmt);
|
|
|
|
bits = le16_get(&fmt.bits);
|
2014-12-30 11:45:05 -06:00
|
|
|
if (enc == WAV_FMT_EXT) {
|
2014-11-15 06:51:02 -06:00
|
|
|
if (csize != WAV_FMT_EXT_SIZE) {
|
|
|
|
log_puts("missing extended format chunk in .wav file\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
if (memcmp(fmt.guid, wav_guid, sizeof(wav_guid)) != 0) {
|
|
|
|
log_puts("unknown format (GUID) in .wav file\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
bps = (bits + 7) / 8;
|
|
|
|
bits = le16_get(&fmt.valbits);
|
|
|
|
enc = le16_get(&fmt.extfmt);
|
|
|
|
} else
|
|
|
|
bps = (bits + 7) / 8;
|
|
|
|
nch = le16_get(&fmt.nch);
|
|
|
|
if (nch == 0) {
|
|
|
|
log_puts("zero number of channels in .wav file\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
rate = le32_get(&fmt.rate);
|
|
|
|
if (rate < RATE_MIN || rate > RATE_MAX) {
|
|
|
|
log_putu(rate);
|
|
|
|
log_puts(": bad sample rate in .wav file\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (bits < BITS_MIN || bits > BITS_MAX) {
|
|
|
|
log_putu(bits);
|
|
|
|
log_puts(": bad number of bits\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (bits > bps * 8) {
|
|
|
|
log_puts("bits larger than bytes-per-sample\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-12-30 11:45:05 -06:00
|
|
|
switch (enc) {
|
|
|
|
case WAV_FMT_PCM:
|
|
|
|
w->enc = ENC_PCM;
|
2014-11-15 06:51:02 -06:00
|
|
|
w->par.bps = bps;
|
|
|
|
w->par.bits = bits;
|
|
|
|
w->par.le = 1;
|
|
|
|
w->par.sig = (bits <= 8) ? 0 : 1; /* ask microsoft why... */
|
|
|
|
w->par.msb = 1;
|
2014-12-30 11:45:05 -06:00
|
|
|
break;
|
|
|
|
case WAV_FMT_ALAW:
|
|
|
|
case WAV_FMT_ULAW:
|
|
|
|
w->enc = (enc == WAV_FMT_ULAW) ? ENC_ULAW : ENC_ALAW;
|
2014-11-15 06:51:02 -06:00
|
|
|
if (bits != 8) {
|
|
|
|
log_puts("mulaw/alaw encoding not 8-bit\n");
|
|
|
|
return 0;
|
|
|
|
}
|
2014-12-30 11:45:05 -06:00
|
|
|
w->par.bits = 8;
|
|
|
|
w->par.bps = 1;
|
2014-11-15 06:51:02 -06:00
|
|
|
w->par.le = ADATA_LE;
|
2014-12-30 11:45:05 -06:00
|
|
|
w->par.sig = 0;
|
|
|
|
w->par.msb = 0;
|
|
|
|
break;
|
|
|
|
case WAV_FMT_FLOAT:
|
2014-12-30 12:17:32 -06:00
|
|
|
w->enc = ENC_FLOAT;
|
2014-12-30 11:45:05 -06:00
|
|
|
if (bits != 32) {
|
|
|
|
log_puts("only 32-bit float supported\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
w->par.bits = 32;
|
|
|
|
w->par.bps = 4;
|
|
|
|
w->par.le = 1;
|
2014-12-30 12:17:32 -06:00
|
|
|
w->par.sig = 0;
|
2014-11-15 06:51:02 -06:00
|
|
|
w->par.msb = 0;
|
2014-12-30 11:45:05 -06:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
log_putu(enc);
|
|
|
|
log_puts(": unsupported encoding in .wav file\n");
|
|
|
|
return 0;
|
2014-11-15 06:51:02 -06:00
|
|
|
}
|
|
|
|
w->nch = nch;
|
|
|
|
w->rate = rate;
|
|
|
|
return 1;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
static int
|
|
|
|
wav_readhdr(struct wav *w)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
struct wavriff riff;
|
|
|
|
struct wavchunk chunk;
|
|
|
|
unsigned int csize, rsize, pos = 0;
|
|
|
|
int fmt_done = 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
if (lseek(w->fd, 0, SEEK_SET) < 0) {
|
|
|
|
log_puts("failed to seek to beginning of .wav file\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (read(w->fd, &riff, sizeof(riff)) != sizeof(riff)) {
|
|
|
|
log_puts("failed to read .wav file riff header\n");
|
2011-10-05 06:21:01 -05:00
|
|
|
return 0;
|
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (memcmp(&riff.magic, &wav_id_riff, 4) != 0 ||
|
|
|
|
memcmp(&riff.type, &wav_id_wave, 4)) {
|
|
|
|
log_puts("not a .wav file\n");
|
|
|
|
return 0;
|
2011-10-05 06:21:01 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
rsize = le32_get(&riff.size);
|
|
|
|
for (;;) {
|
|
|
|
if (pos + sizeof(struct wavchunk) > rsize) {
|
|
|
|
log_puts("missing data chunk in .wav file\n");
|
|
|
|
return 0;
|
2011-10-05 07:14:20 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (read(w->fd, &chunk, sizeof(chunk)) != sizeof(chunk)) {
|
|
|
|
log_puts("failed to read .wav chunk header\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
csize = le32_get(&chunk.size);
|
|
|
|
if (memcmp(chunk.id, wav_id_fmt, 4) == 0) {
|
|
|
|
if (!wav_readfmt(w, csize))
|
|
|
|
return 0;
|
|
|
|
fmt_done = 1;
|
|
|
|
} else if (memcmp(chunk.id, wav_id_data, 4) == 0) {
|
|
|
|
w->startpos = pos + sizeof(riff) + sizeof(chunk);
|
|
|
|
w->endpos = w->startpos + csize;
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
#ifdef DEBUG
|
|
|
|
if (log_level >= 2)
|
|
|
|
log_puts("skipped unknown .wav file chunk\n");
|
2011-10-05 06:21:01 -05:00
|
|
|
#endif
|
2014-11-15 06:51:02 -06:00
|
|
|
}
|
2011-10-05 06:21:01 -05:00
|
|
|
|
2010-08-19 15:38:45 -05:00
|
|
|
/*
|
2014-11-15 06:51:02 -06:00
|
|
|
* next chunk
|
2010-08-19 15:38:45 -05:00
|
|
|
*/
|
2014-11-15 06:51:02 -06:00
|
|
|
pos += sizeof(struct wavchunk) + csize;
|
|
|
|
if (lseek(w->fd, sizeof(riff) + pos, SEEK_SET) < 0) {
|
|
|
|
log_puts("filed to seek to chunk in .wav file\n");
|
|
|
|
return 0;
|
|
|
|
}
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (!fmt_done) {
|
|
|
|
log_puts("missing format chunk in .wav file\n");
|
2010-08-19 15:38:45 -05:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
2014-11-15 06:51:02 -06:00
|
|
|
* Write header and seek to start position
|
2010-08-19 15:38:45 -05:00
|
|
|
*/
|
2014-11-15 06:51:02 -06:00
|
|
|
static int
|
|
|
|
wav_writehdr(struct wav *w)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
struct wavhdr hdr;
|
|
|
|
|
|
|
|
memset(&hdr, 0, sizeof(struct wavhdr));
|
|
|
|
memcpy(hdr.riff.magic, wav_id_riff, 4);
|
|
|
|
memcpy(hdr.riff.type, wav_id_wave, 4);
|
|
|
|
le32_set(&hdr.riff.size, w->endpos - sizeof(hdr.riff));
|
|
|
|
memcpy(hdr.fmt_hdr.id, wav_id_fmt, 4);
|
|
|
|
le32_set(&hdr.fmt_hdr.size, sizeof(hdr.fmt));
|
|
|
|
le16_set(&hdr.fmt.fmt, 1);
|
|
|
|
le16_set(&hdr.fmt.nch, w->nch);
|
|
|
|
le32_set(&hdr.fmt.rate, w->rate);
|
|
|
|
le32_set(&hdr.fmt.byterate, w->rate * w->par.bps * w->nch);
|
|
|
|
le16_set(&hdr.fmt.blkalign, w->par.bps * w->nch);
|
|
|
|
le16_set(&hdr.fmt.bits, w->par.bits);
|
|
|
|
memcpy(hdr.data_hdr.id, wav_id_data, 4);
|
|
|
|
le32_set(&hdr.data_hdr.size, w->endpos - w->startpos);
|
|
|
|
|
|
|
|
if (lseek(w->fd, 0, SEEK_SET) < 0) {
|
|
|
|
log_puts("failed to seek back to .wav file header\n");
|
2010-08-19 15:38:45 -05:00
|
|
|
return 0;
|
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (write(w->fd, &hdr, sizeof(hdr)) != sizeof(hdr)) {
|
|
|
|
log_puts("failed to write .wav file header\n");
|
2010-08-19 15:38:45 -05:00
|
|
|
return 0;
|
2014-11-15 06:51:02 -06:00
|
|
|
}
|
|
|
|
w->curpos = w->startpos;
|
2010-08-19 15:38:45 -05:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
size_t
|
|
|
|
wav_read(struct wav *w, void *data, size_t count)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
off_t maxread;
|
|
|
|
ssize_t n;
|
2014-12-30 11:45:05 -06:00
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
if (w->endpos >= 0) {
|
|
|
|
maxread = w->endpos - w->curpos;
|
|
|
|
if (maxread == 0) {
|
2010-08-19 15:38:45 -05:00
|
|
|
#ifdef DEBUG
|
2014-11-15 06:51:02 -06:00
|
|
|
if (log_level >= 3) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": end reached\n");
|
|
|
|
}
|
2010-08-19 15:38:45 -05:00
|
|
|
#endif
|
2014-11-15 06:51:02 -06:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
if (count > maxread)
|
|
|
|
count = maxread;
|
|
|
|
}
|
|
|
|
n = read(w->fd, data, count);
|
|
|
|
if (n < 0) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": couldn't read\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
w->curpos += n;
|
|
|
|
return n;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
size_t
|
|
|
|
wav_write(struct wav *w, void *data, size_t count)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
off_t maxwrite;
|
|
|
|
int n;
|
2010-08-19 15:38:45 -05:00
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
if (w->maxpos >= 0) {
|
|
|
|
maxwrite = w->maxpos - w->curpos;
|
|
|
|
if (maxwrite == 0) {
|
2010-08-19 15:38:45 -05:00
|
|
|
#ifdef DEBUG
|
2014-11-15 06:51:02 -06:00
|
|
|
if (log_level >= 3) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": max file size reached\n");
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (count > maxwrite)
|
|
|
|
count = maxwrite;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
n = write(w->fd, data, count);
|
|
|
|
if (n < 0) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": couldn't write\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
w->curpos += n;
|
|
|
|
if (w->endpos < w->curpos)
|
|
|
|
w->endpos = w->curpos;
|
|
|
|
return n;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
int
|
|
|
|
wav_seek(struct wav *w, off_t pos)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
pos += w->startpos;
|
|
|
|
if (w->endpos >= 0 && pos > w->endpos) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": attempt to seek ouside file boundaries\n");
|
|
|
|
return 0;
|
|
|
|
}
|
2010-08-19 15:38:45 -05:00
|
|
|
|
2014-11-15 06:51:02 -06:00
|
|
|
/*
|
|
|
|
* seek only if needed to avoid errors with pipes & sockets
|
|
|
|
*/
|
|
|
|
if (pos != w->curpos) {
|
|
|
|
if (lseek(w->fd, pos, SEEK_SET) < 0) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": couldn't seek\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
w->curpos = pos;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
return 1;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2014-11-15 06:51:02 -06:00
|
|
|
wav_close(struct wav *w)
|
2010-08-19 15:38:45 -05:00
|
|
|
{
|
2014-11-15 06:51:02 -06:00
|
|
|
if (w->flags & WAV_FWRITE) {
|
|
|
|
if (w->hdr == HDR_WAV)
|
|
|
|
wav_writehdr(w);
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
close(w->fd);
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
|
|
|
|
2011-10-05 06:21:01 -05:00
|
|
|
int
|
2014-11-15 06:51:02 -06:00
|
|
|
wav_open(struct wav *w, char *path, int hdr, int flags,
|
|
|
|
struct aparams *par, int rate, int nch)
|
2011-06-03 05:05:47 -05:00
|
|
|
{
|
2011-10-05 06:21:01 -05:00
|
|
|
char *ext;
|
2014-11-15 06:51:02 -06:00
|
|
|
struct wavhdr dummy;
|
|
|
|
|
|
|
|
w->par = *par;
|
|
|
|
w->rate = rate;
|
|
|
|
w->nch = nch;
|
|
|
|
w->flags = flags;
|
|
|
|
w->hdr = hdr;
|
|
|
|
if (hdr == HDR_AUTO) {
|
|
|
|
w->hdr = HDR_RAW;
|
|
|
|
ext = strrchr(path, '.');
|
2011-10-05 06:21:01 -05:00
|
|
|
if (ext != NULL) {
|
|
|
|
ext++;
|
2014-11-15 06:51:02 -06:00
|
|
|
if (strcasecmp(ext, "wav") == 0)
|
|
|
|
w->hdr = HDR_WAV;
|
2011-10-05 06:21:01 -05:00
|
|
|
}
|
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (w->flags == WAV_FREAD) {
|
|
|
|
if (strcmp(path, "-") == 0) {
|
|
|
|
w->path = "stdin";
|
|
|
|
w->fd = STDIN_FILENO;
|
|
|
|
} else {
|
|
|
|
w->path = path;
|
|
|
|
w->fd = open(w->path, O_RDONLY, 0);
|
|
|
|
if (w->fd < 0) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": failed to open for reading\n");
|
|
|
|
return 0;
|
|
|
|
}
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (w->hdr == HDR_WAV) {
|
|
|
|
if (!wav_readhdr(w))
|
|
|
|
goto bad_close;
|
|
|
|
} else {
|
|
|
|
w->startpos = 0;
|
|
|
|
w->endpos = -1; /* read until EOF */
|
2014-12-30 11:45:05 -06:00
|
|
|
w->enc = ENC_PCM;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
w->curpos = w->startpos;
|
|
|
|
} else if (flags == WAV_FWRITE) {
|
|
|
|
if (strcmp(path, "-") == 0) {
|
|
|
|
w->path = "stdout";
|
|
|
|
w->fd = STDOUT_FILENO;
|
|
|
|
} else {
|
|
|
|
w->path = path;
|
|
|
|
w->fd = open(w->path, O_WRONLY | O_TRUNC | O_CREAT, 0666);
|
|
|
|
if (w->fd < 0) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": failed to create file\n");
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2011-10-05 06:21:01 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
if (w->hdr == HDR_WAV) {
|
|
|
|
w->par.bps = (w->par.bits + 7) >> 3;
|
|
|
|
if (w->par.bits > 8) {
|
|
|
|
w->par.le = 1;
|
|
|
|
w->par.sig = 1;
|
|
|
|
} else
|
|
|
|
w->par.sig = 0;
|
|
|
|
if (w->par.bits & 7)
|
|
|
|
w->par.msb = 1;
|
|
|
|
w->endpos = w->startpos = sizeof(struct wavhdr);
|
|
|
|
w->maxpos = WAV_MAXPOS;
|
|
|
|
memset(&dummy, 0xd0, sizeof(struct wavhdr));
|
|
|
|
if (write(w->fd, &dummy, sizeof(struct wavhdr)) < 0) {
|
|
|
|
log_puts(w->path);
|
|
|
|
log_puts(": failed reserve space for .wav header\n");
|
|
|
|
goto bad_close;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
w->endpos = w->startpos = 0;
|
|
|
|
w->maxpos = -1;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|
2014-11-15 06:51:02 -06:00
|
|
|
w->curpos = w->startpos;
|
|
|
|
} else {
|
2010-08-19 15:38:45 -05:00
|
|
|
#ifdef DEBUG
|
2014-11-15 06:51:02 -06:00
|
|
|
log_puts("wav_open: wrong flags\n");
|
|
|
|
panic();
|
2010-08-19 15:38:45 -05:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
return 1;
|
2014-11-15 06:51:02 -06:00
|
|
|
bad_close:
|
|
|
|
close(w->fd);
|
|
|
|
return 0;
|
2010-08-19 15:38:45 -05:00
|
|
|
}
|