1
0
mirror of https://github.com/ericonr/sndio.git synced 2024-02-18 04:45:21 -06:00
sndio/aucat/afile.c

829 lines
18 KiB
C
Raw Normal View History

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.
*/
#include <fcntl.h>
2010-08-19 15:38:45 -05:00
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
2010-08-19 15:38:45 -05:00
#include <unistd.h>
2015-01-05 11:01:29 -06:00
#include "afile.h"
#include "utils.h"
2010-08-19 15:38:45 -05: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
typedef struct {
unsigned char ld[4];
} le32_t;
typedef struct {
unsigned char lw[2];
} le16_t;
2015-01-06 04:41:24 -06:00
typedef struct {
unsigned char bd[4];
} be32_t;
typedef struct {
unsigned char bw[2];
} be16_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;
};
2015-01-06 04:41:24 -06:00
struct aiff_form {
char magic[4];
be32_t size;
char type[4];
};
struct aiff_chunk {
char id[4];
be32_t size;
};
struct aiff_comm {
be16_t nch;
be32_t nfr;
be16_t bits;
/* rate in 80-bit floating point */
be16_t rate_ex;
be32_t rate_hi;
be32_t rate_lo;
2015-01-06 05:49:08 -06:00
char comp_id[4];
/* followed by stuff we don't care about */
2015-01-06 04:41:24 -06:00
};
2015-01-06 05:49:08 -06:00
#define AIFF_COMM_BASESZ (offsetof(struct aiff_comm, comp_id))
2015-01-06 04:41:24 -06:00
struct aiff_data {
be32_t offs;
be32_t blksz;
};
struct aiff_hdr {
struct aiff_form form; /* 00..11 */
struct aiff_chunk comm_hdr; /* 12..20 */
struct aiff_comm comm;
struct aiff_chunk data_hdr;
struct aiff_data data;
};
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
2015-01-06 04:41:24 -06:00
char aiff_id_form[4] = {'F', 'O', 'R', 'M'};
char aiff_id_aiff[4] = { 'A', 'I', 'F', 'F' };
2015-01-06 05:49:08 -06:00
char aiff_id_aifc[4] = { 'A', 'I', 'F', 'C' };
2015-01-06 04:41:24 -06:00
char aiff_id_data[4] = { 'S', 'S', 'N', 'D' };
char aiff_id_comm[4] = { 'C', 'O', 'M', 'M' };
static inline unsigned int
le16_get(le16_t *p)
2010-08-19 15:38:45 -05:00
{
return p->lw[0] | p->lw[1] << 8;
2010-08-19 15:38:45 -05:00
}
static inline void
le16_set(le16_t *p, unsigned int v)
2010-08-19 15:38:45 -05:00
{
p->lw[0] = v;
p->lw[1] = v >> 8;
2010-08-19 15:38:45 -05:00
}
static inline unsigned int
le32_get(le32_t *p)
2010-08-19 15:38:45 -05: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
}
static inline void
le32_set(le32_t *p, unsigned int v)
2010-08-19 15:38:45 -05: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
}
2015-01-06 04:41:24 -06:00
static inline unsigned int
be16_get(be16_t *p)
{
return p->bw[1] | p->bw[0] << 8;
}
static inline void
be16_set(be16_t *p, unsigned int v)
{
p->bw[1] = v;
p->bw[0] = v >> 8;
}
static inline unsigned int
be32_get(be32_t *p)
{
return p->bd[3] |
p->bd[2] << 8 |
p->bd[1] << 16 |
p->bd[0] << 24;
}
static inline void
be32_set(be32_t *p, unsigned int v)
{
p->bd[3] = v;
p->bd[2] = v >> 8;
p->bd[1] = v >> 16;
p->bd[0] = v >> 24;
}
static int
2015-01-05 11:01:29 -06:00
afile_wav_readfmt(struct afile *f, unsigned int csize)
2010-08-19 15:38:45 -05:00
{
struct wavfmt fmt;
unsigned int nch, rate, bits, bps, enc;
2010-08-19 15:38:45 -05: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
}
if (csize > WAV_FMT_EXT_SIZE)
csize = WAV_FMT_EXT_SIZE;
2015-01-05 11:01:29 -06:00
if (read(f->fd, &fmt, csize) != csize) {
log_puts("failed to read .wav format chun\n");
return 0;
}
enc = le16_get(&fmt.fmt);
bits = le16_get(&fmt.bits);
2014-12-30 11:45:05 -06:00
if (enc == WAV_FMT_EXT) {
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
}
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
}
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
}
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:
2015-01-05 11:01:29 -06:00
f->enc = ENC_PCM;
f->par.bps = bps;
f->par.bits = bits;
f->par.le = 1;
f->par.sig = (bits <= 8) ? 0 : 1; /* ask microsoft why... */
f->par.msb = 1;
2014-12-30 11:45:05 -06:00
break;
case WAV_FMT_ALAW:
case WAV_FMT_ULAW:
2015-01-05 11:01:29 -06:00
f->enc = (enc == WAV_FMT_ULAW) ? ENC_ULAW : ENC_ALAW;
if (bits != 8) {
log_puts("mulaw/alaw encoding not 8-bit\n");
return 0;
}
2015-01-05 11:01:29 -06:00
f->par.bits = 8;
f->par.bps = 1;
f->par.le = ADATA_LE;
f->par.sig = 0;
f->par.msb = 0;
2014-12-30 11:45:05 -06:00
break;
case WAV_FMT_FLOAT:
2015-01-05 11:01:29 -06:00
f->enc = ENC_FLOAT;
2014-12-30 11:45:05 -06:00
if (bits != 32) {
log_puts("only 32-bit float supported\n");
return 0;
}
2015-01-05 11:01:29 -06:00
f->par.bits = 32;
f->par.bps = 4;
f->par.le = 1;
f->par.sig = 0;
f->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;
}
2015-01-05 11:01:29 -06:00
f->nch = nch;
f->rate = rate;
return 1;
2010-08-19 15:38:45 -05:00
}
static int
2015-01-05 11:01:29 -06:00
afile_wav_readhdr(struct afile *f)
2010-08-19 15:38:45 -05: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
2015-01-05 11:01:29 -06:00
if (lseek(f->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
}
2015-01-05 11:01:29 -06:00
if (read(f->fd, &riff, sizeof(riff)) != sizeof(riff)) {
log_puts("failed to read .wav file riff header\n");
return 0;
}
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;
}
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
}
2015-01-05 11:01:29 -06:00
if (read(f->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) {
2015-01-05 11:01:29 -06:00
if (!afile_wav_readfmt(f, csize))
return 0;
fmt_done = 1;
} else if (memcmp(chunk.id, wav_id_data, 4) == 0) {
2015-01-05 11:01:29 -06:00
f->startpos = pos + sizeof(riff) + sizeof(chunk);
f->endpos = f->startpos + csize;
break;
} else {
#ifdef DEBUG
if (log_level >= 2)
log_puts("skipped unknown .wav file chunk\n");
#endif
}
2010-08-19 15:38:45 -05:00
/*
* next chunk
2010-08-19 15:38:45 -05:00
*/
pos += sizeof(struct wavchunk) + csize;
2015-01-05 11:01:29 -06:00
if (lseek(f->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
}
if (!fmt_done) {
log_puts("missing format chunk in .wav file\n");
2010-08-19 15:38:45 -05:00
return 0;
}
return 1;
}
/*
* Write header and seek to start position
2010-08-19 15:38:45 -05:00
*/
static int
2015-01-05 11:01:29 -06:00
afile_wav_writehdr(struct afile *f)
2010-08-19 15:38:45 -05: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);
2015-01-05 11:01:29 -06:00
le32_set(&hdr.riff.size, f->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);
2015-01-05 11:01:29 -06:00
le16_set(&hdr.fmt.nch, f->nch);
le32_set(&hdr.fmt.rate, f->rate);
le32_set(&hdr.fmt.byterate, f->rate * f->par.bps * f->nch);
le16_set(&hdr.fmt.blkalign, f->par.bps * f->nch);
le16_set(&hdr.fmt.bits, f->par.bits);
memcpy(hdr.data_hdr.id, wav_id_data, 4);
2015-01-05 11:01:29 -06:00
le32_set(&hdr.data_hdr.size, f->endpos - f->startpos);
2015-01-05 11:01:29 -06:00
if (lseek(f->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;
}
2015-01-05 11:01:29 -06:00
if (write(f->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;
}
2015-01-05 11:01:29 -06:00
f->curpos = f->startpos;
2010-08-19 15:38:45 -05:00
return 1;
}
2015-01-06 04:41:24 -06:00
static int
afile_aiff_readcomm(struct afile *f, unsigned int csize, unsigned int *nfr)
{
struct aiff_comm comm;
2015-01-06 05:49:08 -06:00
unsigned int nch, rate, bits, bps;
2015-01-06 04:41:24 -06:00
unsigned int e, m;
2015-01-06 05:49:08 -06:00
if (csize == AIFF_COMM_BASESZ) {
/* nothing */
} else if (csize >= sizeof(struct aiff_comm)) {
csize = sizeof(struct aiff_comm);
} else {
2015-01-06 04:41:24 -06:00
log_putu(csize);
log_puts(": bugus comm chunk size\n");
return 0;
}
if (read(f->fd, &comm, csize) != csize) {
log_puts("failed to read .aiff comm chunk\n");
return 0;
}
nch = be16_get(&comm.nch);
if (nch == 0) {
log_puts("zero number of channels in .aiff file\n");
return 0;
}
e = be16_get(&comm.rate_ex);
m = be32_get(&comm.rate_hi);
if (e < 0x3fff || e > 0x3fff + 31) {
log_puts(f->path);
log_puts(": bad sample rate in .aiff file\n");
return 0;
}
rate = m >> (0x3fff + 31 - e);
if (rate < RATE_MIN || rate > RATE_MAX) {
log_putu(rate);
log_puts(": sample rate out of range in .aiff file\n");
return 0;
}
bits = be16_get(&comm.bits);
if (bits < BITS_MIN || bits > BITS_MAX) {
log_putu(bits);
log_puts(": bad number of bits in .aiff file\n");
return 0;
}
2015-01-06 05:49:08 -06:00
if (csize == sizeof(struct aiff_comm)) {
if (memcmp(comm.comp_id, "NONE", 4) == 0) {
f->enc = ENC_PCM;
} else if (memcmp(comm.comp_id, "fl32", 4) == 0) {
f->enc = ENC_FLOAT;
} else if (memcmp(comm.comp_id, "ulaw", 4) == 0) {
f->enc = ENC_ULAW;
} else if (memcmp(comm.comp_id, "alaw", 4) == 0) {
f->enc = ENC_ALAW;
} else {
log_puts("unsupported encoding of .aiff file\n");
return 0;
}
} else
f->enc = ENC_PCM;
switch (f->enc) {
case ENC_PCM:
f->par.bps = (bits + 7) / 8;
f->par.bits = bits;
f->par.le = 0;
f->par.sig = 1;
f->par.msb = 1;
break;
case ENC_ALAW:
case ENC_ULAW:
if (bits != 8) {
log_puts("mulaw/alaw encoding not 8-bit\n");
return 0;
}
f->par.bits = 8;
f->par.bps = 1;
f->par.le = ADATA_LE;
f->par.sig = 0;
f->par.msb = 0;
break;
case ENC_FLOAT:
if (bits != 32) {
log_puts("only 32-bit float supported\n");
return 0;
}
f->par.bits = 32;
f->par.bps = 4;
f->par.le = 0;
f->par.sig = 0;
f->par.msb = 0;
break;
default:
log_putu(f->enc);
log_puts(": unsupported encoding in .wav file\n");
return 0;
}
2015-01-06 04:41:24 -06:00
f->nch = nch;
*nfr = be32_get(&comm.nfr);
return 1;
}
static int
afile_aiff_readhdr(struct afile *f)
{
struct aiff_form form;
struct aiff_chunk chunk;
unsigned int csize, rsize, nfr, pos = 0;
int comm_done = 0;
if (lseek(f->fd, 0, SEEK_SET) < 0) {
log_puts("failed to seek to beginning of .aiff file\n");
return 0;
}
if (read(f->fd, &form, sizeof(form)) != sizeof(form)) {
log_puts("failed to read .aiff file form header\n");
return 0;
}
if (memcmp(&form.magic, &aiff_id_form, 4) != 0 ||
2015-01-06 05:49:08 -06:00
(memcmp(&form.type, &aiff_id_aiff, 4) != 0 &&
memcmp(&form.type, &aiff_id_aifc, 4) != 0)) {
2015-01-06 04:41:24 -06:00
log_puts("not a .aiff file\n");
return 0;
}
rsize = be32_get(&form.size);
for (;;) {
if (pos + sizeof(struct aiff_chunk) > rsize) {
log_puts("missing data chunk in .aiff file\n");
return 0;
}
if (read(f->fd, &chunk, sizeof(chunk)) != sizeof(chunk)) {
log_puts("failed to read .aiff chunk header\n");
return 0;
}
csize = be32_get(&chunk.size);
if (memcmp(chunk.id, aiff_id_comm, 4) == 0) {
if (!afile_aiff_readcomm(f, csize, &nfr))
return 0;
comm_done = 1;
} else if (memcmp(chunk.id, aiff_id_data, 4) == 0) {
2015-01-06 05:52:10 -06:00
f->startpos = pos + sizeof(form) + sizeof(chunk) +
sizeof(struct aiff_data);
/* XXX: parse offset */
2015-01-06 04:41:24 -06:00
break;
} else {
#ifdef DEBUG
if (log_level >= 2)
log_puts("skipped unknown .aiff file chunk\n");
#endif
}
/*
* next chunk
*/
pos += sizeof(struct aiff_chunk) + csize;
if (lseek(f->fd, sizeof(form) + pos, SEEK_SET) < 0) {
log_puts("filed to seek to chunk in .aiff file\n");
return 0;
}
}
if (!comm_done) {
log_puts("missing comm chunk in .wav file\n");
return 0;
}
f->endpos = f->startpos + f->par.bps * f->nch * nfr;
return 1;
}
/*
* Write header and seek to start position
*/
static int
afile_aiff_writehdr(struct afile *f)
{
struct aiff_hdr hdr;
unsigned int bpf;
unsigned int e, m;
/* convert rate to 80-bit float (exponent and fraction part) */
m = f->rate;
e = 0x3fff + 31;
while ((m & 0x80000000) == 0) {
e--;
m <<= 1;
}
/* bytes per frame */
bpf = f->nch * f->par.bps;
memset(&hdr, 0, sizeof(struct aiff_hdr));
memcpy(hdr.form.magic, aiff_id_form, 4);
memcpy(hdr.form.type, aiff_id_aiff, 4);
be32_set(&hdr.form.size, f->endpos - sizeof(hdr.form));
memcpy(hdr.comm_hdr.id, aiff_id_comm, 4);
be32_set(&hdr.comm_hdr.size, sizeof(hdr.comm));
be16_set(&hdr.comm.nch, f->nch);
be16_set(&hdr.comm.bits, f->par.bits);
be16_set(&hdr.comm.rate_ex, e);
be32_set(&hdr.comm.rate_hi, m);
be32_set(&hdr.comm.rate_lo, 0);
be32_set(&hdr.comm.nfr, (f->endpos - f->startpos) / bpf);
2015-01-06 05:49:08 -06:00
memcpy(&hdr.comm.comp_id, "NONE", 4);
2015-01-06 04:41:24 -06:00
memcpy(hdr.data_hdr.id, aiff_id_data, 4);
be32_set(&hdr.data_hdr.size, f->endpos - f->startpos);
be32_set(&hdr.data.offs, 0);
be32_set(&hdr.data.blksz, 0);
if (lseek(f->fd, 0, SEEK_SET) < 0) {
log_puts("failed to seek back to .aiff file header\n");
return 0;
}
if (write(f->fd, &hdr, sizeof(hdr)) != sizeof(hdr)) {
log_puts("failed to write .aiff file header\n");
return 0;
}
f->curpos = f->startpos;
return 1;
}
size_t
2015-01-05 11:01:29 -06:00
afile_read(struct afile *f, void *data, size_t count)
2010-08-19 15:38:45 -05:00
{
off_t maxread;
ssize_t n;
2014-12-30 11:45:05 -06:00
2015-01-05 11:01:29 -06:00
if (f->endpos >= 0) {
maxread = f->endpos - f->curpos;
if (maxread == 0) {
2010-08-19 15:38:45 -05:00
#ifdef DEBUG
if (log_level >= 3) {
2015-01-05 11:01:29 -06:00
log_puts(f->path);
log_puts(": end reached\n");
}
2010-08-19 15:38:45 -05:00
#endif
return 0;
}
if (count > maxread)
count = maxread;
}
2015-01-05 11:01:29 -06:00
n = read(f->fd, data, count);
if (n < 0) {
2015-01-05 11:01:29 -06:00
log_puts(f->path);
log_puts(": couldn't read\n");
return 0;
2010-08-19 15:38:45 -05:00
}
2015-01-05 11:01:29 -06:00
f->curpos += n;
return n;
2010-08-19 15:38:45 -05:00
}
size_t
2015-01-05 11:01:29 -06:00
afile_write(struct afile *f, void *data, size_t count)
2010-08-19 15:38:45 -05:00
{
off_t maxwrite;
int n;
2010-08-19 15:38:45 -05:00
2015-01-05 11:01:29 -06:00
if (f->maxpos >= 0) {
maxwrite = f->maxpos - f->curpos;
if (maxwrite == 0) {
2010-08-19 15:38:45 -05:00
#ifdef DEBUG
if (log_level >= 3) {
2015-01-05 11:01:29 -06:00
log_puts(f->path);
log_puts(": max file size reached\n");
}
#endif
return 0;
2010-08-19 15:38:45 -05:00
}
if (count > maxwrite)
count = maxwrite;
2010-08-19 15:38:45 -05:00
}
2015-01-05 11:01:29 -06:00
n = write(f->fd, data, count);
if (n < 0) {
2015-01-05 11:01:29 -06:00
log_puts(f->path);
log_puts(": couldn't write\n");
return 0;
2010-08-19 15:38:45 -05:00
}
2015-01-05 11:01:29 -06:00
f->curpos += n;
if (f->endpos < f->curpos)
f->endpos = f->curpos;
return n;
2010-08-19 15:38:45 -05:00
}
int
2015-01-05 11:01:29 -06:00
afile_seek(struct afile *f, off_t pos)
2010-08-19 15:38:45 -05:00
{
2015-01-05 11:01:29 -06:00
pos += f->startpos;
if (f->endpos >= 0 && pos > f->endpos) {
log_puts(f->path);
log_puts(": attempt to seek ouside file boundaries\n");
return 0;
}
2010-08-19 15:38:45 -05:00
/*
* seek only if needed to avoid errors with pipes & sockets
*/
2015-01-05 11:01:29 -06:00
if (pos != f->curpos) {
if (lseek(f->fd, pos, SEEK_SET) < 0) {
log_puts(f->path);
log_puts(": couldn't seek\n");
return 0;
}
2015-01-05 11:01:29 -06:00
f->curpos = pos;
2010-08-19 15:38:45 -05:00
}
return 1;
2010-08-19 15:38:45 -05:00
}
void
2015-01-05 11:01:29 -06:00
afile_close(struct afile *f)
2010-08-19 15:38:45 -05:00
{
2015-01-05 11:01:29 -06:00
if (f->flags & WAV_FWRITE) {
if (f->hdr == HDR_WAV)
afile_wav_writehdr(f);
2015-01-06 04:41:24 -06:00
else if (f->hdr == HDR_AIFF)
afile_aiff_writehdr(f);
2010-08-19 15:38:45 -05:00
}
2015-01-05 11:01:29 -06:00
close(f->fd);
2010-08-19 15:38:45 -05:00
}
int
2015-01-05 11:01:29 -06:00
afile_open(struct afile *f, char *path, int hdr, int flags,
struct aparams *par, int rate, int nch)
{
char *ext;
struct wavhdr dummy;
2015-01-05 11:01:29 -06:00
f->par = *par;
f->rate = rate;
f->nch = nch;
f->flags = flags;
f->hdr = hdr;
if (hdr == HDR_AUTO) {
2015-01-05 11:01:29 -06:00
f->hdr = HDR_RAW;
ext = strrchr(path, '.');
if (ext != NULL) {
ext++;
2015-01-06 04:41:24 -06:00
if (strcasecmp(ext, "aif") == 0 ||
2015-01-06 05:49:08 -06:00
strcasecmp(ext, "aiff") == 0 ||
strcasecmp(ext, "aifc") == 0)
2015-01-06 04:41:24 -06:00
f->hdr = HDR_AIFF;
else if (strcasecmp(ext, "wav") == 0)
2015-01-05 11:01:29 -06:00
f->hdr = HDR_WAV;
}
}
2015-01-05 11:01:29 -06:00
if (f->flags == WAV_FREAD) {
if (strcmp(path, "-") == 0) {
2015-01-05 11:01:29 -06:00
f->path = "stdin";
f->fd = STDIN_FILENO;
} else {
2015-01-05 11:01:29 -06:00
f->path = path;
f->fd = open(f->path, O_RDONLY, 0);
if (f->fd < 0) {
log_puts(f->path);
log_puts(": failed to open for reading\n");
return 0;
}
2010-08-19 15:38:45 -05:00
}
2015-01-05 11:01:29 -06:00
if (f->hdr == HDR_WAV) {
if (!afile_wav_readhdr(f))
goto bad_close;
2015-01-06 04:41:24 -06:00
} else if (f->hdr == HDR_AIFF) {
if (!afile_aiff_readhdr(f))
goto bad_close;
} else {
2015-01-05 11:01:29 -06:00
f->startpos = 0;
f->endpos = -1; /* read until EOF */
f->enc = ENC_PCM;
2010-08-19 15:38:45 -05:00
}
2015-01-05 11:01:29 -06:00
f->curpos = f->startpos;
} else if (flags == WAV_FWRITE) {
if (strcmp(path, "-") == 0) {
2015-01-05 11:01:29 -06:00
f->path = "stdout";
f->fd = STDOUT_FILENO;
} else {
2015-01-05 11:01:29 -06:00
f->path = path;
f->fd = open(f->path, O_WRONLY | O_TRUNC | O_CREAT, 0666);
if (f->fd < 0) {
log_puts(f->path);
log_puts(": failed to create file\n");
return 0;
2010-08-19 15:38:45 -05:00
}
}
2015-01-05 11:01:29 -06:00
if (f->hdr == HDR_WAV) {
f->par.bps = (f->par.bits + 7) >> 3;
if (f->par.bits > 8) {
f->par.le = 1;
f->par.sig = 1;
} else
2015-01-05 11:01:29 -06:00
f->par.sig = 0;
if (f->par.bits & 7)
f->par.msb = 1;
f->endpos = f->startpos = sizeof(struct wavhdr);
f->maxpos = WAV_MAXPOS;
memset(&dummy, 0xd0, sizeof(struct wavhdr));
2015-01-05 11:01:29 -06:00
if (write(f->fd, &dummy, sizeof(struct wavhdr)) < 0) {
log_puts(f->path);
log_puts(": failed reserve space for .wav header\n");
goto bad_close;
}
2015-01-06 04:41:24 -06:00
} else if (f->hdr == HDR_AIFF) {
f->par.bps = (f->par.bits + 7) >> 3;
if (f->par.bps > 1)
f->par.le = 0;
f->par.sig = 1;
if (f->par.bits & 7)
f->par.msb = 1;
f->endpos = f->startpos = sizeof(struct aiff_hdr);
f->maxpos = WAV_MAXPOS;
memset(&dummy, 0xd0, sizeof(struct aiff_hdr));
if (write(f->fd, &dummy, sizeof(struct aiff_hdr)) < 0) {
log_puts(f->path);
log_puts(": failed reserve space for .aiff header\n");
goto bad_close;
}
} else {
2015-01-05 11:01:29 -06:00
f->endpos = f->startpos = 0;
f->maxpos = -1;
2010-08-19 15:38:45 -05:00
}
2015-01-05 11:01:29 -06:00
f->curpos = f->startpos;
} else {
2010-08-19 15:38:45 -05:00
#ifdef DEBUG
2015-01-05 11:01:29 -06:00
log_puts("afile_open: wrong flags\n");
panic();
2010-08-19 15:38:45 -05:00
#endif
}
return 1;
bad_close:
2015-01-05 11:01:29 -06:00
close(f->fd);
return 0;
2010-08-19 15:38:45 -05:00
}