ref: 44dc2abe48eaded6e27f4a1ae92fa2666d38993b
dir: /src/stio.c/
#include "st.h"
#include "st_i.h"
#include "stconfig.h"
#ifdef HAVE_STRING_H
#include <string.h>
#endif
#ifdef HAVE_ERRNO_H
#include <errno.h>
#endif
#include <stdlib.h>
#include <sys/types.h> /* for fstat() */
#include <sys/stat.h> /* for fstat() */
#ifdef _MSC_VER
/*
* __STDC__ is defined, so these symbols aren't created.
*/
#define S_IFMT _S_IFMT
#define S_IFREG _S_IFREG
#define fstat _fstat
#endif
/* Based from zlib's minigzip: */
#if defined(WIN32) || defined(__NT__)
# include <fcntl.h>
# include <io.h>
#ifndef O_BINARY
#define O_BINARY _O_BINARY
#endif
# define SET_BINARY_MODE(file) setmode(fileno(file), O_BINARY)
#else
# define SET_BINARY_MODE(file)
#endif
static int is_seekable(ft_t ft)
{
struct stat st;
fstat(fileno(ft->fp), &st);
return ((st.st_mode & S_IFMT) == S_IFREG);
}
/* check that all settings have been given */
static int st_checkformat(ft_t ft)
{
ft->st_errno = ST_SUCCESS;
if (ft->info.rate == 0)
{
st_fail_errno(ft,ST_EFMT,"sampling rate was not specified");
return ST_EOF;
}
if (ft->info.size == -1)
{
st_fail_errno(ft,ST_EFMT,"data size was not specified");
return ST_EOF;
}
if (ft->info.encoding == -1)
{
st_fail_errno(ft,ST_EFMT,"data encoding was not specified");
return ST_EOF;
}
if ((ft->info.size <= 0) || (ft->info.size > ST_INFO_SIZE_MAX))
{
st_fail_errno(ft,ST_EFMT,"data size %i is invalid");
return ST_EOF;
}
if (ft->info.encoding <= 0 || ft->info.encoding > ST_ENCODING_MAX)
{
st_fail_errno(ft,ST_EFMT,"data encoding %i is invalid");
return ST_EOF;
}
return ST_SUCCESS;
}
ft_t st_open_read(const char *path, const st_signalinfo_t *info,
const char *filetype)
{
ft_t ft;
ft = (ft_t)calloc(sizeof(struct st_soundstream), 1);
if (!ft)
return NULL;
ft->filename = strdup(path);
/* Let auto effect do the work if user is not overriding. */
if (!filetype)
ft->filetype = strdup("auto");
else
ft->filetype = strdup(filetype);
if (!ft->filename || !ft->filetype)
goto input_error;
if (st_gettype(ft) != ST_SUCCESS)
{
st_warn("Unknown input file format for '%s': %s",
ft->filename,
ft->st_errstr);
goto input_error;
}
ft->info.size = -1;
ft->info.encoding = -1;
ft->info.channels = -1;
if (info)
ft->info = *info;
/* FIXME: Remove ft->swap from code */
ft->swap = ft->info.swap;
ft->mode = 'r';
if (!(ft->h->flags & ST_FILE_NOSTDIO))
{
/* Open file handler based on input name. Used stdin file handler
* if the filename is "-"
*/
if (!strcmp(ft->filename, "-"))
{
SET_BINARY_MODE(stdin);
ft->fp = stdin;
}
else if ((ft->fp = fopen(ft->filename, "rb")) == NULL)
{
st_warn("Can't open input file '%s': %s", ft->filename,
strerror(errno));
goto input_error;
}
/* See if this file is seekable or not */
ft->seekable = is_seekable(ft);
}
/* Read and write starters can change their formats. */
if ((*ft->h->startread)(ft) != ST_SUCCESS)
{
st_warn("Failed reading %s: %s", ft->filename, ft->st_errstr);
goto input_error;
}
/* Go a head and assume 1 channel audio if nothing is detected.
* This is because libst usually doesn't set this for mono file
* formats (for historical reasons).
*/
if (ft->info.channels == -1)
ft->info.channels = 1;
if (st_checkformat(ft) )
{
st_warn("bad input format for file %s: %s", ft->filename,
ft->st_errstr);
goto input_error;
}
return ft;
input_error:
if (ft->filename)
free(ft->filename);
if (ft->filetype)
free(ft->filetype);
free(ft);
return NULL;
}
#if defined(DOS) || defined(WIN32)
#define LASTCHAR '\\'
#else
#define LASTCHAR '/'
#endif
ft_t st_open_write_instr(const char *path, const st_signalinfo_t *info,
const char *filetype, const char *comment,
const st_instrinfo_t *instr,
const st_loopinfo_t *loops)
{
ft_t ft;
int i;
ft = (ft_t)calloc(sizeof(struct st_soundstream), 1);
if (!ft )
return NULL;
ft->filename = strdup(path);
/* Let auto effect do the work if user is not overriding. */
if (!filetype)
{
char *chop;
int len;
len = strlen(ft->filename);
/* Use filename extension to determine audio type.
* Search for the last '.' appearing in the filename, same
* as for input files.
*/
chop = ft->filename + len;
while (chop > ft->filename && *chop != LASTCHAR && *chop != '.')
chop--;
if (*chop == '.')
{
chop++;
ft->filetype = strdup(chop);
}
}
else
ft->filetype = strdup(filetype);
if (!ft->filename)
{
st_warn("Unknown output file format for 'NULL'.");
goto output_error;
}
if (!ft->filetype || st_gettype(ft) != ST_SUCCESS)
{
st_warn("Unknown output file format for '%s': %s",
ft->filename,
ft->st_errstr);
goto output_error;
}
ft->info.size = -1;
ft->info.encoding = -1;
ft->info.channels = -1;
if (info)
ft->info = *info;
ft->mode = 'w';
if (!(ft->h->flags & ST_FILE_NOSTDIO))
{
/* Open file handler based on output name. Used stdout file handler
* if the filename is "-"
*/
if (!strcmp(ft->filename, "-"))
{
SET_BINARY_MODE(stdout);
ft->fp = stdout;
}
else if ((ft->fp = fopen(ft->filename, "wb")) == NULL)
{
st_warn("Can't open output file '%s': %s", ft->filename,
strerror(errno));
goto output_error;
}
/* stdout tends to be line-buffered. Override this */
/* to be Full Buffering. */
/* FIXME: Use buffer size from ft structure */
if (setvbuf (ft->fp, NULL, _IOFBF, sizeof(char)*ST_BUFSIZ))
{
st_warn("Can't set write buffer");
goto output_error;
}
/* See if this file is seekable or not */
ft->seekable = is_seekable(ft);
}
if (ft->comment == NULL && comment != NULL)
ft->comment = strdup(comment);
else
ft->comment = strdup("Processed by SoX");
if (loops)
{
for (i = 0; i < ST_MAX_NLOOPS; i++)
{
ft->loops[i] = loops[i];
}
}
/* leave SMPTE # alone since it's absolute */
if (instr)
ft->instr = *instr;
/* FIXME: Remove ft->swap from code */
ft->swap = ft->info.swap;
/* Read and write starters can change their formats. */
if ((*ft->h->startwrite)(ft) != ST_SUCCESS)
{
st_warn("Failed writing %s: %s", ft->filename, ft->st_errstr);
goto output_error;
}
if (st_checkformat(ft) )
{
st_warn("bad output format for file %s: %s", ft->filename,
ft->st_errstr);
goto output_error;
}
return ft;
output_error:
if (ft->filename)
free(ft->filename);
if (ft->filetype)
free(ft->filetype);
free(ft);
return NULL;
}
ft_t st_open_write(const char *path, const st_signalinfo_t *info,
const char *filetype, const char *comment)
{
return st_open_write_instr(path, info, filetype, comment, NULL, NULL);
}
st_ssize_t st_read(ft_t ft, st_sample_t *buf, st_ssize_t len)
{
return (*ft->h->read)(ft, buf, len);
}
st_ssize_t st_write(ft_t ft, st_sample_t *buf, st_ssize_t len)
{
return (*ft->h->write)(ft, buf, len);
}
int st_close(ft_t ft)
{
int rc;
if (ft->mode == 'r')
rc = (*ft->h->stopread)(ft);
else
rc = (*ft->h->stopwrite)(ft);
if (!(ft->h->flags & ST_FILE_NOSTDIO))
{
fclose(ft->fp);
}
if (ft->filename)
free(ft->filename);
if (ft->filetype)
free(ft->filetype);
/* Currently, since startread() mallocs comments, stopread
* is expected to also free it.
*/
if (ft->mode == 'w' && ft->comment)
free(ft->comment);
return rc;
}
int st_seek(ft_t ft, st_size_t offset, int whence)
{
/* One day, ST_SEEK_CUR and ST_SEEK_END should be impelemented */
if (whence != ST_SEEK_SET)
return ST_EOF;
/* FIXME: Should return this */
/* return ST_EINVAL; */
/* If file is a seekable file and this handler supports seeking,
* the invoke handlers function.
*/
if (ft->seekable && (ft->h->flags & ST_FILE_SEEK))
return (*ft->h->seek)(ft, offset);
else
return ST_EOF;
/* FIXME: Should return this */
/* return ST_EBADF; */
}